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
|
/* 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";
// The purpose of this test is to see that the site security service properly
// writes its state file.
const EXPECTED_ENTRIES = 5;
const EXPECTED_HSTS_COLUMNS = 3;
var gProfileDir = null;
var gExpectingWrites = true;
// For reference, the format of the state file is a list of:
// <domain name> <expiration time in milliseconds>,<sts status>,<includeSubdomains>
// separated by newlines ('\n')
function checkStateWritten(aSubject, aTopic, aData) {
if (aData == CLIENT_AUTH_FILE_NAME) {
return;
}
equal(aData, SSS_STATE_FILE_NAME);
ok(gExpectingWrites);
let stateFile = gProfileDir.clone();
stateFile.append(SSS_STATE_FILE_NAME);
ok(stateFile.exists());
let stateFileContents = readFile(stateFile);
// the last line is removed because it's just a trailing newline
let lines = stateFileContents.split("\n").slice(0, -1);
equal(lines.length, EXPECTED_ENTRIES);
let sites = {}; // a map of domain name -> [the entry in the state file]
for (let line of lines) {
let parts = line.split("\t");
let host = parts[0];
let entry = parts[3].split(",");
let expectedColumns = EXPECTED_HSTS_COLUMNS;
equal(entry.length, expectedColumns);
sites[host] = entry;
}
// While we're still processing headers, multiple writes of the backing data
// can be scheduled, and thus we can receive multiple data-storage-written
// notifications. In these cases, the data may not be as we expect. We only
// care about the final one being correct, however, so we return and wait for
// the next event if things aren't as we expect.
// each sites[url][1] should be SecurityPropertySet (i.e. 1).
// sites[url][2] corresponds to includeSubdomains, so every other one should
// be set (i.e. 1);
if (sites["includesubdomains.preloaded.test:HSTS"][1] != 1) {
return;
}
if (sites["includesubdomains.preloaded.test:HSTS"][2] != 0) {
return;
}
if (sites["a.example.com:HSTS"][1] != 1) {
return;
}
if (sites["a.example.com:HSTS"][2] != 1) {
return;
}
if (sites["b.example.com:HSTS"][1] != 1) {
return;
}
if (sites["b.example.com:HSTS"][2] != 0) {
return;
}
if (sites["c.c.example.com:HSTS"][1] != 1) {
return;
}
if (sites["c.c.example.com:HSTS"][2] != 1) {
return;
}
if (sites["d.example.com:HSTS"][1] != 1) {
return;
}
if (sites["d.example.com:HSTS"][2] != 0) {
return;
}
// If we get here, the file was as expected and we no longer expect any
// data-storage-written notifications.
gExpectingWrites = false;
// Process the headers again to test that seeing them again after such a
// short delay doesn't cause another write.
process_headers();
// Wait a bit before finishing the test, to see if another write happens.
do_timeout(2000, function () {
do_test_finished();
});
}
function process_headers() {
let SSService = Cc["@mozilla.org/ssservice;1"].getService(
Ci.nsISiteSecurityService
);
let uris = [
Services.io.newURI("http://includesubdomains.preloaded.test"),
Services.io.newURI("http://a.example.com"),
Services.io.newURI("http://b.example.com"),
Services.io.newURI("http://c.c.example.com"),
Services.io.newURI("http://d.example.com"),
];
for (let i = 0; i < 1000; i++) {
let uriIndex = i % uris.length;
// vary max-age, but have it be within one day of one year
let maxAge = "max-age=" + (i + 31536000);
// have every other URI set includeSubdomains
let includeSubdomains = uriIndex % 2 == 1 ? "; includeSubdomains" : "";
SSService.processHeader(uris[uriIndex], maxAge + includeSubdomains);
}
}
function run_test() {
Services.prefs.setIntPref("test.datastorage.write_timer_ms", 100);
gProfileDir = do_get_profile();
process_headers();
do_test_pending();
Services.obs.addObserver(checkStateWritten, "data-storage-written");
}
|
