gecko-dev/dom/worklet/Worklet.cpp

392 строки
9.8 KiB
C++

/* -*- 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 "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