1 files changed, 96 insertions, 0 deletions
diff --git a/chrome/test/base/chrome_process_util.h b/chrome/test/base/chrome_process_util.h
new file mode 100644
index 0000000..df82abd
--- /dev/null
+++ b/chrome/test/base/chrome_process_util.h
@@ -0,0 +1,96 @@
+// Copyright (c) 2011 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_BASE_CHROME_PROCESS_UTIL_H_
+#define CHROME_TEST_BASE_CHROME_PROCESS_UTIL_H_
+#pragma once
+
+#include <vector>
+
+#include "base/memory/scoped_ptr.h"
+#include "base/process_util.h"
+
+typedef std::vector<base::ProcessId> ChromeProcessList;
+
+// Returns the executable name of the current Chrome browser process.
+const FilePath::CharType* GetRunningBrowserExecutableName();
+
+// Returns the executable name of the current Chrome helper process.
+const FilePath::CharType* GetRunningHelperExecutableName();
+
+// 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_;
+
+  ~ChromeTestProcessMetrics();
+
+ private:
+  explicit ChromeTestProcessMetrics(base::ProcessHandle 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_BASE_CHROME_PROCESS_UTIL_H_