|
|
// pdlcnfig.cpp : Defines the class behaviors for the application.
//
#include "stdafx.h"
#include "pdlcnfig.h"
#include "OutPage.h"
#include "SetPage.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;#endif
/////////////////////////////////////////////////////////////////////////////
// CPdlConfigApp
BEGIN_MESSAGE_MAP(CPdlConfigApp, CWinApp) //{{AFX_MSG_MAP(CPdlConfigApp)
// NOTE - the ClassWizard will add and remove mapping macros here.
// DO NOT EDIT what you see in these blocks of generated code!
//}}AFX_MSG
ON_COMMAND(ID_HELP, CWinApp::OnHelp)END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CPdlConfigApp construction
CPdlConfigApp::CPdlConfigApp(){ // TODO: add construction code here,
// Place all significant initialization in InitInstance
}
/////////////////////////////////////////////////////////////////////////////
// The one and only CPdlConfigApp object
CPdlConfigApp theApp;
/////////////////////////////////////////////////////////////////////////////
// Test for installation of the service.
LONG CPdlConfigApp::PerfLogServiceStatus(){ HKEY hKeyLogService; LONG lStatus;
// try opening the key to the service
lStatus = RegOpenKeyEx (HKEY_LOCAL_MACHINE, TEXT("SYSTEM\\CurrentControlSet\\Services\\PerfDataLog"), 0, KEY_READ | KEY_WRITE, &hKeyLogService);
if (lStatus == ERROR_SUCCESS) { // don't keep the key open
RegCloseKey (hKeyLogService); }
return lStatus;} LONG CPdlConfigApp::ServiceFilesCopied(){ HANDLE hFile = INVALID_HANDLE_VALUE; LONG lStatus = ERROR_SUCCESS; TCHAR szFullPathName[MAX_PATH];
ExpandEnvironmentStrings ( TEXT("%windir%\\system32\\pdlsvc.exe"), szFullPathName, MAX_PATH);
hFile = CreateFile ( szFullPathName, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); if (hFile == INVALID_HANDLE_VALUE) { lStatus = GetLastError(); } else { CloseHandle (hFile); lStatus = ERROR_SUCCESS; }
return lStatus;} /////////////////////////////////////////////////////////////////////////////
// CPdlConfigApp initialization
LONG CPdlConfigApp::CreatePerfDataLogService (){ LONG lStatus = ERROR_SUCCESS; HKEY hKeyServices = NULL; HKEY hKeyPerfLog = NULL; HKEY hKeyPerfAlert = NULL; DWORD dwWaitLimit = 20; SC_HANDLE hSC; SC_HANDLE hLogService;
// create service
hSC = OpenSCManager (NULL, NULL, SC_MANAGER_CREATE_SERVICE);
if (hSC == NULL) { // display error message
lStatus = GetLastError(); }
if (lStatus == ERROR_SUCCESS) { hLogService = CreateService (hSC, TEXT("PerfDataLog"), TEXT("Performance Data Log"), SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS, SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL, TEXT("%systemroot%\\system32\\pdlsvc.exe"), NULL, NULL, NULL, NULL, NULL);
if (hLogService == NULL) { lStatus = GetLastError(); } else { lStatus = ERROR_SUCCESS; CloseServiceHandle (hLogService); }
if (lStatus == ERROR_SUCCESS) { // wait until the registry is updated before continuing
while (lStatus = RegOpenKeyEx ( HKEY_LOCAL_MACHINE, TEXT("System\\CurrentControlSet\\Services\\PerfDataLog"), 0, KEY_READ | KEY_WRITE, &hKeyPerfLog) != ERROR_SUCCESS) { Sleep (500); // wait .5 seconds and try again
if (--dwWaitLimit == 0) { AfxMessageBox (IDS_SC_CREATE_ERROR); break; } }
if (lStatus == ERROR_SUCCESS) { RegCloseKey (hKeyPerfLog); } }
CloseServiceHandle (hSC); }
return lStatus;} LONG CPdlConfigApp::InitPerfDataLogRegistry (){ LONG lStatus = ERROR_SUCCESS;
HKEY hKeyPerfLog = NULL; HKEY hKeyLogQueries = NULL; HKEY hKeyLogQueriesDefault = NULL;
HKEY hKeyPerfAlert = NULL; HKEY hKeyAlertQueries = NULL; HKEY hKeyAlertQueriesDefault = NULL;
HKEY hKeyEventLogApplication = NULL; HKEY hKeyEventLogPerfDataLog = NULL; HKEY hKeyEventLogPerfDataAlert = NULL;
DWORD dwValue; DWORD dwDisposition;
// open registry key
lStatus = RegOpenKeyEx ( HKEY_LOCAL_MACHINE, TEXT("System\\CurrentControlSet\\Services\\PerfDataLog"), 0, KEY_READ | KEY_WRITE, &hKeyPerfLog); ASSERT (lStatus == ERROR_SUCCESS); if (lStatus != ERROR_SUCCESS) goto Close_And_Exit;
// add registry subkeys for Log Queries
lStatus = RegCreateKeyEx ( hKeyPerfLog, TEXT("Log Queries"), 0, NULL, REG_OPTION_NON_VOLATILE, KEY_READ | KEY_WRITE, NULL, &hKeyLogQueries, &dwDisposition); ASSERT (lStatus == ERROR_SUCCESS); if (lStatus != ERROR_SUCCESS) goto Close_And_Exit;
lStatus = RegCreateKeyEx ( hKeyLogQueries, TEXT("Default"), 0, NULL, REG_OPTION_NON_VOLATILE, KEY_READ | KEY_WRITE, NULL, &hKeyLogQueriesDefault, &dwDisposition); ASSERT (lStatus == ERROR_SUCCESS); if (lStatus != ERROR_SUCCESS) goto Close_And_Exit;
// open registry key
lStatus = RegOpenKeyEx ( HKEY_LOCAL_MACHINE, TEXT("System\\CurrentControlSet\\Services\\EventLog\\Application"), 0, KEY_READ | KEY_WRITE, &hKeyEventLogApplication); ASSERT (lStatus == ERROR_SUCCESS); if (lStatus != ERROR_SUCCESS) goto Close_And_Exit;
// add registry subkeys for event log
lStatus = RegCreateKeyEx ( hKeyEventLogApplication, TEXT("PerfDataLog"), 0, NULL, REG_OPTION_NON_VOLATILE, KEY_READ | KEY_WRITE, NULL, &hKeyEventLogPerfDataLog, &dwDisposition); ASSERT (lStatus == ERROR_SUCCESS); if (lStatus != ERROR_SUCCESS) goto Close_And_Exit;
lStatus = RegSetValueEx (hKeyEventLogPerfDataLog, TEXT("EventMessageFile"), 0, REG_SZ, (BYTE *)TEXT("%systemroot%\\system32\\pdlsvc.exe"), (lstrlen(TEXT("%systemroot%\\system32\\pdlsvc.exe"))+1) * sizeof (TCHAR)); ASSERT (lStatus == ERROR_SUCCESS); if (lStatus != ERROR_SUCCESS) goto Close_And_Exit;
dwValue = 7; lStatus = RegSetValueEx (hKeyEventLogPerfDataLog, TEXT("TypesSupported"), 0, REG_DWORD, (BYTE *)&dwValue, sizeof (DWORD)); ASSERT (lStatus == ERROR_SUCCESS); if (lStatus != ERROR_SUCCESS) goto Close_And_Exit;
Close_And_Exit:
if (hKeyPerfLog != NULL) RegCloseKey (hKeyPerfLog); if (hKeyLogQueries != NULL) RegCloseKey (hKeyLogQueries); if (hKeyLogQueriesDefault != NULL) RegCloseKey (hKeyLogQueriesDefault);
if (hKeyEventLogApplication != NULL) RegCloseKey (hKeyEventLogApplication); if (hKeyEventLogPerfDataLog != NULL) RegCloseKey (hKeyEventLogPerfDataLog);
return lStatus;} /////////////////////////////////////////////////////////////////////////////
// CPdlConfigApp initialization
BOOL CPdlConfigApp::InitInstance(){ LONG lServiceStatus; CString csMessage; BOOL bReturn = TRUE;
// Standard initialization
// If you are not using these features and wish to reduce the size
// of your final executable, you should remove from the following
// the specific initialization routines you do not need.
#ifdef _AFXDLL
Enable3dControls(); // Call this when using MFC in a shared DLL
#else
Enable3dControlsStatic(); // Call this when linking to MFC statically
#endif
lServiceStatus = PerfLogServiceStatus();
if (lServiceStatus == ERROR_FILE_NOT_FOUND) { lServiceStatus = ServiceFilesCopied(); if (lServiceStatus == ERROR_SUCCESS) { if (AfxMessageBox (IDS_QUERY_INSTALL, MB_OKCANCEL | MB_ICONQUESTION) == IDOK) { lServiceStatus = CreatePerfDataLogService(); if (lServiceStatus == ERROR_SUCCESS) { lServiceStatus = InitPerfDataLogRegistry(); } if (lServiceStatus != ERROR_SUCCESS) { csMessage.FormatMessage (lServiceStatus); AfxMessageBox (csMessage); bReturn = FALSE; } else { // the service was successfully installed so display
// message
AfxMessageBox (IDS_SERVICE_INSTALLED); } } else { // the service is not installed and the user doesn't
// want it installed so exit
bReturn = FALSE; } } else { // then the service has not yet been installed so bail
AfxMessageBox (IDS_SERVICE_NOT_INSTALLED); bReturn = FALSE; } } else if (lServiceStatus == ERROR_ACCESS_DENIED) { AfxMessageBox (IDS_ACCESS_DENIED, MB_OK | MB_ICONEXCLAMATION); bReturn = FALSE; } else if (lServiceStatus != ERROR_SUCCESS) { AfxMessageBox (IDS_REGISTRY_ERROR, MB_OK | MB_ICONEXCLAMATION); bReturn = FALSE; } else { bReturn = TRUE; }
if (bReturn) { CPropertySheet PSheet; COutputPropPage POutput; CSettingsPropPage PSettings;
m_pMainWnd = &PSheet;
csMessage.LoadString (IDS_PROPERTY_SHEET_CAPTION); PSheet.SetTitle(csMessage); PSheet.AddPage(&PSettings); PSheet.AddPage(&POutput);
INT_PTR nResponse = PSheet.DoModal(); if (nResponse == IDOK) { // TODO: Place code here to handle when the dialog is
// dismissed with OK
} else if (nResponse == IDCANCEL) { // TODO: Place code here to handle when the dialog is
// dismissed with Cancel
} bReturn = FALSE; }
// Since the dialog has been closed, return FALSE so that we exit the
// application, rather than start the application's message pump.
return bReturn;}
|
