mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2636 lines
82 KiB
2636 lines
82 KiB
//==========================================================================//
|
|
// Includes //
|
|
//==========================================================================//
|
|
|
|
|
|
#include <string.h> // strupr
|
|
#include <stdio.h> // for sprintf.
|
|
#include "perfmon.h"
|
|
#include "utils.h"
|
|
|
|
#include "pmemory.h" // for MemoryXXX (mallloc-type) routines
|
|
#include "playback.h" // for PlayingBackLog
|
|
#include "perfdata.h" // external declarations for this file
|
|
#include "system.h" // for DeleteUnusedSystems
|
|
#include "line.h" // for LineFind???()
|
|
#include "perfmops.h" // for AppendObjectToValueList()
|
|
#include "perfmsg.h" // for event log messages
|
|
|
|
//==========================================================================//
|
|
// Constants //
|
|
//==========================================================================//
|
|
|
|
TCHAR NamesKey[] = L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Perflib";
|
|
TCHAR Counters[] = L"Counters";
|
|
TCHAR Help[] = L"Help";
|
|
TCHAR LastHelp[] = L"Last Help";
|
|
TCHAR LastCounter[] = L"Last Counter";
|
|
TCHAR SysVersion[] = L"Version";
|
|
TCHAR CounterNameStr[] = L"Counter ";
|
|
TCHAR ExplainNameStr[] = L"Explain ";
|
|
|
|
#define szPerfSubkey (NULL)
|
|
TCHAR NULL_NAME[] = L" ";
|
|
#define RESERVED 0L
|
|
|
|
TCHAR DefaultLangId[4] ;
|
|
TCHAR EnglishLangId[4] ;
|
|
|
|
static HANDLE *lpHandles ;
|
|
static int NumberOfHandles = 0 ;
|
|
|
|
//==========================================================================//
|
|
// Macros //
|
|
//==========================================================================//
|
|
|
|
|
|
|
|
//==========================================================================//
|
|
// Local Data //
|
|
//==========================================================================//
|
|
|
|
|
|
// When the conversion of this code is complete, this will be the *only*
|
|
// allocated copy of the performance data. It will monotonically grow
|
|
// to hold the largest of the system's performance data.
|
|
|
|
// PPERFDATA pGlobalPerfData ;
|
|
|
|
//==========================================================================//
|
|
// Local Functions //
|
|
//==========================================================================//
|
|
|
|
NTSTATUS
|
|
AddNamesToArray (
|
|
LPTSTR pNames,
|
|
DWORD dwLastID,
|
|
LPWSTR *lpCounterId
|
|
) ;
|
|
|
|
//======================================//
|
|
// Object Accessors //
|
|
//======================================//
|
|
|
|
|
|
void
|
|
ObjectName (
|
|
PPERFSYSTEM pSystem,
|
|
PPERFOBJECT pObject,
|
|
LPTSTR lpszName,
|
|
int iLen
|
|
)
|
|
{
|
|
strclr (lpszName) ;
|
|
QueryPerformanceName (pSystem,
|
|
pObject->ObjectNameTitleIndex,
|
|
0, iLen, lpszName, FALSE) ;
|
|
}
|
|
|
|
|
|
|
|
//======================================//
|
|
// Counter Accessors //
|
|
//======================================//
|
|
|
|
|
|
PPERFINSTANCEDEF
|
|
FirstInstance(
|
|
PPERFOBJECT pObjectDef
|
|
)
|
|
{
|
|
return (PPERFINSTANCEDEF )
|
|
((PCHAR) pObjectDef + pObjectDef->DefinitionLength);
|
|
}
|
|
|
|
|
|
PPERFINSTANCEDEF
|
|
NextInstance(
|
|
PPERFINSTANCEDEF pInstDef
|
|
)
|
|
{
|
|
PERF_COUNTER_BLOCK *pCounterBlock;
|
|
|
|
pCounterBlock = (PERF_COUNTER_BLOCK *)
|
|
((PCHAR) pInstDef + pInstDef->ByteLength);
|
|
|
|
return (PPERFINSTANCEDEF )
|
|
((PCHAR) pCounterBlock + pCounterBlock->ByteLength);
|
|
}
|
|
|
|
|
|
|
|
|
|
LPWSTR
|
|
GetInstanceName(
|
|
PPERFINSTANCEDEF pInstDef
|
|
)
|
|
{
|
|
return (LPWSTR) ((PCHAR) pInstDef + pInstDef->NameOffset);
|
|
}
|
|
|
|
DWORD
|
|
GetAnsiInstanceName (
|
|
PPERFINSTANCEDEF pInstance,
|
|
LPWSTR lpszInstance,
|
|
DWORD dwCodePage
|
|
)
|
|
{
|
|
LPSTR szSource;
|
|
DWORD dwLength;
|
|
|
|
szSource = (LPSTR)GetInstanceName(pInstance);
|
|
|
|
// the locale should be set here
|
|
|
|
// pInstance->NameLength == the number of bytes (chars) in the string
|
|
dwLength = mbstowcs (lpszInstance, szSource, pInstance->NameLength);
|
|
lpszInstance[dwLength] = 0; // null terminate string buffer
|
|
|
|
return dwLength;
|
|
}
|
|
|
|
DWORD
|
|
GetUnicodeInstanceName (
|
|
PPERFINSTANCEDEF pInstance,
|
|
LPWSTR lpszInstance
|
|
)
|
|
{
|
|
LPWSTR wszSource;
|
|
DWORD dwLength;
|
|
|
|
wszSource = GetInstanceName(pInstance) ;
|
|
|
|
// pInstance->NameLength == length of string in BYTES so adjust to
|
|
// number of wide characters here
|
|
dwLength = pInstance->NameLength / sizeof(WCHAR);
|
|
|
|
wcsncpy (lpszInstance,
|
|
(LPWSTR)wszSource,
|
|
dwLength);
|
|
|
|
lpszInstance[dwLength] = 0;
|
|
|
|
return (DWORD)(wcslen(lpszInstance)); // just incase there's null's in the string
|
|
|
|
}
|
|
|
|
void
|
|
GetInstanceNameStr (
|
|
PPERFINSTANCEDEF pInstance,
|
|
LPWSTR lpszInstance,
|
|
DWORD dwCodePage
|
|
)
|
|
{
|
|
DWORD dwCharSize;
|
|
DWORD dwLength;
|
|
|
|
if (dwCodePage > 0) {
|
|
dwCharSize = sizeof(CHAR);
|
|
dwLength = GetAnsiInstanceName (pInstance, lpszInstance, dwCodePage);
|
|
} else { // it's a UNICODE name
|
|
dwCharSize = sizeof(WCHAR);
|
|
dwLength = GetUnicodeInstanceName (pInstance, lpszInstance);
|
|
}
|
|
// sanity check here...
|
|
// the returned string length (in characters) plus the terminating NULL
|
|
// should be the same as the specified length in bytes divided by the
|
|
// character size. If not then the codepage and instance data type
|
|
// don't line up so test that here
|
|
|
|
if ((dwLength + 1) != (pInstance->NameLength / dwCharSize)) {
|
|
// something isn't quite right so try the "other" type of string type
|
|
if (dwCharSize == sizeof(CHAR)) {
|
|
// then we tried to read it as an ASCII string and that didn't work
|
|
// so try it as a UNICODE (if that doesn't work give up and return
|
|
// it any way.
|
|
dwLength = GetUnicodeInstanceName (pInstance, lpszInstance);
|
|
} else if (dwCharSize == sizeof(WCHAR)) {
|
|
// then we tried to read it as a UNICODE string and that didn't work
|
|
// so try it as an ASCII string (if that doesn't work give up and return
|
|
// it any way.
|
|
dwLength = GetAnsiInstanceName (pInstance, lpszInstance, dwCodePage);
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
GetPerfComputerName(
|
|
PPERFDATA pPerfData,
|
|
LPTSTR lpszComputerName
|
|
)
|
|
{
|
|
lstrcpy(lpszComputerName, szComputerPrefix) ;
|
|
if (pPerfData) {
|
|
wcsncpy (&lpszComputerName[2],
|
|
(LPWSTR)((PBYTE) pPerfData + pPerfData->SystemNameOffset),
|
|
pPerfData->SystemNameLength/sizeof(WCHAR)) ;
|
|
} else {
|
|
lpszComputerName[0] = TEXT('\0') ;
|
|
}
|
|
}
|
|
|
|
|
|
//==========================================================================//
|
|
// Exported Functions //
|
|
//==========================================================================//
|
|
|
|
|
|
int
|
|
CounterIndex (
|
|
PPERFCOUNTERDEF pCounterToFind,
|
|
PPERFOBJECT pObject
|
|
)
|
|
/*
|
|
Effect: Return the index ("counter number") of pCounterToFind
|
|
within pObject. If the counter doesnt belong to pObject,
|
|
return -1.
|
|
*/
|
|
{
|
|
PPERFCOUNTERDEF pCounter ;
|
|
UINT iCounter ;
|
|
|
|
for (iCounter = 0, pCounter = FirstCounter (pObject) ;
|
|
iCounter < pObject->NumCounters ;
|
|
iCounter++, pCounter = NextCounter (pCounter)) {
|
|
if (pCounter->CounterNameTitleIndex ==
|
|
pCounterToFind->CounterNameTitleIndex)
|
|
return (iCounter) ;
|
|
}
|
|
|
|
return (-1) ;
|
|
}
|
|
|
|
|
|
HKEY
|
|
OpenSystemPerfData (
|
|
IN LPCTSTR lpszSystem
|
|
)
|
|
{
|
|
HKEY hKey = NULL;
|
|
LONG lStatus;
|
|
|
|
|
|
lStatus = ERROR_CANTOPEN; // default error if none is returned
|
|
|
|
if (IsLocalComputer(lpszSystem) || PlayingBackLog()) {
|
|
bCloseLocalMachine = TRUE ;
|
|
SetLastError (ERROR_SUCCESS);
|
|
return HKEY_PERFORMANCE_DATA;
|
|
} else {
|
|
// Must be a remote system
|
|
try {
|
|
lStatus = RegConnectRegistry (
|
|
(LPTSTR)lpszSystem,
|
|
HKEY_PERFORMANCE_DATA,
|
|
&hKey);
|
|
}finally {
|
|
if (lStatus != ERROR_SUCCESS) {
|
|
SetLastError (lStatus);
|
|
hKey = NULL;
|
|
}
|
|
}
|
|
}
|
|
return (hKey);
|
|
|
|
}
|
|
|
|
LPWSTR
|
|
*AddNewName(
|
|
HKEY hKeyNames,
|
|
PCOUNTERTEXT pCounterInfo,
|
|
LPWSTR CounterBuffer,
|
|
LPWSTR HelpBuffer,
|
|
DWORD dwLastId,
|
|
DWORD dwCounterSize,
|
|
DWORD dwHelpSize,
|
|
LANGID LangIdUsed
|
|
)
|
|
{
|
|
LPWSTR *lpReturnValue;
|
|
LPWSTR *lpCounterId;
|
|
LPWSTR lpCounterNames;
|
|
LPWSTR lpHelpText;
|
|
DWORD dwArraySize;
|
|
DWORD dwBufferSize;
|
|
DWORD dwValueType;
|
|
LONG lWin32Status = ERROR_SUCCESS;
|
|
NTSTATUS Status;
|
|
DWORD dwLastError;
|
|
|
|
dwArraySize = (dwLastId + 1 ) * sizeof(LPWSTR);
|
|
lpReturnValue = MemoryAllocate (dwArraySize + dwCounterSize + dwHelpSize);
|
|
|
|
if (!lpReturnValue)
|
|
goto ERROR_EXIT;
|
|
|
|
// initialize pointers into buffer
|
|
|
|
lpCounterId = lpReturnValue;
|
|
lpCounterNames = (LPWSTR)((LPBYTE)lpCounterId + dwArraySize);
|
|
lpHelpText = (LPWSTR)((LPBYTE)lpCounterNames + dwCounterSize);
|
|
|
|
// read counters into memory
|
|
|
|
dwBufferSize = dwCounterSize;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
CounterBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPVOID)lpCounterNames,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) goto ERROR_EXIT;
|
|
|
|
if (bExplainTextButtonHit) {
|
|
dwBufferSize = dwHelpSize;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
HelpBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPVOID)lpHelpText,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) goto ERROR_EXIT;
|
|
}
|
|
|
|
// load counter array items
|
|
Status = AddNamesToArray (lpCounterNames, dwLastId, lpCounterId);
|
|
if (Status != ERROR_SUCCESS) goto ERROR_EXIT;
|
|
|
|
if (bExplainTextButtonHit) {
|
|
Status = AddNamesToArray (lpHelpText, dwLastId, lpCounterId);
|
|
}
|
|
|
|
if (Status != ERROR_SUCCESS) goto ERROR_EXIT;
|
|
|
|
if (pCounterInfo) {
|
|
pCounterInfo->dwLastId = dwLastId;
|
|
pCounterInfo->dwLangId = LangIdUsed;
|
|
pCounterInfo->dwHelpSize = dwHelpSize;
|
|
pCounterInfo->dwCounterSize = dwCounterSize;
|
|
}
|
|
|
|
return lpReturnValue;
|
|
|
|
ERROR_EXIT:
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
dwLastError = GetLastError();
|
|
}
|
|
|
|
if (lpReturnValue) {
|
|
MemoryFree ((LPVOID)lpReturnValue);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
|
|
LPWSTR
|
|
*BuildNewNameTable(
|
|
PPERFSYSTEM pSystem,
|
|
LPWSTR lpszLangId, // unicode value of Language subkey
|
|
PCOUNTERTEXT pCounterInfo,
|
|
LANGID iLangId, // lang ID of the lpszLangId
|
|
DWORD dwLastId
|
|
)
|
|
/*++
|
|
|
|
BuildNewNameTable
|
|
|
|
Arguments:
|
|
|
|
lpszLangId
|
|
The unicode id of the language to look up. (English is 0x409)
|
|
|
|
Return Value:
|
|
|
|
pointer to an allocated table. (the caller must free it when finished!)
|
|
the table is an array of pointers to zero terminated strings. NULL is
|
|
returned if an error occured.
|
|
|
|
--*/
|
|
{
|
|
LONG lWin32Status;
|
|
DWORD dwValueType;
|
|
DWORD dwLastError;
|
|
DWORD dwBufferSize;
|
|
DWORD dwCounterSize;
|
|
DWORD dwHelpSize;
|
|
HKEY hKeyNames;
|
|
TCHAR CounterBuffer [MiscTextLen] ;
|
|
TCHAR ExplainBuffer [MiscTextLen] ;
|
|
TCHAR subLangId [ShortTextLen] ;
|
|
LANGID LangIdUsed = iLangId;
|
|
|
|
|
|
//initialize local variables
|
|
hKeyNames = pSystem->sysDataKey;
|
|
|
|
|
|
// check for null arguments and insert defaults if necessary
|
|
if (!lpszLangId) {
|
|
lpszLangId = DefaultLangId;
|
|
LangIdUsed = iLanguage ;
|
|
}
|
|
|
|
// get size of counter names and add that to the arrays
|
|
lstrcpy (CounterBuffer, CounterNameStr);
|
|
lstrcat (CounterBuffer, lpszLangId);
|
|
|
|
lstrcpy (ExplainBuffer, ExplainNameStr);
|
|
lstrcat (ExplainBuffer, lpszLangId);
|
|
|
|
dwBufferSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
CounterBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
// check for ACCESS_DENIED error first since if it's here
|
|
// it will be returned on all subsequent calls, we might as well
|
|
// bail out now.
|
|
|
|
if (lWin32Status == ERROR_ACCESS_DENIED) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
// try take out the country ID
|
|
LangIdUsed = MAKELANGID (LangIdUsed & 0x0ff, LANG_NEUTRAL);
|
|
TSPRINTF (subLangId, TEXT("%03x"), LangIdUsed);
|
|
lstrcpy (CounterBuffer, CounterNameStr);
|
|
lstrcat (CounterBuffer, subLangId);
|
|
|
|
lstrcpy (ExplainBuffer, ExplainNameStr);
|
|
lstrcat (ExplainBuffer, subLangId);
|
|
|
|
dwBufferSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
CounterBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwBufferSize);
|
|
}
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
// try the EnglishLangId
|
|
if (!strsame(EnglishLangId, subLangId)) {
|
|
|
|
lstrcpy (CounterBuffer, CounterNameStr);
|
|
lstrcat (CounterBuffer, EnglishLangId);
|
|
|
|
lstrcpy (ExplainBuffer, ExplainNameStr);
|
|
lstrcat (ExplainBuffer, EnglishLangId);
|
|
|
|
LangIdUsed = iEnglishLanguage ;
|
|
|
|
dwBufferSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
CounterBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwBufferSize);
|
|
}
|
|
}
|
|
|
|
// Fail, too bad...
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
dwCounterSize = dwBufferSize;
|
|
|
|
// If ExplainText is needed, then
|
|
// get size of help text and add that to the arrays
|
|
|
|
if (bExplainTextButtonHit) {
|
|
dwBufferSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
ExplainBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) goto BNT_BAILOUT;
|
|
|
|
dwHelpSize = dwBufferSize;
|
|
} else {
|
|
dwHelpSize = 0;
|
|
}
|
|
|
|
return (AddNewName(
|
|
hKeyNames,
|
|
pCounterInfo,
|
|
CounterBuffer,
|
|
ExplainBuffer,
|
|
dwLastId,
|
|
dwCounterSize,
|
|
dwHelpSize,
|
|
LangIdUsed));
|
|
|
|
|
|
BNT_BAILOUT:
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
dwLastError = GetLastError();
|
|
// set the LastError value since a null pointer will
|
|
// be returned which doesn't tell much to the caller
|
|
SetLastError (lWin32Status);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
|
|
LPWSTR *
|
|
BuildOldNameTable(
|
|
HKEY hKeyRegistry, // handle to registry db with counter names
|
|
LPWSTR lpszLangId, // unicode value of Language subkey
|
|
PCOUNTERTEXT pCounterInfo,
|
|
LANGID iLangId, // lang ID of the lpszLangId
|
|
DWORD dwLastId
|
|
)
|
|
/*++
|
|
|
|
BuildOldNameTable
|
|
|
|
Arguments:
|
|
|
|
hKeyRegistry
|
|
Handle to an open registry (this can be local or remote.) and
|
|
is the value returned by RegConnectRegistry or a default key.
|
|
|
|
lpszLangId
|
|
The unicode id of the language to look up. (English is 0x409)
|
|
|
|
Return Value:
|
|
|
|
pointer to an allocated table. (the caller must free it when finished!)
|
|
the table is an array of pointers to zero terminated strings. NULL is
|
|
returned if an error occured.
|
|
|
|
--*/
|
|
{
|
|
LPWSTR *lpReturnValue = NULL;
|
|
|
|
LONG lWin32Status;
|
|
DWORD dwValueType;
|
|
DWORD dwLastError;
|
|
DWORD dwBufferSize;
|
|
DWORD dwCounterSize;
|
|
DWORD dwHelpSize;
|
|
HKEY hKeyNames;
|
|
TCHAR tempBuffer [MiscTextLen] ;
|
|
TCHAR subLangId [ShortTextLen] ;
|
|
LPWSTR lpValueNameString;
|
|
LANGID LangIdUsed = iLangId;
|
|
TCHAR Slash[2];
|
|
|
|
//initialize local variables
|
|
hKeyNames = NULL;
|
|
Slash[0] = L'\\';
|
|
Slash[1] = L'\0';
|
|
|
|
// check for null arguments and insert defaults if necessary
|
|
if (!lpszLangId) {
|
|
lpszLangId = DefaultLangId;
|
|
LangIdUsed = iLanguage ;
|
|
}
|
|
|
|
// get size of string buffer
|
|
lpValueNameString = tempBuffer ;
|
|
|
|
lstrcpy (lpValueNameString, NamesKey);
|
|
lstrcat (lpValueNameString, Slash);
|
|
lstrcat (lpValueNameString, lpszLangId);
|
|
|
|
lWin32Status = RegOpenKeyEx (
|
|
hKeyRegistry,
|
|
lpValueNameString,
|
|
RESERVED,
|
|
KEY_READ,
|
|
&hKeyNames);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
// check for ACCESS_DENIED error first since if it's here
|
|
// it will be returned on all subsequent calls, we might as well
|
|
// bail out now.
|
|
|
|
if (lWin32Status == ERROR_ACCESS_DENIED) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
// try take out the country ID
|
|
LangIdUsed = MAKELANGID (LangIdUsed & 0x0ff, LANG_NEUTRAL);
|
|
TSPRINTF (subLangId, TEXT("%03x"), LangIdUsed);
|
|
lstrcpy (lpValueNameString, NamesKey);
|
|
lstrcat (lpValueNameString, Slash);
|
|
lstrcat (lpValueNameString, subLangId);
|
|
|
|
lWin32Status = RegOpenKeyEx (
|
|
hKeyRegistry,
|
|
lpValueNameString,
|
|
RESERVED,
|
|
KEY_READ,
|
|
&hKeyNames);
|
|
}
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
// try the EnglishLangId
|
|
if (!strsame(EnglishLangId, subLangId)) {
|
|
|
|
lstrcpy (lpValueNameString, NamesKey);
|
|
lstrcat (lpValueNameString, Slash);
|
|
lstrcat (lpValueNameString, EnglishLangId);
|
|
|
|
LangIdUsed = iEnglishLanguage ;
|
|
|
|
lWin32Status = RegOpenKeyEx (
|
|
hKeyRegistry,
|
|
lpValueNameString,
|
|
RESERVED,
|
|
KEY_READ,
|
|
&hKeyNames);
|
|
}
|
|
}
|
|
|
|
// Fail, too bad...
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
// get size of counter names and add that to the arrays
|
|
|
|
|
|
dwBufferSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
Counters,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) goto BNT_BAILOUT;
|
|
|
|
dwCounterSize = dwBufferSize;
|
|
|
|
// If ExplainText is needed, then
|
|
// get size of help text and add that to the arrays
|
|
|
|
if (bExplainTextButtonHit) {
|
|
dwBufferSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
Help,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) goto BNT_BAILOUT;
|
|
|
|
dwHelpSize = dwBufferSize;
|
|
} else {
|
|
dwHelpSize = 0;
|
|
}
|
|
|
|
lpReturnValue = AddNewName(
|
|
hKeyNames,
|
|
pCounterInfo,
|
|
Counters,
|
|
Help,
|
|
dwLastId,
|
|
dwCounterSize,
|
|
dwHelpSize,
|
|
LangIdUsed);
|
|
|
|
RegCloseKey (hKeyNames);
|
|
|
|
return lpReturnValue;
|
|
|
|
BNT_BAILOUT:
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
dwLastError = GetLastError();
|
|
// set the LastError value since a null pointer will
|
|
// be returned which doesn't tell much to the caller
|
|
SetLastError (lWin32Status);
|
|
}
|
|
|
|
if (lpReturnValue) {
|
|
MemoryFree ((LPVOID)lpReturnValue);
|
|
}
|
|
|
|
if (hKeyNames) RegCloseKey (hKeyNames);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
|
|
LPWSTR *
|
|
BuildNameTable(
|
|
PPERFSYSTEM pSysInfo,
|
|
HKEY hKeyRegistry, // handle to registry db with counter names
|
|
LPWSTR lpszLangId, // unicode value of Language subkey
|
|
PCOUNTERTEXT pCounterInfo,
|
|
LANGID iLangId // lang ID of the lpszLangId
|
|
)
|
|
/*++
|
|
|
|
BuildNameTable
|
|
|
|
Arguments:
|
|
|
|
hKeyRegistry
|
|
Handle to an open registry (this can be local or remote.) and
|
|
is the value returned by RegConnectRegistry or a default key.
|
|
|
|
lpszLangId
|
|
The unicode id of the language to look up. (English is 0x409)
|
|
|
|
Return Value:
|
|
|
|
pointer to an allocated table. (the caller must free it when finished!)
|
|
the table is an array of pointers to zero terminated strings. NULL is
|
|
returned if an error occured.
|
|
|
|
--*/
|
|
{
|
|
|
|
LPWSTR *lpReturnValue;
|
|
LONG lWin32Status;
|
|
DWORD dwLastError;
|
|
DWORD dwValueType;
|
|
DWORD dwLastHelp;
|
|
DWORD dwLastCounter;
|
|
DWORD dwLastId;
|
|
DWORD dwBufferSize;
|
|
HKEY hKeyValue;
|
|
DWORD dwSystemVersion;
|
|
|
|
|
|
//initialize local variables
|
|
lpReturnValue = NULL;
|
|
hKeyValue = NULL;
|
|
|
|
|
|
// open registry to get number of items for computing array size
|
|
|
|
lWin32Status = RegOpenKeyEx (
|
|
hKeyRegistry,
|
|
NamesKey,
|
|
RESERVED,
|
|
KEY_READ,
|
|
&hKeyValue);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
// get number of items
|
|
|
|
dwBufferSize = sizeof (dwLastHelp);
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyValue,
|
|
LastHelp,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPBYTE)&dwLastHelp,
|
|
&dwBufferSize);
|
|
|
|
if ((lWin32Status != ERROR_SUCCESS) || (dwValueType != REG_DWORD)) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
dwBufferSize = sizeof (dwLastCounter);
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyValue,
|
|
LastCounter,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPBYTE)&dwLastCounter,
|
|
&dwBufferSize);
|
|
|
|
if ((lWin32Status != ERROR_SUCCESS) || (dwValueType != REG_DWORD)) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
if (dwLastCounter >= dwLastHelp) {
|
|
dwLastId = dwLastCounter;
|
|
} else {
|
|
dwLastId = dwLastHelp;
|
|
}
|
|
|
|
// get system version
|
|
dwBufferSize = sizeof (dwSystemVersion);
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyValue,
|
|
SysVersion,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPBYTE)&dwSystemVersion,
|
|
&dwBufferSize);
|
|
|
|
if ((lWin32Status != ERROR_SUCCESS) || (dwValueType != REG_DWORD)) {
|
|
pSysInfo->SysVersion = 0x10000;
|
|
} else {
|
|
pSysInfo->SysVersion = dwSystemVersion;
|
|
}
|
|
|
|
if (pSysInfo->SysVersion <= 0x10000) {
|
|
lpReturnValue = BuildOldNameTable (
|
|
hKeyRegistry,
|
|
lpszLangId,
|
|
pCounterInfo,
|
|
iLangId,
|
|
dwLastId) ;
|
|
} else {
|
|
lpReturnValue = BuildNewNameTable (
|
|
pSysInfo,
|
|
lpszLangId,
|
|
pCounterInfo,
|
|
iLangId,
|
|
dwLastId) ;
|
|
}
|
|
|
|
RegCloseKey (hKeyValue);
|
|
return lpReturnValue;
|
|
|
|
BNT_BAILOUT:
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
dwLastError = GetLastError();
|
|
// set the LastError value since a null pointer will
|
|
// be returned which doesn't tell much to the caller
|
|
SetLastError (lWin32Status);
|
|
}
|
|
if (hKeyValue) RegCloseKey (hKeyValue);
|
|
return NULL;
|
|
}
|
|
|
|
DWORD
|
|
GetSystemKey (
|
|
PPERFSYSTEM pSysInfo,
|
|
HKEY *phKeyMachine
|
|
)
|
|
{
|
|
DWORD dwStatus;
|
|
|
|
// connect to system registry
|
|
|
|
if (IsLocalComputer(pSysInfo->sysName) ||
|
|
(PlayingBackLog() && PlaybackLog.pBaseCounterNames)) {
|
|
*phKeyMachine = HKEY_LOCAL_MACHINE;
|
|
} else {
|
|
try {
|
|
dwStatus = RegConnectRegistry (
|
|
pSysInfo->sysName,
|
|
HKEY_LOCAL_MACHINE,
|
|
phKeyMachine);
|
|
|
|
if (dwStatus != ERROR_SUCCESS) {
|
|
if (PlayingBackLog()) {
|
|
// If remote machine is not on and we are
|
|
// playing back log, then, use the counters from
|
|
// local machine.
|
|
*phKeyMachine = HKEY_LOCAL_MACHINE;
|
|
} else {
|
|
return dwStatus;
|
|
}
|
|
}
|
|
}finally {
|
|
; // nothing
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
DWORD
|
|
GetSystemNames(
|
|
PPERFSYSTEM pSysInfo
|
|
)
|
|
{
|
|
HKEY hKeyMachine = 0;
|
|
DWORD dwStatus;
|
|
|
|
if (dwStatus = GetSystemKey (pSysInfo, &hKeyMachine)) {
|
|
return dwStatus;
|
|
}
|
|
|
|
// if here, then hKeyMachine is an open key to the system's
|
|
// HKEY_LOCAL_MACHINE registry database
|
|
|
|
// only one language is supported by this approach.
|
|
// multiple language support would:
|
|
// 1. enumerate language keys
|
|
// and for each key:
|
|
// 2. allocate memory for structures
|
|
// 3. call BuildNameTable for each lang key.
|
|
|
|
pSysInfo->CounterInfo.pNextTable = NULL;
|
|
pSysInfo->CounterInfo.dwLangId = iLanguage ; // default Lang ID
|
|
|
|
if (PlayingBackLog() && PlaybackLog.pBaseCounterNames) {
|
|
pSysInfo->CounterInfo.TextString = LogBuildNameTable (pSysInfo) ;
|
|
} else {
|
|
pSysInfo->CounterInfo.TextString = BuildNameTable (
|
|
pSysInfo,
|
|
hKeyMachine,
|
|
NULL, // use default
|
|
&pSysInfo->CounterInfo,
|
|
0);
|
|
}
|
|
|
|
if (hKeyMachine && hKeyMachine != HKEY_LOCAL_MACHINE) {
|
|
RegCloseKey (hKeyMachine) ;
|
|
}
|
|
|
|
if (pSysInfo->CounterInfo.TextString == NULL) {
|
|
return GetLastError();
|
|
} else {
|
|
return ERROR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
BOOL
|
|
GetHelpText(
|
|
PPERFSYSTEM pSysInfo
|
|
)
|
|
{
|
|
LPWSTR *lpCounterId;
|
|
LPWSTR lpHelpText;
|
|
LONG lWin32Status;
|
|
DWORD dwValueType;
|
|
DWORD dwArraySize;
|
|
DWORD dwBufferSize;
|
|
DWORD dwCounterSize;
|
|
DWORD dwHelpSize;
|
|
NTSTATUS Status;
|
|
DWORD dwLastId;
|
|
TCHAR Slash[2];
|
|
|
|
HKEY hKeyNames;
|
|
|
|
TCHAR SysLangId [ShortTextLen] ;
|
|
TCHAR ValueNameString [MiscTextLen] ;
|
|
HKEY hKeyMachine = 0;
|
|
DWORD dwStatus;
|
|
|
|
SetHourglassCursor() ;
|
|
|
|
//initialize local variables
|
|
lpHelpText = NULL;
|
|
hKeyNames = hKeyMachine = NULL;
|
|
Slash[0] = L'\\';
|
|
Slash[1] = L'\0';
|
|
|
|
dwBufferSize = 0;
|
|
|
|
TSPRINTF (SysLangId, TEXT("%03x"), pSysInfo->CounterInfo.dwLangId) ;
|
|
|
|
if (pSysInfo->SysVersion <= 0x10000) {
|
|
// old version, get help from registry
|
|
if (dwStatus = GetSystemKey (pSysInfo, &hKeyMachine)) {
|
|
goto ERROR_EXIT;
|
|
}
|
|
|
|
lstrcpy (ValueNameString, NamesKey);
|
|
lstrcat (ValueNameString, Slash);
|
|
lstrcat (ValueNameString, SysLangId);
|
|
|
|
lWin32Status = RegOpenKeyEx (
|
|
hKeyMachine,
|
|
ValueNameString,
|
|
RESERVED,
|
|
KEY_READ,
|
|
&hKeyNames);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) goto ERROR_EXIT;
|
|
} else {
|
|
// new system version, get it from the HKEY_PERFORMANCE
|
|
hKeyNames = pSysInfo->sysDataKey;
|
|
lstrcpy (ValueNameString, ExplainNameStr);
|
|
lstrcat (ValueNameString, SysLangId);
|
|
}
|
|
|
|
dwHelpSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
pSysInfo->SysVersion <= 0x010000 ? Help : ValueNameString,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwHelpSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS || dwHelpSize == 0) goto ERROR_EXIT;
|
|
|
|
dwLastId = pSysInfo->CounterInfo.dwLastId;
|
|
dwArraySize = (dwLastId + 1) * sizeof (LPWSTR);
|
|
dwCounterSize = pSysInfo->CounterInfo.dwCounterSize;
|
|
|
|
// allocate another memory to get the help text
|
|
lpHelpText = MemoryAllocate (dwHelpSize);
|
|
if (!lpHelpText) goto ERROR_EXIT;
|
|
|
|
dwBufferSize = dwHelpSize;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
pSysInfo->SysVersion <= 0x010000 ? Help : ValueNameString,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPVOID)lpHelpText,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) goto ERROR_EXIT;
|
|
|
|
// setup the help text pointers
|
|
lpCounterId = pSysInfo->CounterInfo.TextString;
|
|
Status = AddNamesToArray (lpHelpText, dwLastId, lpCounterId) ;
|
|
if (Status != ERROR_SUCCESS) goto ERROR_EXIT;
|
|
|
|
pSysInfo->CounterInfo.dwHelpSize = dwHelpSize;
|
|
|
|
if (pSysInfo->SysVersion <= 0x010000)
|
|
RegCloseKey (hKeyNames);
|
|
|
|
if (hKeyMachine && hKeyMachine != HKEY_LOCAL_MACHINE) {
|
|
RegCloseKey (hKeyMachine) ;
|
|
}
|
|
|
|
pSysInfo->CounterInfo.HelpTextString = lpHelpText;
|
|
|
|
SetArrowCursor() ;
|
|
|
|
return TRUE;
|
|
|
|
ERROR_EXIT:
|
|
|
|
SetArrowCursor() ;
|
|
|
|
if (lpHelpText) {
|
|
MemoryFree ((LPVOID)lpHelpText);
|
|
}
|
|
|
|
if (hKeyNames) {
|
|
RegCloseKey (hKeyNames);
|
|
}
|
|
if (hKeyMachine && hKeyMachine != HKEY_LOCAL_MACHINE) {
|
|
RegCloseKey (hKeyMachine) ;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
//
|
|
// QueryPerformanceName - Get a title, given an index
|
|
//
|
|
// Inputs:
|
|
//
|
|
// pSysInfo - Pointer to sysinfo struct for the
|
|
// system in question
|
|
//
|
|
// dwTitleIndex - Index of Title entry
|
|
//
|
|
// LangID - language in which title should be displayed
|
|
//
|
|
// cbTitle - # of char in the lpTitle buffer
|
|
//
|
|
// lpTitle - pointer to a buffer to receive the
|
|
// Title
|
|
//
|
|
// Help - TRUE is help is desired, else counter or
|
|
// object is assumed
|
|
DWORD
|
|
QueryPerformanceName(
|
|
PPERFSYSTEM pSysInfo,
|
|
DWORD dwTitleIndex,
|
|
LANGID LangID,
|
|
DWORD cbTitle,
|
|
LPTSTR lpTitle,
|
|
BOOL Help
|
|
)
|
|
{
|
|
LPWSTR lpTitleFound;
|
|
NTSTATUS Status;
|
|
BOOL bGetTextSuccess = TRUE ;
|
|
|
|
DBG_UNREFERENCED_PARAMETER(LangID);
|
|
|
|
if (Help && pSysInfo->CounterInfo.dwHelpSize == 0) {
|
|
// we have not get the help text yet, go get it
|
|
bGetTextSuccess = GetHelpText (pSysInfo);
|
|
}
|
|
|
|
if (!bGetTextSuccess) {
|
|
Status = ERROR_INVALID_NAME;
|
|
goto ErrorExit;
|
|
}
|
|
|
|
if ((dwTitleIndex > 0) && (dwTitleIndex <= pSysInfo->CounterInfo.dwLastId)) {
|
|
// then title should be found in the array
|
|
lpTitleFound = pSysInfo->CounterInfo.TextString[dwTitleIndex];
|
|
if (!lpTitleFound) {
|
|
// no entry for this index
|
|
Status = ERROR_INVALID_NAME;
|
|
} else if ((DWORD)lstrlen(lpTitleFound) < cbTitle) {
|
|
lstrcpy (lpTitle, lpTitleFound);
|
|
return (ERROR_SUCCESS);
|
|
} else {
|
|
Status = ERROR_MORE_DATA;
|
|
}
|
|
} else {
|
|
|
|
Status = ERROR_INVALID_NAME;
|
|
}
|
|
|
|
ErrorExit:
|
|
// if here, then an error occured, so return a blank
|
|
|
|
if ((DWORD)lstrlen (NULL_NAME) < cbTitle) {
|
|
lstrcpy (lpTitle, NULL_NAME);
|
|
}
|
|
|
|
return Status; // title not returned
|
|
|
|
}
|
|
|
|
|
|
LONG
|
|
GetSystemPerfData (
|
|
IN HKEY hKeySystem,
|
|
IN LPTSTR lpszValue,
|
|
OUT PPERFDATA pPerfData,
|
|
OUT PDWORD pdwPerfDataLen
|
|
)
|
|
{
|
|
LONG lError ;
|
|
DWORD Type ;
|
|
|
|
// have to pass in a Type to RegQueryValueEx(W) or else it
|
|
// will crash
|
|
lError = RegQueryValueEx (hKeySystem, lpszValue, NULL, &Type,
|
|
(LPSTR) pPerfData, pdwPerfDataLen) ;
|
|
return (lError) ;
|
|
}
|
|
|
|
|
|
BOOL
|
|
CloseSystemPerfData (
|
|
HKEY hKeySystem
|
|
)
|
|
{
|
|
return (TRUE) ;
|
|
}
|
|
|
|
|
|
int
|
|
CBLoadObjects (
|
|
HWND hWndCB,
|
|
PPERFDATA pPerfData,
|
|
PPERFSYSTEM pSysInfo,
|
|
DWORD dwDetailLevel,
|
|
LPTSTR lpszDefaultObject,
|
|
BOOL bIncludeAll
|
|
)
|
|
/*
|
|
Effect: Load into the combo box CB one item for each Object in
|
|
pPerfData. For each item, look up the object's name in
|
|
the registry strings associated with pSysInfo, and
|
|
attach the object to the data field of the CB item.
|
|
|
|
Dont add those objects that are more detailed than
|
|
dwDetailLevel.
|
|
|
|
Set the current selected CB item to the object named
|
|
lpszDefaultObject, or to the default object specified in
|
|
pPerfData if lpszDefaultObject is NULL.
|
|
*/
|
|
{
|
|
UINT i ;
|
|
INT_PTR iIndex ;
|
|
PPERFOBJECT pObject ;
|
|
TCHAR szObject [PerfObjectLen + 1] ;
|
|
TCHAR szDefaultObject [PerfObjectLen + 1] ;
|
|
|
|
CBReset (hWndCB) ;
|
|
strclr (szDefaultObject) ;
|
|
|
|
pObject = FirstObject (pPerfData) ;
|
|
|
|
for (i = 0, pObject = FirstObject (pPerfData) ;
|
|
i < pPerfData->NumObjectTypes ;
|
|
i++, pObject = NextObject (pObject)) { // for
|
|
if (pObject->DetailLevel <= dwDetailLevel) { // if
|
|
strclr (szObject) ;
|
|
QueryPerformanceName (pSysInfo, pObject->ObjectNameTitleIndex,
|
|
0, PerfObjectLen, szObject, FALSE) ;
|
|
|
|
// if szObject not empty, add it to the Combo-box
|
|
if (!strsame(szObject, NULL_NAME)) {
|
|
iIndex = CBAdd (hWndCB, szObject) ;
|
|
CBSetData (hWndCB, iIndex, (DWORD_PTR) pObject) ;
|
|
|
|
if ((LONG)pObject->ObjectNameTitleIndex == pPerfData->DefaultObject)
|
|
lstrcpy (szDefaultObject, szObject) ;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
if (bIncludeAll) {
|
|
StringLoad (IDS_ALLOBJECTS, szObject) ;
|
|
CBInsert (hWndCB, 0, szObject) ;
|
|
// assume "ALL" is default unless overridden
|
|
lstrcpy (szDefaultObject, szObject) ;
|
|
}
|
|
|
|
if (lpszDefaultObject)
|
|
lstrcpy (szDefaultObject, lpszDefaultObject) ;
|
|
|
|
iIndex = CBFind (hWndCB, szDefaultObject) ;
|
|
CBSetSelection (hWndCB, (iIndex != CB_ERR) ? iIndex : 0) ;
|
|
|
|
return (i) ;
|
|
}
|
|
|
|
|
|
int
|
|
LBLoadObjects (
|
|
HWND hWndLB,
|
|
PPERFDATA pPerfData,
|
|
PPERFSYSTEM pSysInfo,
|
|
DWORD dwDetailLevel,
|
|
LPTSTR lpszDefaultObject,
|
|
BOOL bIncludeAll
|
|
)
|
|
/*
|
|
Effect: Load into the list box LB one item for each Object in
|
|
pPerfData. For each item, look up the object's name in
|
|
the registry strings associated with pSysInfo, and
|
|
attach the object to the data field of the LB item.
|
|
|
|
Dont add those objects that are more detailed than
|
|
dwDetailLevel.
|
|
|
|
Set the current selected LB item to the object named
|
|
lpszDefaultObject, or to the default object specified in
|
|
pPerfData if lpszDefaultObject is NULL.
|
|
*/
|
|
{
|
|
UINT i ;
|
|
INT_PTR iIndex = 0;
|
|
PPERFOBJECT pObject ;
|
|
TCHAR szObject [PerfObjectLen + 1] ;
|
|
TCHAR szDefaultObject [PerfObjectLen + 1] ;
|
|
|
|
LBReset (hWndLB) ;
|
|
strclr (szDefaultObject) ;
|
|
|
|
pObject = FirstObject (pPerfData) ;
|
|
|
|
for (i = 0, pObject = FirstObject (pPerfData) ;
|
|
i < pPerfData->NumObjectTypes ;
|
|
i++, pObject = NextObject (pObject)) {
|
|
if (pObject->DetailLevel <= dwDetailLevel) {
|
|
strclr (szObject) ;
|
|
QueryPerformanceName (pSysInfo, pObject->ObjectNameTitleIndex,
|
|
0, PerfObjectLen, szObject, FALSE) ;
|
|
|
|
// if szObject is not empty, add it to the listbox
|
|
if (!strsame(szObject, NULL_NAME)) {
|
|
iIndex = LBAdd (hWndLB, szObject) ;
|
|
LBSetData (hWndLB, iIndex, (DWORD_PTR) pObject) ;
|
|
|
|
if ((LONG)pObject->ObjectNameTitleIndex == pPerfData->DefaultObject)
|
|
lstrcpy (szDefaultObject, szObject) ;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
if (bIncludeAll) {
|
|
StringLoad (IDS_ALLOBJECTS, szObject) ;
|
|
LBInsert (hWndLB, 0, szObject) ;
|
|
LBSetData (hWndLB, iIndex, 0) ;
|
|
// assume "ALL" is default unless overridden
|
|
lstrcpy (szDefaultObject, szObject) ;
|
|
}
|
|
|
|
if (lpszDefaultObject)
|
|
lstrcpy (szDefaultObject, lpszDefaultObject) ;
|
|
|
|
iIndex = LBFind (hWndLB, szDefaultObject) ;
|
|
LBSetSelection (hWndLB, (iIndex != LB_ERR) ? iIndex : 0) ;
|
|
|
|
return (i) ;
|
|
}
|
|
|
|
|
|
/***************************************************************************\
|
|
* GetObjectDef()
|
|
*
|
|
* Entry: pointer to data block and the number of the object type
|
|
* Exit: returns a pointer to the specified object type definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFOBJECT
|
|
GetObjectDef(
|
|
PPERFDATA pDataBlock,
|
|
DWORD NumObjectType
|
|
)
|
|
{
|
|
DWORD NumTypeDef;
|
|
|
|
PPERFOBJECT pObjectDef;
|
|
|
|
pObjectDef = FirstObject(pDataBlock);
|
|
|
|
for ( NumTypeDef = 0;
|
|
NumTypeDef < pDataBlock->NumObjectTypes;
|
|
NumTypeDef++ ) {
|
|
|
|
if ( NumTypeDef == NumObjectType ) {
|
|
|
|
return pObjectDef;
|
|
}
|
|
pObjectDef = NextObject(pObjectDef);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetObjectDefByTitleIndex()
|
|
*
|
|
* Entry: pointer to data block and the title index of the object type
|
|
* Exit: returns a pointer to the specified object type definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFOBJECT
|
|
GetObjectDefByTitleIndex(
|
|
PPERFDATA pDataBlock,
|
|
DWORD ObjectTypeTitleIndex
|
|
)
|
|
{
|
|
DWORD NumTypeDef;
|
|
|
|
PPERFOBJECT pObjectDef;
|
|
|
|
pObjectDef = FirstObject(pDataBlock);
|
|
|
|
for ( NumTypeDef = 0;
|
|
NumTypeDef < pDataBlock->NumObjectTypes;
|
|
NumTypeDef++ ) {
|
|
|
|
if ( pObjectDef->ObjectNameTitleIndex == ObjectTypeTitleIndex ) {
|
|
|
|
return pObjectDef;
|
|
}
|
|
pObjectDef = NextObject(pObjectDef);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetObjectDefByName()
|
|
*
|
|
* Entry: pointer to data block and the name of the object type
|
|
* Exit: returns a pointer to the specified object type definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFOBJECT
|
|
GetObjectDefByName(
|
|
PPERFSYSTEM pSystem,
|
|
PPERFDATA pDataBlock,
|
|
LPTSTR pObjectName
|
|
)
|
|
{
|
|
DWORD NumTypeDef;
|
|
TCHAR szObjectName [PerfObjectLen + 1] ;
|
|
|
|
PPERFOBJECT pObjectDef;
|
|
|
|
pObjectDef = FirstObject(pDataBlock);
|
|
for ( NumTypeDef = 0;
|
|
NumTypeDef < pDataBlock->NumObjectTypes;
|
|
NumTypeDef++ ) {
|
|
|
|
ObjectName (pSystem, pObjectDef, szObjectName, PerfObjectLen) ;
|
|
if (strsame (szObjectName, pObjectName) ) {
|
|
|
|
return pObjectDef;
|
|
}
|
|
pObjectDef = NextObject(pObjectDef);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetObjectIdByName()
|
|
*
|
|
* Entry: pointer to data block and the name of the object type
|
|
* Exit: returns the Object title index for the specified Object Name
|
|
*
|
|
\***************************************************************************/
|
|
|
|
DWORD
|
|
GetObjectIdByName(
|
|
PPERFSYSTEM pSystem,
|
|
PPERFDATA pDataBlock,
|
|
LPTSTR pObjectName
|
|
)
|
|
{
|
|
DWORD NumTypeDef;
|
|
TCHAR szObjectName [PerfObjectLen + 1] ;
|
|
|
|
PPERFOBJECT pObjectDef;
|
|
|
|
pObjectDef = FirstObject(pDataBlock);
|
|
for ( NumTypeDef = 0;
|
|
NumTypeDef < pDataBlock->NumObjectTypes;
|
|
NumTypeDef++ ) {
|
|
|
|
ObjectName (pSystem, pObjectDef, szObjectName, PerfObjectLen) ;
|
|
if (strsame (szObjectName, pObjectName) ) {
|
|
|
|
return pObjectDef->ObjectNameTitleIndex;
|
|
}
|
|
pObjectDef = NextObject(pObjectDef);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
/***************************************************************************\
|
|
* GetCounterDef()
|
|
*
|
|
* Entry: pointer to object type definition the number of the Counter
|
|
* definition
|
|
* Exit: returns a pointer to the specified Counter definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFCOUNTERDEF
|
|
GetCounterDef(
|
|
PPERFOBJECT pObjectDef,
|
|
DWORD NumCounter
|
|
)
|
|
{
|
|
DWORD NumCtrDef;
|
|
|
|
PPERFCOUNTERDEF pCounterDef;
|
|
|
|
pCounterDef = FirstCounter(pObjectDef);
|
|
|
|
for ( NumCtrDef = 0;
|
|
NumCtrDef < pObjectDef->NumCounters;
|
|
NumCtrDef++ ) {
|
|
|
|
if ( NumCtrDef == NumCounter ) {
|
|
|
|
return pCounterDef;
|
|
}
|
|
pCounterDef = NextCounter(pCounterDef);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetCounterNumByTitleIndex()
|
|
*
|
|
* Entry: pointer to object type definition and the title index of
|
|
* the name of the Counter definition
|
|
* Exit: returns the number of the specified Counter definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
LONG
|
|
GetCounterNumByTitleIndex(
|
|
PPERFOBJECT pObjectDef,
|
|
DWORD CounterTitleIndex
|
|
)
|
|
{
|
|
DWORD NumCtrDef;
|
|
|
|
PPERFCOUNTERDEF pCounterDef;
|
|
|
|
pCounterDef = FirstCounter(pObjectDef);
|
|
|
|
for ( NumCtrDef = 0;
|
|
NumCtrDef < pObjectDef->NumCounters;
|
|
NumCtrDef++ ) {
|
|
|
|
if ( pCounterDef->CounterNameTitleIndex == CounterTitleIndex ) {
|
|
|
|
return NumCtrDef;
|
|
}
|
|
pCounterDef = NextCounter(pCounterDef);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetCounterData()
|
|
*
|
|
* Entry: pointer to object definition and number of counter, must be
|
|
* an object with no instances
|
|
* Exit: returns a pointer to the data
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PVOID
|
|
GetCounterData(
|
|
PPERFOBJECT pObjectDef,
|
|
PPERFCOUNTERDEF pCounterDef
|
|
)
|
|
{
|
|
|
|
PERF_COUNTER_BLOCK *pCtrBlock;
|
|
|
|
pCtrBlock = (PERF_COUNTER_BLOCK *)((PCHAR)pObjectDef +
|
|
pObjectDef->DefinitionLength);
|
|
|
|
return (PVOID)((PCHAR)pCtrBlock + pCounterDef->CounterOffset);
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetInstanceCounterData()
|
|
*
|
|
* Entry: pointer to object definition and number of counter, and a pointer
|
|
* to the instance for which the data is to be retrieved
|
|
* Exit: returns a pointer to the data
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PVOID
|
|
GetInstanceCounterData(
|
|
PPERFOBJECT pObjectDef,
|
|
PPERFINSTANCEDEF pInstanceDef,
|
|
PPERFCOUNTERDEF pCounterDef
|
|
)
|
|
{
|
|
|
|
PERF_COUNTER_BLOCK *pCtrBlock;
|
|
|
|
pCtrBlock = (PERF_COUNTER_BLOCK *)((PCHAR)pInstanceDef +
|
|
pInstanceDef->ByteLength);
|
|
|
|
return (PVOID)((PCHAR)pCtrBlock + pCounterDef->CounterOffset);
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetNextInstance()
|
|
*
|
|
* Entry: pointer to instance definition
|
|
* Exit: returns a pointer to the next instance definition. If none,
|
|
* points to byte past this instance
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFINSTANCEDEF
|
|
GetNextInstance(
|
|
PPERFINSTANCEDEF pInstDef
|
|
)
|
|
{
|
|
PERF_COUNTER_BLOCK *pCtrBlock;
|
|
|
|
pCtrBlock = (PERF_COUNTER_BLOCK *)
|
|
((PCHAR) pInstDef + pInstDef->ByteLength);
|
|
|
|
return (PPERFINSTANCEDEF )
|
|
((PCHAR) pCtrBlock + pCtrBlock->ByteLength);
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetInstance()
|
|
*
|
|
* Entry: pointer to object type definition, the name of the instance,
|
|
* the name of the parent object type, and the parent instance index.
|
|
* The name of the parent object type is NULL if no parent.
|
|
* Exit: returns a pointer to the specified instance definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFINSTANCEDEF
|
|
GetInstance(
|
|
PPERFOBJECT pObjectDef,
|
|
LONG InstanceNumber
|
|
)
|
|
{
|
|
|
|
PPERFINSTANCEDEF pInstanceDef;
|
|
LONG NumInstance;
|
|
|
|
if (!pObjectDef) {
|
|
return 0;
|
|
}
|
|
|
|
pInstanceDef = FirstInstance(pObjectDef);
|
|
|
|
for ( NumInstance = 0;
|
|
NumInstance < pObjectDef->NumInstances;
|
|
NumInstance++ ) {
|
|
if ( InstanceNumber == NumInstance ) {
|
|
return pInstanceDef;
|
|
}
|
|
pInstanceDef = GetNextInstance(pInstanceDef);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetInstanceByUniqueID()
|
|
*
|
|
* Entry: pointer to object type definition, and
|
|
* the unique ID of the instance.
|
|
* Exit: returns a pointer to the specified instance definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFINSTANCEDEF
|
|
GetInstanceByUniqueID(
|
|
PPERFOBJECT pObjectDef,
|
|
LONG UniqueID,
|
|
DWORD dwIndex
|
|
)
|
|
{
|
|
|
|
PPERFINSTANCEDEF pInstanceDef;
|
|
DWORD dwLocalIndex;
|
|
|
|
LONG NumInstance;
|
|
|
|
pInstanceDef = FirstInstance(pObjectDef);
|
|
dwLocalIndex = dwIndex;
|
|
|
|
for ( NumInstance = 0;
|
|
NumInstance < pObjectDef->NumInstances;
|
|
NumInstance++ ) {
|
|
|
|
if ( pInstanceDef->UniqueID == UniqueID ) {
|
|
if (dwLocalIndex == 0) {
|
|
return pInstanceDef;
|
|
} else {
|
|
--dwLocalIndex;
|
|
}
|
|
}
|
|
pInstanceDef = GetNextInstance(pInstanceDef);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
/***************************************************************************\
|
|
* GetInstanceByNameUsingParentTitleIndex()
|
|
*
|
|
* Entry: pointer to object type definition, the name of the instance,
|
|
* and the name of the parent instance.
|
|
* The name of the parent instance is NULL if no parent.
|
|
* Exit: returns a pointer to the specified instance definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFINSTANCEDEF
|
|
GetInstanceByNameUsingParentTitleIndex(
|
|
PPERFDATA pDataBlock,
|
|
PPERFOBJECT pObjectDef,
|
|
LPTSTR pInstanceName,
|
|
LPTSTR pParentName,
|
|
DWORD dwIndex
|
|
)
|
|
{
|
|
BOOL fHaveParent;
|
|
PPERFOBJECT pParentObj;
|
|
|
|
PPERFINSTANCEDEF pParentInst,
|
|
pInstanceDef;
|
|
|
|
LONG NumInstance;
|
|
TCHAR InstanceName[256];
|
|
DWORD dwLocalIndex;
|
|
|
|
|
|
fHaveParent = FALSE;
|
|
pInstanceDef = FirstInstance(pObjectDef);
|
|
dwLocalIndex = dwIndex;
|
|
|
|
memset(InstanceName, 0, sizeof(TCHAR) * 256);
|
|
for ( NumInstance = 0;
|
|
NumInstance < pObjectDef->NumInstances;
|
|
NumInstance++ ) {
|
|
|
|
GetInstanceNameStr(pInstanceDef, InstanceName, pObjectDef->CodePage);
|
|
if ( lstrcmpi(InstanceName, pInstanceName) == 0 ) {
|
|
|
|
// Instance name matches
|
|
|
|
if ( pParentName == NULL ) {
|
|
|
|
// No parent, we're done if this is the right "copy"
|
|
|
|
if (dwLocalIndex == 0) {
|
|
return pInstanceDef;
|
|
} else {
|
|
--dwLocalIndex;
|
|
}
|
|
|
|
} else {
|
|
|
|
// Must match parent as well
|
|
|
|
pParentObj = GetObjectDefByTitleIndex(
|
|
pDataBlock,
|
|
pInstanceDef->ParentObjectTitleIndex);
|
|
|
|
if (!pParentObj) {
|
|
// can't locate the parent, forget it
|
|
break ;
|
|
}
|
|
|
|
// Object type of parent found; now find parent
|
|
// instance
|
|
|
|
pParentInst = GetInstance(pParentObj,
|
|
pInstanceDef->ParentObjectInstance);
|
|
|
|
if (!pParentInst) {
|
|
// can't locate the parent instance, forget it
|
|
break ;
|
|
}
|
|
|
|
GetInstanceNameStr(pParentInst,InstanceName, pParentObj->CodePage);
|
|
if ( lstrcmpi(InstanceName, pParentName) == 0 ) {
|
|
|
|
// Parent Instance Name matches that passed in
|
|
if (dwLocalIndex == 0) {
|
|
return pInstanceDef;
|
|
} else {
|
|
--dwLocalIndex;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
pInstanceDef = GetNextInstance(pInstanceDef);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* GetInstanceByName()
|
|
*
|
|
* Entry: pointer to object type definition, the name of the instance,
|
|
* and the name of the parent instance.
|
|
* The name of the parent instance is NULL if no parent.
|
|
* Exit: returns a pointer to the specified instance definition
|
|
*
|
|
\***************************************************************************/
|
|
|
|
PPERFINSTANCEDEF
|
|
GetInstanceByName(
|
|
PPERFDATA pDataBlock,
|
|
PPERFOBJECT pObjectDef,
|
|
LPTSTR pInstanceName,
|
|
LPTSTR pParentName,
|
|
DWORD dwIndex
|
|
)
|
|
{
|
|
BOOL fHaveParent;
|
|
|
|
PPERFOBJECT pParentObj;
|
|
|
|
PPERFINSTANCEDEF pParentInst,
|
|
pInstanceDef;
|
|
|
|
LONG NumInstance;
|
|
TCHAR InstanceName[256];
|
|
DWORD dwLocalIndex;
|
|
|
|
fHaveParent = FALSE;
|
|
pInstanceDef = FirstInstance(pObjectDef);
|
|
dwLocalIndex = dwIndex;
|
|
|
|
memset(InstanceName, 0, sizeof(TCHAR) * 256);
|
|
for ( NumInstance = 0;
|
|
NumInstance < pObjectDef->NumInstances;
|
|
NumInstance++ ) {
|
|
|
|
GetInstanceNameStr(pInstanceDef,InstanceName, pObjectDef->CodePage);
|
|
if ( lstrcmpi(InstanceName, pInstanceName) == 0 ) {
|
|
|
|
// Instance name matches
|
|
|
|
if ( !pInstanceDef->ParentObjectTitleIndex ) {
|
|
|
|
// No parent, we're done
|
|
|
|
if (dwLocalIndex == 0) {
|
|
return pInstanceDef;
|
|
} else {
|
|
--dwLocalIndex;
|
|
}
|
|
|
|
} else {
|
|
|
|
// Must match parent as well
|
|
|
|
pParentObj = GetObjectDefByTitleIndex(
|
|
pDataBlock,
|
|
pInstanceDef->ParentObjectTitleIndex);
|
|
|
|
// Object type of parent found; now find parent
|
|
// instance
|
|
|
|
if (pParentObj == NULL)
|
|
continue;
|
|
pParentInst = GetInstance(pParentObj,
|
|
pInstanceDef->ParentObjectInstance);
|
|
|
|
GetInstanceNameStr(pParentInst,InstanceName, pParentObj->CodePage);
|
|
if ( lstrcmpi(InstanceName, pParentName) == 0 ) {
|
|
// Parent Instance Name matches that passed in
|
|
|
|
if (dwLocalIndex == 0) {
|
|
return pInstanceDef;
|
|
} else {
|
|
--dwLocalIndex;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
pInstanceDef = GetNextInstance(pInstanceDef);
|
|
}
|
|
return 0;
|
|
} // GetInstanceByName
|
|
|
|
|
|
BOOL
|
|
FailedLineData (
|
|
PPERFDATA pPerfData,
|
|
PLINE pLine
|
|
)
|
|
/*
|
|
This routine handles the case where there is no data for a
|
|
system.
|
|
*/
|
|
{
|
|
LARGE_INTEGER liDummy ;
|
|
|
|
// System no longer exists.
|
|
liDummy.LowPart = liDummy.HighPart = 0;
|
|
if (pLine->lnCounterType == PERF_COUNTER_TIMER_INV) {
|
|
// Timer inverse with Performance Counter as timer
|
|
pLine->lnaOldCounterValue[0] = pLine->lnOldTime ;
|
|
pLine->lnaCounterValue[0] = pLine->lnNewTime ;
|
|
} else if (pLine->lnCounterType == PERF_100NSEC_TIMER_INV ||
|
|
pLine->lnCounterType == PERF_100NSEC_MULTI_TIMER_INV) {
|
|
// Timer inverse with System Time as timer
|
|
pLine->lnaOldCounterValue[0] = pLine->lnOldTime100Ns ;
|
|
pLine->lnaCounterValue[0] = pLine->lnNewTime100Ns ;
|
|
} else {
|
|
// Normal timer
|
|
pLine->lnaOldCounterValue[0] =
|
|
pLine->lnaCounterValue[0] =
|
|
pLine->lnaOldCounterValue[1] =
|
|
pLine->lnaCounterValue[1] = liDummy ;
|
|
}
|
|
return TRUE ;
|
|
|
|
}
|
|
|
|
|
|
BOOL
|
|
UpdateLineData (
|
|
PPERFDATA pPerfData,
|
|
PLINE pLine,
|
|
PPERFSYSTEM pSystem
|
|
)
|
|
/*
|
|
Assert: pPerfData holds the performance data for the same
|
|
system as pLine.
|
|
*/
|
|
{
|
|
PPERFOBJECT pObject ;
|
|
PPERFINSTANCEDEF pInstanceDef ;
|
|
PPERFCOUNTERDEF pCounterDef ;
|
|
PPERFCOUNTERDEF pCounterDef2 ;
|
|
PDWORD pCounterValue ;
|
|
PDWORD pCounterValue2 = NULL;
|
|
UINT iCounterIndex ;
|
|
LARGE_INTEGER liDummy[2] ;
|
|
|
|
// Use Object time units if available, otherwise use system
|
|
// performance timer
|
|
|
|
pLine->lnOldTime = pLine->lnNewTime;
|
|
|
|
pLine->lnOldTime100Ns = pLine->lnNewTime100Ns;
|
|
pLine->lnNewTime100Ns = pPerfData->PerfTime100nSec;
|
|
|
|
pLine->lnPerfFreq = pPerfData->PerfFreq ;
|
|
|
|
if ((pLine->lnObject.TotalByteLength == 0) && !(PlayingBackLog())) {
|
|
// this is the case when openning a setting file and the remote
|
|
// system is not up at that time. We have all the names but no
|
|
// pObject, pCounter, etc. So, we have to re-built the linedata.
|
|
PPERFOBJECT pObject ;
|
|
PPERFCOUNTERDEF pCounter ;
|
|
PPERFINSTANCEDEF pInstance ;
|
|
|
|
pObject = LineFindObject (pSystem, pPerfData, pLine) ;
|
|
if (!pObject) {
|
|
//Something wrong, this object is still not there...
|
|
return FALSE ;
|
|
}
|
|
pCounter = LineFindCounter (pSystem, pObject, pPerfData, pLine) ;
|
|
if (!pCounter) {
|
|
return FALSE ;
|
|
}
|
|
if (pObject &&
|
|
pLine->lnObject.NumInstances > 0 &&
|
|
pLine->lnInstanceName == NULL) {
|
|
return FALSE ;
|
|
}
|
|
pInstance = LineFindInstance (pPerfData, pObject, pLine) ;
|
|
if (!pInstance) {
|
|
if (pLine->lnParentObjName) {
|
|
MemoryFree (pLine->lnParentObjName) ;
|
|
pLine->lnParentObjName = NULL ;
|
|
}
|
|
}
|
|
|
|
pLine->lnCounterType = pCounter->CounterType;
|
|
pLine->lnCounterLength = pCounter->CounterSize;
|
|
|
|
if (pSystem->lpszValue && strsame (pSystem->lpszValue, L"Global")) {
|
|
DWORD dwBufferSize = MemorySize (pSystem->lpszValue) ;
|
|
memset (pSystem->lpszValue, 0, dwBufferSize) ;
|
|
}
|
|
|
|
AppendObjectToValueList (
|
|
pLine->lnObject.ObjectNameTitleIndex,
|
|
pSystem->lpszValue);
|
|
|
|
|
|
}
|
|
|
|
pObject = GetObjectDefByTitleIndex(
|
|
pPerfData,
|
|
pLine->lnObject.ObjectNameTitleIndex);
|
|
|
|
if (!pObject) {
|
|
// Object Type no longer exists. This is possible if we are
|
|
// looking at a log file which has not always collected all
|
|
// the same data, such as appending measurements of different
|
|
// object types.
|
|
|
|
pCounterValue =
|
|
pCounterValue2 = (PDWORD) liDummy;
|
|
liDummy[0].QuadPart = 0;
|
|
|
|
|
|
pLine->lnNewTime = pPerfData->PerfTime;
|
|
|
|
if (pLine->lnCounterType == PERF_COUNTER_TIMER_INV) {
|
|
// Timer inverse with Performance Counter as timer
|
|
pLine->lnaOldCounterValue[0] = pLine->lnOldTime ;
|
|
pLine->lnaCounterValue[0] = pLine->lnNewTime ;
|
|
} else if (pLine->lnCounterType == PERF_100NSEC_TIMER_INV ||
|
|
pLine->lnCounterType == PERF_100NSEC_MULTI_TIMER_INV) {
|
|
// Timer inverse with System Time as timer
|
|
pLine->lnaOldCounterValue[0] = pLine->lnOldTime100Ns ;
|
|
pLine->lnaCounterValue[0] = pLine->lnNewTime100Ns ;
|
|
} else {
|
|
// Normal timer or counter
|
|
pLine->lnaOldCounterValue[0] =
|
|
pLine->lnaCounterValue[0] =
|
|
pLine->lnaOldCounterValue[1] =
|
|
pLine->lnaCounterValue[1] = liDummy[0] ;
|
|
}
|
|
return TRUE ;
|
|
} else {
|
|
pCounterDef = &pLine->lnCounterDef ;
|
|
|
|
// if (pObject->PerfFreq.QuadPart > 0) {
|
|
if (pCounterDef->CounterType & PERF_OBJECT_TIMER) {
|
|
pLine->lnNewTime = pObject->PerfTime;
|
|
} else {
|
|
pLine->lnNewTime = pPerfData->PerfTime;
|
|
}
|
|
|
|
iCounterIndex = CounterIndex (pCounterDef, pObject) ;
|
|
|
|
// Get second counter, only if we are not at
|
|
// the end of the counters; some computations
|
|
// require a second counter
|
|
|
|
if (iCounterIndex < pObject->NumCounters-1 && iCounterIndex != -1) {
|
|
pCounterDef2 = GetCounterDef(pObject, iCounterIndex+1);
|
|
} else {
|
|
pCounterDef2 = NULL;
|
|
}
|
|
|
|
if (pObject->NumInstances > 0) {
|
|
|
|
if ( pLine->lnUniqueID != PERF_NO_UNIQUE_ID ) {
|
|
pInstanceDef = GetInstanceByUniqueID(pObject,
|
|
pLine->lnUniqueID,
|
|
pLine->dwInstanceIndex);
|
|
} else {
|
|
|
|
pInstanceDef =
|
|
GetInstanceByNameUsingParentTitleIndex(
|
|
pPerfData,
|
|
pObject,
|
|
pLine->lnInstanceName,
|
|
pLine->lnPINName,
|
|
pLine->dwInstanceIndex);
|
|
}
|
|
|
|
if (pInstanceDef) {
|
|
pLine->lnInstanceDef = *pInstanceDef;
|
|
pCounterValue = GetInstanceCounterData(pObject,
|
|
pInstanceDef,
|
|
pCounterDef);
|
|
if ( pCounterDef2 ) {
|
|
pCounterValue2 = GetInstanceCounterData(pObject,
|
|
pInstanceDef,
|
|
pCounterDef2);
|
|
}
|
|
} else {
|
|
pCounterValue =
|
|
pCounterValue2 = (PDWORD) liDummy;
|
|
liDummy[0].QuadPart = 0;
|
|
liDummy[1].QuadPart = 0;
|
|
}
|
|
|
|
// Got everything...
|
|
|
|
} // instances exist, look at them for counter blocks
|
|
|
|
else {
|
|
pCounterValue = GetCounterData(pObject, pCounterDef);
|
|
if (pCounterDef2) {
|
|
pCounterValue2 = GetCounterData(pObject, pCounterDef2);
|
|
}
|
|
|
|
} // counter def search when no instances
|
|
}
|
|
|
|
pLine->lnaOldCounterValue[0] = pLine->lnaCounterValue[0] ;
|
|
|
|
if (pLine->lnCounterLength <= 4) {
|
|
// HighPart was initialize to 0
|
|
pLine->lnaCounterValue[0].HighPart = 0;
|
|
pLine->lnaCounterValue[0].LowPart = *pCounterValue;
|
|
} else {
|
|
pLine->lnaCounterValue[0] = *(LARGE_INTEGER UNALIGNED *) pCounterValue;
|
|
}
|
|
|
|
// Get second counter, only if we are not at
|
|
// the end of the counters; some computations
|
|
// require a second counter
|
|
|
|
if ( pCounterDef2 ) {
|
|
pLine->lnaOldCounterValue[1] =
|
|
pLine->lnaCounterValue[1] ;
|
|
if (pCounterDef2->CounterSize <= 4) {
|
|
// HighPart was initialize to 0
|
|
pLine->lnaCounterValue[1].HighPart = 0;
|
|
pLine->lnaCounterValue[1].LowPart = *pCounterValue2;
|
|
} else
|
|
pLine->lnaCounterValue[1] =
|
|
*((LARGE_INTEGER UNALIGNED *) pCounterValue2);
|
|
}
|
|
return (TRUE) ;
|
|
} // UpdateLineData
|
|
|
|
|
|
|
|
BOOL
|
|
UpdateSystemData (
|
|
PPERFSYSTEM pSystem,
|
|
PPERFDATA *ppPerfData
|
|
)
|
|
{
|
|
#define PERF_SYSTEM_TIMEOUT (60L * 1000L)
|
|
long lError ;
|
|
DWORD Status ;
|
|
DWORD Size;
|
|
BOOL TimeToCheck = FALSE ;
|
|
LONGLONG llLastTimeStamp;
|
|
|
|
if (!ppPerfData)
|
|
return (FALSE) ;
|
|
|
|
while (TRUE) {
|
|
if (pSystem->FailureTime) {
|
|
DWORD CurrentTickCount = GetTickCount() ;
|
|
|
|
if (CurrentTickCount < pSystem->FailureTime) {
|
|
// wrap-around case
|
|
if (CurrentTickCount >= PERF_SYSTEM_TIMEOUT)
|
|
TimeToCheck = TRUE ;
|
|
else if ( ~pSystem->FailureTime >= PERF_SYSTEM_TIMEOUT)
|
|
TimeToCheck = TRUE ;
|
|
else if (CurrentTickCount + (~pSystem->FailureTime) >= PERF_SYSTEM_TIMEOUT)
|
|
TimeToCheck = TRUE ;
|
|
} else {
|
|
if (CurrentTickCount - pSystem->FailureTime >= PERF_SYSTEM_TIMEOUT)
|
|
TimeToCheck = TRUE ;
|
|
}
|
|
|
|
if (TimeToCheck) {
|
|
// free any memory hanging off this system
|
|
SystemFree (pSystem, FALSE) ;
|
|
|
|
// get the registry info
|
|
pSystem->sysDataKey = OpenSystemPerfData(pSystem->sysName) ;
|
|
|
|
Status = !ERROR_SUCCESS ;
|
|
if (pSystem->sysDataKey) {
|
|
Status = GetSystemNames(pSystem);
|
|
}
|
|
|
|
if (Status != ERROR_SUCCESS) {
|
|
// something wrong in getting the registry info,
|
|
// remote system must be still down (??)
|
|
pSystem->FailureTime = GetTickCount();
|
|
|
|
// Free any memory that may have created
|
|
SystemFree (pSystem, FALSE) ;
|
|
|
|
return (FALSE) ;
|
|
}
|
|
|
|
// time to check again
|
|
pSystem->FailureTime = 0 ;
|
|
} else {
|
|
// not time to check again
|
|
return (FALSE) ;
|
|
}
|
|
}
|
|
|
|
if (pSystem->FailureTime == 0 ) {
|
|
Size = MemorySize ((LPMEMORY)*ppPerfData);
|
|
|
|
// save the last sample timestamp for this system
|
|
|
|
if (pSystem->pSystemPerfData != NULL) {
|
|
llLastTimeStamp =
|
|
pSystem->pSystemPerfData->PerfTime.QuadPart;
|
|
}
|
|
|
|
lError = GetSystemPerfData (pSystem->sysDataKey,
|
|
pSystem->lpszValue,
|
|
*ppPerfData,
|
|
&Size) ;
|
|
if ((!lError) &&
|
|
(Size > 0) &&
|
|
(*ppPerfData)->Signature[0] == (WCHAR)'P' &&
|
|
(*ppPerfData)->Signature[1] == (WCHAR)'E' &&
|
|
(*ppPerfData)->Signature[2] == (WCHAR)'R' &&
|
|
(*ppPerfData)->Signature[3] == (WCHAR)'F' ) {
|
|
if (pSystem->dwSystemState == SYSTEM_OK) {
|
|
if (pSystem->pSystemPerfData != NULL) {
|
|
if (pSystem->pSystemPerfData->PerfTime.QuadPart <
|
|
llLastTimeStamp) {
|
|
if (bReportEvents) {
|
|
// then a system error occured, so log it
|
|
dwMessageDataBytes = 0;
|
|
szMessageArray[wMessageIndex++] = pSystem->sysName;
|
|
ReportEvent (hEventLog,
|
|
EVENTLOG_ERROR_TYPE, // error type
|
|
0, // category (not used)
|
|
(DWORD)PERFMON_ERROR_TIMESTAMP, // event,
|
|
NULL, // SID (not used),
|
|
wMessageIndex, // number of strings
|
|
0, // sizeof raw data
|
|
szMessageArray, // message text array
|
|
NULL); // raw data
|
|
}
|
|
}
|
|
}
|
|
return (TRUE) ;
|
|
} else if (pSystem->dwSystemState == SYSTEM_DOWN ||
|
|
pSystem->dwSystemState == SYSTEM_DOWN_RPT ) {
|
|
if (TimeToCheck && bReportEvents) {
|
|
// then the system just came back so display the message
|
|
wMessageIndex = 0;
|
|
szMessageArray[wMessageIndex++] = pSystem->sysName;
|
|
ReportEvent (hEventLog,
|
|
EVENTLOG_INFORMATION_TYPE, // error type
|
|
0, // category (not used)
|
|
(DWORD)PERFMON_INFO_SYSTEM_RESTORED, // event,
|
|
NULL, // SID (not used),
|
|
wMessageIndex, // number of strings
|
|
0, // sizeof raw data
|
|
szMessageArray, // message text array
|
|
NULL); // raw data
|
|
}
|
|
pSystem->dwSystemState = SYSTEM_RECONNECT ;
|
|
} else if (pSystem->dwSystemState == SYSTEM_RECONNECT_RPT) {
|
|
pSystem->dwSystemState = SYSTEM_OK ;
|
|
}
|
|
// for SYSTEM_RECONNECT case, we want to wait for Alert
|
|
// view to report the re-connection first
|
|
return (TRUE) ;
|
|
}
|
|
|
|
if (lError == ERROR_MORE_DATA) {
|
|
Size = MemorySize ((LPMEMORY)*ppPerfData) + dwPerfDataIncrease;
|
|
*ppPerfData = MemoryResize ((LPMEMORY)*ppPerfData, Size);
|
|
|
|
if (!*ppPerfData) {
|
|
if (pSystem->dwSystemState != SYSTEM_DOWN_RPT) {
|
|
if (bReportEvents) {
|
|
// then the system just came back so display the message
|
|
wMessageIndex = 0;
|
|
dwMessageDataBytes = 0;
|
|
szMessageArray[wMessageIndex++] = pSystem->sysName;
|
|
dwMessageData[dwMessageDataBytes++] = Size;
|
|
dwMessageData[dwMessageDataBytes++] = GetLastError();
|
|
dwMessageDataBytes *= sizeof(DWORD);
|
|
ReportEvent (hEventLog,
|
|
EVENTLOG_ERROR_TYPE, // error type
|
|
0, // category (not used)
|
|
(DWORD)PERFMON_ERROR_PERF_DATA_ALLOC, // event,
|
|
NULL, // SID (not used),
|
|
wMessageIndex, // number of strings
|
|
dwMessageDataBytes, // sizeof raw data
|
|
szMessageArray, // message text array
|
|
(LPVOID)&dwMessageData[0]); // raw data
|
|
}
|
|
pSystem->dwSystemState = SYSTEM_DOWN ;
|
|
}
|
|
pSystem->FailureTime = GetTickCount();
|
|
return (FALSE) ;
|
|
}
|
|
} else {
|
|
if (pSystem->dwSystemState != SYSTEM_DOWN_RPT) {
|
|
if (bReportEvents) {
|
|
// then the system just came back so display the message
|
|
wMessageIndex = 0;
|
|
dwMessageDataBytes = 0;
|
|
szMessageArray[wMessageIndex++] = pSystem->sysName;
|
|
dwMessageData[dwMessageDataBytes++] = lError;
|
|
dwMessageDataBytes *= sizeof(DWORD);
|
|
ReportEvent (hEventLog,
|
|
EVENTLOG_WARNING_TYPE, // error type
|
|
0, // category (not used)
|
|
(DWORD)PERFMON_ERROR_SYSTEM_OFFLINE, // event,
|
|
NULL, // SID (not used),
|
|
wMessageIndex, // number of strings
|
|
dwMessageDataBytes, // sizeof raw data
|
|
szMessageArray, // message text array
|
|
(LPVOID)&dwMessageData[0]); // raw data
|
|
}
|
|
pSystem->dwSystemState = SYSTEM_DOWN ;
|
|
}
|
|
pSystem->FailureTime = GetTickCount();
|
|
return (FALSE) ;
|
|
} // else
|
|
} // if
|
|
} // while
|
|
}
|
|
|
|
|
|
|
|
BOOL
|
|
FailedLinesForSystem (
|
|
LPTSTR lpszSystem,
|
|
PPERFDATA pPerfData,
|
|
PLINE pLineFirst
|
|
)
|
|
{
|
|
PLINE pLine ;
|
|
BOOL bMatchFound = FALSE ; // no line from this system
|
|
|
|
for (pLine = pLineFirst; pLine; pLine = pLine->pLineNext) {
|
|
if (strsamei (lpszSystem, pLine->lnSystemName)) {
|
|
FailedLineData (pPerfData, pLine) ;
|
|
if (pLine->bFirstTime) {
|
|
pLine->bFirstTime-- ;
|
|
}
|
|
bMatchFound = TRUE ; // one or more lines from this system
|
|
}
|
|
}
|
|
|
|
return (bMatchFound) ;
|
|
}
|
|
|
|
|
|
BOOL UpdateLinesForSystem (LPTSTR lpszSystem,
|
|
PPERFDATA pPerfData,
|
|
PLINE pLineFirst,
|
|
PPERFSYSTEM pSystem)
|
|
{
|
|
PLINE pLine ;
|
|
BOOL bMatchFound = FALSE ; // no line from this system
|
|
|
|
for (pLine = pLineFirst; pLine; pLine = pLine->pLineNext) {
|
|
if (strsamei (lpszSystem, pLine->lnSystemName)) {
|
|
UpdateLineData (pPerfData, pLine, pSystem) ;
|
|
if (pLine->bFirstTime) {
|
|
pLine->bFirstTime-- ;
|
|
}
|
|
bMatchFound = TRUE ; // one or more lines from this system
|
|
}
|
|
}
|
|
|
|
return (bMatchFound) ;
|
|
}
|
|
|
|
|
|
BOOL
|
|
PerfDataInitializeInstance (void)
|
|
{
|
|
// pGlobalPerfData = MemoryAllocate (STARTING_SYSINFO_SIZE) ;
|
|
// return (pGlobalPerfData != NULL) ;
|
|
return (TRUE) ;
|
|
}
|
|
|
|
NTSTATUS
|
|
AddNamesToArray (
|
|
LPTSTR lpNames,
|
|
DWORD dwLastId,
|
|
LPWSTR *lpCounterId
|
|
)
|
|
{
|
|
LPWSTR lpThisName;
|
|
LPWSTR lpStopString;
|
|
DWORD dwThisCounter;
|
|
NTSTATUS Status = ERROR_SUCCESS;
|
|
|
|
for (lpThisName = lpNames; *lpThisName; ) {
|
|
|
|
// first string should be an integer (in decimal unicode digits)
|
|
dwThisCounter = wcstoul(lpThisName, &lpStopString, 10);
|
|
|
|
if ((dwThisCounter == 0) || (dwThisCounter == ULONG_MAX)) {
|
|
Status += 1;
|
|
goto ADD_BAILOUT; // bad entry
|
|
}
|
|
|
|
// point to corresponding counter name
|
|
|
|
while (*lpThisName++);
|
|
|
|
if (dwThisCounter <= dwLastId) {
|
|
|
|
// and load array element;
|
|
|
|
lpCounterId[dwThisCounter] = lpThisName;
|
|
|
|
}
|
|
|
|
while (*lpThisName++);
|
|
|
|
}
|
|
|
|
ADD_BAILOUT:
|
|
return (Status) ;
|
|
}
|
|
|
|
// try the new way of getting data...
|
|
|
|
BOOL
|
|
UpdateLines (
|
|
PPPERFSYSTEM ppSystemFirst,
|
|
PLINE pLineFirst
|
|
)
|
|
{
|
|
PPERFSYSTEM pSystem ;
|
|
int iNoUseSystemDetected = 0 ;
|
|
int NumberOfSystems = 0 ;
|
|
DWORD WaitStatus ;
|
|
HANDLE *lpPacketHandles ;
|
|
|
|
// allocate the handle array for multiple wait
|
|
if (NumberOfHandles == 0) {
|
|
NumberOfHandles = MAXIMUM_WAIT_OBJECTS ;
|
|
lpHandles = (HANDLE *) MemoryAllocate (NumberOfHandles * sizeof (HANDLE)) ;
|
|
if (!lpHandles) {
|
|
// out of memory, can't go on
|
|
NumberOfHandles = 0 ;
|
|
return FALSE ;
|
|
}
|
|
}
|
|
|
|
for (pSystem = *ppSystemFirst; pSystem; pSystem = pSystem->pSystemNext) {
|
|
|
|
// lock the state data mutex, should be quick unless this thread
|
|
// is still getting data from last time
|
|
if (pSystem->hStateDataMutex == 0)
|
|
continue ;
|
|
|
|
WaitStatus = WaitForSingleObject(pSystem->hStateDataMutex, 100L);
|
|
if (WaitStatus == WAIT_OBJECT_0) {
|
|
ResetEvent (pSystem->hPerfDataEvent) ;
|
|
pSystem->StateData = WAIT_FOR_PERF_DATA ;
|
|
|
|
// add this to the wait
|
|
if (NumberOfSystems >= NumberOfHandles) {
|
|
NumberOfHandles += MAXIMUM_WAIT_OBJECTS ;
|
|
lpHandles = (HANDLE *) MemoryResize (
|
|
lpHandles,
|
|
NumberOfHandles * sizeof (HANDLE)) ;
|
|
if (!lpHandles) {
|
|
// out of memory, can't go on
|
|
NumberOfHandles = 0 ;
|
|
return FALSE ;
|
|
}
|
|
}
|
|
|
|
lpHandles [NumberOfSystems] = pSystem->hPerfDataEvent ;
|
|
NumberOfSystems++ ;
|
|
|
|
// Send Message to thread to take a data sample
|
|
PostThreadMessage (
|
|
pSystem->dwThreadID,
|
|
WM_GET_PERF_DATA,
|
|
(WPARAM)0,
|
|
(LPARAM)0) ;
|
|
|
|
ReleaseMutex(pSystem->hStateDataMutex);
|
|
} else {
|
|
// wait timed out, report error
|
|
if (bReportEvents) {
|
|
wMessageIndex = 0;
|
|
dwMessageDataBytes = 0;
|
|
szMessageArray[wMessageIndex++] = pSystem->sysName;
|
|
ReportEvent (hEventLog,
|
|
EVENTLOG_ERROR_TYPE, // error type
|
|
0, // category (not used)
|
|
(DWORD)PERFMON_ERROR_LOCK_TIMEOUT, // event,
|
|
NULL, // SID (not used),
|
|
wMessageIndex, // number of strings
|
|
0, // sizeof raw data
|
|
szMessageArray, // message text array
|
|
NULL); // raw data
|
|
}
|
|
}
|
|
}
|
|
|
|
Sleep (10); // give the data collection thread a chance to get going
|
|
|
|
// wait for all the data
|
|
if (NumberOfSystems) {
|
|
// increase timeout if we are monitoring lots of systems
|
|
// For every additional 5 systems, add 5 more seconds
|
|
lpPacketHandles = lpHandles ;
|
|
do {
|
|
WaitStatus = WaitForMultipleObjects (
|
|
min (NumberOfSystems, MAXIMUM_WAIT_OBJECTS),
|
|
lpPacketHandles,
|
|
TRUE, // wait for all objects
|
|
DataTimeOut + (NumberOfSystems / 5) * DEFAULT_DATA_TIMEOUT);
|
|
|
|
if (WaitStatus == WAIT_TIMEOUT ||
|
|
NumberOfSystems <= MAXIMUM_WAIT_OBJECTS) {
|
|
//if (WaitStatus == WAIT_TIMEOUT)
|
|
// mike2(TEXT("WaitTimeOut for %ld systems\n"), NumberOfSystems) ;
|
|
|
|
break ;
|
|
}
|
|
|
|
// more systems --> more to wait
|
|
NumberOfSystems -= MAXIMUM_WAIT_OBJECTS ;
|
|
lpPacketHandles += MAXIMUM_WAIT_OBJECTS ;
|
|
} while (TRUE) ;
|
|
|
|
for (pSystem = *ppSystemFirst; pSystem; pSystem = pSystem->pSystemNext) {
|
|
|
|
if (pSystem->hStateDataMutex == 0)
|
|
continue ;
|
|
|
|
// lock the state data mutex
|
|
WaitStatus = WaitForSingleObject(pSystem->hStateDataMutex, 100L);
|
|
if (WaitStatus == WAIT_OBJECT_0) {
|
|
if (pSystem->StateData != PERF_DATA_READY) {
|
|
if (!FailedLinesForSystem (pSystem->sysName,
|
|
pSystem->pSystemPerfData,
|
|
pLineFirst)) {
|
|
if (!bAddLineInProgress) {
|
|
// mark this system as no-longer-needed
|
|
iNoUseSystemDetected++ ;
|
|
pSystem->bSystemNoLongerNeeded = TRUE ;
|
|
}
|
|
}
|
|
} else {
|
|
if (!UpdateLinesForSystem (pSystem->sysName,
|
|
pSystem->pSystemPerfData,
|
|
pLineFirst,
|
|
pSystem)) {
|
|
if (!bAddLineInProgress) {
|
|
// mark this system as no-longer-needed
|
|
iNoUseSystemDetected++ ;
|
|
pSystem->bSystemNoLongerNeeded = TRUE ;
|
|
}
|
|
}
|
|
}
|
|
pSystem->StateData = IDLE_STATE ;
|
|
ReleaseMutex(pSystem->hStateDataMutex);
|
|
} else {
|
|
if (!FailedLinesForSystem (pSystem->sysName,
|
|
pSystem->pSystemPerfData,
|
|
pLineFirst)) {
|
|
if (!bAddLineInProgress) {
|
|
// mark this system as no-longer-needed
|
|
iNoUseSystemDetected++ ;
|
|
pSystem->bSystemNoLongerNeeded = TRUE ;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// check for un-used systems
|
|
if (iNoUseSystemDetected) {
|
|
// some unused system(s) detected.
|
|
DeleteUnusedSystems (ppSystemFirst, iNoUseSystemDetected) ;
|
|
}
|
|
}
|
|
|
|
return (TRUE) ;
|
|
}
|
|
|
|
void
|
|
PerfDataThread (
|
|
PPERFSYSTEM pSystem
|
|
)
|
|
{
|
|
MSG msg ;
|
|
BOOL bGetPerfData ;
|
|
DWORD WaitStatus ;
|
|
|
|
while (GetMessage (&msg, NULL, 0, 0)) {
|
|
if (LOWORD(msg.message) == WM_GET_PERF_DATA) {
|
|
|
|
// this system has been marked as no long used,
|
|
// forget about getting data and continue until
|
|
// we get to the WM_FREE_SYSTEM msg
|
|
if (pSystem->bSystemNoLongerNeeded)
|
|
continue ;
|
|
|
|
bGetPerfData = FALSE ;
|
|
|
|
if (!bAddLineInProgress ||
|
|
(pSystem->lpszValue &&
|
|
!strsame (pSystem->lpszValue, L"Global"))) {
|
|
bGetPerfData = UpdateSystemData (pSystem, &(pSystem->pSystemPerfData)) ;
|
|
}
|
|
|
|
WaitStatus = WaitForSingleObject(pSystem->hStateDataMutex, 1000L);
|
|
if (WaitStatus == WAIT_OBJECT_0) {
|
|
if (pSystem->StateData == WAIT_FOR_PERF_DATA) {
|
|
pSystem->StateData = bGetPerfData ?
|
|
PERF_DATA_READY : PERF_DATA_FAIL ;
|
|
} else {
|
|
//mike2(TEXT("Thread - System = %s, WaitStatus = %d\n"),
|
|
//pSystem->sysName, WaitStatus) ;
|
|
}
|
|
ReleaseMutex(pSystem->hStateDataMutex);
|
|
SetEvent (pSystem->hPerfDataEvent) ;
|
|
}
|
|
} // WM_GET_PERF_DATA MSG
|
|
|
|
else if (LOWORD(msg.message) == WM_FREE_SYSTEM) {
|
|
//mike2(TEXT("Thread - System = %s closing\n"),
|
|
//pSystem->sysName) ;
|
|
// do the memory cleanup during SystemFree stage
|
|
// cleanup all the data collection variables
|
|
if (pSystem->hPerfDataEvent)
|
|
CloseHandle (pSystem->hPerfDataEvent) ;
|
|
|
|
if (pSystem->hStateDataMutex)
|
|
CloseHandle (pSystem->hStateDataMutex) ;
|
|
|
|
if (pSystem->pSystemPerfData)
|
|
MemoryFree ((LPMEMORY)pSystem->pSystemPerfData);
|
|
|
|
if (pSystem->lpszValue) {
|
|
MemoryFree (pSystem->lpszValue);
|
|
pSystem->lpszValue = NULL ;
|
|
}
|
|
|
|
CloseHandle (pSystem->hThread);
|
|
|
|
MemoryFree (pSystem) ;
|
|
break ; // get out of message loop
|
|
} // WM_FREE_SYSTEM MSG
|
|
}
|
|
|
|
ExitThread (TRUE) ;
|
|
}
|