blob: 210c3bd5f6bb775659e677ceb29dad2ed67d22ae (
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
94
95
96
97
98
|
// Copyright (c) 2010 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.
#ifndef CHROME_TEST_CHROME_PROCESS_UTIL_H_
#define CHROME_TEST_CHROME_PROCESS_UTIL_H_
#pragma once
#include <vector>
#include "base/file_path.h"
#include "base/process_util.h"
#include "base/scoped_ptr.h"
typedef std::vector<base::ProcessId> ChromeProcessList;
// Returns a vector of PIDs of all chrome processes (main and renderers etc)
// based on |browser_pid|, the PID of the main browser process.
ChromeProcessList GetRunningChromeProcesses(base::ProcessId browser_pid);
// Attempts to terminate all chrome processes launched by (and including)
// |browser_pid|.
void TerminateAllChromeProcesses(base::ProcessId browser_pid);
// A wrapper class for tests to use in fetching process metrics.
// Delegates everything we need to base::ProcessMetrics, except
// memory stats on Mac (which have to parse ps output due to privilege
// restrictions, behavior we don't want in base). Long-term, if
// the production base::ProcessMetrics gets updated to return
// acceptable metrics on Mac, this class should disappear.
class ChromeTestProcessMetrics {
public:
static ChromeTestProcessMetrics* CreateProcessMetrics(
base::ProcessHandle process) {
return new ChromeTestProcessMetrics(process);
}
size_t GetPagefileUsage();
size_t GetWorkingSetSize();
size_t GetPeakPagefileUsage() {
return process_metrics_->GetPeakPagefileUsage();
}
size_t GetPeakWorkingSetSize() {
return process_metrics_->GetPeakWorkingSetSize();
}
bool GetIOCounters(base::IoCounters* io_counters) {
return process_metrics_->GetIOCounters(io_counters);
}
base::ProcessHandle process_handle_;
private:
explicit ChromeTestProcessMetrics(base::ProcessHandle process) {
#if !defined(OS_MACOSX)
process_metrics_.reset(
base::ProcessMetrics::CreateProcessMetrics(process));
#else
process_metrics_.reset(
base::ProcessMetrics::CreateProcessMetrics(process, NULL));
#endif
process_handle_ = process;
}
scoped_ptr<base::ProcessMetrics> process_metrics_;
DISALLOW_COPY_AND_ASSIGN(ChromeTestProcessMetrics);
};
#if defined(OS_MACOSX)
// These types and API are here to fetch the information about a set of running
// processes by ID on the Mac. There are also APIs in base, but fetching the
// information for another process requires privileges that a normal executable
// does not have. This API fetches the data by spawning ps (which is setuid so
// it has the needed privileges) and processing its output. The API is provided
// here because we don't want code spawning processes like this in base, where
// someone writing cross platform code might use it without realizing that it's
// a heavyweight call on the Mac.
struct MacChromeProcessInfo {
base::ProcessId pid;
int rsz_in_kb;
int vsz_in_kb;
};
typedef std::vector<MacChromeProcessInfo> MacChromeProcessInfoList;
// Any ProcessId that info can't be found for will be left out.
MacChromeProcessInfoList GetRunningMacProcessInfo(
const ChromeProcessList& process_list);
#endif // defined(OS_MACOSX)
#endif // CHROME_TEST_CHROME_PROCESS_UTIL_H_
|
