mirror of
https://repo.dactyloidae.xyz/Dactyloidae/UXP.git
synced 2026-09-06 15:58:39 +09:00
import FIREFOX_52_6_0esr_RELEASE from mozilla-esr52 hg repo
This commit is contained in:
Roy Tam
commit
dcd9973243
150858 changed files with 23884658 additions and 0 deletions
392
dom/worklet/Worklet.cpp
Normal file
392
dom/worklet/Worklet.cpp
Normal file
|
|
@ -0,0 +1,392 @@
|
|||
/* -*- 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 "Worklet.h"
|
||||
#include "WorkletGlobalScope.h"
|
||||
#include "mozilla/dom/WorkletBinding.h"
|
||||
#include "mozilla/dom/BlobBinding.h"
|
||||
#include "mozilla/dom/Fetch.h"
|
||||
#include "mozilla/dom/PromiseNativeHandler.h"
|
||||
#include "mozilla/dom/RegisterWorkletBindings.h"
|
||||
#include "mozilla/dom/Response.h"
|
||||
#include "mozilla/dom/ScriptSettings.h"
|
||||
#include "nsIInputStreamPump.h"
|
||||
#include "nsIThreadRetargetableRequest.h"
|
||||
#include "nsNetUtil.h"
|
||||
#include "nsScriptLoader.h"
|
||||
#include "xpcprivate.h"
|
||||
|
||||
namespace mozilla {
|
||||
namespace dom {
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// WorkletFetchHandler
|
||||
|
||||
class WorkletFetchHandler : public PromiseNativeHandler
|
||||
, public nsIStreamLoaderObserver
|
||||
{
|
||||
public:
|
||||
NS_DECL_ISUPPORTS
|
||||
|
||||
static already_AddRefed<Promise>
|
||||
Fetch(Worklet* aWorklet, const nsAString& aModuleURL, ErrorResult& aRv)
|
||||
{
|
||||
MOZ_ASSERT(aWorklet);
|
||||
|
||||
nsCOMPtr<nsIGlobalObject> global =
|
||||
do_QueryInterface(aWorklet->GetParentObject());
|
||||
MOZ_ASSERT(global);
|
||||
|
||||
RefPtr<Promise> promise = Promise::Create(global, aRv);
|
||||
if (NS_WARN_IF(aRv.Failed())) {
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
nsCOMPtr<nsPIDOMWindowInner> window = aWorklet->GetParentObject();
|
||||
MOZ_ASSERT(window);
|
||||
|
||||
nsCOMPtr<nsIDocument> doc;
|
||||
doc = window->GetExtantDoc();
|
||||
if (!doc) {
|
||||
promise->MaybeReject(NS_ERROR_FAILURE);
|
||||
return promise.forget();
|
||||
}
|
||||
|
||||
nsCOMPtr<nsIURI> baseURI = doc->GetBaseURI();
|
||||
nsCOMPtr<nsIURI> resolvedURI;
|
||||
nsresult rv = NS_NewURI(getter_AddRefs(resolvedURI), aModuleURL, nullptr, baseURI);
|
||||
if (NS_WARN_IF(NS_FAILED(rv))) {
|
||||
promise->MaybeReject(rv);
|
||||
return promise.forget();
|
||||
}
|
||||
|
||||
nsAutoCString spec;
|
||||
rv = resolvedURI->GetSpec(spec);
|
||||
if (NS_WARN_IF(NS_FAILED(rv))) {
|
||||
promise->MaybeReject(rv);
|
||||
return promise.forget();
|
||||
}
|
||||
|
||||
// Maybe we already have an handler for this URI
|
||||
{
|
||||
WorkletFetchHandler* handler = aWorklet->GetImportFetchHandler(spec);
|
||||
if (handler) {
|
||||
handler->AddPromise(promise);
|
||||
return promise.forget();
|
||||
}
|
||||
}
|
||||
|
||||
RequestOrUSVString request;
|
||||
request.SetAsUSVString().Rebind(aModuleURL.Data(), aModuleURL.Length());
|
||||
|
||||
RequestInit init;
|
||||
|
||||
RefPtr<Promise> fetchPromise = FetchRequest(global, request, init, aRv);
|
||||
if (NS_WARN_IF(aRv.Failed())) {
|
||||
promise->MaybeReject(aRv);
|
||||
return promise.forget();
|
||||
}
|
||||
|
||||
RefPtr<WorkletFetchHandler> handler =
|
||||
new WorkletFetchHandler(aWorklet, aModuleURL, promise);
|
||||
fetchPromise->AppendNativeHandler(handler);
|
||||
|
||||
aWorklet->AddImportFetchHandler(spec, handler);
|
||||
return promise.forget();
|
||||
}
|
||||
|
||||
virtual void
|
||||
ResolvedCallback(JSContext* aCx, JS::Handle<JS::Value> aValue) override
|
||||
{
|
||||
if (!aValue.isObject()) {
|
||||
RejectPromises(NS_ERROR_FAILURE);
|
||||
return;
|
||||
}
|
||||
|
||||
RefPtr<Response> response;
|
||||
nsresult rv = UNWRAP_OBJECT(Response, &aValue.toObject(), response);
|
||||
if (NS_WARN_IF(NS_FAILED(rv))) {
|
||||
RejectPromises(NS_ERROR_FAILURE);
|
||||
return;
|
||||
}
|
||||
|
||||
if (!response->Ok()) {
|
||||
RejectPromises(NS_ERROR_DOM_NETWORK_ERR);
|
||||
return;
|
||||
}
|
||||
|
||||
nsCOMPtr<nsIInputStream> inputStream;
|
||||
response->GetBody(getter_AddRefs(inputStream));
|
||||
if (!inputStream) {
|
||||
RejectPromises(NS_ERROR_DOM_NETWORK_ERR);
|
||||
return;
|
||||
}
|
||||
|
||||
nsCOMPtr<nsIInputStreamPump> pump;
|
||||
rv = NS_NewInputStreamPump(getter_AddRefs(pump), inputStream);
|
||||
if (NS_WARN_IF(NS_FAILED(rv))) {
|
||||
RejectPromises(rv);
|
||||
return;
|
||||
}
|
||||
|
||||
nsCOMPtr<nsIStreamLoader> loader;
|
||||
rv = NS_NewStreamLoader(getter_AddRefs(loader), this);
|
||||
if (NS_WARN_IF(NS_FAILED(rv))) {
|
||||
RejectPromises(rv);
|
||||
return;
|
||||
}
|
||||
|
||||
rv = pump->AsyncRead(loader, nullptr);
|
||||
if (NS_WARN_IF(NS_FAILED(rv))) {
|
||||
RejectPromises(rv);
|
||||
return;
|
||||
}
|
||||
|
||||
nsCOMPtr<nsIThreadRetargetableRequest> rr = do_QueryInterface(pump);
|
||||
if (rr) {
|
||||
nsCOMPtr<nsIEventTarget> sts =
|
||||
do_GetService(NS_STREAMTRANSPORTSERVICE_CONTRACTID);
|
||||
rv = rr->RetargetDeliveryTo(sts);
|
||||
if (NS_FAILED(rv)) {
|
||||
NS_WARNING("Failed to dispatch the nsIInputStreamPump to a IO thread.");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
NS_IMETHOD
|
||||
OnStreamComplete(nsIStreamLoader* aLoader, nsISupports* aContext,
|
||||
nsresult aStatus, uint32_t aStringLen,
|
||||
const uint8_t* aString) override
|
||||
{
|
||||
MOZ_ASSERT(NS_IsMainThread());
|
||||
|
||||
if (NS_FAILED(aStatus)) {
|
||||
RejectPromises(aStatus);
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
char16_t* scriptTextBuf;
|
||||
size_t scriptTextLength;
|
||||
nsresult rv =
|
||||
nsScriptLoader::ConvertToUTF16(nullptr, aString, aStringLen,
|
||||
NS_LITERAL_STRING("UTF-8"), nullptr,
|
||||
scriptTextBuf, scriptTextLength);
|
||||
if (NS_WARN_IF(NS_FAILED(rv))) {
|
||||
RejectPromises(rv);
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
// Moving the ownership of the buffer
|
||||
JS::SourceBufferHolder buffer(scriptTextBuf, scriptTextLength,
|
||||
JS::SourceBufferHolder::GiveOwnership);
|
||||
|
||||
AutoJSAPI jsapi;
|
||||
jsapi.Init();
|
||||
|
||||
RefPtr<WorkletGlobalScope> globalScope =
|
||||
mWorklet->GetOrCreateGlobalScope(jsapi.cx());
|
||||
MOZ_ASSERT(globalScope);
|
||||
|
||||
AutoEntryScript aes(globalScope, "Worklet");
|
||||
JSContext* cx = aes.cx();
|
||||
|
||||
JS::Rooted<JSObject*> globalObj(cx, globalScope->GetGlobalJSObject());
|
||||
|
||||
(void) new XPCWrappedNativeScope(cx, globalObj);
|
||||
|
||||
JS::CompileOptions compileOptions(cx);
|
||||
compileOptions.setIntroductionType("Worklet");
|
||||
compileOptions.setFileAndLine(NS_ConvertUTF16toUTF8(mURL).get(), 0);
|
||||
compileOptions.setVersion(JSVERSION_DEFAULT);
|
||||
compileOptions.setIsRunOnce(true);
|
||||
|
||||
// We only need the setNoScriptRval bit when compiling off-thread here,
|
||||
// since otherwise nsJSUtils::EvaluateString will set it up for us.
|
||||
compileOptions.setNoScriptRval(true);
|
||||
|
||||
JS::Rooted<JS::Value> unused(cx);
|
||||
if (!JS::Evaluate(cx, compileOptions, buffer, &unused)) {
|
||||
ErrorResult error;
|
||||
error.StealExceptionFromJSContext(cx);
|
||||
RejectPromises(error.StealNSResult());
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
// All done.
|
||||
ResolvePromises();
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
virtual void
|
||||
RejectedCallback(JSContext* aCx, JS::Handle<JS::Value> aValue) override
|
||||
{
|
||||
RejectPromises(NS_ERROR_DOM_NETWORK_ERR);
|
||||
}
|
||||
|
||||
private:
|
||||
WorkletFetchHandler(Worklet* aWorklet, const nsAString& aURL,
|
||||
Promise* aPromise)
|
||||
: mWorklet(aWorklet)
|
||||
, mStatus(ePending)
|
||||
, mErrorStatus(NS_OK)
|
||||
, mURL(aURL)
|
||||
{
|
||||
MOZ_ASSERT(aWorklet);
|
||||
MOZ_ASSERT(aPromise);
|
||||
|
||||
mPromises.AppendElement(aPromise);
|
||||
}
|
||||
|
||||
~WorkletFetchHandler()
|
||||
{}
|
||||
|
||||
void
|
||||
AddPromise(Promise* aPromise)
|
||||
{
|
||||
MOZ_ASSERT(aPromise);
|
||||
|
||||
switch (mStatus) {
|
||||
case ePending:
|
||||
mPromises.AppendElement(aPromise);
|
||||
return;
|
||||
|
||||
case eRejected:
|
||||
MOZ_ASSERT(NS_FAILED(mErrorStatus));
|
||||
aPromise->MaybeReject(mErrorStatus);
|
||||
return;
|
||||
|
||||
case eResolved:
|
||||
aPromise->MaybeResolveWithUndefined();
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
RejectPromises(nsresult aResult)
|
||||
{
|
||||
MOZ_ASSERT(mStatus == ePending);
|
||||
MOZ_ASSERT(NS_FAILED(aResult));
|
||||
|
||||
for (uint32_t i = 0; i < mPromises.Length(); ++i) {
|
||||
mPromises[i]->MaybeReject(aResult);
|
||||
}
|
||||
mPromises.Clear();
|
||||
|
||||
mStatus = eRejected;
|
||||
mErrorStatus = aResult;
|
||||
mWorklet = nullptr;
|
||||
}
|
||||
|
||||
void
|
||||
ResolvePromises()
|
||||
{
|
||||
MOZ_ASSERT(mStatus == ePending);
|
||||
|
||||
for (uint32_t i = 0; i < mPromises.Length(); ++i) {
|
||||
mPromises[i]->MaybeResolveWithUndefined();
|
||||
}
|
||||
mPromises.Clear();
|
||||
|
||||
mStatus = eResolved;
|
||||
mWorklet = nullptr;
|
||||
}
|
||||
|
||||
RefPtr<Worklet> mWorklet;
|
||||
nsTArray<RefPtr<Promise>> mPromises;
|
||||
|
||||
enum {
|
||||
ePending,
|
||||
eRejected,
|
||||
eResolved
|
||||
} mStatus;
|
||||
|
||||
nsresult mErrorStatus;
|
||||
|
||||
nsString mURL;
|
||||
};
|
||||
|
||||
NS_IMPL_ISUPPORTS(WorkletFetchHandler, nsIStreamLoaderObserver)
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Worklet
|
||||
|
||||
NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(Worklet, mWindow, mScope)
|
||||
NS_IMPL_CYCLE_COLLECTING_ADDREF(Worklet)
|
||||
NS_IMPL_CYCLE_COLLECTING_RELEASE(Worklet)
|
||||
|
||||
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(Worklet)
|
||||
NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
|
||||
NS_INTERFACE_MAP_ENTRY(nsISupports)
|
||||
NS_INTERFACE_MAP_END
|
||||
|
||||
Worklet::Worklet(nsPIDOMWindowInner* aWindow, nsIPrincipal* aPrincipal)
|
||||
: mWindow(aWindow)
|
||||
, mPrincipal(aPrincipal)
|
||||
{
|
||||
MOZ_ASSERT(aWindow);
|
||||
MOZ_ASSERT(aPrincipal);
|
||||
|
||||
#ifdef RELEASE_OR_BETA
|
||||
MOZ_CRASH("This code should not go to release/beta yet!");
|
||||
#endif
|
||||
}
|
||||
|
||||
Worklet::~Worklet()
|
||||
{}
|
||||
|
||||
JSObject*
|
||||
Worklet::WrapObject(JSContext* aCx, JS::Handle<JSObject*> aGivenProto)
|
||||
{
|
||||
return WorkletBinding::Wrap(aCx, this, aGivenProto);
|
||||
}
|
||||
|
||||
already_AddRefed<Promise>
|
||||
Worklet::Import(const nsAString& aModuleURL, ErrorResult& aRv)
|
||||
{
|
||||
return WorkletFetchHandler::Fetch(this, aModuleURL, aRv);
|
||||
}
|
||||
|
||||
WorkletGlobalScope*
|
||||
Worklet::GetOrCreateGlobalScope(JSContext* aCx)
|
||||
{
|
||||
if (!mScope) {
|
||||
mScope = new WorkletGlobalScope(mWindow);
|
||||
|
||||
JS::Rooted<JSObject*> global(aCx);
|
||||
NS_ENSURE_TRUE(mScope->WrapGlobalObject(aCx, mPrincipal, &global), nullptr);
|
||||
|
||||
JSAutoCompartment ac(aCx, global);
|
||||
|
||||
// Init Web IDL bindings
|
||||
if (!RegisterWorkletBindings(aCx, global)) {
|
||||
mScope = nullptr;
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
JS_FireOnNewGlobalObject(aCx, global);
|
||||
}
|
||||
|
||||
return mScope;
|
||||
}
|
||||
|
||||
WorkletFetchHandler*
|
||||
Worklet::GetImportFetchHandler(const nsACString& aURI)
|
||||
{
|
||||
return mImportHandlers.GetWeak(aURI);
|
||||
}
|
||||
|
||||
void
|
||||
Worklet::AddImportFetchHandler(const nsACString& aURI,
|
||||
WorkletFetchHandler* aHandler)
|
||||
{
|
||||
MOZ_ASSERT(aHandler);
|
||||
MOZ_ASSERT(!mImportHandlers.GetWeak(aURI));
|
||||
|
||||
mImportHandlers.Put(aURI, aHandler);
|
||||
}
|
||||
|
||||
} // dom namespace
|
||||
} // mozilla namespace
|
||||
Loading…
Add table
Add a link
Reference in a new issue