| 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 |
| 11 #include <algorithm> | 11 #include <algorithm> |
| 12 #include <limits> | 12 #include <limits> |
| 13 #include <string> | 13 #include <string> |
| 14 | 14 |
| 15 #include "base/files/file_path.h" | 15 #include "base/files/file_path.h" |
| 16 #include "base/files/file_util.h" | 16 #include "base/files/file_util.h" |
| 17 #include "base/md5.h" | 17 #include "base/md5.h" |
| 18 #include "base/process/kill.h" | 18 #include "base/process/kill.h" |
| 19 #include "base/process/launch.h" | 19 #include "base/process/launch.h" |
| 20 #include "base/process/process_handle.h" | 20 #include "base/process/process.h" |
| 21 #include "base/strings/string_util.h" | 21 #include "base/strings/string_util.h" |
| 22 #include "base/strings/utf_string_conversions.h" | 22 #include "base/strings/utf_string_conversions.h" |
| 23 #include "base/win/registry.h" | 23 #include "base/win/registry.h" |
| 24 #include "base/win/scoped_comptr.h" | 24 #include "base/win/scoped_comptr.h" |
| 25 #include "base/win/scoped_handle.h" | 25 #include "base/win/scoped_handle.h" |
| 26 #include "base/win/win_util.h" | 26 #include "base/win/win_util.h" |
| 27 #include "chrome/installer/util/browser_distribution.h" | 27 #include "chrome/installer/util/browser_distribution.h" |
| 28 #include "chrome/installer/util/install_util.h" | 28 #include "chrome/installer/util/install_util.h" |
| 29 #include "chrome/installer/util/util_constants.h" | 29 #include "chrome/installer/util/util_constants.h" |
| 30 #include "google_update/google_update_idl.h" | 30 #include "google_update/google_update_idl.h" |
| (...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 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::Process process; |
| 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 base::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 process = base::LaunchProcess(update_command, launch_options); |
| 119 &process_handle)) { | 119 if (!process.IsValid()) { |
| 120 AtlTrace("%hs. Failed to launch command to finalize update; " | 120 AtlTrace("%hs. Failed to launch command to finalize update; " |
| 121 "error %u.\n", __FUNCTION__, ::GetLastError()); | 121 "error %u.\n", __FUNCTION__, ::GetLastError()); |
| 122 } | 122 } |
| 123 } | 123 } |
| 124 } else { | 124 } else { |
| 125 // Run the update command via Google Update. | 125 // Run the update command via Google Update. |
| 126 HRESULT hr = S_OK; | 126 HRESULT hr = S_OK; |
| 127 base::win::ScopedComPtr<IProcessLauncher> process_launcher; | 127 base::win::ScopedComPtr<IProcessLauncher> process_launcher; |
| 128 hr = process_launcher.CreateInstance(__uuidof(ProcessLauncherClass)); | 128 hr = process_launcher.CreateInstance(__uuidof(ProcessLauncherClass)); |
| 129 if (FAILED(hr)) { | 129 if (FAILED(hr)) { |
| 130 AtlTrace("%hs. Failed to Create ProcessLauncher; hr=0x%X.\n", | 130 AtlTrace("%hs. Failed to Create ProcessLauncher; hr=0x%X.\n", |
| 131 __FUNCTION__, hr); | 131 __FUNCTION__, hr); |
| 132 } else { | 132 } else { |
| 133 ULONG_PTR handle = 0; | 133 ULONG_PTR handle = 0; |
| 134 BrowserDistribution* dist = BrowserDistribution::GetDistribution(); | 134 BrowserDistribution* dist = BrowserDistribution::GetDistribution(); |
| 135 hr = process_launcher->LaunchCmdElevated( | 135 hr = process_launcher->LaunchCmdElevated( |
| 136 dist->GetAppGuid().c_str(), kRenameCommandValue, | 136 dist->GetAppGuid().c_str(), kRenameCommandValue, |
| 137 GetCurrentProcessId(), &handle); | 137 GetCurrentProcessId(), &handle); |
| 138 if (FAILED(hr)) { | 138 if (FAILED(hr)) { |
| 139 AtlTrace("%hs. Failed to launch command to finalize update; " | 139 AtlTrace("%hs. Failed to launch command to finalize update; " |
| 140 "hr=0x%X.\n", __FUNCTION__, hr); | 140 "hr=0x%X.\n", __FUNCTION__, hr); |
| 141 } else { | 141 } else { |
| 142 process_handle.Set(reinterpret_cast<base::ProcessHandle>(handle)); | 142 process = base::Process(reinterpret_cast<base::ProcessHandle>(handle)); |
| 143 } | 143 } |
| 144 } | 144 } |
| 145 } | 145 } |
| 146 | 146 |
| 147 // Wait for the update to complete and report the results. | 147 // Wait for the update to complete and report the results. |
| 148 if (process_handle.IsValid()) { | 148 if (process.IsValid()) { |
| 149 int exit_code = 0; | 149 int exit_code = 0; |
| 150 // WaitForExitCode will close the handle in all cases. | 150 if (!base::WaitForExitCodeWithTimeout( |
| 151 if (!base::WaitForExitCode(process_handle.Take(), &exit_code)) { | 151 process.Handle(), &exit_code, |
| 152 base::TimeDelta::FromMilliseconds(INFINITE))) { |
| 152 AtlTrace("%hs. Failed to get result when finalizing update.\n", | 153 AtlTrace("%hs. Failed to get result when finalizing update.\n", |
| 153 __FUNCTION__); | 154 __FUNCTION__); |
| 154 } else if (exit_code != installer::RENAME_SUCCESSFUL) { | 155 } else if (exit_code != installer::RENAME_SUCCESSFUL) { |
| 155 AtlTrace("%hs. Failed to finalize update with exit code %d.\n", | 156 AtlTrace("%hs. Failed to finalize update with exit code %d.\n", |
| 156 __FUNCTION__, exit_code); | 157 __FUNCTION__, exit_code); |
| 157 } else { | 158 } else { |
| 158 AtlTrace("%hs. Finalized pending update.\n", __FUNCTION__); | 159 AtlTrace("%hs. Finalized pending update.\n", __FUNCTION__); |
| 159 } | 160 } |
| 160 } | 161 } |
| 161 #endif | 162 #endif |
| 162 } | 163 } |
| 163 | 164 |
| 164 } // delegate_execute | 165 } // delegate_execute |
| OLD | NEW |