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

chrome/installer/util/app_command.cc

 // Copyright (c) 2011 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.
 
 #include "chrome/installer/util/app_command.h"
 
 #include "base/logging.h"
 #include "base/win/registry.h"
 #include "chrome/installer/util/google_update_constants.h"
 #include "chrome/installer/util/work_item_list.h"
 
 namespace installer {
 
+namespace {
+
+// Adds a work item to set |value_name| to DWORD 1 if |value_data| is true;
+// adds a work item to remove |value_name| otherwise.
+WorkItem* AddSetOptionalBoolRegValueWorkItem(
+    HKEY predefined_root,
+    const string16& key_path,
+    const string16& value_name,
+    bool value_data,
+    WorkItemList* item_list) {
+  if (value_data) {
+    return item_list->AddSetRegValueWorkItem(predefined_root,
+                                             key_path,
+                                             value_name,
+                                             static_cast<DWORD>(1),
+                                             true);
+  } else {
+    return item_list->AddDeleteRegValueWorkItem(predefined_root,
+                                                key_path,
+                                                value_name);
+  }
+}
+
+}  // namespace
+
 AppCommand::AppCommand()
     : sends_pings_(false),
-      is_web_accessible_(false) {
+      is_web_accessible_(false),
+      is_auto_run_on_os_upgrade_(false) {
 }
 
-AppCommand::AppCommand(const std::wstring& command_line,
-                       bool sends_pings,
-                       bool is_web_accessible)
+AppCommand::AppCommand(const string16& command_line)
     : command_line_(command_line),
-      sends_pings_(sends_pings),
-      is_web_accessible_(is_web_accessible) {
+      sends_pings_(false),
+      is_web_accessible_(false),
+      is_auto_run_on_os_upgrade_(false) {
 }
 
 bool AppCommand::Initialize(const base::win::RegKey& key) {
   if (!key.Valid()) {
     LOG(DFATAL) << "Cannot initialize an AppCommand from an invalid key.";
     return false;
   }
 
   LONG result = ERROR_SUCCESS;
-  std::wstring cmd_line;
+  string16 cmd_line;
   DWORD sends_pings = 0;
   DWORD is_web_acc = 0;
+  DWORD is_auto_run_on_os_upgrade = 0;
 
   result = key.ReadValue(google_update::kRegCommandLineField, &cmd_line);
   if (result != ERROR_SUCCESS) {
     LOG(WARNING) << "Error reading " << google_update::kRegCommandLineField
                  << " value from registry: " << result;
     return false;
   }
 
-  result = key.ReadValueDW(google_update::kRegSendsPingsField, &sends_pings);
-  if (result != ERROR_SUCCESS) {
-    LOG(WARNING) << "Error reading " << google_update::kRegSendsPingsField
-                 << " value from registry: " << result;
-    return false;
-  }
-
-  result = key.ReadValueDW(google_update::kRegWebAccessibleField, &is_web_acc);
-  if (result != ERROR_SUCCESS) {
-    LOG(WARNING) << "Error reading " << google_update::kRegWebAccessibleField
-                 << " value from registry: " << result;
-    return false;
-  }
+  // Note: ReadValueDW only modifies its out param on success.
+  key.ReadValueDW(google_update::kRegSendsPingsField, &sends_pings);
+  key.ReadValueDW(google_update::kRegWebAccessibleField, &is_web_acc);
+  key.ReadValueDW(google_update::kRegAutoRunOnOSUpgradeField,
+                  &is_auto_run_on_os_upgrade);
 
   command_line_.swap(cmd_line);
   sends_pings_ = (sends_pings != 0);
   is_web_accessible_ = (is_web_acc != 0);
+  is_auto_run_on_os_upgrade_ = (is_auto_run_on_os_upgrade != 0);
 
   return true;
 }
 
 void AppCommand::AddWorkItems(HKEY predefined_root,
-                              const std::wstring& command_path,
+                              const string16& command_path,
                               WorkItemList* item_list) const {
-  const DWORD sends_pings = sends_pings_ ? 1U : 0U;
-  const DWORD is_web_accessible = is_web_accessible_ ? 1U : 0U;
-
   item_list->AddCreateRegKeyWorkItem(predefined_root, command_path)
-      ->set_log_message("creating quick-enable-cf command registry key");
+      ->set_log_message("creating AppCommand registry key");
   item_list->AddSetRegValueWorkItem(predefined_root, command_path,
                                     google_update::kRegCommandLineField,
                                     command_line_, true)
-      ->set_log_message("setting quick-enable-cf CommandLine registry value");
-  item_list->AddSetRegValueWorkItem(predefined_root, command_path,
-                                    google_update::kRegSendsPingsField,
-                                    sends_pings, true)
-      ->set_log_message("setting quick-enable-cf SendsPings registry value");
-  item_list->AddSetRegValueWorkItem(predefined_root, command_path,
-                                    google_update::kRegWebAccessibleField,
-                                    is_web_accessible, true)
-      ->set_log_message("setting quick-enable-cf WebAccessible registry value");
+      ->set_log_message("setting AppCommand CommandLine registry value");
+  AddSetOptionalBoolRegValueWorkItem(predefined_root,
+                                     command_path,
+                                     google_update::kRegSendsPingsField,
+                                     sends_pings_,
+                                     item_list)
+      ->set_log_message("setting AppCommand SendsPings registry value");
+  AddSetOptionalBoolRegValueWorkItem(predefined_root,
+                                     command_path,
+                                     google_update::kRegWebAccessibleField,
+                                     is_web_accessible_,
+                                     item_list)
+      ->set_log_message("setting AppCommand WebAccessible registry value");
+  AddSetOptionalBoolRegValueWorkItem(predefined_root,
+                                     command_path,
+                                     google_update::kRegAutoRunOnOSUpgradeField,
+                                     is_auto_run_on_os_upgrade_,
+                                     item_list)
+      ->set_log_message("setting AppCommand AutoRunOnOSUpgrade registry value");
 }
 
 }  // namespace installer