From c407dc5cd9bdc5668497f21b26b09d988ab439de Mon Sep 17 00:00:00 2001 From: Ben Murdoch Date: Thu, 29 Jul 2010 17:14:53 +0100 Subject: Merge Chromium src@r53293 Change-Id: Ia79acf8670f385cee48c45b0a75371d8e950af34 --- chrome/browser/process_info_snapshot_mac.cc | 168 ++++++++++++++++++++++++++++ 1 file changed, 168 insertions(+) create mode 100644 chrome/browser/process_info_snapshot_mac.cc (limited to 'chrome/browser/process_info_snapshot_mac.cc') diff --git a/chrome/browser/process_info_snapshot_mac.cc b/chrome/browser/process_info_snapshot_mac.cc new file mode 100644 index 0000000..732f22e --- /dev/null +++ b/chrome/browser/process_info_snapshot_mac.cc @@ -0,0 +1,168 @@ +// 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/browser/process_info_snapshot.h" + +#include + +#include "base/logging.h" +#include "base/string_util.h" +#include "base/thread.h" + +// Implementation for the Mac; calls '/bin/ps' for information when +// |Sample()| is called. + +// Default constructor. +ProcessInfoSnapshot::ProcessInfoSnapshot() { } + +// Destructor: just call |Reset()| to release everything. +ProcessInfoSnapshot::~ProcessInfoSnapshot() { + Reset(); +} + +const size_t ProcessInfoSnapshot::kMaxPidListSize = 1000; + +// Capture the information by calling '/bin/ps'. +// Note: we ignore the "tsiz" (text size) display option of ps because it's +// always zero (tested on 10.5 and 10.6). +bool ProcessInfoSnapshot::Sample(std::vector pid_list) { + const char* kPsPathName = "/bin/ps"; + Reset(); + + // Nothing to do if no PIDs given. + if (pid_list.size() == 0) + return true; + if (pid_list.size() > kMaxPidListSize) { + // The spec says |pid_list| *must* not have more than this many entries. + NOTREACHED(); + return false; + } + + std::vector argv; + argv.push_back(kPsPathName); + // Get PID, PPID, (real) UID, effective UID, resident set size, virtual memory + // size, and command. + argv.push_back("-o"); + argv.push_back("pid=,ppid=,ruid=,uid=,rss=,vsz=,comm="); + // Only display the specified PIDs. + for (std::vector::iterator it = pid_list.begin(); + it != pid_list.end(); ++it) { + argv.push_back("-p"); + argv.push_back(Int64ToString(static_cast(*it))); + } + + std::string output; + CommandLine command_line(argv); + // Limit output read to a megabyte for safety. + if (!base::GetAppOutputRestricted(command_line, &output, 1024 * 1024)) { + LOG(ERROR) << "Failure running " << kPsPathName << " to acquire data."; + return false; + } + + std::istringstream in(output, std::istringstream::in); + std::string line; + + // Process lines until done. + while (true) { + ProcInfoEntry proc_info; + + // The format is as specified above to ps (see ps(1)): + // "-o pid=,ppid=,ruid=,uid=,rss=,vsz=,comm=". + // Try to read the PID; if we get it, we should be able to get the rest of + // the line. + in >> proc_info.pid; + if (in.eof()) + break; + in >> proc_info.ppid; + in >> proc_info.uid; + in >> proc_info.euid; + in >> proc_info.rss; + in >> proc_info.vsize; + in.ignore(1, ' '); // Eat the space. + std::getline(in, proc_info.command); // Get the rest of the line. + if (!in.good()) { + LOG(ERROR) << "Error parsing output from " << kPsPathName << "."; + return false; + } + + // Make sure the new PID isn't already in our list. + if (proc_info_entries_.find(proc_info.pid) != proc_info_entries_.end()) { + LOG(ERROR) << "Duplicate PID in output from " << kPsPathName << "."; + return false; + } + + if (!proc_info.pid || ! proc_info.vsize) { + LOG(WARNING) << "Invalid data from " << kPsPathName << "."; + return false; + } + + // Record the process information. + proc_info_entries_[proc_info.pid] = proc_info; + } + + return true; +} + +// Clear all the stored information. +void ProcessInfoSnapshot::Reset() { + proc_info_entries_.clear(); +} + +bool ProcessInfoSnapshot::GetProcInfo(int pid, + ProcInfoEntry* proc_info) const { + std::map::const_iterator it = proc_info_entries_.find(pid); + if (it == proc_info_entries_.end()) + return false; + + *proc_info = it->second; + return true; +} + +bool ProcessInfoSnapshot::GetCommittedKBytesOfPID( + int pid, + base::CommittedKBytes* usage) const { + // Try to avoid crashing on a bug; stats aren't usually so crucial. + if (!usage) { + NOTREACHED(); + return false; + } + + // Failure of |GetProcInfo()| is "normal", due to racing. + ProcInfoEntry proc_info; + if (!GetProcInfo(pid, &proc_info)) { + usage->priv = 0; + usage->mapped = 0; + usage->image = 0; + return false; + } + + usage->priv = proc_info.vsize; + usage->mapped = 0; + usage->image = 0; + return true; +} + +bool ProcessInfoSnapshot::GetWorkingSetKBytesOfPID( + int pid, + base::WorkingSetKBytes* ws_usage) const { + // Try to avoid crashing on a bug; stats aren't usually so crucial. + if (!ws_usage) { + NOTREACHED(); + return false; + } + + // Failure of |GetProcInfo()| is "normal", due to racing. + ProcInfoEntry proc_info; + if (!GetProcInfo(pid, &proc_info)) { + ws_usage->priv = 0; + ws_usage->shareable = 0; + ws_usage->shared = 0; + return false; + } + + ws_usage->priv = 0; + ws_usage->shareable = proc_info.rss; + ws_usage->shared = 0; + return true; +} -- cgit v1.1