Source code
Revision control
Copy as Markdown
Other Tools
/* -*- 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
#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, bool aFrameOnly,
AllowCallback&& aAllowCallback, CancelCallback&& aCancelCallback)
: ContentPermissionRequestBase(aNodePrincipal, aWindow,
"dom.storage_access"_ns,
"storage-access"_ns),
mAllowCallback(std::move(aAllowCallback)),
mCancelCallback(std::move(aCancelCallback)),
mCallbackCalled(false) {
mOptions.SetLength(2);
if (aTopLevelBaseDomain.isSome()) {
nsCString option = aTopLevelBaseDomain.value();
mOptions.ElementAt(0) = NS_ConvertUTF8toUTF16(option);
}
if (aFrameOnly) {
mOptions.ElementAt(1) = u"1"_ns;
}
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(), true, std::move(aAllowCallback),
std::move(aCancelCallback));
}
already_AddRefed<StorageAccessPermissionRequest>
StorageAccessPermissionRequest::Create(
nsPIDOMWindowInner* aWindow, nsIPrincipal* aPrincipal,
const Maybe<nsCString>& aTopLevelBaseDomain, bool aFrameOnly,
AllowCallback&& aAllowCallback, CancelCallback&& aCancelCallback) {
if (!aWindow) {
return nullptr;
}
if (!aPrincipal) {
return nullptr;
}
RefPtr<StorageAccessPermissionRequest> request =
new StorageAccessPermissionRequest(
aWindow, aPrincipal, aTopLevelBaseDomain, aFrameOnly,
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