| 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 182 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 193 if (!::GetExitCodeProcess(handle, &temp_code)) | 193 if (!::GetExitCodeProcess(handle, &temp_code)) |
| 194 return false; | 194 return false; |
| 195 | 195 |
| 196 *exit_code = temp_code; | 196 *exit_code = temp_code; |
| 197 return true; | 197 return true; |
| 198 } | 198 } |
| 199 | 199 |
| 200 bool WaitForProcessesToExit(const FilePath::StringType& executable_name, | 200 bool WaitForProcessesToExit(const FilePath::StringType& executable_name, |
| 201 base::TimeDelta wait, | 201 base::TimeDelta wait, |
| 202 const ProcessFilter* filter) { | 202 const ProcessFilter* filter) { |
| 203 const ProcessEntry* entry; | |
| 204 bool result = true; | 203 bool result = true; |
| 205 DWORD start_time = GetTickCount(); | 204 DWORD start_time = GetTickCount(); |
| 206 | 205 |
| 207 NamedProcessIterator iter(executable_name, filter); | 206 NamedProcessIterator iter(executable_name, filter); |
| 208 while ((entry = iter.NextProcessEntry())) { | 207 for (const ProcessEntry* entry = iter.NextProcessEntry(); entry; |
| 208 entry = iter.NextProcessEntry()) { |
| 209 DWORD remaining_wait = std::max<int64>( | 209 DWORD remaining_wait = std::max<int64>( |
| 210 0, wait.InMilliseconds() - (GetTickCount() - start_time)); | 210 0, wait.InMilliseconds() - (GetTickCount() - start_time)); |
| 211 HANDLE process = OpenProcess(SYNCHRONIZE, | 211 HANDLE process = OpenProcess(SYNCHRONIZE, |
| 212 FALSE, | 212 FALSE, |
| 213 entry->th32ProcessID); | 213 entry->th32ProcessID); |
| 214 DWORD wait_result = WaitForSingleObject(process, remaining_wait); | 214 DWORD wait_result = WaitForSingleObject(process, remaining_wait); |
| 215 CloseHandle(process); | 215 CloseHandle(process); |
| 216 result = result && (wait_result == WAIT_OBJECT_0); | 216 result &= (wait_result == WAIT_OBJECT_0); |
| 217 } | 217 } |
| 218 | 218 |
| 219 return result; | 219 return result; |
| 220 } | 220 } |
| 221 | 221 |
| 222 bool WaitForSingleProcess(ProcessHandle handle, base::TimeDelta wait) { | 222 bool WaitForSingleProcess(ProcessHandle handle, base::TimeDelta wait) { |
| 223 int exit_code; | 223 int exit_code; |
| 224 if (!WaitForExitCodeWithTimeout(handle, &exit_code, wait)) | 224 return WaitForExitCodeWithTimeout(handle, &exit_code, wait) && exit_code == 0; |
| 225 return false; | |
| 226 return exit_code == 0; | |
| 227 } | 225 } |
| 228 | 226 |
| 229 bool CleanupProcesses(const FilePath::StringType& executable_name, | 227 bool CleanupProcesses(const FilePath::StringType& executable_name, |
| 230 base::TimeDelta wait, | 228 base::TimeDelta wait, |
| 231 int exit_code, | 229 int exit_code, |
| 232 const ProcessFilter* filter) { | 230 const ProcessFilter* filter) { |
| 233 bool exited_cleanly = WaitForProcessesToExit(executable_name, wait, filter); | 231 if (WaitForProcessesToExit(executable_name, wait, filter)) |
| 234 if (!exited_cleanly) | 232 return true; |
| 235 KillProcesses(executable_name, exit_code, filter); | 233 KillProcesses(executable_name, exit_code, filter); |
| 236 return exited_cleanly; | 234 return false; |
| 237 } | 235 } |
| 238 | 236 |
| 239 void EnsureProcessTerminated(ProcessHandle process) { | 237 void EnsureProcessTerminated(ProcessHandle process) { |
| 240 DCHECK(process != GetCurrentProcess()); | 238 DCHECK(process != GetCurrentProcess()); |
| 241 | 239 |
| 242 // If already signaled, then we are done! | 240 // If already signaled, then we are done! |
| 243 if (WaitForSingleObject(process, 0) == WAIT_OBJECT_0) { | 241 if (WaitForSingleObject(process, 0) == WAIT_OBJECT_0) { |
| 244 CloseHandle(process); | 242 CloseHandle(process); |
| 245 return; | 243 return; |
| 246 } | 244 } |
| 247 | 245 |
| 248 MessageLoop::current()->PostDelayedTask( | 246 MessageLoop::current()->PostDelayedTask( |
| 249 FROM_HERE, | 247 FROM_HERE, |
| 250 base::Bind(&TimerExpiredTask::TimedOut, | 248 base::Bind(&TimerExpiredTask::TimedOut, |
| 251 base::Owned(new TimerExpiredTask(process))), | 249 base::Owned(new TimerExpiredTask(process))), |
| 252 base::TimeDelta::FromMilliseconds(kWaitInterval)); | 250 base::TimeDelta::FromMilliseconds(kWaitInterval)); |
| 253 } | 251 } |
| 254 | 252 |
| 255 } // namespace base | 253 } // namespace base |
| OLD | NEW |