summaryrefslogtreecommitdiffstats
path: root/dom/url/URLMainThread.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'dom/url/URLMainThread.cpp')
-rw-r--r--dom/url/URLMainThread.cpp133
1 files changed, 133 insertions, 0 deletions
diff --git a/dom/url/URLMainThread.cpp b/dom/url/URLMainThread.cpp
new file mode 100644
index 0000000000..95afd0db45
--- /dev/null
+++ b/dom/url/URLMainThread.cpp
@@ -0,0 +1,133 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "URLMainThread.h"
+
+#include "mozilla/dom/BindingUtils.h"
+#include "mozilla/dom/Blob.h"
+#include "mozilla/dom/BlobURLProtocolHandler.h"
+#include "mozilla/Unused.h"
+#include "nsContentUtils.h"
+#include "nsNetUtil.h"
+#include "nsThreadUtils.h"
+#include "mozilla/dom/Document.h"
+
+namespace mozilla::dom {
+
+/* static */
+void URLMainThread::CreateObjectURL(const GlobalObject& aGlobal, Blob& aBlob,
+ nsAString& aResult, ErrorResult& aRv) {
+ MOZ_ASSERT(NS_IsMainThread());
+
+ nsCOMPtr<nsIGlobalObject> global = do_QueryInterface(aGlobal.GetAsSupports());
+ if (NS_WARN_IF(!global)) {
+ aRv.Throw(NS_ERROR_FAILURE);
+ return;
+ }
+
+ nsAutoString partKey;
+ if (nsCOMPtr<nsPIDOMWindowInner> owner = do_QueryInterface(global)) {
+ if (Document* doc = owner->GetExtantDoc()) {
+ nsCOMPtr<nsICookieJarSettings> cookieJarSettings =
+ doc->CookieJarSettings();
+
+ cookieJarSettings->GetPartitionKey(partKey);
+ }
+ }
+
+ nsCOMPtr<nsIPrincipal> principal =
+ nsContentUtils::ObjectPrincipal(aGlobal.Get());
+
+ nsAutoCString url;
+ aRv = BlobURLProtocolHandler::AddDataEntry(
+ aBlob.Impl(), principal, NS_ConvertUTF16toUTF8(partKey), url);
+ if (NS_WARN_IF(aRv.Failed())) {
+ return;
+ }
+
+ global->RegisterHostObjectURI(url);
+ CopyASCIItoUTF16(url, aResult);
+}
+
+/* static */
+void URLMainThread::CreateObjectURL(const GlobalObject& aGlobal,
+ MediaSource& aSource, nsAString& aResult,
+ ErrorResult& aRv) {
+ MOZ_ASSERT(NS_IsMainThread());
+
+ nsCOMPtr<nsIGlobalObject> global = do_QueryInterface(aGlobal.GetAsSupports());
+ if (NS_WARN_IF(!global)) {
+ aRv.Throw(NS_ERROR_FAILURE);
+ return;
+ }
+
+ nsAutoString partKey;
+ if (nsCOMPtr<nsPIDOMWindowInner> owner = do_QueryInterface(global)) {
+ if (Document* doc = owner->GetExtantDoc()) {
+ nsCOMPtr<nsICookieJarSettings> cookieJarSettings =
+ doc->CookieJarSettings();
+
+ cookieJarSettings->GetPartitionKey(partKey);
+ }
+ }
+
+ nsCOMPtr<nsIPrincipal> principal =
+ nsContentUtils::ObjectPrincipal(aGlobal.Get());
+
+ nsAutoCString url;
+ aRv = BlobURLProtocolHandler::AddDataEntry(
+ &aSource, principal, NS_ConvertUTF16toUTF8(partKey), url);
+ if (NS_WARN_IF(aRv.Failed())) {
+ return;
+ }
+
+ nsCOMPtr<nsIRunnable> revocation = NS_NewRunnableFunction(
+ "dom::URLMainThread::CreateObjectURL",
+ [url] { BlobURLProtocolHandler::RemoveDataEntry(url); });
+
+ nsContentUtils::RunInStableState(revocation.forget());
+
+ CopyASCIItoUTF16(url, aResult);
+}
+
+/* static */
+void URLMainThread::RevokeObjectURL(const GlobalObject& aGlobal,
+ const nsAString& aURL, ErrorResult& aRv) {
+ MOZ_ASSERT(NS_IsMainThread());
+ nsCOMPtr<nsIGlobalObject> global = do_QueryInterface(aGlobal.GetAsSupports());
+ if (!global) {
+ aRv.Throw(NS_ERROR_FAILURE);
+ return;
+ }
+
+ nsAutoString partKey;
+ if (nsCOMPtr<nsPIDOMWindowInner> owner = do_QueryInterface(global)) {
+ if (Document* doc = owner->GetExtantDoc()) {
+ nsCOMPtr<nsICookieJarSettings> cookieJarSettings =
+ doc->CookieJarSettings();
+
+ cookieJarSettings->GetPartitionKey(partKey);
+ }
+ }
+
+ NS_LossyConvertUTF16toASCII asciiurl(aURL);
+
+ if (BlobURLProtocolHandler::RemoveDataEntry(
+ asciiurl, nsContentUtils::ObjectPrincipal(aGlobal.Get()),
+ NS_ConvertUTF16toUTF8(partKey))) {
+ global->UnregisterHostObjectURI(asciiurl);
+ }
+}
+
+/* static */
+bool URLMainThread::IsValidObjectURL(const GlobalObject& aGlobal,
+ const nsAString& aURL, ErrorResult& aRv) {
+ MOZ_ASSERT(NS_IsMainThread());
+ NS_LossyConvertUTF16toASCII asciiurl(aURL);
+ return BlobURLProtocolHandler::HasDataEntry(asciiurl);
+}
+
+} // namespace mozilla::dom