141 lines
3.7 KiB
C++
141 lines
3.7 KiB
C++
//+---------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 2000-2001.
|
|
//
|
|
// File: debug.cpp
|
|
//
|
|
// Contents: Debugging support
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
#include "stdafx.h"
|
|
|
|
#if DBG == 1
|
|
static int indentLevel = 0;
|
|
|
|
#define DEBUGKEY L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\AdminDebug\\LogHours"
|
|
|
|
#define DEBUG_OUTPUT_NONE 0
|
|
#define DEBUG_OUTPUT_ERROR 1
|
|
#define DEBUG_OUTPUT_WARNING 2
|
|
#define DEBUG_OUTPUT_TRACE 3
|
|
#define DEBUGLEVEL L"debugOutput"
|
|
|
|
static bool g_fDebugOutputLevelInit = false;
|
|
static unsigned long g_ulDebugOutput = DEBUG_OUTPUT_NONE;
|
|
|
|
void __cdecl _TRACE (int level, const wchar_t *format, ... )
|
|
{
|
|
if ( g_ulDebugOutput > DEBUG_OUTPUT_NONE )
|
|
{
|
|
va_list arglist;
|
|
const size_t DEBUG_BUF_LEN = 512;
|
|
WCHAR Buffer[DEBUG_BUF_LEN];
|
|
int cb;
|
|
|
|
if ( level < 0 )
|
|
indentLevel += level;
|
|
//
|
|
// Format the output into a buffer and then print it.
|
|
//
|
|
wstring strTabs;
|
|
|
|
for (int nLevel = 0; nLevel < indentLevel; nLevel++)
|
|
strTabs += L" ";
|
|
|
|
OutputDebugStringW (strTabs.c_str ());
|
|
|
|
va_start(arglist, format);
|
|
|
|
cb = _vsnwprintf (Buffer, DEBUG_BUF_LEN, format, arglist);
|
|
if ( cb )
|
|
{
|
|
OutputDebugStringW (Buffer);
|
|
}
|
|
|
|
va_end(arglist);
|
|
|
|
if ( level > 0 )
|
|
indentLevel += level;
|
|
}
|
|
}
|
|
|
|
|
|
PCSTR
|
|
StripDirPrefixA(
|
|
PCSTR pszPathName
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Strip the directory prefix off a filename (ANSI version)
|
|
|
|
Arguments:
|
|
|
|
pstrFilename - Pointer to filename string
|
|
|
|
Return Value:
|
|
|
|
Pointer to the last component of a filename (without directory prefix)
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD dwLen = lstrlenA(pszPathName);
|
|
|
|
pszPathName += dwLen - 1; // go to the end
|
|
|
|
while (*pszPathName != '\\' && dwLen--)
|
|
{
|
|
pszPathName--;
|
|
}
|
|
|
|
return pszPathName + 1;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
// Function: CheckInit
|
|
//
|
|
// Synopsis: Performs debugging library initialization
|
|
// including reading the registry for the desired infolevel
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
void CheckDebugOutputLevel ()
|
|
{
|
|
if ( g_fDebugOutputLevelInit )
|
|
return;
|
|
g_fDebugOutputLevelInit = true;
|
|
HKEY hKey = 0;
|
|
DWORD dwDisposition = 0;
|
|
LONG lResult = ::RegCreateKeyEx (HKEY_LOCAL_MACHINE, // handle of an open key
|
|
DEBUGKEY, // address of subkey name
|
|
0, // reserved
|
|
L"", // address of class string
|
|
REG_OPTION_NON_VOLATILE, // special options flag
|
|
KEY_ALL_ACCESS, // desired security access
|
|
NULL, // address of key security structure
|
|
&hKey, // address of buffer for opened handle
|
|
&dwDisposition); // address of disposition value buffer
|
|
if (lResult == ERROR_SUCCESS)
|
|
{
|
|
DWORD dwSize = sizeof(unsigned long);
|
|
lResult = RegQueryValueExW (hKey, DEBUGLEVEL, NULL, NULL,
|
|
(LPBYTE)&g_ulDebugOutput, &dwSize);
|
|
if (lResult != ERROR_SUCCESS)
|
|
{
|
|
g_ulDebugOutput = DEBUG_OUTPUT_NONE;
|
|
if ( ERROR_FILE_NOT_FOUND == lResult )
|
|
{
|
|
RegSetValueExW (hKey, DEBUGLEVEL, 0, REG_DWORD,
|
|
(LPBYTE)&g_ulDebugOutput, sizeof (g_ulDebugOutput));
|
|
}
|
|
}
|
|
RegCloseKey(hKey);
|
|
}
|
|
}
|
|
|
|
#endif // if DBG
|