windows-nt/Source/XPSP1/NT/admin/snapin/certmgr/debug.cpp
2020-09-26 16:20:57 +08:00

139 lines
3.5 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 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