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
|
"use strict";
var { WebRequest } = ChromeUtils.importESModule(
"resource://gre/modules/WebRequest.sys.mjs"
);
var { ExtensionParent } = ChromeUtils.importESModule(
"resource://gre/modules/ExtensionParent.sys.mjs"
);
const server = createHttpServer({ hosts: ["example.com"] });
server.registerPathHandler("/", (request, response) => {
response.setStatusLine(request.httpVersion, 200, "OK");
if (request.hasHeader("Cookie")) {
let value = request.getHeader("Cookie");
if (value == "blinky=1") {
response.setHeader("Set-Cookie", "dinky=1", false);
}
response.write("cookie-present");
} else {
response.setHeader("Set-Cookie", "foopy=1", false);
response.write("cookie-not-present");
}
});
const URL = "http://example.com/";
var countBefore = 0;
var countAfter = 0;
function onBeforeSendHeaders(details) {
if (details.url != URL) {
return undefined;
}
countBefore++;
info(`onBeforeSendHeaders ${details.url}`);
let found = false;
let headers = [];
for (let { name, value } of details.requestHeaders) {
info(`Saw header ${name} '${value}'`);
if (name == "Cookie") {
equal(value, "foopy=1", "Cookie is correct");
headers.push({ name, value: "blinky=1" });
found = true;
} else {
headers.push({ name, value });
}
}
ok(found, "Saw cookie header");
equal(countBefore, 1, "onBeforeSendHeaders hit once");
return { requestHeaders: headers };
}
function onResponseStarted(details) {
if (details.url != URL) {
return;
}
countAfter++;
info(`onResponseStarted ${details.url}`);
let found = false;
for (let { name, value } of details.responseHeaders) {
info(`Saw header ${name} '${value}'`);
if (name == "set-cookie") {
equal(value, "dinky=1", "Cookie is correct");
found = true;
}
}
ok(found, "Saw cookie header");
equal(countAfter, 1, "onResponseStarted hit once");
}
add_task(async function setup() {
// When WebRequest.jsm is used directly instead of through ext-webRequest.js,
// ExtensionParent.apiManager is not automatically initialized. Do it here.
await ExtensionParent.apiManager.lazyInit();
});
add_task(async function filter_urls() {
// First load the URL so that we set cookie foopy=1.
let contentPage = await ExtensionTestUtils.loadContentPage(URL);
await contentPage.close();
// Now load with WebRequest set up.
WebRequest.onBeforeSendHeaders.addListener(onBeforeSendHeaders, null, [
"blocking",
"requestHeaders",
]);
WebRequest.onResponseStarted.addListener(onResponseStarted, null, [
"responseHeaders",
]);
contentPage = await ExtensionTestUtils.loadContentPage(URL);
await contentPage.close();
WebRequest.onBeforeSendHeaders.removeListener(onBeforeSendHeaders);
WebRequest.onResponseStarted.removeListener(onResponseStarted);
});
|
