|
|
//
// Microsoft Corporation 1998
//
// DEBUG.CPP - Debug library
//
#include "main.h"
#if DBG
//
// Global Variable containing the debugging level.
//
DWORD dwDebugLevel;
//
// Debug strings
//
const TCHAR c_szGPTDemo[] = TEXT("RIGPSNAP(%x): "); const TCHAR c_szCRLF[] = TEXT("\r\n");
//
// Registry debug information
//
#define DEBUG_REG_LOCATION TEXT("Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon")
#define DEBUG_KEY_NAME TEXT("RIGPSNAP")
//*************************************************************
//
// InitDebugSupport()
//
// Purpose: Sets the debugging level.
// Also checks the registry for a debugging level.
//
// Parameters: None
//
// Return: void
//
// Comments:
//
//
// History: Date Author Comment
// 5/25/95 ericflo Created
//
//*************************************************************
void InitDebugSupport(void) { LONG lResult; HKEY hKey; DWORD dwType, dwSize;
//
// Initialize the debug level to normal
//
dwDebugLevel = DL_NORMAL;
//
// Check the registry
//
lResult = RegOpenKeyEx( HKEY_LOCAL_MACHINE, DEBUG_REG_LOCATION, 0, KEY_READ, &hKey);
if (lResult == ERROR_SUCCESS) {
dwSize = sizeof(dwDebugLevel); RegQueryValueEx(hKey, DEBUG_KEY_NAME, NULL, &dwType, (LPBYTE)&dwDebugLevel, &dwSize);
RegCloseKey(hKey); } }
//*************************************************************
//
// DebugMsg()
//
// Purpose: Displays debug messages based on the debug level
// and type of debug message.
//
// Parameters: mask - debug message type
// pszMsg - debug message
// ... - variable number of parameters
//
// Return: void
//
//
// Comments:
//
//
// History: Date Author Comment
// 5/25/95 ericflo Created
//
//*************************************************************
void _DebugMsg(UINT mask, LPCTSTR pszMsg, ...) { BOOL bOutput; TCHAR szDebugTitle[30]; TCHAR szDebugBuffer[2*MAX_PATH+40]; va_list marker; DWORD dwErrCode;
//
// Save the last error code (so the debug output doesn't change it).
//
dwErrCode = GetLastError();
//
// Detemerine the correct amount of debug output
//
switch (LOWORD(dwDebugLevel)) {
case DL_VERBOSE: bOutput = TRUE; break;
case DL_NORMAL:
//
// Normal debug output. Don't
// display verbose stuff, but
// do display warnings/asserts.
//
if (mask != DM_VERBOSE) { bOutput = TRUE; } else { bOutput = FALSE; } break;
case DL_NONE: default:
//
// Only display asserts
//
if (mask == DM_ASSERT) { bOutput = TRUE; } else { bOutput = FALSE; } break; }
//
// Display the error message if appropriate
//
if (bOutput) { wsprintf (szDebugTitle, c_szGPTDemo, GetCurrentProcessId()); OutputDebugString(szDebugTitle);
va_start(marker, pszMsg); wvsprintf(szDebugBuffer, pszMsg, marker); OutputDebugString(szDebugBuffer); OutputDebugString(c_szCRLF); va_end(marker);
if (dwDebugLevel & DL_LOGFILE) { HANDLE hFile; DWORD dwBytesWritten;
hFile = CreateFile(TEXT("c:\\rigpsnap.log"), GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile != INVALID_HANDLE_VALUE) {
if (SetFilePointer (hFile, 0, NULL, FILE_END) != 0xFFFFFFFF) {
WriteFile (hFile, (LPCVOID) szDebugTitle, lstrlen (szDebugTitle) * sizeof(TCHAR), &dwBytesWritten, NULL);
WriteFile (hFile, (LPCVOID) szDebugBuffer, lstrlen (szDebugBuffer) * sizeof(TCHAR), &dwBytesWritten, NULL);
WriteFile (hFile, (LPCVOID) c_szCRLF, lstrlen (c_szCRLF) * sizeof(TCHAR), &dwBytesWritten, NULL); }
CloseHandle (hFile); }
}
}
//
// Restore the last error code
//
SetLastError(dwErrCode);
//
// Break to the debugger if appropriate
//
if (mask == DM_ASSERT) { DebugBreak(); } }
#endif // DBG
|