windows-xp/Source/XPSP1/NT/enduser/windows.com/wuv3/wsdu/wsdueng/log.cpp

#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;
}