| OLD | NEW |
| (Empty) | |
| 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 |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/installer/launcher_support/chrome_launcher_support.h" |
| 6 |
| 7 #include <windows.h> |
| 8 #include <tchar.h> |
| 9 #include "base/file_path.h" |
| 10 #include "base/file_util.h" |
| 11 #ifndef OFFICIAL_BUILD |
| 12 #include "base/path_service.h" |
| 13 #endif |
| 14 #include "base/string16.h" |
| 15 #include "base/win/registry.h" |
| 16 |
| 17 namespace { |
| 18 |
| 19 // TODO(erikwright): These constants are duplicated all over the place. |
| 20 // Consolidate them somehow. |
| 21 const wchar_t kChromeRegClientStateKey[] = |
| 22 L"Software\\Google\\Update\\ClientState\\" |
| 23 L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; |
| 24 |
| 25 const wchar_t kUninstallStringField[] = L"UninstallString"; |
| 26 |
| 27 const wchar_t kChromeExe[] = L"chrome.exe"; |
| 28 |
| 29 #ifndef OFFICIAL_BUILD |
| 30 FilePath GetDevelopmentChrome() { |
| 31 FilePath current_directory; |
| 32 if (PathService::Get(base::DIR_EXE, ¤t_directory)) { |
| 33 FilePath chrome_exe_path(current_directory.Append(kChromeExe)); |
| 34 if (file_util::PathExists(chrome_exe_path)) |
| 35 return chrome_exe_path; |
| 36 } |
| 37 return FilePath(); |
| 38 } |
| 39 #endif |
| 40 |
| 41 } // namespace |
| 42 |
| 43 namespace chrome_launcher_support { |
| 44 |
| 45 FilePath GetAnyChromePath() { |
| 46 FilePath chrome_path; |
| 47 #ifndef OFFICIAL_BUILD |
| 48 // For development mode, chrome.exe should be in same dir as the stub. |
| 49 chrome_path = GetDevelopmentChrome(); |
| 50 #endif |
| 51 if (chrome_path.empty()) |
| 52 chrome_path = GetChromePathForInstallationLevel(SYSTEM_LEVEL_INSTALLATION); |
| 53 if (chrome_path.empty()) |
| 54 chrome_path = GetChromePathForInstallationLevel(USER_LEVEL_INSTALLATION); |
| 55 |
| 56 return chrome_path; |
| 57 } |
| 58 |
| 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 |
| OLD | NEW |