1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// 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_auto_run_on_os_upgrade_(false) {
}
AppCommand::AppCommand(const string16& command_line)
: command_line_(command_line),
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;
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;
}
// 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 string16& command_path,
WorkItemList* item_list) const {
item_list->AddCreateRegKeyWorkItem(predefined_root, command_path)
->set_log_message("creating AppCommand registry key");
item_list->AddSetRegValueWorkItem(predefined_root, command_path,
google_update::kRegCommandLineField,
command_line_, true)
->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
|
