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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
/* 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";
const { PermissionTestUtils } = ChromeUtils.importESModule(
"resource://testing-common/PermissionTestUtils.sys.mjs"
);
const TEST_URI = "https://example.com";
const TEST_PATH = getRootDirectory(gTestPath).replace(
"chrome://mochitests/content",
TEST_URI
);
const AUTOMATIC_DOWNLOAD_TOPIC = "blocked-automatic-download";
var MockFilePicker = SpecialPowers.MockFilePicker;
MockFilePicker.init(window.browsingContext);
registerCleanupFunction(() => MockFilePicker.cleanup());
let gTempDownloadDir;
add_setup(async function () {
// Create temp directory
let time = new Date().getTime();
let tempDir = Services.dirsvc.get("TmpD", Ci.nsIFile);
tempDir.append(time);
gTempDownloadDir = tempDir;
Services.prefs.setIntPref("browser.download.folderList", 2);
Services.prefs.setComplexValue("browser.download.dir", Ci.nsIFile, tempDir);
PermissionTestUtils.add(
TEST_URI,
"automatic-download",
Services.perms.UNKNOWN_ACTION
);
await SpecialPowers.pushPrefEnv({
set: [
// We disable browser.download.always_ask_before_handling_new_types here
// since the test expects the download to be saved directly to disk and
// not prompted by the UnknownContentType window.
["browser.download.always_ask_before_handling_new_types", false],
["browser.download.enable_spam_prevention", true],
],
});
registerCleanupFunction(async () => {
Services.prefs.clearUserPref("browser.download.folderList");
Services.prefs.clearUserPref("browser.download.dir");
await IOUtils.remove(tempDir.path, { recursive: true });
});
});
add_task(async function check_download_spam_permissions() {
const INITIAL_TABS_COUNT = gBrowser.tabs.length;
let publicList = await Downloads.getList(Downloads.PUBLIC);
let downloadFinishedPromise = promiseDownloadFinished(
publicList,
true /* stop the download from openning */
);
let blockedDownloadsCount = 0;
let blockedDownloadsURI = "";
let automaticDownloadObserver = {
observe: function automatic_download_observe(aSubject, aTopic, aData) {
if (aTopic === AUTOMATIC_DOWNLOAD_TOPIC) {
blockedDownloadsCount++;
blockedDownloadsURI = aData;
}
},
};
Services.obs.addObserver(automaticDownloadObserver, AUTOMATIC_DOWNLOAD_TOPIC);
let newTab = await BrowserTestUtils.openNewForegroundTab(
gBrowser,
TEST_PATH + "test_spammy_page.html"
);
registerCleanupFunction(async () => {
DownloadIntegration.downloadSpamProtection.removeDownloadSpamForWindow(
TEST_URI,
window
);
DownloadsPanel.hidePanel();
await publicList.removeFinished();
BrowserTestUtils.removeTab(newTab);
Services.obs.removeObserver(
automaticDownloadObserver,
AUTOMATIC_DOWNLOAD_TOPIC
);
});
let download = await downloadFinishedPromise;
await TestUtils.waitForCondition(
() => gBrowser.tabs.length == INITIAL_TABS_COUNT + 1
);
is(
PermissionTestUtils.testPermission(TEST_URI, "automatic-download"),
Services.perms.PROMPT_ACTION,
"The permission to prompt the user should be stored."
);
ok(
await IOUtils.exists(download.target.path),
"One file should be downloaded"
);
let aCopyFilePath = download.target.path.replace(".pdf", "(1).pdf");
is(
await IOUtils.exists(aCopyFilePath),
false,
"An other file should be blocked"
);
info("Will wait for blockedDownloadsCount to be >= 99");
await TestUtils.waitForCondition(() => blockedDownloadsCount >= 99);
is(blockedDownloadsCount, 99, "Browser should block 99 downloads");
is(
blockedDownloadsURI,
TEST_URI,
"The test URI should have blocked automatic downloads"
);
await savelink();
});
// Check to ensure that a link saved manually is not blocked.
async function savelink() {
let publicList = await Downloads.getList(Downloads.PUBLIC);
let menu = document.getElementById("contentAreaContextMenu");
let popupShown = BrowserTestUtils.waitForEvent(menu, "popupshown");
BrowserTestUtils.synthesizeMouse(
"#image",
5,
5,
{ type: "contextmenu", button: 2 },
gBrowser.selectedBrowser
);
await popupShown;
await new Promise(resolve => {
MockFilePicker.showCallback = function () {
resolve();
let file = gTempDownloadDir.clone();
file.append("file_with__funny_name.png");
MockFilePicker.setFiles([file]);
return Ci.nsIFilePicker.returnOK;
};
let menuitem = menu.querySelector("#context-savelink");
menu.activateItem(menuitem);
});
await promiseDownloadFinished(
publicList,
true // stop the download from openning
);
}
|
