1 files changed, 244 insertions, 0 deletions
diff --git a/dom/security/test/unit/test_deserialization_format_before_100.js b/dom/security/test/unit/test_deserialization_format_before_100.js
new file mode 100644
index 0000000000..a21ae4838a
--- /dev/null
+++ b/dom/security/test/unit/test_deserialization_format_before_100.js
@@ -0,0 +1,244 @@
+/* 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";
+
+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.writeStringZ("https://example.com/"); // spec
+      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.writeStringZ("https://example.com/"); // spec
+      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.writeStringZ("https://example.com/"); // spec
+      stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
+      stream.writeBoolean(false); // sendReferrer
+      stream.writeBoolean(true); // isComputed
+      stream.writeStringZ("https://example.com/"); // computedReferrer
+      stream.writeBoolean(true); // initialized
+      stream.writeBoolean(false); // overridePolicyByDefault
+    },
+    {
+      referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
+      sendReferrer: false,
+      computedReferrerSpec: "https://example.com/",
+    }
+  );
+
+  // 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.writeStringZ("https://example.com/"); // spec
+      stream.write32(Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN); // policy
+      stream.writeBoolean(true); // sendReferrer
+      stream.writeBoolean(true); // isComputed
+      stream.writeStringZ("https://example.com/"); // computedReferrer
+      stream.writeBoolean(true); // initialized
+      stream.writeBoolean(false); // overridePolicyByDefault
+    },
+    {
+      referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
+      sendReferrer: true,
+      computedReferrerSpec: "https://example.com/",
+    }
+  );
+});
+
+// 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.writeStringZ("https://example.com/"); // spec
+      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.writeStringZ("https://example.com/"); // spec
+      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.writeStringZ("https://example.com/"); // spec
+      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.writeStringZ("https://example.com/"); // computedReferrer
+      stream.writeBoolean(true); // initialized
+      stream.writeBoolean(false); // overridePolicyByDefault
+    },
+    {
+      referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
+      sendReferrer: false,
+      computedReferrerSpec: "https://example.com/",
+    }
+  );
+
+  // 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.writeStringZ("https://example.com/"); // spec
+      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.writeStringZ("https://example.com/"); // computedReferrer
+      stream.writeBoolean(true); // initialized
+      stream.writeBoolean(false); // overridePolicyByDefault
+    },
+    {
+      referrerPolicy: Ci.nsIReferrerInfo.STRICT_ORIGIN_WHEN_CROSS_ORIGIN,
+      sendReferrer: true,
+      computedReferrerSpec: "https://example.com/",
+    }
+  );
+
+  // 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.writeStringZ("https://example.com/"); // spec
+      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,
+    }
+  );
+});