| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/launcher_support/chrome_launcher_support.h" | 5 #include "chrome/installer/launcher_support/chrome_launcher_support.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 #include <tchar.h> | 8 #include <tchar.h> |
| 9 #include "base/file_path.h" | 9 #include "base/file_path.h" |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| 11 #include "base/logging.h" |
| 11 #ifndef OFFICIAL_BUILD | 12 #ifndef OFFICIAL_BUILD |
| 12 #include "base/path_service.h" | 13 #include "base/path_service.h" |
| 13 #endif | 14 #endif |
| 14 #include "base/string16.h" | 15 #include "base/string16.h" |
| 15 #include "base/win/registry.h" | 16 #include "base/win/registry.h" |
| 16 | 17 |
| 18 namespace chrome_launcher_support { |
| 19 |
| 17 namespace { | 20 namespace { |
| 18 | 21 |
| 19 // TODO(erikwright): These constants are duplicated all over the place. | 22 // TODO(huangs) Refactor the constants: http://crbug.com/148538 |
| 20 // Consolidate them somehow. | 23 const wchar_t kGoogleRegClientStateKey[] = |
| 21 const wchar_t kChromeRegClientStateKey[] = | 24 L"Software\\Google\\Update\\ClientState\\"; |
| 22 L"Software\\Google\\Update\\ClientState\\" | |
| 23 L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; | |
| 24 | 25 |
| 26 // Copied from chrome_appid.cc. |
| 27 const wchar_t kBinariesAppGuid[] = L"{4DC8B4CA-1BDA-483e-B5FA-D3C12E15B62D}"; |
| 28 |
| 29 // Copied from google_chrome_distribution.cc. |
| 30 const wchar_t kBrowserAppGuid[] = L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; |
| 31 |
| 32 // Copied from util_constants.cc. |
| 25 const wchar_t kUninstallStringField[] = L"UninstallString"; | 33 const wchar_t kUninstallStringField[] = L"UninstallString"; |
| 26 | |
| 27 const wchar_t kChromeExe[] = L"chrome.exe"; | 34 const wchar_t kChromeExe[] = L"chrome.exe"; |
| 28 | 35 |
| 29 #ifndef OFFICIAL_BUILD | 36 #ifndef OFFICIAL_BUILD |
| 30 FilePath GetDevelopmentChrome() { | 37 FilePath GetDevelopmentChrome() { |
| 31 FilePath current_directory; | 38 FilePath current_directory; |
| 32 if (PathService::Get(base::DIR_EXE, ¤t_directory)) { | 39 if (PathService::Get(base::DIR_EXE, ¤t_directory)) { |
| 33 FilePath chrome_exe_path(current_directory.Append(kChromeExe)); | 40 FilePath chrome_exe_path(current_directory.Append(kChromeExe)); |
| 34 if (file_util::PathExists(chrome_exe_path)) | 41 if (file_util::PathExists(chrome_exe_path)) |
| 35 return chrome_exe_path; | 42 return chrome_exe_path; |
| 36 } | 43 } |
| 37 return FilePath(); | 44 return FilePath(); |
| 38 } | 45 } |
| 39 #endif | 46 #endif |
| 40 | 47 |
| 48 // Reads the path to setup.exe from the value "UninstallString" within the |
| 49 // specified product's "ClientState" registry key. Returns an empty FilePath if |
| 50 // an error occurs or the product is not installed at the specified level. |
| 51 FilePath GetSetupExeFromRegistry(InstallationLevel level, |
| 52 const wchar_t* app_guid) { |
| 53 HKEY root_key = (level == USER_LEVEL_INSTALLATION) ? |
| 54 HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE; |
| 55 string16 subkey(kGoogleRegClientStateKey); |
| 56 subkey.append(app_guid); |
| 57 base::win::RegKey reg_key; |
| 58 if (reg_key.Open(root_key, subkey.c_str(), |
| 59 KEY_QUERY_VALUE) == ERROR_SUCCESS) { |
| 60 string16 uninstall; |
| 61 if (reg_key.ReadValue(kUninstallStringField, &uninstall) == ERROR_SUCCESS) { |
| 62 FilePath setup_exe_path(uninstall); |
| 63 if (file_util::PathExists(setup_exe_path)) |
| 64 return setup_exe_path; |
| 65 } |
| 66 } |
| 67 return FilePath(); |
| 68 } |
| 69 |
| 41 } // namespace | 70 } // namespace |
| 42 | 71 |
| 43 namespace chrome_launcher_support { | 72 FilePath GetSetupExeForInstallationLevel(InstallationLevel level) { |
| 73 // Look in the registry for Chrome Binaries first. |
| 74 FilePath setup_exe_path(GetSetupExeFromRegistry(level, kBinariesAppGuid)); |
| 75 // If the above fails, look in the registry for Chrome next. |
| 76 if (setup_exe_path.empty()) |
| 77 setup_exe_path = GetSetupExeFromRegistry(level, kBrowserAppGuid); |
| 78 // If we fail again, then setup_exe_path would be empty. |
| 79 return setup_exe_path; |
| 80 } |
| 81 |
| 82 FilePath GetChromePathForInstallationLevel(InstallationLevel level) { |
| 83 FilePath setup_exe_path(GetSetupExeForInstallationLevel(level)); |
| 84 if (!setup_exe_path.empty()) { |
| 85 // The uninstall path contains the path to setup.exe which is two levels |
| 86 // down from chrome.exe. Move up two levels (plus one to drop the file |
| 87 // name) and look for chrome.exe from there. |
| 88 FilePath chrome_exe_path( |
| 89 setup_exe_path.DirName().DirName().DirName().Append(kChromeExe)); |
| 90 if (!file_util::PathExists(chrome_exe_path)) { |
| 91 // By way of mild future proofing, look up one to see if there's a |
| 92 // chrome.exe in the version directory |
| 93 chrome_exe_path = chrome_exe_path.DirName().DirName().Append(kChromeExe); |
| 94 } |
| 95 if (file_util::PathExists(chrome_exe_path)) |
| 96 return chrome_exe_path; |
| 97 } |
| 98 return FilePath(); |
| 99 } |
| 44 | 100 |
| 45 FilePath GetAnyChromePath() { | 101 FilePath GetAnyChromePath() { |
| 46 FilePath chrome_path; | 102 FilePath chrome_path; |
| 47 #ifndef OFFICIAL_BUILD | 103 #ifndef OFFICIAL_BUILD |
| 48 // For development mode, chrome.exe should be in same dir as the stub. | 104 // For development mode, chrome.exe should be in same dir as the stub. |
| 49 chrome_path = GetDevelopmentChrome(); | 105 chrome_path = GetDevelopmentChrome(); |
| 50 #endif | 106 #endif |
| 51 if (chrome_path.empty()) | 107 if (chrome_path.empty()) |
| 52 chrome_path = GetChromePathForInstallationLevel(SYSTEM_LEVEL_INSTALLATION); | 108 chrome_path = GetChromePathForInstallationLevel(SYSTEM_LEVEL_INSTALLATION); |
| 53 if (chrome_path.empty()) | 109 if (chrome_path.empty()) |
| 54 chrome_path = GetChromePathForInstallationLevel(USER_LEVEL_INSTALLATION); | 110 chrome_path = GetChromePathForInstallationLevel(USER_LEVEL_INSTALLATION); |
| 55 | |
| 56 return chrome_path; | 111 return chrome_path; |
| 57 } | 112 } |
| 58 | 113 |
| 59 FilePath GetChromePathForInstallationLevel(InstallationLevel level) { | |
| 60 using base::win::RegKey; | |
| 61 HKEY root_key = (level == USER_LEVEL_INSTALLATION ? | |
| 62 HKEY_CURRENT_USER : | |
| 63 HKEY_LOCAL_MACHINE); | |
| 64 RegKey reg_key(root_key, kChromeRegClientStateKey, KEY_QUERY_VALUE); | |
| 65 | |
| 66 FilePath chrome_exe_path; | |
| 67 | |
| 68 if (reg_key.Valid()) { | |
| 69 // Now grab the uninstall string from the appropriate ClientState key | |
| 70 // and use that as the base for a path to chrome.exe. | |
| 71 string16 uninstall; | |
| 72 if (reg_key.ReadValue(kUninstallStringField, &uninstall) == ERROR_SUCCESS) { | |
| 73 // The uninstall path contains the path to setup.exe which is two levels | |
| 74 // down from chrome.exe. Move up two levels (plus one to drop the file | |
| 75 // name) and look for chrome.exe from there. | |
| 76 FilePath uninstall_path(uninstall); | |
| 77 chrome_exe_path = | |
| 78 uninstall_path.DirName().DirName().DirName().Append(kChromeExe); | |
| 79 if (!file_util::PathExists(chrome_exe_path)) { | |
| 80 // By way of mild future proofing, look up one to see if there's a | |
| 81 // chrome.exe in the version directory | |
| 82 chrome_exe_path = | |
| 83 uninstall_path.DirName().DirName().Append(kChromeExe); | |
| 84 } | |
| 85 } | |
| 86 } | |
| 87 | |
| 88 if (file_util::PathExists(chrome_exe_path)) | |
| 89 return chrome_exe_path; | |
| 90 | |
| 91 return FilePath(); | |
| 92 } | |
| 93 | |
| 94 } // namespace chrome_launcher_support | 114 } // namespace chrome_launcher_support |
| OLD | NEW |