97 lines
2.8 KiB
C++
97 lines
2.8 KiB
C++
|
/*****************************************************************************\
|
||
|
|
FILE: logging.cpp
|
||
|
|
|
||
|
|
DESCRIPTION:
|
||
|
|
Logging helper functions
|
||
|
|
|
||
|
|
BryanSt 4/23/2001 (Bryan Starbuck)
|
||
|
|
Copyright (C) Microsoft Corp 2001-2001. All rights reserved.
|
||
|
|
\*****************************************************************************/
|
||
|
|
|
||
|
|
#include <shlwapi.h> // SHGetValue, wnsprintf, SHFormatDateTime
|
||
|
|
|
||
|
|
|
||
|
|
#ifndef ARRAYSIZE
|
||
|
|
#define ARRAYSIZE(a) (sizeof((a))/sizeof((a)[0]))
|
||
|
|
#endif // ARRAYSIZE
|
||
|
|
|
||
|
|
|
||
|
|
HANDLE g_hLogFile = INVALID_HANDLE_VALUE;
|
||
|
|
void WriteToLogFileA(LPCSTR pszMessage, ...)
|
||
|
|
{
|
||
|
|
va_list vaParamList;
|
||
|
|
va_start(vaParamList, pszMessage);
|
||
|
|
|
||
|
|
if (INVALID_HANDLE_VALUE == g_hLogFile)
|
||
|
|
{
|
||
|
|
TCHAR szPath[MAX_PATH];
|
||
|
|
|
||
|
|
if (GetWindowsDirectory(szPath, ARRAYSIZE(szPath)))
|
||
|
|
{
|
||
|
|
PathAppend(szPath, TEXT("AutoPlay.log"));
|
||
|
|
g_hLogFile = CreateFile(szPath, (GENERIC_READ | GENERIC_WRITE), FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
||
|
|
if (INVALID_HANDLE_VALUE != g_hLogFile)
|
||
|
|
{
|
||
|
|
CHAR szTimeDate[MAX_PATH];
|
||
|
|
CHAR szHeader[MAX_PATH];
|
||
|
|
FILETIME ftCurrentUTC;
|
||
|
|
FILETIME ftCurrent;
|
||
|
|
SYSTEMTIME stCurrent;
|
||
|
|
DWORD cbWritten;
|
||
|
|
|
||
|
|
SetFilePointer(g_hLogFile, 0, NULL, FILE_END);
|
||
|
|
|
||
|
|
GetLocalTime(&stCurrent);
|
||
|
|
SystemTimeToFileTime(&stCurrent, &ftCurrent);
|
||
|
|
LocalFileTimeToFileTime(&ftCurrent, &ftCurrentUTC);
|
||
|
|
SHFormatDateTimeA(&ftCurrentUTC, NULL, szTimeDate, ARRAYSIZE(szTimeDate));
|
||
|
|
|
||
|
|
TCHAR szProcess[MAX_PATH];
|
||
|
|
if (!GetModuleFileName(NULL, szProcess, ARRAYSIZE(szProcess)))
|
||
|
|
{
|
||
|
|
szProcess[0] = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
wnsprintfA(szHeader, ARRAYSIZE(szHeader), "\r\n\r\n%hs - (%ls)\r\n", szTimeDate, /*szUserName,*/ szProcess);
|
||
|
|
WriteFile(g_hLogFile, szHeader, lstrlenA(szHeader), &cbWritten, NULL);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (INVALID_HANDLE_VALUE != g_hLogFile)
|
||
|
|
{
|
||
|
|
CHAR szMessage[4000];
|
||
|
|
DWORD cbWritten;
|
||
|
|
wvsprintfA(szMessage, pszMessage, vaParamList);
|
||
|
|
WriteFile(g_hLogFile, szMessage, lstrlenA(szMessage), &cbWritten, NULL);
|
||
|
|
FlushFileBuffers(g_hLogFile);
|
||
|
|
}
|
||
|
|
|
||
|
|
va_end(vaParamList);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void WriteToLogFileW(LPCWSTR pszMessage)
|
||
|
|
{
|
||
|
|
DWORD cchSize = (lstrlen(pszMessage) + 1);
|
||
|
|
LPSTR pszMessageAnsi = (LPSTR) LocalAlloc(LPTR, cchSize);
|
||
|
|
|
||
|
|
if (pszMessageAnsi)
|
||
|
|
{
|
||
|
|
SHUnicodeToAnsi(pszMessage, pszMessageAnsi, cchSize);
|
||
|
|
WriteToLogFileA(pszMessageAnsi);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void CloseLogFile(void)
|
||
|
|
{
|
||
|
|
if (INVALID_HANDLE_VALUE != g_hLogFile)
|
||
|
|
{
|
||
|
|
HANDLE hLogFile = g_hLogFile;
|
||
|
|
g_hLogFile = INVALID_HANDLE_VALUE;
|
||
|
|
CloseHandle(hLogFile);
|
||
|
|
}
|
||
|
|
}
|