Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
- Manifest: netwerk/test/browser/browser.toml
"use strict";
const PROMPT_ALLOW_BUTTON = -1;
const PROMPT_NOT_NOW_BUTTON = 0;
const { HttpServer } = ChromeUtils.importESModule(
);
const baseURL = getRootDirectory(gTestPath).replace(
);
async function restorePermissions() {
info("Restoring permissions");
Services.obs.notifyObservers(null, "testonly-reload-permissions-from-disk");
Services.perms.removeAll();
}
add_setup(async function () {
await SpecialPowers.pushPrefEnv({
set: [
["permissions.manager.defaultsUrl", ""],
["network.websocket.delay-failed-reconnects", false],
["network.websocket.max-connections", 1000],
["network.lna.block_trackers", true],
["network.lna.blocking", true],
["network.http.rcwn.enabled", false],
],
});
Services.obs.notifyObservers(null, "testonly-reload-permissions-from-disk");
const server = new HttpServer();
server.start(21555);
registerServerHandlers(server);
registerCleanupFunction(async () => {
await restorePermissions();
await new Promise(resolve => {
server.stop(resolve);
});
});
});
requestLongerTimeout(10);
function clickDoorhangerButton(buttonIndex, browser, notificationID) {
let popup = PopupNotifications.getNotification(notificationID, browser);
let notification = popup?.owner?.panel?.childNodes?.[0];
ok(notification, "Notification popup is available");
if (buttonIndex === PROMPT_ALLOW_BUTTON) {
ok(true, "Triggering main action (allow)");
notification.button.doCommand();
} else {
ok(true, "Triggering secondary action (deny)");
notification.secondaryButton.doCommand();
}
}
function observeAndCheck(testType, rand, expectedStatus, message) {
return new Promise(resolve => {
const observer = {
observe(subject, topic) {
if (topic !== "http-on-stop-request") {
return;
}
let channel = subject.QueryInterface(Ci.nsIHttpChannel);
if (!channel || channel.URI.spec !== url) {
return;
}
is(channel.status, expectedStatus, message);
Services.obs.removeObserver(observer, "http-on-stop-request");
resolve();
},
};
Services.obs.addObserver(observer, "http-on-stop-request");
});
}
const testCases = [
{
type: "fetch",
allowStatus: Cr.NS_OK,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
{
type: "xhr",
allowStatus: Cr.NS_OK,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
{
type: "img",
allowStatus: Cr.NS_OK,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
{
type: "video",
allowStatus: Cr.NS_OK,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
{
type: "audio",
allowStatus: Cr.NS_OK,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
{
type: "iframe",
allowStatus: Cr.NS_OK,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
{
type: "script",
allowStatus: Cr.NS_OK,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
{
type: "font",
allowStatus: Cr.NS_OK,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
{
type: "websocket",
allowStatus: Cr.NS_ERROR_WEBSOCKET_CONNECTION_REFUSED,
denyStatus: Cr.NS_ERROR_LOCAL_NETWORK_ACCESS_DENIED,
},
];
function registerServerHandlers(server) {
server.registerPathHandler("/", (request, response) => {
const params = new URLSearchParams(request.queryString);
const type = params.get("type");
response.setHeader("Access-Control-Allow-Origin", "*", false);
switch (type) {
case "img":
response.setHeader("Content-Type", "image/gif", false);
response.setStatusLine(request.httpVersion, 200, "OK");
response.write(
atob("R0lGODlhAQABAIAAAAAAAP///ywAAAAAAQABAAACAUwAOw==")
);
break;
case "audio":
response.setHeader("Content-Type", "audio/wav", false);
response.setStatusLine(request.httpVersion, 200, "OK");
response.write(
atob("UklGRhYAAABXQVZFZm10IBAAAAABAAEAIlYAAESsAAACABAAZGF0YQAAAAA=")
);
break;
case "video":
response.setHeader("Content-Type", "video/mp4", false);
response.setStatusLine(request.httpVersion, 200, "OK");
response.write(
atob(
"GkXfo0AgQoaBAUL3gQFC8oEEQvOBCEKCQAR3ZWJtQoeBAkKFgQIYU4BnQI0VSalmQCgq17FAAw9CQE2AQAZ3aGFtbXlXQUAGd2hhbW15RIlACECPQAAAAAAAFlSua0AxrkAu14EBY8WBAZyBACK1nEADdW5khkAFVl9WUDglhohAA1ZQOIOBAeBABrCBCLqBCB9DtnVAIueBAKNAHIEAAIAwAQCdASoIAAgAAUAmJaQAA3AA/vz0AAA="
)
);
break;
default:
response.setHeader("Content-Type", "text/plain", false);
response.setStatusLine(request.httpVersion, 200, "OK");
response.write("hello");
}
});
}
async function runSingleTestCase(
test,
rand,
expectedStatus,
description,
userAction = null,
notificationID = null
) {
info(description);
const promise = observeAndCheck(test.type, rand, expectedStatus, description);
const tab = await BrowserTestUtils.openNewForegroundTab(
gBrowser,
`${baseURL}page_with_non_trackers.html?test=${test.type}&rand=${rand}`
);
if (userAction && notificationID) {
const buttonNum =
userAction === "allow" ? PROMPT_ALLOW_BUTTON : PROMPT_NOT_NOW_BUTTON;
await BrowserTestUtils.waitForEvent(PopupNotifications.panel, "popupshown");
clickDoorhangerButton(buttonNum, gBrowser.selectedBrowser, notificationID);
}
await promise;
gBrowser.removeTab(tab);
}
async function runPromptedLnaTest(test, overrideLabel, notificationID) {
const promptActions = ["allow", "deny"];
for (const userAction of promptActions) {
const rand = Math.random();
const expectedStatus =
userAction === "allow" ? test.allowStatus : test.denyStatus;
await runSingleTestCase(
test,
rand,
expectedStatus,
`LNA test (${overrideLabel}) for ${test.type} with user action: ${userAction}`,
userAction,
notificationID
);
// Wait some time for cache entry to be updated
// XXX(valentin) though this should not be necessary.
// eslint-disable-next-line mozilla/no-arbitrary-setTimeout
await new Promise(resolve => setTimeout(resolve, 300));
// Now run the test again with cached main document
await runSingleTestCase(
test,
rand,
expectedStatus,
`LNA test (${overrideLabel}) for ${test.type} with user action: ${userAction}`,
userAction,
notificationID
);
}
}
add_task(async function test_lna_prompt_behavior() {
// Non-LNA test: no prompt expected
for (const test of testCases) {
const rand = Math.random();
await runSingleTestCase(
test,
rand,
test.allowStatus,
`Non-LNA test for ${test.type}`
);
}
// Public -> Local test (localhost permission)
Services.prefs.setCharPref(
"network.lna.address_space.public.override",
"127.0.0.1:4443"
);
for (const test of testCases) {
await runPromptedLnaTest(test, "public", "localhost");
}
// Public -> Private (local-network permission)
Services.prefs.setCharPref(
"network.lna.address_space.private.override",
"127.0.0.1:21555"
);
for (const test of testCases) {
await runPromptedLnaTest(test, "private", "local-network");
}
Services.prefs.clearUserPref("network.lna.address_space.private.override");
});
add_task(async function test_lna_cancellation_during_prompt() {
info("Testing LNA cancellation during permission prompt");
// Disable RCWN but enable caching for this test
await SpecialPowers.pushPrefEnv({
set: [
["network.http.rcwn.enabled", false],
["browser.cache.disk.enable", true],
["browser.cache.memory.enable", true],
["network.lna.address_space.public.override", "127.0.0.1:4443"],
],
});
const testType = "fetch";
const rand1 = Math.random();
// Test 1: Cancel request during LNA prompt and verify proper cleanup
info(
"Step 1: Making request that will trigger LNA prompt, then cancelling it"
);
// Open tab and wait for LNA prompt
const tab1 = await BrowserTestUtils.openNewForegroundTab(
gBrowser,
`${baseURL}page_with_non_trackers.html?test=${testType}&rand=${rand1}`
);
// Wait for the LNA permission prompt to appear
await BrowserTestUtils.waitForEvent(PopupNotifications.panel, "popupshown");
info("LNA permission prompt appeared");
gBrowser.removeTab(tab1);
// Navigate to a new URL (which should cancel the pending request)
const tab2 = await BrowserTestUtils.openNewForegroundTab(
gBrowser,
`${baseURL}page_with_non_trackers.html?test=${testType}&rand=${rand1}`
);
info("Navigated to new URL, request should be cancelled");
// Wait for the navigation to complete
await BrowserTestUtils.waitForEvent(PopupNotifications.panel, "popupshown");
clickDoorhangerButton(
PROMPT_ALLOW_BUTTON,
gBrowser.selectedBrowser,
"localhost"
);
// Close the first tab now that we're done with it
gBrowser.removeTab(tab2);
// The main test objective is complete - we verified that cancellation
// during LNA prompt works without hanging channels. The navigation
// completed successfully, which means our fix is working correctly.
info(
"Test completed successfully - cancellation during LNA prompt handled correctly"
);
await SpecialPowers.popPrefEnv();
});