blob: 461194f1c04c272bdf87aba36fbda914016e96bc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/test/chrome_process_util.h"
#include <vector>
#include "base/process_util.h"
#include "base/time.h"
#include "chrome/common/chrome_constants.h"
#include "chrome/common/result_codes.h"
using base::Time;
using base::TimeDelta;
namespace {
class ChromeProcessFilter : public base::ProcessFilter {
public:
explicit ChromeProcessFilter(base::ProcessId browser_pid)
: browser_pid_(browser_pid) {}
virtual bool Includes(base::ProcessId pid, base::ProcessId parent_pid) const {
// Match browser process itself and its children.
return browser_pid_ == pid || browser_pid_ == parent_pid;
}
private:
base::ProcessId browser_pid_;
DISALLOW_COPY_AND_ASSIGN(ChromeProcessFilter);
};
} // namespace
void TerminateAllChromeProcesses(const FilePath& data_dir) {
// Total time the function will wait for chrome processes
// to terminate after it told them to do so.
const TimeDelta kExitTimeout = TimeDelta::FromMilliseconds(5000);
ChromeProcessList process_pids(GetRunningChromeProcesses(data_dir));
std::vector<base::ProcessHandle> handles;
{
ChromeProcessList::const_iterator it;
for (it = process_pids.begin(); it != process_pids.end(); ++it) {
base::ProcessHandle handle;
// Ignore processes for which we can't open the handle. We don't guarantee
// that all processes will terminate, only try to do so.
if (base::OpenProcessHandle(*it, &handle))
handles.push_back(handle);
}
}
std::vector<base::ProcessHandle>::const_iterator it;
for (it = handles.begin(); it != handles.end(); ++it)
base::KillProcess(*it, ResultCodes::TASKMAN_KILL, false);
const Time start = Time::Now();
for (it = handles.begin();
it != handles.end() && Time::Now() - start < kExitTimeout;
++it) {
// TODO(phajdan.jr): Fix int/int64 problems with TimeDelta::InMilliseconds.
int wait_time_ms = static_cast<int>((Time::Now() - start).InMilliseconds());
base::WaitForSingleProcess(*it, wait_time_ms);
}
for (it = handles.begin(); it != handles.end(); ++it)
base::CloseProcessHandle(*it);
}
ChromeProcessList GetRunningChromeProcesses(const FilePath& data_dir) {
ChromeProcessList result;
base::ProcessId browser_pid = ChromeBrowserProcessId(data_dir);
if (browser_pid < 0)
return result;
ChromeProcessFilter filter(browser_pid);
base::NamedProcessIterator it(chrome::kBrowserProcessExecutableName, &filter);
const ProcessEntry* process_entry;
while ((process_entry = it.NextProcessEntry())) {
#if defined(OS_WIN)
result.push_back(process_entry->th32ProcessID);
#elif defined(OS_POSIX)
result.push_back(process_entry->pid);
#endif
}
return result;
}
|
