15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/test/base/chrome_process_util.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h" 11bbcdd45c55eb7c4641ab97aef9889b0fc828e7d3Ben Murdoch#include "base/process/launch.h" 122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/strings/string_number_conversions.h" 132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/strings/string_split.h" 14868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h" 15868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)MacChromeProcessInfoList GetRunningMacProcessInfo( 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const ChromeProcessList& process_list) { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MacChromeProcessInfoList result; 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Build up the ps command line 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<std::string> cmdline; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cmdline.push_back("ps"); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cmdline.push_back("-o"); 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cmdline.push_back("pid=,rss=,vsz="); // fields we need, no headings 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ChromeProcessList::const_iterator process_iter; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (process_iter = process_list.begin(); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) process_iter != process_list.end(); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++process_iter) { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cmdline.push_back("-p"); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cmdline.push_back(base::StringPrintf("%d", *process_iter)); 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Invoke it 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string ps_output; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!base::GetAppOutput(CommandLine(cmdline), &ps_output)) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return result; // All the pids might have exited 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Process the results 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<std::string> ps_output_lines; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::SplitString(ps_output, '\n', &ps_output_lines); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<std::string>::const_iterator line_iter; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (line_iter = ps_output_lines.begin(); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) line_iter != ps_output_lines.end(); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++line_iter) { 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string line(CollapseWhitespaceASCII(*line_iter, false)); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<std::string> values; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::SplitString(line, ' ', &values); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (values.size() == 3) { 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MacChromeProcessInfo proc_info; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pid; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::StringToInt(values[0], &pid); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) proc_info.pid = pid; 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::StringToInt(values[1], &proc_info.rsz_in_kb); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::StringToInt(values[2], &proc_info.vsz_in_kb); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (proc_info.pid && proc_info.rsz_in_kb && proc_info.vsz_in_kb) 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result.push_back(proc_info); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return result; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Common interface for fetching memory values from parsed ps output. 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// We fill in both values we may get called for, even though our 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// callers typically only care about one, just to keep the code 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// simple and because this is a test. 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static bool GetMemoryValuesHack(uint32 process_id, 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t* virtual_size, 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t* working_set_size) { 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(virtual_size && working_set_size); 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<base::ProcessId> processes; 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) processes.push_back(process_id); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MacChromeProcessInfoList process_info = GetRunningMacProcessInfo(processes); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (process_info.empty()) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool found_process = false; 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *virtual_size = 0; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *working_set_size = 0; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MacChromeProcessInfoList::iterator it = process_info.begin(); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (; it != process_info.end(); ++it) { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it->pid != static_cast<base::ProcessId>(process_id)) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) continue; 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) found_process = true; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *virtual_size = it->vsz_in_kb * 1024; 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *working_set_size = it->rsz_in_kb * 1024; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return found_process; 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)size_t ChromeTestProcessMetrics::GetPagefileUsage() { 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t virtual_size; 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t working_set_size; 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetMemoryValuesHack(process_handle_, &virtual_size, &working_set_size); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return virtual_size; 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)size_t ChromeTestProcessMetrics::GetWorkingSetSize() { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t virtual_size; 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t working_set_size; 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetMemoryValuesHack(process_handle_, &virtual_size, &working_set_size); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return working_set_size; 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 110