1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
/* -*- 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 "nsCOMPtr.h"
#include "nsCycleCollectionParticipant.h"
#include "nsHashKeys.h"
#include "nsIAddonPolicyService.h"
#include "nsAtom.h"
#include "nsIDOMEventListener.h"
#include "nsIMemoryReporter.h"
#include "nsIObserver.h"
#include "nsIObserverService.h"
#include "nsISupports.h"
#include "nsPointerHashKeys.h"
#include "nsRefPtrHashtable.h"
#include "nsTHashtable.h"
class nsIChannel;
class nsIObserverService;
class nsIPIDOMWindowInner;
class nsIPIDOMWindowOuter;
namespace mozilla {
namespace dom {
class ContentFrameMessageManager;
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 nsIDOMEventListener,
public nsIMemoryReporter {
public:
NS_DECL_CYCLE_COLLECTION_CLASS_AMBIGUOUS(ExtensionPolicyService,
nsIAddonPolicyService)
NS_DECL_CYCLE_COLLECTING_ISUPPORTS
NS_DECL_NSIADDONPOLICYSERVICE
NS_DECL_NSIOBSERVER
NS_DECL_NSIDOMEVENTLISTENER
NS_DECL_NSIMEMORYREPORTER
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;
nsTHashtable<nsRefPtrHashKey<dom::ContentFrameMessageManager>>
mMessageManagers;
nsRefPtrHashtable<nsPtrHashKey<const extensions::DocumentObserver>,
extensions::DocumentObserver>
mObservers;
nsCOMPtr<nsIObserverService> mObs;
nsString mBaseCSP;
nsString mDefaultCSP;
};
} // namespace mozilla
#endif // mozilla_ExtensionPolicyService_h
|
