mirror of
https://gitlab.winehq.org/wine/wine-gecko.git
synced 2024-09-13 09:24:08 -07:00
2207 lines
66 KiB
C++
2207 lines
66 KiB
C++
/* 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 "ServiceWorkerManager.h"
|
|
|
|
#include "nsIDOMEventTarget.h"
|
|
#include "nsIDocument.h"
|
|
#include "nsIScriptSecurityManager.h"
|
|
#include "nsPIDOMWindow.h"
|
|
|
|
#include "jsapi.h"
|
|
|
|
#include "mozilla/dom/BindingUtils.h"
|
|
#include "mozilla/dom/DOMError.h"
|
|
#include "mozilla/dom/ErrorEvent.h"
|
|
#include "mozilla/dom/InstallEventBinding.h"
|
|
#include "mozilla/dom/PromiseNativeHandler.h"
|
|
|
|
#include "nsContentUtils.h"
|
|
#include "nsNetUtil.h"
|
|
#include "nsProxyRelease.h"
|
|
#include "nsTArray.h"
|
|
|
|
#include "RuntimeService.h"
|
|
#include "ServiceWorker.h"
|
|
#include "ServiceWorkerClient.h"
|
|
#include "ServiceWorkerRegistration.h"
|
|
#include "ServiceWorkerEvents.h"
|
|
#include "WorkerInlines.h"
|
|
#include "WorkerPrivate.h"
|
|
#include "WorkerRunnable.h"
|
|
#include "WorkerScope.h"
|
|
|
|
using namespace mozilla;
|
|
using namespace mozilla::dom;
|
|
|
|
BEGIN_WORKERS_NAMESPACE
|
|
|
|
NS_IMPL_ISUPPORTS0(ServiceWorkerRegistrationInfo)
|
|
|
|
UpdatePromise::UpdatePromise()
|
|
: mState(Pending)
|
|
{
|
|
MOZ_COUNT_CTOR(UpdatePromise);
|
|
}
|
|
|
|
UpdatePromise::~UpdatePromise()
|
|
{
|
|
MOZ_COUNT_DTOR(UpdatePromise);
|
|
}
|
|
|
|
void
|
|
UpdatePromise::AddPromise(Promise* aPromise)
|
|
{
|
|
MOZ_ASSERT(mState == Pending);
|
|
mPromises.AppendElement(aPromise);
|
|
}
|
|
|
|
void
|
|
UpdatePromise::ResolveAllPromises(const nsACString& aScriptSpec, const nsACString& aScope)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(mState == Pending);
|
|
mState = Resolved;
|
|
RuntimeService* rs = RuntimeService::GetOrCreateService();
|
|
MOZ_ASSERT(rs);
|
|
|
|
nsTArray<WeakPtr<Promise>> array;
|
|
array.SwapElements(mPromises);
|
|
for (uint32_t i = 0; i < array.Length(); ++i) {
|
|
WeakPtr<Promise>& pendingPromise = array.ElementAt(i);
|
|
if (pendingPromise) {
|
|
nsRefPtr<Promise> kungfuDeathGrip = pendingPromise.get();
|
|
nsCOMPtr<nsIGlobalObject> go =
|
|
do_QueryInterface(pendingPromise->GetParentObject());
|
|
MOZ_ASSERT(go);
|
|
|
|
AutoSafeJSContext cx;
|
|
JS::Rooted<JSObject*> global(cx, go->GetGlobalJSObject());
|
|
JSAutoCompartment ac(cx, global);
|
|
|
|
GlobalObject domGlobal(cx, global);
|
|
|
|
// The service worker is created and kept alive as a SharedWorker.
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
nsresult rv = rs->CreateServiceWorker(domGlobal,
|
|
NS_ConvertUTF8toUTF16(aScriptSpec),
|
|
aScope,
|
|
getter_AddRefs(serviceWorker));
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
pendingPromise->MaybeReject(NS_ERROR_DOM_ABORT_ERR);
|
|
continue;
|
|
}
|
|
|
|
// Since ServiceWorkerRegistration is only exposed to windows we can be
|
|
// certain about this cast.
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(go);
|
|
nsRefPtr<ServiceWorkerRegistration> swr =
|
|
new ServiceWorkerRegistration(window, NS_ConvertUTF8toUTF16(aScope));
|
|
|
|
pendingPromise->MaybeResolve(swr);
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
UpdatePromise::RejectAllPromises(nsresult aRv)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(mState == Pending);
|
|
mState = Rejected;
|
|
|
|
nsTArray<WeakPtr<Promise>> array;
|
|
array.SwapElements(mPromises);
|
|
for (uint32_t i = 0; i < array.Length(); ++i) {
|
|
WeakPtr<Promise>& pendingPromise = array.ElementAt(i);
|
|
if (pendingPromise) {
|
|
// Since ServiceWorkerRegistration is only exposed to windows we can be
|
|
// certain about this cast.
|
|
nsCOMPtr<nsPIDOMWindow> window =
|
|
do_QueryInterface(pendingPromise->GetParentObject());
|
|
MOZ_ASSERT(window);
|
|
nsRefPtr<DOMError> domError = new DOMError(window, aRv);
|
|
pendingPromise->MaybeRejectBrokenly(domError);
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
UpdatePromise::RejectAllPromises(const ErrorEventInit& aErrorDesc)
|
|
{
|
|
MOZ_ASSERT(mState == Pending);
|
|
mState = Rejected;
|
|
|
|
nsTArray<WeakPtr<Promise>> array;
|
|
array.SwapElements(mPromises);
|
|
for (uint32_t i = 0; i < array.Length(); ++i) {
|
|
WeakPtr<Promise>& pendingPromise = array.ElementAt(i);
|
|
if (pendingPromise) {
|
|
// Since ServiceWorkerRegistration is only exposed to windows we can be
|
|
// certain about this cast.
|
|
nsCOMPtr<nsIGlobalObject> go = do_QueryInterface(pendingPromise->GetParentObject());
|
|
MOZ_ASSERT(go);
|
|
|
|
AutoJSAPI jsapi;
|
|
jsapi.Init(go);
|
|
|
|
JSContext* cx = jsapi.cx();
|
|
|
|
JS::Rooted<JSString*> stack(cx, JS_GetEmptyString(JS_GetRuntime(cx)));
|
|
|
|
JS::Rooted<JS::Value> fnval(cx);
|
|
if (!ToJSValue(cx, aErrorDesc.mFilename, &fnval)) {
|
|
pendingPromise->MaybeReject(NS_ERROR_FAILURE);
|
|
continue;
|
|
}
|
|
JS::Rooted<JSString*> fn(cx, fnval.toString());
|
|
|
|
JS::Rooted<JS::Value> msgval(cx);
|
|
if (!ToJSValue(cx, aErrorDesc.mMessage, &msgval)) {
|
|
pendingPromise->MaybeReject(NS_ERROR_FAILURE);
|
|
continue;
|
|
}
|
|
JS::Rooted<JSString*> msg(cx, msgval.toString());
|
|
|
|
JS::Rooted<JS::Value> error(cx);
|
|
if (!JS::CreateError(cx, JSEXN_ERR, stack, fn, aErrorDesc.mLineno,
|
|
aErrorDesc.mColno, nullptr, msg, &error)) {
|
|
pendingPromise->MaybeReject(NS_ERROR_FAILURE);
|
|
continue;
|
|
}
|
|
|
|
pendingPromise->MaybeReject(cx, error);
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
ServiceWorkerRegistrationInfo::Clear()
|
|
{
|
|
if (mInstallingWorker) {
|
|
// FIXME(nsm): Terminate installing worker.
|
|
// Bug 1043701 Set state to redundant.
|
|
// Fire statechange.
|
|
mInstallingWorker = nullptr;
|
|
// FIXME(nsm): Abort any inflight requests from installing worker.
|
|
}
|
|
|
|
if (mWaitingWorker) {
|
|
// FIXME(nsm): Bug 1043701 Set state to redundant.
|
|
// Fire statechange.
|
|
mWaitingWorker = nullptr;
|
|
}
|
|
|
|
if (mCurrentWorker) {
|
|
// FIXME(nsm): Bug 1043701 Set state to redundant.
|
|
mCurrentWorker = nullptr;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
MOZ_ASSERT(swm);
|
|
swm->InvalidateServiceWorkerRegistrationWorker(this,
|
|
WhichServiceWorker::INSTALLING_WORKER |
|
|
WhichServiceWorker::WAITING_WORKER |
|
|
WhichServiceWorker::ACTIVE_WORKER);
|
|
}
|
|
|
|
class FinishFetchOnMainThreadRunnable : public nsRunnable
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerUpdateInstance> mUpdateInstance;
|
|
public:
|
|
explicit FinishFetchOnMainThreadRunnable
|
|
(const nsMainThreadPtrHandle<ServiceWorkerUpdateInstance>& aUpdateInstance)
|
|
: mUpdateInstance(aUpdateInstance)
|
|
{ }
|
|
|
|
NS_IMETHOD
|
|
Run() MOZ_OVERRIDE;
|
|
};
|
|
|
|
class FinishSuccessfulFetchWorkerRunnable : public WorkerRunnable
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerUpdateInstance> mUpdateInstance;
|
|
public:
|
|
FinishSuccessfulFetchWorkerRunnable(WorkerPrivate* aWorkerPrivate,
|
|
const nsMainThreadPtrHandle<ServiceWorkerUpdateInstance>& aUpdateInstance)
|
|
: WorkerRunnable(aWorkerPrivate, WorkerThreadModifyBusyCount),
|
|
mUpdateInstance(aUpdateInstance)
|
|
{
|
|
AssertIsOnMainThread();
|
|
}
|
|
|
|
bool
|
|
WorkerRun(JSContext* aCx, WorkerPrivate* aWorkerPrivate)
|
|
{
|
|
aWorkerPrivate->AssertIsOnWorkerThread();
|
|
if (!aWorkerPrivate->WorkerScriptExecutedSuccessfully()) {
|
|
return true;
|
|
}
|
|
|
|
nsRefPtr<FinishFetchOnMainThreadRunnable> r =
|
|
new FinishFetchOnMainThreadRunnable(mUpdateInstance);
|
|
NS_DispatchToMainThread(r);
|
|
return true;
|
|
}
|
|
};
|
|
|
|
// Allows newer calls to Update() to 'abort' older calls.
|
|
// Each call to Update() creates the instance which handles creating the
|
|
// worker and queues up a runnable to resolve the update promise once the
|
|
// worker has successfully been parsed.
|
|
class ServiceWorkerUpdateInstance MOZ_FINAL : public nsISupports
|
|
{
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> mRegistration;
|
|
nsCString mScriptSpec;
|
|
nsCOMPtr<nsPIDOMWindow> mWindow;
|
|
|
|
bool mAborted;
|
|
|
|
~ServiceWorkerUpdateInstance() {}
|
|
|
|
public:
|
|
NS_DECL_ISUPPORTS
|
|
|
|
ServiceWorkerUpdateInstance(ServiceWorkerRegistrationInfo *aRegistration,
|
|
nsPIDOMWindow* aWindow)
|
|
: mRegistration(aRegistration),
|
|
// Capture the current script spec in case register() gets called.
|
|
mScriptSpec(aRegistration->mScriptSpec),
|
|
mWindow(aWindow),
|
|
mAborted(false)
|
|
{
|
|
AssertIsOnMainThread();
|
|
}
|
|
|
|
const nsCString&
|
|
GetScriptSpec() const
|
|
{
|
|
return mScriptSpec;
|
|
}
|
|
|
|
void
|
|
Abort()
|
|
{
|
|
MOZ_ASSERT(!mAborted);
|
|
mAborted = true;
|
|
}
|
|
|
|
void
|
|
Update()
|
|
{
|
|
AssertIsOnMainThread();
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
MOZ_ASSERT(swm);
|
|
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
nsresult rv = swm->CreateServiceWorkerForWindow(mWindow,
|
|
mScriptSpec,
|
|
mRegistration->mScope,
|
|
getter_AddRefs(serviceWorker));
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
swm->RejectUpdatePromiseObservers(mRegistration, rv);
|
|
return;
|
|
}
|
|
|
|
nsMainThreadPtrHandle<ServiceWorkerUpdateInstance> handle(
|
|
new nsMainThreadPtrHolder<ServiceWorkerUpdateInstance>(this));
|
|
// FIXME(nsm): Deal with error case (worker failed to download, redirect,
|
|
// parse) in error handler patch.
|
|
nsRefPtr<FinishSuccessfulFetchWorkerRunnable> r =
|
|
new FinishSuccessfulFetchWorkerRunnable(serviceWorker->GetWorkerPrivate(), handle);
|
|
|
|
AutoSafeJSContext cx;
|
|
if (!r->Dispatch(cx)) {
|
|
swm->RejectUpdatePromiseObservers(mRegistration, NS_ERROR_FAILURE);
|
|
}
|
|
}
|
|
|
|
void
|
|
FetchDone()
|
|
{
|
|
AssertIsOnMainThread();
|
|
if (mAborted) {
|
|
return;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
MOZ_ASSERT(swm);
|
|
swm->FinishFetch(mRegistration, mWindow);
|
|
}
|
|
};
|
|
|
|
NS_IMPL_ISUPPORTS0(ServiceWorkerUpdateInstance)
|
|
|
|
NS_IMETHODIMP
|
|
FinishFetchOnMainThreadRunnable::Run()
|
|
{
|
|
AssertIsOnMainThread();
|
|
mUpdateInstance->FetchDone();
|
|
return NS_OK;
|
|
}
|
|
|
|
ServiceWorkerRegistrationInfo::ServiceWorkerRegistrationInfo(const nsACString& aScope)
|
|
: mControlledDocumentsCounter(0),
|
|
mScope(aScope),
|
|
mPendingUninstall(false)
|
|
{ }
|
|
|
|
ServiceWorkerRegistrationInfo::~ServiceWorkerRegistrationInfo()
|
|
{
|
|
if (IsControllingDocuments()) {
|
|
NS_WARNING("ServiceWorkerRegistrationInfo is still controlling documents. This can be a bug or a leak in ServiceWorker API or in any other API that takes the document alive.");
|
|
}
|
|
}
|
|
|
|
//////////////////////////
|
|
// ServiceWorkerManager //
|
|
//////////////////////////
|
|
|
|
NS_IMPL_ADDREF(ServiceWorkerManager)
|
|
NS_IMPL_RELEASE(ServiceWorkerManager)
|
|
|
|
NS_INTERFACE_MAP_BEGIN(ServiceWorkerManager)
|
|
NS_INTERFACE_MAP_ENTRY(nsIServiceWorkerManager)
|
|
if (aIID.Equals(NS_GET_IID(ServiceWorkerManager)))
|
|
foundInterface = static_cast<nsIServiceWorkerManager*>(this);
|
|
else
|
|
NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIServiceWorkerManager)
|
|
NS_INTERFACE_MAP_END
|
|
|
|
ServiceWorkerManager::ServiceWorkerManager()
|
|
{
|
|
}
|
|
|
|
ServiceWorkerManager::~ServiceWorkerManager()
|
|
{
|
|
// The map will assert if it is not empty when destroyed.
|
|
mDomainMap.EnumerateRead(CleanupServiceWorkerInformation, nullptr);
|
|
mDomainMap.Clear();
|
|
}
|
|
|
|
/* static */ PLDHashOperator
|
|
ServiceWorkerManager::CleanupServiceWorkerInformation(const nsACString& aDomain,
|
|
ServiceWorkerDomainInfo* aDomainInfo,
|
|
void *aUnused)
|
|
{
|
|
aDomainInfo->mServiceWorkerRegistrationInfos.Clear();
|
|
return PL_DHASH_NEXT;
|
|
}
|
|
|
|
/*
|
|
* Implements the async aspects of the register algorithm.
|
|
*/
|
|
class RegisterRunnable : public nsRunnable
|
|
{
|
|
nsCOMPtr<nsPIDOMWindow> mWindow;
|
|
const nsCString mScope;
|
|
nsCOMPtr<nsIURI> mScriptURI;
|
|
nsRefPtr<Promise> mPromise;
|
|
public:
|
|
RegisterRunnable(nsPIDOMWindow* aWindow, const nsCString aScope,
|
|
nsIURI* aScriptURI, Promise* aPromise)
|
|
: mWindow(aWindow), mScope(aScope), mScriptURI(aScriptURI),
|
|
mPromise(aPromise)
|
|
{ }
|
|
|
|
NS_IMETHODIMP
|
|
Run()
|
|
{
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
nsRefPtr<ServiceWorkerManager::ServiceWorkerDomainInfo> domainInfo = swm->GetDomainInfo(mScriptURI);
|
|
if (!domainInfo) {
|
|
nsCString domain;
|
|
nsresult rv = mScriptURI->GetHost(domain);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
mPromise->MaybeReject(rv);
|
|
return NS_OK;
|
|
}
|
|
|
|
domainInfo = new ServiceWorkerManager::ServiceWorkerDomainInfo;
|
|
swm->mDomainMap.Put(domain, domainInfo);
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration =
|
|
domainInfo->GetRegistration(mScope);
|
|
|
|
nsCString spec;
|
|
nsresult rv = mScriptURI->GetSpec(spec);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
mPromise->MaybeReject(rv);
|
|
return NS_OK;
|
|
}
|
|
|
|
if (registration) {
|
|
registration->mPendingUninstall = false;
|
|
if (spec.Equals(registration->mScriptSpec)) {
|
|
// There is an existing update in progress. Resolve with whatever it
|
|
// results in.
|
|
if (registration->HasUpdatePromise()) {
|
|
registration->AddUpdatePromiseObserver(mPromise);
|
|
return NS_OK;
|
|
}
|
|
|
|
// There is no update in progress and since SW updating is upto the UA,
|
|
// we will not update right now. Simply resolve with whatever worker we
|
|
// have.
|
|
nsRefPtr<ServiceWorkerInfo> info = registration->Newest();
|
|
if (info) {
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
nsresult rv =
|
|
swm->CreateServiceWorkerForWindow(mWindow,
|
|
info->GetScriptSpec(),
|
|
registration->mScope,
|
|
getter_AddRefs(serviceWorker));
|
|
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistration> swr =
|
|
new ServiceWorkerRegistration(mWindow,
|
|
NS_ConvertUTF8toUTF16(registration->mScope));
|
|
|
|
mPromise->MaybeResolve(swr);
|
|
return NS_OK;
|
|
}
|
|
}
|
|
} else {
|
|
registration = domainInfo->CreateNewRegistration(mScope);
|
|
}
|
|
|
|
registration->mScriptSpec = spec;
|
|
|
|
rv = swm->Update(registration, mWindow);
|
|
MOZ_ASSERT(registration->HasUpdatePromise());
|
|
|
|
// We append this register() call's promise after calling Update() because
|
|
// we don't want this one to be aborted when the others (existing updates
|
|
// for the same registration) are aborted. Update() sets a new
|
|
// UpdatePromise on the registration.
|
|
registration->mUpdatePromise->AddPromise(mPromise);
|
|
|
|
return rv;
|
|
}
|
|
};
|
|
|
|
// If we return an error code here, the ServiceWorkerContainer will
|
|
// automatically reject the Promise.
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::Register(const nsAString& aScope,
|
|
const nsAString& aScriptURL,
|
|
nsISupports** aPromise)
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
// XXXnsm Don't allow chrome callers for now, we don't support chrome
|
|
// ServiceWorkers.
|
|
MOZ_ASSERT(!nsContentUtils::IsCallerChrome());
|
|
|
|
nsCOMPtr<nsIGlobalObject> sgo = GetEntryGlobal();
|
|
MOZ_ASSERT(sgo, "Register() should only be called from a valid entry settings object!");
|
|
|
|
ErrorResult result;
|
|
nsRefPtr<Promise> promise = Promise::Create(sgo, result);
|
|
if (result.Failed()) {
|
|
return result.ErrorCode();
|
|
}
|
|
|
|
nsCOMPtr<nsIDocument> doc = GetEntryDocument();
|
|
if (!doc) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
// Although the spec says that the same-origin checks should also be done
|
|
// asynchronously, we do them in sync because the Promise created by the
|
|
// WebIDL infrastructure due to a returned error will be resolved
|
|
// asynchronously. We aren't making any internal state changes in these
|
|
// checks, so ordering of multiple calls is not affected.
|
|
|
|
nsCOMPtr<nsIURI> documentURI = doc->GetBaseURI();
|
|
|
|
bool httpsNeeded = true;
|
|
|
|
// FIXME(nsm): Bug 1003991. Disable check when devtools are open.
|
|
if (Preferences::GetBool("dom.serviceWorkers.testing.enabled")) {
|
|
httpsNeeded = false;
|
|
}
|
|
|
|
// No https needed for localhost.
|
|
if (httpsNeeded) {
|
|
nsAutoCString host;
|
|
result = documentURI->GetHost(host);
|
|
if (NS_WARN_IF(result.Failed())) {
|
|
return result.ErrorCode();
|
|
}
|
|
|
|
if (host.Equals("127.0.0.1") ||
|
|
host.Equals("localhost") ||
|
|
host.Equals("::1")) {
|
|
httpsNeeded = false;
|
|
}
|
|
}
|
|
|
|
if (httpsNeeded) {
|
|
bool isHttps;
|
|
result = documentURI->SchemeIs("https", &isHttps);
|
|
if (result.Failed() || !isHttps) {
|
|
NS_WARNING("ServiceWorker registration from insecure websites is not allowed.");
|
|
return NS_ERROR_DOM_SECURITY_ERR;
|
|
}
|
|
}
|
|
|
|
nsCOMPtr<nsIURI> scriptURI;
|
|
result = NS_NewURI(getter_AddRefs(scriptURI), aScriptURL, nullptr, documentURI);
|
|
if (NS_WARN_IF(result.Failed())) {
|
|
return result.ErrorCode();
|
|
}
|
|
|
|
// Data URLs are not allowed.
|
|
nsCOMPtr<nsIPrincipal> documentPrincipal = doc->NodePrincipal();
|
|
|
|
result = documentPrincipal->CheckMayLoad(scriptURI, true /* report */,
|
|
false /* allowIfInheritsPrincipal */);
|
|
if (result.Failed()) {
|
|
return NS_ERROR_DOM_SECURITY_ERR;
|
|
}
|
|
|
|
nsCOMPtr<nsIURI> scopeURI;
|
|
result = NS_NewURI(getter_AddRefs(scopeURI), aScope, nullptr, documentURI);
|
|
if (NS_WARN_IF(result.Failed())) {
|
|
return NS_ERROR_DOM_SECURITY_ERR;
|
|
}
|
|
|
|
result = documentPrincipal->CheckMayLoad(scopeURI, true /* report */,
|
|
false /* allowIfInheritsPrinciple */);
|
|
if (result.Failed()) {
|
|
return NS_ERROR_DOM_SECURITY_ERR;
|
|
}
|
|
|
|
nsCString cleanedScope;
|
|
result = scopeURI->GetSpecIgnoringRef(cleanedScope);
|
|
if (NS_WARN_IF(result.Failed())) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryObject(sgo);
|
|
if (!window) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsRefPtr<nsIRunnable> registerRunnable =
|
|
new RegisterRunnable(window, cleanedScope, scriptURI, promise);
|
|
promise.forget(aPromise);
|
|
return NS_DispatchToCurrentThread(registerRunnable);
|
|
}
|
|
|
|
/*
|
|
* Implements the async aspects of the getRegistrations algorithm.
|
|
*/
|
|
class GetRegistrationsRunnable : public nsRunnable
|
|
{
|
|
nsCOMPtr<nsPIDOMWindow> mWindow;
|
|
nsRefPtr<Promise> mPromise;
|
|
public:
|
|
GetRegistrationsRunnable(nsPIDOMWindow* aWindow, Promise* aPromise)
|
|
: mWindow(aWindow), mPromise(aPromise)
|
|
{ }
|
|
|
|
NS_IMETHODIMP
|
|
Run()
|
|
{
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
|
|
nsIDocument* doc = mWindow->GetExtantDoc();
|
|
if (!doc) {
|
|
mPromise->MaybeReject(NS_ERROR_UNEXPECTED);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsCOMPtr<nsIURI> docURI = doc->GetDocumentURI();
|
|
if (!docURI) {
|
|
mPromise->MaybeReject(NS_ERROR_UNEXPECTED);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsCOMPtr<nsIPrincipal> principal = doc->NodePrincipal();
|
|
if (!principal) {
|
|
mPromise->MaybeReject(NS_ERROR_UNEXPECTED);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsTArray<nsRefPtr<ServiceWorkerRegistration>> array;
|
|
|
|
nsRefPtr<ServiceWorkerManager::ServiceWorkerDomainInfo> domainInfo =
|
|
swm->GetDomainInfo(docURI);
|
|
|
|
if (!domainInfo) {
|
|
mPromise->MaybeResolve(array);
|
|
return NS_OK;
|
|
}
|
|
|
|
for (uint32_t i = 0; i < domainInfo->mOrderedScopes.Length(); ++i) {
|
|
NS_ConvertUTF8toUTF16 scope(domainInfo->mOrderedScopes[i]);
|
|
nsRefPtr<ServiceWorkerRegistration> swr =
|
|
new ServiceWorkerRegistration(mWindow, scope);
|
|
|
|
array.AppendElement(swr);
|
|
}
|
|
|
|
mPromise->MaybeResolve(array);
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
// If we return an error code here, the ServiceWorkerContainer will
|
|
// automatically reject the Promise.
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetRegistrations(nsIDOMWindow* aWindow,
|
|
nsISupports** aPromise)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aWindow);
|
|
|
|
// XXXnsm Don't allow chrome callers for now, we don't support chrome
|
|
// ServiceWorkers.
|
|
MOZ_ASSERT(!nsContentUtils::IsCallerChrome());
|
|
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(aWindow);
|
|
if (!window) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsCOMPtr<nsIGlobalObject> sgo = do_QueryInterface(window);
|
|
ErrorResult result;
|
|
nsRefPtr<Promise> promise = Promise::Create(sgo, result);
|
|
if (result.Failed()) {
|
|
return result.ErrorCode();
|
|
}
|
|
|
|
nsRefPtr<nsIRunnable> runnable =
|
|
new GetRegistrationsRunnable(window, promise);
|
|
promise.forget(aPromise);
|
|
return NS_DispatchToCurrentThread(runnable);
|
|
}
|
|
|
|
/*
|
|
* Implements the async aspects of the getRegistration algorithm.
|
|
*/
|
|
class GetRegistrationRunnable : public nsRunnable
|
|
{
|
|
nsCOMPtr<nsPIDOMWindow> mWindow;
|
|
nsRefPtr<Promise> mPromise;
|
|
nsString mDocumentURL;
|
|
|
|
public:
|
|
GetRegistrationRunnable(nsPIDOMWindow* aWindow, Promise* aPromise,
|
|
const nsAString& aDocumentURL)
|
|
: mWindow(aWindow), mPromise(aPromise), mDocumentURL(aDocumentURL)
|
|
{ }
|
|
|
|
NS_IMETHODIMP
|
|
Run()
|
|
{
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
|
|
nsIDocument* doc = mWindow->GetExtantDoc();
|
|
if (!doc) {
|
|
mPromise->MaybeReject(NS_ERROR_UNEXPECTED);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsCOMPtr<nsIURI> docURI = doc->GetDocumentURI();
|
|
if (!docURI) {
|
|
mPromise->MaybeReject(NS_ERROR_UNEXPECTED);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsCOMPtr<nsIURI> uri;
|
|
nsresult rv = NS_NewURI(getter_AddRefs(uri), mDocumentURL, nullptr, docURI);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
mPromise->MaybeReject(rv);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsCOMPtr<nsIPrincipal> principal = doc->NodePrincipal();
|
|
if (!principal) {
|
|
mPromise->MaybeReject(NS_ERROR_UNEXPECTED);
|
|
return NS_OK;
|
|
}
|
|
|
|
rv = principal->CheckMayLoad(uri, true /* report */,
|
|
false /* allowIfInheritsPrinciple */);
|
|
if (NS_FAILED(rv)) {
|
|
mPromise->MaybeReject(NS_ERROR_DOM_SECURITY_ERR);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration =
|
|
swm->GetServiceWorkerRegistrationInfo(uri);
|
|
|
|
if (!registration) {
|
|
mPromise->MaybeResolve(JS::UndefinedHandleValue);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_ConvertUTF8toUTF16 scope(registration->mScope);
|
|
nsRefPtr<ServiceWorkerRegistration> swr =
|
|
new ServiceWorkerRegistration(mWindow, scope);
|
|
mPromise->MaybeResolve(swr);
|
|
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
// If we return an error code here, the ServiceWorkerContainer will
|
|
// automatically reject the Promise.
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetRegistration(nsIDOMWindow* aWindow,
|
|
const nsAString& aDocumentURL,
|
|
nsISupports** aPromise)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aWindow);
|
|
|
|
// XXXnsm Don't allow chrome callers for now, we don't support chrome
|
|
// ServiceWorkers.
|
|
MOZ_ASSERT(!nsContentUtils::IsCallerChrome());
|
|
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(aWindow);
|
|
if (!window) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsCOMPtr<nsIGlobalObject> sgo = do_QueryInterface(window);
|
|
ErrorResult result;
|
|
nsRefPtr<Promise> promise = Promise::Create(sgo, result);
|
|
if (result.Failed()) {
|
|
return result.ErrorCode();
|
|
}
|
|
|
|
nsRefPtr<nsIRunnable> runnable =
|
|
new GetRegistrationRunnable(window, promise, aDocumentURL);
|
|
promise.forget(aPromise);
|
|
return NS_DispatchToCurrentThread(runnable);
|
|
}
|
|
|
|
class GetReadyPromiseRunnable : public nsRunnable
|
|
{
|
|
nsCOMPtr<nsPIDOMWindow> mWindow;
|
|
nsRefPtr<Promise> mPromise;
|
|
|
|
public:
|
|
GetReadyPromiseRunnable(nsPIDOMWindow* aWindow, Promise* aPromise)
|
|
: mWindow(aWindow), mPromise(aPromise)
|
|
{ }
|
|
|
|
NS_IMETHODIMP
|
|
Run()
|
|
{
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
|
|
nsIDocument* doc = mWindow->GetExtantDoc();
|
|
if (!doc) {
|
|
mPromise->MaybeReject(NS_ERROR_UNEXPECTED);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsCOMPtr<nsIURI> docURI = doc->GetDocumentURI();
|
|
if (!docURI) {
|
|
mPromise->MaybeReject(NS_ERROR_UNEXPECTED);
|
|
return NS_OK;
|
|
}
|
|
|
|
if (!swm->CheckReadyPromise(mWindow, docURI, mPromise)) {
|
|
swm->StorePendingReadyPromise(mWindow, docURI, mPromise);
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetReadyPromise(nsIDOMWindow* aWindow,
|
|
nsISupports** aPromise)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aWindow);
|
|
|
|
// XXXnsm Don't allow chrome callers for now, we don't support chrome
|
|
// ServiceWorkers.
|
|
MOZ_ASSERT(!nsContentUtils::IsCallerChrome());
|
|
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(aWindow);
|
|
if (!window) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
MOZ_ASSERT(!mPendingReadyPromises.Contains(window));
|
|
|
|
nsCOMPtr<nsIGlobalObject> sgo = do_QueryInterface(window);
|
|
ErrorResult result;
|
|
nsRefPtr<Promise> promise = Promise::Create(sgo, result);
|
|
if (result.Failed()) {
|
|
return result.ErrorCode();
|
|
}
|
|
|
|
nsRefPtr<nsIRunnable> runnable =
|
|
new GetReadyPromiseRunnable(window, promise);
|
|
promise.forget(aPromise);
|
|
return NS_DispatchToCurrentThread(runnable);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::RemoveReadyPromise(nsIDOMWindow* aWindow)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aWindow);
|
|
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(aWindow);
|
|
if (!window) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
mPendingReadyPromises.Remove(aWindow);
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::StorePendingReadyPromise(nsPIDOMWindow* aWindow,
|
|
nsIURI* aURI,
|
|
Promise* aPromise)
|
|
{
|
|
PendingReadyPromise* data;
|
|
|
|
// We should not have 2 pending promises for the same window.
|
|
MOZ_ASSERT(!mPendingReadyPromises.Get(aWindow, &data));
|
|
|
|
data = new PendingReadyPromise(aURI, aPromise);
|
|
mPendingReadyPromises.Put(aWindow, data);
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::CheckPendingReadyPromises()
|
|
{
|
|
mPendingReadyPromises.Enumerate(CheckPendingReadyPromisesEnumerator, this);
|
|
}
|
|
|
|
PLDHashOperator
|
|
ServiceWorkerManager::CheckPendingReadyPromisesEnumerator(
|
|
nsISupports* aSupports,
|
|
nsAutoPtr<PendingReadyPromise>& aData,
|
|
void* aPtr)
|
|
{
|
|
ServiceWorkerManager* aSwm = static_cast<ServiceWorkerManager*>(aPtr);
|
|
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(aSupports);
|
|
|
|
if (aSwm->CheckReadyPromise(window, aData->mURI, aData->mPromise)) {
|
|
return PL_DHASH_REMOVE;
|
|
}
|
|
|
|
return PL_DHASH_NEXT;
|
|
}
|
|
|
|
bool
|
|
ServiceWorkerManager::CheckReadyPromise(nsPIDOMWindow* aWindow,
|
|
nsIURI* aURI, Promise* aPromise)
|
|
{
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration =
|
|
GetServiceWorkerRegistrationInfo(aURI);
|
|
|
|
if (registration && registration->mCurrentWorker) {
|
|
NS_ConvertUTF8toUTF16 scope(registration->mScope);
|
|
nsRefPtr<ServiceWorkerRegistration> swr =
|
|
new ServiceWorkerRegistration(aWindow, scope);
|
|
aPromise->MaybeResolve(swr);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::RejectUpdatePromiseObservers(ServiceWorkerRegistrationInfo* aRegistration,
|
|
nsresult aRv)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aRegistration->HasUpdatePromise());
|
|
aRegistration->mUpdatePromise->RejectAllPromises(aRv);
|
|
aRegistration->mUpdatePromise = nullptr;
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::RejectUpdatePromiseObservers(ServiceWorkerRegistrationInfo* aRegistration,
|
|
const ErrorEventInit& aErrorDesc)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aRegistration->HasUpdatePromise());
|
|
aRegistration->mUpdatePromise->RejectAllPromises(aErrorDesc);
|
|
aRegistration->mUpdatePromise = nullptr;
|
|
}
|
|
|
|
/*
|
|
* Update() does not return the Promise that the spec says it should. Callers
|
|
* may access the registration's (new) Promise after calling this method.
|
|
*/
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::Update(ServiceWorkerRegistrationInfo* aRegistration,
|
|
nsPIDOMWindow* aWindow)
|
|
{
|
|
if (aRegistration->HasUpdatePromise()) {
|
|
NS_WARNING("Already had a UpdatePromise. Aborting that one!");
|
|
AbortCurrentUpdate(aRegistration);
|
|
}
|
|
|
|
if (aRegistration->mInstallingWorker) {
|
|
// FIXME(nsm): Terminate the worker. We still haven't figured out worker
|
|
// instance ownership when not associated with a window, so let's wait on
|
|
// this.
|
|
// FIXME(nsm): We should be setting the state on the actual worker
|
|
// instance.
|
|
// FIXME(nsm): Fire "statechange" on installing worker instance.
|
|
aRegistration->mInstallingWorker = nullptr;
|
|
InvalidateServiceWorkerRegistrationWorker(aRegistration,
|
|
WhichServiceWorker::INSTALLING_WORKER);
|
|
}
|
|
|
|
aRegistration->mUpdatePromise = new UpdatePromise();
|
|
// FIXME(nsm): Bug 931249. If we don't need to fetch & install, resolve
|
|
// promise and skip this.
|
|
// FIXME(nsm): Bug 931249. Force cache update if > 1 day.
|
|
|
|
aRegistration->mUpdateInstance =
|
|
new ServiceWorkerUpdateInstance(aRegistration, aWindow);
|
|
aRegistration->mUpdateInstance->Update();
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::AbortCurrentUpdate(ServiceWorkerRegistrationInfo* aRegistration)
|
|
{
|
|
MOZ_ASSERT(aRegistration->HasUpdatePromise());
|
|
RejectUpdatePromiseObservers(aRegistration, NS_ERROR_DOM_ABORT_ERR);
|
|
MOZ_ASSERT(aRegistration->mUpdateInstance);
|
|
aRegistration->mUpdateInstance->Abort();
|
|
aRegistration->mUpdateInstance = nullptr;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::Unregister(nsIServiceWorkerUnregisterCallback* aCallback,
|
|
const nsAString& aScope)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aCallback);
|
|
|
|
nsCOMPtr<nsIURI> scopeURI;
|
|
nsresult rv = NS_NewURI(getter_AddRefs(scopeURI), aScope, nullptr, nullptr);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return NS_ERROR_DOM_SECURITY_ERR;
|
|
}
|
|
|
|
/*
|
|
* Implements the async aspects of the unregister algorithm.
|
|
*/
|
|
class UnregisterRunnable : public nsRunnable
|
|
{
|
|
nsCOMPtr<nsIServiceWorkerUnregisterCallback> mCallback;
|
|
nsCOMPtr<nsIURI> mScopeURI;
|
|
|
|
public:
|
|
UnregisterRunnable(nsIServiceWorkerUnregisterCallback* aCallback,
|
|
nsIURI* aScopeURI)
|
|
: mCallback(aCallback), mScopeURI(aScopeURI)
|
|
{
|
|
AssertIsOnMainThread();
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
Run()
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
|
|
nsRefPtr<ServiceWorkerManager::ServiceWorkerDomainInfo> domainInfo =
|
|
swm->GetDomainInfo(mScopeURI);
|
|
MOZ_ASSERT(domainInfo);
|
|
|
|
nsCString spec;
|
|
nsresult rv = mScopeURI->GetSpecIgnoringRef(spec);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return mCallback->UnregisterFailed();
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration;
|
|
if (!domainInfo->mServiceWorkerRegistrationInfos.Get(spec,
|
|
getter_AddRefs(registration))) {
|
|
return mCallback->UnregisterSucceeded(false);
|
|
}
|
|
|
|
MOZ_ASSERT(registration);
|
|
|
|
registration->mPendingUninstall = true;
|
|
rv = mCallback->UnregisterSucceeded(true);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
// The "Wait until no document is using registration" can actually be
|
|
// handled by [[HandleDocumentUnload]] in Bug 1041340, so we simply check
|
|
// if the document is currently in use here.
|
|
if (!registration->IsControllingDocuments()) {
|
|
if (!registration->mPendingUninstall) {
|
|
return NS_OK;
|
|
}
|
|
|
|
registration->Clear();
|
|
domainInfo->RemoveRegistration(registration);
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
nsRefPtr<nsIRunnable> unregisterRunnable =
|
|
new UnregisterRunnable(aCallback, scopeURI);
|
|
return NS_DispatchToCurrentThread(unregisterRunnable);
|
|
}
|
|
|
|
/* static */
|
|
already_AddRefed<ServiceWorkerManager>
|
|
ServiceWorkerManager::GetInstance()
|
|
{
|
|
nsCOMPtr<nsIServiceWorkerManager> swm = mozilla::services::GetServiceWorkerManager();
|
|
nsRefPtr<ServiceWorkerManager> concrete = do_QueryObject(swm);
|
|
return concrete.forget();
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::ResolveRegisterPromises(ServiceWorkerRegistrationInfo* aRegistration,
|
|
const nsACString& aWorkerScriptSpec)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aRegistration->HasUpdatePromise());
|
|
if (aRegistration->mUpdatePromise->IsRejected()) {
|
|
aRegistration->mUpdatePromise = nullptr;
|
|
return;
|
|
}
|
|
|
|
aRegistration->mUpdatePromise->ResolveAllPromises(aWorkerScriptSpec,
|
|
aRegistration->mScope);
|
|
aRegistration->mUpdatePromise = nullptr;
|
|
}
|
|
|
|
// Must NS_Free() aString
|
|
void
|
|
ServiceWorkerManager::FinishFetch(ServiceWorkerRegistrationInfo* aRegistration,
|
|
nsPIDOMWindow* aWindow)
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
MOZ_ASSERT(aRegistration->HasUpdatePromise());
|
|
MOZ_ASSERT(aRegistration->mUpdateInstance);
|
|
aRegistration->mUpdateInstance = nullptr;
|
|
if (aRegistration->mUpdatePromise->IsRejected()) {
|
|
aRegistration->mUpdatePromise = nullptr;
|
|
return;
|
|
}
|
|
|
|
// We have skipped Steps 3-8.3 of the Update algorithm here!
|
|
|
|
nsRefPtr<ServiceWorker> worker;
|
|
nsresult rv = CreateServiceWorkerForWindow(aWindow,
|
|
aRegistration->mScriptSpec,
|
|
aRegistration->mScope,
|
|
getter_AddRefs(worker));
|
|
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
RejectUpdatePromiseObservers(aRegistration, rv);
|
|
return;
|
|
}
|
|
|
|
ResolveRegisterPromises(aRegistration, aRegistration->mScriptSpec);
|
|
|
|
nsRefPtr<ServiceWorkerInfo> info = new ServiceWorkerInfo(aRegistration->mScriptSpec);
|
|
Install(aRegistration, info);
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::HandleError(JSContext* aCx,
|
|
const nsACString& aScope,
|
|
const nsAString& aWorkerURL,
|
|
nsString aMessage,
|
|
nsString aFilename,
|
|
nsString aLine,
|
|
uint32_t aLineNumber,
|
|
uint32_t aColumnNumber,
|
|
uint32_t aFlags)
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
nsCOMPtr<nsIURI> uri;
|
|
nsresult rv = NS_NewURI(getter_AddRefs(uri), aScope, nullptr, nullptr);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(uri);
|
|
if (!domainInfo) {
|
|
return;
|
|
}
|
|
|
|
nsCString scope;
|
|
scope.Assign(aScope);
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration = domainInfo->GetRegistration(scope);
|
|
MOZ_ASSERT(registration);
|
|
|
|
RootedDictionary<ErrorEventInit> init(aCx);
|
|
init.mMessage = aMessage;
|
|
init.mFilename = aFilename;
|
|
init.mLineno = aLineNumber;
|
|
init.mColno = aColumnNumber;
|
|
|
|
// If the worker was the one undergoing registration, we reject the promises,
|
|
// otherwise we fire events on the ServiceWorker instances.
|
|
|
|
// If there is an update in progress and the worker that errored is the same one
|
|
// that is being updated, it is a sufficient test for 'this worker is being
|
|
// registered'.
|
|
// FIXME(nsm): Except the case where an update is found for a worker, in
|
|
// which case we'll need some other association than simply the URL.
|
|
if (registration->mUpdateInstance &&
|
|
registration->mUpdateInstance->GetScriptSpec().Equals(NS_ConvertUTF16toUTF8(aWorkerURL))) {
|
|
RejectUpdatePromiseObservers(registration, init);
|
|
// We don't need to abort here since the worker has already run.
|
|
registration->mUpdateInstance = nullptr;
|
|
} else {
|
|
// FIXME(nsm): Bug 983497 Fire 'error' on ServiceWorkerContainers.
|
|
}
|
|
}
|
|
|
|
class FinishInstallRunnable MOZ_FINAL : public nsRunnable
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> mRegistration;
|
|
|
|
public:
|
|
explicit FinishInstallRunnable(
|
|
const nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo>& aRegistration)
|
|
: mRegistration(aRegistration)
|
|
{
|
|
MOZ_ASSERT(!NS_IsMainThread());
|
|
}
|
|
|
|
NS_IMETHOD
|
|
Run() MOZ_OVERRIDE
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
swm->FinishInstall(mRegistration.get());
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
class FinishActivationRunnable : public nsRunnable
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> mRegistration;
|
|
|
|
public:
|
|
explicit FinishActivationRunnable(const nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo>& aRegistration)
|
|
: mRegistration(aRegistration)
|
|
{
|
|
MOZ_ASSERT(!NS_IsMainThread());
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
Run()
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
// FinishActivate takes ownership of the passed info.
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
swm->FinishActivate(mRegistration.get());
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
class CancelServiceWorkerInstallationRunnable MOZ_FINAL : public nsRunnable
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> mRegistration;
|
|
|
|
public:
|
|
explicit CancelServiceWorkerInstallationRunnable(
|
|
const nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo>& aRegistration)
|
|
: mRegistration(aRegistration)
|
|
{
|
|
}
|
|
|
|
NS_IMETHOD
|
|
Run() MOZ_OVERRIDE
|
|
{
|
|
AssertIsOnMainThread();
|
|
// FIXME(nsm): Change installing worker state to redundant.
|
|
// FIXME(nsm): Fire statechange.
|
|
mRegistration->mInstallingWorker = nullptr;
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
swm->InvalidateServiceWorkerRegistrationWorker(mRegistration,
|
|
WhichServiceWorker::INSTALLING_WORKER);
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
/*
|
|
* Used to handle InstallEvent::waitUntil() and proceed with installation.
|
|
*/
|
|
class FinishInstallHandler MOZ_FINAL : public PromiseNativeHandler
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> mRegistration;
|
|
|
|
virtual
|
|
~FinishInstallHandler()
|
|
{ }
|
|
|
|
public:
|
|
explicit FinishInstallHandler(
|
|
const nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo>& aRegistration)
|
|
: mRegistration(aRegistration)
|
|
{
|
|
MOZ_ASSERT(!NS_IsMainThread());
|
|
}
|
|
|
|
void
|
|
ResolvedCallback(JSContext* aCx, JS::Handle<JS::Value> aValue) MOZ_OVERRIDE
|
|
{
|
|
WorkerPrivate* workerPrivate = GetCurrentThreadWorkerPrivate();
|
|
MOZ_ASSERT(workerPrivate);
|
|
workerPrivate->AssertIsOnWorkerThread();
|
|
|
|
nsRefPtr<FinishInstallRunnable> r = new FinishInstallRunnable(mRegistration);
|
|
NS_DispatchToMainThread(r);
|
|
}
|
|
|
|
void
|
|
RejectedCallback(JSContext* aCx, JS::Handle<JS::Value> aValue) MOZ_OVERRIDE
|
|
{
|
|
nsRefPtr<CancelServiceWorkerInstallationRunnable> r =
|
|
new CancelServiceWorkerInstallationRunnable(mRegistration);
|
|
NS_DispatchToMainThread(r);
|
|
}
|
|
};
|
|
|
|
class FinishActivateHandler : public PromiseNativeHandler
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> mRegistration;
|
|
|
|
public:
|
|
explicit FinishActivateHandler(const nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo>& aRegistration)
|
|
: mRegistration(aRegistration)
|
|
{
|
|
MOZ_ASSERT(!NS_IsMainThread());
|
|
}
|
|
|
|
virtual
|
|
~FinishActivateHandler()
|
|
{ }
|
|
|
|
void
|
|
ResolvedCallback(JSContext* aCx, JS::Handle<JS::Value> aValue) MOZ_OVERRIDE
|
|
{
|
|
WorkerPrivate* workerPrivate = GetCurrentThreadWorkerPrivate();
|
|
MOZ_ASSERT(workerPrivate);
|
|
workerPrivate->AssertIsOnWorkerThread();
|
|
|
|
nsRefPtr<FinishActivationRunnable> r = new FinishActivationRunnable(mRegistration);
|
|
NS_DispatchToMainThread(r);
|
|
}
|
|
|
|
void
|
|
RejectedCallback(JSContext* aCx, JS::Handle<JS::Value> aValue) MOZ_OVERRIDE
|
|
{
|
|
// FIXME(nsm). Spec is undefined.
|
|
}
|
|
};
|
|
|
|
/*
|
|
* Fires 'install' event on the ServiceWorkerGlobalScope. Modifies busy count
|
|
* since it fires the event. This is ok since there can't be nested
|
|
* ServiceWorkers, so the parent thread -> worker thread requirement for
|
|
* runnables is satisfied.
|
|
*/
|
|
class InstallEventRunnable MOZ_FINAL : public WorkerRunnable
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> mRegistration;
|
|
nsCString mScope;
|
|
|
|
public:
|
|
InstallEventRunnable(
|
|
WorkerPrivate* aWorkerPrivate,
|
|
const nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo>& aRegistration)
|
|
: WorkerRunnable(aWorkerPrivate, WorkerThreadModifyBusyCount),
|
|
mRegistration(aRegistration),
|
|
mScope(aRegistration.get()->mScope) // copied for access on worker thread.
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aWorkerPrivate);
|
|
}
|
|
|
|
bool
|
|
WorkerRun(JSContext* aCx, WorkerPrivate* aWorkerPrivate)
|
|
{
|
|
MOZ_ASSERT(aWorkerPrivate);
|
|
return DispatchInstallEvent(aCx, aWorkerPrivate);
|
|
}
|
|
|
|
private:
|
|
bool
|
|
DispatchInstallEvent(JSContext* aCx, WorkerPrivate* aWorkerPrivate)
|
|
{
|
|
aWorkerPrivate->AssertIsOnWorkerThread();
|
|
MOZ_ASSERT(aWorkerPrivate->IsServiceWorker());
|
|
InstallEventInit init;
|
|
init.mBubbles = false;
|
|
init.mCancelable = true;
|
|
|
|
// FIXME(nsm): Bug 982787 pass previous active worker.
|
|
|
|
nsRefPtr<EventTarget> target = aWorkerPrivate->GlobalScope();
|
|
nsRefPtr<InstallEvent> event =
|
|
InstallEvent::Constructor(target, NS_LITERAL_STRING("install"), init);
|
|
|
|
event->SetTrusted(true);
|
|
|
|
nsRefPtr<Promise> waitUntilPromise;
|
|
|
|
nsresult rv = target->DispatchDOMEvent(nullptr, event, nullptr, nullptr);
|
|
|
|
nsCOMPtr<nsIGlobalObject> sgo = aWorkerPrivate->GlobalScope();
|
|
if (NS_SUCCEEDED(rv)) {
|
|
waitUntilPromise = event->GetPromise();
|
|
if (!waitUntilPromise) {
|
|
ErrorResult rv;
|
|
waitUntilPromise =
|
|
Promise::Resolve(sgo,
|
|
aCx, JS::UndefinedHandleValue, rv);
|
|
}
|
|
} else {
|
|
ErrorResult rv;
|
|
// Continue with a canceled install.
|
|
waitUntilPromise = Promise::Reject(sgo, aCx,
|
|
JS::UndefinedHandleValue, rv);
|
|
}
|
|
|
|
nsRefPtr<FinishInstallHandler> handler =
|
|
new FinishInstallHandler(mRegistration);
|
|
waitUntilPromise->AppendNativeHandler(handler);
|
|
return true;
|
|
}
|
|
};
|
|
|
|
class ActivateEventRunnable : public WorkerRunnable
|
|
{
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> mRegistration;
|
|
|
|
public:
|
|
ActivateEventRunnable(WorkerPrivate* aWorkerPrivate,
|
|
const nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo>& aRegistration)
|
|
: WorkerRunnable(aWorkerPrivate, WorkerThreadModifyBusyCount),
|
|
mRegistration(aRegistration)
|
|
{
|
|
MOZ_ASSERT(aWorkerPrivate);
|
|
}
|
|
|
|
bool
|
|
WorkerRun(JSContext* aCx, WorkerPrivate* aWorkerPrivate)
|
|
{
|
|
MOZ_ASSERT(aWorkerPrivate);
|
|
return DispatchActivateEvent(aCx, aWorkerPrivate);
|
|
}
|
|
|
|
private:
|
|
bool
|
|
DispatchActivateEvent(JSContext* aCx, WorkerPrivate* aWorkerPrivate)
|
|
{
|
|
MOZ_ASSERT(aWorkerPrivate->IsServiceWorker());
|
|
nsRefPtr<EventTarget> target = do_QueryObject(aWorkerPrivate->GlobalScope());
|
|
|
|
// FIXME(nsm): Set activeWorker to the correct thing.
|
|
EventInit init;
|
|
init.mBubbles = false;
|
|
init.mCancelable = true;
|
|
nsRefPtr<InstallPhaseEvent> event =
|
|
InstallPhaseEvent::Constructor(target, NS_LITERAL_STRING("activate"), init);
|
|
|
|
event->SetTrusted(true);
|
|
|
|
nsRefPtr<Promise> waitUntilPromise;
|
|
|
|
nsresult rv = target->DispatchDOMEvent(nullptr, event, nullptr, nullptr);
|
|
if (NS_SUCCEEDED(rv)) {
|
|
waitUntilPromise = event->GetPromise();
|
|
if (!waitUntilPromise) {
|
|
ErrorResult rv;
|
|
nsCOMPtr<nsIGlobalObject> global =
|
|
do_QueryObject(aWorkerPrivate->GlobalScope());
|
|
waitUntilPromise =
|
|
Promise::Resolve(global,
|
|
aCx, JS::UndefinedHandleValue, rv);
|
|
}
|
|
} else {
|
|
ErrorResult rv;
|
|
nsCOMPtr<nsIGlobalObject> global =
|
|
do_QueryObject(aWorkerPrivate->GlobalScope());
|
|
// Continue with a canceled install.
|
|
waitUntilPromise = Promise::Reject(global, aCx,
|
|
JS::UndefinedHandleValue, rv);
|
|
}
|
|
|
|
nsRefPtr<FinishActivateHandler> handler = new FinishActivateHandler(mRegistration);
|
|
waitUntilPromise->AppendNativeHandler(handler);
|
|
return true;
|
|
}
|
|
};
|
|
|
|
void
|
|
ServiceWorkerManager::Install(ServiceWorkerRegistrationInfo* aRegistration,
|
|
ServiceWorkerInfo* aServiceWorkerInfo)
|
|
{
|
|
AssertIsOnMainThread();
|
|
aRegistration->mInstallingWorker = aServiceWorkerInfo;
|
|
MOZ_ASSERT(aRegistration->mInstallingWorker);
|
|
InvalidateServiceWorkerRegistrationWorker(aRegistration,
|
|
WhichServiceWorker::INSTALLING_WORKER);
|
|
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> handle(
|
|
new nsMainThreadPtrHolder<ServiceWorkerRegistrationInfo>(aRegistration));
|
|
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
nsresult rv =
|
|
CreateServiceWorker(aServiceWorkerInfo->GetScriptSpec(),
|
|
aRegistration->mScope,
|
|
getter_AddRefs(serviceWorker));
|
|
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
aRegistration->mInstallingWorker = nullptr;
|
|
// We don't need to invalidate here since the upper one will have done it.
|
|
return;
|
|
}
|
|
|
|
nsRefPtr<InstallEventRunnable> r =
|
|
new InstallEventRunnable(serviceWorker->GetWorkerPrivate(), handle);
|
|
|
|
AutoSafeJSContext cx;
|
|
r->Dispatch(cx);
|
|
|
|
// When this function exits, although we've lost references to the ServiceWorker,
|
|
// which means the underlying WorkerPrivate has no references, the worker
|
|
// will stay alive due to the modified busy count until the install event has
|
|
// been dispatched.
|
|
// NOTE: The worker spec does not require Promises to keep a worker alive, so
|
|
// the waitUntil() construct by itself will not keep a worker alive beyond
|
|
// the event dispatch. On the other hand, networking, IDB and so on do keep
|
|
// the worker alive, so the waitUntil() is only relevant if the Promise is
|
|
// gated on those actions. I (nsm) am not sure if it is worth requiring
|
|
// a special spec mention saying the install event should keep the worker
|
|
// alive indefinitely purely on the basis of calling waitUntil(), since
|
|
// a wait is likely to be required only when performing networking or storage
|
|
// transactions in the first place.
|
|
|
|
FireEventOnServiceWorkerRegistrations(aRegistration,
|
|
NS_LITERAL_STRING("updatefound"));
|
|
}
|
|
|
|
class ActivationRunnable : public nsRunnable
|
|
{
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> mRegistration;
|
|
public:
|
|
explicit ActivationRunnable(ServiceWorkerRegistrationInfo* aRegistration)
|
|
: mRegistration(aRegistration)
|
|
{
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
Run() MOZ_OVERRIDE
|
|
{
|
|
if (mRegistration->mCurrentWorker) {
|
|
// FIXME(nsm). Steps 3.1-3.4 of the algorithm.
|
|
}
|
|
|
|
mRegistration->mCurrentWorker = mRegistration->mWaitingWorker.forget();
|
|
|
|
nsRefPtr<ServiceWorkerManager> swm = ServiceWorkerManager::GetInstance();
|
|
swm->InvalidateServiceWorkerRegistrationWorker(mRegistration,
|
|
WhichServiceWorker::ACTIVE_WORKER | WhichServiceWorker::WAITING_WORKER);
|
|
if (!mRegistration->mCurrentWorker) {
|
|
// FIXME(nsm): Just got unregistered!
|
|
return NS_OK;
|
|
}
|
|
|
|
swm->CheckPendingReadyPromises();
|
|
|
|
// FIXME(nsm): Steps 7 of the algorithm.
|
|
|
|
swm->FireEventOnServiceWorkerRegistrations(mRegistration,
|
|
NS_LITERAL_STRING("controllerchange"));
|
|
|
|
MOZ_ASSERT(mRegistration->mCurrentWorker);
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
nsresult rv =
|
|
swm->CreateServiceWorker(mRegistration->mCurrentWorker->GetScriptSpec(),
|
|
mRegistration->mScope,
|
|
getter_AddRefs(serviceWorker));
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
nsMainThreadPtrHandle<ServiceWorkerRegistrationInfo> handle(
|
|
new nsMainThreadPtrHolder<ServiceWorkerRegistrationInfo>(mRegistration));
|
|
|
|
nsRefPtr<ActivateEventRunnable> r =
|
|
new ActivateEventRunnable(serviceWorker->GetWorkerPrivate(), handle);
|
|
|
|
AutoSafeJSContext cx;
|
|
if (!r->Dispatch(cx)) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
void
|
|
ServiceWorkerManager::FinishInstall(ServiceWorkerRegistrationInfo* aRegistration)
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
if (aRegistration->mWaitingWorker) {
|
|
// FIXME(nsm): Actually update the state of active ServiceWorker instances.
|
|
}
|
|
|
|
if (!aRegistration->mInstallingWorker) {
|
|
// It is possible that while this run of [[Install]] was waiting for
|
|
// the worker to handle the install event, some page called register() with
|
|
// a different script leading to [[Update]] terminating the
|
|
// installingWorker and setting it to null. The FinishInstallRunnable may
|
|
// already have been dispatched, hence the check.
|
|
return;
|
|
}
|
|
|
|
aRegistration->mWaitingWorker = aRegistration->mInstallingWorker.forget();
|
|
MOZ_ASSERT(aRegistration->mWaitingWorker);
|
|
InvalidateServiceWorkerRegistrationWorker(aRegistration,
|
|
WhichServiceWorker::WAITING_WORKER | WhichServiceWorker::INSTALLING_WORKER);
|
|
|
|
// FIXME(nsm): Actually update state of active ServiceWorker instances to
|
|
// installed.
|
|
// FIXME(nsm): Fire statechange on the instances.
|
|
|
|
// FIXME(nsm): Handle replace().
|
|
|
|
if (!aRegistration->IsControllingDocuments()) {
|
|
nsRefPtr<ActivationRunnable> r =
|
|
new ActivationRunnable(aRegistration);
|
|
|
|
nsresult rv = NS_DispatchToMainThread(r);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
// FIXME(nsm): Handle error.
|
|
// How likely is this to happen and can we really do anything about it?
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::FinishActivate(ServiceWorkerRegistrationInfo* aRegistration)
|
|
{
|
|
// FIXME(nsm): Set aRegistration->mCurrentWorker state to activated.
|
|
// Fire statechange.
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::CreateServiceWorkerForWindow(nsPIDOMWindow* aWindow,
|
|
const nsACString& aScriptSpec,
|
|
const nsACString& aScope,
|
|
ServiceWorker** aServiceWorker)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aWindow);
|
|
|
|
RuntimeService* rs = RuntimeService::GetOrCreateService();
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
|
|
nsCOMPtr<nsIGlobalObject> sgo = do_QueryInterface(aWindow);
|
|
|
|
AutoSafeJSContext cx;
|
|
JS::Rooted<JSObject*> jsGlobal(cx, sgo->GetGlobalJSObject());
|
|
JSAutoCompartment ac(cx, jsGlobal);
|
|
|
|
GlobalObject global(cx, jsGlobal);
|
|
nsresult rv = rs->CreateServiceWorker(global,
|
|
NS_ConvertUTF8toUTF16(aScriptSpec),
|
|
aScope,
|
|
getter_AddRefs(serviceWorker));
|
|
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
serviceWorker.forget(aServiceWorker);
|
|
return rv;
|
|
}
|
|
|
|
already_AddRefed<ServiceWorkerRegistrationInfo>
|
|
ServiceWorkerManager::GetServiceWorkerRegistrationInfo(nsPIDOMWindow* aWindow)
|
|
{
|
|
nsCOMPtr<nsIDocument> document = aWindow->GetExtantDoc();
|
|
return GetServiceWorkerRegistrationInfo(document);
|
|
}
|
|
|
|
already_AddRefed<ServiceWorkerRegistrationInfo>
|
|
ServiceWorkerManager::GetServiceWorkerRegistrationInfo(nsIDocument* aDoc)
|
|
{
|
|
nsCOMPtr<nsIURI> documentURI = aDoc->GetDocumentURI();
|
|
return GetServiceWorkerRegistrationInfo(documentURI);
|
|
}
|
|
|
|
already_AddRefed<ServiceWorkerRegistrationInfo>
|
|
ServiceWorkerManager::GetServiceWorkerRegistrationInfo(nsIURI* aURI)
|
|
{
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(aURI);
|
|
if (!domainInfo) {
|
|
return nullptr;
|
|
}
|
|
|
|
nsCString spec;
|
|
nsresult rv = aURI->GetSpec(spec);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return nullptr;
|
|
}
|
|
|
|
nsCString scope = FindScopeForPath(domainInfo->mOrderedScopes, spec);
|
|
if (scope.IsEmpty()) {
|
|
return nullptr;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration;
|
|
domainInfo->mServiceWorkerRegistrationInfos.Get(scope, getter_AddRefs(registration));
|
|
// ordered scopes and registrations better be in sync.
|
|
MOZ_ASSERT(registration);
|
|
|
|
if (registration->mPendingUninstall) {
|
|
return nullptr;
|
|
}
|
|
return registration.forget();
|
|
}
|
|
|
|
/* static */ void
|
|
ServiceWorkerManager::AddScope(nsTArray<nsCString>& aList, const nsACString& aScope)
|
|
{
|
|
for (uint32_t i = 0; i < aList.Length(); ++i) {
|
|
const nsCString& current = aList[i];
|
|
|
|
// Perfect match!
|
|
if (aScope.Equals(current)) {
|
|
return;
|
|
}
|
|
|
|
// Sort by length, with longest match first.
|
|
// /foo/bar should be before /foo/
|
|
// Similarly /foo/b is between the two.
|
|
if (StringBeginsWith(aScope, current)) {
|
|
aList.InsertElementAt(i, aScope);
|
|
return;
|
|
}
|
|
}
|
|
|
|
aList.AppendElement(aScope);
|
|
}
|
|
|
|
/* static */ nsCString
|
|
ServiceWorkerManager::FindScopeForPath(nsTArray<nsCString>& aList, const nsACString& aPath)
|
|
{
|
|
nsCString match;
|
|
|
|
for (uint32_t i = 0; i < aList.Length(); ++i) {
|
|
const nsCString& current = aList[i];
|
|
if (StringBeginsWith(aPath, current)) {
|
|
match = current;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return match;
|
|
}
|
|
|
|
/* static */ void
|
|
ServiceWorkerManager::RemoveScope(nsTArray<nsCString>& aList, const nsACString& aScope)
|
|
{
|
|
aList.RemoveElement(aScope);
|
|
}
|
|
|
|
already_AddRefed<ServiceWorkerManager::ServiceWorkerDomainInfo>
|
|
ServiceWorkerManager::GetDomainInfo(nsIDocument* aDoc)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aDoc);
|
|
nsCOMPtr<nsIURI> documentURI = aDoc->GetDocumentURI();
|
|
return GetDomainInfo(documentURI);
|
|
}
|
|
|
|
already_AddRefed<ServiceWorkerManager::ServiceWorkerDomainInfo>
|
|
ServiceWorkerManager::GetDomainInfo(nsIURI* aURI)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aURI);
|
|
|
|
nsCString domain;
|
|
nsresult rv = aURI->GetHost(domain);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return nullptr;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo;
|
|
mDomainMap.Get(domain, getter_AddRefs(domainInfo));
|
|
return domainInfo.forget();
|
|
}
|
|
|
|
already_AddRefed<ServiceWorkerManager::ServiceWorkerDomainInfo>
|
|
ServiceWorkerManager::GetDomainInfo(const nsCString& aURL)
|
|
{
|
|
AssertIsOnMainThread();
|
|
nsCOMPtr<nsIURI> uri;
|
|
nsresult rv = NS_NewURI(getter_AddRefs(uri), aURL, nullptr, nullptr);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return nullptr;
|
|
}
|
|
|
|
return GetDomainInfo(uri);
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::MaybeStartControlling(nsIDocument* aDoc)
|
|
{
|
|
AssertIsOnMainThread();
|
|
if (!Preferences::GetBool("dom.serviceWorkers.enabled")) {
|
|
return;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(aDoc);
|
|
if (!domainInfo) {
|
|
return;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration =
|
|
GetServiceWorkerRegistrationInfo(aDoc);
|
|
if (registration && registration->mCurrentWorker) {
|
|
MOZ_ASSERT(!domainInfo->mControlledDocuments.Contains(aDoc));
|
|
registration->StartControllingADocument();
|
|
// Use the already_AddRefed<> form of Put to avoid the addref-deref since
|
|
// we don't need the registration pointer in this function anymore.
|
|
domainInfo->mControlledDocuments.Put(aDoc, registration.forget());
|
|
}
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::MaybeStopControlling(nsIDocument* aDoc)
|
|
{
|
|
MOZ_ASSERT(aDoc);
|
|
if (!Preferences::GetBool("dom.serviceWorkers.enabled")) {
|
|
return;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(aDoc);
|
|
if (!domainInfo) {
|
|
return;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration;
|
|
domainInfo->mControlledDocuments.Remove(aDoc, getter_AddRefs(registration));
|
|
// A document which was uncontrolled does not maintain that state itself, so
|
|
// it will always call MaybeStopControlling() even if there isn't an
|
|
// associated registration. So this check is required.
|
|
if (registration) {
|
|
registration->StopControllingADocument();
|
|
}
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetScopeForUrl(const nsAString& aUrl, nsAString& aScope)
|
|
{
|
|
nsCOMPtr<nsIURI> uri;
|
|
nsresult rv = NS_NewURI(getter_AddRefs(uri), aUrl, nullptr, nullptr);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> r = GetServiceWorkerRegistrationInfo(uri);
|
|
if (!r) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
aScope = NS_ConvertUTF8toUTF16(r->mScope);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::AddRegistrationEventListener(nsIURI* aDocumentURI, nsIDOMEventTarget* aListener)
|
|
{
|
|
MOZ_ASSERT(aDocumentURI);
|
|
AssertIsOnMainThread();
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(aDocumentURI);
|
|
if (!domainInfo) {
|
|
nsCString domain;
|
|
nsresult rv = aDocumentURI->GetHost(domain);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
domainInfo = new ServiceWorkerDomainInfo;
|
|
mDomainMap.Put(domain, domainInfo);
|
|
}
|
|
|
|
MOZ_ASSERT(domainInfo);
|
|
|
|
// TODO: this is very very bad:
|
|
ServiceWorkerRegistration* registration = static_cast<ServiceWorkerRegistration*>(aListener);
|
|
MOZ_ASSERT(!domainInfo->mServiceWorkerRegistrations.Contains(registration));
|
|
domainInfo->mServiceWorkerRegistrations.AppendElement(registration);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::RemoveRegistrationEventListener(nsIURI* aDocumentURI, nsIDOMEventTarget* aListener)
|
|
{
|
|
AssertIsOnMainThread();
|
|
MOZ_ASSERT(aDocumentURI);
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(aDocumentURI);
|
|
if (!domainInfo) {
|
|
return NS_OK;
|
|
}
|
|
|
|
ServiceWorkerRegistration* registration = static_cast<ServiceWorkerRegistration*>(aListener);
|
|
MOZ_ASSERT(domainInfo->mServiceWorkerRegistrations.Contains(registration));
|
|
domainInfo->mServiceWorkerRegistrations.RemoveElement(registration);
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::FireEventOnServiceWorkerRegistrations(
|
|
ServiceWorkerRegistrationInfo* aRegistration,
|
|
const nsAString& aName)
|
|
{
|
|
AssertIsOnMainThread();
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo =
|
|
GetDomainInfo(aRegistration->mScriptSpec);
|
|
|
|
if (domainInfo) {
|
|
nsTObserverArray<ServiceWorkerRegistration*>::ForwardIterator it(domainInfo->mServiceWorkerRegistrations);
|
|
while (it.HasMore()) {
|
|
nsRefPtr<ServiceWorkerRegistration> target = it.GetNext();
|
|
nsIURI* targetURI = target->GetDocumentURI();
|
|
if (!targetURI) {
|
|
NS_WARNING("Controlled domain cannot have page with null URI!");
|
|
continue;
|
|
}
|
|
|
|
nsCString path;
|
|
nsresult rv = targetURI->GetSpec(path);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
continue;
|
|
}
|
|
|
|
nsCString scope = FindScopeForPath(domainInfo->mOrderedScopes, path);
|
|
if (scope.IsEmpty() ||
|
|
!scope.Equals(aRegistration->mScope)) {
|
|
continue;
|
|
}
|
|
|
|
target->DispatchTrustedEvent(aName);
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
* This is used for installing, waiting and active.
|
|
*/
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetServiceWorkerForScope(nsIDOMWindow* aWindow,
|
|
const nsAString& aScope,
|
|
WhichServiceWorker aWhichWorker,
|
|
nsISupports** aServiceWorker)
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(aWindow);
|
|
if (!window) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsCOMPtr<nsIDocument> doc = window->GetExtantDoc();
|
|
MOZ_ASSERT(doc);
|
|
|
|
///////////////////////////////////////////
|
|
// Security check
|
|
nsCString scope = NS_ConvertUTF16toUTF8(aScope);
|
|
nsCOMPtr<nsIURI> scopeURI;
|
|
// We pass nullptr as the base URI since scopes obtained from
|
|
// ServiceWorkerRegistrations MUST be fully qualified URIs.
|
|
nsresult rv = NS_NewURI(getter_AddRefs(scopeURI), scope, nullptr, nullptr);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return NS_ERROR_DOM_SECURITY_ERR;
|
|
}
|
|
|
|
nsCOMPtr<nsIPrincipal> documentPrincipal = doc->NodePrincipal();
|
|
rv = documentPrincipal->CheckMayLoad(scopeURI, true /* report */,
|
|
false /* allowIfInheritsPrinciple */);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return NS_ERROR_DOM_SECURITY_ERR;
|
|
}
|
|
////////////////////////////////////////////
|
|
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(scope);
|
|
if (!domainInfo) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration =
|
|
domainInfo->GetRegistration(scope);
|
|
if (!registration) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerInfo> info;
|
|
if (aWhichWorker == WhichServiceWorker::INSTALLING_WORKER) {
|
|
info = registration->mInstallingWorker;
|
|
} else if (aWhichWorker == WhichServiceWorker::WAITING_WORKER) {
|
|
info = registration->mWaitingWorker;
|
|
} else if (aWhichWorker == WhichServiceWorker::ACTIVE_WORKER) {
|
|
info = registration->mCurrentWorker;
|
|
} else {
|
|
MOZ_CRASH("Invalid worker type");
|
|
}
|
|
|
|
if (!info) {
|
|
return NS_ERROR_DOM_NOT_FOUND_ERR;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
rv = CreateServiceWorkerForWindow(window,
|
|
info->GetScriptSpec(),
|
|
registration->mScope,
|
|
getter_AddRefs(serviceWorker));
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
serviceWorker.forget(aServiceWorker);
|
|
return NS_OK;
|
|
}
|
|
|
|
/*
|
|
* The .controller is for the registration associated with the document when
|
|
* the document was loaded.
|
|
*/
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetDocumentController(nsIDOMWindow* aWindow, nsISupports** aServiceWorker)
|
|
{
|
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(aWindow);
|
|
MOZ_ASSERT(window);
|
|
if (!window || !window->GetExtantDoc()) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsCOMPtr<nsIDocument> doc = window->GetExtantDoc();
|
|
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(doc);
|
|
if (!domainInfo) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration;
|
|
if (!domainInfo->mControlledDocuments.Get(doc, getter_AddRefs(registration))) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
// If the document is controlled, the current worker MUST be non-null.
|
|
MOZ_ASSERT(registration->mCurrentWorker);
|
|
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
nsresult rv = CreateServiceWorkerForWindow(window,
|
|
registration->mCurrentWorker->GetScriptSpec(),
|
|
registration->mScope,
|
|
getter_AddRefs(serviceWorker));
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
serviceWorker.forget(aServiceWorker);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetInstalling(nsIDOMWindow* aWindow,
|
|
const nsAString& aScope,
|
|
nsISupports** aServiceWorker)
|
|
{
|
|
return GetServiceWorkerForScope(aWindow, aScope,
|
|
WhichServiceWorker::INSTALLING_WORKER,
|
|
aServiceWorker);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetWaiting(nsIDOMWindow* aWindow,
|
|
const nsAString& aScope,
|
|
nsISupports** aServiceWorker)
|
|
{
|
|
return GetServiceWorkerForScope(aWindow, aScope,
|
|
WhichServiceWorker::WAITING_WORKER,
|
|
aServiceWorker);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::GetActive(nsIDOMWindow* aWindow,
|
|
const nsAString& aScope,
|
|
nsISupports** aServiceWorker)
|
|
{
|
|
return GetServiceWorkerForScope(aWindow, aScope,
|
|
WhichServiceWorker::ACTIVE_WORKER,
|
|
aServiceWorker);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ServiceWorkerManager::CreateServiceWorker(const nsACString& aScriptSpec,
|
|
const nsACString& aScope,
|
|
ServiceWorker** aServiceWorker)
|
|
{
|
|
AssertIsOnMainThread();
|
|
|
|
WorkerPrivate::LoadInfo info;
|
|
nsresult rv = NS_NewURI(getter_AddRefs(info.mBaseURI), aScriptSpec, nullptr, nullptr);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
info.mResolvedScriptURI = info.mBaseURI;
|
|
|
|
rv = info.mBaseURI->GetHost(info.mDomain);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
// FIXME(nsm): Create correct principal based on app-ness.
|
|
// Would it make sense to store the nsIPrincipal of the first register() in
|
|
// the ServiceWorkerRegistrationInfo and use that?
|
|
nsIScriptSecurityManager* ssm = nsContentUtils::GetSecurityManager();
|
|
rv = ssm->GetNoAppCodebasePrincipal(info.mBaseURI, getter_AddRefs(info.mPrincipal));
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
AutoSafeJSContext cx;
|
|
|
|
nsRefPtr<ServiceWorker> serviceWorker;
|
|
RuntimeService* rs = RuntimeService::GetService();
|
|
if (!rs) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
rv = rs->CreateServiceWorkerFromLoadInfo(cx, info, NS_ConvertUTF8toUTF16(aScriptSpec), aScope,
|
|
getter_AddRefs(serviceWorker));
|
|
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
serviceWorker.forget(aServiceWorker);
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
ServiceWorkerManager::InvalidateServiceWorkerRegistrationWorker(ServiceWorkerRegistrationInfo* aRegistration,
|
|
WhichServiceWorker aWhichOnes)
|
|
{
|
|
AssertIsOnMainThread();
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo =
|
|
GetDomainInfo(aRegistration->mScriptSpec);
|
|
|
|
if (domainInfo) {
|
|
nsTObserverArray<ServiceWorkerRegistration*>::ForwardIterator it(domainInfo->mServiceWorkerRegistrations);
|
|
while (it.HasMore()) {
|
|
nsRefPtr<ServiceWorkerRegistration> target = it.GetNext();
|
|
|
|
nsIURI* targetURI = target->GetDocumentURI();
|
|
nsCString path;
|
|
nsresult rv = targetURI->GetSpec(path);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
continue;
|
|
}
|
|
|
|
nsCString scope = FindScopeForPath(domainInfo->mOrderedScopes, path);
|
|
if (scope.IsEmpty() ||
|
|
!scope.Equals(aRegistration->mScope)) {
|
|
continue;
|
|
}
|
|
|
|
target->InvalidateWorkerReference(aWhichOnes);
|
|
}
|
|
}
|
|
}
|
|
|
|
namespace {
|
|
|
|
class MOZ_STACK_CLASS FilterRegistrationData
|
|
{
|
|
public:
|
|
FilterRegistrationData(nsTArray<uint64_t>* aDocuments,
|
|
ServiceWorkerRegistrationInfo* aRegistration)
|
|
: mDocuments(aDocuments),
|
|
mRegistration(aRegistration)
|
|
{
|
|
}
|
|
|
|
nsTArray<uint64_t>* mDocuments;
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> mRegistration;
|
|
};
|
|
|
|
static PLDHashOperator
|
|
EnumControlledDocuments(nsISupports* aKey,
|
|
ServiceWorkerRegistrationInfo* aRegistration,
|
|
void* aData)
|
|
{
|
|
FilterRegistrationData* data = static_cast<FilterRegistrationData*>(aData);
|
|
if (data->mRegistration != aRegistration) {
|
|
return PL_DHASH_NEXT;
|
|
}
|
|
nsCOMPtr<nsIDocument> document = do_QueryInterface(aKey);
|
|
if (!document || !document->GetInnerWindow()) {
|
|
return PL_DHASH_NEXT;
|
|
}
|
|
|
|
data->mDocuments->AppendElement(document->GetInnerWindow()->WindowID());
|
|
return PL_DHASH_NEXT;
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
void
|
|
ServiceWorkerManager::GetServicedClients(const nsCString& aScope,
|
|
nsTArray<uint64_t>* aControlledDocuments)
|
|
{
|
|
nsRefPtr<ServiceWorkerDomainInfo> domainInfo = GetDomainInfo(aScope);
|
|
nsRefPtr<ServiceWorkerRegistrationInfo> registration =
|
|
domainInfo->GetRegistration(aScope);
|
|
|
|
if (!registration) {
|
|
// The registration was removed, leave the array empty.
|
|
return;
|
|
}
|
|
|
|
FilterRegistrationData data(aControlledDocuments, registration);
|
|
|
|
domainInfo->mControlledDocuments.EnumerateRead(EnumControlledDocuments,
|
|
&data);
|
|
}
|
|
|
|
END_WORKERS_NAMESPACE
|