TnbEventLogger.h

[詳解]

#pragma once
#include "TnbSingleton.h"
#include "TnbStrEx.h"
#include "TnbRegistryAccessor.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CEventLogger
{
    SINGLETON_CONSTRUCTOR(CEventLogger)
    {
        m_sourceName = GetProcessName();
    }
public:
 
    void SetSourceName(LPCTSTR lpszSourceName)
    {
        m_sourceName = lpszSourceName;
    }
 
    LPCTSTR GetSourceName(void) const
    {
        return m_sourceName;
    }
 
    bool RegistEventSource(LPCTSTR lpszModuleName = NULL)
    {
        HMODULE h = NULL;
        if ( lpszModuleName != NULL )
        {
            h = ::LoadLibrary(lpszModuleName);
        }
        CStr path;
        DWORD dwRc = ::GetModuleFileName(h, path.GetBuffer(MAX_PATH), MAX_PATH);
        path.ReleaseBuffer();
        if ( dwRc > 0 )
        {
            CRegistryAccessor reg(HKEY_LOCAL_MACHINE, m_KeyName());
            CRegistryAccessor::CSection sec = reg[m_sourceName];
            sec.WriteString(_T("EventMessageFile"), path);
        }
        return (dwRc > 0);
    }
 
    void UnregistEventSource(void)
    {
        if ( ! m_sourceName.IsEmpty() )
        {
            CRegistryAccessor reg(HKEY_LOCAL_MACHINE, m_KeyName());
            reg.DeleteSection(m_sourceName);
        }
    }
 
    bool Information(DWORD eventId, LPCTSTR lpszFmt, ...) const
    {
        va_list args;
        va_start(args, lpszFmt);
        bool r = m_Write(EVENTLOG_INFORMATION_TYPE, eventId, 0, NULL, lpszFmt, args);
        va_end(args);
        return r;
    }
 
    bool Information(DWORD eventId, size_t dataSize, LPCVOID lpData, LPCTSTR lpszFmt, ...) const
    {
        ASSERT( ! ::IsBadReadPtr(lpData, dataSize) );
        va_list args;
        va_start(args, lpszFmt);
        bool r = m_Write(EVENTLOG_INFORMATION_TYPE, eventId, dataSize, lpData, lpszFmt, args);
        va_end(args);
        return r;
    }
 
    bool Warning(DWORD eventId, LPCTSTR lpszFmt, ...) const
    {
        va_list args;
        va_start(args, lpszFmt);
        bool r = m_Write(EVENTLOG_WARNING_TYPE, eventId, 0, NULL, lpszFmt, args);
        va_end(args);
        return r;
    }
 
    bool Warning(DWORD eventId, size_t dataSize, LPCVOID lpData, LPCTSTR lpszFmt, ...) const
    {
        ASSERT( ! ::IsBadReadPtr(lpData, dataSize) );
        va_list args;
        va_start(args, lpszFmt);
        bool r = m_Write(EVENTLOG_WARNING_TYPE, eventId, dataSize, lpData, lpszFmt, args);
        va_end(args);
        return r;
    }
 
    bool Error(DWORD eventId, LPCTSTR lpszFmt, ...) const
    {
        va_list args;
        va_start(args, lpszFmt);
        bool r = m_Write(EVENTLOG_ERROR_TYPE, eventId, 0, NULL, lpszFmt, args);
        va_end(args);
        return r;
    }
 
    bool Error(DWORD eventId, size_t dataSize, LPCVOID lpData, LPCTSTR lpszFmt, ...) const
    {
        ASSERT( ! ::IsBadReadPtr(lpData, dataSize) );
        va_list args;
        va_start(args, lpszFmt);
        bool r = m_Write(EVENTLOG_ERROR_TYPE, eventId, dataSize, lpData, lpszFmt, args);
        va_end(args);
        return r;
    }
 
private:
    // キー名
    LPCTSTR m_KeyName(void) const
    {
        return _T("SYSTEM\\CurrentControlSet\\Services\\Eventlog\\Application");
    }
    // 書き込み
    bool m_Write(WORD type, DWORD eventId, size_t binarySize, LPCVOID lpBinary, LPCTSTR lpszFmt, va_list V) const
    {
        bool r = false;
        CStr str;
        str.FormatV(lpszFmt, V);
        LPCTSTR lpszMsg = str;
        HANDLE h = ::RegisterEventSource(NULL, m_sourceName);
        if ( h == NULL )
        {
            _GetLastError("RegisterEventSource");
            return false;
        }
        r = !! ::ReportEvent(h, type, 0, eventId, NULL, 1, static_cast<DWORD>(binarySize), &lpszMsg, const_cast<LPVOID>(lpBinary));
        if ( ! r )
        {
            _GetLastError("ReportEvent");
        }
        ::DeregisterEventSource(h);
        return r;
    }
 
    CStr        m_sourceName;   
};
 
 
 
}; // TNB