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
12 namespace installer {
13
14 // static
15 // Associate bool member variables with registry entries.
16 const 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
AppCommand()27 AppCommand::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
AppCommand(const string16 & command_line)34 AppCommand::AppCommand(const 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
Initialize(const base::win::RegKey & key)42 bool 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 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
AddWorkItems(HKEY predefined_root,const string16 & command_path,WorkItemList * item_list) const70 void AppCommand::AddWorkItems(HKEY predefined_root,
71 const string16& command_path,
72 WorkItemList* item_list) const {
73 item_list->AddCreateRegKeyWorkItem(predefined_root, command_path)
74 ->set_log_message("creating AppCommand registry key");
75 item_list->AddSetRegValueWorkItem(predefined_root, command_path,
76 google_update::kRegCommandLineField,
77 command_line_, true)
78 ->set_log_message("setting AppCommand CommandLine registry value");
79
80 for (int i = 0; i < arraysize(kNameBoolVars); ++i) {
81 const wchar_t* var_name = kNameBoolVars[i].name;
82 bool var_data = this->*(kNameBoolVars[i].data);
83
84 // Adds a work item to set |var_name| to DWORD 1 if |var_data| is true;
85 // adds a work item to remove |var_name| otherwise.
86 if (var_data) {
87 item_list->AddSetRegValueWorkItem(predefined_root,
88 command_path,
89 var_name,
90 static_cast<DWORD>(1),
91 true);
92 } else {
93 item_list->AddDeleteRegValueWorkItem(predefined_root,
94 command_path,
95 var_name);
96 }
97 }
98 }
99
100 } // namespace installer
101