More memory stats code cleanup:...

chrome/test/perf/mem_usage_win.cc

-// Copyright (c) 2006-2008 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/perf/mem_usage.h"
-
-#include <windows.h>
-#include <psapi.h>
-
-#include "base/file_path.h"
-#include "base/path_service.h"
-#include "base/process_util.h"
-#include "base/scoped_ptr.h"
-#include "chrome/common/chrome_constants.h"
-#include "chrome/common/chrome_paths.h"
-#include "chrome/test/chrome_process_util.h"
-
-// GetPerformanceInfo is not available on WIN2K.  So we'll
-// load it on-the-fly.
-const wchar_t kPsapiDllName[] = L"psapi.dll";
-typedef BOOL (WINAPI *GetPerformanceInfoFunction) (
-    PPERFORMANCE_INFORMATION pPerformanceInformation,
-    DWORD cb);
-
-static BOOL InternalGetPerformanceInfo(
-    PPERFORMANCE_INFORMATION pPerformanceInformation, DWORD cb) {
-  static GetPerformanceInfoFunction GetPerformanceInfo_func = NULL;
-  if (!GetPerformanceInfo_func) {
-    HMODULE psapi_dll = ::GetModuleHandle(kPsapiDllName);
-    if (psapi_dll)
-      GetPerformanceInfo_func = reinterpret_cast<GetPerformanceInfoFunction>(
-          GetProcAddress(psapi_dll, "GetPerformanceInfo"));
-
-    if (!GetPerformanceInfo_func) {
-      // The function could be loaded!
-      memset(pPerformanceInformation, 0, cb);
-      return FALSE;
-    }
-  }
-  return GetPerformanceInfo_func(pPerformanceInformation, cb);
-}
-
-
-size_t GetSystemCommitCharge() {
-  // Get the System Page Size.
-  SYSTEM_INFO system_info;
-  GetSystemInfo(&system_info);
-
-  PERFORMANCE_INFORMATION info;
-  if (InternalGetPerformanceInfo(&info, sizeof(info)))
-    return info.CommitTotal * system_info.dwPageSize;
-  return -1;
-}
-
-void PrintChromeMemoryUsageInfo() {
-  printf("\n");
-
-  FilePath data_dir;
-  PathService::Get(chrome::DIR_USER_DATA, &data_dir);
-  int browser_process_pid = ChromeBrowserProcessId(data_dir);
-  ChromeProcessList chrome_processes(GetRunningChromeProcesses(data_dir));
-
-  ChromeProcessList::const_iterator it;
-  for (it = chrome_processes.begin(); it != chrome_processes.end(); ++it) {
-    base::ProcessHandle process_handle;
-    if (!base::OpenPrivilegedProcessHandle(*it, &process_handle)) {
-      NOTREACHED();
-    }
-
-    // TODO(sgk):  if/when base::ProcessMetrics can return real memory
-    // stats on mac, convert to:
-    //
-    // scoped_ptr<base::ProcessMetrics> process_metrics;
-    // process_metrics.reset(
-    //     base::ProcessMetrics::CreateProcessMetrics(process_handle));
-    scoped_ptr<ChromeTestProcessMetrics> process_metrics;
-    process_metrics.reset(
-        ChromeTestProcessMetrics::CreateProcessMetrics(process_handle));
-
-    size_t peak_virtual_size = process_metrics->GetPeakPagefileUsage();
-    size_t current_virtual_size = process_metrics->GetPagefileUsage();
-    size_t peak_working_set_size = process_metrics->GetPeakWorkingSetSize();
-    size_t current_working_set_size = process_metrics->GetWorkingSetSize();
-
-    if (*it == browser_process_pid) {
-      wprintf(L"browser_vm_peak = %d\n", peak_virtual_size);
-      wprintf(L"browser_vm_current = %d\n", current_virtual_size);
-      wprintf(L"browser_ws_peak = %d\n", peak_working_set_size);
-      wprintf(L"browser_ws_final = %d\n", current_working_set_size);
-    } else {
-      wprintf(L"render_vm_peak = %d\n", peak_virtual_size);
-      wprintf(L"render_vm_current = %d\n", current_virtual_size);
-      wprintf(L"render_ws_peak = %d\n", peak_working_set_size);
-      wprintf(L"render_ws_final = %d\n", current_working_set_size);
-    }
-  };
-}