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
|
'use strict';
{
let runningScripts = new Map();
var RequestUtil = {
async executeOnStart(request, details) {
let {requestId, tabId, frameId} = request;
details = Object.assign({
runAt: "document_start",
frameId,
}, details);
browser.tabs.executeScript(tabId, details);
return;
let filter = browser.webRequest.filterResponseData(requestId);
filter.onstart = event => {
browser.tabs.executeScript(tabId, details);
debug("Execute on start", details);
filter.write(new Uint8Array());
};
filter.ondata = event => {
filter.write(event.data);
filter.disconnect();
}
},
async executeOnStartCS(request, details) {
let {url, requestId, tabId, frameId} = request;
let urlObj = new URL(url);
if (urlObj.hash || urlObj.port || urlObj.username) {
urlObj.hash = urlObj.port = urlObj.username = "";
url = urlObj.toString();
}
let wr = browser.webRequest;
let filter = {
urls: [`${urlObj.origin}/*`],
types: ["main_frame", "sub_frame", "object"]
};
let finalize;
let cleanup = r => {
if (cleanup && r.requestId === requestId) {
wr.onCompleted.removeListener(cleanup);
wr.onErrorOccurred.removeListener(cleanup);
cleanup = null;
if (finalize) {
finalize();
}
}
};
wr.onCompleted.addListener(cleanup, filter);
wr.onErrorOccurred.addListener(cleanup, filter);
details = Object.assign({
runAt: "document_start",
frameId,
}, details);
if (browser.contentScripts) {
let js = [{}];
if (details.file) js[0].file = details.file;
else if (details.code) js[0].code = details.code;
let settings = {
"runAt": details.runAt,
js,
matches: [url],
allFrames: frameId !== 0,
}
// let's try to avoid duplicates
let key = JSON.stringify(settings);
if (runningScripts.has(key)) {
let scriptRef = runningScripts.get(key);
scriptRef.count++;
return;
}
if (settings.allFrames) {
// let's check whether the same script is registered for top frames:
// if it is, let's unregister it first to avoid duplicates
settings.allFrames = false;
let topKey = JSON.stringify(settings);
settings.allFrames = true;
if (runningScripts.has(topKey)) {
let topScript = runningScripts.get(topKey);
try {
topScript.unregister();
} catch (e) {
error(e);
} finally {
runningScripts.delete(topKey);
}
}
}
let script = await browser.contentScripts.register(settings);
debug("Content script %o registered.", settings);
finalize = () => {
debug("Finalizing content script %o...", settings);
try {
script.unregister();
runningScripts.delete(key);
debug("Content script %o unregistered!", settings);
} finally {
finalize = null;
}
}
runningScripts.set(key, script);
if (!cleanup) { // the request has already been interrupted
finalize();
}
return;
}
function listener(r) {
if (r.requestId === requestId) {
browser.tabs.executeScript(tabId, details);
finalize();
finalize = null;
}
}
finalize = () => {
wr.onResponseStarted.removeListener(listener);
}
wr.onResponseStarted.addListener(listener, filter);
debug("Executing %o", details);
},
}
}
|