mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-05 20:15:58 +00:00
648 lines
21 KiB
C++
648 lines
21 KiB
C++
/* -*- Mode: C++; indent-tabs-mode: nil; c-basic-offset: 4 -*-
|
|
*
|
|
* ***** BEGIN LICENSE BLOCK *****
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
*
|
|
* The contents of this file are subject to the Mozilla Public License Version
|
|
* 1.1 (the "License"); you may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
* http://www.mozilla.org/MPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS IS" basis,
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
* for the specific language governing rights and limitations under the
|
|
* License.
|
|
*
|
|
* The Original Code is wm.cpp, released
|
|
* November 15, 2000.
|
|
*
|
|
* The Initial Developer of the Original Code is
|
|
* Netscape Communications Corporation.
|
|
* Portions created by the Initial Developer are Copyright (C) 2000
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
* Chris Waterson <waterson@netscape.com>
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
/*
|
|
* This program tracks a process's working memory usage using the
|
|
* ``performance'' entries in the Win32 registry. It borrows from
|
|
* the ``pviewer'' source code in the MS SDK.
|
|
*/
|
|
|
|
#include <assert.h>
|
|
#include <windows.h>
|
|
#include <winperf.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#define PN_PROCESS 1
|
|
#define PN_PROCESS_CPU 2
|
|
#define PN_PROCESS_PRIV 3
|
|
#define PN_PROCESS_USER 4
|
|
#define PN_PROCESS_WORKING_SET 5
|
|
#define PN_PROCESS_PEAK_WS 6
|
|
#define PN_PROCESS_PRIO 7
|
|
#define PN_PROCESS_ELAPSE 8
|
|
#define PN_PROCESS_ID 9
|
|
#define PN_PROCESS_PRIVATE_PAGE 10
|
|
#define PN_PROCESS_VIRTUAL_SIZE 11
|
|
#define PN_PROCESS_PEAK_VS 12
|
|
#define PN_PROCESS_FAULT_COUNT 13
|
|
#define PN_THREAD 14
|
|
#define PN_THREAD_CPU 15
|
|
#define PN_THREAD_PRIV 16
|
|
#define PN_THREAD_USER 17
|
|
#define PN_THREAD_START 18
|
|
#define PN_THREAD_SWITCHES 19
|
|
#define PN_THREAD_PRIO 20
|
|
#define PN_THREAD_BASE_PRIO 21
|
|
#define PN_THREAD_ELAPSE 22
|
|
#define PN_THREAD_DETAILS 23
|
|
#define PN_THREAD_PC 24
|
|
#define PN_IMAGE 25
|
|
#define PN_IMAGE_NOACCESS 26
|
|
#define PN_IMAGE_READONLY 27
|
|
#define PN_IMAGE_READWRITE 28
|
|
#define PN_IMAGE_WRITECOPY 29
|
|
#define PN_IMAGE_EXECUTABLE 30
|
|
#define PN_IMAGE_EXE_READONLY 31
|
|
#define PN_IMAGE_EXE_READWRITE 32
|
|
#define PN_IMAGE_EXE_WRITECOPY 33
|
|
#define PN_PROCESS_ADDRESS_SPACE 34
|
|
#define PN_PROCESS_PRIVATE_NOACCESS 35
|
|
#define PN_PROCESS_PRIVATE_READONLY 36
|
|
#define PN_PROCESS_PRIVATE_READWRITE 37
|
|
#define PN_PROCESS_PRIVATE_WRITECOPY 38
|
|
#define PN_PROCESS_PRIVATE_EXECUTABLE 39
|
|
#define PN_PROCESS_PRIVATE_EXE_READONLY 40
|
|
#define PN_PROCESS_PRIVATE_EXE_READWRITE 41
|
|
#define PN_PROCESS_PRIVATE_EXE_WRITECOPY 42
|
|
#define PN_PROCESS_MAPPED_NOACCESS 43
|
|
#define PN_PROCESS_MAPPED_READONLY 44
|
|
#define PN_PROCESS_MAPPED_READWRITE 45
|
|
#define PN_PROCESS_MAPPED_WRITECOPY 46
|
|
#define PN_PROCESS_MAPPED_EXECUTABLE 47
|
|
#define PN_PROCESS_MAPPED_EXE_READONLY 48
|
|
#define PN_PROCESS_MAPPED_EXE_READWRITE 49
|
|
#define PN_PROCESS_MAPPED_EXE_WRITECOPY 50
|
|
#define PN_PROCESS_IMAGE_NOACCESS 51
|
|
#define PN_PROCESS_IMAGE_READONLY 52
|
|
#define PN_PROCESS_IMAGE_READWRITE 53
|
|
#define PN_PROCESS_IMAGE_WRITECOPY 54
|
|
#define PN_PROCESS_IMAGE_EXECUTABLE 55
|
|
#define PN_PROCESS_IMAGE_EXE_READONLY 56
|
|
#define PN_PROCESS_IMAGE_EXE_READWRITE 57
|
|
#define PN_PROCESS_IMAGE_EXE_WRITECOPY 58
|
|
|
|
struct entry_t {
|
|
int e_key;
|
|
int e_index;
|
|
char* e_title;
|
|
};
|
|
|
|
entry_t entries[] = {
|
|
{ PN_PROCESS, 0, TEXT("Process") },
|
|
{ PN_PROCESS_CPU, 0, TEXT("% Processor Time") },
|
|
{ PN_PROCESS_PRIV, 0, TEXT("% Privileged Time") },
|
|
{ PN_PROCESS_USER, 0, TEXT("% User Time") },
|
|
{ PN_PROCESS_WORKING_SET, 0, TEXT("Working Set") },
|
|
{ PN_PROCESS_PEAK_WS, 0, TEXT("Working Set Peak") },
|
|
{ PN_PROCESS_PRIO, 0, TEXT("Priority Base") },
|
|
{ PN_PROCESS_ELAPSE, 0, TEXT("Elapsed Time") },
|
|
{ PN_PROCESS_ID, 0, TEXT("ID Process") },
|
|
{ PN_PROCESS_PRIVATE_PAGE, 0, TEXT("Private Bytes") },
|
|
{ PN_PROCESS_VIRTUAL_SIZE, 0, TEXT("Virtual Bytes") },
|
|
{ PN_PROCESS_PEAK_VS, 0, TEXT("Virtual Bytes Peak") },
|
|
{ PN_PROCESS_FAULT_COUNT, 0, TEXT("Page Faults/sec") },
|
|
{ PN_THREAD, 0, TEXT("Thread") },
|
|
{ PN_THREAD_CPU, 0, TEXT("% Processor Time") },
|
|
{ PN_THREAD_PRIV, 0, TEXT("% Privileged Time") },
|
|
{ PN_THREAD_USER, 0, TEXT("% User Time") },
|
|
{ PN_THREAD_START, 0, TEXT("Start Address") },
|
|
{ PN_THREAD_SWITCHES, 0, TEXT("Con0, TEXT Switches/sec") },
|
|
{ PN_THREAD_PRIO, 0, TEXT("Priority Current") },
|
|
{ PN_THREAD_BASE_PRIO, 0, TEXT("Priority Base") },
|
|
{ PN_THREAD_ELAPSE, 0, TEXT("Elapsed Time") },
|
|
{ PN_THREAD_DETAILS, 0, TEXT("Thread Details") },
|
|
{ PN_THREAD_PC, 0, TEXT("User PC") },
|
|
{ PN_IMAGE, 0, TEXT("Image") },
|
|
{ PN_IMAGE_NOACCESS, 0, TEXT("No Access") },
|
|
{ PN_IMAGE_READONLY, 0, TEXT("Read Only") },
|
|
{ PN_IMAGE_READWRITE, 0, TEXT("Read/Write") },
|
|
{ PN_IMAGE_WRITECOPY, 0, TEXT("Write Copy") },
|
|
{ PN_IMAGE_EXECUTABLE, 0, TEXT("Executable") },
|
|
{ PN_IMAGE_EXE_READONLY, 0, TEXT("Exec Read Only") },
|
|
{ PN_IMAGE_EXE_READWRITE, 0, TEXT("Exec Read/Write") },
|
|
{ PN_IMAGE_EXE_WRITECOPY, 0, TEXT("Exec Write Copy") },
|
|
{ PN_PROCESS_ADDRESS_SPACE, 0, TEXT("Process Address Space") },
|
|
{ PN_PROCESS_PRIVATE_NOACCESS, 0, TEXT("Reserved Space No Access") },
|
|
{ PN_PROCESS_PRIVATE_READONLY, 0, TEXT("Reserved Space Read Only") },
|
|
{ PN_PROCESS_PRIVATE_READWRITE, 0, TEXT("Reserved Space Read/Write") },
|
|
{ PN_PROCESS_PRIVATE_WRITECOPY, 0, TEXT("Reserved Space Write Copy") },
|
|
{ PN_PROCESS_PRIVATE_EXECUTABLE, 0, TEXT("Reserved Space Executable") },
|
|
{ PN_PROCESS_PRIVATE_EXE_READONLY, 0, TEXT("Reserved Space Exec Read Only") },
|
|
{ PN_PROCESS_PRIVATE_EXE_READWRITE, 0, TEXT("Reserved Space Exec Read/Write") },
|
|
{ PN_PROCESS_PRIVATE_EXE_WRITECOPY, 0, TEXT("Reserved Space Exec Write Copy") },
|
|
{ PN_PROCESS_MAPPED_NOACCESS, 0, TEXT("Mapped Space No Access") },
|
|
{ PN_PROCESS_MAPPED_READONLY, 0, TEXT("Mapped Space Read Only") },
|
|
{ PN_PROCESS_MAPPED_READWRITE, 0, TEXT("Mapped Space Read/Write") },
|
|
{ PN_PROCESS_MAPPED_WRITECOPY, 0, TEXT("Mapped Space Write Copy") },
|
|
{ PN_PROCESS_MAPPED_EXECUTABLE, 0, TEXT("Mapped Space Executable") },
|
|
{ PN_PROCESS_MAPPED_EXE_READONLY, 0, TEXT("Mapped Space Exec Read Only") },
|
|
{ PN_PROCESS_MAPPED_EXE_READWRITE, 0, TEXT("Mapped Space Exec Read/Write") },
|
|
{ PN_PROCESS_MAPPED_EXE_WRITECOPY, 0, TEXT("Mapped Space Exec Write Copy") },
|
|
{ PN_PROCESS_IMAGE_NOACCESS, 0, TEXT("Image Space No Access") },
|
|
{ PN_PROCESS_IMAGE_READONLY, 0, TEXT("Image Space Read Only") },
|
|
{ PN_PROCESS_IMAGE_READWRITE, 0, TEXT("Image Space Read/Write") },
|
|
{ PN_PROCESS_IMAGE_WRITECOPY, 0, TEXT("Image Space Write Copy") },
|
|
{ PN_PROCESS_IMAGE_EXECUTABLE, 0, TEXT("Image Space Executable") },
|
|
{ PN_PROCESS_IMAGE_EXE_READONLY, 0, TEXT("Image Space Exec Read Only") },
|
|
{ PN_PROCESS_IMAGE_EXE_READWRITE, 0, TEXT("Image Space Exec Read/Write") },
|
|
{ PN_PROCESS_IMAGE_EXE_WRITECOPY, 0, TEXT("Image Space Exec Write Copy") },
|
|
{ 0, 0, 0 },
|
|
};
|
|
|
|
#define NENTRIES ((sizeof(entries) / sizeof(entry_t)) - 1)
|
|
|
|
static int
|
|
key_for_index(int key)
|
|
{
|
|
entry_t* entry = entries + NENTRIES / 2;
|
|
unsigned int step = 64 / 4; // XXX
|
|
|
|
while (step) {
|
|
if (key < entry->e_key)
|
|
entry -= step;
|
|
else if (key > entry->e_key)
|
|
entry += step;
|
|
|
|
if (key == entry->e_key)
|
|
return entry->e_index;
|
|
|
|
step >>= 1;
|
|
}
|
|
|
|
assert(false);
|
|
return 0;
|
|
}
|
|
|
|
|
|
class auto_hkey {
|
|
protected:
|
|
HKEY hkey;
|
|
|
|
HKEY* begin_assignment() {
|
|
if (hkey) {
|
|
::RegCloseKey(hkey);
|
|
hkey = 0;
|
|
}
|
|
return &hkey;
|
|
}
|
|
|
|
public:
|
|
auto_hkey() : hkey(0) {}
|
|
~auto_hkey() { ::RegCloseKey(hkey); }
|
|
|
|
HKEY get() const { return hkey; }
|
|
operator HKEY() const { return get(); }
|
|
|
|
friend HKEY*
|
|
getter_Acquires(auto_hkey& hkey);
|
|
};
|
|
|
|
static HKEY*
|
|
getter_Acquires(auto_hkey& hkey)
|
|
{
|
|
return hkey.begin_assignment();
|
|
}
|
|
|
|
|
|
static int
|
|
get_perf_titles(char*& buffer, char**& titles, int& last_title_index)
|
|
{
|
|
DWORD result;
|
|
|
|
// Open the perflib key to find out the last counter's index and
|
|
// system version.
|
|
auto_hkey perflib_hkey;
|
|
result = ::RegOpenKeyEx(HKEY_LOCAL_MACHINE,
|
|
TEXT("software\\microsoft\\windows nt\\currentversion\\perflib"),
|
|
0,
|
|
KEY_READ,
|
|
getter_Acquires(perflib_hkey));
|
|
|
|
if (result != ERROR_SUCCESS)
|
|
return result;
|
|
|
|
// Get the last counter's index so we know how much memory to
|
|
// allocate for titles
|
|
DWORD data_size = sizeof(DWORD);
|
|
DWORD type;
|
|
result = ::RegQueryValueEx(perflib_hkey,
|
|
TEXT("Last Counter"),
|
|
0,
|
|
&type,
|
|
reinterpret_cast<BYTE*>(&last_title_index),
|
|
&data_size);
|
|
|
|
if (result != ERROR_SUCCESS)
|
|
return result;
|
|
|
|
// Find system version, for system earlier than 1.0a, there's no
|
|
// version value.
|
|
int version;
|
|
result = ::RegQueryValueEx(perflib_hkey,
|
|
TEXT("Version"),
|
|
0,
|
|
&type,
|
|
reinterpret_cast<BYTE*>(&version),
|
|
&data_size);
|
|
|
|
bool is_nt_10 = (result == ERROR_SUCCESS);
|
|
|
|
// Now, get ready for the counter names and indexes.
|
|
char* counter_value_name;
|
|
auto_hkey counter_autohkey;
|
|
HKEY counter_hkey;
|
|
if (is_nt_10) {
|
|
// NT 1.0, so make hKey2 point to ...\perflib\009 and get
|
|
// the counters from value "Counters"
|
|
counter_value_name = TEXT("Counters");
|
|
result = ::RegOpenKeyEx(HKEY_LOCAL_MACHINE,
|
|
TEXT("software\\microsoft\\windows nt\\currentversion\\perflib\\009"),
|
|
0,
|
|
KEY_READ,
|
|
getter_Acquires(counter_autohkey));
|
|
|
|
if (result != ERROR_SUCCESS)
|
|
return result;
|
|
|
|
counter_hkey = counter_autohkey;
|
|
}
|
|
else {
|
|
// NT 1.0a or later. Get the counters in key HKEY_PERFORMANCE_KEY
|
|
// and from value "Counter 009"
|
|
counter_value_name = TEXT("Counter 009");
|
|
counter_hkey = HKEY_PERFORMANCE_DATA;
|
|
}
|
|
|
|
// Find out the size of the data.
|
|
result = ::RegQueryValueEx(counter_hkey,
|
|
counter_value_name,
|
|
0,
|
|
&type,
|
|
0,
|
|
&data_size);
|
|
|
|
if (result != ERROR_SUCCESS)
|
|
return result;
|
|
|
|
// Allocate memory
|
|
buffer = new char[data_size];
|
|
titles = new char*[last_title_index + 1];
|
|
for (int i = 0; i <= last_title_index; ++i)
|
|
titles[i] = 0;
|
|
|
|
// Query the data
|
|
result = ::RegQueryValueEx(counter_hkey,
|
|
counter_value_name,
|
|
0,
|
|
&type,
|
|
reinterpret_cast<BYTE*>(buffer),
|
|
&data_size);
|
|
if (result != ERROR_SUCCESS)
|
|
return result;
|
|
|
|
// Setup the titles array of pointers to point to beginning of
|
|
// each title string.
|
|
char* title = buffer;
|
|
int len;
|
|
|
|
while (len = lstrlen(title)) {
|
|
int index = atoi(title);
|
|
title += len + 1;
|
|
|
|
if (index <= last_title_index)
|
|
titles[index] = title;
|
|
|
|
#ifdef DEBUG
|
|
printf("%d=%s\n", index, title);
|
|
#endif
|
|
|
|
title += lstrlen(title) + 1;
|
|
}
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
static void
|
|
init_entries()
|
|
{
|
|
char* buffer;
|
|
char** titles;
|
|
int last = 0;
|
|
|
|
DWORD result = get_perf_titles(buffer, titles, last);
|
|
|
|
assert(result == ERROR_SUCCESS);
|
|
|
|
for (entry_t* entry = entries; entry->e_key != 0; ++entry) {
|
|
for (int index = 0; index <= last; ++index) {
|
|
if (titles[index] && 0 == lstrcmpi(titles[index], entry->e_title)) {
|
|
entry->e_index = index;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (entry->e_index == 0) {
|
|
fprintf(stderr, "warning: unable to find index for ``%s''\n", entry->e_title);
|
|
}
|
|
}
|
|
|
|
delete[] buffer;
|
|
delete[] titles;
|
|
}
|
|
|
|
|
|
|
|
static DWORD
|
|
get_perf_data(HKEY perf_hkey, char* object_index, PERF_DATA_BLOCK** data, DWORD* size)
|
|
{
|
|
if (! *data)
|
|
*data = reinterpret_cast<PERF_DATA_BLOCK*>(new char[*size]);
|
|
|
|
DWORD result;
|
|
|
|
while (1) {
|
|
DWORD type;
|
|
DWORD real_size = *size;
|
|
|
|
result = ::RegQueryValueEx(perf_hkey,
|
|
object_index,
|
|
0,
|
|
&type,
|
|
reinterpret_cast<BYTE*>(*data),
|
|
&real_size);
|
|
|
|
if (result != ERROR_MORE_DATA)
|
|
break;
|
|
|
|
delete[] *data;
|
|
*size += 1024;
|
|
*data = reinterpret_cast<PERF_DATA_BLOCK*>(new char[*size]);
|
|
|
|
if (! *data)
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
static const PERF_OBJECT_TYPE*
|
|
first_object(const PERF_DATA_BLOCK* data)
|
|
{
|
|
return data
|
|
? reinterpret_cast<const PERF_OBJECT_TYPE*>(reinterpret_cast<const char*>(data) + data->HeaderLength)
|
|
: 0;
|
|
}
|
|
|
|
static const PERF_OBJECT_TYPE*
|
|
next_object(const PERF_OBJECT_TYPE* object)
|
|
{
|
|
return object
|
|
? reinterpret_cast<const PERF_OBJECT_TYPE*>(reinterpret_cast<const char*>(object) + object->TotalByteLength)
|
|
: 0;
|
|
}
|
|
|
|
const PERF_OBJECT_TYPE*
|
|
find_object(const PERF_DATA_BLOCK* data, DWORD index)
|
|
{
|
|
const PERF_OBJECT_TYPE* object = first_object(data);
|
|
if (! object)
|
|
return 0;
|
|
|
|
for (int i = 0; i < data->NumObjectTypes; ++i) {
|
|
if (object->ObjectNameTitleIndex == index)
|
|
return object;
|
|
|
|
object = next_object(object);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
static const PERF_COUNTER_DEFINITION*
|
|
first_counter(const PERF_OBJECT_TYPE* object)
|
|
{
|
|
return object
|
|
? reinterpret_cast<const PERF_COUNTER_DEFINITION*>(reinterpret_cast<const char*>(object) + object->HeaderLength)
|
|
: 0;
|
|
}
|
|
|
|
static const PERF_COUNTER_DEFINITION*
|
|
next_counter(const PERF_COUNTER_DEFINITION* counter)
|
|
{
|
|
return counter ?
|
|
reinterpret_cast<const PERF_COUNTER_DEFINITION*>(reinterpret_cast<const char*>(counter) + counter->ByteLength)
|
|
: 0;
|
|
}
|
|
|
|
|
|
static const PERF_COUNTER_DEFINITION*
|
|
find_counter(const PERF_OBJECT_TYPE* object, int index)
|
|
{
|
|
const PERF_COUNTER_DEFINITION* counter =
|
|
first_counter(object);
|
|
|
|
if (! counter)
|
|
return 0;
|
|
|
|
for (int i; i < object->NumCounters; ++i) {
|
|
if (counter->CounterNameTitleIndex == index)
|
|
return counter;
|
|
|
|
counter = next_counter(counter);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
static const PERF_INSTANCE_DEFINITION*
|
|
first_instance(const PERF_OBJECT_TYPE* object)
|
|
{
|
|
return object
|
|
? reinterpret_cast<const PERF_INSTANCE_DEFINITION*>(reinterpret_cast<const char*>(object) + object->DefinitionLength)
|
|
: 0;
|
|
}
|
|
|
|
|
|
static const PERF_INSTANCE_DEFINITION*
|
|
next_instance(const PERF_INSTANCE_DEFINITION* instance)
|
|
{
|
|
if (instance) {
|
|
const PERF_COUNTER_BLOCK* counter_block =
|
|
reinterpret_cast<const PERF_COUNTER_BLOCK*>(reinterpret_cast<const char*>(instance) + instance->ByteLength);
|
|
|
|
return reinterpret_cast<const PERF_INSTANCE_DEFINITION*>(reinterpret_cast<const char*>(counter_block) + counter_block->ByteLength);
|
|
}
|
|
else {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
|
|
static const wchar_t*
|
|
instance_name(const PERF_INSTANCE_DEFINITION* instance)
|
|
{
|
|
return instance
|
|
? reinterpret_cast<const wchar_t*>(reinterpret_cast<const char*>(instance) + instance->NameOffset)
|
|
: 0;
|
|
}
|
|
|
|
|
|
static const void*
|
|
counter_data(const PERF_INSTANCE_DEFINITION* instance,
|
|
const PERF_COUNTER_DEFINITION* counter)
|
|
{
|
|
if (counter && instance) {
|
|
const PERF_COUNTER_BLOCK* counter_block;
|
|
counter_block = reinterpret_cast<const PERF_COUNTER_BLOCK*>(reinterpret_cast<const char*>(instance) + instance->ByteLength);
|
|
return reinterpret_cast<const char*>(counter_block) + counter->CounterOffset;
|
|
}
|
|
else {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
|
|
static bool
|
|
list_process(PERF_DATA_BLOCK* perf_data, wchar_t* process_name)
|
|
{
|
|
const PERF_OBJECT_TYPE* process = find_object(perf_data, key_for_index(PN_PROCESS));
|
|
const PERF_COUNTER_DEFINITION* working_set = find_counter(process, key_for_index(PN_PROCESS_WORKING_SET));
|
|
const PERF_COUNTER_DEFINITION* peak_working_set = find_counter(process, key_for_index(PN_PROCESS_PEAK_WS));
|
|
const PERF_COUNTER_DEFINITION* private_page = find_counter(process, key_for_index(PN_PROCESS_PRIVATE_PAGE));
|
|
const PERF_COUNTER_DEFINITION* virtual_size = find_counter(process, key_for_index(PN_PROCESS_VIRTUAL_SIZE));
|
|
|
|
const PERF_INSTANCE_DEFINITION* instance = first_instance(process);
|
|
int index = 0;
|
|
|
|
bool found = false;
|
|
|
|
while (instance && index < process->NumInstances) {
|
|
const wchar_t* name = instance_name(instance);
|
|
if (lstrcmpW(process_name, name) == 0) {
|
|
printf("%d %d %d %d\n",
|
|
*(static_cast<const int*>(counter_data(instance, working_set))),
|
|
*(static_cast<const int*>(counter_data(instance, peak_working_set))),
|
|
*(static_cast<const int*>(counter_data(instance, private_page))),
|
|
*(static_cast<const int*>(counter_data(instance, virtual_size))));
|
|
|
|
found = true;
|
|
}
|
|
|
|
instance = next_instance(instance);
|
|
++index;
|
|
}
|
|
|
|
if (found) {
|
|
#if 0
|
|
// Dig up address space data.
|
|
PERF_OBJECT_TYPE* address_space = FindObject(costly_data, PX_PROCESS_ADDRESS_SPACE);
|
|
PERF_COUNTER_DEFINITION* image_executable = FindCounter(process, PX_PROCESS_IMAGE_EXECUTABLE);
|
|
PERF_COUNTER_DEFINITION* image_exe_readonly = FindCounter(process, PX_PROCESS_IMAGE_EXE_READONLY);
|
|
PERF_COUNTER_DEFINITION* image_exe_readwrite = FindCounter(process, PX_PROCESS_IMAGE_EXE_READWRITE);
|
|
PERF_COUNTER_DEFINITION* image_exe_writecopy = FindCounter(process, PX_PROCESS_IMAGE_EXE_WRITECOPY);
|
|
#endif
|
|
}
|
|
|
|
return found;
|
|
}
|
|
|
|
|
|
int
|
|
main(int argc, char* argv[])
|
|
{
|
|
wchar_t process_name[32];
|
|
|
|
int interval = 10000; // msec
|
|
|
|
int i = 0;
|
|
while (++i < argc) {
|
|
if (argv[i][0] != '-')
|
|
break;
|
|
|
|
switch (argv[i][1]) {
|
|
case 'i':
|
|
interval = atoi(argv[++i]) * 1000;
|
|
break;
|
|
|
|
default:
|
|
fprintf(stderr, "unknown option `%c'\n", argv[i][1]);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
if (argv[i]) {
|
|
char* p = argv[i];
|
|
wchar_t* q = process_name;
|
|
while (*q++ = wchar_t(*p++))
|
|
continue;
|
|
}
|
|
else {
|
|
fprintf(stderr, "no image name specified\n");
|
|
exit(1);
|
|
}
|
|
|
|
init_entries();
|
|
|
|
PERF_DATA_BLOCK* perf_data = 0;
|
|
PERF_DATA_BLOCK* costly_data = 0;
|
|
DWORD perf_data_size = 50 * 1024;
|
|
DWORD costly_data_size = 100 * 1024;
|
|
|
|
do {
|
|
char buf[64];
|
|
sprintf(buf, "%ld %ld",
|
|
key_for_index(PN_PROCESS),
|
|
key_for_index(PN_THREAD));
|
|
|
|
get_perf_data(HKEY_PERFORMANCE_DATA, buf, &perf_data, &perf_data_size);
|
|
|
|
#if 0
|
|
sprintf(buf, "%ld %ld %ld",
|
|
key_for_index(PN_PROCESS_ADDRESS_SPACE),
|
|
key_for_index(PN_IMAGE),
|
|
key_for_index(PN_THREAD_DETAILS));
|
|
|
|
get_perf_data(HKEY_PERFORMANCE_DATA, buf, &costly_data, &costly_data_size);
|
|
#endif
|
|
|
|
if (! list_process(perf_data, process_name))
|
|
break;
|
|
|
|
_sleep(interval);
|
|
} while (1);
|
|
|
|
return 0;
|
|
}
|
|
|