chrome/test/base/chrome_process_util.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/test/base/chrome_process_util.h"
   6
   7 #include <set>
   8 #include <string>
   9 #include <vector>
  10
  11 #include "base/command_line.h"
  12 #include "base/process/process.h"
  13 #include "base/process/process_iterator.h"
  14 #include "base/time/time.h"
  15 #include "chrome/common/chrome_constants.h"
  16 #include "chrome/test/base/test_switches.h"
  17 #include "content/public/common/result_codes.h"
  18
  19 using base::TimeDelta;
  20 using base::TimeTicks;
  21
  22 void TerminateAllChromeProcesses(const ChromeProcessList& process_pids) {
  23   ChromeProcessList::const_iterator it;
  24   for (it = process_pids.begin(); it != process_pids.end(); ++it) {
  25     base::Process process = base::Process::Open(*it);
  26     if (process.IsValid()) {
  27       // Ignore processes for which we can't open the handle. We don't
  28       // guarantee that all processes will terminate, only try to do so.
  29       process.Terminate(content::RESULT_CODE_KILLED, true);
  30     }
  31   }
  32 }
  33
  34 class ChildProcessFilter : public base::ProcessFilter {
  35  public:
  36   explicit ChildProcessFilter(base::ProcessId parent_pid)
  37       : parent_pids_(&parent_pid, (&parent_pid) + 1) {}
  38
  39   explicit ChildProcessFilter(const std::vector<base::ProcessId>& parent_pids)
  40       : parent_pids_(parent_pids.begin(), parent_pids.end()) {}
  41
  42   bool Includes(const base::ProcessEntry& entry) const override {
  43     return parent_pids_.find(entry.parent_pid()) != parent_pids_.end();
  44   }
  45
  46  private:
  47   const std::set<base::ProcessId> parent_pids_;
  48
  49   DISALLOW_COPY_AND_ASSIGN(ChildProcessFilter);
  50 };
  51
  52 ChromeProcessList GetRunningChromeProcesses(base::ProcessId browser_pid) {
  53   const base::FilePath::CharType* executable_name =
  54       chrome::kBrowserProcessExecutableName;
  55   ChromeProcessList result;
  56   if (browser_pid == static_cast<base::ProcessId>(-1))
  57     return result;
  58
  59   ChildProcessFilter filter(browser_pid);
  60   base::NamedProcessIterator it(executable_name, &filter);
  61   while (const base::ProcessEntry* process_entry = it.NextProcessEntry()) {
  62     result.push_back(process_entry->pid());
  63   }
  64
  65 #if defined(OS_POSIX) && !defined(OS_MACOSX)
  66   // On Unix we might be running with a zygote process for the renderers.
  67   // Because of that we sweep the list of processes again and pick those which
  68   // are children of one of the processes that we've already seen.
  69   {
  70     ChildProcessFilter filter(result);
  71     base::NamedProcessIterator it(executable_name, &filter);
  72     while (const base::ProcessEntry* process_entry = it.NextProcessEntry())
  73       result.push_back(process_entry->pid());
  74   }
  75 #endif  // defined(OS_POSIX) && !defined(OS_MACOSX)
  76
  77 #if defined(OS_POSIX)
  78   // On Mac OS X we run the subprocesses with a different bundle, and
  79   // on Linux via /proc/self/exe, so they end up with a different
  80   // name.  We must collect them in a second pass.
  81   {
  82     base::FilePath::StringType name = chrome::kHelperProcessExecutableName;
  83     ChildProcessFilter filter(browser_pid);
  84     base::NamedProcessIterator it(name, &filter);
  85     while (const base::ProcessEntry* process_entry = it.NextProcessEntry())
  86       result.push_back(process_entry->pid());
  87   }
  88 #endif  // defined(OS_POSIX)
  89
  90   result.push_back(browser_pid);
  91
  92   return result;
  93 }
  94
  95 #if !defined(OS_MACOSX)
  96
  97 size_t ChromeTestProcessMetrics::GetPagefileUsage() {
  98   return process_metrics_->GetPagefileUsage();
  99 }
 100
 101 size_t ChromeTestProcessMetrics::GetWorkingSetSize() {
 102   return process_metrics_->GetWorkingSetSize();
 103 }
 104
 105 #endif  // !defined(OS_MACOSX)
 106
 107 ChromeTestProcessMetrics::~ChromeTestProcessMetrics() {}
 108
 109 ChromeTestProcessMetrics::ChromeTestProcessMetrics(
 110     base::ProcessHandle process) {
 111 #if !defined(OS_MACOSX)
 112   process_metrics_.reset(
 113       base::ProcessMetrics::CreateProcessMetrics(process));
 114 #else
 115   process_metrics_.reset(
 116       base::ProcessMetrics::CreateProcessMetrics(process, NULL));
 117 #endif
 118   process_handle_ = process;
 119 }