252 lines
5.8 KiB
C
252 lines
5.8 KiB
C
//*************************************************************
|
|
//
|
|
// Debugging functions
|
|
//
|
|
// Microsoft Confidential
|
|
// Copyright (c) Microsoft Corporation 1995
|
|
// All rights reserved
|
|
//
|
|
//*************************************************************
|
|
|
|
#include <windows.h>
|
|
#include <strsafe.h>
|
|
#include "debug.h"
|
|
|
|
#if DBG
|
|
|
|
#define ARRAYSIZE(a) (sizeof(a)/sizeof(a[0]))
|
|
|
|
//
|
|
// Global Variable containing the debugging level.
|
|
//
|
|
|
|
DWORD dwDebugLevel;
|
|
|
|
//
|
|
// Debug strings
|
|
//
|
|
|
|
const TCHAR c_szProQuota[] = TEXT("PROQUOTA(%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("ProQuotaDebugLevel")
|
|
|
|
TCHAR szLogFileName[] = L"%SystemRoot%\\Debug\\UserMode\\proquota.log"; // Current log
|
|
|
|
|
|
//*************************************************************
|
|
//
|
|
// 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 = RegOpenKey (HKEY_LOCAL_MACHINE, DEBUG_REG_LOCATION,
|
|
&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;
|
|
HRESULT hr;
|
|
|
|
|
|
//
|
|
// 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) {
|
|
hr = StringCchPrintf(szDebugTitle, ARRAYSIZE(szDebugTitle), c_szProQuota, GetCurrentProcessId());
|
|
if (FAILED(hr))
|
|
return;
|
|
OutputDebugString(szDebugTitle);
|
|
|
|
va_start(marker, pszMsg);
|
|
hr = StringCchVPrintf(szDebugBuffer, ARRAYSIZE(szDebugBuffer), pszMsg, marker);
|
|
if (FAILED(hr))
|
|
return;
|
|
OutputDebugString(szDebugBuffer);
|
|
OutputDebugString(c_szCRLF);
|
|
va_end(marker);
|
|
|
|
if (dwDebugLevel & DL_LOGFILE) {
|
|
HANDLE hFile;
|
|
DWORD dwBytesWritten;
|
|
TCHAR szExpLogFileName[MAX_PATH+1];
|
|
|
|
DWORD dwRet = ExpandEnvironmentStrings ( szLogFileName, szExpLogFileName, ARRAYSIZE(szExpLogFileName));
|
|
|
|
if ( dwRet != 0 && dwRet <= MAX_PATH) {
|
|
|
|
hFile = CreateFile(szExpLogFileName,
|
|
FILE_WRITE_DATA | FILE_APPEND_DATA,
|
|
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
|