The Tor Browser: toolkit/components/downloads/test/unit/test_bug_401430.js@129ffea94266 (annotated)

toolkit/components/downloads/test/unit/test_bug_401430.js@129ffea94266 (annotated)

toolkit/components/downloads/test/unit/test_bug_401430.js

Sat, 03 Jan 2015 20:18:00 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Sat, 03 Jan 2015 20:18:00 +0100
branch: TOR_BUG_3246
changeset 7: 129ffea94266
permissions: -rw-r--r--

Conditionally enable double key logic according to:
private browsing mode or privacy.thirdparty.isolate preference and
implement in GetCookieStringCommon and FindCookie where it counts...
With some reservations of how to convince FindCookie users to test
condition and pass a nullptr when disabling double key logic.

 /* 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/. */
 // This tests the "add to recent documents" functionality of the DM
 const nsIDownloadManager = Ci.nsIDownloadManager;
 const dm = Cc["@mozilla.org/download-manager;1"].getService(nsIDownloadManager);
 // Make sure Unicode is supported:
 // U+00E3 : LATIN SMALL LETTER A WITH TILDE
 // U+041B : CYRILLIC CAPITAL LETTER EL
 // U+3056 : HIRAGANA LETTER ZA
 const resultFileName = "test\u00e3\u041b\u3056" + Date.now() + ".doc";
 // Milliseconds between polls.
 const POLL_REGISTRY_TIMEOUT = 200;
 // Max number of polls.
 const POLL_REGISTRY_MAX_LOOPS = 25;
 function checkResult() {
   // delete the saved file (this doesn't affect the "recent documents" list)
   var resultFile = do_get_file(resultFileName);
   resultFile.remove(false);
   // Need to poll RecentDocs value because the SHAddToRecentDocs call
   // doesn't update the registry immediately.
   do_timeout(POLL_REGISTRY_TIMEOUT, pollRecentDocs);
 }
 var gPollsCount = 0;
 function pollRecentDocs() {
   if (++gPollsCount > POLL_REGISTRY_MAX_LOOPS) {
     do_throw("Maximum time elapsed while polling RecentDocs.");
     do_test_finished();
     return;
   }
   if (checkRecentDocsFor(resultFileName)) {
     print("Document found in RecentDocs");
     do_test_finished();
   }
   else
     do_timeout(POLL_REGISTRY_TIMEOUT, pollRecentDocs);
 }
 function checkRecentDocsFor(aFileName) {
   var recentDocsKey = Cc["@mozilla.org/windows-registry-key;1"].
                         createInstance(Ci.nsIWindowsRegKey);
   var recentDocsPath =
         "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\RecentDocs";
   recentDocsKey.open(Ci.nsIWindowsRegKey.ROOT_KEY_CURRENT_USER,
                      recentDocsPath,
                      Ci.nsIWindowsRegKey.ACCESS_READ);
   var count = recentDocsKey.valueCount;
   for (var i = 0; i < count; ++i) {
     var valueName = recentDocsKey.getValueName(i);
     var binValue = recentDocsKey.readBinaryValue(valueName);
     // "fields" in the data are separated by \0 wide characters, which are
     // returned as two \0 "bytes" by readBinaryValue. Use only the first field.
     var fileNameRaw = binValue.split("\0\0")[0];
     // Convert the filename from UTF-16LE.
     var fileName = "";
     for (var c = 0; c < fileNameRaw.length; c += 2)
       fileName += String.fromCharCode(fileNameRaw.charCodeAt(c) |
                                       fileNameRaw.charCodeAt(c+1) * 256);
     if (aFileName == fileName)
       return true;
   }
   return false;
 }
 var httpserv = null;
 function run_test()
 {
   if (oldDownloadManagerDisabled()) {
     return;
   }
   // This test functionality only implemented on Windows.
   // Is there a better way of doing this?
   var httpPH = Cc["@mozilla.org/network/protocol;1?name=http"].
                getService(Ci.nsIHttpProtocolHandler);
   if (httpPH.platform != "Windows")
     return;
   // Don't finish until the download is finished
   do_test_pending();
   httpserv = new HttpServer();
   httpserv.registerDirectory("/", do_get_cwd());
   httpserv.start(-1);
   var listener = {
     onDownloadStateChange: function test_401430_odsc(aState, aDownload) {
       if (aDownload.state == Ci.nsIDownloadManager.DOWNLOAD_FINISHED) {
         checkResult();
       }
     },
     onStateChange: function(a, b, c, d, e) { },
     onProgressChange: function(a, b, c, d, e, f, g) { },
     onSecurityChange: function(a, b, c, d) { }
   };
   dm.addListener(listener);
   dm.addListener(getDownloadListener());
   // need to save the file to the CWD, because the profile dir is in $TEMP,
   // and Windows apparently doesn't like putting things from $TEMP into
   // the recent files list.
   var dl = addDownload(httpserv,
                        {resultFileName: resultFileName,
                         targetFile: do_get_file(resultFileName, true)});
 }