home *** CD-ROM | disk | FTP | other *** search
/ PC World 2008 February / PCWorld_2008-02_cd.bin / temacd / songbird / Songbird_0.4_windows-i686.exe / xulrunner / components / nsUpdateService.js < prev    next >
Text File  |  2007-12-19  |  108KB  |  3,206 lines

  1. //@line 44 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  2.  
  3. const PREF_APP_UPDATE_ENABLED             = "app.update.enabled";
  4. const PREF_APP_UPDATE_AUTO                = "app.update.auto";
  5. const PREF_APP_UPDATE_MODE                = "app.update.mode";
  6. const PREF_APP_UPDATE_SILENT              = "app.update.silent";
  7. const PREF_APP_UPDATE_INTERVAL            = "app.update.interval";
  8. const PREF_APP_UPDATE_TIMER               = "app.update.timer";
  9. const PREF_APP_UPDATE_IDLETIME            = "app.update.idletime";
  10. const PREF_APP_UPDATE_LOG_BRANCH          = "app.update.log.";
  11. const PREF_APP_UPDATE_URL                 = "app.update.url";
  12. const PREF_APP_UPDATE_URL_OVERRIDE        = "app.update.url.override";
  13. const PREF_APP_UPDATE_URL_DETAILS         = "app.update.url.details";
  14. const PREF_APP_UPDATE_CHANNEL             = "app.update.channel";
  15. const PREF_APP_UPDATE_SHOW_INSTALLED_UI   = "app.update.showInstalledUI";
  16. const PREF_APP_UPDATE_LASTUPDATETIME_FMT  = "app.update.lastUpdateTime.%ID%";
  17. const PREF_GENERAL_USERAGENT_LOCALE       = "general.useragent.locale";
  18. const PREF_APP_UPDATE_INCOMPATIBLE_MODE   = "app.update.incompatible.mode";
  19. const PREF_UPDATE_NEVER_BRANCH            = "app.update.never.";
  20. const PREF_PARTNER_BRANCH                 = "app.partner.";
  21. const PREF_APP_DISTRIBUTION               = "distribution.id";
  22. const PREF_APP_DISTRIBUTION_VERSION       = "distribution.version";
  23.  
  24. const URI_UPDATE_PROMPT_DIALOG  = "chrome://mozapps/content/update/updates.xul";
  25. const URI_UPDATE_HISTORY_DIALOG = "chrome://mozapps/content/update/history.xul";
  26. const URI_BRAND_PROPERTIES      = "chrome://branding/locale/brand.properties";
  27. const URI_UPDATES_PROPERTIES    = "chrome://mozapps/locale/update/updates.properties";
  28. const URI_UPDATE_NS             = "http://www.mozilla.org/2005/app-update";
  29.  
  30. const KEY_APPDIR          = "XCurProcD";
  31. //@line 74 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  32. const KEY_UPDROOT         = "UpdRootD";
  33. const KEY_UAPPDATA        = "UAppData";
  34. //@line 77 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  35.  
  36. const DIR_UPDATES         = "updates";
  37. const FILE_UPDATE_STATUS  = "update.status";
  38. const FILE_UPDATE_ARCHIVE = "update.mar";
  39. const FILE_UPDATE_LOG     = "update.log"
  40. const FILE_UPDATES_DB     = "updates.xml";
  41. const FILE_UPDATE_ACTIVE  = "active-update.xml";
  42. const FILE_PERMS_TEST     = "update.test";
  43. const FILE_LAST_LOG       = "last-update.log";
  44.  
  45. const MODE_RDONLY   = 0x01;
  46. const MODE_WRONLY   = 0x02;
  47. const MODE_CREATE   = 0x08;
  48. const MODE_APPEND   = 0x10;
  49. const MODE_TRUNCATE = 0x20;
  50.  
  51. const PERMS_FILE      = 0644;
  52. const PERMS_DIRECTORY = 0755;
  53.  
  54. const STATE_NONE            = "null";
  55. const STATE_DOWNLOADING     = "downloading";
  56. const STATE_PENDING         = "pending";
  57. const STATE_APPLYING        = "applying";
  58. const STATE_SUCCEEDED       = "succeeded";
  59. const STATE_DOWNLOAD_FAILED = "download-failed";
  60. const STATE_FAILED          = "failed";
  61.  
  62. // From updater/errors.h:
  63. const WRITE_ERROR = 7;
  64.  
  65. const DOWNLOAD_CHUNK_SIZE           = 300000; // bytes
  66. const DOWNLOAD_BACKGROUND_INTERVAL  = 600;    // seconds
  67. const DOWNLOAD_FOREGROUND_INTERVAL  = 0;
  68.  
  69. const TOOLKIT_ID              = "toolkit@mozilla.org";
  70.  
  71. const POST_UPDATE_CONTRACTID = "@mozilla.org/updates/post-update;1";
  72.  
  73. const nsIExtensionManager     = Components.interfaces.nsIExtensionManager;
  74. const nsILocalFile            = Components.interfaces.nsILocalFile;
  75. const nsIUpdateService        = Components.interfaces.nsIUpdateService;
  76. const nsIUpdateItem           = Components.interfaces.nsIUpdateItem;
  77. const nsIPrefLocalizedString  = Components.interfaces.nsIPrefLocalizedString;
  78. const nsIIncrementalDownload  = Components.interfaces.nsIIncrementalDownload;
  79. const nsIFileInputStream      = Components.interfaces.nsIFileInputStream;
  80. const nsIFileOutputStream     = Components.interfaces.nsIFileOutputStream;
  81. const nsICryptoHash           = Components.interfaces.nsICryptoHash;
  82.  
  83. const Node = Components.interfaces.nsIDOMNode;
  84.  
  85. var gApp        = null;
  86. var gPref       = null;
  87. var gABI        = null;
  88. var gOSVersion  = null;
  89. var gConsole    = null;
  90. var gLogEnabled = { };
  91.  
  92. // shared code for suppressing bad cert dialogs
  93. //@line 40 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\shared\src\badCertHandler.js"
  94.  
  95. /**
  96.  * Only allow built-in certs for HTTPS connections.  See bug 340198.
  97.  */
  98. function checkCert(channel) {
  99.   if (!channel.originalURI.schemeIs("https"))  // bypass
  100.     return;
  101.  
  102.   const Ci = Components.interfaces;  
  103.   var cert =
  104.       channel.securityInfo.QueryInterface(Ci.nsISSLStatusProvider).
  105.       SSLStatus.QueryInterface(Ci.nsISSLStatus).serverCert;
  106.  
  107.   var issuer = cert.issuer;
  108.   while (issuer && !cert.equals(issuer)) {
  109.     cert = issuer;
  110.     issuer = cert.issuer;
  111.   }
  112.  
  113.   if (!issuer || issuer.tokenName != "Builtin Object Token")
  114.     throw "cert issuer is not built-in";
  115. }
  116.  
  117. /**
  118.  * This class implements nsIBadCertListener.  It's job is to prevent "bad cert"
  119.  * security dialogs from being shown to the user.  It is better to simply fail
  120.  * if the certificate is bad. See bug 304286.
  121.  */
  122. function BadCertHandler() {
  123. }
  124. BadCertHandler.prototype = {
  125.  
  126.   // nsIChannelEventSink
  127.   onChannelRedirect: function(oldChannel, newChannel, flags) {
  128.     // make sure the certificate of the old channel checks out before we follow
  129.     // a redirect from it.  See bug 340198.
  130.     checkCert(oldChannel);
  131.   },
  132.  
  133.   // nsIInterfaceRequestor
  134.   getInterface: function(iid) {
  135.     return this.QueryInterface(iid);
  136.   },
  137.  
  138.   // nsISupports
  139.   QueryInterface: function(iid) {
  140.     if (!iid.equals(Components.interfaces.nsIChannelEventSink) &&
  141.         !iid.equals(Components.interfaces.nsIInterfaceRequestor) &&
  142.         !iid.equals(Components.interfaces.nsISupports))
  143.       throw Components.results.NS_ERROR_NO_INTERFACE;
  144.     return this;
  145.   }
  146. };
  147. //@line 136 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  148.  
  149. /**
  150.  * Logs a string to the error console.
  151.  * @param   string
  152.  *          The string to write to the error console..
  153.  */
  154. function LOG(module, string) {
  155.   if (module in gLogEnabled || "all" in gLogEnabled) {
  156.     dump("*** " + module + ": " + string + "\n");
  157.     gConsole.logStringMessage(string);
  158.   }
  159. }
  160.  
  161. /**
  162.  * Convert a string containing binary values to hex.
  163.  */
  164. function binaryToHex(input) {
  165.   var result = "";
  166.   for (var i = 0; i < input.length; ++i) {
  167.     var hex = input.charCodeAt(i).toString(16);
  168.     if (hex.length == 1)
  169.       hex = "0" + hex;
  170.     result += hex;
  171.   }
  172.   return result;
  173. }
  174.  
  175. /**
  176.  * Gets a File URL spec for a nsIFile
  177.  * @param   file
  178.  *          The file to get a file URL spec to
  179.  * @returns The file URL spec to the file
  180.  */
  181. function getURLSpecFromFile(file) {
  182.   var ioServ = Components.classes["@mozilla.org/network/io-service;1"]
  183.                          .getService(Components.interfaces.nsIIOService);
  184.   var fph = ioServ.getProtocolHandler("file")
  185.                   .QueryInterface(Components.interfaces.nsIFileProtocolHandler);
  186.   return fph.getURLSpecFromFile(file);
  187. }
  188.  
  189. /**
  190.  * Gets the specified directory at the specified hierarchy under a
  191.  * Directory Service key.
  192.  * @param   key
  193.  *          The Directory Service Key to start from
  194.  * @param   pathArray
  195.  *          An array of path components to locate beneath the directory
  196.  *          specified by |key|
  197.  * @return  nsIFile object for the location specified. If the directory
  198.  *          requested does not exist, it is created, along with any
  199.  *          parent directories that need to be created.
  200.  */
  201. function getDir(key, pathArray) {
  202.   return getDirInternal(key, pathArray, true, false);
  203. }
  204.  
  205. /**
  206.  * Gets the specified directory at the specified hierarchy under a
  207.  * Directory Service key.
  208.  * @param   key
  209.  *          The Directory Service Key to start from
  210.  * @param   pathArray
  211.  *          An array of path components to locate beneath the directory
  212.  *          specified by |key|
  213.  * @return  nsIFile object for the location specified. If the directory
  214.  *          requested does not exist, it is NOT created.
  215.  */
  216. function getDirNoCreate(key, pathArray) {
  217.   return getDirInternal(key, pathArray, false, false);
  218. }
  219.  
  220. /**
  221.  * Gets the specified directory at the specified hierarchy under the
  222.  * update root directory.
  223.  * @param   pathArray
  224.  *          An array of path components to locate beneath the directory
  225.  *          specified by |key|
  226.  * @return  nsIFile object for the location specified. If the directory
  227.  *          requested does not exist, it is created, along with any
  228.  *          parent directories that need to be created.
  229.  */
  230. function getUpdateDir(pathArray) {
  231.   return getDirInternal(KEY_APPDIR, pathArray, true, true);
  232. }
  233.  
  234. /**
  235.  * Gets the specified directory at the specified hierarchy under a
  236.  * Directory Service key.
  237.  * @param   key
  238.  *          The Directory Service Key to start from
  239.  * @param   pathArray
  240.  *          An array of path components to locate beneath the directory
  241.  *          specified by |key|
  242.  * @param   shouldCreate
  243.  *          true if the directory hierarchy specified in |pathArray|
  244.  *          should be created if it does not exist,
  245.  *          false otherwise.
  246.  * @param   update
  247.  *          true if finding the update directory,
  248.  *          false otherwise.
  249.  * @return  nsIFile object for the location specified.
  250.  */
  251. function getDirInternal(key, pathArray, shouldCreate, update) {
  252.   var fileLocator = Components.classes["@mozilla.org/file/directory_service;1"]
  253.                               .getService(Components.interfaces.nsIProperties);
  254.   var dir = fileLocator.get(key, Components.interfaces.nsIFile);
  255. //@line 244 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  256.   if (update) {
  257.     try {
  258.       dir = fileLocator.get(KEY_UPDROOT, Components.interfaces.nsIFile);
  259.     } catch (e) {
  260.     }
  261.   }
  262. //@line 251 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  263.   for (var i = 0; i < pathArray.length; ++i) {
  264.     dir.append(pathArray[i]);
  265.     if (shouldCreate && !dir.exists())
  266.       dir.create(nsILocalFile.DIRECTORY_TYPE, PERMS_DIRECTORY);
  267.   }
  268.   return dir;
  269. }
  270.  
  271. /**
  272.  * Gets the file at the specified hierarchy under a Directory Service key.
  273.  * @param   key
  274.  *          The Directory Service Key to start from
  275.  * @param   pathArray
  276.  *          An array of path components to locate beneath the directory
  277.  *          specified by |key|. The last item in this array must be the
  278.  *          leaf name of a file.
  279.  * @return  nsIFile object for the file specified. The file is NOT created
  280.  *          if it does not exist, however all required directories along
  281.  *          the way are.
  282.  */
  283. function getFile(key, pathArray) {
  284.   var file = getDir(key, pathArray.slice(0, -1));
  285.   file.append(pathArray[pathArray.length - 1]);
  286.   return file;
  287. }
  288.  
  289. /**
  290.  * Gets the file at the specified hierarchy under the update root directory.
  291.  * @param   pathArray
  292.  *          An array of path components to locate beneath the directory
  293.  *          specified by |key|. The last item in this array must be the
  294.  *          leaf name of a file.
  295.  * @return  nsIFile object for the file specified. The file is NOT created
  296.  *          if it does not exist, however all required directories along
  297.  *          the way are.
  298.  */
  299. function getUpdateFile(pathArray) {
  300.   var file = getUpdateDir(pathArray.slice(0, -1));
  301.   file.append(pathArray[pathArray.length - 1]);
  302.   return file;
  303. }
  304.  
  305. /**
  306.  * Closes a Safe Output Stream
  307.  * @param   fos
  308.  *          The Safe Output Stream to close
  309.  */
  310. function closeSafeOutputStream(fos) {
  311.   if (fos instanceof Components.interfaces.nsISafeOutputStream) {
  312.     try {
  313.       fos.finish();
  314.     }
  315.     catch (e) {
  316.       fos.close();
  317.     }
  318.   }
  319.   else
  320.     fos.close();
  321. }
  322.  
  323. /**
  324.  * Returns human readable status text from the updates.properties bundle
  325.  * based on an error code
  326.  * @param   code
  327.  *          The error code to look up human readable status text for
  328.  * @param   defaultCode
  329.  *          The default code to look up should human readable status text
  330.  *          not exist for |code|
  331.  * @returns A human readable status text string
  332.  */
  333. function getStatusTextFromCode(code, defaultCode) {
  334.   var sbs =
  335.       Components.classes["@mozilla.org/intl/stringbundle;1"].
  336.       getService(Components.interfaces.nsIStringBundleService);
  337.   var updateBundle = sbs.createBundle(URI_UPDATES_PROPERTIES);
  338.   var reason = updateBundle.GetStringFromName("checker_error-" + defaultCode);
  339.   try {
  340.     reason = updateBundle.GetStringFromName("checker_error-" + code);
  341.     LOG("General", "Transfer Error: " + reason + ", code: " + code);
  342.   }
  343.   catch (e) {
  344.     // Use the default reason
  345.     LOG("General", "Transfer Error: " + reason + ", code: " + defaultCode);
  346.   }
  347.   return reason;
  348. }
  349.  
  350. /**
  351.  * Get the Active Updates directory
  352.  * @param   key
  353.  *          The Directory Service Key (optional).
  354.  *          If used, don't search local appdata on Win32 and don't create dir.
  355.  * @returns The active updates directory, as a nsIFile object
  356.  */
  357. function getUpdatesDir(key) {
  358.   // Right now, we only support downloading one patch at a time, so we always
  359.   // use the same target directory.
  360.   var fileLocator =
  361.       Components.classes["@mozilla.org/file/directory_service;1"].
  362.       getService(Components.interfaces.nsIProperties);
  363.   var appDir;
  364.   if (key)
  365.     appDir = fileLocator.get(key, Components.interfaces.nsIFile);
  366.   else {
  367.     appDir = fileLocator.get(KEY_APPDIR, Components.interfaces.nsIFile);
  368. //@line 357 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  369.     try {
  370.       appDir = fileLocator.get(KEY_UPDROOT, Components.interfaces.nsIFile);
  371.     } catch (e) {
  372.     }
  373. //@line 362 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  374.   }
  375.   appDir.append(DIR_UPDATES);
  376.   appDir.append("0");
  377.   if (!appDir.exists() && !key)
  378.     appDir.create(nsILocalFile.DIRECTORY_TYPE, PERMS_DIRECTORY);
  379.   return appDir;
  380. }
  381.  
  382. /**
  383.  * Reads the update state from the update.status file in the specified
  384.  * directory.
  385.  * @param   dir
  386.  *          The dir to look for an update.status file in
  387.  * @returns The status value of the update.
  388.  */
  389. function readStatusFile(dir) {
  390.   var statusFile = dir.clone();
  391.   statusFile.append(FILE_UPDATE_STATUS);
  392.   LOG("General", "Reading Status File: " + statusFile.path);
  393.   return readStringFromFile(statusFile) || STATE_NONE;
  394. }
  395.  
  396. /**
  397.  * Writes the current update operation/state to a file in the patch
  398.  * directory, indicating to the patching system that operations need
  399.  * to be performed.
  400.  * @param   dir
  401.  *          The patch directory where the update.status file should be
  402.  *          written.
  403.  * @param   state
  404.  *          The state value to write.
  405.  */
  406. function writeStatusFile(dir, state) {
  407.   var statusFile = dir.clone();
  408.   statusFile.append(FILE_UPDATE_STATUS);
  409.   writeStringToFile(statusFile, state);
  410. }
  411.  
  412. /**
  413.  * Removes the Updates Directory
  414.  * @param   key
  415.  *          The Directory Service Key under which update directory resides
  416.  *          (optional).
  417.  */
  418. function cleanUpUpdatesDir(key) {
  419.   // Bail out if we don't have appropriate permissions
  420.   var updateDir;
  421.   try {
  422.     updateDir = getUpdatesDir(key);
  423.   }
  424.   catch (e) {
  425.     return;
  426.   }
  427.  
  428.   var e = updateDir.directoryEntries;
  429.   while (e.hasMoreElements()) {
  430.     var f = e.getNext().QueryInterface(Components.interfaces.nsIFile);
  431.     // Preserve the last update log file for debugging purposes
  432.     if (f.leafName == FILE_UPDATE_LOG) {
  433.       try {
  434.         var dir = f.parent.parent;
  435.         var logFile = dir.clone();
  436.         logFile.append(FILE_LAST_LOG);
  437.         if (logFile.exists())
  438.           logFile.remove(false);
  439.         f.copyTo(dir, FILE_LAST_LOG);
  440.       }
  441.       catch (e) {
  442.         LOG("General", "Failed to copy file: " + f.path);
  443.       }
  444.     }
  445.     // Now, recursively remove this file.  The recusive removal is really
  446.     // only needed on Mac OSX because this directory will contain a copy of
  447.     // updater.app, which is itself a directory.
  448.     try {
  449.       f.remove(true);
  450.     }
  451.     catch (e) {
  452.       LOG("General", "Failed to remove file: " + f.path);
  453.     }
  454.   }
  455.   try {
  456.     updateDir.remove(false);
  457.   } catch (e) {
  458.     LOG("General", "Failed to remove update directory: " + updateDir.path +
  459.         " - This is almost always bad. Exception = " + e);
  460.     throw e;
  461.   }
  462. }
  463.  
  464. /**
  465.  * Clean up updates list and the updates directory.
  466.  * @param   key
  467.  *          The Directory Service Key under which update directory resides
  468.  *          (optional).
  469.  */
  470. function cleanupActiveUpdate(key) {
  471.   // Move the update from the Active Update list into the Past Updates list.
  472.   var um =
  473.       Components.classes["@mozilla.org/updates/update-manager;1"].
  474.       getService(Components.interfaces.nsIUpdateManager);
  475.   um.activeUpdate = null;
  476.   um.saveUpdates();
  477.  
  478.   // Now trash the updates directory, since we're done with it
  479.   cleanUpUpdatesDir(key);
  480. }
  481.  
  482. /**
  483.  * Gets a preference value, handling the case where there is no default.
  484.  * @param   func
  485.  *          The name of the preference function to call, on nsIPrefBranch
  486.  * @param   preference
  487.  *          The name of the preference
  488.  * @param   defaultValue
  489.  *          The default value to return in the event the preference has
  490.  *          no setting
  491.  * @returns The value of the preference, or undefined if there was no
  492.  *          user or default value.
  493.  */
  494. function getPref(func, preference, defaultValue) {
  495.   try {
  496.     return gPref[func](preference);
  497.   }
  498.   catch (e) {
  499.   }
  500.   return defaultValue;
  501. }
  502.  
  503. /**
  504.  * Gets the current value of the locale.  It's possible for this preference to
  505.  * be localized, so we have to do a little extra work here.  Similar code
  506.  * exists in nsHttpHandler.cpp when building the UA string.
  507.  */
  508. function getLocale() {
  509.   try {
  510.       // Get the default branch
  511.       var prefs = Components.classes["@mozilla.org/preferences-service;1"]
  512.           .getService(Components.interfaces.nsIPrefService);
  513.       var defaultPrefs = prefs.getDefaultBranch(null);
  514.       return defaultPrefs.getCharPref(PREF_GENERAL_USERAGENT_LOCALE);
  515.   } catch (e) {}
  516.  
  517.   return gPref.getCharPref(PREF_GENERAL_USERAGENT_LOCALE);
  518. }
  519.  
  520. /**
  521.  * Read the update channel from defaults only.  We do this to ensure that
  522.  * the channel is tightly coupled with the application and does not apply
  523.  * to other instances of the application that may use the same profile.
  524.  */
  525. function getUpdateChannel() {
  526.   var channel = "default";
  527.   var prefName;
  528.   var prefValue;
  529.  
  530.   var defaults =
  531.       gPref.QueryInterface(Components.interfaces.nsIPrefService).
  532.       getDefaultBranch(null);
  533.   try {
  534.     channel = defaults.getCharPref(PREF_APP_UPDATE_CHANNEL);
  535.   } catch (e) {
  536.     // use default when pref not found
  537.   }
  538.  
  539.   try {
  540.     var partners = gPref.getChildList(PREF_PARTNER_BRANCH, { });
  541.     if (partners.length) {
  542.       channel += "-cck";
  543.       partners.sort();
  544.  
  545.       for each (prefName in partners) {
  546.         prefValue = gPref.getCharPref(prefName);
  547.         channel += "-" + prefValue;
  548.       }
  549.     }
  550.   }
  551.   catch (e) {
  552.     Components.utils.reportError(e);
  553.   }
  554.  
  555.   return channel;
  556. }
  557.  
  558. /* Get the distribution pref values, from defaults only */
  559. function getDistributionPrefValue(aPrefName) {
  560.   var prefValue = "default";
  561.  
  562.   var defaults =
  563.       gPref.QueryInterface(Components.interfaces.nsIPrefService).
  564.       getDefaultBranch(null);
  565.   try {
  566.     prefValue = defaults.getCharPref(aPrefName);
  567.   } catch (e) {
  568.     // use default when pref not found
  569.   }
  570.  
  571.   return prefValue;
  572. }
  573.  
  574. /**
  575.  * An enumeration of items in a JS array.
  576.  * @constructor
  577.  */
  578. function ArrayEnumerator(aItems) {
  579.   this._index = 0;
  580.   if (aItems) {
  581.     for (var i = 0; i < aItems.length; ++i) {
  582.       if (!aItems[i])
  583.         aItems.splice(i, 1);
  584.     }
  585.   }
  586.   this._contents = aItems;
  587. }
  588.  
  589. ArrayEnumerator.prototype = {
  590.   _index: 0,
  591.   _contents: [],
  592.  
  593.   hasMoreElements: function() {
  594.     return this._index < this._contents.length;
  595.   },
  596.  
  597.   getNext: function() {
  598.     return this._contents[this._index++];
  599.   }
  600. };
  601.  
  602. /**
  603.  * Trims a prefix from a string.
  604.  * @param   string
  605.  *          The source string
  606.  * @param   prefix
  607.  *          The prefix to remove.
  608.  * @returns The suffix (string - prefix)
  609.  */
  610. function stripPrefix(string, prefix) {
  611.   return string.substr(prefix.length);
  612. }
  613.  
  614. /**
  615.  * Writes a string of text to a file.  A newline will be appended to the data
  616.  * written to the file.  This function only works with ASCII text.
  617.  */
  618. function writeStringToFile(file, text) {
  619.   var fos =
  620.       Components.classes["@mozilla.org/network/safe-file-output-stream;1"].
  621.       createInstance(nsIFileOutputStream);
  622.   var modeFlags = MODE_WRONLY | MODE_CREATE | MODE_TRUNCATE;
  623.   if (!file.exists())
  624.     file.create(nsILocalFile.NORMAL_FILE_TYPE, PERMS_FILE);
  625.   fos.init(file, modeFlags, PERMS_FILE, 0);
  626.   text += "\n";
  627.   fos.write(text, text.length);
  628.   closeSafeOutputStream(fos);
  629. }
  630.  
  631. /**
  632.  * Reads a string of text from a file.  A trailing newline will be removed
  633.  * before the result is returned.  This function only works with ASCII text.
  634.  */
  635. function readStringFromFile(file) {
  636.   var fis =
  637.       Components.classes["@mozilla.org/network/file-input-stream;1"].
  638.       createInstance(nsIFileInputStream);
  639.   var modeFlags = MODE_RDONLY;
  640.   if (!file.exists())
  641.     return null;
  642.   fis.init(file, modeFlags, PERMS_FILE, 0);
  643.   var sis =
  644.       Components.classes["@mozilla.org/scriptableinputstream;1"].
  645.       createInstance(Components.interfaces.nsIScriptableInputStream);
  646.   sis.init(fis);
  647.   var text = sis.read(sis.available());
  648.   sis.close();
  649.   if (text[text.length - 1] == "\n")
  650.     text = text.slice(0, -1);
  651.   return text;
  652. }
  653.  
  654. function getObserverService()
  655. {
  656.   return Components.classes["@mozilla.org/observer-service;1"]
  657.                    .getService(Components.interfaces.nsIObserverService);
  658. }
  659.  
  660. /**
  661.  * Update Patch
  662.  * @param   patch
  663.  *          A <patch> element to initialize this object with
  664.  * @throws if patch has a size of 0
  665.  * @constructor
  666.  */
  667. function UpdatePatch(patch) {
  668.   this._properties = {};
  669.   for (var i = 0; i < patch.attributes.length; ++i) {
  670.     var attr = patch.attributes.item(i);
  671.     attr.QueryInterface(Components.interfaces.nsIDOMAttr);
  672.     switch (attr.name) {
  673.     case "selected":
  674.       this.selected = attr.value == "true";
  675.       break;
  676.     case "size":
  677.       if (0 == parseInt(attr.value)) {
  678.         LOG("UpdatePatch", "0-sized patch!");
  679.         throw Components.results.NS_ERROR_ILLEGAL_VALUE;
  680.       }
  681.       // fall through
  682.     default:
  683.       this[attr.name] = attr.value;
  684.       break;
  685.     };
  686.   }
  687. }
  688. UpdatePatch.prototype = {
  689.   /**
  690.    * See nsIUpdateService.idl
  691.    */
  692.   serialize: function(updates) {
  693.     var patch = updates.createElementNS(URI_UPDATE_NS, "patch");
  694.     patch.setAttribute("type", this.type);
  695.     patch.setAttribute("URL", this.URL);
  696.     patch.setAttribute("hashFunction", this.hashFunction);
  697.     patch.setAttribute("hashValue", this.hashValue);
  698.     patch.setAttribute("size", this.size);
  699.     patch.setAttribute("selected", this.selected);
  700.     patch.setAttribute("state", this.state);
  701.  
  702.     for (var p in this._properties) {
  703.       if (this._properties[p].present)
  704.         patch.setAttribute(p, this._properties[p].data);
  705.     }
  706.  
  707.     return patch;
  708.   },
  709.  
  710.   /**
  711.    * A hash of custom properties
  712.    */
  713.   _properties: null,
  714.  
  715.   /**
  716.    * See nsIWritablePropertyBag.idl
  717.    */
  718.   setProperty: function(name, value) {
  719.     this._properties[name] = { data: value, present: true };
  720.   },
  721.  
  722.   /**
  723.    * See nsIWritablePropertyBag.idl
  724.    */
  725.   deleteProperty: function(name) {
  726.     if (name in this._properties)
  727.       this._properties[name].present = false;
  728.     else
  729.       throw Components.results.NS_ERROR_FAILURE;
  730.   },
  731.  
  732.   /**
  733.    * See nsIPropertyBag.idl
  734.    */
  735.   get enumerator() {
  736.     var properties = [];
  737.     for (var p in this._properties)
  738.       properties.push(this._properties[p].data);
  739.     return new ArrayEnumerator(properties);
  740.   },
  741.  
  742.   /**
  743.    * See nsIPropertyBag.idl
  744.    */
  745.   getProperty: function(name) {
  746.     if (name in this._properties &&
  747.         this._properties[name].present)
  748.       return this._properties[name].data;
  749.     throw Components.results.NS_ERROR_FAILURE;
  750.   },
  751.  
  752.   /**
  753.    * Returns whether or not the update.status file for this patch exists at the
  754.    * appropriate location.
  755.    */
  756.   get statusFileExists() {
  757.     var statusFile = getUpdatesDir();
  758.     statusFile.append(FILE_UPDATE_STATUS);
  759.     return statusFile.exists();
  760.   },
  761.  
  762.   /**
  763.    * See nsIUpdateService.idl
  764.    */
  765.   get state() {
  766.     if (!this.statusFileExists)
  767.       return STATE_NONE;
  768.     return this._properties.state;
  769.   },
  770.   set state(val) {
  771.     this._properties.state = val;
  772.   },
  773.  
  774.   /**
  775.    * See nsISupports.idl
  776.    */
  777.   QueryInterface: function(iid) {
  778.     if (!iid.equals(Components.interfaces.nsIUpdatePatch) &&
  779.         !iid.equals(Components.interfaces.nsIPropertyBag) &&
  780.         !iid.equals(Components.interfaces.nsIWritablePropertyBag) &&
  781.         !iid.equals(Components.interfaces.nsISupports))
  782.       throw Components.results.NS_ERROR_NO_INTERFACE;
  783.     return this;
  784.   }
  785. };
  786.  
  787. /**
  788.  * Update
  789.  * Implements nsIUpdate
  790.  * @param   update
  791.  *          An <update> element to initialize this object with
  792.  * @throws if the update contains no patches
  793.  * @constructor
  794.  */
  795. function Update(update) {
  796.   this._properties = {};
  797.   this._patches = [];
  798.   this.installDate = 0;
  799.   this.isCompleteUpdate = false;
  800.   this.channel = "default"
  801.  
  802.   // Null <update>, assume this is a message container and do no
  803.   // further initialization
  804.   if (!update)
  805.     return;
  806.  
  807.   for (var i = 0; i < update.childNodes.length; ++i) {
  808.     var patchElement = update.childNodes.item(i);
  809.     if (patchElement.nodeType != Node.ELEMENT_NODE ||
  810.         patchElement.localName != "patch")
  811.       continue;
  812.  
  813.     patchElement.QueryInterface(Components.interfaces.nsIDOMElement);
  814.     try {
  815.       var patch = new UpdatePatch(patchElement);
  816.     } catch (e) {
  817.       continue;
  818.     }
  819.     this._patches.push(patch);
  820.   }
  821.  
  822.   if (0 == this._patches.length)
  823.     throw Components.results.NS_ERROR_ILLEGAL_VALUE;
  824.  
  825.   for (var i = 0; i < update.attributes.length; ++i) {
  826.     var attr = update.attributes.item(i);
  827.     attr.QueryInterface(Components.interfaces.nsIDOMAttr);
  828.     if (attr.name == "installDate" && attr.value)
  829.       this.installDate = parseInt(attr.value);
  830.     else if (attr.name == "isCompleteUpdate")
  831.       this.isCompleteUpdate = attr.value == "true";
  832.     else if (attr.name == "isSecurityUpdate")
  833.       this.isSecurityUpdate = attr.value == "true";
  834.     else if (attr.name == "detailsURL")
  835.       this._detailsURL = attr.value;
  836.     else if (attr.name == "channel")
  837.       this.channel = attr.value;
  838.     else
  839.       this[attr.name] = attr.value;
  840.   }
  841.  
  842.   // The Update Name is either the string provided by the <update> element, or
  843.   // the string: "<App Name> <Update App Version>"
  844.   var name = "";
  845.   if (update.hasAttribute("name"))
  846.     name = update.getAttribute("name");
  847.   else {
  848.     var sbs = Components.classes["@mozilla.org/intl/stringbundle;1"]
  849.                         .getService(Components.interfaces.nsIStringBundleService);
  850.     var brandBundle = sbs.createBundle(URI_BRAND_PROPERTIES);
  851.     var updateBundle = sbs.createBundle(URI_UPDATES_PROPERTIES);
  852.     var appName = brandBundle.GetStringFromName("brandShortName");
  853.     name = updateBundle.formatStringFromName("updateName",
  854.                                              [appName, this.version], 2);
  855.   }
  856.   this.name = name;
  857. }
  858. Update.prototype = {
  859.   /**
  860.    * See nsIUpdateService.idl
  861.    */
  862.   get patchCount() {
  863.     return this._patches.length;
  864.   },
  865.  
  866.   /**
  867.    * See nsIUpdateService.idl
  868.    */
  869.   getPatchAt: function(index) {
  870.     return this._patches[index];
  871.   },
  872.  
  873.   /**
  874.    * See nsIUpdateService.idl
  875.    *
  876.    * We use a copy of the state cached on this object in |_state| only when
  877.    * there is no selected patch, i.e. in the case when we could not load
  878.    * |.activeUpdate| from the update manager for some reason but still have
  879.    * the update.status file to work with.
  880.    */
  881.   _state: "",
  882.   set state(state) {
  883.     if (this.selectedPatch)
  884.       this.selectedPatch.state = state;
  885.     this._state = state;
  886.     return state;
  887.   },
  888.   get state() {
  889.     if (this.selectedPatch)
  890.       return this.selectedPatch.state;
  891.     return this._state;
  892.   },
  893.  
  894.   /**
  895.    * See nsIUpdateService.idl
  896.    */
  897.   errorCode: 0,
  898.  
  899.   /**
  900.    * See nsIUpdateService.idl
  901.    */
  902.   get selectedPatch() {
  903.     for (var i = 0; i < this.patchCount; ++i) {
  904.       if (this._patches[i].selected)
  905.         return this._patches[i];
  906.     }
  907.     return null;
  908.   },
  909.  
  910.   /**
  911.    * See nsIUpdateService.idl
  912.    */
  913.   get detailsURL() {
  914.     if (!this._detailsURL) {
  915.       try {
  916.         // Try using a default details URL supplied by the distribution
  917.         // if the update XML does not supply one.
  918.         var formatter = Components.classes["@mozilla.org/toolkit/URLFormatterService;1"]
  919.                                   .getService(Components.interfaces.nsIURLFormatter);
  920.         return formatter.formatURLPref(PREF_APP_UPDATE_URL_DETAILS);
  921.       }
  922.       catch (e) {
  923.       }
  924.     }
  925.     return this._detailsURL || "";
  926.   },
  927.  
  928.   /**
  929.    * See nsIUpdateService.idl
  930.    */
  931.   serialize: function(updates) {
  932.     var update = updates.createElementNS(URI_UPDATE_NS, "update");
  933.     update.setAttribute("type", this.type);
  934.     update.setAttribute("name", this.name);
  935.     update.setAttribute("version", this.version);
  936.     update.setAttribute("platformVersion", this.platformVersion);
  937.     update.setAttribute("extensionVersion", this.extensionVersion);
  938.     update.setAttribute("detailsURL", this.detailsURL);
  939.     update.setAttribute("licenseURL", this.licenseURL);
  940.     update.setAttribute("serviceURL", this.serviceURL);
  941.     update.setAttribute("installDate", this.installDate);
  942.     update.setAttribute("statusText", this.statusText);
  943.     update.setAttribute("buildID", this.buildID);
  944.     update.setAttribute("isCompleteUpdate", this.isCompleteUpdate);
  945.     update.setAttribute("channel", this.channel);
  946.     updates.documentElement.appendChild(update);
  947.  
  948.     for (var p in this._properties) {
  949.       if (this._properties[p].present)
  950.         update.setAttribute(p, this._properties[p].data);
  951.     }
  952.  
  953.     for (var i = 0; i < this.patchCount; ++i)
  954.       update.appendChild(this.getPatchAt(i).serialize(updates));
  955.  
  956.     return update;
  957.   },
  958.  
  959.   /**
  960.    * A hash of custom properties
  961.    */
  962.   _properties: null,
  963.  
  964.   /**
  965.    * See nsIWritablePropertyBag.idl
  966.    */
  967.   setProperty: function(name, value) {
  968.     this._properties[name] = { data: value, present: true };
  969.   },
  970.  
  971.   /**
  972.    * See nsIWritablePropertyBag.idl
  973.    */
  974.   deleteProperty: function(name) {
  975.     if (name in this._properties)
  976.       this._properties[name].present = false;
  977.     else
  978.       throw Components.results.NS_ERROR_FAILURE;
  979.   },
  980.  
  981.   /**
  982.    * See nsIPropertyBag.idl
  983.    */
  984.   get enumerator() {
  985.     var properties = [];
  986.     for (var p in this._properties)
  987.       properties.push(this._properties[p].data);
  988.     return new ArrayEnumerator(properties);
  989.   },
  990.  
  991.   /**
  992.    * See nsIPropertyBag.idl
  993.    */
  994.   getProperty: function(name) {
  995.     if (name in this._properties &&
  996.         this._properties[name].present)
  997.       return this._properties[name].data;
  998.     throw Components.results.NS_ERROR_FAILURE;
  999.   },
  1000.  
  1001.   /**
  1002.    * See nsISupports.idl
  1003.    */
  1004.   QueryInterface: function(iid) {
  1005.     if (!iid.equals(Components.interfaces.nsIUpdate) &&
  1006.         !iid.equals(Components.interfaces.nsIPropertyBag) &&
  1007.         !iid.equals(Components.interfaces.nsIWritablePropertyBag) &&
  1008.         !iid.equals(Components.interfaces.nsISupports))
  1009.       throw Components.results.NS_ERROR_NO_INTERFACE;
  1010.     return this;
  1011.   }
  1012. };
  1013.  
  1014. /**
  1015.  * UpdateService
  1016.  * A Service for managing the discovery and installation of software updates.
  1017.  * @constructor
  1018.  */
  1019. function UpdateService() {
  1020.   gApp  = Components.classes["@mozilla.org/xre/app-info;1"]
  1021.                     .getService(Components.interfaces.nsIXULAppInfo)
  1022.                     .QueryInterface(Components.interfaces.nsIXULRuntime);
  1023.   gPref = Components.classes["@mozilla.org/preferences-service;1"]
  1024.                     .getService(Components.interfaces.nsIPrefBranch2);
  1025.   gConsole = Components.classes["@mozilla.org/consoleservice;1"]
  1026.                        .getService(Components.interfaces.nsIConsoleService);
  1027.  
  1028.   // Not all builds have a known ABI
  1029.   try {
  1030.     gABI = gApp.XPCOMABI;
  1031.   }
  1032.   catch (e) {
  1033.     LOG("UpdateService", "XPCOM ABI unknown: updates are not possible.");
  1034.   }
  1035.  
  1036.   try {
  1037.     var sysInfo =
  1038.       Components.classes["@mozilla.org/system-info;1"]
  1039.                 .getService(Components.interfaces.nsIPropertyBag2);
  1040.  
  1041.     gOSVersion = encodeURIComponent(sysInfo.getProperty("name") + " " +
  1042.                                     sysInfo.getProperty("version"));
  1043.   }
  1044.   catch (e) {
  1045.     LOG("UpdateService", "OS Version unknown: updates are not possible.");
  1046.   }
  1047.  
  1048. //@line 1045 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  1049.  
  1050.   // Start the update timer only after a profile has been selected so that the
  1051.   // appropriate values for the update check are read from the user's profile.
  1052.   var os = getObserverService();
  1053.  
  1054.   os.addObserver(this, "profile-after-change", false);
  1055.  
  1056.   // Observe xpcom-shutdown to unhook pref branch observers above to avoid
  1057.   // shutdown leaks.
  1058.   os.addObserver(this, "xpcom-shutdown", false);
  1059. }
  1060.  
  1061. UpdateService.prototype = {
  1062.   /**
  1063.    * The downloader we are using to download updates. There is only ever one of
  1064.    * these.
  1065.    */
  1066.   _downloader: null,
  1067.  
  1068.   /**
  1069.    * Handle Observer Service notifications
  1070.    * @param   subject
  1071.    *          The subject of the notification
  1072.    * @param   topic
  1073.    *          The notification name
  1074.    * @param   data
  1075.    *          Additional data
  1076.    */
  1077.   observe: function(subject, topic, data) {
  1078.     var os = getObserverService();
  1079.  
  1080.     switch (topic) {
  1081.     case "profile-after-change":
  1082.       os.removeObserver(this, "profile-after-change");
  1083.       this._start();
  1084.       break;
  1085.     case "xpcom-shutdown":
  1086.       os.removeObserver(this, "xpcom-shutdown");
  1087.  
  1088.       // Release Services
  1089.       gApp      = null;
  1090.       gPref     = null;
  1091.       gConsole  = null;
  1092.       break;
  1093.     }
  1094.   },
  1095.  
  1096.   /**
  1097.    * Start the Update Service
  1098.    */
  1099.   _start: function() {
  1100.     // Start logging
  1101.     this._initLoggingPrefs();
  1102.  
  1103.     // Clean up any extant updates
  1104.     this._postUpdateProcessing();
  1105.  
  1106.     // Register a background update check timer
  1107.     var tm =
  1108.         Components.classes["@mozilla.org/updates/timer-manager;1"]
  1109.                   .getService(Components.interfaces.nsIUpdateTimerManager);
  1110.     var interval = getPref("getIntPref", PREF_APP_UPDATE_INTERVAL, 86400);
  1111.     tm.registerTimer("background-update-timer", this, interval);
  1112.  
  1113.     // Resume fetching...
  1114.     var um = Components.classes["@mozilla.org/updates/update-manager;1"]
  1115.                         .getService(Components.interfaces.nsIUpdateManager);
  1116.     var activeUpdate = um.activeUpdate;
  1117.     if (activeUpdate) {
  1118.       var status = this.downloadUpdate(activeUpdate, true);
  1119.       if (status == STATE_NONE)
  1120.         cleanupActiveUpdate();
  1121.     }
  1122.   },
  1123.  
  1124.   /**
  1125.    * Perform post-processing on updates lingering in the updates directory
  1126.    * from a previous browser session - either report install failures (and
  1127.    * optionally attempt to fetch a different version if appropriate) or
  1128.    * notify the user of install success.
  1129.    */
  1130.   _postUpdateProcessing: function() {
  1131.     // Detect installation failures and notify
  1132.  
  1133.     // Bail out if we don't have appropriate permissions
  1134.     if (!this.canUpdate)
  1135.       return;
  1136.  
  1137.     var status = readStatusFile(getUpdatesDir());
  1138.  
  1139.     // Make sure to cleanup after an update that failed for an unknown reason
  1140.     if (status == "null")
  1141.       status = null;
  1142.  
  1143.     var updRootKey = null;
  1144. //@line 1141 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  1145.     function findPreviousUpdate(key) {
  1146.       var updateDir = getUpdatesDir(key);
  1147.       if (updateDir.exists()) {
  1148.         status = readStatusFile(updateDir);
  1149.         // Previous download should succeed. Otherwise, we will not be here!
  1150.         if (status == STATE_SUCCEEDED)
  1151.           updRootKey = key;
  1152.         else
  1153.           status = null;
  1154.       }
  1155.     }
  1156.  
  1157.     // required when updating from Fx 2.0.0.1 to 2.0.0.3 (or later)
  1158.     // on Windows Vista.
  1159.     if (status == null)
  1160.       findPreviousUpdate(KEY_UAPPDATA);
  1161.  
  1162.     // required to migrate from older versions.
  1163.     if (status == null)
  1164.       findPreviousUpdate(KEY_APPDIR);
  1165. //@line 1162 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  1166.  
  1167.     if (status == STATE_DOWNLOADING) {
  1168.       LOG("UpdateService", "_postUpdateProcessing: Downloading patch, resuming...");
  1169.     }
  1170.     else if (status != null) {
  1171.       // null status means the update.status file is not present, because either:
  1172.       // 1) no update was performed, and so there's no UI to show
  1173.       // 2) an update was attempted but failed during checking, transfer or
  1174.       //    verification, and was cleaned up at that point, and UI notifying of
  1175.       //    that error was shown at that stage.
  1176.       var um =
  1177.           Components.classes["@mozilla.org/updates/update-manager;1"].
  1178.           getService(Components.interfaces.nsIUpdateManager);
  1179.       var prompter =
  1180.           Components.classes["@mozilla.org/updates/update-prompt;1"].
  1181.           createInstance(Components.interfaces.nsIUpdatePrompt);
  1182.  
  1183.       var shouldCleanup = true;
  1184.       var update = um.activeUpdate;
  1185.       if (!update) {
  1186.         update = new Update(null);
  1187.       }
  1188.       update.state = status;
  1189.       var sbs =
  1190.           Components.classes["@mozilla.org/intl/stringbundle;1"].
  1191.           getService(Components.interfaces.nsIStringBundleService);
  1192.       var bundle = sbs.createBundle(URI_UPDATES_PROPERTIES);
  1193.       if (status == STATE_SUCCEEDED) {
  1194.         update.statusText = bundle.GetStringFromName("installSuccess");
  1195.  
  1196.         // Dig through the update history to find the patch that was just
  1197.         // installed and update its metadata.
  1198.         for (var i = 0; i < um.updateCount; ++i) {
  1199.           var umUpdate = um.getUpdateAt(i);
  1200.           if (umUpdate && umUpdate.version == update.version &&
  1201.                           umUpdate.buildID == update.buildID) {
  1202.             umUpdate.statusText = update.statusText;
  1203.             break;
  1204.           }
  1205.         }
  1206.  
  1207.         LOG("UpdateService", "_postUpdateProcessing: Install Succeeded, Showing UI");
  1208.         prompter.showUpdateInstalled(update);
  1209. //@line 1209 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  1210.         // Perform platform-specific post-update processing.
  1211.         if (POST_UPDATE_CONTRACTID in Components.classes) {
  1212.           Components.classes[POST_UPDATE_CONTRACTID].
  1213.               createInstance(Components.interfaces.nsIRunnable).run();
  1214.         }
  1215. //@line 1215 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  1216.  
  1217.         // Done with this update. Clean it up.
  1218.         cleanupActiveUpdate(updRootKey);
  1219.       }
  1220.       else {
  1221.         // If we hit an error, then the error code will be included in the
  1222.         // status string following a colon.  If we had an I/O error, then we
  1223.         // assume that the patch is not invalid, and we restage the patch so
  1224.         // that it can be attempted again the next time we restart.
  1225.         var ary = status.split(": ");
  1226.         update.state = ary[0];
  1227.         if (update.state == STATE_FAILED && ary[1]) {
  1228.           update.errorCode = ary[1];
  1229.           if (update.errorCode == WRITE_ERROR) {
  1230.             prompter.showUpdateError(update);
  1231.             writeStatusFile(getUpdatesDir(), update.state = STATE_PENDING);
  1232.             return;
  1233.           }
  1234.         }
  1235.  
  1236.         // Something went wrong with the patch application process.
  1237.         cleanupActiveUpdate();
  1238.  
  1239.         update.statusText = bundle.GetStringFromName("patchApplyFailure");
  1240.         var oldType = update.selectedPatch ? update.selectedPatch.type
  1241.                                            : "complete";
  1242.         if (update.selectedPatch && oldType == "partial") {
  1243.           // Partial patch application failed, try downloading the complete
  1244.           // update in the background instead.
  1245.           LOG("UpdateService", "_postUpdateProcessing: Install of Partial Patch " +
  1246.               "failed, downloading Complete Patch and maybe showing UI");
  1247.           var status = this.downloadUpdate(update, true);
  1248.           if (status == STATE_NONE)
  1249.             cleanupActiveUpdate();
  1250.         }
  1251.         else {
  1252.           LOG("UpdateService", "_postUpdateProcessing: Install of Complete or " +
  1253.               "only patch failed. Showing error.");
  1254.         }
  1255.         update.QueryInterface(Components.interfaces.nsIWritablePropertyBag);
  1256.         update.setProperty("patchingFailed", oldType);
  1257.         prompter.showUpdateError(update);
  1258.       }
  1259.     }
  1260.     else {
  1261.       LOG("UpdateService", "_postUpdateProcessing: No Status, No Update");
  1262.     }
  1263.   },
  1264.  
  1265.   /**
  1266.    * Initialize Logging preferences, formatted like so:
  1267.    *  app.update.log.<moduleName> = <true|false>
  1268.    */
  1269.   _initLoggingPrefs: function() {
  1270.     try {
  1271.       var ps = Components.classes["@mozilla.org/preferences-service;1"]
  1272.                         .getService(Components.interfaces.nsIPrefService);
  1273.       var logBranch = ps.getBranch(PREF_APP_UPDATE_LOG_BRANCH);
  1274.       var modules = logBranch.getChildList("", { value: 0 });
  1275.  
  1276.       for (var i = 0; i < modules.length; ++i) {
  1277.         if (logBranch.prefHasUserValue(modules[i]))
  1278.           gLogEnabled[modules[i]] = logBranch.getBoolPref(modules[i]);
  1279.       }
  1280.     }
  1281.     catch (e) {
  1282.     }
  1283.   },
  1284.  
  1285.   /**
  1286.    * Notified when a timer fires
  1287.    * @param   timer
  1288.    *          The timer that fired
  1289.    */
  1290.   notify: function(timer) {
  1291.     // If a download is in progress, then do nothing.
  1292.     if (this.isDownloading || this._downloader && this._downloader.patchIsStaged)
  1293.       return;
  1294.  
  1295.     var self = this;
  1296.     var listener = {
  1297.       /**
  1298.        * See nsIUpdateService.idl
  1299.        */
  1300.       onProgress: function(request, position, totalSize) {
  1301.       },
  1302.  
  1303.       /**
  1304.        * See nsIUpdateService.idl
  1305.        */
  1306.       onCheckComplete: function(request, updates, updateCount) {
  1307.         self._selectAndInstallUpdate(updates);
  1308.       },
  1309.  
  1310.       /**
  1311.        * See nsIUpdateService.idl
  1312.        */
  1313.       onError: function(request, update) {
  1314.         LOG("Checker", "Error during background update: " + update.statusText);
  1315.       },
  1316.     }
  1317.     this.backgroundChecker.checkForUpdates(listener, false);
  1318.   },
  1319.  
  1320.   /**
  1321.    * Determine whether or not an update requires user confirmation before it
  1322.    * can be installed.
  1323.    * @param   update
  1324.    *          The update to be installed
  1325.    * @returns true if a prompt UI should be shown asking the user if they want
  1326.    *          to install the update, false if the update should just be
  1327.    *          silently downloaded and installed.
  1328.    */
  1329.   _shouldPrompt: function(update) {
  1330.     // There are two possible outcomes here:
  1331.     // 1. download and install the update automatically
  1332.     // 2. alert the user about the presence of an update before doing anything
  1333.     //
  1334.     // The outcome we follow is determined as follows:
  1335.     //
  1336.     // Note:  all Major updates require notification and confirmation
  1337.     //
  1338.     // Update Type      Mode      Incompatible    Outcome
  1339.     // Major            0         Yes or No       Notify and Confirm
  1340.     // Major            1         No              Notify and Confirm
  1341.     // Major            1         Yes             Notify and Confirm
  1342.     // Major            2         Yes or No       Notify and Confirm
  1343.     // Minor            0         Yes or No       Auto Install
  1344.     // Minor            1         No              Auto Install
  1345.     // Minor            1         Yes             Notify and Confirm
  1346.     // Minor            2         No              Auto Install
  1347.     // Minor            2         Yes             Notify and Confirm
  1348.     //
  1349.     // In addition, if there is a license associated with an update, regardless
  1350.     // of type it must be agreed to.
  1351.     //
  1352.     // If app.update.enabled is set to false, an update check is not performed
  1353.     // at all, and so none of the decision making above is entered into.
  1354.     //
  1355.     if (update.type == "major") {
  1356.       LOG("Checker", "_shouldPrompt: Prompting because it is a major update");
  1357.       return true;
  1358.     }
  1359.  
  1360.     update.QueryInterface(Components.interfaces.nsIPropertyBag);
  1361.     try {
  1362.       var licenseAccepted = update.getProperty("licenseAccepted") == "true";
  1363.     }
  1364.     catch (e) {
  1365.       licenseAccepted = false;
  1366.     }
  1367.  
  1368.     var updateEnabled = getPref("getBoolPref", PREF_APP_UPDATE_ENABLED, true);
  1369.     if (!updateEnabled) {
  1370.       LOG("Checker", "_shouldPrompt: Not prompting because update is " +
  1371.           "disabled");
  1372.       return false;
  1373.     }
  1374.  
  1375.     // User has turned off automatic download and install
  1376.     var autoEnabled = getPref("getBoolPref", PREF_APP_UPDATE_AUTO, true);
  1377.     if (!autoEnabled) {
  1378.       LOG("Checker", "_shouldPrompt: Prompting because auto install is disabled");
  1379.       return true;
  1380.     }
  1381.  
  1382.     switch (getPref("getIntPref", PREF_APP_UPDATE_MODE, 1)) {
  1383.     case 1:
  1384.       // Mode 1 is do not prompt only if there are no incompatibilities
  1385.       // releases
  1386.       LOG("Checker", "_shouldPrompt: Prompting if there are incompatibilities");
  1387.       return !isCompatible(update);
  1388.     case 2:
  1389.       // Mode 2 is do not prompt only if there are no incompatibilities
  1390.       LOG("Checker", "_shouldPrompt: Prompting if there are incompatibilities");
  1391.       return !isCompatible(update);
  1392.     }
  1393.     // Mode 0 is do not prompt regardless of incompatibilities
  1394.     LOG("Checker", "_shouldPrompt: Not prompting the user - they choose to " +
  1395.         "ignore incompatibilities");
  1396.     return false;
  1397.   },
  1398.  
  1399.   /**
  1400.    * Determine which of the specified updates should be installed.
  1401.    * @param   updates
  1402.    *          An array of available updates
  1403.    */
  1404.   selectUpdate: function(updates) {
  1405.     if (updates.length == 0)
  1406.       return null;
  1407.  
  1408.     // Choose the newest of the available minor and major updates.
  1409.     var majorUpdate = null, minorUpdate = null;
  1410.     var newestMinor = updates[0], newestMajor = updates[0];
  1411.  
  1412.     var vc = Components.classes["@mozilla.org/xpcom/version-comparator;1"]
  1413.                        .getService(Components.interfaces.nsIVersionComparator);
  1414.     for (var i = 0; i < updates.length; ++i) {
  1415.       if (updates[i].type == "major" &&
  1416.           vc.compare(newestMajor.version, updates[i].version) <= 0)
  1417.         majorUpdate = newestMajor = updates[i];
  1418.       if (updates[i].type == "minor" &&
  1419.           vc.compare(newestMinor.version, updates[i].version) <= 0)
  1420.         minorUpdate = newestMinor = updates[i];
  1421.     }
  1422.  
  1423.     // IMPORTANT
  1424.     // If there's a minor update, always try and fetch that one first,
  1425.     // otherwise use the newest major update.
  1426.     // selectUpdate() only returns one update.
  1427.     // if major were to trump minor, and we said "never" to the major
  1428.     // we'd never get the minor update, since selectUpdate()
  1429.     // would return the major update that the user said "never" to
  1430.     // (shadowing the important minor update with security fixes)
  1431.     return minorUpdate || majorUpdate;
  1432.   },
  1433.  
  1434.   /**
  1435.    * Determine which of the specified updates should be installed and
  1436.    * begin the download/installation process, optionally prompting the
  1437.    * user for permission if required.
  1438.    * @param   updates
  1439.    *          An array of available updates
  1440.    */
  1441.   _selectAndInstallUpdate: function(updates) {
  1442.     // Don't prompt if there's an active update - the user is already
  1443.     // aware and is downloading, or performed some user action to prevent
  1444.     // notification.
  1445.     var um = Components.classes["@mozilla.org/updates/update-manager;1"]
  1446.                        .getService(Components.interfaces.nsIUpdateManager);
  1447.     if (um.activeUpdate)
  1448.       return;
  1449.  
  1450.     var update = this.selectUpdate(updates, updates.length);
  1451.     if (!update)
  1452.       return;
  1453.  
  1454.     // check if the user said "never" to this version
  1455.     // this check is done here, and not in selectUpdate() so that
  1456.     // the user can get an upgrade they said "never" to if they
  1457.     // manually do "Check for Updates..."
  1458.     // note, selectUpdate() only returns one update.
  1459.     // but in selectUpdate(), minor updates trump major updates
  1460.     // if major trumps minor, and we said "never" to the major
  1461.     // we'd never see the minor update.
  1462.     //
  1463.     // note, the never decision should only apply to major updates
  1464.     // see bug #350636 for a scenario where this could potentially
  1465.     // be an issue
  1466.     //
  1467.     // fix for bug #359093
  1468.     // version might one day come back from AUS as an
  1469.     // arbitrary (and possibly non ascii) string, so we need to encode it
  1470.     var neverPrefName = PREF_UPDATE_NEVER_BRANCH + encodeURIComponent(update.version);
  1471.     var never = getPref("getBoolPref", neverPrefName, false);
  1472.     if (never && update.type == "major")
  1473.       return;
  1474.  
  1475.     if (this._shouldPrompt(update))
  1476.       showPromptIfNoIncompatibilities(update);
  1477.     else {
  1478.       LOG("UpdateService", "_selectAndInstallUpdate: No need to show prompt, just download update");
  1479.       var status = this.downloadUpdate(update, true);
  1480.       if (status == STATE_NONE)
  1481.         cleanupActiveUpdate();
  1482.     }
  1483.   },
  1484.  
  1485.   /**
  1486.    * The Checker used for background update checks.
  1487.    */
  1488.   _backgroundChecker: null,
  1489.  
  1490.   /**
  1491.    * See nsIUpdateService.idl
  1492.    */
  1493.   get backgroundChecker() {
  1494.     if (!this._backgroundChecker)
  1495.       this._backgroundChecker = new Checker();
  1496.     return this._backgroundChecker;
  1497.   },
  1498.  
  1499.   /**
  1500.    * See nsIUpdateService.idl
  1501.    */
  1502.   get canUpdate() {
  1503.     try {
  1504.       var appDirFile = getUpdateFile([FILE_PERMS_TEST]);
  1505.       LOG("UpdateService", "canUpdate?  testing " + appDirFile.path);
  1506.       if (!appDirFile.exists()) {
  1507.         appDirFile.create(nsILocalFile.NORMAL_FILE_TYPE, PERMS_FILE);
  1508.         appDirFile.remove(false);
  1509.       }
  1510.       var updateDir = getUpdatesDir();
  1511.       var upDirFile = updateDir.clone();
  1512.       upDirFile.append(FILE_PERMS_TEST);
  1513.       LOG("UpdateService", "canUpdate?  testing " + upDirFile.path);
  1514.       if (!upDirFile.exists()) {
  1515.         upDirFile.create(nsILocalFile.NORMAL_FILE_TYPE, PERMS_FILE);
  1516.         upDirFile.remove(false);
  1517.       }
  1518. //@line 1518 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  1519.       var sysInfo =
  1520.         Components.classes["@mozilla.org/system-info;1"]
  1521.                   .getService(Components.interfaces.nsIPropertyBag2);
  1522.  
  1523.       // Example windowsVersion:  Windows XP == 5.1
  1524.       var windowsVersion = sysInfo.getProperty("version");
  1525.       LOG("UpdateService", "canUpdate?  windowsVersion = " + windowsVersion);
  1526.  
  1527.       // For Vista, updates can be performed to a location requiring 
  1528.       // admin privileges by requesting elevation via the UAC prompt when 
  1529.       // launching updater.exe if the appDir is under the Program Files 
  1530.       // directory (e.g. C:\Program Files\) and UAC is turned on and 
  1531.       // we can elevate (e.g. user has a split token)
  1532.       //
  1533.       // Note: this does note attempt to handle the case where UAC is
  1534.       // turned on and the installation directory is in a restricted
  1535.       // location that requires admin privileges to update other than 
  1536.       // Program Files.
  1537.  
  1538.       var userCanElevate = false;
  1539.  
  1540.       if (parseFloat(windowsVersion) >= 6) {
  1541.         try {
  1542.           var fileLocator = 
  1543.             Components.classes["@mozilla.org/file/directory_service;1"]
  1544.                       .getService(Components.interfaces.nsIProperties);
  1545.           // KEY_UPDROOT will fail and throw an exception if
  1546.           // appDir is not under the Program Files, so we rely on that
  1547.           var dir = fileLocator.get(KEY_UPDROOT, Components.interfaces.nsIFile);
  1548.           // appDir is under Program Files, so check if the user can elevate
  1549.           userCanElevate = 
  1550.             gApp.QueryInterface(Components.interfaces.nsIWinAppHelper)
  1551.                 .userCanElevate;
  1552.           LOG("UpdateService", 
  1553.               "canUpdate?  on Vista, userCanElevate = " + userCanElevate);
  1554.         }
  1555.         catch (ex) {
  1556.           // When the installation directory is not under Program Files,
  1557.           // fall through to checking if write access to the 
  1558.           // installation directory is available.
  1559.           LOG("UpdateService", 
  1560.               "canUpdate?  on Vista, appDir is not under the Program Files");
  1561.         }
  1562.       }
  1563.  
  1564.       // On Windows, we no longer store the update under the app dir
  1565.       // if the app dir is under C:\Program Files.
  1566.       //
  1567.       // If we are on Windows (including Vista, if we can't elevate)
  1568.       // we need to check that
  1569.       // we can create and remove files from the actual app directory
  1570.       // (like C:\Program Files\Mozilla Firefox).  If we can't
  1571.       // (because this user is not an adminstrator, for example)
  1572.       // canUpdate() should return false.
  1573.       //
  1574.       // For Vista, we perform this check to enable updating the 
  1575.       // application when the user has write access to the installation 
  1576.       // directory under the following scenarios:
  1577.       // 1) the installation directory is not under Program Files 
  1578.       //    (e.g. C:\Program Files)
  1579.       // 2) UAC is turned off
  1580.       // 3) UAC is turned on and the user is not an admin 
  1581.       //    (e.g. the user does not have a split token)
  1582.       // 4) UAC is turned on and the user is already elevated,
  1583.       //    so they can't be elevated again.
  1584.       if (!userCanElevate) {
  1585.         // if we're unable to create the test file
  1586.         // the code below will throw an exception 
  1587.         var actualAppDir = getDir(KEY_APPDIR, []);
  1588.         var actualAppDirFile = actualAppDir.clone();
  1589.         actualAppDirFile.append(FILE_PERMS_TEST);
  1590.         LOG("UpdateService", "canUpdate?  testing " + actualAppDirFile.path);
  1591.         if (!actualAppDirFile.exists()) {
  1592.           actualAppDirFile.create(nsILocalFile.NORMAL_FILE_TYPE, PERMS_FILE);
  1593.           actualAppDirFile.remove(false);
  1594.         }
  1595.       }
  1596. //@line 1596 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  1597.     }
  1598.     catch (e) {
  1599.        LOG("UpdateService", "can't update, no privileges: " + e);
  1600.       // No write privileges to install directory
  1601.       return false;
  1602.     }
  1603.     // If the administrator has locked the app update functionality
  1604.     // OFF - this is not just a user setting, so disable the manual
  1605.     // UI too.
  1606.     var enabled = getPref("getBoolPref", PREF_APP_UPDATE_ENABLED, true);
  1607.     if (!enabled && gPref.prefIsLocked(PREF_APP_UPDATE_ENABLED)) {
  1608.       LOG("UpdateService", "can't update, disabled by pref");
  1609.       return false;
  1610.     }
  1611.  
  1612.     // If we don't know the binary platform we're updating, we can't update.
  1613.     if (!gABI) {
  1614.       LOG("UpdateService", "can't update, unknown ABI");
  1615.       return false;
  1616.     }
  1617.  
  1618.     // If we don't know the OS version we're updating, we can't update.
  1619.     if (!gOSVersion) {
  1620.       LOG("UpdateService", "can't update, unknown OS version");
  1621.       return false;
  1622.     }
  1623.  
  1624.     LOG("UpdateService", "can update");
  1625.     return true;
  1626.   },
  1627.  
  1628.   /**
  1629.    * See nsIUpdateService.idl
  1630.    */
  1631.   addDownloadListener: function(listener) {
  1632.     if (!this._downloader) {
  1633.       LOG("UpdateService", "addDownloadListener: no downloader!\n");
  1634.       return;
  1635.     }
  1636.     this._downloader.addDownloadListener(listener);
  1637.   },
  1638.  
  1639.   /**
  1640.    * See nsIUpdateService.idl
  1641.    */
  1642.   removeDownloadListener: function(listener) {
  1643.     if (!this._downloader) {
  1644.       LOG("UpdateService", "removeDownloadListener: no downloader!\n");
  1645.       return;
  1646.     }
  1647.     this._downloader.removeDownloadListener(listener);
  1648.   },
  1649.  
  1650.   /**
  1651.    * See nsIUpdateService.idl
  1652.    */
  1653.   downloadUpdate: function(update, background) {
  1654.     if (!update)
  1655.       throw Components.results.NS_ERROR_NULL_POINTER;
  1656.     if (this.isDownloading) {
  1657.       if (update.isCompleteUpdate == this._downloader.isCompleteUpdate &&
  1658.           background == this._downloader.background) {
  1659.         LOG("UpdateService", "no support for downloading more than one update at a time");
  1660.         return readStatusFile(getUpdatesDir());
  1661.       }
  1662.       this._downloader.cancel();
  1663.     }
  1664.     this._downloader = new Downloader(background);
  1665.     return this._downloader.downloadUpdate(update);
  1666.   },
  1667.  
  1668.   /**
  1669.    * See nsIUpdateService.idl
  1670.    */
  1671.   pauseDownload: function() {
  1672.     if (this.isDownloading)
  1673.       this._downloader.cancel();
  1674.   },
  1675.  
  1676.   /**
  1677.    * See nsIUpdateService.idl
  1678.    */
  1679.   get isDownloading() {
  1680.     return this._downloader && this._downloader.isBusy;
  1681.   },
  1682.  
  1683.   /**
  1684.    * See nsISupports.idl
  1685.    */
  1686.   QueryInterface: function(iid) {
  1687.     if (!iid.equals(Components.interfaces.nsIApplicationUpdateService) &&
  1688.         !iid.equals(Components.interfaces.nsITimerCallback) &&
  1689.         !iid.equals(Components.interfaces.nsIObserver) &&
  1690.         !iid.equals(Components.interfaces.nsISupports))
  1691.       throw Components.results.NS_ERROR_NO_INTERFACE;
  1692.     return this;
  1693.   }
  1694. };
  1695.  
  1696. /**
  1697.  * A service to manage active and past updates.
  1698.  * @constructor
  1699.  */
  1700. function UpdateManager() {
  1701.   // Ensure the Active Update file is loaded
  1702.   var updates = this._loadXMLFileIntoArray(getUpdateFile([FILE_UPDATE_ACTIVE]));
  1703.   if (updates.length > 0)
  1704.     this._activeUpdate = updates[0];
  1705. }
  1706. UpdateManager.prototype = {
  1707.   /**
  1708.    * All previously downloaded and installed updates, as an array of nsIUpdate
  1709.    * objects.
  1710.    */
  1711.   _updates: null,
  1712.  
  1713.   /**
  1714.    * The current actively downloading/installing update, as a nsIUpdate object.
  1715.    */
  1716.   _activeUpdate: null,
  1717.  
  1718.   /**
  1719.    * Loads an updates.xml formatted file into an array of nsIUpdate items.
  1720.    * @param   file
  1721.    *          A nsIFile for the updates.xml file
  1722.    * @returns The array of nsIUpdate items held in the file.
  1723.    */
  1724.   _loadXMLFileIntoArray: function(file) {
  1725.     if (!file.exists()) {
  1726.       LOG("UpdateManager", "_loadXMLFileIntoArray: XML File does not exist");
  1727.       return [];
  1728.     }
  1729.  
  1730.     var result = [];
  1731.     var fileStream = Components.classes["@mozilla.org/network/file-input-stream;1"]
  1732.                                .createInstance(Components.interfaces.nsIFileInputStream);
  1733.     fileStream.init(file, MODE_RDONLY, PERMS_FILE, 0);
  1734.     try {
  1735.       var parser = Components.classes["@mozilla.org/xmlextras/domparser;1"]
  1736.                             .createInstance(Components.interfaces.nsIDOMParser);
  1737.       var doc = parser.parseFromStream(fileStream, "UTF-8", fileStream.available(), "text/xml");
  1738.  
  1739.       var updateCount = doc.documentElement.childNodes.length;
  1740.       for (var i = 0; i < updateCount; ++i) {
  1741.         var updateElement = doc.documentElement.childNodes.item(i);
  1742.         if (updateElement.nodeType != Node.ELEMENT_NODE ||
  1743.             updateElement.localName != "update")
  1744.           continue;
  1745.  
  1746.         updateElement.QueryInterface(Components.interfaces.nsIDOMElement);
  1747.         try {
  1748.           var update = new Update(updateElement);
  1749.         } catch (e) {
  1750.           LOG("UpdateManager", "_loadXMLFileIntoArray: invalid update");
  1751.           continue;
  1752.         }
  1753.         result.push(new Update(updateElement));
  1754.       }
  1755.     }
  1756.     catch (e) {
  1757.       LOG("UpdateManager", "_loadXMLFileIntoArray: Error constructing update list " +
  1758.           e);
  1759.     }
  1760.     fileStream.close();
  1761.     return result;
  1762.   },
  1763.  
  1764.   /**
  1765.    * Load the update manager, initializing state from state files.
  1766.    */
  1767.   _ensureUpdates: function() {
  1768.     if (!this._updates) {
  1769.       this._updates = this._loadXMLFileIntoArray(getUpdateFile(
  1770.                         [FILE_UPDATES_DB]));
  1771.  
  1772.       // Make sure that any active update is part of our updates list
  1773.       var active = this.activeUpdate;
  1774.       if (active)
  1775.         this._addUpdate(active);
  1776.     }
  1777.   },
  1778.  
  1779.   /**
  1780.    * See nsIUpdateService.idl
  1781.    */
  1782.   getUpdateAt: function(index) {
  1783.     this._ensureUpdates();
  1784.     return this._updates[index];
  1785.   },
  1786.  
  1787.   /**
  1788.    * See nsIUpdateService.idl
  1789.    */
  1790.   get updateCount() {
  1791.     this._ensureUpdates();
  1792.     return this._updates.length;
  1793.   },
  1794.  
  1795.   /**
  1796.    * See nsIUpdateService.idl
  1797.    */
  1798.   get activeUpdate() {
  1799.     if (this._activeUpdate &&
  1800.         this._activeUpdate.channel != getUpdateChannel()) {
  1801.       // User switched channels, clear out any old active updates and remove
  1802.       // partial downloads
  1803.       this._activeUpdate = null;
  1804.  
  1805.       // Destroy the updates directory, since we're done with it.
  1806.       cleanUpUpdatesDir();
  1807.     }
  1808.     return this._activeUpdate;
  1809.   },
  1810.   set activeUpdate(activeUpdate) {
  1811.     this._addUpdate(activeUpdate);
  1812.     this._activeUpdate = activeUpdate;
  1813.     if (!activeUpdate) {
  1814.       // If |activeUpdate| is null, we have updated both lists - the active list
  1815.       // and the history list, so we want to write both files.
  1816.       this.saveUpdates();
  1817.     }
  1818.     else
  1819.       this._writeUpdatesToXMLFile([this._activeUpdate],
  1820.                                   getUpdateFile([FILE_UPDATE_ACTIVE]));
  1821.     return activeUpdate;
  1822.   },
  1823.  
  1824.   /**
  1825.    * Add an update to the Updates list. If the item already exists in the list,
  1826.    * replace the existing value with the new value.
  1827.    * @param   update
  1828.    *          The nsIUpdate object to add.
  1829.    */
  1830.   _addUpdate: function(update) {
  1831.     if (!update)
  1832.       return;
  1833.     this._ensureUpdates();
  1834.     if (this._updates) {
  1835.       for (var i = 0; i < this._updates.length; ++i) {
  1836.         if (this._updates[i] &&
  1837.             this._updates[i].version == update.version &&
  1838.             this._updates[i].buildID == update.buildID) {
  1839.           // Replace the existing entry with the new value, updating
  1840.           // all metadata.
  1841.           this._updates[i] = update;
  1842.           return;
  1843.         }
  1844.       }
  1845.     }
  1846.     // Otherwise add it to the front of the list.
  1847.     if (update)
  1848.       this._updates = [update].concat(this._updates);
  1849.   },
  1850.  
  1851.   /**
  1852.    * Serializes an array of updates to an XML file
  1853.    * @param   updates
  1854.    *          An array of nsIUpdate objects
  1855.    * @param   file
  1856.    *          The nsIFile object to serialize to
  1857.    */
  1858.   _writeUpdatesToXMLFile: function(updates, file) {
  1859.     var fos = Components.classes["@mozilla.org/network/safe-file-output-stream;1"]
  1860.                         .createInstance(Components.interfaces.nsIFileOutputStream);
  1861.     var modeFlags = MODE_WRONLY | MODE_CREATE | MODE_TRUNCATE;
  1862.     if (!file.exists())
  1863.       file.create(nsILocalFile.NORMAL_FILE_TYPE, PERMS_FILE);
  1864.     fos.init(file, modeFlags, PERMS_FILE, 0);
  1865.  
  1866.     try {
  1867.       var parser = Components.classes["@mozilla.org/xmlextras/domparser;1"]
  1868.                             .createInstance(Components.interfaces.nsIDOMParser);
  1869.       const EMPTY_UPDATES_DOCUMENT = "<?xml version=\"1.0\"?><updates xmlns=\"http://www.mozilla.org/2005/app-update\"></updates>";
  1870.       var doc = parser.parseFromString(EMPTY_UPDATES_DOCUMENT, "text/xml");
  1871.  
  1872.       for (var i = 0; i < updates.length; ++i) {
  1873.         if (updates[i])
  1874.           doc.documentElement.appendChild(updates[i].serialize(doc));
  1875.       }
  1876.  
  1877.       var serializer = Components.classes["@mozilla.org/xmlextras/xmlserializer;1"]
  1878.                                 .createInstance(Components.interfaces.nsIDOMSerializer);
  1879.       serializer.serializeToStream(doc.documentElement, fos, null);
  1880.     }
  1881.     catch (e) {
  1882.     }
  1883.  
  1884.     closeSafeOutputStream(fos);
  1885.   },
  1886.  
  1887.   /**
  1888.    * See nsIUpdateService.idl
  1889.    */
  1890.   saveUpdates: function() {
  1891.     this._writeUpdatesToXMLFile([this._activeUpdate],
  1892.                                 getUpdateFile([FILE_UPDATE_ACTIVE]));
  1893.     if (this._updates) {
  1894.       this._writeUpdatesToXMLFile(this._updates.slice(0, 10),
  1895.                                   getUpdateFile([FILE_UPDATES_DB]));
  1896.     }
  1897.   },
  1898.  
  1899.   /**
  1900.    * See nsISupports.idl
  1901.    */
  1902.   QueryInterface: function(iid) {
  1903.     if (!iid.equals(Components.interfaces.nsIUpdateManager) &&
  1904.         !iid.equals(Components.interfaces.nsISupports))
  1905.       throw Components.results.NS_ERROR_NO_INTERFACE;
  1906.     return this;
  1907.   }
  1908. };
  1909.  
  1910.  
  1911. /**
  1912.  * Checker
  1913.  * Checks for new Updates
  1914.  * @constructor
  1915.  */
  1916. function Checker() {
  1917. }
  1918. Checker.prototype = {
  1919.   /**
  1920.    * The XMLHttpRequest object that performs the connection.
  1921.    */
  1922.   _request  : null,
  1923.  
  1924.   /**
  1925.    * The nsIUpdateCheckListener callback
  1926.    */
  1927.   _callback : null,
  1928.  
  1929.   /**
  1930.    * The URL of the update service XML file to connect to that contains details
  1931.    * about available updates.
  1932.    */
  1933.   getUpdateURL: function(force) {
  1934.     this._forced = force;
  1935.  
  1936.     var defaults =
  1937.         gPref.QueryInterface(Components.interfaces.nsIPrefService).
  1938.         getDefaultBranch(null);
  1939.  
  1940.     // Use the override URL if specified.
  1941.     var url = getPref("getCharPref", PREF_APP_UPDATE_URL_OVERRIDE, null);
  1942.  
  1943.     // Otherwise, construct the update URL from component parts.
  1944.     if (!url) {
  1945.       try {
  1946.         url = defaults.getCharPref(PREF_APP_UPDATE_URL);
  1947.       } catch (e) {
  1948.       }
  1949.     }
  1950.  
  1951.     if (!url || url == "") {
  1952.       LOG("Checker", "Update URL not defined");
  1953.       return null;
  1954.     }
  1955.  
  1956.     url = url.replace(/%PRODUCT%/g, gApp.name);
  1957.     url = url.replace(/%VERSION%/g, gApp.version);
  1958.     url = url.replace(/%BUILD_ID%/g, gApp.appBuildID);
  1959.     url = url.replace(/%BUILD_TARGET%/g, gApp.OS + "_" + gABI);
  1960.     url = url.replace(/%OS_VERSION%/g, gOSVersion);
  1961.     url = url.replace(/%LOCALE%/g, getLocale());
  1962.     url = url.replace(/%CHANNEL%/g, getUpdateChannel());
  1963.     url = url.replace(/%PLATFORM_VERSION%/g, gApp.platformVersion);
  1964.     url = url.replace(/%DISTRIBUTION%/g,
  1965.                       getDistributionPrefValue(PREF_APP_DISTRIBUTION));
  1966.     url = url.replace(/%DISTRIBUTION_VERSION%/g,
  1967.                       getDistributionPrefValue(PREF_APP_DISTRIBUTION_VERSION));
  1968.     url = url.replace(/\+/g, "%2B");
  1969.  
  1970.     if (force)
  1971.     url += "?force=1"
  1972.  
  1973.     LOG("Checker", "update url: " + url);
  1974.     return url;
  1975.   },
  1976.  
  1977.   /**
  1978.    * See nsIUpdateService.idl
  1979.    */
  1980.   checkForUpdates: function(listener, force) {
  1981.     if (!listener)
  1982.       throw Components.results.NS_ERROR_NULL_POINTER;
  1983.  
  1984.     if (!this.getUpdateURL(force) || (!this.enabled && !force))
  1985.       return;
  1986.  
  1987.     this._request =
  1988.       Components.classes["@mozilla.org/xmlextras/xmlhttprequest;1"].
  1989.       createInstance(Components.interfaces.nsIXMLHttpRequest);
  1990.     this._request.open("GET", this.getUpdateURL(force), true);
  1991.     this._request.channel.notificationCallbacks = new BadCertHandler();
  1992.     this._request.overrideMimeType("text/xml");
  1993.     this._request.setRequestHeader("Cache-Control", "no-cache");
  1994.  
  1995.     var self = this;
  1996.     this._request.onerror     = function(event) { self.onError(event);    };
  1997.     this._request.onload      = function(event) { self.onLoad(event);     };
  1998.     this._request.onprogress  = function(event) { self.onProgress(event); };
  1999.  
  2000.     LOG("Checker", "checkForUpdates: sending request to " + this.getUpdateURL(force));
  2001.     this._request.send(null);
  2002.  
  2003.     this._callback = listener;
  2004.   },
  2005.  
  2006.   /**
  2007.    * When progress associated with the XMLHttpRequest is received.
  2008.    * @param   event
  2009.    *          The nsIDOMLSProgressEvent for the load.
  2010.    */
  2011.   onProgress: function(event) {
  2012.     LOG("Checker", "onProgress: " + event.position + "/" + event.totalSize);
  2013.     this._callback.onProgress(event.target, event.position, event.totalSize);
  2014.   },
  2015.  
  2016.   /**
  2017.    * Returns an array of nsIUpdate objects discovered by the update check.
  2018.    */
  2019.   get _updates() {
  2020.     var updatesElement = this._request.responseXML.documentElement;
  2021.     if (!updatesElement) {
  2022.       LOG("Checker", "get_updates: empty updates document?!");
  2023.       return [];
  2024.     }
  2025.  
  2026.     if (updatesElement.nodeName != "updates") {
  2027.       LOG("Checker", "get_updates: unexpected node name!");
  2028.       throw "";
  2029.     }
  2030.  
  2031.     var updates = [];
  2032.     for (var i = 0; i < updatesElement.childNodes.length; ++i) {
  2033.       var updateElement = updatesElement.childNodes.item(i);
  2034.       if (updateElement.nodeType != Node.ELEMENT_NODE ||
  2035.           updateElement.localName != "update")
  2036.         continue;
  2037.  
  2038.       updateElement.QueryInterface(Components.interfaces.nsIDOMElement);
  2039.       try {
  2040.         var update = new Update(updateElement);
  2041.       } catch (e) {
  2042.         LOG("Checker", "Invalid <update/>, ignoring...");
  2043.         continue;
  2044.       }
  2045.       update.serviceURL = this.getUpdateURL(this._forced);
  2046.       update.channel = getUpdateChannel();
  2047.       updates.push(update);
  2048.     }
  2049.  
  2050.     return updates;
  2051.   },
  2052.  
  2053.   /**
  2054.    * The XMLHttpRequest succeeded and the document was loaded.
  2055.    * @param   event
  2056.    *          The nsIDOMLSEvent for the load
  2057.    */
  2058.   onLoad: function(event) {
  2059.     LOG("Checker", "onLoad: request completed downloading document");
  2060.  
  2061.     try {
  2062.       checkCert(this._request.channel);
  2063.  
  2064.       // Analyze the resulting DOM and determine the set of updates to install
  2065.       var updates = this._updates;
  2066.  
  2067.       LOG("Checker", "Updates available: " + updates.length);
  2068.  
  2069.       // ... and tell the Update Service about what we discovered.
  2070.       this._callback.onCheckComplete(event.target, updates, updates.length);
  2071.     }
  2072.     catch (e) {
  2073.       LOG("Checker", "There was a problem with the update service URL specified, " +
  2074.           "either the XML file was malformed or it does not exist at the location " +
  2075.           "specified. Exception: " + e);
  2076.       var update = new Update(null);
  2077.       update.statusText = getStatusTextFromCode(404, 404);
  2078.       this._callback.onError(event.target, update);
  2079.     }
  2080.  
  2081.     this._request = null;
  2082.   },
  2083.  
  2084.   /**
  2085.    * There was an error of some kind during the XMLHttpRequest
  2086.    * @param   event
  2087.    *          The nsIDOMLSEvent for the load
  2088.    */
  2089.   onError: function(event) {
  2090.     LOG("Checker", "onError: error during load");
  2091.  
  2092.     var request = event.target;
  2093.     try {
  2094.       var status = request.status;
  2095.     }
  2096.     catch (e) {
  2097.       var req = request.channel.QueryInterface(Components.interfaces.nsIRequest);
  2098.       status = req.status;
  2099.     }
  2100.  
  2101.     // If we can't find an error string specific to this status code,
  2102.     // just use the 200 message from above, which means everything
  2103.     // "looks" fine but there was probably an XML error or a bogus file.
  2104.     var update = new Update(null);
  2105.     update.statusText = getStatusTextFromCode(status, 200);
  2106.     this._callback.onError(request, update);
  2107.  
  2108.     this._request = null;
  2109.   },
  2110.  
  2111.   /**
  2112.    * Whether or not we are allowed to do update checking.
  2113.    */
  2114.   _enabled: true,
  2115.  
  2116.   /**
  2117.    * See nsIUpdateService.idl
  2118.    */
  2119.   get enabled() {
  2120.     var aus =
  2121.         Components.classes["@mozilla.org/updates/update-service;1"].
  2122.         getService(Components.interfaces.nsIApplicationUpdateService);
  2123.     var enabled = getPref("getBoolPref", PREF_APP_UPDATE_ENABLED, true) &&
  2124.                   aus.canUpdate && this._enabled;
  2125.     return enabled;
  2126.   },
  2127.  
  2128.   /**
  2129.    * See nsIUpdateService.idl
  2130.    */
  2131.   stopChecking: function(duration) {
  2132.     // Always stop the current check
  2133.     if (this._request)
  2134.       this._request.abort();
  2135.  
  2136.     const nsIUpdateChecker = Components.interfaces.nsIUpdateChecker;
  2137.     switch (duration) {
  2138.     case nsIUpdateChecker.CURRENT_SESSION:
  2139.       this._enabled = false;
  2140.       break;
  2141.     case nsIUpdateChecker.ANY_CHECKS:
  2142.       this._enabled = false;
  2143.       gPref.setBoolPref(PREF_APP_UPDATE_ENABLED, this._enabled);
  2144.       break;
  2145.     }
  2146.   },
  2147.  
  2148.   /**
  2149.    * See nsISupports.idl
  2150.    */
  2151.   QueryInterface: function(iid) {
  2152.     if (!iid.equals(Components.interfaces.nsIUpdateChecker) &&
  2153.         !iid.equals(Components.interfaces.nsISupports))
  2154.       throw Components.results.NS_ERROR_NO_INTERFACE;
  2155.     return this;
  2156.   }
  2157. };
  2158.  
  2159. /**
  2160.  * Manages the download of updates
  2161.  * @param   background
  2162.  *          Whether or not this downloader is operating in background
  2163.  *          update mode.
  2164.  * @constructor
  2165.  */
  2166. function Downloader(background) {
  2167.   this.background = background;
  2168. }
  2169. Downloader.prototype = {
  2170.   /**
  2171.    * The nsIUpdatePatch that we are downloading
  2172.    */
  2173.   _patch: null,
  2174.  
  2175.   /**
  2176.    * The nsIUpdate that we are downloading
  2177.    */
  2178.   _update: null,
  2179.  
  2180.   /**
  2181.    * The nsIIncrementalDownload object handling the download
  2182.    */
  2183.   _request: null,
  2184.  
  2185.   /**
  2186.    * Whether or not the update being downloaded is a complete replacement of
  2187.    * the user's existing installation or a patch representing the difference
  2188.    * between the new version and the previous version.
  2189.    */
  2190.   isCompleteUpdate: null,
  2191.  
  2192.   /**
  2193.    * Cancels the active download.
  2194.    */
  2195.   cancel: function() {
  2196.     if (this._request &&
  2197.         this._request instanceof Components.interfaces.nsIRequest) {
  2198.       const NS_BINDING_ABORTED = 0x804b0002;
  2199.       this._request.cancel(NS_BINDING_ABORTED);
  2200.     }
  2201.   },
  2202.  
  2203.   /**
  2204.    * Whether or not a patch has been downloaded and staged for installation.
  2205.    */
  2206.   get patchIsStaged() {
  2207.     return readStatusFile(getUpdatesDir()) == STATE_PENDING;
  2208.   },
  2209.  
  2210.   /**
  2211.    * Verify the downloaded file.  We assume that the download is complete at
  2212.    * this point.
  2213.    */
  2214.   _verifyDownload: function() {
  2215.     if (!this._request)
  2216.       return false;
  2217.  
  2218.     var destination = this._request.destination;
  2219.  
  2220.     // Ensure that the file size matches the expected file size.
  2221.     if (destination.fileSize != this._patch.size)
  2222.       return false;
  2223.  
  2224.     var fileStream = Components.classes["@mozilla.org/network/file-input-stream;1"].
  2225.         createInstance(nsIFileInputStream);
  2226.     fileStream.init(destination, MODE_RDONLY, PERMS_FILE, 0);
  2227.  
  2228.     try {
  2229.       var hash = Components.classes["@mozilla.org/security/hash;1"].
  2230.           createInstance(nsICryptoHash);
  2231.       var hashFunction = nsICryptoHash[this._patch.hashFunction.toUpperCase()];
  2232.       if (hashFunction == undefined)
  2233.         throw Components.results.NS_ERROR_UNEXPECTED;
  2234.       hash.init(hashFunction);
  2235.       hash.updateFromStream(fileStream, -1);
  2236.       // NOTE: For now, we assume that the format of _patch.hashValue is hex
  2237.       // encoded binary (such as what is typically output by programs like
  2238.       // sha1sum).  In the future, this may change to base64 depending on how
  2239.       // we choose to compute these hashes.
  2240.       digest = binaryToHex(hash.finish(false));
  2241.     } catch (e) {
  2242.       LOG("Downloader", "failed to compute hash of downloaded update archive");
  2243.       digest = "";
  2244.     }
  2245.  
  2246.     fileStream.close();
  2247.  
  2248.     return digest == this._patch.hashValue.toLowerCase();
  2249.   },
  2250.  
  2251.   /**
  2252.    * Select the patch to use given the current state of updateDir and the given
  2253.    * set of update patches.
  2254.    * @param   update
  2255.    *          A nsIUpdate object to select a patch from
  2256.    * @param   updateDir
  2257.    *          A nsIFile representing the update directory
  2258.    * @returns A nsIUpdatePatch object to download
  2259.    */
  2260.   _selectPatch: function(update, updateDir) {
  2261.     // Given an update to download, we will always try to download the patch
  2262.     // for a partial update over the patch for a full update.
  2263.  
  2264.     /**
  2265.      * Return the first UpdatePatch with the given type.
  2266.      * @param   type
  2267.      *          The type of the patch ("complete" or "partial")
  2268.      * @returns A nsIUpdatePatch object matching the type specified
  2269.      */
  2270.     function getPatchOfType(type) {
  2271.       for (var i = 0; i < update.patchCount; ++i) {
  2272.         var patch = update.getPatchAt(i);
  2273.         if (patch && patch.type == type)
  2274.           return patch;
  2275.       }
  2276.       return null;
  2277.     }
  2278.  
  2279.     // Look to see if any of the patches in the Update object has been
  2280.     // pre-selected for download, otherwise we must figure out which one
  2281.     // to select ourselves.
  2282.     var selectedPatch = update.selectedPatch;
  2283.  
  2284.     var state = readStatusFile(updateDir);
  2285.  
  2286.     // If this is a patch that we know about, then select it.  If it is a patch
  2287.     // that we do not know about, then remove it and use our default logic.
  2288.     var useComplete = false;
  2289.     if (selectedPatch) {
  2290.       LOG("Downloader", "found existing patch [state="+state+"]");
  2291.       switch (state) {
  2292.       case STATE_DOWNLOADING:
  2293.         LOG("Downloader", "resuming download");
  2294.         return selectedPatch;
  2295.       case STATE_PENDING:
  2296.         LOG("Downloader", "already downloaded and staged");
  2297.         return null;
  2298.       default:
  2299.         // Something went wrong when we tried to apply the previous patch.
  2300.         // Try the complete patch next time.
  2301.         if (update && selectedPatch.type == "partial") {
  2302.           useComplete = true;
  2303.         } else {
  2304.           // This is a pretty fatal error.  Just bail.
  2305.           LOG("Downloader", "failed to apply complete patch!");
  2306.           writeStatusFile(updateDir, STATE_NONE);
  2307.           return null;
  2308.         }
  2309.       }
  2310.  
  2311.       selectedPatch = null;
  2312.     }
  2313.  
  2314.     // If we were not able to discover an update from a previous download, we
  2315.     // select the best patch from the given set.
  2316.     var partialPatch = getPatchOfType("partial");
  2317.     if (!useComplete)
  2318.       selectedPatch = partialPatch;
  2319.     if (!selectedPatch) {
  2320.       if (partialPatch)
  2321.         partialPatch.selected = false;
  2322.       selectedPatch = getPatchOfType("complete");
  2323.     }
  2324.  
  2325.     // Restore the updateDir since we may have deleted it.
  2326.     updateDir = getUpdatesDir();
  2327.  
  2328.     // if update only contains a partial patch, selectedPatch == null here if
  2329.     // the partial patch has been attempted and fails and we're trying to get a
  2330.     // complete patch
  2331.     if (selectedPatch)
  2332.       selectedPatch.selected = true;
  2333.  
  2334.     update.isCompleteUpdate = useComplete;
  2335.  
  2336.     // Reset the Active Update object on the Update Manager and flush the
  2337.     // Active Update DB.
  2338.     var um = Components.classes["@mozilla.org/updates/update-manager;1"]
  2339.                        .getService(Components.interfaces.nsIUpdateManager);
  2340.     um.activeUpdate = update;
  2341.  
  2342.     return selectedPatch;
  2343.   },
  2344.  
  2345.   /**
  2346.    * Whether or not we are currently downloading something.
  2347.    */
  2348.   get isBusy() {
  2349.     return this._request != null;
  2350.   },
  2351.  
  2352.   /**
  2353.    * Download and stage the given update.
  2354.    * @param   update
  2355.    *          A nsIUpdate object to download a patch for. Cannot be null.
  2356.    */
  2357.   downloadUpdate: function(update) {
  2358.     if (!update)
  2359.       throw Components.results.NS_ERROR_NULL_POINTER;
  2360.  
  2361.     var updateDir = getUpdatesDir();
  2362.  
  2363.     this._update = update;
  2364.  
  2365.     // This function may return null, which indicates that there are no patches
  2366.     // to download.
  2367.     this._patch = this._selectPatch(update, updateDir);
  2368.     if (!this._patch) {
  2369.       LOG("Downloader", "no patch to download");
  2370.       return readStatusFile(updateDir);
  2371.     }
  2372.     this.isCompleteUpdate = this._patch.type == "complete";
  2373.  
  2374.     var patchFile = updateDir.clone();
  2375.     patchFile.append(FILE_UPDATE_ARCHIVE);
  2376.  
  2377.     var ios = Components.classes["@mozilla.org/network/io-service;1"].
  2378.         getService(Components.interfaces.nsIIOService);
  2379.     var uri = ios.newURI(this._patch.URL, null, null);
  2380.  
  2381.     this._request =
  2382.         Components.classes["@mozilla.org/network/incremental-download;1"].
  2383.         createInstance(nsIIncrementalDownload);
  2384.  
  2385.     LOG("Downloader", "downloadUpdate: Downloading from " + uri.spec + " to " +
  2386.         patchFile.path);
  2387.  
  2388.     var interval = this.background ? DOWNLOAD_BACKGROUND_INTERVAL
  2389.                                    : DOWNLOAD_FOREGROUND_INTERVAL;
  2390.     this._request.init(uri, patchFile, DOWNLOAD_CHUNK_SIZE, interval);
  2391.     this._request.start(this, null);
  2392.  
  2393.     writeStatusFile(updateDir, STATE_DOWNLOADING);
  2394.     this._patch.QueryInterface(Components.interfaces.nsIWritablePropertyBag);
  2395.     this._patch.state = STATE_DOWNLOADING;
  2396.     var um = Components.classes["@mozilla.org/updates/update-manager;1"]
  2397.                        .getService(Components.interfaces.nsIUpdateManager);
  2398.     um.saveUpdates();
  2399.     return STATE_DOWNLOADING;
  2400.   },
  2401.  
  2402.   /**
  2403.    * An array of download listeners to notify when we receive
  2404.    * nsIRequestObserver or nsIProgressEventSink method calls.
  2405.    */
  2406.   _listeners: [],
  2407.  
  2408.   /**
  2409.    * Adds a listener to the download process
  2410.    * @param   listener
  2411.    *          A download listener, implementing nsIRequestObserver and
  2412.    *          nsIProgressEventSink
  2413.    */
  2414.   addDownloadListener: function(listener) {
  2415.     for (var i = 0; i < this._listeners.length; ++i) {
  2416.       if (this._listeners[i] == listener)
  2417.         return;
  2418.     }
  2419.     this._listeners.push(listener);
  2420.   },
  2421.  
  2422.   /**
  2423.    * Removes a download listener
  2424.    * @param   listener
  2425.    *          The listener to remove.
  2426.    */
  2427.   removeDownloadListener: function(listener) {
  2428.     for (var i = 0; i < this._listeners.length; ++i) {
  2429.       if (this._listeners[i] == listener) {
  2430.         this._listeners.splice(i, 1);
  2431.         return;
  2432.       }
  2433.     }
  2434.   },
  2435.  
  2436.   /**
  2437.    * When the async request begins
  2438.    * @param   request
  2439.    *          The nsIRequest object for the transfer
  2440.    * @param   context
  2441.    *          Additional data
  2442.    */
  2443.   onStartRequest: function(request, context) {
  2444.     request.QueryInterface(nsIIncrementalDownload);
  2445.     LOG("Downloader", "onStartRequest: " + request.URI.spec);
  2446.  
  2447.     var listenerCount = this._listeners.length;
  2448.     for (var i = 0; i < listenerCount; ++i)
  2449.       this._listeners[i].onStartRequest(request, context);
  2450.   },
  2451.  
  2452.   /**
  2453.    * When new data has been downloaded
  2454.    * @param   request
  2455.    *          The nsIRequest object for the transfer
  2456.    * @param   context
  2457.    *          Additional data
  2458.    * @param   progress
  2459.    *          The current number of bytes transferred
  2460.    * @param   maxProgress
  2461.    *          The total number of bytes that must be transferred
  2462.    */
  2463.   onProgress: function(request, context, progress, maxProgress) {
  2464.     request.QueryInterface(nsIIncrementalDownload);
  2465.     LOG("Downloader.onProgress", "onProgress: " + request.URI.spec + ", " + progress + "/" + maxProgress);
  2466.  
  2467.     var listenerCount = this._listeners.length;
  2468.     for (var i = 0; i < listenerCount; ++i) {
  2469.       var listener = this._listeners[i];
  2470.       if (listener instanceof Components.interfaces.nsIProgressEventSink)
  2471.         listener.onProgress(request, context, progress, maxProgress);
  2472.     }
  2473.   },
  2474.  
  2475.   /**
  2476.    * When we have new status text
  2477.    * @param   request
  2478.    *          The nsIRequest object for the transfer
  2479.    * @param   context
  2480.    *          Additional data
  2481.    * @param   status
  2482.    *          A status code
  2483.    * @param   statusText
  2484.    *          Human readable version of |status|
  2485.    */
  2486.   onStatus: function(request, context, status, statusText) {
  2487.     request.QueryInterface(nsIIncrementalDownload);
  2488.     LOG("Downloader", "onStatus: " + request.URI.spec + " status = " + status + ", text = " + statusText);
  2489.     var listenerCount = this._listeners.length;
  2490.     for (var i = 0; i < listenerCount; ++i) {
  2491.       var listener = this._listeners[i];
  2492.       if (listener instanceof Components.interfaces.nsIProgressEventSink)
  2493.         listener.onStatus(request, context, status, statusText);
  2494.     }
  2495.   },
  2496.  
  2497.   /**
  2498.    * When data transfer ceases
  2499.    * @param   request
  2500.    *          The nsIRequest object for the transfer
  2501.    * @param   context
  2502.    *          Additional data
  2503.    * @param   status
  2504.    *          Status code containing the reason for the cessation.
  2505.    */
  2506.   onStopRequest: function(request, context, status) {
  2507.     request.QueryInterface(nsIIncrementalDownload);
  2508.     LOG("Downloader", "onStopRequest: " + request.URI.spec + ", status = " + status);
  2509.  
  2510.     var state = this._patch.state;
  2511.     var shouldShowPrompt = false;
  2512.     var deleteActiveUpdate = false;
  2513.     const NS_BINDING_ABORTED = 0x804b0002;
  2514.     const NS_ERROR_ABORT = 0x80004004;
  2515.     if (Components.isSuccessCode(status)) {
  2516.       var sbs =
  2517.           Components.classes["@mozilla.org/intl/stringbundle;1"].
  2518.           getService(Components.interfaces.nsIStringBundleService);
  2519.       var updateStrings = sbs.createBundle(URI_UPDATES_PROPERTIES);
  2520.       if (this._verifyDownload()) {
  2521.         state = STATE_PENDING;
  2522.  
  2523.         // We only need to explicitly show the prompt if this is a backround
  2524.         // download, since otherwise some kind of UI is already visible and
  2525.         // that UI will notify.
  2526.         if (this.background)
  2527.           shouldShowPrompt = true;
  2528.  
  2529.         // Tell the updater.exe we're ready to apply.
  2530.         writeStatusFile(getUpdatesDir(), state);
  2531.         this._update.installDate = (new Date()).getTime();
  2532.         this._update.statusText = updateStrings.
  2533.           GetStringFromName("installPending");
  2534.       } else {
  2535.         LOG("Downloader", "onStopRequest: download verification failed");
  2536.         state = STATE_DOWNLOAD_FAILED;
  2537.  
  2538.         var brandStrings = sbs.createBundle(URI_BRAND_PROPERTIES);
  2539.         var brandShortName = brandStrings.GetStringFromName("brandShortName");
  2540.         this._update.statusText = updateStrings.
  2541.           formatStringFromName("verificationError", [brandShortName], 1);
  2542.  
  2543.         // TODO: use more informative error code here
  2544.         status = Components.results.NS_ERROR_UNEXPECTED;
  2545.  
  2546.         var message = getStatusTextFromCode("verification_failed",
  2547.           "verification_failed");
  2548.         this._update.statusText = message;
  2549.  
  2550.         if (this._update.isCompleteUpdate)
  2551.           deleteActiveUpdate = true;
  2552.  
  2553.         // Destroy the updates directory, since we're done with it.
  2554.         cleanUpUpdatesDir();
  2555.       }
  2556.     }
  2557.     else if (status != NS_BINDING_ABORTED &&
  2558.              status != NS_ERROR_ABORT) {
  2559.       LOG("Downloader", "onStopRequest: Non-verification failure");
  2560.       // Some sort of other failure, log this in the |statusText| property
  2561.       state = STATE_DOWNLOAD_FAILED;
  2562.  
  2563.       // XXXben - if |request| (The Incremental Download) provided a means
  2564.       // for accessing the http channel we could do more here.
  2565.  
  2566.       const NS_BINDING_FAILED = 2152398849;
  2567.       this._update.statusText = getStatusTextFromCode(status,
  2568.         NS_BINDING_FAILED);
  2569.  
  2570.       // Destroy the updates directory, since we're done with it.
  2571.       cleanUpUpdatesDir();
  2572.  
  2573.       deleteActiveUpdate = true;
  2574.     }
  2575.     LOG("Downloader", "Setting state to: " + state);
  2576.     this._patch.state = state;
  2577.     var um =
  2578.         Components.classes["@mozilla.org/updates/update-manager;1"].
  2579.         getService(Components.interfaces.nsIUpdateManager);
  2580.     if (deleteActiveUpdate) {
  2581.       this._update.installDate = (new Date()).getTime();
  2582.       um.activeUpdate = null;
  2583.     }
  2584.     um.saveUpdates();
  2585.  
  2586.     var listenerCount = this._listeners.length;
  2587.     for (var i = 0; i < listenerCount; ++i)
  2588.       this._listeners[i].onStopRequest(request, context, status);
  2589.  
  2590.     this._request = null;
  2591.  
  2592.     if (state == STATE_DOWNLOAD_FAILED) {
  2593.       if (!this._update.isCompleteUpdate) {
  2594.         var allFailed = true;
  2595.  
  2596.         // If we were downloading a patch and the patch verification phase
  2597.         // failed, log this and then commence downloading the complete update.
  2598.         LOG("Downloader", "onStopRequest: Verification of patch failed, downloading complete update");
  2599.         this._update.isCompleteUpdate = true;
  2600.         var status = this.downloadUpdate(this._update);
  2601.  
  2602.         if (status == STATE_NONE) {
  2603.           cleanupActiveUpdate();
  2604.         } else {
  2605.           allFailed = false;
  2606.         }
  2607.         // This will reset the |.state| property on this._update if a new
  2608.         // download initiates.
  2609.       }
  2610.  
  2611.       // if we still fail after trying a complete download, give up completely
  2612.       if (allFailed) {
  2613.         // In all other failure cases, i.e. we're S.O.L. - no more failing over
  2614.         // ...
  2615.  
  2616.         // If this was ever a foreground download, and now there is no UI active
  2617.         // (e.g. because the user closed the download window) and there was an
  2618.         // error, we must notify now. Otherwise we can keep the failure to
  2619.         // ourselves since the user won't be expecting it.
  2620.         try {
  2621.           this._update.QueryInterface(Components.interfaces.nsIWritablePropertyBag);
  2622.           var fgdl = this._update.getProperty("foregroundDownload");
  2623.         }
  2624.         catch (e) {
  2625.         }
  2626.  
  2627.         if (fgdl == "true") {
  2628.           var prompter =
  2629.               Components.classes["@mozilla.org/updates/update-prompt;1"].
  2630.               createInstance(Components.interfaces.nsIUpdatePrompt);
  2631.           this._update.QueryInterface(Components.interfaces.nsIWritablePropertyBag);
  2632.           this._update.setProperty("downloadFailed", "true");
  2633.           prompter.showUpdateError(this._update);
  2634.         }
  2635.       }
  2636.  
  2637.       // the complete download succeeded or total failure was handled, so exit
  2638.       return;
  2639.     }
  2640.  
  2641.     // Do this after *everything* else, since it will likely cause the app
  2642.     // to shut down.
  2643.     if (shouldShowPrompt) {
  2644.       // Notify the user that an update has been downloaded and is ready for
  2645.       // installation (i.e. that they should restart the application). We do
  2646.       // not notify on failed update attempts.
  2647.       var prompter =
  2648.           Components.classes["@mozilla.org/updates/update-prompt;1"].
  2649.           createInstance(Components.interfaces.nsIUpdatePrompt);
  2650.       prompter.showUpdateDownloaded(this._update, true);
  2651.     }
  2652.   },
  2653.  
  2654.   /**
  2655.    * See nsIInterfaceRequestor.idl
  2656.    */
  2657.   getInterface: function(iid) {
  2658.     // The network request may require proxy authentication, so provide the
  2659.     // default nsIAuthPrompt if requested.
  2660.     if (iid.equals(Components.interfaces.nsIAuthPrompt)) {
  2661.       var prompt =
  2662.           Components.classes["@mozilla.org/network/default-auth-prompt;1"].
  2663.           createInstance();
  2664.       return prompt.QueryInterface(iid);
  2665.     }
  2666.     throw Components.results.NS_ERROR_NO_INTERFACE;
  2667.   },
  2668.  
  2669.   /**
  2670.    * See nsISupports.idl
  2671.    */
  2672.   QueryInterface: function(iid) {
  2673.     if (!iid.equals(Components.interfaces.nsIRequestObserver) &&
  2674.         !iid.equals(Components.interfaces.nsIProgressEventSink) &&
  2675.         !iid.equals(Components.interfaces.nsIInterfaceRequestor) &&
  2676.         !iid.equals(Components.interfaces.nsISupports))
  2677.       throw Components.results.NS_ERROR_NO_INTERFACE;
  2678.     return this;
  2679.   }
  2680. };
  2681.  
  2682. /**
  2683.  * A manager for update check timers. Manages timers that fire over long
  2684.  * periods of time (e.g. days, weeks).
  2685.  * @constructor
  2686.  */
  2687. function TimerManager() {
  2688.   getObserverService().addObserver(this, "xpcom-shutdown", false);
  2689.  
  2690.   const nsITimer = Components.interfaces.nsITimer;
  2691.   this._timer = Components.classes["@mozilla.org/timer;1"]
  2692.                           .createInstance(nsITimer);
  2693.   var timerInterval = getPref("getIntPref", PREF_APP_UPDATE_TIMER, 600000);
  2694.   this._timer.initWithCallback(this, timerInterval,
  2695.                                nsITimer.TYPE_REPEATING_SLACK);
  2696. }
  2697. TimerManager.prototype = {
  2698.   /**
  2699.    * See nsIObserver.idl
  2700.    */
  2701.   observe: function(subject, topic, data) {
  2702.     if (topic == "xpcom-shutdown") {
  2703.      getObserverService().removeObserver(this, "xpcom-shutdown");
  2704.  
  2705.       // Release everything we hold onto.
  2706.       for (var timerID in this._timers)
  2707.         delete this._timers[timerID];
  2708.       this._timer = null;
  2709.       this._timers = null;
  2710.     }
  2711.   },
  2712.  
  2713.   /**
  2714.    * The Checker Timer
  2715.    */
  2716.   _timer: null,
  2717.  
  2718.   /**
  2719.    * The set of registered timers.
  2720.    */
  2721.   _timers: { },
  2722.  
  2723.   /**
  2724.    * Called when the checking timer fires.
  2725.    * @param   timer
  2726.    *          The checking timer that fired.
  2727.    */
  2728.   notify: function(timer) {
  2729.     for (var timerID in this._timers) {
  2730.       var timerData = this._timers[timerID];
  2731.       var lastUpdateTime = timerData.lastUpdateTime;
  2732.       var now = Math.round(Date.now() / 1000);
  2733.  
  2734.       // Fudge the lastUpdateTime by some random increment of the update
  2735.       // check interval (e.g. some random slice of 10 minutes) so that when
  2736.       // the time comes to check, we offset each client request by a random
  2737.       // amount so they don't all hit at once. app.update.timer is in milliseconds,
  2738.       // whereas app.update.lastUpdateTime is in seconds
  2739.       var timerInterval = getPref("getIntPref", PREF_APP_UPDATE_TIMER, 600000);
  2740.       lastUpdateTime += Math.round(Math.random() * timerInterval / 1000);
  2741.  
  2742.       if ((now - lastUpdateTime) > timerData.interval &&
  2743.           timerData.callback instanceof Components.interfaces.nsITimerCallback) {
  2744.         timerData.callback.notify(timer);
  2745.         timerData.lastUpdateTime = now;
  2746.         var preference = PREF_APP_UPDATE_LASTUPDATETIME_FMT.replace(/%ID%/, timerID);
  2747.         gPref.setIntPref(preference, now);
  2748.       }
  2749.     }
  2750.   },
  2751.  
  2752.   /**
  2753.    * See nsIUpdateService.idl
  2754.    */
  2755.   registerTimer: function(id, callback, interval) {
  2756.     var preference = PREF_APP_UPDATE_LASTUPDATETIME_FMT.replace(/%ID%/, id);
  2757.     var now = Math.round(Date.now() / 1000);
  2758.     var lastUpdateTime = null;
  2759.     if (gPref.prefHasUserValue(preference)) {
  2760.       lastUpdateTime = gPref.getIntPref(preference);
  2761.     } else {
  2762.       gPref.setIntPref(preference, now);
  2763.       lastUpdateTime = now;
  2764.     }
  2765.     this._timers[id] = { callback       : callback,
  2766.                          interval       : interval,
  2767.                          lastUpdateTime : lastUpdateTime };
  2768.   },
  2769.  
  2770.   /**
  2771.    * See nsISupports.idl
  2772.    */
  2773.   QueryInterface: function(iid) {
  2774.     if (!iid.equals(Components.interfaces.nsIUpdateTimerManager) &&
  2775.         !iid.equals(Components.interfaces.nsITimerCallback) &&
  2776.         !iid.equals(Components.interfaces.nsIObserver) &&
  2777.         !iid.equals(Components.interfaces.nsISupports))
  2778.       throw Components.results.NS_ERROR_NO_INTERFACE;
  2779.     return this;
  2780.   }
  2781. };
  2782.  
  2783. //@line 2783 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  2784. /**
  2785.  * UpdatePrompt
  2786.  * An object which can prompt the user with information about updates, request
  2787.  * action, etc. Embedding clients can override this component with one that
  2788.  * invokes a native front end.
  2789.  * @constructor
  2790.  */
  2791. function UpdatePrompt() {
  2792. }
  2793. UpdatePrompt.prototype = {
  2794.   /**
  2795.    * See nsIUpdateService.idl
  2796.    */
  2797.   checkForUpdates: function() {
  2798.     this._showUI(null, URI_UPDATE_PROMPT_DIALOG, null, "Update:Wizard",
  2799.                  null, null);
  2800.   },
  2801.  
  2802.   /**
  2803.    * See nsIUpdateService.idl
  2804.    */
  2805.   showUpdateAvailable: function(update) {
  2806.     if (this._enabled) {
  2807.       this._showUIWhenIdle(null, URI_UPDATE_PROMPT_DIALOG, null,
  2808.                            "Update:Wizard", "updatesavailable", update);
  2809.     }
  2810.   },
  2811.  
  2812.   /**
  2813.    * See nsIUpdateService.idl
  2814.    */
  2815.   showUpdateDownloaded: function(update, whenIdle) {
  2816.     if (this._enabled) {
  2817.       if (whenIdle)
  2818.         this._showUIWhenIdle(null, URI_UPDATE_PROMPT_DIALOG, null,
  2819.                              "Update:Wizard", "finishedBackground", update);
  2820.       else
  2821.         this._showUI(null, URI_UPDATE_PROMPT_DIALOG, null,
  2822.                      "Update:Wizard", "finishedBackground", update);
  2823.     }
  2824.   },
  2825.  
  2826.   /**
  2827.    * See nsIUpdateService.idl
  2828.    */
  2829.   showUpdateInstalled: function(update) {
  2830.     var showUpdateInstalledUI = getPref("getBoolPref",
  2831.       PREF_APP_UPDATE_SHOW_INSTALLED_UI, true);
  2832.     if (this._enabled && showUpdateInstalledUI) {
  2833.       this._showUI(null, URI_UPDATE_PROMPT_DIALOG, null, "Update:Wizard",
  2834.                    "installed", update);
  2835.     }
  2836.   },
  2837.  
  2838.   /**
  2839.    * See nsIUpdateService.idl
  2840.    */
  2841.   showUpdateError: function(update) {
  2842.     if (this._enabled) {
  2843.       // In some cases, we want to just show a simple alert dialog:
  2844.       if (update.state == STATE_FAILED && update.errorCode == WRITE_ERROR) {
  2845.         var sbs =
  2846.             Components.classes["@mozilla.org/intl/stringbundle;1"].
  2847.             getService(Components.interfaces.nsIStringBundleService);
  2848.         var updateBundle = sbs.createBundle(URI_UPDATES_PROPERTIES);
  2849.         var title = updateBundle.GetStringFromName("updaterIOErrorTitle");
  2850.         var text = updateBundle.formatStringFromName("updaterIOErrorMsg",
  2851.                                                      [gApp.name, gApp.name], 2);
  2852.         var ww =
  2853.             Components.classes["@mozilla.org/embedcomp/window-watcher;1"].
  2854.             getService(Components.interfaces.nsIWindowWatcher);
  2855.         ww.getNewPrompter(null).alert(title, text);
  2856.       } else {
  2857.         this._showUI(null, URI_UPDATE_PROMPT_DIALOG, null, "Update:Wizard",
  2858.                      "errors", update);
  2859.       }
  2860.     }
  2861.   },
  2862.  
  2863.   /**
  2864.    * See nsIUpdateService.idl
  2865.    */
  2866.   showUpdateHistory: function(parent) {
  2867.     this._showUI(parent, URI_UPDATE_HISTORY_DIALOG, "modal,dialog=yes", "Update:History",
  2868.                  null, null);
  2869.   },
  2870.  
  2871.   /**
  2872.    * Whether or not we are enabled (i.e. not in Silent mode)
  2873.    */
  2874.   get _enabled() {
  2875.     return !getPref("getBoolPref", PREF_APP_UPDATE_SILENT, false);
  2876.   },
  2877.  
  2878.   /**
  2879.    * Show the Update Checking UI when the user was idle
  2880.    * @param   parent
  2881.    *          A parent window, can be null
  2882.    * @param   uri
  2883.    *          The URI string of the dialog to show
  2884.    * @param   name
  2885.    *          The Window Name of the dialog to show, in case it is already open
  2886.    *          and can merely be focused
  2887.    * @param   page
  2888.    *          The page of the wizard to be displayed, if one is already open.
  2889.    * @param   update
  2890.    *          An update to pass to the UI in the window arguments.
  2891.    *          Can be null
  2892.    */
  2893.   _showUIWhenIdle: function(parent, uri, features, name, page, update) {
  2894.     var idleService =
  2895.       Components.classes["@mozilla.org/widget/idleservice;1"]
  2896.                 .getService(Components.interfaces.nsIIdleService);
  2897.  
  2898.     const IDLE_TIME = getPref("getIntPref", PREF_APP_UPDATE_IDLETIME, 60);
  2899.     if (idleService.idleTime / 1000 >= IDLE_TIME) {
  2900.       this._showUI(parent, uri, features, name, page, update);
  2901.     } else {
  2902.       var observerService =
  2903.         Components.classes["@mozilla.org/observer-service;1"]
  2904.                   .getService(Components.interfaces.nsIObserverService);
  2905.       var observer = {
  2906.         updatePrompt: this,
  2907.         observe: function (aSubject, aTopic, aData) {
  2908.           switch (aTopic) {
  2909.             case "idle":
  2910.               this.updatePrompt._showUI(parent, uri, features, name, page, update);
  2911.               // fall thru
  2912.             case "quit-application":
  2913.               idleService.removeIdleObserver(this, IDLE_TIME);
  2914.               observerService.removeObserver(this, "quit-application");
  2915.               break;
  2916.           }
  2917.         }
  2918.       };
  2919.       idleService.addIdleObserver(observer, IDLE_TIME);
  2920.       observerService.addObserver(observer, "quit-application", false);
  2921.     }
  2922.   },
  2923.  
  2924.   /**
  2925.    * Show the Update Checking UI
  2926.    * @param   parent
  2927.    *          A parent window, can be null
  2928.    * @param   uri
  2929.    *          The URI string of the dialog to show
  2930.    * @param   name
  2931.    *          The Window Name of the dialog to show, in case it is already open
  2932.    *          and can merely be focused
  2933.    * @param   page
  2934.    *          The page of the wizard to be displayed, if one is already open.
  2935.    * @param   update
  2936.    *          An update to pass to the UI in the window arguments.
  2937.    *          Can be null
  2938.    */
  2939.   _showUI: function(parent, uri, features, name, page, update) {
  2940.     var ary = null;
  2941.     if (update) {
  2942.       ary = Components.classes["@mozilla.org/supports-array;1"]
  2943.                       .createInstance(Components.interfaces.nsISupportsArray);
  2944.       ary.AppendElement(update);
  2945.     }
  2946.  
  2947.     var wm = Components.classes["@mozilla.org/appshell/window-mediator;1"]
  2948.                        .getService(Components.interfaces.nsIWindowMediator);
  2949.     var win = wm.getMostRecentWindow(name);
  2950.     if (win) {
  2951.       if (page && "setCurrentPage" in win)
  2952.         win.setCurrentPage(page);
  2953.       win.focus();
  2954.     }
  2955.     else {
  2956.       var openFeatures = "chrome,centerscreen,dialog=no,resizable=no,titlebar,toolbar=no";
  2957.       if (features)
  2958.         openFeatures += "," + features;
  2959.       var ww = Components.classes["@mozilla.org/embedcomp/window-watcher;1"]
  2960.                          .getService(Components.interfaces.nsIWindowWatcher);
  2961.       ww.openWindow(parent, uri, "", openFeatures, ary);
  2962.     }
  2963.   },
  2964.  
  2965.   /**
  2966.    * See nsISupports.idl
  2967.    */
  2968.   QueryInterface: function(iid) {
  2969.     if (!iid.equals(Components.interfaces.nsIUpdatePrompt) &&
  2970.         !iid.equals(Components.interfaces.nsISupports))
  2971.       throw Components.results.NS_ERROR_NO_INTERFACE;
  2972.     return this;
  2973.   }
  2974. };
  2975. //@line 2975 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  2976.  
  2977. var gModule = {
  2978.   registerSelf: function(componentManager, fileSpec, location, type) {
  2979.     componentManager = componentManager.QueryInterface(Components.interfaces.nsIComponentRegistrar);
  2980.  
  2981.     for (var key in this._objects) {
  2982.       var obj = this._objects[key];
  2983.       componentManager.registerFactoryLocation(obj.CID, obj.className, obj.contractID,
  2984.                                                fileSpec, location, type);
  2985.     }
  2986.  
  2987.     // Make the Update Service a startup observer
  2988.     var categoryManager = Components.classes["@mozilla.org/categorymanager;1"]
  2989.                                     .getService(Components.interfaces.nsICategoryManager);
  2990.     categoryManager.addCategoryEntry("app-startup", this._objects.service.className,
  2991.                                      "service," + this._objects.service.contractID,
  2992.                                      true, true);
  2993.   },
  2994.  
  2995.   getClassObject: function(componentManager, cid, iid) {
  2996.     if (!iid.equals(Components.interfaces.nsIFactory))
  2997.       throw Components.results.NS_ERROR_NOT_IMPLEMENTED;
  2998.  
  2999.     for (var key in this._objects) {
  3000.       if (cid.equals(this._objects[key].CID))
  3001.         return this._objects[key].factory;
  3002.     }
  3003.  
  3004.     throw Components.results.NS_ERROR_NO_INTERFACE;
  3005.   },
  3006.  
  3007.   _objects: {
  3008.     service: { CID        : Components.ID("{B3C290A6-3943-4B89-8BBE-C01EB7B3B311}"),
  3009.                contractID : "@mozilla.org/updates/update-service;1",
  3010.                className  : "Update Service",
  3011.                factory    : makeFactory(UpdateService)
  3012.              },
  3013.     checker: { CID        : Components.ID("{898CDC9B-E43F-422F-9CC4-2F6291B415A3}"),
  3014.                contractID : "@mozilla.org/updates/update-checker;1",
  3015.                className  : "Update Checker",
  3016.                factory    : makeFactory(Checker)
  3017.              },
  3018. //@line 3018 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  3019.     prompt:  { CID        : Components.ID("{27ABA825-35B5-4018-9FDD-F99250A0E722}"),
  3020.                contractID : "@mozilla.org/updates/update-prompt;1",
  3021.                className  : "Update Prompt",
  3022.                factory    : makeFactory(UpdatePrompt)
  3023.              },
  3024. //@line 3024 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  3025.     timers:  { CID        : Components.ID("{B322A5C0-A419-484E-96BA-D7182163899F}"),
  3026.                contractID : "@mozilla.org/updates/timer-manager;1",
  3027.                className  : "Timer Manager",
  3028.                factory    : makeFactory(TimerManager)
  3029.              },
  3030.     manager: { CID        : Components.ID("{093C2356-4843-4C65-8709-D7DBCBBE7DFB}"),
  3031.                contractID : "@mozilla.org/updates/update-manager;1",
  3032.                className  : "Update Manager",
  3033.                factory    : makeFactory(UpdateManager)
  3034.              },
  3035.   },
  3036.  
  3037.   canUnload: function(componentManager) {
  3038.     return true;
  3039.   }
  3040. };
  3041.  
  3042. /**
  3043.  * Creates a factory for instances of an object created using the passed-in
  3044.  * constructor.
  3045.  */
  3046. function makeFactory(ctor) {
  3047.   function ci(outer, iid) {
  3048.     if (outer != null)
  3049.       throw Components.results.NS_ERROR_NO_AGGREGATION;
  3050.     return (new ctor()).QueryInterface(iid);
  3051.   }
  3052.   return { createInstance: ci };
  3053. }
  3054.  
  3055. function NSGetModule(compMgr, fileSpec) {
  3056.   return gModule;
  3057. }
  3058.  
  3059. /**
  3060.  * Determines whether or there are installed addons which are incompatible
  3061.  * with this update.
  3062.  * @param   update
  3063.  *          The update to check compatibility against
  3064.  * @returns true if there are no addons installed that are incompatible with
  3065.  *          the specified update, false otherwise.
  3066.  */
  3067. function isCompatible(update) {
  3068. //@line 3068 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  3069.   var em =
  3070.       Components.classes["@mozilla.org/extensions/manager;1"].
  3071.       getService(nsIExtensionManager);
  3072.   var items = em.getIncompatibleItemList("", update.extensionVersion,
  3073.     update.platformVersion, nsIUpdateItem.TYPE_ADDON, false, { });
  3074.   return items.length == 0;
  3075. //@line 3077 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  3076. }
  3077.  
  3078. /**
  3079.  * Shows a prompt for an update, provided there are no incompatible addons.
  3080.  * If there are, kick off an update check and see if updates are available
  3081.  * that will resolve the incompatibilities.
  3082.  * @param   update
  3083.  *          The available update to show
  3084.  */
  3085. function showPromptIfNoIncompatibilities(update) {
  3086.   function showPrompt(update) {
  3087.     LOG("UpdateService", "_selectAndInstallUpdate: need to prompt user before continuing...");
  3088.     var prompter =
  3089.         Components.classes["@mozilla.org/updates/update-prompt;1"].
  3090.         createInstance(Components.interfaces.nsIUpdatePrompt);
  3091.     prompter.showUpdateAvailable(update);
  3092.   }
  3093.  
  3094. //@line 3096 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  3095.   /**
  3096.    * Determines if an addon is compatible with a particular update.
  3097.    * @param   addon
  3098.    *          The addon to check
  3099.    * @param   version
  3100.    *          The extensionVersion of the update to check for compatibility
  3101.    *          against.
  3102.    * @returns true if the addon is compatible, false otherwise
  3103.    */
  3104.   function addonIsCompatible(addon, version) {
  3105.     var vc =
  3106.         Components.classes["@mozilla.org/xpcom/version-comparator;1"].
  3107.         getService(Components.interfaces.nsIVersionComparator);
  3108.     return (vc.compare(version, addon.minAppVersion) >= 0) &&
  3109.            (vc.compare(version, addon.maxAppVersion) <= 0);
  3110.   }
  3111.  
  3112.   /**
  3113.    * An object implementing nsIAddonUpdateCheckListener that looks for
  3114.    * available updates to addons and if updates are found that will make the
  3115.    * user's installed addon set compatible with the update, suppresses the
  3116.    * prompt that would otherwise be shown.
  3117.    * @param   addons
  3118.    *          An array of incompatible addons that are installed.
  3119.    * @constructor
  3120.    */
  3121.   function Listener(addons) {
  3122.     this._addons = addons;
  3123.   }
  3124.   Listener.prototype = {
  3125.     _addons: null,
  3126.  
  3127.     /**
  3128.      * See nsIUpdateService.idl
  3129.      */
  3130.     onUpdateStarted: function() {
  3131.     },
  3132.     onUpdateEnded: function() {
  3133.       // There are still incompatibilities, even after an extension update
  3134.       // check to see if there were newer, compatible versions available, so
  3135.       // we have to prompt.
  3136.       //
  3137.       // PREF_APP_UPDATE_INCOMPATIBLE_MODE controls the mode in which we
  3138.       // handle incompatibilities:
  3139.       // UPDATE_CHECK_NEWVERSION    We count both VersionInfo updates _and_
  3140.       //      NewerVersion updates against the list of incompatible addons
  3141.       //      installed - i.e. if Foo 1.2 is installed and it is incompatible
  3142.       //      with the update, and we find Foo 2.0 which is but which is not
  3143.       //      yet downloaded or installed, then we do NOT prompt because the
  3144.       //      user can download Foo 2.0 when they restart after the update
  3145.       //      during the mismatch checking UI. This is the default, since it
  3146.       //      suppresses most prompt dialogs.
  3147.       // UPDATE_CHECK_COMPATIBILITY    We count only VersionInfo updates
  3148.       //      against the list of incompatible addons installed - i.e. if the
  3149.       //      situation above with Foo 1.2 and available update to 2.0
  3150.       //      applies, we DO show the prompt since a download operation will
  3151.       //      be required after the update. This is not the default and is
  3152.       //      supplied only as a hidden option for those that want it.
  3153.       var mode = getPref("getIntPref", PREF_APP_UPDATE_INCOMPATIBLE_MODE,
  3154.                          nsIExtensionManager.UPDATE_CHECK_NEWVERSION);
  3155.       if ((mode == nsIExtensionManager.UPDATE_CHECK_NEWVERSION
  3156.            && this._addons.length) || !isCompatible(update))
  3157.         showPrompt(update);
  3158.     },
  3159.     onAddonUpdateStarted: function(addon) {
  3160.     },
  3161.     onAddonUpdateEnded: function(addon, status) {
  3162.       const Ci = Components.interfaces;
  3163.       if (status != Ci.nsIAddonUpdateCheckListener.STATUS_UPDATE)
  3164.         return;
  3165.  
  3166.       var reqVersion = addon.targetAppID == TOOLKIT_ID ?
  3167.                        update.platformVersion :
  3168.                        update.extensionVersion;
  3169.       if (!addonIsCompatible(addon, reqVersion))
  3170.         return;
  3171.  
  3172.       for (var i = 0; i < this._addons.length; ++i) {
  3173.         if (this._addons[i] == addon) {
  3174.           this._addons.splice(i, 1);
  3175.           break;
  3176.         }
  3177.       }
  3178.     },
  3179.     /**
  3180.      * See nsISupports.idl
  3181.      */
  3182.     QueryInterface: function(iid) {
  3183.       if (!iid.equals(Components.interfaces.nsIAddonUpdateCheckListener) &&
  3184.           !iid.equals(Components.interfaces.nsISupports))
  3185.         throw Components.results.NS_ERROR_NO_INTERFACE;
  3186.       return this;
  3187.     }
  3188.   };
  3189.  
  3190.   if (!isCompatible(update)) {
  3191.     var em =
  3192.         Components.classes["@mozilla.org/extensions/manager;1"].
  3193.         getService(nsIExtensionManager);
  3194.     var items = em.getIncompatibleItemList("", update.extensionVersion,
  3195.       update.platformVersion, nsIUpdateItem.TYPE_ADDON, false, { });
  3196.     var listener = new Listener(items);
  3197.     // See documentation on |mode| above.
  3198.     var mode = getPref("getIntPref", PREF_APP_UPDATE_INCOMPATIBLE_MODE,
  3199.                        nsIExtensionManager.UPDATE_CHECK_NEWVERSION);
  3200.     em.update([], 0, mode, listener);
  3201.   }
  3202.   else
  3203. //@line 3205 "c:\builds\xulrunner\xr_trunk_dubya\mozilla\toolkit\mozapps\update\src\nsUpdateService.js.in"
  3204.     showPrompt(update);
  3205. }
  3206.