base
This commit is contained in:
biosvos
@@ -0,0 +1,285 @@
|
||||
// DSLog.cpp: implementation of the Log class.
|
||||
//
|
||||
|
||||
#include "stdafx.h"
|
||||
#include "DSLog.h"
|
||||
|
||||
const DWORD CDSLog::modeDebug = 0x01;
|
||||
const DWORD CDSLog::modeFile = 0x02;
|
||||
const DWORD CDSLog::modeConsole = 0x04;
|
||||
|
||||
const CHAR CDSLog::m_szExt[] = ".log";
|
||||
|
||||
const static int LINE_BUFFER_SIZE = 1024;
|
||||
|
||||
// Create a new dslog object.
|
||||
// nMode - specifies where output should go, using combination
|
||||
// of flags above.
|
||||
// nFlag - the flag
|
||||
// pszFileName - if flag CDSLog::modeFile is specified in the type,
|
||||
// a filename must be specified here.
|
||||
// bAppend - if logging to a file, whether or not to append to any
|
||||
// existing log.
|
||||
// bDaily - if logging to a file, whether or not to replace log every day.
|
||||
|
||||
CDSLog::CDSLog(DWORD dwMode, DWORD dwLevel, LPSTR pszFileName, BOOL bAppend, BOOL bDaily)
|
||||
{
|
||||
m_nLastLogTime = 0;
|
||||
m_szLastLogDay[0] = 0;
|
||||
|
||||
m_pszFileName = NULL;
|
||||
m_hFile = NULL;
|
||||
|
||||
m_pszEventSource = NULL;
|
||||
|
||||
m_bDebugMode = false;
|
||||
m_bFileMode = false;
|
||||
m_bConsoleMode = false;
|
||||
m_bAppend = false;
|
||||
m_bDaily = false;
|
||||
|
||||
m_dwLevel = dwLevel;
|
||||
|
||||
SetFile(pszFileName, bAppend, bDaily);
|
||||
SetMode(dwMode);
|
||||
}
|
||||
|
||||
CDSLog::~CDSLog()
|
||||
{
|
||||
if (m_pszFileName)
|
||||
free(m_pszFileName);
|
||||
|
||||
if (m_pszEventSource)
|
||||
free(m_pszEventSource);
|
||||
|
||||
CloseFile();
|
||||
}
|
||||
|
||||
void CDSLog::EventLog(WORD wType, LPSTR pszFormat, ...)
|
||||
{
|
||||
va_list val;
|
||||
|
||||
va_start(val, pszFormat);
|
||||
|
||||
if (m_pszEventSource) {
|
||||
HANDLE hEventSource;
|
||||
|
||||
CHAR szBuf[256];
|
||||
LPSTR ppszBuf[2];
|
||||
|
||||
vsprintf(szBuf, pszFormat, val);
|
||||
|
||||
hEventSource = RegisterEventSource(NULL, m_pszEventSource);
|
||||
|
||||
ppszBuf[0] = m_pszEventSource;
|
||||
ppszBuf[1] = szBuf;
|
||||
|
||||
if (hEventSource) {
|
||||
ReportEvent(
|
||||
hEventSource, // handle of event source
|
||||
wType, // event type
|
||||
0, // event category
|
||||
0, // event ID
|
||||
NULL, // current user's SID
|
||||
2, // strings in 'strings'
|
||||
0, // no bytes of raw data
|
||||
(const char **)ppszBuf, // array of error strings
|
||||
NULL); // no raw data
|
||||
|
||||
DeregisterEventSource(hEventSource);
|
||||
}
|
||||
}
|
||||
|
||||
if (DSLL_STATE & m_dwLevel)
|
||||
_Print(pszFormat, val);
|
||||
|
||||
va_end(val);
|
||||
}
|
||||
|
||||
void CDSLog::EventLogError(LPSTR pszMsg)
|
||||
{
|
||||
DWORD dwError = GetLastError();
|
||||
TCHAR szBuf[256];
|
||||
|
||||
wsprintf(szBuf, "%s, error code : %d", pszMsg, dwError);
|
||||
EventLog(EVENTLOG_ERROR_TYPE, szBuf);
|
||||
}
|
||||
|
||||
void CDSLog::EventLogWSAError(LPSTR pszMsg)
|
||||
{
|
||||
// DWORD dwError = WSAGetLastError();
|
||||
// TCHAR szBuf[256];
|
||||
|
||||
// wsprintf(szBuf, "%s, WSA error code : %d", pszMsg, dwError);
|
||||
// EventLog(EVENTLOG_ERROR_TYPE, szBuf);
|
||||
}
|
||||
|
||||
void CDSLog::SetMode(DWORD dwMode)
|
||||
{
|
||||
m_bDebugMode = dwMode & modeDebug;
|
||||
|
||||
if (dwMode & modeFile) {
|
||||
if (!m_bFileMode)
|
||||
OpenFile();
|
||||
}
|
||||
else {
|
||||
CloseFile();
|
||||
m_bFileMode = false;
|
||||
}
|
||||
|
||||
if (dwMode & modeConsole) {
|
||||
if (!m_bConsoleMode)
|
||||
AllocConsole();
|
||||
|
||||
m_bConsoleMode = true;
|
||||
}
|
||||
else {
|
||||
m_bConsoleMode = false;
|
||||
}
|
||||
}
|
||||
|
||||
void CDSLog::SetLevel(DWORD dwLevel)
|
||||
{
|
||||
m_dwLevel = dwLevel;
|
||||
}
|
||||
|
||||
void CDSLog::SetFile(LPSTR pszFileName, BOOL bAppend, BOOL bDaily)
|
||||
{
|
||||
CloseFile();
|
||||
|
||||
if (m_pszFileName)
|
||||
free(m_pszFileName);
|
||||
|
||||
if (pszFileName)
|
||||
m_pszFileName = strdup(pszFileName);
|
||||
|
||||
m_bAppend = bAppend;
|
||||
m_bDaily = bDaily;
|
||||
|
||||
if (m_bFileMode)
|
||||
OpenFile();
|
||||
}
|
||||
|
||||
void CDSLog::SetEventSource(LPSTR pszEventSource)
|
||||
{
|
||||
if (m_pszEventSource)
|
||||
free(m_pszEventSource);
|
||||
|
||||
if (pszEventSource)
|
||||
m_pszEventSource = strdup(pszEventSource);
|
||||
}
|
||||
|
||||
void CDSLog::_Print(LPSTR pszFormat, va_list val)
|
||||
{
|
||||
UINT64 nCurrentTime = _time64(0);
|
||||
|
||||
if (m_bFileMode && m_bDaily && m_hFile) {
|
||||
CHAR szDate[7];
|
||||
|
||||
MakeSurfix(szDate);
|
||||
|
||||
if (szDate[6] != m_szLastLogDay[6] || szDate[5] != m_szLastLogDay[5]) {
|
||||
CloseFile();
|
||||
OpenFile();
|
||||
}
|
||||
}
|
||||
|
||||
if (nCurrentTime != m_nLastLogTime) {
|
||||
m_nLastLogTime = nCurrentTime;
|
||||
_PrintLine(_ctime64((__time64_t*)&m_nLastLogTime));
|
||||
}
|
||||
|
||||
// - Write the log message
|
||||
TCHAR szLine[LINE_BUFFER_SIZE];
|
||||
|
||||
vsprintf(szLine, pszFormat, val);
|
||||
_PrintLine(szLine);
|
||||
}
|
||||
|
||||
inline void CDSLog::_PrintLine(LPSTR pszLine)
|
||||
{
|
||||
if (m_bDebugMode)
|
||||
OutputDebugString(pszLine);
|
||||
|
||||
DWORD dwWritten;
|
||||
|
||||
if (m_bConsoleMode)
|
||||
WriteConsole(GetStdHandle(STD_OUTPUT_HANDLE), pszLine, (DWORD)strlen(pszLine), &dwWritten, NULL);
|
||||
|
||||
if (m_bFileMode && m_hFile)
|
||||
WriteFile(m_hFile, pszLine, (DWORD)strlen(pszLine), &dwWritten, NULL);
|
||||
}
|
||||
|
||||
void CDSLog::OpenFile()
|
||||
{
|
||||
if (!m_pszFileName) {
|
||||
m_bDebugMode = true;
|
||||
m_bFileMode = false;
|
||||
|
||||
Print(0, "Error opening log file\n");
|
||||
return;
|
||||
}
|
||||
|
||||
char szFileName[MAX_PATH];
|
||||
|
||||
strcpy(szFileName, m_pszFileName);
|
||||
|
||||
if (m_bDaily) {
|
||||
CHAR szDate[7];
|
||||
|
||||
MakeSurfix(szDate);
|
||||
|
||||
strcat(szFileName, szDate);
|
||||
|
||||
strcpy(m_szLastLogDay, szDate);
|
||||
}
|
||||
|
||||
strcat(szFileName, m_szExt);
|
||||
|
||||
m_bFileMode = true;
|
||||
|
||||
m_hFile = CreateFile(szFileName, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
||||
if (m_hFile == INVALID_HANDLE_VALUE) {
|
||||
m_bDebugMode = true;
|
||||
m_bFileMode = false;
|
||||
|
||||
Print(0, "Error opening log file %s\n", szFileName);
|
||||
return;
|
||||
}
|
||||
|
||||
if (m_bAppend)
|
||||
SetFilePointer(m_hFile, 0, NULL, FILE_END);
|
||||
else
|
||||
SetEndOfFile(m_hFile);
|
||||
}
|
||||
|
||||
void CDSLog::CloseFile()
|
||||
{
|
||||
if (m_hFile) {
|
||||
CloseHandle(m_hFile);
|
||||
m_hFile = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
void CDSLog::MakeSurfix(CHAR *pszBuf)
|
||||
{
|
||||
WORD wMonth, wDay, wYear;
|
||||
|
||||
SYSTEMTIME st;
|
||||
GetLocalTime(&st);
|
||||
|
||||
wYear = st.wYear % 100;
|
||||
wMonth = st.wMonth;
|
||||
wDay = st.wDay;
|
||||
|
||||
pszBuf[0] = (CHAR)(wYear / 10 + '0');
|
||||
pszBuf[1] = (CHAR)(wYear % 10 + '0');
|
||||
|
||||
pszBuf[2] = (CHAR)(wMonth / 10 + '0');
|
||||
pszBuf[3] = (CHAR)(wMonth % 10 + '0');
|
||||
|
||||
pszBuf[4] = (CHAR)(wDay / 10 + '0');
|
||||
pszBuf[5] = (CHAR)(wDay % 10 + '0');
|
||||
|
||||
pszBuf[6] = 0;
|
||||
}
|
||||
Reference in New Issue
Block a user