diff options
Diffstat (limited to '')
16 files changed, 589 insertions, 0 deletions
diff --git a/toolkit/mozapps/downloads/tests/browser/browser.ini b/toolkit/mozapps/downloads/tests/browser/browser.ini new file mode 100644 index 0000000000..da1cf956e0 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/browser.ini @@ -0,0 +1,24 @@ +[DEFAULT] +support-files = + unknownContentType_dialog_layout_data.pif + unknownContentType_dialog_layout_data.pif^headers^ + unknownContentType_dialog_layout_data.txt + unknownContentType_dialog_layout_data.txt^headers^ + head.js + +[browser_save_wrongextension.js] +[browser_unknownContentType_blob.js] +[browser_unknownContentType_delayedbutton.js] +skip-if = + os == "linux" && bits == 64 && debug # Bug 1747285 + os == "linux" && fission && tsan # Bug 1747285 +[browser_unknownContentType_dialog_layout.js] +[browser_unknownContentType_extension.js] +support-files = + unknownContentType.EXE + unknownContentType.EXE^headers^ +[browser_unknownContentType_policy.js] +skip-if = os != 'win' # jnlp file are not considered executable on macOS or Linux +support-files = + example.jnlp + example.jnlp^headers^ diff --git a/toolkit/mozapps/downloads/tests/browser/browser_save_wrongextension.js b/toolkit/mozapps/downloads/tests/browser/browser_save_wrongextension.js new file mode 100644 index 0000000000..ba1b6cde5c --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/browser_save_wrongextension.js @@ -0,0 +1,98 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +"use strict"; + +let url = + "data:text/html,<a id='link' href='http://localhost:8000/thefile.js'>Link</a>"; + +let MockFilePicker = SpecialPowers.MockFilePicker; +MockFilePicker.init(window); + +let httpServer = null; + +add_task(async function test() { + const { HttpServer } = ChromeUtils.import( + "resource://testing-common/httpd.js" + ); + + httpServer = new HttpServer(); + httpServer.start(8000); + + function handleRequest(aRequest, aResponse) { + aResponse.setStatusLine(aRequest.httpVersion, 200); + aResponse.setHeader("Content-Type", "text/plain"); + aResponse.write("Some Text"); + } + httpServer.registerPathHandler("/thefile.js", handleRequest); + + let tab = await BrowserTestUtils.openNewForegroundTab(gBrowser, url); + + let popupShownPromise = BrowserTestUtils.waitForEvent(document, "popupshown"); + + await BrowserTestUtils.synthesizeMouseAtCenter( + "#link", + { type: "contextmenu", button: 2 }, + gBrowser.selectedBrowser + ); + await popupShownPromise; + + let tempDir = createTemporarySaveDirectory(); + let destFile; + + MockFilePicker.displayDirectory = tempDir; + MockFilePicker.showCallback = fp => { + let fileName = fp.defaultString; + destFile = tempDir.clone(); + destFile.append(fileName); + + MockFilePicker.setFiles([destFile]); + MockFilePicker.filterIndex = 1; // kSaveAsType_URL + }; + + let transferCompletePromise = new Promise(resolve => { + mockTransferCallback = resolve; + mockTransferRegisterer.register(); + }); + + registerCleanupFunction(function () { + mockTransferRegisterer.unregister(); + tempDir.remove(true); + }); + + document.getElementById("context-savelink").doCommand(); + + let contextMenu = document.getElementById("contentAreaContextMenu"); + let popupHiddenPromise = BrowserTestUtils.waitForEvent( + document, + "popuphidden" + ); + contextMenu.hidePopup(); + await popupHiddenPromise; + + await transferCompletePromise; + + is(destFile.leafName, "thefile.js", "filename extension is not modified"); + + BrowserTestUtils.removeTab(tab); +}); + +add_task(async () => { + MockFilePicker.cleanup(); + await new Promise(resolve => httpServer.stop(resolve)); +}); + +Services.scriptloader.loadSubScript( + "chrome://mochitests/content/browser/toolkit/content/tests/browser/common/mockTransfer.js", + this +); + +function createTemporarySaveDirectory() { + let saveDir = Services.dirsvc.get("TmpD", Ci.nsIFile); + saveDir.append("testsavedir"); + if (!saveDir.exists()) { + saveDir.create(Ci.nsIFile.DIRECTORY_TYPE, 0o755); + } + return saveDir; +} diff --git a/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_blob.js b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_blob.js new file mode 100644 index 0000000000..7b3900f46f --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_blob.js @@ -0,0 +1,118 @@ +/* Any copyright is dedicated to the Public Domain. + http://creativecommons.org/publicdomain/zero/1.0/ */ + +"use strict"; + +const UCT_URI = "chrome://mozapps/content/downloads/unknownContentType.xhtml"; + +async function promiseDownloadFinished(list) { + return new Promise(resolve => { + list.addView({ + onDownloadChanged(download) { + info("Download changed!"); + if (download.succeeded || download.error) { + info("Download succeeded or errored"); + list.removeView(this); + resolve(download); + } + }, + }); + }); +} + +/** + * Check that both in the download "what do you want to do with this file" + * dialog and in the about:downloads download list, we represent blob URL + * download sources using the principal (origin) that generated the blob. + */ +add_task(async function test_check_blob_origin_representation() { + forcePromptForFiles("text/plain", "txt"); + + await check_blob_origin( + "https://example.org/1", + "example.org", + "example.org" + ); + await check_blob_origin( + "data:text/html,<body>Some Text<br>", + "(data)", + "blob" + ); +}); + +async function check_blob_origin(pageURL, expectedSource, expectedListOrigin) { + await BrowserTestUtils.withNewTab(pageURL, async browser => { + // Ensure we wait for the download to finish: + let downloadList = await Downloads.getList(Downloads.PUBLIC); + let downloadPromise = promiseDownloadFinished(downloadList); + + // Wait for the download prompting dialog + let dialogPromise = BrowserTestUtils.domWindowOpenedAndLoaded( + null, + win => win.document.documentURI == UCT_URI + ); + + // create and click an <a download> link to a txt file. + await SpecialPowers.spawn(browser, [], () => { + // Use `eval` to get a blob URL scoped to content, so that content is + // actually allowed to open it and so we can check the origin is correct. + let url = content.eval(` + window.foo = new Blob(["Hello"], {type: "text/plain"}); + URL.createObjectURL(window.foo)`); + let link = content.document.createElement("a"); + link.href = url; + link.textContent = "Click me, click me, me me me"; + link.download = "my-file.txt"; + content.document.body.append(link); + link.click(); + }); + + // Check what we display in the dialog + let dialogWin = await dialogPromise; + let source = dialogWin.document.getElementById("source"); + is( + source.value, + expectedSource, + "Should list origin as source if available." + ); + + // Close the dialog + let closedPromise = BrowserTestUtils.windowClosed(dialogWin); + // Ensure we're definitely saving (otherwise this depends on mime service + // defaults): + dialogWin.document.getElementById("save").click(); + let dialogNode = dialogWin.document.querySelector("dialog"); + dialogNode.getButton("accept").disabled = false; + dialogNode.acceptDialog(); + await closedPromise; + + // Wait for the download to finish and ensure it is cleared up. + let download = await downloadPromise; + registerCleanupFunction(async () => { + let target = download.target.path; + await download.finalize(); + await IOUtils.remove(target); + }); + + // Check that the same download is displayed correctly in about:downloads. + await BrowserTestUtils.withNewTab("about:downloads", async dlBrowser => { + let doc = dlBrowser.contentDocument; + let listNode = doc.getElementById("downloadsListBox"); + await BrowserTestUtils.waitForMutationCondition( + listNode, + { childList: true, subtree: true, attributeFilter: ["value"] }, + () => + listNode.firstElementChild + ?.querySelector(".downloadDetailsNormal") + ?.getAttribute("value") + ); + let download = listNode.firstElementChild; + let detailString = download.querySelector(".downloadDetailsNormal").value; + Assert.stringContains( + detailString, + expectedListOrigin, + "Should list origin in download list if available." + ); + }); + }); +} diff --git a/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_delayedbutton.js b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_delayedbutton.js new file mode 100644 index 0000000000..426a740cd7 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_delayedbutton.js @@ -0,0 +1,96 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +const UCT_URI = "chrome://mozapps/content/downloads/unknownContentType.xhtml"; +const LOAD_URI = + "http://mochi.test:8888/browser/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.txt"; + +const DIALOG_DELAY = + Services.prefs.getIntPref("security.dialog_enable_delay") + 200; + +let UCTObserver = { + opened: PromiseUtils.defer(), + closed: PromiseUtils.defer(), + + observe(aSubject, aTopic, aData) { + let win = aSubject; + + switch (aTopic) { + case "domwindowopened": + win.addEventListener( + "load", + function onLoad(event) { + // Let the dialog initialize + SimpleTest.executeSoon(function () { + UCTObserver.opened.resolve(win); + }); + }, + { once: true } + ); + break; + + case "domwindowclosed": + if (win.location == UCT_URI) { + this.closed.resolve(); + } + break; + } + }, +}; + +function waitDelay(delay) { + return new Promise((resolve, reject) => { + /* eslint-disable mozilla/no-arbitrary-setTimeout */ + window.setTimeout(resolve, delay); + }); +} + +add_task(async function test_unknownContentType_delayedbutton() { + info("Starting browser_unknownContentType_delayedbutton.js..."); + forcePromptForFiles("text/plain", "txt"); + + Services.ww.registerNotification(UCTObserver); + + await BrowserTestUtils.withNewTab( + { + gBrowser, + url: LOAD_URI, + waitForLoad: false, + waitForStateStop: true, + }, + async function () { + let uctWindow = await UCTObserver.opened.promise; + let dialog = uctWindow.document.getElementById("unknownContentType"); + let ok = dialog.getButton("accept"); + + SimpleTest.is(ok.disabled, true, "button started disabled"); + + await waitDelay(DIALOG_DELAY); + + SimpleTest.is(ok.disabled, false, "button was enabled"); + + let focusOutOfDialog = SimpleTest.promiseFocus(window); + window.focus(); + await focusOutOfDialog; + + SimpleTest.is(ok.disabled, true, "button was disabled"); + + let focusOnDialog = SimpleTest.promiseFocus(uctWindow); + uctWindow.focus(); + await focusOnDialog; + + SimpleTest.is(ok.disabled, true, "button remained disabled"); + + await waitDelay(DIALOG_DELAY); + SimpleTest.is(ok.disabled, false, "button re-enabled after delay"); + + dialog.cancelDialog(); + await UCTObserver.closed.promise; + + Services.ww.unregisterNotification(UCTObserver); + uctWindow = null; + UCTObserver = null; + } + ); +}); diff --git a/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_dialog_layout.js b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_dialog_layout.js new file mode 100644 index 0000000000..577e341f90 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_dialog_layout.js @@ -0,0 +1,103 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +/* + * The unknownContentType popup can have two different layouts depending on + * whether a helper application can be selected or not. + * This tests that both layouts have correct collapsed elements. + */ + +const UCT_URI = "chrome://mozapps/content/downloads/unknownContentType.xhtml"; + +let tests = [ + { + // This URL will trigger the simple UI, where only the Save an Cancel buttons are available + url: "http://mochi.test:8888/browser/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.pif", + elements: { + basicBox: { collapsed: false }, + normalBox: { collapsed: true }, + }, + }, + { + // This URL will trigger the full UI + url: "http://mochi.test:8888/browser/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.txt", + elements: { + basicBox: { collapsed: true }, + normalBox: { collapsed: false }, + }, + }, +]; + +add_task(async function test_unknownContentType_dialog_layout() { + forcePromptForFiles("text/plain", "txt"); + forcePromptForFiles("application/octet-stream", "pif"); + + for (let test of tests) { + let UCTObserver = { + opened: PromiseUtils.defer(), + closed: PromiseUtils.defer(), + + observe(aSubject, aTopic, aData) { + let win = aSubject; + + switch (aTopic) { + case "domwindowopened": + win.addEventListener( + "load", + function onLoad(event) { + // Let the dialog initialize + SimpleTest.executeSoon(function () { + UCTObserver.opened.resolve(win); + }); + }, + { once: true } + ); + break; + + case "domwindowclosed": + if (win.location == UCT_URI) { + this.closed.resolve(); + } + break; + } + }, + }; + + Services.ww.registerNotification(UCTObserver); + await BrowserTestUtils.withNewTab( + { + gBrowser, + url: test.url, + waitForLoad: false, + waitForStateStop: true, + }, + async function () { + let uctWindow = await UCTObserver.opened.promise; + + for (let [id, props] of Object.entries(test.elements)) { + let elem = uctWindow.dialog.dialogElement(id); + for (let [prop, value] of Object.entries(props)) { + SimpleTest.is( + elem[prop], + value, + "Element with id " + + id + + " has property " + + prop + + " set to " + + value + ); + } + } + let focusOnDialog = SimpleTest.promiseFocus(uctWindow); + uctWindow.focus(); + await focusOnDialog; + + uctWindow.document.getElementById("unknownContentType").cancelDialog(); + uctWindow = null; + Services.ww.unregisterNotification(UCTObserver); + } + ); + } +}); diff --git a/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_extension.js b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_extension.js new file mode 100644 index 0000000000..1bb836c1d8 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_extension.js @@ -0,0 +1,58 @@ +/* Any copyright is dedicated to the Public Domain. + http://creativecommons.org/publicdomain/zero/1.0/ */ + +"use strict"; + +const TEST_PATH = getRootDirectory(gTestPath).replace( + "chrome://mochitests/content", + "http://example.com" +); + +/** + * Check that case-sensitivity doesn't cause us to duplicate + * file name extensions. + */ +add_task(async function test_download_filename_extension() { + forcePromptForFiles("application/octet-stream", "exe"); + let windowObserver = BrowserTestUtils.domWindowOpenedAndLoaded(); + + let tab = await BrowserTestUtils.openNewForegroundTab({ + gBrowser, + opening: TEST_PATH + "unknownContentType.EXE", + waitForLoad: false, + }); + let win = await windowObserver; + + let list = await Downloads.getList(Downloads.ALL); + let downloadFinishedPromise = new Promise(resolve => { + list.addView({ + onDownloadChanged(download) { + if (download.stopped) { + list.removeView(this); + resolve(download); + } + }, + }); + }); + + let dialog = win.document.querySelector("dialog"); + dialog.getButton("accept").removeAttribute("disabled"); + dialog.acceptDialog(); + let download = await downloadFinishedPromise; + // We cannot assume that the filename didn't change. + let filename = PathUtils.filename(download.target.path); + Assert.ok( + filename.indexOf(".") == filename.lastIndexOf("."), + "Should not duplicate extension" + ); + Assert.ok(filename.endsWith(".EXE"), "Should not change extension"); + await list.remove(download); + BrowserTestUtils.removeTab(tab); + try { + await IOUtils.remove(download.target.path); + } catch (ex) { + // Ignore errors in removing the file, the system may keep it locked and + // it's not a critical issue. + info("Failed to remove the file " + ex); + } +}); diff --git a/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_policy.js b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_policy.js new file mode 100644 index 0000000000..ccb0d957bb --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/browser_unknownContentType_policy.js @@ -0,0 +1,68 @@ +/* Any copyright is dedicated to the Public Domain. + http://creativecommons.org/publicdomain/zero/1.0/ */ + +"use strict"; + +const { EnterprisePolicyTesting } = ChromeUtils.importESModule( + "resource://testing-common/EnterprisePolicyTesting.sys.mjs" +); + +const TEST_PATH = getRootDirectory(gTestPath).replace( + "chrome://mochitests/content", + "http://example.com" +); + +/** + * Check that policy allows certain extensions to be launched. + */ +add_task(async function test_download_jnlp_policy() { + forcePromptForFiles("application/x-java-jnlp-file", "jnlp"); + let windowObserver = BrowserTestUtils.domWindowOpenedAndLoaded(); + + let tab = await BrowserTestUtils.openNewForegroundTab({ + gBrowser, + opening: TEST_PATH + "example.jnlp", + waitForLoad: false, + }); + let win = await windowObserver; + + let dialog = win.document.querySelector("dialog"); + let normalBox = win.document.getElementById("normalBox"); + let basicBox = win.document.getElementById("basicBox"); + is(normalBox.collapsed, !AppConstants.IS_ESR); + is(basicBox.collapsed, AppConstants.IS_ESR); + dialog.cancelDialog(); + BrowserTestUtils.removeTab(tab); + + await EnterprisePolicyTesting.setupPolicyEngineWithJson({ + policies: { + ExemptDomainFileTypePairsFromFileTypeDownloadWarnings: [ + { + file_extension: "jnlp", + domains: ["example.com"], + }, + ], + }, + }); + + windowObserver = BrowserTestUtils.domWindowOpenedAndLoaded(); + + tab = await BrowserTestUtils.openNewForegroundTab({ + gBrowser, + opening: TEST_PATH + "example.jnlp", + waitForLoad: false, + }); + win = await windowObserver; + + dialog = win.document.querySelector("dialog"); + normalBox = win.document.getElementById("normalBox"); + basicBox = win.document.getElementById("basicBox"); + is(normalBox.collapsed, false); + is(basicBox.collapsed, true); + dialog.cancelDialog(); + BrowserTestUtils.removeTab(tab); + + await EnterprisePolicyTesting.setupPolicyEngineWithJson({ + policies: {}, + }); +}); diff --git a/toolkit/mozapps/downloads/tests/browser/example.jnlp b/toolkit/mozapps/downloads/tests/browser/example.jnlp new file mode 100644 index 0000000000..e69de29bb2 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/example.jnlp diff --git a/toolkit/mozapps/downloads/tests/browser/example.jnlp^headers^ b/toolkit/mozapps/downloads/tests/browser/example.jnlp^headers^ new file mode 100644 index 0000000000..fac0de2095 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/example.jnlp^headers^ @@ -0,0 +1 @@ +Content-Type: application/x-java-jnlp-file diff --git a/toolkit/mozapps/downloads/tests/browser/head.js b/toolkit/mozapps/downloads/tests/browser/head.js new file mode 100644 index 0000000000..a5536e95b2 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/head.js @@ -0,0 +1,17 @@ +/* Any copyright is dedicated to the Public Domain. + http://creativecommons.org/publicdomain/zero/1.0/ */ + +"use strict"; + +function forcePromptForFiles(mime, extension) { + const handlerSvc = Cc["@mozilla.org/uriloader/handler-service;1"].getService( + Ci.nsIHandlerService + ); + const mimeSvc = Cc["@mozilla.org/mime;1"].getService(Ci.nsIMIMEService); + + let txtHandlerInfo = mimeSvc.getFromTypeAndExtension(mime, extension); + txtHandlerInfo.preferredAction = Ci.nsIHandlerInfo.alwaysAsk; + txtHandlerInfo.alwaysAskBeforeHandling = true; + handlerSvc.store(txtHandlerInfo); + registerCleanupFunction(() => handlerSvc.remove(txtHandlerInfo)); +} diff --git a/toolkit/mozapps/downloads/tests/browser/unknownContentType.EXE b/toolkit/mozapps/downloads/tests/browser/unknownContentType.EXE new file mode 100644 index 0000000000..e69de29bb2 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/unknownContentType.EXE diff --git a/toolkit/mozapps/downloads/tests/browser/unknownContentType.EXE^headers^ b/toolkit/mozapps/downloads/tests/browser/unknownContentType.EXE^headers^ new file mode 100644 index 0000000000..09b22facc0 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/unknownContentType.EXE^headers^ @@ -0,0 +1 @@ +Content-Type: application/octet-stream diff --git a/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.pif b/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.pif new file mode 100644 index 0000000000..9353d13126 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.pif @@ -0,0 +1 @@ +Dummy content for unknownContentType_dialog_layout_data.pif diff --git a/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.pif^headers^ b/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.pif^headers^ new file mode 100644 index 0000000000..09b22facc0 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.pif^headers^ @@ -0,0 +1 @@ +Content-Type: application/octet-stream diff --git a/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.txt b/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.txt new file mode 100644 index 0000000000..77e7195596 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.txt @@ -0,0 +1 @@ +Dummy content for unknownContentType_dialog_layout_data.txt diff --git a/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.txt^headers^ b/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.txt^headers^ new file mode 100644 index 0000000000..2a3c472e26 --- /dev/null +++ b/toolkit/mozapps/downloads/tests/browser/unknownContentType_dialog_layout_data.txt^headers^ @@ -0,0 +1,2 @@ +Content-Type: text/plain +Content-Disposition: attachment |