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.
1312 lines
38 KiB
1312 lines
38 KiB
// exctrdlg.cpp : implementation file
|
|
//
|
|
|
|
#ifndef UNICODE
|
|
#define UNICODE 1
|
|
#endif
|
|
#ifndef _UNICODE
|
|
#define _UNICODE 1
|
|
#endif
|
|
|
|
#include "stdafx.h"
|
|
#include "exctrlst.h"
|
|
#include "exctrdlg.h"
|
|
#include "tchar.h"
|
|
|
|
// string constants
|
|
// displayed strings
|
|
static const TCHAR cszNotFound[] = {TEXT("Not Found")};
|
|
static const TCHAR cszNA[] = {TEXT("N/A")};
|
|
// strings that are not displayed
|
|
static const WCHAR cszDisablePerformanceCounters[] = {L"Disable Performance Counters"};
|
|
static const WCHAR cszDefaultLangId[] = {L"009"};
|
|
static const TCHAR cszDoubleBackslash[] = {TEXT("\\\\")};
|
|
static const TCHAR cszSpace[] = {TEXT(" ")};
|
|
static const TCHAR cszSplat[] = {TEXT("*")};
|
|
static const TCHAR cszServIdFmt[] = {TEXT("%d %s")};
|
|
static const TCHAR cszOpen[] = {TEXT("Open")};
|
|
static const TCHAR cszCollect[] = {TEXT("Collect")};
|
|
static const TCHAR cszClose[] = {TEXT("Close")};
|
|
static const TCHAR cszIdFmt[] = {TEXT("0x%8.8x (%d) %s")};
|
|
static const TCHAR cszSortIdFmt[] = {TEXT("0x%8.8x\t%s")};
|
|
static const TCHAR cszTab[] = {TEXT("\t")};
|
|
static const TCHAR cszFirstCounter[] = {TEXT("First Counter")};
|
|
static const TCHAR cszLastCounter[] = {TEXT("Last Counter")};
|
|
static const TCHAR cszFirstHelp[] = {TEXT("First Help")};
|
|
static const TCHAR cszLastHelp[] = {TEXT("Last Help")};
|
|
static const TCHAR cszLibrary[] = {TEXT("Library")};
|
|
static const TCHAR cszPerformance[] = {TEXT("\\Performance")};
|
|
static const TCHAR cszServiceKeyName[] = {TEXT("SYSTEM\\CurrentControlSet\\Services")};
|
|
static const TCHAR cszNamesKey[] = {TEXT("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Perflib")};
|
|
static const TCHAR cszSlash[] = {TEXT("\\")};
|
|
static const WCHAR cszVersionName[] = {L"Version"};
|
|
static const WCHAR cszCounterName[] = {L"Counter "};
|
|
static const WCHAR cszHelpName[] = {L"Explain "};
|
|
static const WCHAR cszCounters[] = {L"Counters"};
|
|
static const TCHAR cszHelp[] = {TEXT("Help")};
|
|
|
|
#ifdef _DEBUG
|
|
#undef THIS_FILE
|
|
static char BASED_CODE THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
BOOL CExctrlstDlg::IndexHasString (
|
|
DWORD dwIndex
|
|
)
|
|
{
|
|
if ((dwIndex <= dwLastElement) && (pNameTable != NULL)) {
|
|
if (pNameTable[dwIndex] != NULL) {
|
|
return TRUE;
|
|
} else {
|
|
return FALSE;
|
|
}
|
|
} else {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
static
|
|
LPWSTR
|
|
*BuildNameTable(
|
|
LPCWSTR szMachineName,
|
|
LPCWSTR lpszLangIdArg, // unicode value of Language subkey
|
|
PDWORD pdwLastItem, // size of array in elements
|
|
PDWORD pdwIdArray // array for index ID's
|
|
)
|
|
/*++
|
|
|
|
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. (default is 409)
|
|
|
|
Return Value:
|
|
|
|
pointer to an allocated table. (the caller must MemoryFree it when finished!)
|
|
the table is an array of pointers to zero terminated strings. NULL is
|
|
returned if an error occured.
|
|
|
|
--*/
|
|
{
|
|
HKEY hKeyRegistry; // handle to registry db with counter names
|
|
|
|
LPWSTR *lpReturnValue;
|
|
LPCWSTR lpszLangId;
|
|
|
|
LPWSTR *lpCounterId;
|
|
LPWSTR lpCounterNames;
|
|
LPWSTR lpHelpText;
|
|
|
|
LPWSTR lpThisName;
|
|
|
|
LONG lWin32Status;
|
|
DWORD dwLastError;
|
|
DWORD dwValueType;
|
|
DWORD dwArraySize;
|
|
DWORD dwBufferSize;
|
|
DWORD dwCounterSize;
|
|
DWORD dwHelpSize;
|
|
DWORD dwThisCounter;
|
|
|
|
DWORD dwSystemVersion;
|
|
DWORD dwLastId;
|
|
DWORD dwLastHelpId;
|
|
|
|
DWORD dwLastCounterIdUsed;
|
|
DWORD dwLastHelpIdUsed;
|
|
|
|
HKEY hKeyValue;
|
|
HKEY hKeyNames;
|
|
|
|
LPWSTR lpValueNameString;
|
|
WCHAR CounterNameBuffer [50];
|
|
WCHAR HelpNameBuffer [50];
|
|
|
|
if (szMachineName != NULL) {
|
|
lWin32Status = RegConnectRegistryW (szMachineName,
|
|
HKEY_LOCAL_MACHINE,
|
|
&hKeyRegistry);
|
|
} else {
|
|
lWin32Status = ERROR_SUCCESS;
|
|
hKeyRegistry = HKEY_LOCAL_MACHINE;
|
|
}
|
|
|
|
lpValueNameString = NULL; //initialize to NULL
|
|
lpReturnValue = NULL;
|
|
hKeyValue = NULL;
|
|
hKeyNames = NULL;
|
|
|
|
// check for null arguments and insert defaults if necessary
|
|
|
|
if (!lpszLangIdArg) {
|
|
lpszLangId = cszDefaultLangId;
|
|
} else {
|
|
lpszLangId = lpszLangIdArg;
|
|
}
|
|
|
|
// open registry to get number of items for computing array size
|
|
|
|
lWin32Status = RegOpenKeyEx (
|
|
hKeyRegistry,
|
|
cszNamesKey,
|
|
RESERVED,
|
|
KEY_READ,
|
|
&hKeyValue);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
// get number of items
|
|
|
|
dwBufferSize = sizeof (dwLastHelpId);
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyValue,
|
|
cszLastHelp,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPBYTE)&dwLastHelpId,
|
|
&dwBufferSize);
|
|
|
|
if ((lWin32Status != ERROR_SUCCESS) || (dwValueType != REG_DWORD)) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
pdwIdArray[2] = dwLastHelpId;
|
|
|
|
// get number of items
|
|
|
|
dwBufferSize = sizeof (dwLastId);
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyValue,
|
|
cszLastCounter,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPBYTE)&dwLastId,
|
|
&dwBufferSize);
|
|
|
|
if ((lWin32Status != ERROR_SUCCESS) || (dwValueType != REG_DWORD)) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
pdwIdArray[0] = dwLastId;
|
|
|
|
if (dwLastId < dwLastHelpId)
|
|
dwLastId = dwLastHelpId;
|
|
|
|
dwArraySize = dwLastId * sizeof(LPWSTR);
|
|
|
|
// get Perflib system version
|
|
dwBufferSize = sizeof (dwSystemVersion);
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyValue,
|
|
cszVersionName,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPBYTE)&dwSystemVersion,
|
|
&dwBufferSize);
|
|
|
|
if ((lWin32Status != ERROR_SUCCESS) || (dwValueType != REG_DWORD)) {
|
|
dwSystemVersion = OLD_VERSION;
|
|
// reset the error status
|
|
lWin32Status = ERROR_SUCCESS;
|
|
}
|
|
|
|
if (dwSystemVersion == OLD_VERSION) {
|
|
// get names from registry
|
|
lpValueNameString = (LPWSTR)HeapAlloc (GetProcessHeap(), 0,
|
|
lstrlen(cszNamesKey) * sizeof (WCHAR) +
|
|
lstrlen(cszSlash) * sizeof (WCHAR) +
|
|
lstrlen(lpszLangId) * sizeof (WCHAR) +
|
|
sizeof (UNICODE_NULL));
|
|
|
|
if (!lpValueNameString) goto BNT_BAILOUT;
|
|
|
|
lstrcpy (lpValueNameString, cszNamesKey);
|
|
lstrcat (lpValueNameString, cszSlash);
|
|
lstrcat (lpValueNameString, lpszLangId);
|
|
|
|
lWin32Status = RegOpenKeyEx (
|
|
hKeyRegistry,
|
|
lpValueNameString,
|
|
RESERVED,
|
|
KEY_READ,
|
|
&hKeyNames);
|
|
} else {
|
|
if (szMachineName[0] == 0) {
|
|
hKeyNames = HKEY_PERFORMANCE_DATA;
|
|
} else {
|
|
lWin32Status = RegConnectRegistry (szMachineName,
|
|
HKEY_PERFORMANCE_DATA,
|
|
&hKeyNames);
|
|
}
|
|
lstrcpy (CounterNameBuffer, cszCounterName);
|
|
lstrcat (CounterNameBuffer, lpszLangId);
|
|
|
|
lstrcpy (HelpNameBuffer, cszHelpName);
|
|
lstrcat (HelpNameBuffer, lpszLangId);
|
|
}
|
|
|
|
// get size of counter names and add that to the arrays
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
dwBufferSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
dwSystemVersion == (DWORD)OLD_VERSION ? cszCounters : CounterNameBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
dwCounterSize = dwBufferSize;
|
|
|
|
// get size of counter names and add that to the arrays
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) goto BNT_BAILOUT;
|
|
|
|
dwBufferSize = 0;
|
|
lWin32Status = RegQueryValueEx (
|
|
hKeyNames,
|
|
dwSystemVersion == (DWORD)OLD_VERSION ? cszHelp : HelpNameBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
NULL,
|
|
&dwBufferSize);
|
|
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
dwHelpSize = dwBufferSize;
|
|
|
|
lpReturnValue = (LPWSTR *)HeapAlloc (GetProcessHeap(), 0,dwArraySize + dwCounterSize + dwHelpSize);
|
|
|
|
if (!lpReturnValue) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
// initialize pointers into buffer
|
|
|
|
lpCounterId = lpReturnValue;
|
|
lpCounterNames = (LPWSTR)((LPBYTE)lpCounterId + dwArraySize);
|
|
lpHelpText = (LPWSTR)((LPBYTE)lpCounterNames + dwCounterSize);
|
|
|
|
// read counters into memory
|
|
|
|
dwBufferSize = dwCounterSize;
|
|
lWin32Status = RegQueryValueExW (
|
|
hKeyNames,
|
|
dwSystemVersion == OLD_VERSION ? cszCounters : CounterNameBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPBYTE)lpCounterNames,
|
|
&dwBufferSize);
|
|
|
|
if (!lpReturnValue) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
dwBufferSize = dwHelpSize;
|
|
lWin32Status = RegQueryValueExW (
|
|
hKeyNames,
|
|
dwSystemVersion == OLD_VERSION ? cszHelp : HelpNameBuffer,
|
|
RESERVED,
|
|
&dwValueType,
|
|
(LPBYTE)lpHelpText,
|
|
&dwBufferSize);
|
|
|
|
if (!lpReturnValue) {
|
|
goto BNT_BAILOUT;
|
|
}
|
|
|
|
dwLastCounterIdUsed = 0;
|
|
dwLastHelpIdUsed = 0;
|
|
|
|
// load counter array items
|
|
|
|
for (lpThisName = lpCounterNames;
|
|
*lpThisName;
|
|
lpThisName += (lstrlen(lpThisName)+1) ) {
|
|
|
|
// first string should be an integer (in decimal unicode digits)
|
|
|
|
dwThisCounter = wcstoul (lpThisName, NULL, 10);
|
|
|
|
if (dwThisCounter == 0) {
|
|
goto BNT_BAILOUT; // bad entry
|
|
}
|
|
|
|
// point to corresponding counter name
|
|
|
|
lpThisName += (lstrlen(lpThisName)+1);
|
|
|
|
// and load array element;
|
|
|
|
lpCounterId[dwThisCounter] = lpThisName;
|
|
|
|
if (dwThisCounter > dwLastCounterIdUsed) dwLastCounterIdUsed = dwThisCounter;
|
|
|
|
}
|
|
|
|
pdwIdArray[1] = dwLastCounterIdUsed;
|
|
|
|
for (lpThisName = lpHelpText;
|
|
*lpThisName;
|
|
lpThisName += (lstrlen(lpThisName)+1) ) {
|
|
|
|
// first string should be an integer (in decimal unicode digits)
|
|
|
|
dwThisCounter = wcstoul (lpThisName, NULL, 10);
|
|
|
|
if (dwThisCounter == 0) {
|
|
goto BNT_BAILOUT; // bad entry
|
|
}
|
|
// point to corresponding counter name
|
|
|
|
lpThisName += (lstrlen(lpThisName)+1);
|
|
|
|
// and load array element;
|
|
|
|
lpCounterId[dwThisCounter] = lpThisName;
|
|
|
|
if (dwThisCounter > dwLastHelpIdUsed) dwLastHelpIdUsed= dwThisCounter;
|
|
}
|
|
|
|
pdwIdArray[3] = dwLastHelpIdUsed;
|
|
|
|
dwLastId = dwLastHelpIdUsed;
|
|
if (dwLastId < dwLastCounterIdUsed) dwLastId = dwLastCounterIdUsed;
|
|
|
|
if (pdwLastItem) *pdwLastItem = dwLastId;
|
|
|
|
HeapFree (GetProcessHeap(), 0, (LPVOID)lpValueNameString);
|
|
RegCloseKey (hKeyValue);
|
|
// if (dwSystemVersion == OLD_VERSION)
|
|
RegCloseKey (hKeyNames);
|
|
|
|
return lpReturnValue;
|
|
|
|
BNT_BAILOUT:
|
|
if (lWin32Status != ERROR_SUCCESS) {
|
|
dwLastError = GetLastError();
|
|
}
|
|
|
|
if (lpValueNameString) {
|
|
HeapFree (GetProcessHeap(), 0, (LPVOID)lpValueNameString);
|
|
}
|
|
|
|
if (lpReturnValue) {
|
|
HeapFree (GetProcessHeap(), 0, (LPVOID)lpReturnValue);
|
|
}
|
|
|
|
if (hKeyValue) RegCloseKey (hKeyValue);
|
|
|
|
// if (dwSystemVersion == OLD_VERSION &&
|
|
// hKeyNames)
|
|
RegCloseKey (hKeyNames);
|
|
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static
|
|
BOOL
|
|
IsMsObject(CString *pLibraryName)
|
|
{
|
|
CString LocalLibraryName;
|
|
|
|
LocalLibraryName = *pLibraryName;
|
|
LocalLibraryName.MakeLower();
|
|
|
|
// for now this just compares known DLL names. valid as of
|
|
// NT v4.0
|
|
if (LocalLibraryName.Find((LPCWSTR)L"perfctrs.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"ftpctrs.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"rasctrs.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"winsctrs.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"sfmctrs.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"atkctrs.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"bhmon.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"tapictrs.dll") >= 0) return TRUE;
|
|
// NT v5.0
|
|
if (LocalLibraryName.Find((LPCWSTR)L"perfdisk.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"perfos.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"perfproc.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"perfnet.dll") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"winspool.drv") >= 0) return TRUE;
|
|
if (LocalLibraryName.Find((LPCWSTR)L"tapiperf.dll") >= 0) return TRUE;
|
|
|
|
return FALSE;
|
|
}
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CExctrlstDlg dialog
|
|
|
|
CExctrlstDlg::CExctrlstDlg(CWnd* pParent /*=NULL*/)
|
|
: CDialog(CExctrlstDlg::IDD, pParent)
|
|
{
|
|
//{{AFX_DATA_INIT(CExctrlstDlg)
|
|
// NOTE: the ClassWizard will add member initialization here
|
|
//}}AFX_DATA_INIT
|
|
// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
|
|
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
|
|
hKeyMachine = HKEY_LOCAL_MACHINE;
|
|
hKeyServices = NULL;
|
|
dwSortOrder = SORT_ORDER_SERVICE;
|
|
bReadWriteAccess = TRUE;
|
|
dwRegAccessMask = KEY_READ | KEY_WRITE;
|
|
pNameTable = NULL;
|
|
dwLastElement = 0;
|
|
dwListBoxHorizExtent = 0;
|
|
dwTabStopCount = 1;
|
|
dwTabStopArray[0] = 85;
|
|
memset (&dwIdArray[0], 0, sizeof(dwIdArray));
|
|
}
|
|
|
|
CExctrlstDlg::~CExctrlstDlg()
|
|
{
|
|
if (pNameTable != NULL) {
|
|
HeapFree (GetProcessHeap(), 0, pNameTable);
|
|
pNameTable = NULL;
|
|
dwLastElement = 0;
|
|
}
|
|
}
|
|
|
|
void CExctrlstDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
//{{AFX_DATA_MAP(CExctrlstDlg)
|
|
// NOTE: the ClassWizard will add DDX and DDV calls here
|
|
//}}AFX_DATA_MAP
|
|
}
|
|
|
|
BEGIN_MESSAGE_MAP(CExctrlstDlg, CDialog)
|
|
//{{AFX_MSG_MAP(CExctrlstDlg)
|
|
ON_WM_PAINT()
|
|
ON_WM_QUERYDRAGICON()
|
|
ON_LBN_SELCHANGE(IDC_EXT_LIST, OnSelchangeExtList)
|
|
ON_WM_DESTROY()
|
|
ON_BN_CLICKED(IDC_REFRESH, OnRefresh)
|
|
ON_BN_CLICKED(IDC_ABOUT, OnAbout)
|
|
ON_EN_KILLFOCUS(IDC_MACHINE_NAME, OnKillfocusMachineName)
|
|
ON_BN_CLICKED(IDC_SORT_LIBRARY, OnSortButton)
|
|
ON_BN_CLICKED(IDC_SORT_SERVICE, OnSortButton)
|
|
ON_BN_CLICKED(IDC_SORT_ID, OnSortButton)
|
|
ON_BN_CLICKED(IDC_ENABLED_BTN, OnEnablePerf)
|
|
ON_WM_SYSCOMMAND()
|
|
//}}AFX_MSG_MAP
|
|
END_MESSAGE_MAP()
|
|
|
|
DWORD CExctrlstDlg::EnablePerfCounters (HKEY hKeyItem, DWORD dwNewValue)
|
|
{
|
|
DWORD dwStatus;
|
|
DWORD dwType;
|
|
DWORD dwValue;
|
|
DWORD dwSize;
|
|
DWORD dwReturn;
|
|
|
|
switch (dwNewValue) {
|
|
case ENABLE_PERF_CTR_QUERY:
|
|
dwType = 0;
|
|
dwSize = sizeof (dwValue);
|
|
dwValue = 0;
|
|
dwStatus = RegQueryValueExW (
|
|
hKeyItem,
|
|
cszDisablePerformanceCounters,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&dwValue,
|
|
&dwSize);
|
|
|
|
if ((dwStatus == ERROR_SUCCESS) && (dwType == REG_DWORD)) {
|
|
switch (dwValue) {
|
|
case 0: dwReturn = ENABLE_PERF_CTR_ENABLE; break;
|
|
case 1: dwReturn = ENABLE_PERF_CTR_DISABLE; break;
|
|
default: dwReturn = 0; break;
|
|
}
|
|
} else {
|
|
// if the value is not present, or not = 1, then the perfctrs
|
|
// are enabled
|
|
dwReturn = ENABLE_PERF_CTR_ENABLE;
|
|
}
|
|
break;
|
|
|
|
case ENABLE_PERF_CTR_ENABLE:
|
|
dwType = REG_DWORD;
|
|
dwSize = sizeof (dwValue);
|
|
dwValue = 0;
|
|
dwStatus = RegSetValueExW (
|
|
hKeyItem,
|
|
cszDisablePerformanceCounters,
|
|
0L,
|
|
dwType,
|
|
(LPBYTE)&dwValue,
|
|
dwSize);
|
|
if (dwStatus == ERROR_SUCCESS) {
|
|
dwReturn = ENABLE_PERF_CTR_ENABLE;
|
|
} else {
|
|
dwReturn = 0;
|
|
}
|
|
break;
|
|
|
|
case ENABLE_PERF_CTR_DISABLE:
|
|
dwType = REG_DWORD;
|
|
dwSize = sizeof (dwValue);
|
|
dwValue = 1;
|
|
dwStatus = RegSetValueExW (
|
|
hKeyItem,
|
|
cszDisablePerformanceCounters,
|
|
0L,
|
|
dwType,
|
|
(LPBYTE)&dwValue,
|
|
dwSize);
|
|
if (dwStatus == ERROR_SUCCESS) {
|
|
dwReturn = ENABLE_PERF_CTR_DISABLE;
|
|
} else {
|
|
dwReturn = 0;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
dwReturn = 0;
|
|
}
|
|
return dwReturn;
|
|
}
|
|
|
|
void CExctrlstDlg::ScanForExtensibleCounters ()
|
|
{
|
|
LONG lStatus = ERROR_SUCCESS;
|
|
LONG lEnumStatus = ERROR_SUCCESS;
|
|
DWORD dwServiceIndex;
|
|
TCHAR szServiceSubKeyName[MAX_PATH];
|
|
TCHAR szPerfSubKeyName[MAX_PATH+20];
|
|
TCHAR szItemText[MAX_PATH];
|
|
TCHAR szListText[MAX_PATH*2];
|
|
DWORD dwNameSize;
|
|
HKEY hKeyPerformance;
|
|
UINT_PTR nListBoxEntry;
|
|
DWORD dwItemSize, dwType, dwValue;
|
|
HCURSOR hOldCursor;
|
|
DWORD dwThisExtent;
|
|
HDC hDcListBox;
|
|
CWnd *pCWndListBox;
|
|
|
|
hOldCursor = ::SetCursor (LoadCursor(NULL, IDC_WAIT));
|
|
|
|
ResetListBox();
|
|
|
|
if (hKeyServices == NULL) {
|
|
// try read/write access
|
|
lStatus = RegOpenKeyEx (hKeyMachine,
|
|
cszServiceKeyName,
|
|
0L,
|
|
dwRegAccessMask,
|
|
&hKeyServices);
|
|
if (lStatus != ERROR_SUCCESS) {
|
|
// try read-only then
|
|
dwRegAccessMask = KEY_READ;
|
|
bReadWriteAccess = FALSE;
|
|
lStatus = RegOpenKeyEx (hKeyMachine,
|
|
cszServiceKeyName,
|
|
0L,
|
|
dwRegAccessMask,
|
|
&hKeyServices);
|
|
if (lStatus != ERROR_SUCCESS) {
|
|
// display Read Only message
|
|
AfxMessageBox (IDS_READ_ONLY);
|
|
} else {
|
|
// fall through with error code
|
|
// display no access message
|
|
AfxMessageBox (IDS_NO_ACCESS);
|
|
}
|
|
}
|
|
} else {
|
|
lStatus = ERROR_SUCCESS;
|
|
}
|
|
|
|
if (lStatus == ERROR_SUCCESS) {
|
|
pCWndListBox = GetDlgItem (IDC_EXT_LIST);
|
|
hDcListBox = ::GetDC (pCWndListBox->m_hWnd);
|
|
if (hDcListBox == NULL) {
|
|
return;
|
|
}
|
|
dwServiceIndex = 0;
|
|
dwNameSize = MAX_PATH;
|
|
while ((lEnumStatus = RegEnumKeyEx (
|
|
hKeyServices,
|
|
dwServiceIndex,
|
|
szServiceSubKeyName,
|
|
&dwNameSize,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL)) == ERROR_SUCCESS) {
|
|
|
|
//try to open the perfkey under this key.
|
|
lstrcpy (szPerfSubKeyName, szServiceSubKeyName);
|
|
lstrcat (szPerfSubKeyName, cszPerformance);
|
|
|
|
lStatus = RegOpenKeyEx (
|
|
hKeyServices,
|
|
szPerfSubKeyName,
|
|
0L,
|
|
dwRegAccessMask,
|
|
&hKeyPerformance);
|
|
|
|
if (lStatus == ERROR_SUCCESS) {
|
|
// look up the library name
|
|
|
|
dwItemSize = MAX_PATH * sizeof(TCHAR);
|
|
dwType = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyPerformance,
|
|
cszLibrary,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&szItemText[0],
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) ||
|
|
((dwType != REG_SZ) && dwType != REG_EXPAND_SZ)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
}
|
|
|
|
dwItemSize = sizeof(DWORD);
|
|
dwType = 0;
|
|
dwValue = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyPerformance,
|
|
cszFirstCounter,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&dwValue,
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) || (dwType != REG_DWORD)) {
|
|
dwValue = 0;
|
|
}
|
|
|
|
// make the string for the list box here depending
|
|
// on the selected sort order.
|
|
|
|
if (dwSortOrder == SORT_ORDER_LIBRARY) {
|
|
lstrcpy(szListText, szItemText);
|
|
lstrcat(szListText, cszTab);
|
|
lstrcat(szListText, szServiceSubKeyName);
|
|
} else if (dwSortOrder == SORT_ORDER_ID) {
|
|
_stprintf (szListText, cszSortIdFmt,
|
|
dwValue, szServiceSubKeyName);
|
|
} else { // default is sort by service
|
|
lstrcpy(szListText, szServiceSubKeyName);
|
|
lstrcat(szListText, cszTab);
|
|
lstrcat(szListText, szItemText);
|
|
}
|
|
|
|
// add this name to the list box
|
|
nListBoxEntry = SendDlgItemMessage(IDC_EXT_LIST,
|
|
LB_ADDSTRING, 0, (LPARAM)&szListText);
|
|
|
|
if (nListBoxEntry != LB_ERR) {
|
|
dwThisExtent = GetTabbedTextExtent (
|
|
hDcListBox,
|
|
szListText,
|
|
lstrlen(szListText),
|
|
(int)dwTabStopCount,
|
|
(int *)&dwTabStopArray[0]);
|
|
|
|
if (dwThisExtent > dwListBoxHorizExtent) {
|
|
dwListBoxHorizExtent = dwThisExtent;
|
|
SendDlgItemMessage(IDC_EXT_LIST,
|
|
LB_SETHORIZONTALEXTENT,
|
|
(WPARAM)LOWORD(dwListBoxHorizExtent), (LPARAM)0);
|
|
}
|
|
// save key to this entry in the registry
|
|
SendDlgItemMessage (IDC_EXT_LIST,
|
|
LB_SETITEMDATA, (WPARAM)nListBoxEntry,
|
|
(LPARAM)hKeyPerformance);
|
|
} else {
|
|
// close the key since there's no point in
|
|
// keeping it open
|
|
RegCloseKey (hKeyPerformance);
|
|
}
|
|
}
|
|
// reset for next loop
|
|
dwServiceIndex++;
|
|
dwNameSize = MAX_PATH;
|
|
}
|
|
::ReleaseDC (pCWndListBox->m_hWnd, hDcListBox);
|
|
}
|
|
nListBoxEntry = SendDlgItemMessage (IDC_EXT_LIST, LB_GETCOUNT);
|
|
if (nListBoxEntry > 0) {
|
|
SendDlgItemMessage (IDC_EXT_LIST, LB_SETCURSEL, 0, 0);
|
|
}
|
|
::SetCursor (hOldCursor);
|
|
|
|
}
|
|
|
|
void CExctrlstDlg::UpdateSystemInfo () {
|
|
TCHAR szItemText[MAX_PATH];
|
|
|
|
_stprintf (szItemText, cszIdFmt,
|
|
dwIdArray[0], dwIdArray[0], cszSpace);
|
|
SetDlgItemText (IDC_LAST_COUNTER_VALUE, szItemText);
|
|
|
|
_stprintf (szItemText, cszIdFmt,
|
|
dwIdArray[1], dwIdArray[1],
|
|
dwIdArray[1] != dwIdArray[0] ? cszSplat : cszSpace);
|
|
SetDlgItemText (IDC_LAST_TEXT_COUNTER_VALUE, szItemText);
|
|
|
|
_stprintf (szItemText, cszIdFmt,
|
|
dwIdArray[2], dwIdArray[2], cszSpace);
|
|
SetDlgItemText (IDC_LAST_HELP_VALUE, szItemText);
|
|
|
|
_stprintf (szItemText, cszIdFmt,
|
|
dwIdArray[3], dwIdArray[3],
|
|
dwIdArray[3] != dwIdArray[2] ? cszSplat : cszSpace);
|
|
SetDlgItemText (IDC_LAST_TEXT_HELP_VALUE, szItemText);
|
|
|
|
}
|
|
|
|
void CExctrlstDlg::UpdateDllInfo () {
|
|
HKEY hKeyItem;
|
|
TCHAR szItemText[MAX_PATH];
|
|
UINT_PTR nSelectedItem;
|
|
LONG lStatus;
|
|
DWORD dwType;
|
|
DWORD dwValue;
|
|
DWORD dwItemSize;
|
|
BOOL bNoIndexValues = FALSE;
|
|
DWORD dwEnabled;
|
|
|
|
CString OpenProcName;
|
|
CString LibraryName;
|
|
|
|
HCURSOR hOldCursor;
|
|
|
|
hOldCursor = ::SetCursor (LoadCursor(NULL, IDC_WAIT));
|
|
|
|
OpenProcName.Empty();
|
|
LibraryName.Empty();
|
|
// update the performance counter information
|
|
|
|
nSelectedItem = SendDlgItemMessage (IDC_EXT_LIST, LB_GETCURSEL);
|
|
|
|
if (nSelectedItem != LB_ERR) {
|
|
// get registry key for the selected item
|
|
hKeyItem = (HKEY)SendDlgItemMessage (IDC_EXT_LIST, LB_GETITEMDATA,
|
|
(WPARAM)nSelectedItem, 0);
|
|
|
|
dwItemSize = MAX_PATH * sizeof(TCHAR);
|
|
dwType = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyItem,
|
|
cszLibrary,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&szItemText[0],
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) ||
|
|
((dwType != REG_SZ) && dwType != REG_EXPAND_SZ)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
} else {
|
|
LibraryName = szItemText;
|
|
}
|
|
SetDlgItemText (IDC_DLL_NAME, szItemText);
|
|
|
|
dwItemSize = MAX_PATH * sizeof(TCHAR);
|
|
dwType = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyItem,
|
|
cszOpen,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&szItemText[0],
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) ||
|
|
((dwType != REG_SZ) && dwType != REG_EXPAND_SZ)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
} else {
|
|
OpenProcName = szItemText;
|
|
}
|
|
SetDlgItemText (IDC_OPEN_PROC, szItemText);
|
|
|
|
dwItemSize = MAX_PATH * sizeof(TCHAR);
|
|
dwType = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyItem,
|
|
cszCollect,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&szItemText[0],
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) ||
|
|
((dwType != REG_SZ) && dwType != REG_EXPAND_SZ)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
}
|
|
SetDlgItemText (IDC_COLLECT_PROC, szItemText);
|
|
|
|
dwItemSize = MAX_PATH * sizeof(TCHAR);
|
|
dwType = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyItem,
|
|
cszClose,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&szItemText[0],
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) ||
|
|
((dwType != REG_SZ) && dwType != REG_EXPAND_SZ)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
}
|
|
SetDlgItemText (IDC_CLOSE_PROC, szItemText);
|
|
|
|
dwItemSize = sizeof(DWORD);
|
|
dwType = 0;
|
|
dwValue = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyItem,
|
|
cszFirstCounter,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&dwValue,
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) || (dwType != REG_DWORD)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
bNoIndexValues = TRUE;
|
|
} else {
|
|
_stprintf (szItemText, cszServIdFmt, dwValue, IndexHasString (dwValue) ? cszSpace : cszSplat);
|
|
}
|
|
SetDlgItemText (IDC_FIRST_CTR_ID, szItemText);
|
|
|
|
dwItemSize = sizeof(DWORD);
|
|
dwType = 0;
|
|
dwValue = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyItem,
|
|
cszLastCounter,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&dwValue,
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) || (dwType != REG_DWORD)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
} else {
|
|
_stprintf (szItemText, cszServIdFmt, dwValue, IndexHasString (dwValue) ? cszSpace : cszSplat);
|
|
}
|
|
SetDlgItemText (IDC_LAST_CTR_ID, szItemText);
|
|
|
|
dwItemSize = sizeof(DWORD);
|
|
dwType = 0;
|
|
dwValue = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyItem,
|
|
cszFirstHelp,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&dwValue,
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) || (dwType != REG_DWORD)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
bNoIndexValues = TRUE;
|
|
} else {
|
|
_stprintf (szItemText, cszServIdFmt, dwValue, IndexHasString (dwValue) ? cszSpace : cszSplat);
|
|
}
|
|
SetDlgItemText (IDC_FIRST_HELP_ID, szItemText);
|
|
|
|
dwItemSize = sizeof(DWORD);
|
|
dwType = 0;
|
|
dwValue = 0;
|
|
lStatus = RegQueryValueEx (
|
|
hKeyItem,
|
|
cszLastHelp,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&dwValue,
|
|
&dwItemSize);
|
|
|
|
if ((lStatus != ERROR_SUCCESS) || (dwType != REG_DWORD)) {
|
|
lstrcpy (szItemText, cszNotFound);
|
|
} else {
|
|
_stprintf (szItemText, cszServIdFmt, dwValue, IndexHasString (dwValue) ? cszSpace : cszSplat);
|
|
}
|
|
SetDlgItemText (IDC_LAST_HELP_ID, szItemText);
|
|
|
|
if (bNoIndexValues) {
|
|
// test to see if this is a "standard" i.e. Microsoft provided
|
|
// extensible counter or simply one that hasn't been completely
|
|
// installed
|
|
if (IsMsObject(&LibraryName)) {
|
|
SetDlgItemText (IDC_FIRST_HELP_ID, cszNA);
|
|
SetDlgItemText (IDC_LAST_HELP_ID, cszNA);
|
|
SetDlgItemText (IDC_FIRST_CTR_ID, cszNA);
|
|
SetDlgItemText (IDC_LAST_CTR_ID, cszNA);
|
|
}
|
|
}
|
|
|
|
GetDlgItem(IDC_ENABLED_BTN)->ShowWindow (bReadWriteAccess ? SW_SHOW : SW_HIDE);
|
|
GetDlgItem(IDC_ENABLED_BTN)->EnableWindow (bReadWriteAccess);
|
|
|
|
dwEnabled = EnablePerfCounters (hKeyItem, ENABLE_PERF_CTR_QUERY);
|
|
|
|
if (bReadWriteAccess) {
|
|
// then set the check box
|
|
CheckDlgButton (IDC_ENABLED_BTN, dwEnabled == ENABLE_PERF_CTR_ENABLE ? 1 : 0);
|
|
GetDlgItem(IDC_DISABLED_TEXT)->ShowWindow (SW_HIDE);
|
|
} else {
|
|
// update the text message
|
|
GetDlgItem(IDC_DISABLED_TEXT)->ShowWindow (
|
|
(!(dwEnabled == ENABLE_PERF_CTR_ENABLE)) ?
|
|
SW_SHOW : SW_HIDE);
|
|
GetDlgItem(IDC_DISABLED_TEXT)->EnableWindow (TRUE);
|
|
}
|
|
|
|
|
|
}
|
|
::SetCursor (hOldCursor);
|
|
}
|
|
|
|
void CExctrlstDlg::ResetListBox ()
|
|
{
|
|
INT_PTR nItemCount;
|
|
INT nThisItem;
|
|
HKEY hKeyItem;
|
|
|
|
nItemCount = SendDlgItemMessage (IDC_EXT_LIST, LB_GETCOUNT);
|
|
nThisItem = 0;
|
|
while (nThisItem > nItemCount) {
|
|
hKeyItem = (HKEY) SendDlgItemMessage (IDC_EXT_LIST,
|
|
LB_GETITEMDATA, (WPARAM)nThisItem);
|
|
RegCloseKey (hKeyItem);
|
|
nThisItem++;
|
|
}
|
|
SendDlgItemMessage (IDC_EXT_LIST, LB_RESETCONTENT);
|
|
dwListBoxHorizExtent = 0;
|
|
SendDlgItemMessage(IDC_EXT_LIST,
|
|
LB_SETHORIZONTALEXTENT,
|
|
(WPARAM)LOWORD(dwListBoxHorizExtent), (LPARAM)0);
|
|
}
|
|
|
|
void CExctrlstDlg::SetSortButtons()
|
|
{
|
|
DWORD dwBtn;
|
|
|
|
switch (dwSortOrder) {
|
|
case SORT_ORDER_LIBRARY: dwBtn = IDC_SORT_LIBRARY; break;
|
|
case SORT_ORDER_SERVICE: dwBtn = IDC_SORT_SERVICE; break;
|
|
case SORT_ORDER_ID: dwBtn = IDC_SORT_ID; break;
|
|
default: dwBtn = IDC_SORT_SERVICE; break;
|
|
}
|
|
|
|
CheckRadioButton (
|
|
IDC_SORT_LIBRARY,
|
|
IDC_SORT_ID,
|
|
dwBtn);
|
|
}
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CExctrlstDlg message handlers
|
|
|
|
BOOL CExctrlstDlg::OnInitDialog()
|
|
{
|
|
HCURSOR hOldCursor;
|
|
DWORD dwLength;
|
|
|
|
hOldCursor = ::SetCursor (::LoadCursor (NULL, IDC_WAIT));
|
|
|
|
CDialog::OnInitDialog();
|
|
CenterWindow();
|
|
|
|
lstrcpy (szThisComputerName, cszDoubleBackslash);
|
|
dwLength = MAX_COMPUTERNAME_LENGTH+1;
|
|
GetComputerName (&szThisComputerName[2], &dwLength);
|
|
|
|
lstrcpy (szComputerName, szThisComputerName);
|
|
|
|
SetDlgItemText (IDC_MACHINE_NAME, szComputerName);
|
|
|
|
hKeyMachine = HKEY_LOCAL_MACHINE;
|
|
|
|
pNameTable = BuildNameTable (
|
|
szComputerName,
|
|
cszDefaultLangId,
|
|
&dwLastElement, // size of array in elements
|
|
&dwIdArray[0]);
|
|
|
|
SendDlgItemMessage (IDC_MACHINE_NAME, EM_LIMITTEXT,
|
|
(WPARAM)MAX_COMPUTERNAME_LENGTH+2, 0); // include 2 leading backslash
|
|
|
|
SendDlgItemMessage (IDC_EXT_LIST, LB_SETTABSTOPS,
|
|
(WPARAM)dwTabStopCount, (LPARAM)&dwTabStopArray[0]);
|
|
|
|
SetSortButtons();
|
|
|
|
ScanForExtensibleCounters(); //.checks for access to the registry
|
|
|
|
UpdateSystemInfo();
|
|
|
|
// set the check box to the appropriate state
|
|
|
|
UpdateDllInfo ();
|
|
|
|
::SetCursor(hOldCursor);
|
|
|
|
return TRUE; // return TRUE unless you set the focus to a control
|
|
}
|
|
|
|
// If you add a minimize button to your dialog, you will need the code below
|
|
// to draw the icon. For MFC applications using the document/view model,
|
|
// this is automatically done for you by the framework.
|
|
|
|
void CExctrlstDlg::OnPaint()
|
|
{
|
|
if (IsIconic())
|
|
{
|
|
CPaintDC dc(this); // device context for painting
|
|
|
|
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
|
|
|
|
// Center icon in client rectangle
|
|
int cxIcon = GetSystemMetrics(SM_CXICON);
|
|
int cyIcon = GetSystemMetrics(SM_CYICON);
|
|
CRect rect;
|
|
GetClientRect(&rect);
|
|
int x = (rect.Width() - cxIcon + 1) / 2;
|
|
int y = (rect.Height() - cyIcon + 1) / 2;
|
|
|
|
// Draw the icon
|
|
dc.DrawIcon(x, y, m_hIcon);
|
|
}
|
|
else
|
|
{
|
|
CDialog::OnPaint();
|
|
}
|
|
}
|
|
|
|
// The system calls this to obtain the cursor to display while the user drags
|
|
// the minimized window.
|
|
HCURSOR CExctrlstDlg::OnQueryDragIcon()
|
|
{
|
|
return (HCURSOR) m_hIcon;
|
|
}
|
|
|
|
void CExctrlstDlg::OnSelchangeExtList()
|
|
{
|
|
UpdateDllInfo ();
|
|
}
|
|
|
|
void CExctrlstDlg::OnDestroy()
|
|
{
|
|
ResetListBox();
|
|
CDialog::OnDestroy();
|
|
}
|
|
|
|
void CExctrlstDlg::OnAbout()
|
|
{
|
|
CAbout dlg;
|
|
dlg.DoModal();
|
|
}
|
|
|
|
void CExctrlstDlg::OnRefresh()
|
|
{
|
|
HCURSOR hOldCursor;
|
|
|
|
hOldCursor = ::SetCursor (::LoadCursor (NULL, IDC_WAIT));
|
|
|
|
ScanForExtensibleCounters();
|
|
if (pNameTable != NULL) {
|
|
HeapFree (GetProcessHeap(), 0, pNameTable);
|
|
pNameTable = NULL;
|
|
dwLastElement = 0;
|
|
}
|
|
|
|
pNameTable = BuildNameTable (
|
|
szComputerName,
|
|
cszDefaultLangId,
|
|
&dwLastElement, // size of array in elements
|
|
&dwIdArray[0]);
|
|
|
|
UpdateSystemInfo();
|
|
|
|
UpdateDllInfo ();
|
|
::SetCursor(hOldCursor);
|
|
}
|
|
|
|
void CExctrlstDlg::OnKillfocusMachineName()
|
|
{
|
|
TCHAR szNewMachineName[MAX_PATH];
|
|
HKEY hKeyNewMachine;
|
|
LONG lStatus;
|
|
HCURSOR hOldCursor;
|
|
|
|
hOldCursor = ::SetCursor (::LoadCursor (NULL, IDC_WAIT));
|
|
|
|
GetDlgItemText (IDC_MACHINE_NAME, szNewMachineName, MAX_PATH);
|
|
|
|
if (lstrcmpi(szComputerName, szNewMachineName) != 0) {
|
|
// a new computer has been entered so try to connect to it
|
|
lStatus = RegConnectRegistry (szNewMachineName,
|
|
HKEY_LOCAL_MACHINE, &hKeyNewMachine);
|
|
if (lStatus == ERROR_SUCCESS) {
|
|
RegCloseKey (hKeyServices); // close the old key
|
|
hKeyServices = NULL; // clear it
|
|
bReadWriteAccess = TRUE; // reset the access variables
|
|
dwRegAccessMask = KEY_READ | KEY_WRITE;
|
|
RegCloseKey (hKeyMachine); // close the old machine
|
|
hKeyMachine = hKeyNewMachine; // update to the new machine
|
|
lstrcpy (szComputerName, szNewMachineName); // update the name
|
|
OnRefresh(); // get new counters
|
|
} else {
|
|
SetDlgItemText (IDC_MACHINE_NAME, szComputerName);
|
|
}
|
|
} else {
|
|
// the machine name has not changed
|
|
}
|
|
::SetCursor (hOldCursor);
|
|
}
|
|
|
|
void CExctrlstDlg::OnSortButton()
|
|
{
|
|
if (IsDlgButtonChecked(IDC_SORT_LIBRARY)) {
|
|
dwSortOrder = SORT_ORDER_LIBRARY;
|
|
} else if (IsDlgButtonChecked(IDC_SORT_SERVICE)) {
|
|
dwSortOrder = SORT_ORDER_SERVICE;
|
|
} else if (IsDlgButtonChecked(IDC_SORT_ID)) {
|
|
dwSortOrder = SORT_ORDER_ID;
|
|
}
|
|
ScanForExtensibleCounters();
|
|
UpdateDllInfo ();
|
|
}
|
|
|
|
void CExctrlstDlg::OnEnablePerf()
|
|
{
|
|
HKEY hKeyItem;
|
|
UINT_PTR nSelectedItem;
|
|
DWORD dwNewValue;
|
|
|
|
nSelectedItem = SendDlgItemMessage (IDC_EXT_LIST, LB_GETCURSEL);
|
|
|
|
if (nSelectedItem != LB_ERR) {
|
|
// get registry key for the selected item
|
|
hKeyItem = (HKEY)SendDlgItemMessage (IDC_EXT_LIST, LB_GETITEMDATA,
|
|
(WPARAM)nSelectedItem, 0);
|
|
|
|
// get selected perf item and the corre
|
|
dwNewValue = IsDlgButtonChecked(IDC_ENABLED_BTN) ?
|
|
ENABLE_PERF_CTR_ENABLE :
|
|
ENABLE_PERF_CTR_DISABLE;
|
|
|
|
if (EnablePerfCounters (hKeyItem, dwNewValue) == 0) {
|
|
MessageBeep(0xFFFFFFFF);
|
|
// then it failed so reset to the curent value
|
|
dwNewValue = EnablePerfCounters (hKeyItem, ENABLE_PERF_CTR_QUERY);
|
|
CheckDlgButton (IDC_ENABLED_BTN, dwNewValue == ENABLE_PERF_CTR_ENABLE ? 1 : 0);
|
|
}
|
|
}
|
|
}
|
|
|
|
void CExctrlstDlg::OnSysCommand(UINT nID, LPARAM lParam)
|
|
{
|
|
switch (nID) {
|
|
case SC_CLOSE:
|
|
EndDialog(IDOK);
|
|
break;
|
|
|
|
default:
|
|
CDialog::OnSysCommand (nID, lParam);
|
|
break;
|
|
}
|
|
}
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CAbout dialog
|
|
|
|
|
|
CAbout::CAbout(CWnd* pParent /*=NULL*/)
|
|
: CDialog(CAbout::IDD, pParent)
|
|
{
|
|
//{{AFX_DATA_INIT(CAbout)
|
|
// NOTE: the ClassWizard will add member initialization here
|
|
//}}AFX_DATA_INIT
|
|
}
|
|
|
|
|
|
void CAbout::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
//{{AFX_DATA_MAP(CAbout)
|
|
// NOTE: the ClassWizard will add DDX and DDV calls here
|
|
//}}AFX_DATA_MAP
|
|
}
|
|
|
|
BOOL CAbout::OnInitDialog()
|
|
{
|
|
CDialog::OnInitDialog();
|
|
|
|
TCHAR buffer[512];
|
|
TCHAR strProgram[1024];
|
|
TCHAR strMicrosoft[1024];
|
|
DWORD dw;
|
|
BYTE* pVersionInfo;
|
|
LPTSTR pVersion = NULL;
|
|
LPTSTR pProduct = NULL;
|
|
LPTSTR pCopyRight = NULL;
|
|
|
|
dw = GetModuleFileName(NULL, strProgram, 1024);
|
|
|
|
if( dw>0 ){
|
|
|
|
dw = GetFileVersionInfoSize( strProgram, &dw );
|
|
if( dw > 0 ){
|
|
|
|
pVersionInfo = (BYTE*)malloc(dw);
|
|
if( NULL != pVersionInfo ){
|
|
if(GetFileVersionInfo( strProgram, 0, dw, pVersionInfo )){
|
|
LPDWORD lptr = NULL;
|
|
VerQueryValue( pVersionInfo, _T("\\VarFileInfo\\Translation"), (void**)&lptr, (UINT*)&dw );
|
|
if( lptr != NULL ){
|
|
_stprintf( buffer, _T("\\StringFileInfo\\%04x%04x\\%s"), LOWORD(*lptr), HIWORD(*lptr), _T("ProductVersion") );
|
|
VerQueryValue( pVersionInfo, buffer, (void**)&pVersion, (UINT*)&dw );
|
|
_stprintf( buffer, _T("\\StringFileInfo\\%04x%04x\\%s"), LOWORD(*lptr), HIWORD(*lptr), _T("OriginalFilename") );
|
|
VerQueryValue( pVersionInfo, buffer, (void**)&pProduct, (UINT*)&dw );
|
|
_stprintf( buffer, _T("\\StringFileInfo\\%04x%04x\\%s"), LOWORD(*lptr), HIWORD(*lptr), _T("LegalCopyright") );
|
|
VerQueryValue( pVersionInfo, buffer, (void**)&pCopyRight, (UINT*)&dw );
|
|
}
|
|
|
|
if( pProduct != NULL && pVersion != NULL && pCopyRight != NULL ){
|
|
GetDlgItem(IDC_COPYRIGHT)->SetWindowText( pCopyRight );
|
|
GetDlgItem(IDC_VERSION)->SetWindowText( pVersion );
|
|
}
|
|
}
|
|
free( pVersionInfo );
|
|
}
|
|
}
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
BEGIN_MESSAGE_MAP(CAbout, CDialog)
|
|
//{{AFX_MSG_MAP(CAbout)
|
|
// NOTE: the ClassWizard will add message map macros here
|
|
//}}AFX_MSG_MAP
|
|
END_MESSAGE_MAP()
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CAbout message handlers
|