121 lines
2.9 KiB
C++
121 lines
2.9 KiB
C++
|
#include "wsdueng.h"
|
||
|
|
|
||
|
|
#define REGKEY_WUV3TEST "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\WindowsUpdate\\wuv3test"
|
||
|
|
|
||
|
|
FILE* CLogger::c_pfile = stdout;
|
||
|
|
int CLogger::c_cnIndent = 0;
|
||
|
|
int CLogger::c_cnLevels = -1;
|
||
|
|
|
||
|
|
CLogger::CLogger(
|
||
|
|
const char* szBlockName /*= 0*/,
|
||
|
|
int nLoggingLevel/*= 0*/,
|
||
|
|
const char* szFileName/*= 0*/,
|
||
|
|
int nLine/*= 0*/
|
||
|
|
)
|
||
|
|
{
|
||
|
|
if (-1 == c_cnLevels)
|
||
|
|
{
|
||
|
|
/*c_cnLevels = 0;
|
||
|
|
HKEY hkey;
|
||
|
|
if (NO_ERROR == RegOpenKeyEx(HKEY_LOCAL_MACHINE, REGKEY_WUV3TEST, 0, KEY_READ, &hkey))
|
||
|
|
{
|
||
|
|
DWORD dwSize = sizeof(c_cnLevels);
|
||
|
|
RegQueryValueEx(hkey, "LogLevel", 0, 0, (LPBYTE)&c_cnLevels, &dwSize);
|
||
|
|
char szLogFile[MAX_PATH] = {0};
|
||
|
|
dwSize = sizeof(szLogFile);
|
||
|
|
RegQueryValueEx(hkey, "LogFile", 0, 0, (LPBYTE)&szLogFile, &dwSize);
|
||
|
|
FILE* pfile = fopen(szLogFile, "at");
|
||
|
|
if (pfile)
|
||
|
|
{
|
||
|
|
c_pfile = pfile;
|
||
|
|
}
|
||
|
|
RegCloseKey(hkey);
|
||
|
|
}*/
|
||
|
|
char sz_LogFileName[MAX_PATH];
|
||
|
|
c_cnLevels = 4; // default to log level 4
|
||
|
|
ExpandEnvironmentStrings(cszLoggingFile, sz_LogFileName, MAX_PATH);
|
||
|
|
FILE* pfile = fopen(sz_LogFileName, "at");
|
||
|
|
if (pfile)
|
||
|
|
{
|
||
|
|
c_pfile = pfile;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
m_szBlockName[0] = 0;
|
||
|
|
//m_fOut = nLoggingLevel < c_cnLevels;
|
||
|
|
m_fOut = TRUE;
|
||
|
|
if (m_fOut && NULL != szBlockName)
|
||
|
|
{
|
||
|
|
lstrcpyn(m_szBlockName, szBlockName, sizeOfArray(m_szBlockName));
|
||
|
|
//out("%s %s(%d)", szBlockName, szFileName, nLine);
|
||
|
|
out("%s", szBlockName);
|
||
|
|
m_dwStartTick = GetTickCount();
|
||
|
|
c_cnIndent ++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
CLogger::~CLogger()
|
||
|
|
{
|
||
|
|
if (c_pfile && m_fOut && NULL != m_szBlockName[0])
|
||
|
|
{
|
||
|
|
c_cnIndent --;
|
||
|
|
out("~%s (%d msecs)", m_szBlockName, GetTickCount() - m_dwStartTick);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void __cdecl CLogger::out(const char *szFormat, ...)
|
||
|
|
{
|
||
|
|
if (m_fOut)
|
||
|
|
{
|
||
|
|
va_list va;
|
||
|
|
va_start (va, szFormat);
|
||
|
|
v_out(szFormat, va);
|
||
|
|
va_end (va);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void __cdecl CLogger::error(const char *szFormat, ...)
|
||
|
|
{
|
||
|
|
if (m_fOut)
|
||
|
|
{
|
||
|
|
va_list va;
|
||
|
|
va_start (va, szFormat);
|
||
|
|
char szOut[4 * 1024];
|
||
|
|
wvsprintf(szOut, szFormat, va);
|
||
|
|
va_end (va);
|
||
|
|
out("ERROR - %s", szOut);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void __cdecl CLogger::out1(const char *szFormat, ...)
|
||
|
|
{
|
||
|
|
CLogger logger;
|
||
|
|
va_list va;
|
||
|
|
va_start (va, szFormat);
|
||
|
|
logger.v_out(szFormat, va);
|
||
|
|
va_end (va);
|
||
|
|
}
|
||
|
|
|
||
|
|
void CLogger::v_out( const char* szFormat, va_list va)
|
||
|
|
{
|
||
|
|
char szOut[5 * 1024];
|
||
|
|
char* pszOut = szOut;
|
||
|
|
// Indent first
|
||
|
|
for(int i = 0; i < c_cnIndent; i ++)
|
||
|
|
*(pszOut ++) = '\t';
|
||
|
|
wvsprintf(pszOut, szFormat, va);
|
||
|
|
// move the file pointer to the end
|
||
|
|
if (0 == fseek(c_pfile, 0, SEEK_END))
|
||
|
|
{
|
||
|
|
fprintf(c_pfile, "%s\n", szOut);
|
||
|
|
fflush(c_pfile);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void CLogger::close(void)
|
||
|
|
{
|
||
|
|
if (c_pfile) fclose(c_pfile);
|
||
|
|
c_pfile = NULL;
|
||
|
|
c_cnLevels = -1;
|
||
|
|
}
|
||
|
|
|