Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
- Manifest: dom/security/test/unit/xpcshell.toml
/* 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
"use strict";
const ReferrerInfo = Components.Constructor(
"@mozilla.org/referrer-info;1",
"nsIReferrerInfo",
"init"
);
async function runTest(setupFunc, expected) {
let objectOutStream = Cc["@mozilla.org/binaryoutputstream;1"].createInstance(
Ci.nsIObjectOutputStream
);
let pipe = Cc["@mozilla.org/pipe;1"].createInstance(Ci.nsIPipe);
pipe.init(
false /* non-blocking input */,
false /* non-blocking output */,
0 /* segment size */,
0 /* max segments */
);
objectOutStream.setOutputStream(pipe.outputStream);
setupFunc(objectOutStream);
let objectInStream = Cc["@mozilla.org/binaryinputstream;1"].createInstance(
Ci.nsIObjectInputStream
);
objectInStream.setInputStream(pipe.inputStream);
let referrerInfo = new ReferrerInfo(Ci.nsIReferrerInfo.EMPTY);
try {
referrerInfo.read(objectInStream);
} catch (e) {
Assert.ok(false, "Shouldn't fail when deserializing.");
return;
}
Assert.ok(true, "Successfully deserialize the referrerInfo.");
let { referrerPolicy, sendReferrer, computedReferrerSpec } = expected;
Assert.equal(
referrerInfo.referrerPolicy,
referrerPolicy,
"The referrerInfo has the expected referrer policy."
);
Assert.equal(
referrerInfo.sendReferrer,
sendReferrer,
"The referrerInfo has the expected sendReferrer value."
);
if (computedReferrerSpec) {
Assert.equal(
referrerInfo.computedReferrerSpec,
computedReferrerSpec,
"The referrerInfo has the expected computedReferrerSpec value."
);
}
}
// Test deserializing referrer info with the old format.
add_task(async function test_deserializeOldReferrerInfo() {
// Test with a regular old format.
await runTest(
stream => {
// Write to the output stream with the old format.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
stream.writeBoolean(false); // sendReferrer
stream.writeBoolean(false); // isComputed
stream.writeBoolean(true); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
sendReferrer: false,
}
);
// Test with an old format with `sendReferrer` is true.
await runTest(
stream => {
// Write to the output stream with the old format.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
stream.writeBoolean(true); // sendReferrer
stream.writeBoolean(false); // isComputed
stream.writeBoolean(true); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
sendReferrer: true,
}
);
// Test with an old format with a computed Referrer.
await runTest(
stream => {
// Write to the output stream with the old format with a string.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
stream.writeBoolean(false); // sendReferrer
stream.writeBoolean(true); // isComputed
stream.writeBoolean(true); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
sendReferrer: false,
}
);
// Test with an old format with a computed Referrer and sendReferrer as true.
await runTest(
stream => {
// Write to the output stream with the old format with a string.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
stream.writeBoolean(true); // sendReferrer
stream.writeBoolean(true); // isComputed
stream.writeBoolean(true); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
sendReferrer: true,
}
);
});
// Test deserializing referrer info with the current format.
add_task(async function test_deserializeReferrerInfo() {
// Test with a current format.
await runTest(
stream => {
// Write to the output stream with the new format.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // original policy
stream.writeBoolean(false); // sendReferrer
stream.writeBoolean(false); // isComputed
stream.writeBoolean(true); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
sendReferrer: false,
}
);
// Test with a current format with sendReferrer as true.
await runTest(
stream => {
// Write to the output stream with the new format.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // original policy
stream.writeBoolean(true); // sendReferrer
stream.writeBoolean(false); // isComputed
stream.writeBoolean(true); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
sendReferrer: true,
}
);
// Test with a current format with a computedReferrer.
await runTest(
stream => {
// Write to the output stream with the new format with a string.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // original policy
stream.writeBoolean(false); // sendReferrer
stream.writeBoolean(true); // isComputed
stream.writeBoolean(true); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
sendReferrer: false,
}
);
// Test with a current format with a computedReferrer and sendReferrer as true.
await runTest(
stream => {
// Write to the output stream with the new format with a string.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // original policy
stream.writeBoolean(true); // sendReferrer
stream.writeBoolean(true); // isComputed
stream.writeBoolean(true); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
sendReferrer: true,
}
);
// Test with a current format that the tailing bytes are all zero.
await runTest(
stream => {
// Write to the output stream with the new format with a string.
stream.writeBoolean(true); // nonNull
stream.write32(Ci.nsIReferrerInfo.EMPTY); // policy
stream.write32(Ci.nsIReferrerInfo.EMPTY); // original policy
stream.writeBoolean(false); // sendReferrer
stream.writeBoolean(false); // isComputed
stream.writeBoolean(false); // initialized
stream.writeBoolean(false); // overridePolicyByDefault
},
{
referrerPolicy: Ci.nsIReferrerInfo.EMPTY,
sendReferrer: false,
}
);
});