forked from mirrors/gecko-dev
fa6ffb7af2
New XPCOM idl interfaces to define the components used to: - represent an API request (call to API methods, add/remove API event listener, get the value of an API property getter) - define the API request handler defined by privileged JS code running on the main thread New ExtensionAPIBase shared stub methods and helper methods to create and forward the API requests from the worker to the main thread and translate the results got from the API request handler in the value to return or resolve (or the errors to be raised) to the extension code running on the worker thread. Depends on D80604 Differential Revision: https://phabricator.services.mozilla.com/D75311
127 lines
3.7 KiB
C++
127 lines
3.7 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2; -*- */
|
|
/* 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/. */
|
|
|
|
#ifndef mozilla_ExtensionPolicyService_h
|
|
#define mozilla_ExtensionPolicyService_h
|
|
|
|
#include "mozilla/MemoryReporting.h"
|
|
#include "mozilla/extensions/WebExtensionPolicy.h"
|
|
#include "mozIExtensionProcessScript.h"
|
|
#include "nsCOMPtr.h"
|
|
#include "nsCycleCollectionParticipant.h"
|
|
#include "nsHashKeys.h"
|
|
#include "nsIAddonPolicyService.h"
|
|
#include "nsAtom.h"
|
|
#include "nsIMemoryReporter.h"
|
|
#include "nsIObserver.h"
|
|
#include "nsIObserverService.h"
|
|
#include "nsISupports.h"
|
|
#include "nsPointerHashKeys.h"
|
|
#include "nsRefPtrHashtable.h"
|
|
#include "nsTHashSet.h"
|
|
|
|
class nsIChannel;
|
|
class nsIObserverService;
|
|
|
|
class nsIPIDOMWindowInner;
|
|
class nsIPIDOMWindowOuter;
|
|
|
|
namespace mozilla {
|
|
namespace dom {
|
|
class Promise;
|
|
} // namespace dom
|
|
namespace extensions {
|
|
class DocInfo;
|
|
class DocumentObserver;
|
|
class WebExtensionContentScript;
|
|
} // namespace extensions
|
|
|
|
using extensions::DocInfo;
|
|
using extensions::WebExtensionPolicy;
|
|
|
|
class ExtensionPolicyService final : public nsIAddonPolicyService,
|
|
public nsIObserver,
|
|
public nsIMemoryReporter {
|
|
public:
|
|
NS_DECL_CYCLE_COLLECTION_CLASS_AMBIGUOUS(ExtensionPolicyService,
|
|
nsIAddonPolicyService)
|
|
NS_DECL_CYCLE_COLLECTING_ISUPPORTS
|
|
NS_DECL_NSIADDONPOLICYSERVICE
|
|
NS_DECL_NSIOBSERVER
|
|
NS_DECL_NSIMEMORYREPORTER
|
|
|
|
static mozIExtensionProcessScript& ProcessScript();
|
|
|
|
static ExtensionPolicyService& GetSingleton();
|
|
|
|
static already_AddRefed<ExtensionPolicyService> GetInstance() {
|
|
return do_AddRef(&GetSingleton());
|
|
}
|
|
|
|
WebExtensionPolicy* GetByID(const nsAtom* aAddonId) {
|
|
return mExtensions.GetWeak(aAddonId);
|
|
}
|
|
|
|
WebExtensionPolicy* GetByID(const nsAString& aAddonId) {
|
|
RefPtr<nsAtom> atom = NS_AtomizeMainThread(aAddonId);
|
|
return GetByID(atom);
|
|
}
|
|
|
|
WebExtensionPolicy* GetByURL(const extensions::URLInfo& aURL);
|
|
|
|
WebExtensionPolicy* GetByHost(const nsACString& aHost) const {
|
|
return mExtensionHosts.GetWeak(aHost);
|
|
}
|
|
|
|
void GetAll(nsTArray<RefPtr<WebExtensionPolicy>>& aResult);
|
|
|
|
bool RegisterExtension(WebExtensionPolicy& aPolicy);
|
|
bool UnregisterExtension(WebExtensionPolicy& aPolicy);
|
|
|
|
bool RegisterObserver(extensions::DocumentObserver& aPolicy);
|
|
bool UnregisterObserver(extensions::DocumentObserver& aPolicy);
|
|
|
|
bool UseRemoteExtensions() const;
|
|
bool IsExtensionProcess() const;
|
|
|
|
nsresult InjectContentScripts(WebExtensionPolicy* aExtension);
|
|
|
|
protected:
|
|
virtual ~ExtensionPolicyService();
|
|
|
|
private:
|
|
ExtensionPolicyService();
|
|
|
|
void RegisterObservers();
|
|
void UnregisterObservers();
|
|
|
|
void CheckRequest(nsIChannel* aChannel);
|
|
void CheckDocument(dom::Document* aDocument);
|
|
|
|
void CheckContentScripts(const DocInfo& aDocInfo, bool aIsPreload);
|
|
|
|
already_AddRefed<dom::Promise> ExecuteContentScript(
|
|
nsPIDOMWindowInner* aWindow,
|
|
extensions::WebExtensionContentScript& aScript);
|
|
|
|
RefPtr<dom::Promise> ExecuteContentScripts(
|
|
JSContext* aCx, nsPIDOMWindowInner* aWindow,
|
|
const nsTArray<RefPtr<extensions::WebExtensionContentScript>>& aScripts);
|
|
|
|
nsRefPtrHashtable<nsPtrHashKey<const nsAtom>, WebExtensionPolicy> mExtensions;
|
|
nsRefPtrHashtable<nsCStringHashKey, WebExtensionPolicy> mExtensionHosts;
|
|
|
|
nsRefPtrHashtable<nsPtrHashKey<const extensions::DocumentObserver>,
|
|
extensions::DocumentObserver>
|
|
mObservers;
|
|
|
|
nsCOMPtr<nsIObserverService> mObs;
|
|
|
|
nsString mDefaultCSP;
|
|
};
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif // mozilla_ExtensionPolicyService_h
|