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
|
"use strict";
const server = createHttpServer();
const gServerUrl = `http://localhost:${server.identity.primaryPort}`;
server.registerPathHandler("/dummy", (request, response) => {
response.setStatusLine(request.httpVersion, 200, "OK");
response.write("ok");
});
add_task(async function test_redirect_property() {
function background(serverUrl) {
browser.webRequest.onBeforeRequest.addListener(
() => {
return { redirectUrl: `${serverUrl}/dummy` };
},
{ urls: ["*://localhost/*"] },
["blocking"]
);
}
let ext = ExtensionTestUtils.loadExtension({
manifest: {
browser_specific_settings: { gecko: { id: "redirect@test" } },
permissions: ["webRequest", "webRequestBlocking", "<all_urls>"],
},
background: `(${background})("${gServerUrl}")`,
});
await ext.startup();
let data = await new Promise(resolve => {
let ssm = Services.scriptSecurityManager;
let channel = NetUtil.newChannel({
uri: `${gServerUrl}/redirect`,
loadingPrincipal:
ssm.createContentPrincipalFromOrigin("http://localhost"),
contentPolicyType: Ci.nsIContentPolicy.TYPE_XMLHTTPREQUEST,
securityFlags: Ci.nsILoadInfo.SEC_ALLOW_CROSS_ORIGIN_SEC_CONTEXT_IS_NULL,
});
channel.asyncOpen({
QueryInterface: ChromeUtils.generateQI(["nsIStreamListener"]),
onStartRequest(request) {},
onStopRequest(request, statusCode) {
let properties = request.QueryInterface(Ci.nsIPropertyBag);
let id = properties.getProperty("redirectedByExtension");
resolve({ id, url: request.QueryInterface(Ci.nsIChannel).URI.spec });
},
onDataAvailable() {},
});
});
Assert.equal(`${gServerUrl}/dummy`, data.url, "request redirected");
Assert.equal(
ext.id,
data.id,
"extension id attached to channel property bag"
);
await ext.unload();
});
|
