Callback flow to register Chrome and update shortcuts after OS upgrade to Windows 8

chrome/installer/setup/install_worker.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 //
 // This file contains the definitions of the installer functions that build
 // the WorkItemList used to install the application.
 
 #include "chrome/installer/setup/install_worker.h"
 
 #include <oaidl.h>
@@ skipping 301 matching lines @@
   for (size_t i = 0; i < products.size(); ++i) {
     const Product& p = *products[i];
     if (p.is_chrome_frame()) {
       AddChromeFrameWorkItems(original_state, installer_state, setup_path,
                               new_version, p, list);
     }
     if (p.is_chrome_app_host()) {
       AddInstallAppCommandWorkItems(installer_state, original_state,
                                     &setup_path, &new_version, p, list);
     }
+    if (p.is_chrome()) {
+      // In Chrome Browser, add registry entries so that when an OS upgrade
+      // occurs, Google Update will call setup.exe with --on-os-upgrade switch,
+      // which leads to ChromeBrowserOnOsUpgrade() being called.
+      CommandLine cmd_line(CommandLine::NO_PROGRAM);
+      RenderChromeBrowserOsUpgradeCommand(installer_state, setup_path,
+                                          new_version, &cmd_line);
+      AddOsUpgradeWorkItems(original_state, installer_state, p, cmd_line, list);
+    }
   }
 }
 
 // Mirror oeminstall the first time anything is installed multi.  There is no
 // need to update the value on future install/update runs since this value never
 // changes.  Note that the value is removed by Google Update after EULA
 // acceptance is processed.
 void AddOemInstallWorkItems(const InstallationState& original_state,
                             const InstallerState& installer_state,
                             WorkItemList* install_list) {
@@ skipping 1254 matching lines @@
   AddGenericQuickEnableWorkItems(installer_state,
                                  machine_state,
                                  setup_path,
                                  new_version,
                                  work_item_list,
                                  false,  // have_child_product
                                  cmd_line,
                                  kCmdQuickEnableApplicationHost);
 }
 
+void RenderChromeBrowserOsUpgradeCommand(

[gab, 2012/08/28 16:08:19]

s/Render/Get

[gab, 2012/08/28 16:08:19]

Remove "ChromeBrowser" from function name

[grt (UTC plus 2), 2012/08/28 19:35:39]

suggestion: you can make this product-independent by passing in the const
Product& product instance and using product.AppendProductFlags(cmd_line) in
place of the manual appends of --chrome and --multi-install.  in that case, i
don't know that there's any good reason not to roll it into
AddOsUpgradeWorkItems, so i propose that you do that too.

[huangs, 2012/08/29 17:02:54]

Took grt@'s approach.

+    const InstallerState& installer_state,
+    const FilePath& setup_path,
+    const Version& new_version,
+    CommandLine* cmd_line) {
+  FilePath installer_path(installer_state
+      .GetInstallerDirectory(new_version).Append(setup_path.BaseName()));
+  cmd_line->SetProgram(installer_path);
+  cmd_line->AppendSwitch(installer::switches::kOnOsUpgrade);
+    cmd_line->AppendSwitch(installer::switches::kChrome);

[gab, 2012/08/28 16:08:19]

nit: unindent 2 spaces

[huangs, 2012/08/29 17:02:54]

Done.

+  if (installer_state.system_install())
+    cmd_line->AppendSwitch(installer::switches::kSystemLevel);
+  if (installer_state.is_multi_install())
+    cmd_line->AppendSwitch(installer::switches::kMultiInstall);
+  if (installer_state.verbose_logging())
+    cmd_line->AppendSwitch(installer::switches::kVerboseLogging);
+}
+
+void AddOsUpgradeWorkItems(const InstallationState& original_state,
+                           const InstallerState& installer_state,
+                           const Product& product,
+                           const CommandLine& cmd_line,
+                           WorkItemList* install_list) {
+  const HKEY root_key = installer_state.root_key();
+  std::wstring cmd_key(product.distribution()->GetVersionKey());

[gab, 2012/08/28 16:08:19]

std::wstring -> string16

[huangs, 2012/08/29 17:02:54]

Done.  Also replacing everywhere else in file.

On 2012/08/28 16:08:19, gab wrote:

+  cmd_key.append(1, L'\\').append(google_update::kRegCommandsKey)

[gab, 2012/08/28 16:08:19]

use
.push_back(FilePath::kSeparators[0])
instead of
.append(1, L'\\')

[huangs, 2012/08/29 17:02:54]

Okay, but .push_back() returns void, and cannot be chained.  So I'm using
.append(1, FilePath::kSeparators[0]) instead.

On 2012/08/28 16:08:19, gab wrote:

+      .append(1, L'\\').append(kCmdOnOsUpgrade);
+  if (installer_state.operation() != InstallerState::UNINSTALL) {
+    install_list->set_log_message("Adding OS upgrade command");
+    AppCommand cmd(cmd_line.GetCommandLineString(), false, false, true);
+    cmd.AddWorkItems(installer_state.root_key(), cmd_key, install_list);
+  } else {
+    install_list->set_log_message("Removing OS upgrade command");
+    install_list->AddDeleteRegKeyWorkItem(root_key, cmd_key);
+  }
+}
+
 }  // namespace installer