435 lines
7.8 KiB
C++
435 lines
7.8 KiB
C++
|
// File: DebSpew.cpp
|
|||
|
|
|
|||
|
|
#include "precomp.h"
|
|||
|
|
#include <confreg.h>
|
|||
|
|
#include <RegEntry.h>
|
|||
|
|
|
|||
|
|
|
|||
|
|
#ifdef DEBUG /* THE WHOLE FILE! */
|
|||
|
|
|
|||
|
|
#if defined (_M_IX86)
|
|||
|
|
#define _DbgBreak() __asm { int 3 }
|
|||
|
|
#else
|
|||
|
|
#define _DbgBreak() DebugBreak()
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
/* Types
|
|||
|
|
********/
|
|||
|
|
|
|||
|
|
PCSTR g_pcszSpewModule = NULL;
|
|||
|
|
|
|||
|
|
/* debug flags */
|
|||
|
|
|
|||
|
|
typedef enum _debugdebugflags
|
|||
|
|
{
|
|||
|
|
DEBUG_DFL_ENABLE_TRACE_MESSAGES = 0x0001,
|
|||
|
|
|
|||
|
|
DEBUG_DFL_LOG_TRACE_MESSAGES = 0x0002,
|
|||
|
|
|
|||
|
|
DEBUG_DFL_ENABLE_CALL_TRACING = 0x0008,
|
|||
|
|
|
|||
|
|
DEBUG_DFL_INDENT = 0x2000,
|
|||
|
|
|
|||
|
|
ALL_DEBUG_DFLAGS = (DEBUG_DFL_ENABLE_TRACE_MESSAGES |
|
|||
|
|
DEBUG_DFL_LOG_TRACE_MESSAGES |
|
|||
|
|
DEBUG_DFL_ENABLE_CALL_TRACING |
|
|||
|
|
DEBUG_DFL_INDENT)
|
|||
|
|
}
|
|||
|
|
DEBUGDEBUGFLAGS;
|
|||
|
|
|
|||
|
|
|
|||
|
|
/* Global Variables
|
|||
|
|
*******************/
|
|||
|
|
|
|||
|
|
|
|||
|
|
#pragma data_seg(DATA_SEG_PER_INSTANCE)
|
|||
|
|
|
|||
|
|
/* parameters used by SpewOut() */
|
|||
|
|
|
|||
|
|
DWORD g_dwSpewFlags = 0;
|
|||
|
|
UINT g_uSpewSev = 0;
|
|||
|
|
UINT g_uSpewLine = 0;
|
|||
|
|
PCSTR g_pcszSpewFile = NULL;
|
|||
|
|
|
|||
|
|
HDBGZONE ghDbgZone = NULL;
|
|||
|
|
|
|||
|
|
|
|||
|
|
/* TLS slot used to store stack depth for SpewOut() indentation */
|
|||
|
|
|
|||
|
|
#ifdef _DBGSTACK
|
|||
|
|
DWORD s_dwStackDepthSlot = TLS_OUT_OF_INDEXES;
|
|||
|
|
|
|||
|
|
/* hack stack depth counter used until s_dwStackDepthSlot is not available */
|
|||
|
|
|
|||
|
|
ULONG_PTR s_ulcHackStackDepth = 0;
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
/* debug flags */
|
|||
|
|
|
|||
|
|
DWORD s_dwDebugModuleFlags = 0;
|
|||
|
|
|
|||
|
|
#pragma data_seg()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
/***************************** Private Functions *****************************/
|
|||
|
|
|
|||
|
|
/* Module Prototypes
|
|||
|
|
********************/
|
|||
|
|
|
|||
|
|
BOOL IsValidSpewSev(UINT);
|
|||
|
|
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
** IsValidSpewSev()
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
** Arguments:
|
|||
|
|
**
|
|||
|
|
** Returns:
|
|||
|
|
**
|
|||
|
|
** Side Effects: none
|
|||
|
|
*/
|
|||
|
|
BOOL IsValidSpewSev(UINT uSpewSev)
|
|||
|
|
{
|
|||
|
|
BOOL bResult;
|
|||
|
|
|
|||
|
|
switch (uSpewSev)
|
|||
|
|
{
|
|||
|
|
case SPEW_TRACE:
|
|||
|
|
case SPEW_CALLTRACE:
|
|||
|
|
case SPEW_WARNING:
|
|||
|
|
case SPEW_ERROR:
|
|||
|
|
case SPEW_FATAL:
|
|||
|
|
bResult = TRUE;
|
|||
|
|
break;
|
|||
|
|
|
|||
|
|
default:
|
|||
|
|
ERROR_OUT(("IsValidSpewSev(): Invalid debug spew severity %u.",
|
|||
|
|
uSpewSev));
|
|||
|
|
bResult = FALSE;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return(bResult);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
/****************************** Public Functions *****************************/
|
|||
|
|
|
|||
|
|
|
|||
|
|
DWORD NMINTERNAL GetDebugOutputFlags(VOID)
|
|||
|
|
{
|
|||
|
|
return s_dwDebugModuleFlags;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
VOID NMINTERNAL SetDebugOutputFlags(DWORD dw)
|
|||
|
|
{
|
|||
|
|
ASSERT(FLAGS_ARE_VALID(dw, ALL_DEBUG_DFLAGS));
|
|||
|
|
s_dwDebugModuleFlags = dw;
|
|||
|
|
|
|||
|
|
// Save changed data back to registry
|
|||
|
|
RegEntry re(DEBUG_KEY, HKEY_LOCAL_MACHINE);
|
|||
|
|
re.SetValue(REGVAL_DBG_SPEWFLAGS, dw);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
** InitDebugModule()
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
** Arguments:
|
|||
|
|
**
|
|||
|
|
** Returns:
|
|||
|
|
**
|
|||
|
|
** Side Effects: none
|
|||
|
|
*/
|
|||
|
|
BOOL NMINTERNAL InitDebugModule(PCSTR pcszSpewModule)
|
|||
|
|
{
|
|||
|
|
RegEntry re(DEBUG_KEY, HKEY_LOCAL_MACHINE);
|
|||
|
|
|
|||
|
|
s_dwDebugModuleFlags = re.GetNumber(REGVAL_DBG_SPEWFLAGS, DEFAULT_DBG_SPEWFLAGS);
|
|||
|
|
|
|||
|
|
|
|||
|
|
g_pcszSpewModule = pcszSpewModule;
|
|||
|
|
|
|||
|
|
#ifdef _DBGSTACK
|
|||
|
|
|
|||
|
|
ASSERT(s_dwStackDepthSlot == TLS_OUT_OF_INDEXES);
|
|||
|
|
|
|||
|
|
s_dwStackDepthSlot = TlsAlloc();
|
|||
|
|
|
|||
|
|
if (s_dwStackDepthSlot != TLS_OUT_OF_INDEXES)
|
|||
|
|
{
|
|||
|
|
EVAL(TlsSetValue(s_dwStackDepthSlot, (PVOID)s_ulcHackStackDepth));
|
|||
|
|
|
|||
|
|
TRACE_OUT(("InitDebugModule(): Using thread local storage slot %lu for debug stack depth counter.",
|
|||
|
|
s_dwStackDepthSlot));
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
WARNING_OUT(("InitDebugModule(): TlsAlloc() failed to allocate thread local storage for debug stack depth counter."));
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
if (NULL == ghDbgZone)
|
|||
|
|
{
|
|||
|
|
PSTR rgsz[4];
|
|||
|
|
rgsz[0] = (PSTR) pcszSpewModule;
|
|||
|
|
|
|||
|
|
ASSERT(0 == ZONE_WARNING);
|
|||
|
|
rgsz[1+ZONE_WARNING] = "Warning";
|
|||
|
|
|
|||
|
|
ASSERT(1 == ZONE_TRACE);
|
|||
|
|
rgsz[1+ZONE_TRACE] = "Trace";
|
|||
|
|
|
|||
|
|
ASSERT(2 == ZONE_FUNCTION);
|
|||
|
|
rgsz[1+ZONE_FUNCTION] = "Function";
|
|||
|
|
|
|||
|
|
// Initialize standard debug settings with warning enabled by default
|
|||
|
|
DbgInitEx(&ghDbgZone, rgsz, 3, 0x01);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return TRUE;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
** ExitDebugModule()
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
** Arguments:
|
|||
|
|
**
|
|||
|
|
** Returns:
|
|||
|
|
**
|
|||
|
|
** Side Effects: none
|
|||
|
|
*/
|
|||
|
|
void NMINTERNAL ExitDebugModule(void)
|
|||
|
|
{
|
|||
|
|
#ifdef _DBGSTACK
|
|||
|
|
|
|||
|
|
if (s_dwStackDepthSlot != TLS_OUT_OF_INDEXES)
|
|||
|
|
{
|
|||
|
|
s_ulcHackStackDepth = ((ULONG_PTR)TlsGetValue(s_dwStackDepthSlot));
|
|||
|
|
|
|||
|
|
/* Leave s_ulcHackStackDepth == 0 if TlsGetValue() fails. */
|
|||
|
|
|
|||
|
|
EVAL(TlsFree(s_dwStackDepthSlot));
|
|||
|
|
s_dwStackDepthSlot = TLS_OUT_OF_INDEXES;
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
g_pcszSpewModule = NULL;
|
|||
|
|
DBGDEINIT(&ghDbgZone);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
** StackEnter()
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
** Arguments:
|
|||
|
|
**
|
|||
|
|
** Returns:
|
|||
|
|
**
|
|||
|
|
** Side Effects: none
|
|||
|
|
*/
|
|||
|
|
void NMINTERNAL StackEnter(void)
|
|||
|
|
{
|
|||
|
|
#ifdef _DBGSTACK
|
|||
|
|
|
|||
|
|
if (s_dwStackDepthSlot != TLS_OUT_OF_INDEXES)
|
|||
|
|
{
|
|||
|
|
ULONG_PTR ulcDepth;
|
|||
|
|
|
|||
|
|
ulcDepth = ((ULONG_PTR)TlsGetValue(s_dwStackDepthSlot));
|
|||
|
|
|
|||
|
|
ASSERT(ulcDepth < ULONG_MAX);
|
|||
|
|
|
|||
|
|
EVAL(TlsSetValue(s_dwStackDepthSlot, (PVOID)(ulcDepth + 1)));
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
ASSERT(s_ulcHackStackDepth < ULONG_MAX);
|
|||
|
|
s_ulcHackStackDepth++;
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
** StackLeave()
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
** Arguments:
|
|||
|
|
**
|
|||
|
|
** Returns:
|
|||
|
|
**
|
|||
|
|
** Side Effects: none
|
|||
|
|
*/
|
|||
|
|
void NMINTERNAL StackLeave(void)
|
|||
|
|
{
|
|||
|
|
#ifdef _DBGSTACK
|
|||
|
|
|
|||
|
|
if (s_dwStackDepthSlot != TLS_OUT_OF_INDEXES)
|
|||
|
|
{
|
|||
|
|
ULONG_PTR ulcDepth;
|
|||
|
|
|
|||
|
|
ulcDepth = ((ULONG_PTR)TlsGetValue(s_dwStackDepthSlot));
|
|||
|
|
|
|||
|
|
if (EVAL(ulcDepth > 0))
|
|||
|
|
EVAL(TlsSetValue(s_dwStackDepthSlot, (PVOID)(ulcDepth - 1)));
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
if (EVAL(s_ulcHackStackDepth > 0))
|
|||
|
|
s_ulcHackStackDepth--;
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
** GetStackDepth()
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
**
|
|||
|
|
** Arguments:
|
|||
|
|
**
|
|||
|
|
** Returns:
|
|||
|
|
**
|
|||
|
|
** Side Effects: none
|
|||
|
|
*/
|
|||
|
|
ULONG_PTR NMINTERNAL GetStackDepth(void)
|
|||
|
|
{
|
|||
|
|
ULONG_PTR ulcDepth = 0;
|
|||
|
|
#ifdef _DBGSTACK
|
|||
|
|
|
|||
|
|
if (s_dwStackDepthSlot != TLS_OUT_OF_INDEXES)
|
|||
|
|
ulcDepth = (ULONG)((ULONG_PTR)TlsGetValue(s_dwStackDepthSlot));
|
|||
|
|
else
|
|||
|
|
ulcDepth = s_ulcHackStackDepth;
|
|||
|
|
#endif
|
|||
|
|
return(ulcDepth);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
/* _ D B G Z P R I N T M S G */
|
|||
|
|
/*-------------------------------------------------------------------------
|
|||
|
|
%%Function: _DbgZPrintMsg
|
|||
|
|
|
|||
|
|
-------------------------------------------------------------------------*/
|
|||
|
|
static VOID _DbgZPrintMsg(UINT iZone, PSTR pszFormat, va_list arglist)
|
|||
|
|
{
|
|||
|
|
PCSTR pcszSpewPrefix;
|
|||
|
|
char szModule[128];
|
|||
|
|
|
|||
|
|
if (g_pcszSpewModule)
|
|||
|
|
{
|
|||
|
|
switch (iZone)
|
|||
|
|
{
|
|||
|
|
case ZONE_TRACE:
|
|||
|
|
pcszSpewPrefix = "Trace";
|
|||
|
|
break;
|
|||
|
|
case ZONE_FUNCTION:
|
|||
|
|
pcszSpewPrefix = "Func ";
|
|||
|
|
break;
|
|||
|
|
case ZONE_WARNING:
|
|||
|
|
pcszSpewPrefix = "Warn ";
|
|||
|
|
break;
|
|||
|
|
default:
|
|||
|
|
pcszSpewPrefix = "?????";
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
wsprintfA(szModule, "%s:%s", g_pcszSpewModule, pcszSpewPrefix);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
// No module nonsense, empty prefix
|
|||
|
|
wsprintfA(szModule, "%s", "");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (IS_FLAG_CLEAR(s_dwDebugModuleFlags, DEBUG_DFL_INDENT))
|
|||
|
|
{
|
|||
|
|
// Don't indent output
|
|||
|
|
DbgPrintf(szModule, pszFormat, arglist);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
PCSTR pcszIndent;
|
|||
|
|
ULONG_PTR ulcStackDepth;
|
|||
|
|
char szFormat[512];
|
|||
|
|
static char _szSpewLeader[] = " ";
|
|||
|
|
|
|||
|
|
ulcStackDepth = GetStackDepth();
|
|||
|
|
if (ulcStackDepth > sizeof(_szSpewLeader))
|
|||
|
|
ulcStackDepth = sizeof(_szSpewLeader);
|
|||
|
|
|
|||
|
|
pcszIndent = _szSpewLeader + sizeof(_szSpewLeader) - ulcStackDepth;
|
|||
|
|
|
|||
|
|
wsprintfA(szFormat, "%s%s", pcszIndent, pszFormat);
|
|||
|
|
DbgPrintf(szModule, szFormat, arglist);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
VOID WINAPI DbgZPrintError(PSTR pszFormat,...)
|
|||
|
|
{
|
|||
|
|
va_list v1;
|
|||
|
|
va_start(v1, pszFormat);
|
|||
|
|
|
|||
|
|
_DbgZPrintMsg(ZONE_WARNING, pszFormat, v1);
|
|||
|
|
va_end(v1);
|
|||
|
|
|
|||
|
|
_DbgBreak();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
VOID WINAPI DbgZPrintWarning(PSTR pszFormat,...)
|
|||
|
|
{
|
|||
|
|
if (GETZONEMASK(ghDbgZone) & ZONE_WARNING_FLAG)
|
|||
|
|
{
|
|||
|
|
va_list v1;
|
|||
|
|
va_start(v1, pszFormat);
|
|||
|
|
|
|||
|
|
_DbgZPrintMsg(ZONE_WARNING, pszFormat, v1);
|
|||
|
|
va_end(v1);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
VOID WINAPI DbgZPrintTrace(PSTR pszFormat,...)
|
|||
|
|
{
|
|||
|
|
if (GETZONEMASK(ghDbgZone) & ZONE_TRACE_FLAG)
|
|||
|
|
{
|
|||
|
|
va_list v1;
|
|||
|
|
va_start(v1, pszFormat);
|
|||
|
|
_DbgZPrintMsg(ZONE_TRACE, pszFormat, v1);
|
|||
|
|
va_end(v1);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
VOID WINAPI DbgZPrintFunction(PSTR pszFormat,...)
|
|||
|
|
{
|
|||
|
|
if (GETZONEMASK(ghDbgZone) & ZONE_FUNCTION_FLAG)
|
|||
|
|
{
|
|||
|
|
va_list v1;
|
|||
|
|
va_start(v1, pszFormat);
|
|||
|
|
_DbgZPrintMsg(ZONE_FUNCTION, pszFormat, v1);
|
|||
|
|
va_end(v1);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
#endif /* DEBUG */
|
|||
|
|
|
|||
|
|
|