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
159
160
161
162
163
164
165
166
|
/*
* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/
*/
// The functions in this file will run in the content process in a test
// scope.
/* eslint-env mozilla/simpletest */
/* global ContentTaskUtils, content */
const { Services } = ChromeUtils.import("resource://gre/modules/Services.jsm");
const { NetUtil } = ChromeUtils.import("resource://gre/modules/NetUtil.jsm");
const info = console.log;
export var HTTPS_EXAMPLE_ORG = "https://example.org";
export var HTTPS_EXAMPLE_COM = "https://example.com";
export var HTTP_EXAMPLE_COM = "http://example.com";
export function browserTestPath(uri) {
return uri + "/browser/netwerk/test/browser/";
}
export function waitForAllExpectedTests() {
return ContentTaskUtils.waitForCondition(() => {
return content.testDone === true;
});
}
export function cleanupObservers() {
Services.obs.notifyObservers(null, "cookie-content-filter-cleanup");
}
export async function preclean_test() {
// enable all cookies for the set-cookie trigger via setCookieStringFromHttp
Services.prefs.setIntPref("network.cookie.cookieBehavior", 0);
Services.prefs.setBoolPref(
"network.cookieJarSettings.unblocked_for_testing",
true
);
Services.prefs.setBoolPref("network.cookie.sameSite.laxByDefault", false);
Services.prefs.setBoolPref(
"network.cookie.sameSite.noneRequiresSecure",
false
);
Services.prefs.setBoolPref("network.cookie.sameSite.schemeful", false);
Services.cookies.removeAll();
}
export async function cleanup_test() {
Services.prefs.clearUserPref("network.cookie.cookieBehavior");
Services.prefs.clearUserPref(
"network.cookieJarSettings.unblocked_for_testing"
);
Services.prefs.clearUserPref("network.cookie.sameSite.laxByDefault");
Services.prefs.clearUserPref("network.cookie.sameSite.noneRequiresSecure");
Services.prefs.clearUserPref("network.cookie.sameSite.schemeful");
Services.cookies.removeAll();
}
export async function fetchHelper(url, cookie, secure, domain = "") {
let headers = new Headers();
headers.append("return-set-cookie", cookie);
if (!secure) {
headers.append("return-insecure-cookie", cookie);
}
if (domain != "") {
headers.append("return-cookie-domain", domain);
}
info("fetching " + url);
await fetch(url, { headers });
}
// cookie header strings with multiple name=value pairs delimited by \n
// will trigger multiple "cookie-changed" signals
export function triggerSetCookieFromHttp(uri, cookie, fpd = "", ucd = 0) {
info("about to trigger set-cookie: " + uri + " " + cookie);
let channel = NetUtil.newChannel({
uri,
loadUsingSystemPrincipal: true,
contentPolicyType: Ci.nsIContentPolicy.TYPE_DOCUMENT,
});
if (fpd != "") {
channel.loadInfo.originAttributes = { firstPartyDomain: fpd };
}
if (ucd != 0) {
channel.loadInfo.originAttributes = { userContextId: ucd };
}
Services.cookies.setCookieStringFromHttp(uri, cookie, channel);
}
export async function triggerSetCookieFromHttpPrivate(uri, cookie) {
info("about to trigger set-cookie: " + uri + " " + cookie);
let channel = NetUtil.newChannel({
uri,
loadUsingSystemPrincipal: true,
contentPolicyType: Ci.nsIContentPolicy.TYPE_DOCUMENT,
}).QueryInterface(Ci.nsIPrivateBrowsingChannel);
channel.loadInfo.originAttributes = { privateBrowsingId: 1 };
channel.setPrivate(true);
Services.cookies.setCookieStringFromHttp(uri, cookie, channel);
}
// observer/listener function that will be run on the content processes
// listens and checks for the expected cookies
export function checkExpectedCookies(expected, browserName) {
const COOKIE_FILTER_TEST_MESSAGE = "cookie-content-filter-test";
const COOKIE_FILTER_TEST_CLEANUP = "cookie-content-filter-cleanup";
// Counting the expected number of tests is vital to the integrity of these
// tests due to the fact that this test suite relies on triggering tests
// to occur on multiple content processes.
// As such, test modifications/bugs often lead to silent failures.
// Hence, we count to ensure we didn't break anything
// To reduce risk here, we modularize each test as much as possible to
// increase liklihood that a silent failure will trigger a no-test
// error/warning
content.testDone = false;
let testNumber = 0;
// setup observer that continues listening/testing
function obs(subject, topic) {
// cleanup trigger recieved -> tear down the observer
if (topic == COOKIE_FILTER_TEST_CLEANUP) {
info("cleaning up: " + browserName);
Services.obs.removeObserver(obs, COOKIE_FILTER_TEST_MESSAGE);
Services.obs.removeObserver(obs, COOKIE_FILTER_TEST_CLEANUP);
return;
}
// test trigger recv'd -> perform test on cookie contents
if (topic == COOKIE_FILTER_TEST_MESSAGE) {
info("Checking if cookie visible: " + browserName);
let result = content.document.cookie;
let resultStr =
"Result " +
result +
" == expected: " +
expected[testNumber] +
" in " +
browserName;
ok(result == expected[testNumber], resultStr);
testNumber++;
if (testNumber >= expected.length) {
info("finishing browser tests: " + browserName);
content.testDone = true;
}
return;
}
ok(false, "Didn't handle cookie message properly"); //
}
info("setting up observers: " + browserName);
Services.obs.addObserver(obs, COOKIE_FILTER_TEST_MESSAGE);
Services.obs.addObserver(obs, COOKIE_FILTER_TEST_CLEANUP);
}
|
