The Tor Browser: toolkit/mozapps/update/tests/unit_aus_update/updateRootDirMigration_win.js@129ffea94266 (annotated)

toolkit/mozapps/update/tests/unit_aus_update/updateRootDirMigration_win.js@129ffea94266 (annotated)

toolkit/mozapps/update/tests/unit_aus_update/updateRootDirMigration_win.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/.
  */
 const PREF_APP_UPDATE_MIGRATE_APP_DIR = "app.update.migrated.updateDir";
 function clearTaskbarIDHash(exePath, appInfoName) {
   let registry = AUS_Cc["@mozilla.org/windows-registry-key;1"].
                  createInstance(AUS_Ci.nsIWindowsRegKey);
   try {
     registry.open(AUS_Ci.nsIWindowsRegKey.ROOT_KEY_CURRENT_USER,
                   "Software\\Mozilla\\" + appInfoName + "\\TaskBarIDs",
                   AUS_Ci.nsIWindowsRegKey.ACCESS_ALL);
     registry.removeValue(exePath);
   } catch (e) {
   }
   finally {
     registry.close();
   }
 }
 function setTaskbarIDHash(exePath, hash, appInfoName) {
   let registry = AUS_Cc["@mozilla.org/windows-registry-key;1"].
                  createInstance(AUS_Ci.nsIWindowsRegKey);
   try {
     registry.create(AUS_Ci.nsIWindowsRegKey.ROOT_KEY_CURRENT_USER,
                     "Software\\Mozilla\\" + appInfoName + "\\TaskBarIDs",
                     AUS_Ci.nsIWindowsRegKey.ACCESS_WRITE);
     registry.writeStringValue(exePath, hash);
   } catch (e) {
   }
   finally {
     registry.close();
   }
 };
 function getMigrated() {
   var migrated = 0;
   try {
     migrated = Services.prefs.getBoolPref(PREF_APP_UPDATE_MIGRATE_APP_DIR);
   } catch (e) {
   }
   return migrated;
 }
 /* General Update Manager Tests */
 function run_test() {
   setupTestCommon();
   standardInit();
   var appinfo = AUS_Cc["@mozilla.org/xre/app-info;1"].
                 getService(AUS_Ci.nsIXULAppInfo).
                 QueryInterface(AUS_Ci.nsIXULRuntime);
    // Obtain the old update root leaf
   var updateLeafName;
   var exeFile = FileUtils.getFile(XRE_EXECUTABLE_FILE, []);
   var programFiles = FileUtils.getFile("ProgF", []);
   if (exeFile.path.substring(0, programFiles.path.length).toLowerCase() ==
       programFiles.path.toLowerCase()) {
     updateLeafName = exeFile.parent.leafName;
   } else {
     updateLeafName = appinfo.name;
   }
   // Obtain the old update root
   var oldUpdateRoot;
   if (appinfo.vendor) {
     oldUpdateRoot = FileUtils.getDir("LocalAppData", [appinfo.vendor,
                                                       appinfo.name,
                                                       updateLeafName], false);
   } else {
     oldUpdateRoot = FileUtils.getDir("LocalAppData", [appinfo.name,
                                                       updateLeafName], false);
   }
   // Obtain the new update root
   var newUpdateRoot = FileUtils.getDir("UpdRootD", [], false);
   ///////////////////////////////////////////////////////////
   // Setting a taskbar ID without the old update dir existing should set the
   // pref so a migration isn't retried.
   // Remove the old and new update root directories
   try {
     oldUpdateRoot.remove(true);
   } catch (e) {
   }
   try {
     newUpdateRoot.remove(true);
   } catch (e) {
   }
   Services.prefs.setBoolPref(PREF_APP_UPDATE_MIGRATE_APP_DIR, false);
   setTaskbarIDHash(exeFile.parent.path, "AAAAAAAA", appinfo.name);
   initUpdateServiceStub();
   do_check_eq(getMigrated(), 1);
   ///////////////////////////////////////////////////////////
   // An application without a taskbar ID should bail early without a pref
   // being set, so that if a taskbar is set for the application, the migration
   // will be retried.
   Services.prefs.setBoolPref(PREF_APP_UPDATE_MIGRATE_APP_DIR, false);
   clearTaskbarIDHash(exeFile.parent.path, appinfo.name);
   initUpdateServiceStub();
   do_check_eq(getMigrated(), 0);
   ///////////////////////////////////////////////////////////
   // Migrating files should work
   Services.prefs.setBoolPref(PREF_APP_UPDATE_MIGRATE_APP_DIR, false);
   setTaskbarIDHash(exeFile.parent.path, "AAAAAAAA", appinfo.name);
   var oldUpdateDirs = oldUpdateRoot.clone();
   oldUpdateDirs.append("updates");
   oldUpdateDirs.append("0");
   oldUpdateDirs.create(AUS_Ci.nsIFile.DIRECTORY_TYPE, FileUtils.PERMS_DIRECTORY);
   // Get an array of all of the files we want to migrate.
   // We do this to create them in the old update directory.
   var filesToMigrate = [FILE_UPDATES_DB, FILE_UPDATE_ACTIVE,
                         ["updates", FILE_LAST_LOG], ["updates", FILE_BACKUP_LOG],
                         ["updates", "0", FILE_UPDATE_STATUS]];
   // Move each of those files to the new directory
   filesToMigrate.forEach(relPath => {
     let oldFile = oldUpdateRoot.clone();
     if (relPath instanceof Array) {
       relPath.forEach(relPathPart => {
         oldFile.append(relPathPart);
       });
     } else {
       oldFile.append(relPath);
     }
     oldFile.create(AUS_Ci.nsIFile.NORMAL_FILE_TYPE, PERMS_FILE);
   });
   // Do the migration
   initUpdateServiceStub();
   doTestFinish();
   return;
   // Now verify that each of the files exist in the new update directory
   filesToMigrate.forEach(relPath => {
     let newFile = newUpdateRoot.clone();
     let oldFile = oldUpdateRoot.clone();
     if (relPath instanceof Array) {
       relPath.forEach(relPathPart => {
         newFile.append(relPathPart);
         oldFile.append(relPathPart);
       });
     } else {
       newFile.append(relPath);
       oldFile.append(relPath);
     }
     // The file should be mimgrated, except for FILE_UPDATE_STATUS
     // which gets consumed by post update after it is migrated..
     if (newFile.leafName != FILE_UPDATE_STATUS) {
       do_check_true(newFile.exists());
     }
     do_check_false(oldFile.exists());
   });
   doTestFinish();
 }
 function end_test() {
   var appinfo = AUS_Cc["@mozilla.org/xre/app-info;1"].
                 getService(AUS_Ci.nsIXULAppInfo).
                 QueryInterface(AUS_Ci.nsIXULRuntime);
   var exeFile = FileUtils.getFile(XRE_EXECUTABLE_FILE, []);
   clearTaskbarIDHash(exeFile.parent.path, appinfo.name);
 }