| 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/installation_state.h" | 5 #include "chrome/installer/util/installation_state.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/string_util.h" | 8 #include "base/string_util.h" |
| 9 #include "base/version.h" | 9 #include "base/version.h" |
| 10 #include "base/win/registry.h" | 10 #include "base/win/registry.h" |
| 11 #include "chrome/installer/util/google_update_constants.h" | 11 #include "chrome/installer/util/google_update_constants.h" |
| 12 #include "chrome/installer/util/install_util.h" | 12 #include "chrome/installer/util/install_util.h" |
| 13 | 13 |
| 14 namespace installer { | 14 namespace installer { |
| 15 | 15 |
| 16 ProductState::ProductState() | 16 ProductState::ProductState() |
| 17 : uninstall_command_(CommandLine::NO_PROGRAM), | 17 : uninstall_command_(CommandLine::NO_PROGRAM), |
| 18 msi_(false), | 18 msi_(false), |
| 19 multi_install_(false) { | 19 multi_install_(false) { |
| 20 } | 20 } |
| 21 | 21 |
| 22 bool ProductState::Initialize(bool system_install, | 22 bool ProductState::Initialize(bool system_install, |
| 23 BrowserDistribution::Type type) { | 23 BrowserDistribution::Type type) { |
| 24 return Initialize(system_install, | 24 return Initialize(system_install, |
| 25 BrowserDistribution::GetSpecificDistribution(type)); | 25 BrowserDistribution::GetSpecificDistribution(type)); |
| 26 } | 26 } |
| 27 | 27 |
| 28 // Initializes |commands| from the "Commands" subkey of |version_key|. |
| 29 // Returns false if there is no "Commands" subkey or on error. |
| 30 // static |
| 31 bool ProductState::InitializeCommands(const base::win::RegKey& version_key, |
| 32 AppCommands* commands) { |
| 33 static const DWORD kAccess = KEY_ENUMERATE_SUB_KEYS | KEY_QUERY_VALUE; |
| 34 base::win::RegKey commands_key; |
| 35 |
| 36 if (commands_key.Open(version_key.Handle(), google_update::kRegCommandsKey, |
| 37 kAccess) == ERROR_SUCCESS) |
| 38 return commands->Initialize(commands_key); |
| 39 return false; |
| 40 } |
| 41 |
| 28 bool ProductState::Initialize(bool system_install, | 42 bool ProductState::Initialize(bool system_install, |
| 29 BrowserDistribution* distribution) { | 43 BrowserDistribution* distribution) { |
| 30 const std::wstring version_key(distribution->GetVersionKey()); | 44 const std::wstring version_key(distribution->GetVersionKey()); |
| 31 const std::wstring state_key(distribution->GetStateKey()); | 45 const std::wstring state_key(distribution->GetStateKey()); |
| 32 const HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; | 46 const HKEY root_key = system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; |
| 33 base::win::RegKey key(root_key, version_key.c_str(), KEY_QUERY_VALUE); | 47 base::win::RegKey key(root_key, version_key.c_str(), KEY_QUERY_VALUE); |
| 34 std::wstring version_str; | 48 std::wstring version_str; |
| 35 if (key.ReadValue(google_update::kRegVersionField, | 49 if (key.ReadValue(google_update::kRegVersionField, |
| 36 &version_str) == ERROR_SUCCESS) { | 50 &version_str) == ERROR_SUCCESS) { |
| 37 version_.reset(Version::GetVersionFromString(WideToASCII(version_str))); | 51 version_.reset(Version::GetVersionFromString(WideToASCII(version_str))); |
| 38 if (version_.get() != NULL) { | 52 if (version_.get() != NULL) { |
| 39 // The product is installed. | 53 // The product is installed. |
| 40 if (key.ReadValue(google_update::kRegOldVersionField, | 54 if (key.ReadValue(google_update::kRegOldVersionField, |
| 41 &version_str) == ERROR_SUCCESS) { | 55 &version_str) == ERROR_SUCCESS) { |
| 42 old_version_.reset( | 56 old_version_.reset( |
| 43 Version::GetVersionFromString(WideToASCII(version_str))); | 57 Version::GetVersionFromString(WideToASCII(version_str))); |
| 44 } else { | 58 } else { |
| 45 old_version_.reset(); | 59 old_version_.reset(); |
| 46 } | 60 } |
| 47 if (key.ReadValue(google_update::kRegRenameCmdField, | 61 if (key.ReadValue(google_update::kRegRenameCmdField, |
| 48 &rename_cmd_) != ERROR_SUCCESS) | 62 &rename_cmd_) != ERROR_SUCCESS) |
| 49 rename_cmd_.clear(); | 63 rename_cmd_.clear(); |
| 64 if (!InitializeCommands(key, &commands_)) |
| 65 commands_.Clear(); |
| 50 // Read from the ClientState key. | 66 // Read from the ClientState key. |
| 51 channel_.set_value(std::wstring()); | 67 channel_.set_value(std::wstring()); |
| 52 uninstall_command_ = CommandLine(CommandLine::NO_PROGRAM); | 68 uninstall_command_ = CommandLine(CommandLine::NO_PROGRAM); |
| 53 msi_ = false; | 69 msi_ = false; |
| 54 multi_install_ = false; | 70 multi_install_ = false; |
| 55 if (key.Open(root_key, state_key.c_str(), | 71 if (key.Open(root_key, state_key.c_str(), |
| 56 KEY_QUERY_VALUE) == ERROR_SUCCESS) { | 72 KEY_QUERY_VALUE) == ERROR_SUCCESS) { |
| 57 std::wstring setup_path; | 73 std::wstring setup_path; |
| 58 std::wstring uninstall_arguments; | 74 std::wstring uninstall_arguments; |
| 59 // "ap" will be absent if not managed by Google Update. | 75 // "ap" will be absent if not managed by Google Update. |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 92 return *version_; | 108 return *version_; |
| 93 } | 109 } |
| 94 | 110 |
| 95 ProductState& ProductState::CopyFrom(const ProductState& other) { | 111 ProductState& ProductState::CopyFrom(const ProductState& other) { |
| 96 channel_.set_value(other.channel_.value()); | 112 channel_.set_value(other.channel_.value()); |
| 97 version_.reset(other.version_.get() == NULL ? NULL : other.version_->Clone()); | 113 version_.reset(other.version_.get() == NULL ? NULL : other.version_->Clone()); |
| 98 old_version_.reset( | 114 old_version_.reset( |
| 99 other.old_version_.get() == NULL ? NULL : other.old_version_->Clone()); | 115 other.old_version_.get() == NULL ? NULL : other.old_version_->Clone()); |
| 100 rename_cmd_ = other.rename_cmd_; | 116 rename_cmd_ = other.rename_cmd_; |
| 101 uninstall_command_ = other.uninstall_command_; | 117 uninstall_command_ = other.uninstall_command_; |
| 118 commands_.CopyFrom(other.commands_); |
| 102 msi_ = other.msi_; | 119 msi_ = other.msi_; |
| 103 multi_install_ = other.multi_install_; | 120 multi_install_ = other.multi_install_; |
| 104 | 121 |
| 105 return *this; | 122 return *this; |
| 106 } | 123 } |
| 107 | 124 |
| 108 InstallationState::InstallationState() { | 125 InstallationState::InstallationState() { |
| 109 } | 126 } |
| 110 | 127 |
| 111 // static | 128 // static |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 143 | 160 |
| 144 const ProductState* InstallationState::GetProductState( | 161 const ProductState* InstallationState::GetProductState( |
| 145 bool system_install, | 162 bool system_install, |
| 146 BrowserDistribution::Type type) const { | 163 BrowserDistribution::Type type) const { |
| 147 const ProductState& product_state = (system_install ? system_products_ : | 164 const ProductState& product_state = (system_install ? system_products_ : |
| 148 user_products_)[IndexFromDistType(type)]; | 165 user_products_)[IndexFromDistType(type)]; |
| 149 return product_state.version_.get() == NULL ? NULL : &product_state; | 166 return product_state.version_.get() == NULL ? NULL : &product_state; |
| 150 } | 167 } |
| 151 | 168 |
| 152 } // namespace installer | 169 } // namespace installer |
| OLD | NEW |