-
Notifications
You must be signed in to change notification settings - Fork 54
/
features.js
156 lines (120 loc) · 6.17 KB
/
features.js
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
var tests = [
// Config is an array with 4 elements:
// - annotation blacklist
// - annotation whitelist
// - tracking blacklist
// - tracking whitelist
// All disabled.
{ config: [ false, false, false, false ], loadExpected: true, annotationExpected: false },
// Just whitelisted.
{ config: [ false, false, false, true ], loadExpected: true, annotationExpected: false },
// Just blacklisted.
{ config: [ false, false, true, false ], loadExpected: false, annotationExpected: false },
// whitelist + blacklist => whitelist wins
{ config: [ false, false, true, true ], loadExpected: true, annotationExpected: false },
// just annotated in whitelist.
{ config: [ false, true, false, false ], loadExpected: true, annotationExpected: false },
// TP and annotation whitelisted.
{ config: [ false, true, false, true ], loadExpected: true, annotationExpected: false },
// Annotation whitelisted, but TP blacklisted.
{ config: [ false, true, true, false ], loadExpected: false, annotationExpected: false },
// Annotation whitelisted. TP blacklisted and whitelisted: whitelist wins.
{ config: [ false, true, true, true ], loadExpected: true, annotationExpected: false },
// Just blacklist annotated.
{ config: [ true, false, false, false ], loadExpected: true, annotationExpected: true },
// annotated but TP whitelisted.
{ config: [ true, false, false, true ], loadExpected: true, annotationExpected: true },
// annotated and blacklisted.
{ config: [ true, false, true, false ], loadExpected: false, annotationExpected: false },
// annotated, TP blacklisted and whitelisted: whitelist wins.
{ config: [ true, false, true, true ], loadExpected: true, annotationExpected: true },
// annotated in white and blacklist.
{ config: [ true, true, false, false ], loadExpected: true, annotationExpected: false },
// annotated in white and blacklist. TP Whiteslited
{ config: [ true, true, false, true ], loadExpected: true, annotationExpected: false },
// everywhere. TP whitelist wins.
{ config: [ true, true, true, true ], loadExpected: true, annotationExpected: false },
];
function prefBlacklistValue(value) {
return value ? "example.com" : "";
}
function prefWhitelistValue(value) {
return value ? "mochi.test" : "";
}
async function runTest(test, expectedFlag, expectedTrackingResource, prefs) {
let config = [
[ "urlclassifier.trackingAnnotationTable.testEntries", prefBlacklistValue(test.config[0]) ],
[ "urlclassifier.features.fingerprinting.annotate.blacklistHosts", prefBlacklistValue(test.config[0]) ],
[ "urlclassifier.features.cryptomining.annotate.blacklistHosts", prefBlacklistValue(test.config[0]) ],
[ "urlclassifier.trackingAnnotationWhitelistTable.testEntries", prefWhitelistValue(test.config[1]) ],
[ "urlclassifier.features.fingerprinting.annotate.whitelistHosts", prefWhitelistValue(test.config[1]) ],
[ "urlclassifier.features.cryptomining.annotate.whitelistHosts", prefWhitelistValue(test.config[1]) ],
[ "urlclassifier.trackingTable.testEntries", prefBlacklistValue(test.config[2]) ],
[ "urlclassifier.features.fingerprinting.blacklistHosts", prefBlacklistValue(test.config[2]) ],
[ "urlclassifier.features.cryptomining.blacklistHosts", prefBlacklistValue(test.config[2]) ],
[ "urlclassifier.trackingWhitelistTable.testEntries", prefWhitelistValue(test.config[3]) ],
[ "urlclassifier.features.fingerprinting.whitelistHosts", prefWhitelistValue(test.config[3]) ],
[ "urlclassifier.features.cryptomining.whitelistHosts", prefWhitelistValue(test.config[3]) ],
];
info("Testing: " + config.toSource() + "\n");
await SpecialPowers.pushPrefEnv({set: config.concat(prefs) });
// This promise will be resolved when the chromeScript knows if the channel
// is annotated or not.
let annotationPromise;
if (test.loadExpected) {
info("We want to have annotation information");
annotationPromise = new Promise(resolve => {
chromeScript.addMessageListener("last-channel-flags",
data => resolve(data),
{once: true});
});
}
// Let's load an image with a random query string, just to avoid network cache.
let result = await new Promise(resolve => {
let image = new Image();
image.src = "http://example.com/tests/toolkit/components/url-classifier/tests/mochitest/raptor.jpg?" + Math.random();
image.onload = _ => resolve(true);
image.onerror = _ => resolve(false);
});
is(result, test.loadExpected, "The loading happened correctly");
if (annotationPromise) {
let data = await annotationPromise;
is(!!data.classificationFlags, test.annotationExpected, "The annotation happened correctly");
if (test.annotationExpected) {
is(data.classificationFlags, expectedFlag, "Correct flag");
is(data.isTrackingResource, expectedTrackingResource, "Tracking resource flag matches");
}
}
}
var chromeScript;
function runTests(flag, prefs, trackingResource) {
chromeScript = SpecialPowers.loadChromeScript(_ => {
const {Services} = ChromeUtils.import("resource://gre/modules/Services.jsm");
function onExamResp(subject, topic, data) {
let channel = subject.QueryInterface(Ci.nsIHttpChannel);
if (!channel ||
!channel.URI.spec.startsWith("http://example.com/tests/toolkit/components/url-classifier/tests/mochitest/raptor.jpg")) {
return;
}
// eslint-disable-next-line no-undef
sendAsyncMessage("last-channel-flags", {
classificationFlags: channel.classificationFlags,
isTrackingResource: channel.isTrackingResource(),
});
}
// eslint-disable-next-line no-undef
addMessageListener("done", __ => {
Services.obs.removeObserver(onExamResp, "http-on-examine-response");
});
Services.obs.addObserver(onExamResp, "http-on-examine-response");
// eslint-disable-next-line no-undef
sendAsyncMessage("start-test");
});
chromeScript.addMessageListener("start-test", async _ => {
for (let test in tests) {
await runTest(tests[test], flag, trackingResource, prefs);
}
chromeScript.sendAsyncMessage("done");
SimpleTest.finish();
}, {once: true});
}