| 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 "win8/delegate_execute/chrome_util.h" | 5 #include "win8/delegate_execute/chrome_util.h" |
| 6 | 6 |
| 7 #include <atlbase.h> | 7 #include <atlbase.h> |
| 8 #include <shlobj.h> | 8 #include <shlobj.h> |
| 9 #include <windows.h> | 9 #include <windows.h> |
| 10 | 10 |
| (...skipping 18 matching lines...) Expand all Loading... |
| 29 #if defined(GOOGLE_CHROME_BUILD) | 29 #if defined(GOOGLE_CHROME_BUILD) |
| 30 const wchar_t kAppUserModelId[] = L"Chrome"; | 30 const wchar_t kAppUserModelId[] = L"Chrome"; |
| 31 #else // GOOGLE_CHROME_BUILD | 31 #else // GOOGLE_CHROME_BUILD |
| 32 const wchar_t kAppUserModelId[] = L"Chromium"; | 32 const wchar_t kAppUserModelId[] = L"Chromium"; |
| 33 #endif // GOOGLE_CHROME_BUILD | 33 #endif // GOOGLE_CHROME_BUILD |
| 34 | 34 |
| 35 #if defined(GOOGLE_CHROME_BUILD) | 35 #if defined(GOOGLE_CHROME_BUILD) |
| 36 | 36 |
| 37 // TODO(grt): These constants live in installer_util. Consider moving them | 37 // TODO(grt): These constants live in installer_util. Consider moving them |
| 38 // into common_constants to allow for reuse. | 38 // into common_constants to allow for reuse. |
| 39 const FilePath::CharType kNewChromeExe[] = FILE_PATH_LITERAL("new_chrome.exe"); | 39 const base::FilePath::CharType kNewChromeExe[] = |
| 40 FILE_PATH_LITERAL("new_chrome.exe"); |
| 40 const wchar_t kRenameCommandValue[] = L"cmd"; | 41 const wchar_t kRenameCommandValue[] = L"cmd"; |
| 41 const wchar_t kChromeAppGuid[] = L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; | 42 const wchar_t kChromeAppGuid[] = L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; |
| 42 const wchar_t kRegPathChromeClient[] = | 43 const wchar_t kRegPathChromeClient[] = |
| 43 L"Software\\Google\\Update\\Clients\\" | 44 L"Software\\Google\\Update\\Clients\\" |
| 44 L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; | 45 L"{8A69D345-D564-463c-AFF1-A69D9E530F96}"; |
| 45 const int kExitCodeRenameSuccessful = 23; | 46 const int kExitCodeRenameSuccessful = 23; |
| 46 | 47 |
| 47 // Returns the name of the global event used to detect if |chrome_exe| is in | 48 // Returns the name of the global event used to detect if |chrome_exe| is in |
| 48 // use by a browser process. | 49 // use by a browser process. |
| 49 // TODO(grt): Move this somewhere central so it can be used by both this | 50 // TODO(grt): Move this somewhere central so it can be used by both this |
| 50 // IsBrowserRunning (below) and IsBrowserAlreadyRunning (browser_util_win.cc). | 51 // IsBrowserRunning (below) and IsBrowserAlreadyRunning (browser_util_win.cc). |
| 51 string16 GetEventName(const FilePath& chrome_exe) { | 52 string16 GetEventName(const base::FilePath& chrome_exe) { |
| 52 static wchar_t const kEventPrefix[] = L"Global\\"; | 53 static wchar_t const kEventPrefix[] = L"Global\\"; |
| 53 const size_t prefix_len = arraysize(kEventPrefix) - 1; | 54 const size_t prefix_len = arraysize(kEventPrefix) - 1; |
| 54 string16 name; | 55 string16 name; |
| 55 name.reserve(prefix_len + chrome_exe.value().size()); | 56 name.reserve(prefix_len + chrome_exe.value().size()); |
| 56 name.assign(kEventPrefix, prefix_len); | 57 name.assign(kEventPrefix, prefix_len); |
| 57 name.append(chrome_exe.value()); | 58 name.append(chrome_exe.value()); |
| 58 std::replace(name.begin() + prefix_len, name.end(), '\\', '!'); | 59 std::replace(name.begin() + prefix_len, name.end(), '\\', '!'); |
| 59 std::transform(name.begin() + prefix_len, name.end(), | 60 std::transform(name.begin() + prefix_len, name.end(), |
| 60 name.begin() + prefix_len, tolower); | 61 name.begin() + prefix_len, tolower); |
| 61 return name; | 62 return name; |
| 62 } | 63 } |
| 63 | 64 |
| 64 // Returns true if |chrome_exe| is in use by a browser process. In this case, | 65 // Returns true if |chrome_exe| is in use by a browser process. In this case, |
| 65 // "in use" means past ChromeBrowserMainParts::PreMainMessageLoopRunImpl. | 66 // "in use" means past ChromeBrowserMainParts::PreMainMessageLoopRunImpl. |
| 66 bool IsBrowserRunning(const FilePath& chrome_exe) { | 67 bool IsBrowserRunning(const base::FilePath& chrome_exe) { |
| 67 base::win::ScopedHandle handle(::OpenEvent( | 68 base::win::ScopedHandle handle(::OpenEvent( |
| 68 SYNCHRONIZE, FALSE, GetEventName(chrome_exe).c_str())); | 69 SYNCHRONIZE, FALSE, GetEventName(chrome_exe).c_str())); |
| 69 if (handle.IsValid()) | 70 if (handle.IsValid()) |
| 70 return true; | 71 return true; |
| 71 DWORD last_error = ::GetLastError(); | 72 DWORD last_error = ::GetLastError(); |
| 72 if (last_error != ERROR_FILE_NOT_FOUND) { | 73 if (last_error != ERROR_FILE_NOT_FOUND) { |
| 73 AtlTrace("%hs. Failed to open browser event; error %u.\n", __FUNCTION__, | 74 AtlTrace("%hs. Failed to open browser event; error %u.\n", __FUNCTION__, |
| 74 last_error); | 75 last_error); |
| 75 } | 76 } |
| 76 return false; | 77 return false; |
| 77 } | 78 } |
| 78 | 79 |
| 79 // Returns true if the file new_chrome.exe exists in the same directory as | 80 // Returns true if the file new_chrome.exe exists in the same directory as |
| 80 // |chrome_exe|. | 81 // |chrome_exe|. |
| 81 bool NewChromeExeExists(const FilePath& chrome_exe) { | 82 bool NewChromeExeExists(const base::FilePath& chrome_exe) { |
| 82 FilePath new_chrome_exe(chrome_exe.DirName().Append(kNewChromeExe)); | 83 base::FilePath new_chrome_exe(chrome_exe.DirName().Append(kNewChromeExe)); |
| 83 return file_util::PathExists(new_chrome_exe); | 84 return file_util::PathExists(new_chrome_exe); |
| 84 } | 85 } |
| 85 | 86 |
| 86 bool GetUpdateCommand(bool is_per_user, string16* update_command) { | 87 bool GetUpdateCommand(bool is_per_user, string16* update_command) { |
| 87 const HKEY root = is_per_user ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE; | 88 const HKEY root = is_per_user ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE; |
| 88 base::win::RegKey key(root, kRegPathChromeClient, KEY_QUERY_VALUE); | 89 base::win::RegKey key(root, kRegPathChromeClient, KEY_QUERY_VALUE); |
| 89 | 90 |
| 90 return key.ReadValue(kRenameCommandValue, update_command) == ERROR_SUCCESS; | 91 return key.ReadValue(kRenameCommandValue, update_command) == ERROR_SUCCESS; |
| 91 } | 92 } |
| 92 | 93 |
| 93 #endif // GOOGLE_CHROME_BUILD | 94 #endif // GOOGLE_CHROME_BUILD |
| 94 | 95 |
| 95 // TODO(grt): This code also lives in installer_util. Refactor for reuse. | 96 // TODO(grt): This code also lives in installer_util. Refactor for reuse. |
| 96 bool IsPerUserInstall(const FilePath& chrome_exe) { | 97 bool IsPerUserInstall(const base::FilePath& chrome_exe) { |
| 97 wchar_t program_files_path[MAX_PATH] = {0}; | 98 wchar_t program_files_path[MAX_PATH] = {0}; |
| 98 if (SUCCEEDED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL, | 99 if (SUCCEEDED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL, |
| 99 SHGFP_TYPE_CURRENT, program_files_path))) { | 100 SHGFP_TYPE_CURRENT, program_files_path))) { |
| 100 return !StartsWith(chrome_exe.value().c_str(), program_files_path, false); | 101 return !StartsWith(chrome_exe.value().c_str(), program_files_path, false); |
| 101 } else { | 102 } else { |
| 102 NOTREACHED(); | 103 NOTREACHED(); |
| 103 } | 104 } |
| 104 return true; | 105 return true; |
| 105 } | 106 } |
| 106 | 107 |
| (...skipping 74 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 181 suffix->reserve(base32_md5.length() + 1); | 182 suffix->reserve(base32_md5.length() + 1); |
| 182 suffix->assign(1, L'.'); | 183 suffix->assign(1, L'.'); |
| 183 suffix->append(base32_md5); | 184 suffix->append(base32_md5); |
| 184 return true; | 185 return true; |
| 185 } | 186 } |
| 186 | 187 |
| 187 } // namespace | 188 } // namespace |
| 188 | 189 |
| 189 namespace delegate_execute { | 190 namespace delegate_execute { |
| 190 | 191 |
| 191 void UpdateChromeIfNeeded(const FilePath& chrome_exe) { | 192 void UpdateChromeIfNeeded(const base::FilePath& chrome_exe) { |
| 192 #if defined(GOOGLE_CHROME_BUILD) | 193 #if defined(GOOGLE_CHROME_BUILD) |
| 193 // Nothing to do if a browser is already running or if there's no | 194 // Nothing to do if a browser is already running or if there's no |
| 194 // new_chrome.exe. | 195 // new_chrome.exe. |
| 195 if (IsBrowserRunning(chrome_exe) || !NewChromeExeExists(chrome_exe)) | 196 if (IsBrowserRunning(chrome_exe) || !NewChromeExeExists(chrome_exe)) |
| 196 return; | 197 return; |
| 197 | 198 |
| 198 base::ProcessHandle process_handle = base::kNullProcessHandle; | 199 base::ProcessHandle process_handle = base::kNullProcessHandle; |
| 199 | 200 |
| 200 if (IsPerUserInstall(chrome_exe)) { | 201 if (IsPerUserInstall(chrome_exe)) { |
| 201 // Read the update command from the registry. | 202 // Read the update command from the registry. |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 246 AtlTrace("%hs. Failed to finalize update with exit code %d.\n", | 247 AtlTrace("%hs. Failed to finalize update with exit code %d.\n", |
| 247 __FUNCTION__, exit_code); | 248 __FUNCTION__, exit_code); |
| 248 } else { | 249 } else { |
| 249 AtlTrace("%hs. Finalized pending update.\n", __FUNCTION__); | 250 AtlTrace("%hs. Finalized pending update.\n", __FUNCTION__); |
| 250 } | 251 } |
| 251 } | 252 } |
| 252 #endif | 253 #endif |
| 253 } | 254 } |
| 254 | 255 |
| 255 // TODO(gab): This code also lives in shell_util. Refactor for reuse. | 256 // TODO(gab): This code also lives in shell_util. Refactor for reuse. |
| 256 string16 GetAppId(const FilePath& chrome_exe) { | 257 string16 GetAppId(const base::FilePath& chrome_exe) { |
| 257 string16 app_id(kAppUserModelId); | 258 string16 app_id(kAppUserModelId); |
| 258 string16 suffix; | 259 string16 suffix; |
| 259 if (IsPerUserInstall(chrome_exe) && | 260 if (IsPerUserInstall(chrome_exe) && |
| 260 !GetUserSpecificRegistrySuffix(&suffix)) { | 261 !GetUserSpecificRegistrySuffix(&suffix)) { |
| 261 AtlTrace("%hs. GetUserSpecificRegistrySuffix failed.\n", | 262 AtlTrace("%hs. GetUserSpecificRegistrySuffix failed.\n", |
| 262 __FUNCTION__); | 263 __FUNCTION__); |
| 263 } | 264 } |
| 264 return app_id.append(suffix); | 265 return app_id.append(suffix); |
| 265 } | 266 } |
| 266 | 267 |
| 267 } // delegate_execute | 268 } // delegate_execute |
| OLD | NEW |