1 files changed, 194 insertions, 0 deletions
diff --git a/browser/components/sessionstore/test/browser_formdata.js b/browser/components/sessionstore/test/browser_formdata.js
new file mode 100644
index 000000000..ce1272888
--- /dev/null
+++ b/browser/components/sessionstore/test/browser_formdata.js
@@ -0,0 +1,194 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+requestLongerTimeout(2);
+
+/**
+ * This test ensures that form data collection respects the privacy level as
+ * set by the user.
+ */
+add_task(function test_formdata() {
+  const URL = "http://mochi.test:8888/browser/browser/components/" +
+              "sessionstore/test/browser_formdata_sample.html";
+
+  const OUTER_VALUE = "browser_formdata_" + Math.random();
+  const INNER_VALUE = "browser_formdata_" + Math.random();
+
+  // Creates a tab, loads a page with some form fields,
+  // modifies their values and closes the tab.
+  function createAndRemoveTab() {
+    return Task.spawn(function () {
+      // Create a new tab.
+      let tab = gBrowser.addTab(URL);
+      let browser = tab.linkedBrowser;
+      yield promiseBrowserLoaded(browser);
+
+      // Modify form data.
+      yield setInputValue(browser, {id: "txt", value: OUTER_VALUE});
+      yield setInputValue(browser, {id: "txt", value: INNER_VALUE, frame: 0});
+
+      // Remove the tab.
+      yield promiseRemoveTab(tab);
+    });
+  }
+
+  yield createAndRemoveTab();
+  let [{state: {formdata}}] = JSON.parse(ss.getClosedTabData(window));
+  is(formdata.id.txt, OUTER_VALUE, "outer value is correct");
+  is(formdata.children[0].id.txt, INNER_VALUE, "inner value is correct");
+
+  // Disable saving data for encrypted sites.
+  Services.prefs.setIntPref("browser.sessionstore.privacy_level", 1);
+
+  yield createAndRemoveTab();
+  [{state: {formdata}}] = JSON.parse(ss.getClosedTabData(window));
+  is(formdata.id.txt, OUTER_VALUE, "outer value is correct");
+  ok(!formdata.children, "inner value was *not* stored");
+
+  // Disable saving data for any site.
+  Services.prefs.setIntPref("browser.sessionstore.privacy_level", 2);
+
+  yield createAndRemoveTab();
+  [{state: {formdata}}] = JSON.parse(ss.getClosedTabData(window));
+  ok(!formdata, "form data has *not* been stored");
+
+  // Restore the default privacy level.
+  Services.prefs.clearUserPref("browser.sessionstore.privacy_level");
+});
+
+/**
+ * This test ensures that a malicious website can't trick us into restoring
+ * form data into a wrong website and that we always check the stored URL
+ * before doing so.
+ */
+add_task(function test_url_check() {
+  const URL = "data:text/html;charset=utf-8,<input%20id=input>";
+  const VALUE = "value-" + Math.random();
+
+  // Create a tab with an iframe containing an input field.
+  let tab = gBrowser.addTab(URL);
+  let browser = tab.linkedBrowser;
+  yield promiseBrowserLoaded(browser);
+
+  // Restore a tab state with a given form data url.
+  function restoreStateWithURL(url) {
+    let state = {entries: [{url: URL}], formdata: {id: {input: VALUE}}};
+
+    if (url) {
+      state.formdata.url = url;
+    }
+
+    return promiseTabState(tab, state).then(() => getInputValue(browser, "input"));
+  }
+
+  // Check that the form value is restored with the correct URL.
+  is((yield restoreStateWithURL(URL)), VALUE, "form data restored");
+
+  // Check that the form value is *not* restored with the wrong URL.
+  is((yield restoreStateWithURL(URL + "?")), "", "form data not restored");
+  is((yield restoreStateWithURL()), "", "form data not restored");
+
+  // Cleanup.
+  gBrowser.removeTab(tab);
+});
+
+/**
+ * This test ensures that collecting form data works as expected when having
+ * nested frame sets.
+ */
+add_task(function test_nested() {
+  const URL = "data:text/html;charset=utf-8," +
+              "<iframe src='data:text/html;charset=utf-8," +
+              "<input autofocus=true>'/>";
+
+  const FORM_DATA = {
+    children: [{
+      xpath: {"/xhtml:html/xhtml:body/xhtml:input": "M"},
+      url: "data:text/html;charset=utf-8,<input%20autofocus=true>"
+    }]
+  };
+
+  // Create a tab with an iframe containing an input field.
+  let tab = gBrowser.selectedTab = gBrowser.addTab(URL);
+  let browser = tab.linkedBrowser;
+  yield promiseBrowserLoaded(browser);
+
+  // Modify the input field's value.
+  yield sendMessage(browser, "ss-test:sendKeyEvent", {key: "m", frame: 0});
+
+  // Remove the tab and check that we stored form data correctly.
+  yield promiseRemoveTab(tab);
+  let [{state: {formdata}}] = JSON.parse(ss.getClosedTabData(window));
+  is(JSON.stringify(formdata), JSON.stringify(FORM_DATA),
+    "formdata for iframe stored correctly");
+
+  // Restore the closed tab.
+  tab = ss.undoCloseTab(window, 0);
+  browser = tab.linkedBrowser;
+  yield promiseTabRestored(tab);
+
+  // Check that the input field has the right value.
+  yield TabStateFlusher.flush(browser);
+  ({formdata} = JSON.parse(ss.getTabState(tab)));
+  is(JSON.stringify(formdata), JSON.stringify(FORM_DATA),
+    "formdata for iframe restored correctly");
+
+  // Cleanup.
+  gBrowser.removeTab(tab);
+});
+
+/**
+ * This test ensures that collecting form data for documents with
+ * designMode=on works as expected.
+ */
+add_task(function test_design_mode() {
+  const URL = "data:text/html;charset=utf-8,<h1>mozilla</h1>" +
+              "<script>document.designMode='on'</script>";
+
+  // Load a tab with an editable document.
+  let tab = gBrowser.selectedTab = gBrowser.addTab(URL);
+  let browser = tab.linkedBrowser;
+  yield promiseBrowserLoaded(browser);
+
+  // Modify the document content.
+  yield sendMessage(browser, "ss-test:sendKeyEvent", {key: "m"});
+
+  // Close and restore the tab.
+  yield promiseRemoveTab(tab);
+  tab = ss.undoCloseTab(window, 0);
+  browser = tab.linkedBrowser;
+  yield promiseTabRestored(tab);
+
+  // Check that the innerHTML value was restored.
+  let html = yield getInnerHTML(browser);
+  let expected = "<h1>Mmozilla</h1><script>document.designMode='on'</script>";
+  is(html, expected, "editable document has been restored correctly");
+
+  // Close and restore the tab.
+  yield promiseRemoveTab(tab);
+  tab = ss.undoCloseTab(window, 0);
+  browser = tab.linkedBrowser;
+  yield promiseTabRestored(tab);
+
+  // Check that the innerHTML value was restored.
+  html = yield getInnerHTML(browser);
+  expected = "<h1>Mmozilla</h1><script>document.designMode='on'</script>";
+  is(html, expected, "editable document has been restored correctly");
+
+  // Cleanup.
+  gBrowser.removeTab(tab);
+});
+
+function getInputValue(browser, id) {
+  return sendMessage(browser, "ss-test:getInputValue", {id: id});
+}
+
+function setInputValue(browser, data) {
+  return sendMessage(browser, "ss-test:setInputValue", data);
+}
+
+function getInnerHTML(browser) {
+  return sendMessage(browser, "ss-test:getInnerHTML", {selector: "body"});
+}