Index: chrome/installer/mini_installer/configuration.cc |
diff --git a/chrome/installer/mini_installer/configuration.cc b/chrome/installer/mini_installer/configuration.cc |
index f6c793dbca3edf35274ca7d2b0fd362b3e0411fd..0af0275e691fd9d5efd15f74d013d849c376cc18 100644 |
--- a/chrome/installer/mini_installer/configuration.cc |
+++ b/chrome/installer/mini_installer/configuration.cc |
@@ -11,6 +11,7 @@ |
#include "chrome/installer/mini_installer/appid.h" |
#include "chrome/installer/mini_installer/mini_installer_constants.h" |
#include "chrome/installer/mini_installer/mini_installer_resource.h" |
+#include "chrome/installer/mini_installer/mini_string.h" |
#include "chrome/installer/mini_installer/regkey.h" |
namespace mini_installer { |
@@ -36,56 +37,10 @@ Configuration::~Configuration() { |
Clear(); |
} |
-// When multi_install is true, we are potentially: |
-// 1. Performing a multi-install of some product(s) on a clean machine. |
-// Neither the product(s) nor the multi-installer will have a |
-// ClientState key in the registry, so there is no key to be modified. |
-// 2. Upgrading an existing multi-install. The multi-installer will have |
-// a ClientState key in the registry. Only it need be modified. |
-// 3. Migrating a single-install into a multi-install. The product will |
-// have a ClientState key in the registry. Only it need be modified. |
-// To handle all cases, we inspect the product's ClientState to see if it |
-// exists and its "ap" value does not contain "-multi". This is case 3, |
-// so we modify the product's ClientState. Otherwise, we check the |
-// multi-installer's ClientState and modify it if it exists. |
-// TODO(bcwhite): Write a unit test for this that uses registry virtualization. |
-void Configuration::SetChromeAppGuid() { |
- const HKEY root_key = |
- is_system_level_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; |
- const wchar_t* app_guid = |
- has_chrome_frame_ ? |
- google_update::kChromeFrameAppGuid : |
- is_side_by_side_ ? google_update::kSxSAppGuid |
- : google_update::kAppGuid; |
- |
- // This is the value for single-install and case 3. |
- chrome_app_guid_ = app_guid; |
- |
- if (is_multi_install_) { |
- ValueString value; |
- LONG ret = ERROR_SUCCESS; |
- if (ReadClientStateRegistryValue(root_key, app_guid, &ret, value)) { |
- // The product has a client state key. See if it's a single-install. |
- if (ret == ERROR_FILE_NOT_FOUND || |
- (ret == ERROR_SUCCESS && |
- !FindTagInStr(value.get(), kMultiInstallTag, NULL))) { |
- // yes -- case 3: use the existing key. |
- return; |
- } |
- } |
- // error, case 1, or case 2: modify the multi-installer's key. |
- chrome_app_guid_ = google_update::kMultiInstallAppGuid; |
- } |
-} |
- |
-bool Configuration::ReadClientStateRegistryValue( |
- const HKEY root_key, const wchar_t* app_guid, |
- LONG* retval, ValueString& value) { |
- RegKey key; |
- if (!OpenClientStateKey(root_key, app_guid, KEY_QUERY_VALUE, &key)) |
- return false; |
- *retval = key.ReadSZValue(kApRegistryValue, value.get(), value.capacity()); |
- return true; |
+bool Configuration::Initialize(HMODULE module) { |
+ Clear(); |
+ ReadResources(module); |
+ return ParseCommandLine(::GetCommandLine()); |
} |
const wchar_t* Configuration::program() const { |
@@ -101,51 +56,40 @@ void Configuration::Clear() { |
command_line_ = NULL; |
operation_ = INSTALL_PRODUCT; |
argument_count_ = 0; |
- has_chrome_ = false; |
- has_chrome_frame_ = false; |
- is_multi_install_ = false; |
is_system_level_ = false; |
is_side_by_side_ = false; |
+ is_updating_multi_chrome_ = false; |
previous_version_ = NULL; |
} |
-bool Configuration::Initialize(HMODULE module) { |
- Clear(); |
- ReadResources(module); |
- return ParseCommandLine(::GetCommandLine()); |
-} |
- |
// |command_line| is shared with this instance in the sense that this |
// instance may refer to it at will throughout its lifetime, yet it will |
// not release it. |
bool Configuration::ParseCommandLine(const wchar_t* command_line) { |
command_line_ = command_line; |
args_ = ::CommandLineToArgvW(command_line_, &argument_count_); |
- if (args_ != NULL) { |
- for (int i = 1; i < argument_count_; ++i) { |
- if (0 == ::lstrcmpi(args_[i], L"--chrome-sxs")) |
- is_side_by_side_ = true; |
- else if (0 == ::lstrcmpi(args_[i], L"--chrome")) |
- has_chrome_ = true; |
- else if (0 == ::lstrcmpi(args_[i], L"--chrome-frame")) |
- has_chrome_frame_ = true; |
- else if (0 == ::lstrcmpi(args_[i], L"--multi-install")) |
- is_multi_install_ = true; |
- else if (0 == ::lstrcmpi(args_[i], L"--system-level")) |
- is_system_level_ = true; |
- else if (0 == ::lstrcmpi(args_[i], L"--cleanup")) |
- operation_ = CLEANUP; |
- } |
+ if (!args_) |
+ return false; |
- if (!is_system_level_) |
- is_system_level_ = GetGoogleUpdateIsMachineEnvVar(); |
- SetChromeAppGuid(); |
- if (!is_multi_install_) { |
- has_chrome_ = !has_chrome_frame_; |
+ for (int i = 1; i < argument_count_; ++i) { |
+ if (0 == ::lstrcmpi(args_[i], L"--system-level")) { |
+ is_system_level_ = true; |
+#if defined(GOOGLE_CHROME_BUILD) |
+ } else if (0 == ::lstrcmpi(args_[i], L"--chrome-sxs")) { |
+ is_side_by_side_ = true; |
+ chrome_app_guid_ = google_update::kSxSAppGuid; |
+#endif |
+ } else if (0 == ::lstrcmpi(args_[i], L"--cleanup")) { |
+ operation_ = CLEANUP; |
} |
} |
- return args_ != NULL; |
+ if (!is_system_level_) |
+ is_system_level_ = GetGoogleUpdateIsMachineEnvVar(); |
+ |
+ is_updating_multi_chrome_ = IsUpdatingMultiChrome(); |
+ |
+ return true; |
} |
void Configuration::ReadResources(HMODULE module) { |
@@ -177,4 +121,29 @@ void Configuration::ReadResources(HMODULE module) { |
previous_version_ = version_string; |
} |
+bool Configuration::IsUpdatingMultiChrome() const { |
+#if defined(GOOGLE_CHROME_BUILD) |
+ // SxS/canary does not support multi-install. |
+ if (is_side_by_side_) |
+ return false; |
+ |
+ // Is Chrome already installed as multi-install? |
+ const HKEY root = is_system_level_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER; |
+ StackString<128> value; |
+ RegKey key; |
+ return (OpenClientsKey(root, google_update::kAppGuid, KEY_QUERY_VALUE, |
huangs
2017/01/31 17:40:04
I'd prefer using |chrome_app_guid_| instead of goo
grt (UTC plus 2)
2017/02/02 08:17:15
Done.
|
+ &key) == ERROR_SUCCESS && |
+ key.ReadSZValue(kPvRegistryValue, value.get(), value.capacity()) == |
+ ERROR_SUCCESS && |
+ value.length() != 0 && |
+ OpenClientStateKey(root, google_update::kAppGuid, KEY_QUERY_VALUE, |
+ &key) == ERROR_SUCCESS && |
+ key.ReadSZValue(kUninstallArgumentsRegistryValue, value.get(), |
+ value.capacity()) == ERROR_SUCCESS && |
+ value.findi(L"--multi-install") != nullptr); |
+#else |
+ return false; |
+#endif |
+} |
+ |
} // namespace mini_installer |