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 <windows.h> | 7 #include <windows.h> |
8 #include <atlbase.h> | 8 #include <atlbase.h> |
9 #include <shlobj.h> | 9 #include <shlobj.h> |
10 | 10 |
(...skipping 27 matching lines...) Expand all Loading... |
38 const base::FilePath::CharType kNewChromeExe[] = | 38 const base::FilePath::CharType kNewChromeExe[] = |
39 FILE_PATH_LITERAL("new_chrome.exe"); | 39 FILE_PATH_LITERAL("new_chrome.exe"); |
40 const wchar_t kRenameCommandValue[] = L"cmd"; | 40 const wchar_t kRenameCommandValue[] = L"cmd"; |
41 const wchar_t kRegPathChromeClientBase[] = | 41 const wchar_t kRegPathChromeClientBase[] = |
42 L"Software\\Google\\Update\\Clients\\"; | 42 L"Software\\Google\\Update\\Clients\\"; |
43 | 43 |
44 // Returns the name of the global event used to detect if |chrome_exe| is in | 44 // Returns the name of the global event used to detect if |chrome_exe| is in |
45 // use by a browser process. | 45 // use by a browser process. |
46 // TODO(grt): Move this somewhere central so it can be used by both this | 46 // TODO(grt): Move this somewhere central so it can be used by both this |
47 // IsBrowserRunning (below) and IsBrowserAlreadyRunning (browser_util_win.cc). | 47 // IsBrowserRunning (below) and IsBrowserAlreadyRunning (browser_util_win.cc). |
48 string16 GetEventName(const base::FilePath& chrome_exe) { | 48 base::string16 GetEventName(const base::FilePath& chrome_exe) { |
49 static wchar_t const kEventPrefix[] = L"Global\\"; | 49 static wchar_t const kEventPrefix[] = L"Global\\"; |
50 const size_t prefix_len = arraysize(kEventPrefix) - 1; | 50 const size_t prefix_len = arraysize(kEventPrefix) - 1; |
51 string16 name; | 51 base::string16 name; |
52 name.reserve(prefix_len + chrome_exe.value().size()); | 52 name.reserve(prefix_len + chrome_exe.value().size()); |
53 name.assign(kEventPrefix, prefix_len); | 53 name.assign(kEventPrefix, prefix_len); |
54 name.append(chrome_exe.value()); | 54 name.append(chrome_exe.value()); |
55 std::replace(name.begin() + prefix_len, name.end(), '\\', '!'); | 55 std::replace(name.begin() + prefix_len, name.end(), '\\', '!'); |
56 std::transform(name.begin() + prefix_len, name.end(), | 56 std::transform(name.begin() + prefix_len, name.end(), |
57 name.begin() + prefix_len, tolower); | 57 name.begin() + prefix_len, tolower); |
58 return name; | 58 return name; |
59 } | 59 } |
60 | 60 |
61 // Returns true if |chrome_exe| is in use by a browser process. In this case, | 61 // Returns true if |chrome_exe| is in use by a browser process. In this case, |
(...skipping 11 matching lines...) Expand all Loading... |
73 return false; | 73 return false; |
74 } | 74 } |
75 | 75 |
76 // Returns true if the file new_chrome.exe exists in the same directory as | 76 // Returns true if the file new_chrome.exe exists in the same directory as |
77 // |chrome_exe|. | 77 // |chrome_exe|. |
78 bool NewChromeExeExists(const base::FilePath& chrome_exe) { | 78 bool NewChromeExeExists(const base::FilePath& chrome_exe) { |
79 base::FilePath new_chrome_exe(chrome_exe.DirName().Append(kNewChromeExe)); | 79 base::FilePath new_chrome_exe(chrome_exe.DirName().Append(kNewChromeExe)); |
80 return base::PathExists(new_chrome_exe); | 80 return base::PathExists(new_chrome_exe); |
81 } | 81 } |
82 | 82 |
83 bool GetUpdateCommand(bool is_per_user, string16* update_command) { | 83 bool GetUpdateCommand(bool is_per_user, base::string16* update_command) { |
84 const HKEY root = is_per_user ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE; | 84 const HKEY root = is_per_user ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE; |
85 BrowserDistribution* dist = BrowserDistribution::GetDistribution(); | 85 BrowserDistribution* dist = BrowserDistribution::GetDistribution(); |
86 string16 reg_path_chrome_client = kRegPathChromeClientBase; | 86 base::string16 reg_path_chrome_client = kRegPathChromeClientBase; |
87 reg_path_chrome_client.append(dist->GetAppGuid()); | 87 reg_path_chrome_client.append(dist->GetAppGuid()); |
88 base::win::RegKey key(root, reg_path_chrome_client.c_str(), KEY_QUERY_VALUE); | 88 base::win::RegKey key(root, reg_path_chrome_client.c_str(), KEY_QUERY_VALUE); |
89 | 89 |
90 return key.ReadValue(kRenameCommandValue, update_command) == ERROR_SUCCESS; | 90 return key.ReadValue(kRenameCommandValue, update_command) == ERROR_SUCCESS; |
91 } | 91 } |
92 | 92 |
93 #endif // GOOGLE_CHROME_BUILD | 93 #endif // GOOGLE_CHROME_BUILD |
94 | 94 |
95 } // namespace | 95 } // namespace |
96 | 96 |
97 namespace delegate_execute { | 97 namespace delegate_execute { |
98 | 98 |
99 void UpdateChromeIfNeeded(const base::FilePath& chrome_exe) { | 99 void UpdateChromeIfNeeded(const base::FilePath& chrome_exe) { |
100 #if defined(GOOGLE_CHROME_BUILD) | 100 #if defined(GOOGLE_CHROME_BUILD) |
101 // Nothing to do if a browser is already running or if there's no | 101 // Nothing to do if a browser is already running or if there's no |
102 // new_chrome.exe. | 102 // new_chrome.exe. |
103 if (IsBrowserRunning(chrome_exe) || !NewChromeExeExists(chrome_exe)) | 103 if (IsBrowserRunning(chrome_exe) || !NewChromeExeExists(chrome_exe)) |
104 return; | 104 return; |
105 | 105 |
106 base::win::ScopedHandle process_handle; | 106 base::win::ScopedHandle process_handle; |
107 | 107 |
108 if (InstallUtil::IsPerUserInstall(chrome_exe.value().c_str())) { | 108 if (InstallUtil::IsPerUserInstall(chrome_exe.value().c_str())) { |
109 // Read the update command from the registry. | 109 // Read the update command from the registry. |
110 string16 update_command; | 110 base::string16 update_command; |
111 if (!GetUpdateCommand(true, &update_command)) { | 111 if (!GetUpdateCommand(true, &update_command)) { |
112 AtlTrace("%hs. Failed to read update command from registry.\n", | 112 AtlTrace("%hs. Failed to read update command from registry.\n", |
113 __FUNCTION__); | 113 __FUNCTION__); |
114 } else { | 114 } else { |
115 // Run the update command. | 115 // Run the update command. |
116 base::LaunchOptions launch_options; | 116 base::LaunchOptions launch_options; |
117 launch_options.start_hidden = true; | 117 launch_options.start_hidden = true; |
118 if (!base::LaunchProcess(update_command, launch_options, | 118 if (!base::LaunchProcess(update_command, launch_options, |
119 &process_handle)) { | 119 &process_handle)) { |
120 AtlTrace("%hs. Failed to launch command to finalize update; " | 120 AtlTrace("%hs. Failed to launch command to finalize update; " |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
155 AtlTrace("%hs. Failed to finalize update with exit code %d.\n", | 155 AtlTrace("%hs. Failed to finalize update with exit code %d.\n", |
156 __FUNCTION__, exit_code); | 156 __FUNCTION__, exit_code); |
157 } else { | 157 } else { |
158 AtlTrace("%hs. Finalized pending update.\n", __FUNCTION__); | 158 AtlTrace("%hs. Finalized pending update.\n", __FUNCTION__); |
159 } | 159 } |
160 } | 160 } |
161 #endif | 161 #endif |
162 } | 162 } |
163 | 163 |
164 } // delegate_execute | 164 } // delegate_execute |
OLD | NEW |