#include "StdAfx.h"
|
#include "LogManager.h"
|
|
#ifdef _DEBUG
|
#define new DEBUG_NEW
|
#undef THIS_FILE
|
static char THIS_FILE[] = __FILE__;
|
#endif
|
|
|
CLogManager::CLogManager(int LogID, TCHAR* NameOfMemoryMap, unsigned int ProcessID )
|
{
|
CString strNameOfMemoryMap = _T("");
|
strNameOfMemoryMap.Format(_T("%s%02d"),NameOfMemoryMap,ProcessID);
|
m_nProcessID = ProcessID;
|
m_nLogID = LogID;
|
|
//ÆÄÀÏ ¸ÊÇÎ »ý¼º
|
m_hFMap = ::CreateFileMapping(INVALID_HANDLE_VALUE,NULL,PAGE_READWRITE,0,
|
SHAREDMEM_SIZE,strNameOfMemoryMap);
|
|
//¸¸ÀÏ ´Ù¸¥ ¾îÇø®ÄÉÀ̼ǿ¡¼ ÀÌ¹Ì »ý¼ºÇß´Ù¸é
|
if(::GetLastError() == ERROR_ALREADY_EXISTS)
|
{
|
//»ý¼ºµÈ ÆÄÀÏ ¸ÊÇÎÀ» Open
|
m_hFMap = ::OpenFileMapping(FILE_MAP_ALL_ACCESS,FALSE,strNameOfMemoryMap);
|
}
|
|
//¸ÊÇÎ ÆÄÀÏ¿¡¼ °øÀ¯¸Þ¸ð¸® Æ÷ÀÎÅ͸¦ °¡Á®¿Â´Ù.
|
m_pSharedMem = (char*)::MapViewOfFile(m_hFMap,FILE_MAP_ALL_ACCESS,
|
0,0,SHAREDMEM_SIZE);
|
|
//À妽º·Î »ç¿ëÇÒ °øÀ¯ ¸Þ¸ð¸®
|
m_nIndex = (unsigned int*)m_pSharedMem;
|
(*m_nIndex) = 0;
|
m_pSharedMem += 4;
|
|
//Ä¿¸Çµå·Î »ç¿ëÇÒ °øÀ¯¸Þ¸ð¸®
|
m_Command1 =(unsigned char*)m_pSharedMem;
|
*m_Command1 = 0;
|
|
//3byte ¿¹¾à °ø°£
|
//Ç×ÈÄ Ä¿¸Çµå Ãß°¡·Î »ç¿ëÇÒ ¼ö ÀÖ½À´Ï´Ù.
|
|
|
m_pSharedMem += 4;
|
|
//Å©¸®Æ¼Äà ¼½¼Ç ÃʱâÈ
|
m_csMemoryMap = CreateMutex(NULL, FALSE, _T("DIT_LOGVIEW_MUTEX"));
|
InitializeCriticalSection(&m_csLocalLogFile);
|
}
|
|
CLogManager::~CLogManager(void)
|
{
|
//Å©¸®Æ¼Äà ¼½¼Ç ¼Ò¸ê
|
CloseHandle(m_csMemoryMap);
|
DeleteCriticalSection(&m_csLocalLogFile);
|
|
//À妽º¿Í Ä¿¸Çµå °ø°£À» µÇµ¹¸°´Ù.
|
m_pSharedMem -= 8;
|
|
//UnMap ÇÑ´Ù.
|
if( m_pSharedMem != NULL) ::UnmapViewOfFile(m_pSharedMem);
|
if( m_hFMap != NULL) ::CloseHandle(m_hFMap);
|
|
|
}
|
|
void CLogManager::WriteLog( unsigned int Level, TCHAR* str,...)
|
{
|
WaitForSingleObject(m_csMemoryMap, INFINITE);
|
|
memset(m_TempLog,0x00, sizeof(TCHAR)*MAX_STRING_SIZE*2);
|
|
va_list arg;
|
va_start(arg, str);
|
|
PDITLOG ptrLog;
|
//unsigned int* ptrLevel;
|
unsigned int index = 0;
|
|
index = (*m_nIndex)%SHARED_LOG_COUNT;
|
|
ptrLog = (PDITLOG)(m_pSharedMem+(index*ONE_LOG_SIZE));
|
|
// ptrLog->nLogID = m_nLogID;
|
ptrLog->nLogID = Level;
|
|
//·Î±× ·¹º§ ÀúÀå
|
ptrLog->nLevel = Level;
|
//¿¹¾àµ¥ÀÌÅÍ
|
// ptrLog->nProcess = m_nProcessID;
|
ptrLog->nProcess = Level;
|
//·Î±× ½Ã°£ ÀúÀå
|
GetLocalTime(&(ptrLog->nCurTime));
|
//±âÁ¸ ·Î±× ÃʱâÈ
|
memset(ptrLog->logMessage,0x00,sizeof(TCHAR)*MAX_STRING_SIZE);
|
//½Å±Ô ·Î±× ³»¿ë ÀúÀå
|
_vstprintf_s(m_TempLog, MAX_STRING_SIZE*2, str, arg);
|
|
memcpy(ptrLog->logMessage,m_TempLog,sizeof(TCHAR)*(MAX_STRING_SIZE-1));
|
//À妽º Áõ°¡
|
(*m_nIndex)++;
|
|
ReleaseMutex(m_csMemoryMap);
|
}
|
|
void CLogManager::ReadLog(TCHAR* str, unsigned int curIndex, int* curLogID,
|
unsigned int* curLevel,unsigned int* curProcess, PSYSTEMTIME curTime)
|
{
|
unsigned int index = 0;
|
PDITLOG ptrLog;
|
|
index = (curIndex)%SHARED_LOG_COUNT;
|
//Àоî¿Ã ·Î±×ÀÇ Æ÷ÀÎÅ͸¦ °¡Á®¿Â´Ù.
|
ptrLog = (PDITLOG)(m_pSharedMem+(index*ONE_LOG_SIZE));
|
//·Î±× ·¹º§ °¡Á®¿Â´Ù.
|
*curLevel = ptrLog->nLevel;
|
//·Î±× ¾ÆÀ̵𸦠°¡Á®¿Â´Ù.
|
*curLogID = ptrLog->nLogID;
|
//ÇÁ·Î¼¼½º id ¿¹¾à¿µ¿ª µ¥ÀÌÅÍ
|
*curProcess = ptrLog->nProcess;
|
//·Î±×½Ã°£ °¡Á®¿Â´Ù.
|
memcpy( curTime, &(ptrLog->nCurTime), sizeof(SYSTEMTIME));
|
//·Î±×¹®ÀÚ¿ °¡Á®¿Â´Ù.
|
memcpy( str, ptrLog->logMessage, MAX_STRING_SIZE*sizeof(TCHAR));
|
}
|
|
//ÀÌ ÇÔ¼ö´Â °¡±ÞÀû »ç¿ëÇÏÁö ¾Ê´Â °ÍÀÌ ÁÁ½À´Ï´Ù.
|
void CLogManager::WriteLocalLog(TCHAR* str,...)
|
{
|
//»ç¿ë ¾îÇø®ÄÉÀ̼ǿ¡¼ ·ÎÄ÷Π¹Ù·Î ·Î±× ÆÄÀÏÀ» ÀÛ¼ºÇÕ´Ï´Ù.
|
va_list list;
|
TCHAR szText[MAX_STRING_SIZE] = {0, };
|
|
va_start(list, str);
|
_vstprintf_s(szText, str, list);
|
va_end(list);
|
|
// ÆÄÀÏ¿¡ ¾²±â
|
SYSTEMTIME st;
|
GetLocalTime(&st);
|
|
static TCHAR strMessage[MAX_STRING_SIZE];
|
memset(strMessage, 0x00, MAX_STRING_SIZE*sizeof(TCHAR));
|
_stprintf_s(strMessage, _T("[%02d:%02d:%02d] %s\r\n"), st.wHour, st.wMinute, st.wSecond, szText);
|
|
//·Î±× ÆÄÀϸí°ú À§Ä¡ »ý¼º
|
static TCHAR strFileName[MAX_PATH];
|
memset(strFileName, 0x00, MAX_PATH*sizeof(TCHAR));
|
_stprintf_s(strFileName, _T("%sLocal%d%02d%02d.log"), LOG_FILE_PATH, st.wYear, st.wMonth, st.wDay);
|
|
DWORD dwWritten;
|
HANDLE hFile;
|
|
//°æÀï»óÅ ¹æÁö¸¦ À§ÇÑ Å©¸®Æ¼Äà °´Ã¼ »ç¿ë
|
EnterCriticalSection(&m_csLocalLogFile);
|
hFile = CreateFile(strFileName, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
int nFilePointer = SetFilePointer(hFile, 0, NULL, FILE_END);
|
|
#ifdef UNICODE
|
if(!nFilePointer)
|
{
|
// À¯´ÏÄÚµå ÆÄÀÏÀÇ ½ÃÀÛÀº BOM(0xFEFF) À̾î¾ß ÇÑ´Ù.
|
TCHAR strCommand = 0xFEFF;
|
WriteFile(hFile, &strCommand, sizeof(TCHAR), &dwWritten, NULL);
|
}
|
#endif
|
|
//·ÎÄà ·Î±×ÆÄÀÏ ÀÛ¼º
|
WriteFile(hFile, strMessage, _tcslen(strMessage)*sizeof(TCHAR), &dwWritten, NULL);
|
CloseHandle(hFile);
|
LeaveCriticalSection(&m_csLocalLogFile);
|
}
|
void CLogManager::SaveBinary(TCHAR* FilePath)
|
{
|
// ÆÄÀÏ¿¡ ¾²±â
|
SYSTEMTIME st;
|
GetLocalTime(&st);
|
|
//ÆÄÀÏ °æ·Î »ý¼º
|
static TCHAR strFileName[MAX_PATH];
|
memset(strFileName, 0x00, MAX_PATH*sizeof(TCHAR));
|
_stprintf_s(strFileName, _T("%s%d%02d%02d-%02d%02d%02d.map"), FilePath, st.wYear, st.wMonth, st.wDay, st.wHour, st.wMinute, st.wSecond);
|
|
DWORD dwWritten;
|
HANDLE hFile;
|
|
//¹ÙÀ̳ʸ® ¸ÊÀ» °¡Á®¿É´Ï´Ù.
|
char* ptrMap = GetMapDataAddress();
|
|
EnterCriticalSection(&m_csLocalLogFile);
|
hFile = CreateFile(strFileName, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
|
int nFilePointer = SetFilePointer(hFile, 0, NULL, FILE_END);
|
|
//64K ¹ÙÀ̳ʸ® µ¥ÀÌÅ͸¦ ÆÄÀÏ¿¡ ÀúÀåÇÕ´Ï´Ù.
|
WriteFile(hFile, ptrMap, DIT_MAP_SIZE, &dwWritten, NULL);
|
|
CloseHandle(hFile);
|
LeaveCriticalSection(&m_csLocalLogFile);
|
}
|
|
void CLogManager::GetFirstCommand(unsigned char* pCommand)
|
{
|
//Ä¿¸Çµå°ªÀ» ¾ò¾î¿É´Ï´Ù.
|
*pCommand = *m_Command1;
|
}
|
|
void CLogManager::SetFirstCommand(unsigned char Command)
|
{
|
//Ä¿¸Çµå °ªÀ» ÁöÁ¤ÇÕ´Ï´Ù.
|
*m_Command1 = Command;
|
}
|
