OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/installer/util/product.h" | 5 #include "chrome/installer/util/product.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 | 8 |
9 #include "base/command_line.h" | 9 #include "base/command_line.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
(...skipping 108 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
119 | 119 |
120 const MasterPreferences& prefs = MasterPreferences::ForCurrentProcess(); | 120 const MasterPreferences& prefs = MasterPreferences::ForCurrentProcess(); |
121 | 121 |
122 bool is_msi = false; | 122 bool is_msi = false; |
123 prefs.GetBool(installer::master_preferences::kMsi, &is_msi); | 123 prefs.GetBool(installer::master_preferences::kMsi, &is_msi); |
124 | 124 |
125 if (!is_msi) { | 125 if (!is_msi) { |
126 // We didn't find it in the preferences, try looking in the registry. | 126 // We didn't find it in the preferences, try looking in the registry. |
127 HKEY reg_root = system_level() ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; | 127 HKEY reg_root = system_level() ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; |
128 RegKey key; | 128 RegKey key; |
129 if (key.Open(reg_root, distribution_->GetStateKey().c_str(), KEY_READ)) { | 129 if (key.Open(reg_root, distribution_->GetStateKey().c_str(), |
130 DWORD msi_value; | 130 KEY_READ) == ERROR_SUCCESS) { |
131 if (key.ReadValueDW(google_update::kRegMSIField, &msi_value) && | 131 DWORD msi_value = 0; |
132 msi_value != 0) { | 132 key.ReadValueDW(google_update::kRegMSIField, &msi_value); |
133 msi_ = true; | 133 msi_ = msi_value != 0; |
134 } | |
135 } | 134 } |
136 } else { | 135 } else { |
137 msi_ = true; | 136 msi_ = true; |
138 } | 137 } |
139 cache_state_ |= MSI_STATE; | 138 cache_state_ |= MSI_STATE; |
140 } | 139 } |
141 | 140 |
142 return msi_; | 141 return msi_; |
143 } | 142 } |
144 | 143 |
145 bool Product::SetMsiMarker(bool set) const { | 144 bool Product::SetMsiMarker(bool set) const { |
146 bool success = false; | |
147 HKEY reg_root = system_level() ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; | 145 HKEY reg_root = system_level() ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; |
148 RegKey client_state_key; | 146 RegKey client_state_key; |
149 if (client_state_key.Open(reg_root, distribution_->GetStateKey().c_str(), | 147 LONG result = client_state_key.Open(reg_root, |
150 KEY_READ | KEY_WRITE)) { | 148 distribution_->GetStateKey().c_str(), KEY_READ | KEY_WRITE); |
151 DWORD msi_value = set ? 1 : 0; | 149 if (result == ERROR_SUCCESS) { |
152 if (client_state_key.WriteValue(google_update::kRegMSIField, msi_value)) { | 150 result = client_state_key.WriteValue(google_update::kRegMSIField, |
153 success = true; | 151 set ? 1 : 0); |
154 } else { | |
155 LOG(ERROR) << "Could not write MSI value to client state key."; | |
156 } | |
157 } else { | |
158 LOG(ERROR) << "SetMsiMarker: Could not open client state key!"; | |
159 } | 152 } |
160 | 153 |
161 return success; | 154 LOG_IF(ERROR, result != ERROR_SUCCESS) << "Failed to Open or Write MSI value" |
| 155 "to client state key. error: " << result; |
| 156 |
| 157 return (result == ERROR_SUCCESS); |
162 } | 158 } |
163 | 159 |
164 bool Product::ShouldCreateUninstallEntry() const { | 160 bool Product::ShouldCreateUninstallEntry() const { |
165 if (IsMsi()) { | 161 if (IsMsi()) { |
166 // MSI installations will manage their own uninstall shortcuts. | 162 // MSI installations will manage their own uninstall shortcuts. |
167 return false; | 163 return false; |
168 } | 164 } |
169 | 165 |
170 return distribution_->ShouldCreateUninstallEntry(); | 166 return distribution_->ShouldCreateUninstallEntry(); |
171 } | 167 } |
(...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
244 if (!distribution) { | 240 if (!distribution) { |
245 NOTREACHED(); | 241 NOTREACHED(); |
246 return false; | 242 return false; |
247 } | 243 } |
248 | 244 |
249 return AddDistribution(distribution); | 245 return AddDistribution(distribution); |
250 } | 246 } |
251 | 247 |
252 } // namespace installer | 248 } // namespace installer |
253 | 249 |
OLD | NEW |