1 files changed, 108 insertions, 0 deletions
diff --git a/chrome/test/base/chrome_process_util_mac.cc b/chrome/test/base/chrome_process_util_mac.cc
new file mode 100644
index 0000000..e935ad5
--- /dev/null
+++ b/chrome/test/base/chrome_process_util_mac.cc
@@ -0,0 +1,108 @@
+// 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.
+
+#include "chrome/test/base/chrome_process_util.h"
+
+#include <string>
+#include <vector>
+
+#include "base/command_line.h"
+#include "base/process_util.h"
+#include "base/string_number_conversions.h"
+#include "base/string_split.h"
+#include "base/string_util.h"
+
+MacChromeProcessInfoList GetRunningMacProcessInfo(
+    const ChromeProcessList &process_list) {
+  MacChromeProcessInfoList result;
+
+  // Build up the ps command line
+  std::vector<std::string> cmdline;
+  cmdline.push_back("ps");
+  cmdline.push_back("-o");
+  cmdline.push_back("pid=,rsz=,vsz=");  // fields we need, no headings
+  ChromeProcessList::const_iterator process_iter;
+  for (process_iter = process_list.begin();
+       process_iter != process_list.end();
+       ++process_iter) {
+    cmdline.push_back("-p");
+    cmdline.push_back(StringPrintf("%d", *process_iter));
+  }
+
+  // Invoke it
+  std::string ps_output;
+  if (!base::GetAppOutput(CommandLine(cmdline), &ps_output))
+    return result;  // All the pids might have exited
+
+  // Process the results
+  std::vector<std::string> ps_output_lines;
+  base::SplitString(ps_output, '\n', &ps_output_lines);
+  std::vector<std::string>::const_iterator line_iter;
+  for (line_iter = ps_output_lines.begin();
+       line_iter != ps_output_lines.end();
+       ++line_iter) {
+    std::string line(CollapseWhitespaceASCII(*line_iter, false));
+    std::vector<std::string> values;
+    base::SplitString(line, ' ', &values);
+    if (values.size() == 3) {
+      MacChromeProcessInfo proc_info;
+      int pid;
+      base::StringToInt(values[0], &pid);
+      proc_info.pid = pid;
+      base::StringToInt(values[1], &proc_info.rsz_in_kb);
+      base::StringToInt(values[2], &proc_info.vsz_in_kb);
+      if (proc_info.pid && proc_info.rsz_in_kb && proc_info.vsz_in_kb)
+        result.push_back(proc_info);
+    }
+  }
+
+  return result;
+}
+
+// Common interface for fetching memory values from parsed ps output.
+// We fill in both values we may get called for, even though our
+// callers typically only care about one, just to keep the code
+// simple and because this is a test.
+static bool GetMemoryValuesHack(uint32 process_id,
+                          size_t* virtual_size,
+                          size_t* working_set_size) {
+  DCHECK(virtual_size && working_set_size);
+
+  std::vector<base::ProcessId> processes;
+  processes.push_back(process_id);
+
+  MacChromeProcessInfoList process_info = GetRunningMacProcessInfo(processes);
+  if (process_info.empty())
+    return false;
+
+  bool found_process = false;
+  *virtual_size = 0;
+  *working_set_size = 0;
+
+  MacChromeProcessInfoList::iterator it = process_info.begin();
+  for (; it != process_info.end(); ++it) {
+    if (it->pid != static_cast<base::ProcessId>(process_id))
+      continue;
+    found_process = true;
+    *virtual_size = it->vsz_in_kb * 1024;
+    *working_set_size = it->rsz_in_kb * 1024;
+    break;
+  }
+
+  return found_process;
+}
+
+size_t ChromeTestProcessMetrics::GetPagefileUsage() {
+  size_t virtual_size;
+  size_t working_set_size;
+  GetMemoryValuesHack(process_handle_, &virtual_size, &working_set_size);
+  return virtual_size;
+}
+
+size_t ChromeTestProcessMetrics::GetWorkingSetSize() {
+  size_t virtual_size;
+  size_t working_set_size;
+  GetMemoryValuesHack(process_handle_, &virtual_size, &working_set_size);
+  return working_set_size;
+}