1// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include "chrome/installer/util/app_command.h" 6 7#include "base/logging.h" 8#include "base/win/registry.h" 9#include "chrome/installer/util/google_update_constants.h" 10#include "chrome/installer/util/work_item_list.h" 11 12namespace installer { 13 14// static 15// Associate bool member variables with registry entries. 16const AppCommand::NamedBoolVar AppCommand::kNameBoolVars[] = { 17 {&AppCommand::sends_pings_, 18 google_update::kRegSendsPingsField}, 19 {&AppCommand::is_web_accessible_, 20 google_update::kRegWebAccessibleField}, 21 {&AppCommand::is_auto_run_on_os_upgrade_, 22 google_update::kRegAutoRunOnOSUpgradeField}, 23 {&AppCommand::is_run_as_user_, 24 google_update::kRegRunAsUserField}, 25}; 26 27AppCommand::AppCommand() 28 : sends_pings_(false), 29 is_web_accessible_(false), 30 is_auto_run_on_os_upgrade_(false), 31 is_run_as_user_(false) { 32} 33 34AppCommand::AppCommand(const base::string16& command_line) 35 : command_line_(command_line), 36 sends_pings_(false), 37 is_web_accessible_(false), 38 is_auto_run_on_os_upgrade_(false), 39 is_run_as_user_(false) { 40} 41 42bool AppCommand::Initialize(const base::win::RegKey& key) { 43 if (!key.Valid()) { 44 LOG(DFATAL) << "Cannot initialize an AppCommand from an invalid key."; 45 return false; 46 } 47 48 LONG result = ERROR_SUCCESS; 49 base::string16 cmd_line; 50 51 result = key.ReadValue(google_update::kRegCommandLineField, &cmd_line); 52 if (result != ERROR_SUCCESS) { 53 LOG(WARNING) << "Error reading " << google_update::kRegCommandLineField 54 << " value from registry: " << result; 55 return false; 56 } 57 58 command_line_.swap(cmd_line); 59 60 for (int i = 0; i < arraysize(kNameBoolVars); ++i) { 61 DWORD value = 0; // Set default to false. 62 // Note: ReadValueDW only modifies out param on success. 63 key.ReadValueDW(kNameBoolVars[i].name, &value); 64 this->*(kNameBoolVars[i].data) = (value != 0); 65 } 66 67 return true; 68} 69 70void AppCommand::AddWorkItems(HKEY predefined_root, 71 const base::string16& command_path, 72 WorkItemList* item_list) const { 73 // Command_path is derived from GetRegCommandKey which always returns 74 // value from GetVersionKey() which should be 32-bit hive. 75 item_list->AddCreateRegKeyWorkItem( 76 predefined_root, command_path, KEY_WOW64_32KEY) 77 ->set_log_message("creating AppCommand registry key"); 78 item_list->AddSetRegValueWorkItem(predefined_root, 79 command_path, 80 KEY_WOW64_32KEY, 81 google_update::kRegCommandLineField, 82 command_line_, 83 true) 84 ->set_log_message("setting AppCommand CommandLine registry value"); 85 86 for (int i = 0; i < arraysize(kNameBoolVars); ++i) { 87 const wchar_t* var_name = kNameBoolVars[i].name; 88 bool var_data = this->*(kNameBoolVars[i].data); 89 90 // Adds a work item to set |var_name| to DWORD 1 if |var_data| is true; 91 // adds a work item to remove |var_name| otherwise. 92 if (var_data) { 93 item_list->AddSetRegValueWorkItem(predefined_root, 94 command_path, 95 KEY_WOW64_32KEY, 96 var_name, 97 static_cast<DWORD>(1), 98 true); 99 } else { 100 item_list->AddDeleteRegValueWorkItem( 101 predefined_root, command_path, KEY_WOW64_32KEY, var_name); 102 } 103 } 104} 105 106} // namespace installer 107