Source code
Revision control
Copy as Markdown
Other Tools
Test Info: Warnings
- This test gets skipped with pattern: os == 'win' && socketprocess_networking && fission OR os == 'mac' && socketprocess_networking && fission OR os == 'mac' && debug OR os == 'linux' && socketprocess_networking OR os == 'android' && debug
- Manifest: toolkit/components/extensions/test/xpcshell/xpcshell-remote.toml includes toolkit/components/extensions/test/xpcshell/xpcshell-common.toml
- Manifest: toolkit/components/extensions/test/xpcshell/xpcshell.toml includes toolkit/components/extensions/test/xpcshell/xpcshell-common.toml
"use strict";
const HOSTS = new Set(["example.com", "example.org"]);
const server = createHttpServer({ hosts: HOSTS });
server.registerPathHandler("/return_headers.sjs", (request, response) => {
response.setHeader("Content-Type", "text/plain", false);
let headers = {};
for (let { data: header } of request.headers) {
headers[header] = request.getHeader(header);
}
response.write(JSON.stringify(headers));
});
server.registerPathHandler("/dummy", (request, response) => {
response.setStatusLine(request.httpVersion, 200, "OK");
response.write("ok");
});
function getExtension(permission = "<all_urls>") {
return ExtensionTestUtils.loadExtension({
manifest: {
permissions: ["webRequest", "webRequestBlocking", permission],
},
background() {
browser.webRequest.onBeforeSendHeaders.addListener(
details => {
details.requestHeaders.push({ name: "Host", value: "example.org" });
return { requestHeaders: details.requestHeaders };
},
{ urls: ["<all_urls>"] },
["blocking", "requestHeaders"]
);
},
});
}
add_task(async function test_host_header_accepted() {
let extension = getExtension();
await extension.startup();
let headers = JSON.parse(
await ExtensionTestUtils.fetch(
FETCH_ORIGIN,
`${BASE_URL}/return_headers.sjs`
)
);
equal(headers.host, "example.org", "Host header was set on request");
await extension.unload();
});
add_task(async function test_host_header_denied() {
let extension = getExtension(`${BASE_URL}/`);
await extension.startup();
let headers = JSON.parse(
await ExtensionTestUtils.fetch(
FETCH_ORIGIN,
`${BASE_URL}/return_headers.sjs`
)
);
equal(headers.host, "example.com", "Host header was not set on request");
await extension.unload();
});
add_task(async function test_host_header_restricted() {
Services.prefs.setCharPref(
"extensions.webextensions.restrictedDomains",
"example.org"
);
registerCleanupFunction(() => {
Services.prefs.clearUserPref("extensions.webextensions.restrictedDomains");
});
let extension = getExtension();
await extension.startup();
let headers = JSON.parse(
await ExtensionTestUtils.fetch(
FETCH_ORIGIN,
`${BASE_URL}/return_headers.sjs`
)
);
equal(headers.host, "example.com", "Host header was not set on request");
await extension.unload();
});