OLD | NEW |
---|---|
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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 "base/process/kill.h" | 5 #include "base/process/kill.h" |
6 | 6 |
7 #include <io.h> | 7 #include <io.h> |
8 #include <windows.h> | 8 #include <windows.h> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
(...skipping 171 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
182 } | 182 } |
183 | 183 |
184 bool WaitForExitCode(ProcessHandle handle, int* exit_code) { | 184 bool WaitForExitCode(ProcessHandle handle, int* exit_code) { |
185 // TODO(rvargas) crbug.com/417532: Remove this function. | 185 // TODO(rvargas) crbug.com/417532: Remove this function. |
186 Process process(handle); | 186 Process process(handle); |
187 return process.WaitForExit(exit_code); | 187 return process.WaitForExit(exit_code); |
188 } | 188 } |
189 | 189 |
190 bool WaitForExitCodeWithTimeout(ProcessHandle handle, | 190 bool WaitForExitCodeWithTimeout(ProcessHandle handle, |
191 int* exit_code, | 191 int* exit_code, |
192 base::TimeDelta timeout) { | 192 TimeDelta timeout) { |
193 if (::WaitForSingleObject( | 193 if (::WaitForSingleObject( |
194 handle, static_cast<DWORD>(timeout.InMilliseconds())) != WAIT_OBJECT_0) | 194 handle, static_cast<DWORD>(timeout.InMilliseconds())) != WAIT_OBJECT_0) |
195 return false; | 195 return false; |
196 DWORD temp_code; // Don't clobber out-parameters in case of failure. | 196 DWORD temp_code; // Don't clobber out-parameters in case of failure. |
197 if (!::GetExitCodeProcess(handle, &temp_code)) | 197 if (!::GetExitCodeProcess(handle, &temp_code)) |
198 return false; | 198 return false; |
199 | 199 |
200 *exit_code = temp_code; | 200 *exit_code = temp_code; |
201 return true; | 201 return true; |
202 } | 202 } |
203 | 203 |
204 bool WaitForProcessesToExit(const FilePath::StringType& executable_name, | 204 bool WaitForProcessesToExit(const FilePath::StringType& executable_name, |
205 base::TimeDelta wait, | 205 TimeDelta wait, |
206 const ProcessFilter* filter) { | 206 const ProcessFilter* filter) { |
207 bool result = true; | 207 bool result = true; |
208 DWORD start_time = GetTickCount(); | 208 DWORD start_time = GetTickCount(); |
209 | 209 |
210 NamedProcessIterator iter(executable_name, filter); | 210 NamedProcessIterator iter(executable_name, filter); |
211 for (const ProcessEntry* entry = iter.NextProcessEntry(); entry; | 211 for (const ProcessEntry* entry = iter.NextProcessEntry(); entry; |
212 entry = iter.NextProcessEntry()) { | 212 entry = iter.NextProcessEntry()) { |
213 DWORD remaining_wait = static_cast<DWORD>(std::max( | 213 DWORD remaining_wait = static_cast<DWORD>(std::max( |
214 static_cast<int64>(0), | 214 static_cast<int64>(0), |
215 wait.InMilliseconds() - (GetTickCount() - start_time))); | 215 wait.InMilliseconds() - (GetTickCount() - start_time))); |
216 HANDLE process = OpenProcess(SYNCHRONIZE, | 216 HANDLE process = OpenProcess(SYNCHRONIZE, |
217 FALSE, | 217 FALSE, |
218 entry->th32ProcessID); | 218 entry->th32ProcessID); |
219 DWORD wait_result = WaitForSingleObject(process, remaining_wait); | 219 DWORD wait_result = WaitForSingleObject(process, remaining_wait); |
220 CloseHandle(process); | 220 CloseHandle(process); |
221 result &= (wait_result == WAIT_OBJECT_0); | 221 result &= (wait_result == WAIT_OBJECT_0); |
222 } | 222 } |
223 | 223 |
224 return result; | 224 return result; |
225 } | 225 } |
226 | 226 |
227 bool WaitForSingleProcess(ProcessHandle handle, base::TimeDelta wait) { | |
228 int exit_code; | |
229 return WaitForExitCodeWithTimeout(handle, &exit_code, wait) && exit_code == 0; | |
Lei Zhang
2015/02/18 02:49:45
On Windows, we used to check the exit code, but wi
rvargas (doing something else)
2015/02/18 23:17:58
I don't think it matters. I added EXPECT_EQ(0, exi
| |
230 } | |
231 | |
232 bool CleanupProcesses(const FilePath::StringType& executable_name, | 227 bool CleanupProcesses(const FilePath::StringType& executable_name, |
233 base::TimeDelta wait, | 228 TimeDelta wait, |
234 int exit_code, | 229 int exit_code, |
235 const ProcessFilter* filter) { | 230 const ProcessFilter* filter) { |
236 if (WaitForProcessesToExit(executable_name, wait, filter)) | 231 if (WaitForProcessesToExit(executable_name, wait, filter)) |
237 return true; | 232 return true; |
238 KillProcesses(executable_name, exit_code, filter); | 233 KillProcesses(executable_name, exit_code, filter); |
239 return false; | 234 return false; |
240 } | 235 } |
241 | 236 |
242 void EnsureProcessTerminated(Process process) { | 237 void EnsureProcessTerminated(Process process) { |
243 DCHECK(!process.is_current()); | 238 DCHECK(!process.is_current()); |
244 | 239 |
245 // If already signaled, then we are done! | 240 // If already signaled, then we are done! |
246 if (WaitForSingleObject(process.Handle(), 0) == WAIT_OBJECT_0) { | 241 if (WaitForSingleObject(process.Handle(), 0) == WAIT_OBJECT_0) { |
247 return; | 242 return; |
248 } | 243 } |
249 | 244 |
250 MessageLoop::current()->PostDelayedTask( | 245 MessageLoop::current()->PostDelayedTask( |
251 FROM_HERE, | 246 FROM_HERE, |
252 base::Bind(&TimerExpiredTask::TimedOut, | 247 Bind(&TimerExpiredTask::TimedOut, |
253 base::Owned(new TimerExpiredTask(process.Pass()))), | 248 Owned(new TimerExpiredTask(process.Pass()))), |
254 base::TimeDelta::FromMilliseconds(kWaitInterval)); | 249 TimeDelta::FromMilliseconds(kWaitInterval)); |
255 } | 250 } |
256 | 251 |
257 } // namespace base | 252 } // namespace base |
OLD | NEW |