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