The Tor Browser / file revision

 /* 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)});

 }