var BASE_URL = 'example.com/tests/dom/base/test/img_referrer_testserver.sjs';
const IMG_BYTES = atob(
"iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAYAAACNbyblAAAAHElEQVQI12" +
"P4//8/w38GIAXDIBKE0DHxgljNBAAO9TXL0Y4OHwAAAABJRU5ErkJggg==");
function createTestUrl(aPolicy, aAction, aName, aContent) {
var content = aContent || 'text';
return 'http://' + BASE_URL + '?' +
'action=' + aAction + '&' +
'policy=' + aPolicy + '&' +
'name=' + aName + '&' +
'content=' + content;
}
function createTestPage(aHead, aImgPolicy, aName) {
var _createTestUrl = createTestUrl.bind(null, aImgPolicy, 'test', aName);
return '\n\
'+
aHead +
'\n\
 + ')
\n\
\n\
\n\
';
}
// Creates the following test cases for the specified referrer
// policy combination:
// ![]()
with referrer
function createTest(aPolicy, aImgPolicy, aName) {
var headString = '';
if (aPolicy) {
headString += '';
}
headString += '';
return createTestPage(headString, aImgPolicy, aName);
}
// testing regular load img with referrer policy
// speculative parser should not kick in here
function createTest2(aImgPolicy, name) {
return createTestPage('', aImgPolicy, name);
}
function createTest3(aImgPolicy1, aImgPolicy2, aImgPolicy3, aName) {
return '\n\
\n\
\n\
 + ')
\n\
 + ')
\n\
 + ')
\n\
\n\
\n\
';
}
function createTestPage2(aHead, aPolicy, aName) {
return '\n\
'+
aHead +
'\n\
\n\
\n\
\n\
';
}
function createTestPage3(aHead, aPolicy, aName) {
return '\n\
'+
aHead +
'\n\
\n\
\n\
';
}
function createTestPage4(aHead, aPolicy, aName) {
return '\n\
'+
aHead +
'\n\
\n\
\n\
';
}
function createSetAttributeTest1(aPolicy, aImgPolicy, aName) {
var headString = '';
headString += '';
headString += '';
return createTestPage3(headString, aImgPolicy, aName);
}
function createSetAttributeTest2(aPolicy, aImgPolicy, aName) {
var headString = '';
headString += '';
headString += '';
return createTestPage4(headString, aImgPolicy, aName);
}
function createTest4(aPolicy, aName) {
var headString = '';
headString += '';
headString += '';
return createTestPage2(headString, aPolicy, aName);
}
function createTest5(aPolicy, aName) {
var headString = '';
headString += '';
return createTestPage2(headString, aPolicy, aName);
}
function handleRequest(request, response) {
var sharedKey = 'img_referrer_testserver.sjs';
var params = request.queryString.split('&');
var action = params[0].split('=')[1];
response.setHeader('Cache-Control', 'no-cache', false);
response.setHeader('Content-Type', 'text/html; charset=utf-8', false);
if (action === 'resetState') {
var state = getSharedState(sharedKey);
state = {};
setSharedState(sharedKey, JSON.stringify(state));
response.write("");
return;
}
if (action === 'test') {
// ?action=test&policy=origin&name=name&content=content
var policy = params[1].split('=')[1];
var name = params[2].split('=')[1];
var content = params[3].split('=')[1];
var result = getSharedState(sharedKey);
if (result === '') {
result = {};
} else {
result = JSON.parse(result);
}
if (!result["tests"]) {
result["tests"] = {};
}
var referrerLevel = "none";
var test = {}
if (request.hasHeader('Referer')) {
let referrer = request.getHeader('Referer');
if (referrer.indexOf("img_referrer_testserver") > 0) {
referrerLevel = "full";
} else if (referrer == "http://mochi.test:8888/") {
referrerLevel = "origin";
}
test.referrer = request.getHeader('Referer');
} else {
test.referrer = '';
}
test.policy = referrerLevel;
test.expected = policy;
result["tests"][name] = test;
setSharedState(sharedKey, JSON.stringify(result));
if (content === 'image') {
response.setHeader("Content-Type", "image/png");
response.write(IMG_BYTES);
}
return;
}
if (action === 'get-test-results') {
// ?action=get-result
response.write(getSharedState(sharedKey));
return;
}
if (action === 'generate-img-policy-test') {
// ?action=generate-img-policy-test&imgPolicy=b64-encoded-string&name=name&policy=b64-encoded-string
var imgPolicy = unescape(params[1].split('=')[1]);
var name = unescape(params[2].split('=')[1]);
var metaPolicy = '';
if (params[3]) {
metaPolicy = params[3].split('=')[1];
}
response.write(createTest(metaPolicy, imgPolicy, name));
return;
}
if (action === 'generate-img-policy-test2') {
// ?action=generate-img-policy-test2&imgPolicy=b64-encoded-string&name=name
var imgPolicy = unescape(params[1].split('=')[1]);
var name = unescape(params[2].split('=')[1]);
response.write(createTest2(imgPolicy, name));
return;
}
if (action === 'generate-img-policy-test3') {
// ?action=generate-img-policy-test3&imgPolicy1=b64-encoded-string&imgPolicy2=b64-encoded-string&imgPolicy3=b64-encoded-string&name=name
var imgPolicy1 = unescape(params[1].split('=')[1]);
var imgPolicy2 = unescape(params[2].split('=')[1]);
var imgPolicy3 = unescape(params[3].split('=')[1]);
var name = unescape(params[4].split('=')[1]);
response.write(createTest3(imgPolicy1, imgPolicy2, imgPolicy3, name));
return;
}
if (action === 'generate-img-policy-test4') {
// ?action=generate-img-policy-test4&imgPolicy=b64-encoded-string&name=name
var policy = unescape(params[1].split('=')[1]);
var name = unescape(params[2].split('=')[1]);
response.write(createTest4(policy, name));
return;
}
if (action === 'generate-img-policy-test5') {
// ?action=generate-img-policy-test5&policy=b64-encoded-string&name=name
var policy = unescape(params[1].split('=')[1]);
var name = unescape(params[2].split('=')[1]);
response.write(createTest5(policy, name));
return;
}
if (action === 'generate-setAttribute-test1') {
// ?action=generate-setAttribute-test1&policy=b64-encoded-string&name=name
var imgPolicy = unescape(params[1].split('=')[1]);
var policy = unescape(params[2].split('=')[1]);
var name = unescape(params[3].split('=')[1]);
response.write(createSetAttributeTest1(policy, imgPolicy, name));
return;
}
if (action === 'generate-setAttribute-test2') {
// ?action=generate-setAttribute-test2&policy=b64-encoded-string&name=name
var imgPolicy = unescape(params[1].split('=')[1]);
var policy = unescape(params[2].split('=')[1]);
var name = unescape(params[3].split('=')[1]);
response.write(createSetAttributeTest2(policy, imgPolicy, name));
return;
}
response.write("I don't know action "+action);
return;
}