Source code
Revision control
Copy as Markdown
Other Tools
/* 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
#include "ExtensionAPIRequest.h"
#include "mozilla/dom/ClientInfo.h"
#include "mozilla/extensions/WebExtensionPolicy.h"
#include "mozilla/ipc/BackgroundUtils.h" // PrincipalInfoToPrincipal
namespace mozilla {
namespace extensions {
// mozIExtensionServiceWorkerInfo
NS_IMPL_ISUPPORTS(ExtensionServiceWorkerInfo, mozIExtensionServiceWorkerInfo)
NS_IMETHODIMP
ExtensionServiceWorkerInfo::GetPrincipal(nsIPrincipal** aPrincipal) {
MOZ_ASSERT(NS_IsMainThread());
NS_ENSURE_ARG_POINTER(aPrincipal);
auto principalOrErr = PrincipalInfoToPrincipal(mClientInfo.PrincipalInfo());
if (NS_WARN_IF(principalOrErr.isErr())) {
return NS_ERROR_UNEXPECTED;
}
nsCOMPtr<nsIPrincipal> principal = principalOrErr.unwrap();
principal.forget(aPrincipal);
return NS_OK;
}
NS_IMETHODIMP
ExtensionServiceWorkerInfo::GetScriptURL(nsAString& aScriptURL) {
MOZ_ASSERT(NS_IsMainThread());
aScriptURL = NS_ConvertUTF8toUTF16(mClientInfo.URL());
return NS_OK;
}
NS_IMETHODIMP
ExtensionServiceWorkerInfo::GetClientInfoId(nsAString& aClientInfoId) {
MOZ_ASSERT(NS_IsMainThread());
aClientInfoId = NS_ConvertUTF8toUTF16(mClientInfo.Id().ToString().get());
return NS_OK;
}
NS_IMETHODIMP
ExtensionServiceWorkerInfo::GetDescriptorId(uint64_t* aDescriptorId) {
MOZ_ASSERT(NS_IsMainThread());
*aDescriptorId = mDescriptorId;
return NS_OK;
}
// mozIExtensionAPIRequest
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(ExtensionAPIRequest)
NS_INTERFACE_MAP_ENTRY(mozIExtensionAPIRequest)
NS_INTERFACE_MAP_ENTRY(nsISupports)
NS_INTERFACE_MAP_END
NS_IMPL_CYCLE_COLLECTION_CLASS(ExtensionAPIRequest)
NS_IMPL_CYCLE_COLLECTING_ADDREF(ExtensionAPIRequest)
NS_IMPL_CYCLE_COLLECTING_RELEASE(ExtensionAPIRequest)
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(ExtensionAPIRequest)
NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mEventListener)
NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mSWInfo)
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END
NS_IMPL_CYCLE_COLLECTION_TRACE_BEGIN(ExtensionAPIRequest)
NS_IMPL_CYCLE_COLLECTION_TRACE_JS_MEMBER_CALLBACK(mArgs)
NS_IMPL_CYCLE_COLLECTION_TRACE_JS_MEMBER_CALLBACK(mNormalizedArgs)
NS_IMPL_CYCLE_COLLECTION_TRACE_JS_MEMBER_CALLBACK(mStack)
NS_IMPL_CYCLE_COLLECTION_TRACE_END
NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(ExtensionAPIRequest)
NS_IMPL_CYCLE_COLLECTION_UNLINK(mEventListener)
NS_IMPL_CYCLE_COLLECTION_UNLINK(mSWInfo)
tmp->mStack.setUndefined();
tmp->mArgs.setUndefined();
tmp->mNormalizedArgs.setUndefined();
NS_IMPL_CYCLE_COLLECTION_UNLINK_END
ExtensionAPIRequest::ExtensionAPIRequest(
const mozIExtensionAPIRequest::RequestType aRequestType,
const ExtensionAPIRequestTarget& aRequestTarget) {
MOZ_ASSERT(NS_IsMainThread());
mRequestType = aRequestType;
mRequestTarget = aRequestTarget;
mozilla::HoldJSObjects(this);
}
void ExtensionAPIRequest::Init(Maybe<dom::ClientInfo>& aSWClientInfo,
const uint64_t aSWDescriptorId,
JS::Handle<JS::Value> aRequestArgs,
JS::Handle<JS::Value> aCallerStack) {
MOZ_ASSERT(NS_IsMainThread());
mSWClientInfo = aSWClientInfo;
mSWDescriptorId = aSWDescriptorId;
mArgs.set(aRequestArgs);
mStack.set(aCallerStack);
mNormalizedArgs.setUndefined();
}
NS_IMETHODIMP
ExtensionAPIRequest::ToString(nsACString& aResult) {
aResult.Truncate();
nsAutoCString requestType;
nsAutoCString apiNamespace;
nsAutoCString apiName;
GetRequestType(requestType);
GetApiNamespace(apiNamespace);
GetApiName(apiName);
if (mRequestTarget.mObjectType.IsEmpty()) {
aResult.AppendPrintf("[ExtensionAPIRequest %s %s.%s]", requestType.get(),
apiNamespace.get(), apiName.get());
} else {
nsAutoCString objectType;
nsAutoCString objectId;
GetApiObjectType(objectType);
GetApiObjectId(objectId);
aResult.AppendPrintf("[ExtensionAPIRequest %s %s.%s.%s (%s)]",
requestType.get(), apiNamespace.get(),
objectType.get(), apiName.get(), objectId.get());
}
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetRequestType(nsACString& aRequestTypeName) {
MOZ_ASSERT(NS_IsMainThread());
switch (mRequestType) {
case mozIExtensionAPIRequest::RequestType::CALL_FUNCTION:
aRequestTypeName = "callFunction"_ns;
break;
case mozIExtensionAPIRequest::RequestType::CALL_FUNCTION_NO_RETURN:
aRequestTypeName = "callFunctionNoReturn"_ns;
break;
case mozIExtensionAPIRequest::RequestType::CALL_FUNCTION_ASYNC:
aRequestTypeName = "callAsyncFunction"_ns;
break;
case mozIExtensionAPIRequest::RequestType::ADD_LISTENER:
aRequestTypeName = "addListener"_ns;
break;
case mozIExtensionAPIRequest::RequestType::REMOVE_LISTENER:
aRequestTypeName = "removeListener"_ns;
break;
case mozIExtensionAPIRequest::RequestType::GET_PROPERTY:
aRequestTypeName = "getProperty"_ns;
break;
default:
return NS_ERROR_UNEXPECTED;
}
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetApiNamespace(nsACString& aApiNamespace) {
MOZ_ASSERT(NS_IsMainThread());
aApiNamespace.Assign(NS_ConvertUTF16toUTF8(mRequestTarget.mNamespace));
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetApiName(nsACString& aApiName) {
MOZ_ASSERT(NS_IsMainThread());
aApiName.Assign(NS_ConvertUTF16toUTF8(mRequestTarget.mMethod));
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetApiObjectType(nsACString& aApiObjectType) {
MOZ_ASSERT(NS_IsMainThread());
aApiObjectType.Assign(NS_ConvertUTF16toUTF8(mRequestTarget.mObjectType));
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetApiObjectId(nsACString& aApiObjectId) {
MOZ_ASSERT(NS_IsMainThread());
aApiObjectId.Assign(NS_ConvertUTF16toUTF8(mRequestTarget.mObjectId));
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetArgs(JSContext* aCx,
JS::MutableHandle<JS::Value> aRetval) {
MOZ_ASSERT(NS_IsMainThread());
aRetval.set(mArgs);
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetNormalizedArgs(JSContext* aCx,
JS::MutableHandle<JS::Value> aRetval) {
MOZ_ASSERT(NS_IsMainThread());
aRetval.set(mNormalizedArgs);
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::SetNormalizedArgs(JSContext* aCx,
JS::Handle<JS::Value> aNormalizedArgs) {
MOZ_ASSERT(NS_IsMainThread());
mNormalizedArgs.set(aNormalizedArgs);
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetCallerSavedFrame(
JSContext* aCx, JS::MutableHandle<JS::Value> aSavedFrame) {
MOZ_ASSERT(NS_IsMainThread());
aSavedFrame.set(mStack);
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetServiceWorkerInfo(
mozIExtensionServiceWorkerInfo** aSWInfo) {
MOZ_ASSERT(NS_IsMainThread());
NS_ENSURE_ARG_POINTER(aSWInfo);
if (mSWClientInfo.isSome() && !mSWInfo) {
mSWInfo = new ExtensionServiceWorkerInfo(*mSWClientInfo, mSWDescriptorId);
}
NS_IF_ADDREF(*aSWInfo = mSWInfo);
return NS_OK;
}
NS_IMETHODIMP
ExtensionAPIRequest::GetEventListener(mozIExtensionEventListener** aListener) {
MOZ_ASSERT(NS_IsMainThread());
NS_ENSURE_ARG_POINTER(aListener);
NS_IF_ADDREF(*aListener = mEventListener);
return NS_OK;
}
} // namespace extensions
} // namespace mozilla