59 lines
2.1 KiB
JavaScript
59 lines
2.1 KiB
JavaScript
var RESOURCES_DIR = "/beacon/resources/";
|
|
|
|
var referrerOrigin = self.location.origin + '/';
|
|
var referrerUrl = self.location.href;
|
|
|
|
function testReferrerHeader(testBase, expectedReferrer, mayBeBlockedAsMixedContent = false) {
|
|
var id = self.token();
|
|
var testUrl = testBase + "inspect-header.py?header=referer&cmd=put&id=" + id;
|
|
|
|
promise_test(function(test) {
|
|
const sentBeacon = navigator.sendBeacon(testUrl);
|
|
if (mayBeBlockedAsMixedContent && !sentBeacon)
|
|
return Promise.resolve();
|
|
assert_true(sentBeacon, "SendBeacon Succeeded");
|
|
return pollResult(expectedReferrer, id) .then(result => {
|
|
assert_equals(result, expectedReferrer, "Correct referrer header result");
|
|
});
|
|
}, "Test referer header " + testBase);
|
|
}
|
|
|
|
function testOriginHeader(testBase, expectedOrigin, addBody) {
|
|
var id = self.token();
|
|
var testUrl = testBase + "inspect-header.py?header=origin&cmd=put&id=" + id;
|
|
|
|
promise_test(function(test) {
|
|
const sentBeacon = navigator.sendBeacon(testUrl, addBody ? "data" : undefined);
|
|
assert_true(sentBeacon, "SendBeacon Succeeded");
|
|
return pollResult(expectedOrigin, id, "origin") .then(result => {
|
|
assert_equals(result, expectedOrigin, "Correct origin header result");
|
|
});
|
|
}, "Test origin header " + testBase + (addBody ? " - with body" : " - without body"));
|
|
}
|
|
|
|
// SendBeacon is an asynchronous and non-blocking request to a web server.
|
|
// We may have to create a poll loop to get result from server
|
|
function pollResult(expectedReferrer, id, headerName) {
|
|
if (!headerName)
|
|
headerName = "referer";
|
|
var checkUrl = RESOURCES_DIR + "inspect-header.py?header=" + headerName + "&cmd=get&id=" + id;
|
|
|
|
return new Promise(resolve => {
|
|
function checkResult() {
|
|
fetch(checkUrl).then(
|
|
function(response) {
|
|
assert_equals(response.status, 200, "Inspect header response's status is 200");
|
|
let result = response.headers.get("x-request-" + headerName);
|
|
|
|
if (result != undefined) {
|
|
resolve(result);
|
|
} else {
|
|
step_timeout(checkResult.bind(this), 100);
|
|
}
|
|
});
|
|
}
|
|
|
|
checkResult();
|
|
});
|
|
|
|
}
|