1 files changed, 161 insertions, 0 deletions
diff --git a/dom/base/StorageAccessPermissionRequest.cpp b/dom/base/StorageAccessPermissionRequest.cpp
new file mode 100644
index 0000000000..094cd87393
--- /dev/null
+++ b/dom/base/StorageAccessPermissionRequest.cpp
@@ -0,0 +1,161 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim:set ts=2 sw=2 sts=2 et cindent: */
+/* 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 "StorageAccessPermissionRequest.h"
+#include "nsGlobalWindowInner.h"
+#include "mozilla/StaticPrefs_dom.h"
+#include <cstdlib>
+
+namespace mozilla::dom {
+
+NS_IMPL_CYCLE_COLLECTION_INHERITED(StorageAccessPermissionRequest,
+                                   ContentPermissionRequestBase)
+
+NS_IMPL_ISUPPORTS_CYCLE_COLLECTION_INHERITED_0(StorageAccessPermissionRequest,
+                                               ContentPermissionRequestBase)
+
+StorageAccessPermissionRequest::StorageAccessPermissionRequest(
+    nsPIDOMWindowInner* aWindow, nsIPrincipal* aNodePrincipal,
+    const Maybe<nsCString>& aTopLevelBaseDomain, AllowCallback&& aAllowCallback,
+    CancelCallback&& aCancelCallback)
+    : ContentPermissionRequestBase(aNodePrincipal, aWindow,
+                                   "dom.storage_access"_ns,
+                                   "storage-access"_ns),
+      mAllowCallback(std::move(aAllowCallback)),
+      mCancelCallback(std::move(aCancelCallback)),
+      mCallbackCalled(false) {
+  if (aTopLevelBaseDomain.isSome()) {
+    mOptions.AppendElement(NS_ConvertUTF8toUTF16(aTopLevelBaseDomain.value()));
+  }
+  mPermissionRequests.AppendElement(PermissionRequest(mType, mOptions));
+}
+
+NS_IMETHODIMP
+StorageAccessPermissionRequest::Cancel() {
+  if (!mCallbackCalled) {
+    mCallbackCalled = true;
+    mCancelCallback();
+  }
+  return NS_OK;
+}
+
+NS_IMETHODIMP
+StorageAccessPermissionRequest::Allow(JS::Handle<JS::Value> aChoices) {
+  nsTArray<PermissionChoice> choices;
+  nsresult rv = TranslateChoices(aChoices, mPermissionRequests, choices);
+  if (NS_FAILED(rv)) {
+    return rv;
+  }
+
+  // There is no support to allow grants automatically from the prompting code
+  // path.
+
+  if (!mCallbackCalled) {
+    mCallbackCalled = true;
+    if (choices.Length() == 1 && choices[0].choice().EqualsLiteral("allow")) {
+      mAllowCallback();
+    }
+  }
+  return NS_OK;
+}
+
+NS_IMETHODIMP
+StorageAccessPermissionRequest::GetTypes(nsIArray** aTypes) {
+  return nsContentPermissionUtils::CreatePermissionArray(mType, mOptions,
+                                                         aTypes);
+}
+
+RefPtr<StorageAccessPermissionRequest::AutoGrantDelayPromise>
+StorageAccessPermissionRequest::MaybeDelayAutomaticGrants() {
+  RefPtr<AutoGrantDelayPromise::Private> p =
+      new AutoGrantDelayPromise::Private(__func__);
+
+  unsigned simulatedDelay = CalculateSimulatedDelay();
+  if (simulatedDelay) {
+    nsCOMPtr<nsITimer> timer;
+    RefPtr<AutoGrantDelayPromise::Private> promise = p;
+    nsresult rv = NS_NewTimerWithFuncCallback(
+        getter_AddRefs(timer),
+        [](nsITimer* aTimer, void* aClosure) -> void {
+          auto* promise =
+              static_cast<AutoGrantDelayPromise::Private*>(aClosure);
+          promise->Resolve(true, __func__);
+          NS_RELEASE(aTimer);
+          NS_RELEASE(promise);
+        },
+        promise, simulatedDelay, nsITimer::TYPE_ONE_SHOT,
+        "DelayedAllowAutoGrantCallback");
+    if (NS_WARN_IF(NS_FAILED(rv))) {
+      p->Reject(false, __func__);
+    } else {
+      // Leak the references here! We'll release them inside the callback.
+      Unused << timer.forget();
+      Unused << promise.forget();
+    }
+  } else {
+    p->Resolve(false, __func__);
+  }
+  return p;
+}
+
+already_AddRefed<StorageAccessPermissionRequest>
+StorageAccessPermissionRequest::Create(nsPIDOMWindowInner* aWindow,
+                                       AllowCallback&& aAllowCallback,
+                                       CancelCallback&& aCancelCallback) {
+  if (!aWindow) {
+    return nullptr;
+  }
+  nsGlobalWindowInner* win = nsGlobalWindowInner::Cast(aWindow);
+
+  return Create(aWindow, win->GetPrincipal(), std::move(aAllowCallback),
+                std::move(aCancelCallback));
+}
+
+already_AddRefed<StorageAccessPermissionRequest>
+StorageAccessPermissionRequest::Create(nsPIDOMWindowInner* aWindow,
+                                       nsIPrincipal* aPrincipal,
+                                       AllowCallback&& aAllowCallback,
+                                       CancelCallback&& aCancelCallback) {
+  return Create(aWindow, aPrincipal, Nothing(), std::move(aAllowCallback),
+                std::move(aCancelCallback));
+}
+
+already_AddRefed<StorageAccessPermissionRequest>
+StorageAccessPermissionRequest::Create(
+    nsPIDOMWindowInner* aWindow, nsIPrincipal* aPrincipal,
+    const Maybe<nsCString>& aTopLevelBaseDomain, AllowCallback&& aAllowCallback,
+    CancelCallback&& aCancelCallback) {
+  if (!aWindow) {
+    return nullptr;
+  }
+
+  if (!aPrincipal) {
+    return nullptr;
+  }
+
+  RefPtr<StorageAccessPermissionRequest> request =
+      new StorageAccessPermissionRequest(
+          aWindow, aPrincipal, aTopLevelBaseDomain, std::move(aAllowCallback),
+          std::move(aCancelCallback));
+  return request.forget();
+}
+
+unsigned StorageAccessPermissionRequest::CalculateSimulatedDelay() {
+  if (!StaticPrefs::dom_storage_access_auto_grants_delayed()) {
+    return 0;
+  }
+
+  // Generate a random time value that is at least 0 and and most 3 seconds.
+  std::srand(static_cast<unsigned>(PR_Now()));
+
+  const unsigned kMin = 0;
+  const unsigned kMax = 3000;
+  const unsigned random = std::abs(std::rand());
+
+  return kMin + random % (kMax - kMin);
+}
+
+}  // namespace mozilla::dom