The Tor Browser: netwerk/test/unit/test_bug248970_cookie.js@ac0c01689b40 (annotated)

netwerk/test/unit/test_bug248970_cookie.js@ac0c01689b40 (annotated)

netwerk/test/unit/test_bug248970_cookie.js

Thu, 15 Jan 2015 15:59:08 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Thu, 15 Jan 2015 15:59:08 +0100
branch: TOR_BUG_9701
changeset 10: ac0c01689b40
permissions: -rw-r--r--

Implement a real Private Browsing Mode condition by changing the API/ABI;
This solves Tor bug #9701, complying with disk avoidance documented in
https://www.torproject.org/projects/torbrowser/design/#disk-avoidance.

 /* 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/. */
 Cu.import("resource://testing-common/httpd.js");
 Cu.import("resource://gre/modules/Services.jsm");
 var httpserver;
 function inChildProcess() {
   return Cc["@mozilla.org/xre/app-info;1"]
            .getService(Ci.nsIXULRuntime)
            .processType != Ci.nsIXULRuntime.PROCESS_TYPE_DEFAULT;
 }
 function makeChan(path) {
   var ios = Cc["@mozilla.org/network/io-service;1"].getService(Ci.nsIIOService);
   var chan = ios.newChannel("http://localhost:" + httpserver.identity.primaryPort + "/" + path, null, null)
                 .QueryInterface(Ci.nsIHttpChannel);
   return chan;
 }
 function setup_chan(path, isPrivate, callback) {
   var chan = makeChan(path);
   chan.QueryInterface(Ci.nsIPrivateBrowsingChannel).setPrivate(isPrivate);
   chan.asyncOpen(new ChannelListener(callback), null);
  }
 function set_cookie(value, callback) {
   return setup_chan('set?cookie=' + value, false, callback);
 }
 function set_private_cookie(value, callback) {
   return setup_chan('set?cookie=' + value, true, callback);
 }
 function check_cookie_presence(value, isPrivate, expected, callback) {
   var chan = setup_chan('present?cookie=' + value.replace('=','|'), isPrivate, function(req) {
     req.QueryInterface(Ci.nsIHttpChannel);
     do_check_eq(req.responseStatus, expected ? 200 : 404);
     callback(req);
   });
 }
 function presentHandler(metadata, response) {
   var present = false;
   var match = /cookie=([^&]*)/.exec(metadata.queryString);
   if (match) {
     try {
       present = metadata.getHeader("Cookie").indexOf(match[1].replace("|","=")) != -1;
     } catch (x) {
     }
   }
   response.setStatusLine("1.0", present ? 200 : 404, "");
 }
 function setHandler(metadata, response) {
   response.setStatusLine("1.0", 200, "Cookie set");
   var match = /cookie=([^&]*)/.exec(metadata.queryString);
   if (match) {
     response.setHeader("Set-Cookie", match[1]);
   }
 }
 function run_test() {
   // Allow all cookies if the pref service is available in this process.
   if (!inChildProcess())
     Services.prefs.setIntPref("network.cookie.cookieBehavior", 0);
   httpserver = new HttpServer();
   httpserver.registerPathHandler("/set", setHandler);
   httpserver.registerPathHandler("/present", presentHandler);
   httpserver.start(-1);
   do_test_pending();
   function check_cookie(req) {
     req.QueryInterface(Ci.nsIHttpChannel);
     do_check_eq(req.responseStatus, 200);
     try {
       do_check_true(req.getResponseHeader("Set-Cookie") != "", "expected a Set-Cookie header");
     } catch (x) {
       do_throw("missing Set-Cookie header");
     }
     runNextTest();
   }
   let tests = [];
   function runNextTest() {
     do_execute_soon(tests.shift());
   }
   tests.push(function() {
     set_cookie("C1=V1", check_cookie);
   });
   tests.push(function() {
     set_private_cookie("C2=V2", check_cookie);
   });
   tests.push(function() {
     // Check that the first cookie is present in a non-private request
     check_cookie_presence("C1=V1", false, true, runNextTest);
   });
   tests.push(function() {
     // Check that the second cookie is present in a private request
     check_cookie_presence("C2=V2", true, true, runNextTest);
   });
   tests.push(function() {
     // Check that the first cookie is not present in a private request
     check_cookie_presence("C1=V1", true, false, runNextTest);
   });
   tests.push(function() {
     // Check that the second cookie is not present in a non-private request
     check_cookie_presence("C2=V2", false, false, runNextTest);
   });
   // The following test only works in a non-e10s situation at the moment,
   // since the notification needs to run in the parent process but there is
   // no existing mechanism to make that happen.
   if (!inChildProcess()) {
     tests.push(function() {
       // Simulate all private browsing instances being closed
       var obsvc = Cc["@mozilla.org/observer-service;1"].
         getService(Ci.nsIObserverService);
       obsvc.notifyObservers(null, "last-pb-context-exited", null);
       // Check that all private cookies are now unavailable in new private requests
       check_cookie_presence("C2=V2", true, false, runNextTest);
     });
   }
   tests.push(function() { httpserver.stop(do_test_finished); });
   runNextTest();
 }