894 lines
25 KiB
C
894 lines
25 KiB
C
|
/*++
|
||
|
|
||
|
Copyright (c) 2001 Microsoft Corporation
|
||
|
|
||
|
Module Name:
|
||
|
|
||
|
omlog.c
|
||
|
|
||
|
Abstract:
|
||
|
|
||
|
logging routines for the object log stream
|
||
|
|
||
|
this code was jumpstarted from the clusrtl logging code. Most notably, it
|
||
|
is UTF-8 encoded which allows full Unicode without the cost of writing
|
||
|
16b. for each character.
|
||
|
|
||
|
remaining issues to solve: proper truncation of log file based on current
|
||
|
starting session. We'd like to remember as many sessions as possible but
|
||
|
since I've chosen UTF-8, I can't jump to the middle of the file and start
|
||
|
looking around. One possibility strategy is using an alternate NTFS stream
|
||
|
to record the starting offsets of sessions. If that stream didn't exist or
|
||
|
we're on a FAT FS, then we'll continue with the current strategy.
|
||
|
|
||
|
Author:
|
||
|
|
||
|
Charlie Wickham (charlwi) 07-May-2001
|
||
|
|
||
|
Environment:
|
||
|
|
||
|
User Mode
|
||
|
|
||
|
Revision History:
|
||
|
|
||
|
--*/
|
||
|
|
||
|
#include "omp.h"
|
||
|
|
||
|
//
|
||
|
// when this sequence is at the beginning of the file, it indicates that the
|
||
|
// file is UTF8 encoded
|
||
|
//
|
||
|
#define UTF8_BOM "\x0EF\x0BB\x0BF"
|
||
|
|
||
|
//
|
||
|
// Private Data
|
||
|
//
|
||
|
|
||
|
DWORD OmpLogFileLimit;
|
||
|
DWORD OmpCurrentSessionStart;
|
||
|
BOOL OmpLogToFile = FALSE;
|
||
|
HANDLE OmpLogFileHandle = NULL;
|
||
|
DWORD OmpProcessId;
|
||
|
DWORD OmpCurrentSessionOffset;
|
||
|
|
||
|
PCLRTL_WORK_QUEUE OmpLoggerWorkQueue;
|
||
|
|
||
|
//
|
||
|
// structure used to pass formatted buffers to work queue routine
|
||
|
//
|
||
|
typedef struct _OM_LOG_BUFFER_DESC {
|
||
|
DWORD TimeBytes;
|
||
|
DWORD MsgBytes;
|
||
|
PCHAR TimeBuffer;
|
||
|
PCHAR MsgBuffer;
|
||
|
} OM_LOG_BUFFER_DESC, *POM_LOG_BUFFER_DESC;
|
||
|
|
||
|
#define MAX_NUMBER_LENGTH 20
|
||
|
|
||
|
// Specify maximum file size ( DWORD / 1MB )
|
||
|
|
||
|
#define MAX_FILE_SIZE ( 0xFFFFF000 / ( 1024 * 1024 ) )
|
||
|
|
||
|
DWORD OmpLogFileLimit = ( 1 * 1024 * 1024 ); // 1 MB default
|
||
|
DWORD OmpLogFileLoWater = 0;
|
||
|
|
||
|
//
|
||
|
// internal functions
|
||
|
//
|
||
|
DWORD
|
||
|
OmpTruncateFile(
|
||
|
IN HANDLE FileHandle,
|
||
|
IN DWORD FileSize,
|
||
|
IN LPDWORD LastPosition
|
||
|
)
|
||
|
|
||
|
/*++
|
||
|
|
||
|
Routine Description:
|
||
|
|
||
|
Truncate a file by copying the portion starting at LastPosition and ending
|
||
|
at EOF to the front of the file and setting the file's EOF pointer to the
|
||
|
end of the new chunk. We always keep the current session even that means
|
||
|
growing larger than the file limit.
|
||
|
|
||
|
For now, we have no good way of finding all the sessions within the file,
|
||
|
so if the file must be truncated, we whack it back to the beginning of the
|
||
|
current session. If time permits, I'll add something more intelligent
|
||
|
later on.
|
||
|
|
||
|
Arguments:
|
||
|
|
||
|
FileHandle - File handle.
|
||
|
|
||
|
FileSize - Current End of File.
|
||
|
|
||
|
LastPosition - On input, specifies the starting position in the file from
|
||
|
which the copy begins. On output, it is set to the new EOF
|
||
|
|
||
|
Return Value:
|
||
|
|
||
|
New end of file.
|
||
|
|
||
|
--*/
|
||
|
|
||
|
{
|
||
|
//
|
||
|
// The following buffer size should never be more than 1/4 the size of the
|
||
|
// file.
|
||
|
//
|
||
|
#define BUFFER_SIZE ( 64 * 1024 )
|
||
|
DWORD bytesLeft;
|
||
|
DWORD endPosition = sizeof( UTF8_BOM ) - 1;
|
||
|
DWORD bufferSize;
|
||
|
DWORD bytesRead;
|
||
|
DWORD bytesWritten;
|
||
|
DWORD fileSizeHigh = 0;
|
||
|
DWORD readPosition;
|
||
|
DWORD writePosition;
|
||
|
PVOID dataBuffer;
|
||
|
|
||
|
//
|
||
|
// current session is already at beginning of file so bale now...
|
||
|
//
|
||
|
if ( OmpCurrentSessionOffset == sizeof( UTF8_BOM ) - 1) {
|
||
|
return FileSize;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// don't truncate the current session, i.e., always copy from the start of
|
||
|
// the current session
|
||
|
//
|
||
|
if ( *LastPosition > OmpCurrentSessionOffset ) {
|
||
|
*LastPosition = OmpCurrentSessionOffset;
|
||
|
}
|
||
|
|
||
|
if ( *LastPosition > FileSize ) {
|
||
|
//
|
||
|
// something's confused; the spot we're supposed to copy from is at or
|
||
|
// past the current EOF. reset the entire file
|
||
|
//
|
||
|
goto error_exit;
|
||
|
}
|
||
|
|
||
|
dataBuffer = LocalAlloc( LMEM_FIXED, BUFFER_SIZE );
|
||
|
if ( !dataBuffer ) {
|
||
|
goto error_exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// calc number of bytes to move
|
||
|
//
|
||
|
bytesLeft = FileSize - *LastPosition;
|
||
|
endPosition = bytesLeft + sizeof( UTF8_BOM ) - 1;
|
||
|
|
||
|
//
|
||
|
// Point back to last position for reading.
|
||
|
//
|
||
|
readPosition = *LastPosition;
|
||
|
writePosition = sizeof( UTF8_BOM ) - 1;
|
||
|
|
||
|
while ( bytesLeft ) {
|
||
|
if ( bytesLeft >= BUFFER_SIZE ) {
|
||
|
bufferSize = BUFFER_SIZE;
|
||
|
} else {
|
||
|
bufferSize = bytesLeft;
|
||
|
}
|
||
|
bytesLeft -= bufferSize;
|
||
|
|
||
|
SetFilePointer( FileHandle,
|
||
|
readPosition,
|
||
|
&fileSizeHigh,
|
||
|
FILE_BEGIN );
|
||
|
|
||
|
if ( ReadFile( FileHandle,
|
||
|
dataBuffer,
|
||
|
bufferSize,
|
||
|
&bytesRead,
|
||
|
NULL ) )
|
||
|
{
|
||
|
SetFilePointer( FileHandle,
|
||
|
writePosition,
|
||
|
&fileSizeHigh,
|
||
|
FILE_BEGIN );
|
||
|
WriteFile( FileHandle,
|
||
|
dataBuffer,
|
||
|
bytesRead,
|
||
|
&bytesWritten,
|
||
|
NULL );
|
||
|
} else {
|
||
|
endPosition = sizeof( UTF8_BOM ) - 1;
|
||
|
break;
|
||
|
}
|
||
|
readPosition += bytesRead;
|
||
|
writePosition += bytesWritten;
|
||
|
}
|
||
|
|
||
|
LocalFree( dataBuffer );
|
||
|
|
||
|
//
|
||
|
// for now, we only only one truncate by setting the current position to
|
||
|
// the beginning of file.
|
||
|
//
|
||
|
OmpCurrentSessionOffset = sizeof( UTF8_BOM ) - 1;
|
||
|
|
||
|
error_exit:
|
||
|
|
||
|
//
|
||
|
// Force end of file to get set.
|
||
|
//
|
||
|
SetFilePointer( FileHandle,
|
||
|
endPosition,
|
||
|
&fileSizeHigh,
|
||
|
FILE_BEGIN );
|
||
|
|
||
|
SetEndOfFile( FileHandle );
|
||
|
|
||
|
*LastPosition = endPosition;
|
||
|
|
||
|
return(endPosition);
|
||
|
|
||
|
} // OmpTruncateFile
|
||
|
|
||
|
VOID
|
||
|
OmpLoggerWorkThread(
|
||
|
IN PCLRTL_WORK_ITEM WorkItem,
|
||
|
IN DWORD Status,
|
||
|
IN DWORD BytesTransferred,
|
||
|
IN ULONG_PTR IoContext
|
||
|
)
|
||
|
|
||
|
/*++
|
||
|
|
||
|
Routine Description:
|
||
|
|
||
|
work queue worker routine. actually does the write to the file.
|
||
|
|
||
|
Arguments:
|
||
|
|
||
|
standard ClRtl thread args; we only care about WorkItem
|
||
|
|
||
|
Return Value:
|
||
|
|
||
|
None
|
||
|
|
||
|
--*/
|
||
|
|
||
|
{
|
||
|
DWORD fileSize;
|
||
|
DWORD fileSizeHigh;
|
||
|
DWORD tsBytesWritten;
|
||
|
DWORD msgBytesWritten;
|
||
|
|
||
|
POM_LOG_BUFFER_DESC bufDesc = (POM_LOG_BUFFER_DESC)(WorkItem->Context);
|
||
|
|
||
|
fileSize = GetFileSize( OmpLogFileHandle, &fileSizeHigh );
|
||
|
ASSERT( fileSizeHigh == 0 ); // We're only using DWORDs!
|
||
|
|
||
|
if ( fileSize > OmpLogFileLimit ) {
|
||
|
fileSize = OmpTruncateFile( OmpLogFileHandle, fileSize, &OmpLogFileLoWater );
|
||
|
}
|
||
|
|
||
|
SetFilePointer( OmpLogFileHandle,
|
||
|
fileSize,
|
||
|
&fileSizeHigh,
|
||
|
FILE_BEGIN );
|
||
|
|
||
|
WriteFile(OmpLogFileHandle,
|
||
|
bufDesc->TimeBuffer,
|
||
|
bufDesc->TimeBytes,
|
||
|
&tsBytesWritten,
|
||
|
NULL);
|
||
|
|
||
|
WriteFile(OmpLogFileHandle,
|
||
|
bufDesc->MsgBuffer,
|
||
|
bufDesc->MsgBytes,
|
||
|
&msgBytesWritten,
|
||
|
NULL);
|
||
|
|
||
|
//
|
||
|
// if we haven't set the lo water mark, wait until the file size has
|
||
|
// crossed the halfway mark and set it to the beginning of the line we
|
||
|
// just wrote.
|
||
|
//
|
||
|
if ( OmpLogFileLoWater == 0 && (fileSize > (OmpLogFileLimit / 2)) ) {
|
||
|
OmpLogFileLoWater = fileSize;
|
||
|
|
||
|
ASSERT( OmpLogFileLoWater >= OmpCurrentSessionOffset );
|
||
|
}
|
||
|
|
||
|
} // OmpLoggerWorkThread
|
||
|
|
||
|
VOID
|
||
|
OmpLogPrint(
|
||
|
LPWSTR FormatString,
|
||
|
...
|
||
|
)
|
||
|
|
||
|
/*++
|
||
|
|
||
|
Routine Description:
|
||
|
|
||
|
Prints a message to the object config log file.
|
||
|
|
||
|
Arguments:
|
||
|
|
||
|
FormatString - The initial message string to print.
|
||
|
|
||
|
Any FormatMessage-compatible arguments to be inserted in the message
|
||
|
before it is logged.
|
||
|
|
||
|
Return Value:
|
||
|
|
||
|
None.
|
||
|
|
||
|
--*/
|
||
|
|
||
|
{
|
||
|
PWCHAR unicodeOutput = NULL;
|
||
|
PCHAR timestampBuffer;
|
||
|
DWORD timestampBytes;
|
||
|
PCHAR utf8Buffer;
|
||
|
DWORD utf8Bytes;
|
||
|
PWCHAR unicodeBuffer;
|
||
|
DWORD unicodeBytes;
|
||
|
DWORD status = ERROR_SUCCESS;
|
||
|
|
||
|
SYSTEMTIME Time;
|
||
|
ULONG_PTR ArgArray[9];
|
||
|
va_list ArgList;
|
||
|
|
||
|
//
|
||
|
// init the variable arg list
|
||
|
//
|
||
|
va_start(ArgList, FormatString);
|
||
|
|
||
|
if ( !OmpLogToFile ) {
|
||
|
va_end(ArgList);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
GetSystemTime(&Time);
|
||
|
|
||
|
ArgArray[0] = OmpProcessId;
|
||
|
ArgArray[1] = GetCurrentThreadId();
|
||
|
ArgArray[2] = Time.wYear;
|
||
|
ArgArray[3] = Time.wMonth;
|
||
|
ArgArray[4] = Time.wDay;
|
||
|
ArgArray[5] = Time.wHour;
|
||
|
ArgArray[6] = Time.wMinute;
|
||
|
ArgArray[7] = Time.wSecond;
|
||
|
ArgArray[8] = Time.wMilliseconds;
|
||
|
|
||
|
//
|
||
|
// we can get away with formatting it as ANSI since our data is all numbers
|
||
|
//
|
||
|
timestampBytes = FormatMessageA(FORMAT_MESSAGE_FROM_STRING |
|
||
|
FORMAT_MESSAGE_ARGUMENT_ARRAY |
|
||
|
FORMAT_MESSAGE_ALLOCATE_BUFFER,
|
||
|
"%1!08lx!.%2!08lx!::%3!02d!/%4!02d!/%5!02d!-%6!02d!:%7!02d!:%8!02d!.%9!03d! ",
|
||
|
0,
|
||
|
0,
|
||
|
(LPSTR)×tampBuffer,
|
||
|
50,
|
||
|
(va_list*)&ArgArray);
|
||
|
|
||
|
if ( timestampBytes == 0 ) {
|
||
|
va_end(ArgList);
|
||
|
// WmiTrace("Prefix format failed, %d: %!ARSTR!", GetLastError(), FormatString);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// format the message in unicode
|
||
|
//
|
||
|
try {
|
||
|
unicodeBytes = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER
|
||
|
| FORMAT_MESSAGE_FROM_STRING,
|
||
|
FormatString,
|
||
|
0,
|
||
|
0,
|
||
|
(LPWSTR)&unicodeOutput,
|
||
|
512,
|
||
|
&ArgList);
|
||
|
}
|
||
|
except ( EXCEPTION_EXECUTE_HANDLER ) {
|
||
|
unicodeBytes = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER
|
||
|
| FORMAT_MESSAGE_FROM_STRING
|
||
|
| FORMAT_MESSAGE_ARGUMENT_ARRAY,
|
||
|
L"LOGERROR(exception): Could not print message: %1!ws!",
|
||
|
0,
|
||
|
0,
|
||
|
(LPWSTR)&unicodeOutput,
|
||
|
512,
|
||
|
(va_list *) &FormatString );
|
||
|
}
|
||
|
va_end(ArgList);
|
||
|
|
||
|
if (unicodeBytes != 0) {
|
||
|
PCLRTL_WORK_ITEM workQItem;
|
||
|
|
||
|
//
|
||
|
// convert the output to UTF-8; first get the size to see if it will
|
||
|
// fit in our stack buffer.
|
||
|
//
|
||
|
utf8Bytes = WideCharToMultiByte(CP_UTF8,
|
||
|
0, // dwFlags
|
||
|
unicodeOutput,
|
||
|
unicodeBytes,
|
||
|
NULL,
|
||
|
0,
|
||
|
NULL, // lpDefaultChar
|
||
|
NULL); // lpUsedDefaultChar
|
||
|
|
||
|
utf8Buffer = LocalAlloc( LMEM_FIXED, utf8Bytes );
|
||
|
if ( utf8Buffer == NULL ) {
|
||
|
status = ERROR_NOT_ENOUGH_MEMORY;
|
||
|
goto error_exit;
|
||
|
}
|
||
|
|
||
|
utf8Bytes = WideCharToMultiByte(CP_UTF8,
|
||
|
0, // dwFlags
|
||
|
unicodeOutput,
|
||
|
unicodeBytes,
|
||
|
utf8Buffer,
|
||
|
utf8Bytes,
|
||
|
NULL, // lpDefaultChar
|
||
|
NULL); // lpUsedDefaultChar
|
||
|
|
||
|
workQItem = (PCLRTL_WORK_ITEM)LocalAlloc(LMEM_FIXED,
|
||
|
sizeof( CLRTL_WORK_ITEM ) + sizeof( OM_LOG_BUFFER_DESC ));
|
||
|
|
||
|
if ( workQItem != NULL ) {
|
||
|
POM_LOG_BUFFER_DESC bufDesc = (POM_LOG_BUFFER_DESC)(workQItem + 1);
|
||
|
|
||
|
bufDesc->TimeBytes = timestampBytes;
|
||
|
bufDesc->TimeBuffer = timestampBuffer;
|
||
|
bufDesc->MsgBytes = utf8Bytes;
|
||
|
bufDesc->MsgBuffer = utf8Buffer;
|
||
|
|
||
|
ClRtlInitializeWorkItem( workQItem, OmpLoggerWorkThread, bufDesc );
|
||
|
status = ClRtlPostItemWorkQueue( OmpLoggerWorkQueue, workQItem, 0, 0 );
|
||
|
|
||
|
} else {
|
||
|
status = ERROR_NOT_ENOUGH_MEMORY;
|
||
|
goto error_exit;
|
||
|
}
|
||
|
// WmiTrace("%!level! %!ARSTR!", *(UCHAR*)&LogLevel, AnsiString.Buffer + timestampBytes);
|
||
|
} else {
|
||
|
// WmiTrace("Format returned 0 bytes: %!ARSTR!", FormatString);
|
||
|
status = GetLastError();
|
||
|
}
|
||
|
|
||
|
error_exit:
|
||
|
if ( unicodeOutput != NULL ) {
|
||
|
LocalFree( unicodeOutput );
|
||
|
}
|
||
|
|
||
|
return;
|
||
|
|
||
|
} // OmpLogPrint
|
||
|
|
||
|
|
||
|
//
|
||
|
// exported (within OM) functions
|
||
|
//
|
||
|
|
||
|
VOID
|
||
|
OmpOpenObjectLog(
|
||
|
VOID
|
||
|
)
|
||
|
|
||
|
/*++
|
||
|
|
||
|
Routine Description:
|
||
|
|
||
|
Use the clusterlog environment variable to open another file that contains
|
||
|
object mgr name to ID mapping info. If the routine fails, the failure is
|
||
|
logged in the cluster log but no logging will be done to the object log
|
||
|
file.
|
||
|
|
||
|
NOTE: access to the file is synchronized since this routine is assumed to
|
||
|
be called only once by OmInit. Arguments:
|
||
|
|
||
|
Arguments:
|
||
|
|
||
|
None
|
||
|
|
||
|
Return Value:
|
||
|
|
||
|
None
|
||
|
|
||
|
--*/
|
||
|
|
||
|
{
|
||
|
WCHAR logFileBuffer[MAX_PATH];
|
||
|
LPWSTR logFileName = NULL;
|
||
|
LPWSTR objectLogExtension = L".obj";
|
||
|
DWORD status = ERROR_SUCCESS;
|
||
|
DWORD defaultLogSize = 1; // in MB
|
||
|
DWORD envLength;
|
||
|
WCHAR logFileSize[MAX_NUMBER_LENGTH];
|
||
|
DWORD logSize;
|
||
|
LPWSTR lpszBakFileName = NULL;
|
||
|
DWORD fileSizeHigh = 0;
|
||
|
DWORD fileSizeLow;
|
||
|
DWORD bytesWritten;
|
||
|
PWCHAR dot;
|
||
|
|
||
|
UNICODE_STRING logFileString;
|
||
|
|
||
|
//
|
||
|
// see if logging has been specified; get a buffer big enough that will
|
||
|
// hold the object log name
|
||
|
//
|
||
|
envLength = GetEnvironmentVariable(L"ClusterLog",
|
||
|
logFileBuffer,
|
||
|
sizeof(logFileBuffer)/sizeof(WCHAR));
|
||
|
if ( envLength > (( sizeof(logFileBuffer) + sizeof(objectLogExtension)) / sizeof(WCHAR)) ) {
|
||
|
//
|
||
|
// allocate a larger buffer since our static one wasn't big enough
|
||
|
//
|
||
|
logFileName = LocalAlloc( LMEM_FIXED,
|
||
|
envLength * sizeof( WCHAR ) );
|
||
|
if ( logFileName == NULL ) {
|
||
|
ClRtlLogPrint(LOG_UNUSUAL,
|
||
|
"[OM] Unable to get memory for Object log filename buffer\n");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
envLength = GetEnvironmentVariable(L"ClusterLog",
|
||
|
logFileName,
|
||
|
envLength);
|
||
|
if ( envLength == 0 ) {
|
||
|
ClRtlLogPrint(LOG_UNUSUAL,
|
||
|
"[OM] Unable to read ClusterLog environment variable\n");
|
||
|
|
||
|
goto error_exit;
|
||
|
}
|
||
|
} else if ( envLength != 0 ) {
|
||
|
logFileName = logFileBuffer;
|
||
|
}
|
||
|
|
||
|
if ( logFileName == NULL ) {
|
||
|
//
|
||
|
// logging is turned off or we can't determine where to put the file.
|
||
|
//
|
||
|
goto error_exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Try to get a limit on the log file size. This number is the number of
|
||
|
// MB.
|
||
|
//
|
||
|
envLength = GetEnvironmentVariable(L"ClusterLogSize",
|
||
|
logFileSize,
|
||
|
sizeof(logFileSize)/sizeof(WCHAR));
|
||
|
|
||
|
if ( envLength != 0 && envLength < MAX_NUMBER_LENGTH ) {
|
||
|
RtlInitUnicodeString( &logFileString, logFileSize );
|
||
|
status = RtlUnicodeStringToInteger( &logFileString,
|
||
|
10,
|
||
|
&logSize );
|
||
|
if ( NT_SUCCESS( status ) ) {
|
||
|
OmpLogFileLimit = logSize;
|
||
|
}
|
||
|
} else {
|
||
|
OmpLogFileLimit = defaultLogSize;
|
||
|
}
|
||
|
|
||
|
status = ERROR_SUCCESS;
|
||
|
|
||
|
if ( OmpLogFileLimit == 0 ) {
|
||
|
goto error_exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// make the file size no bigger than one-eighth the size of the normal log
|
||
|
// file but no less than 256KB
|
||
|
//
|
||
|
if ( OmpLogFileLimit > MAX_FILE_SIZE ) {
|
||
|
OmpLogFileLimit = MAX_FILE_SIZE;
|
||
|
}
|
||
|
OmpLogFileLimit = ( OmpLogFileLimit * ( 1024 * 1024 )) / 8;
|
||
|
if ( OmpLogFileLimit < ( 256 * 1024 )) {
|
||
|
OmpLogFileLimit = 256 * 1024;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// replace the filename with the object log name; deal with the use of
|
||
|
// forward slashes
|
||
|
//
|
||
|
dot = wcsrchr( logFileName, L'.' );
|
||
|
|
||
|
if ( dot != NULL ) {
|
||
|
wcscpy( dot, objectLogExtension );
|
||
|
} else {
|
||
|
wcscat( logFileName, objectLogExtension );
|
||
|
}
|
||
|
|
||
|
OmpLogFileHandle = CreateFile(logFileName,
|
||
|
GENERIC_READ | GENERIC_WRITE,
|
||
|
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
||
|
NULL,
|
||
|
OPEN_ALWAYS,
|
||
|
0,
|
||
|
NULL );
|
||
|
|
||
|
if ( OmpLogFileHandle == INVALID_HANDLE_VALUE ) {
|
||
|
status = GetLastError();
|
||
|
|
||
|
ClRtlLogPrint(LOG_UNUSUAL,
|
||
|
"[OM] Open of object log file failed. Error %1!u!\n",
|
||
|
status);
|
||
|
goto error_exit;
|
||
|
} else {
|
||
|
|
||
|
//
|
||
|
// write UTF-8 header to beginning of file and get the offset of the
|
||
|
// EOF; we never want to reset the start of the file after this point.
|
||
|
//
|
||
|
WriteFile( OmpLogFileHandle, UTF8_BOM, sizeof( UTF8_BOM ) - 1, &bytesWritten, NULL );
|
||
|
|
||
|
OmpCurrentSessionOffset = SetFilePointer( OmpLogFileHandle, 0, NULL, FILE_END );
|
||
|
if ( OmpCurrentSessionOffset == INVALID_SET_FILE_POINTER ) {
|
||
|
ClRtlLogPrint(LOG_UNUSUAL,
|
||
|
"[OM] Unable to get object log end of file position. error %1!u!.\n",
|
||
|
GetLastError());
|
||
|
|
||
|
CloseHandle( OmpLogFileHandle );
|
||
|
goto error_exit;
|
||
|
}
|
||
|
|
||
|
OmpLogToFile = TRUE;
|
||
|
OmpProcessId = GetCurrentProcessId();
|
||
|
|
||
|
//
|
||
|
// determine the initial low water mark. We have 3 cases
|
||
|
// we need to handle:
|
||
|
// 1) log size is less than 1/2 limit
|
||
|
// 2) log size is within limit but more than 1/2 limit
|
||
|
// 3) log size is greater than limit
|
||
|
//
|
||
|
// case 1 requires nothing special; the low water mark will be updated
|
||
|
// on the next log write.
|
||
|
//
|
||
|
// for case 2, we need to find the beginning of a line near 1/2 the
|
||
|
// current limit. for case 3, the place to start looking is current
|
||
|
// log size - 1/2 limit. In this case, the log will be truncated
|
||
|
// before the first write occurs, so we need to take the last 1/2
|
||
|
// limit bytes and copy them down to the front.
|
||
|
//
|
||
|
|
||
|
//
|
||
|
// For now, set the low water mark to be the current offset. When it
|
||
|
// is time to wrap, we'll lose everything but the current session.
|
||
|
//
|
||
|
// the problem is that we're dealing with UTF8 and we can't just jump
|
||
|
// in the middle of the file and start looking around (we might hit
|
||
|
// the 2nd byte of a DBCS sequence). For now, we'll leave
|
||
|
// OmpLogFileLoWater set to zero. It will get updated when the 1/2 way
|
||
|
// threshold is crossed.
|
||
|
//
|
||
|
OmpLogFileLoWater = OmpCurrentSessionOffset;
|
||
|
#if 0
|
||
|
|
||
|
fileSizeLow = GetFileSize( OmpLogFileHandle, &fileSizeHigh );
|
||
|
if ( fileSizeLow < ( OmpLogFileLimit / 2 )) {
|
||
|
//
|
||
|
// case 1: leave low water at zero; it will be updated with next
|
||
|
// log write
|
||
|
//
|
||
|
;
|
||
|
} else {
|
||
|
#define LOGBUF_SIZE 1024
|
||
|
CHAR buffer[LOGBUF_SIZE];
|
||
|
LONG currentPosition;
|
||
|
DWORD bytesRead;
|
||
|
|
||
|
if ( fileSizeLow < OmpLogFileLimit ) {
|
||
|
//
|
||
|
// case 2; start looking at the 1/2 the current limit to find
|
||
|
// the starting position
|
||
|
//
|
||
|
currentPosition = OmpLogFileLimit / 2;
|
||
|
} else {
|
||
|
//
|
||
|
// case 3: start at current size minus 1/2 limit to find our
|
||
|
// starting position.
|
||
|
//
|
||
|
currentPosition = fileSizeLow - ( OmpLogFileLimit / 2 );
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// backup from the initial file position, read in a block and look
|
||
|
// for the start of a session. When we find one, backup to the
|
||
|
// beginning of that line. Use that as the initial starting
|
||
|
// position when we finally truncate the file.
|
||
|
//
|
||
|
OmpLogFileLoWater = 0;
|
||
|
currentPosition -= LOGBUF_SIZE;
|
||
|
|
||
|
SetFilePointer(OmpLogFileHandle,
|
||
|
currentPosition,
|
||
|
&fileSizeHigh,
|
||
|
FILE_BEGIN);
|
||
|
|
||
|
do {
|
||
|
|
||
|
if ( ReadFile(OmpLogFileHandle,
|
||
|
buffer,
|
||
|
LOGBUF_SIZE - 1,
|
||
|
&bytesRead,
|
||
|
NULL ) )
|
||
|
{
|
||
|
PCHAR p = buffer;
|
||
|
PCHAR newp;
|
||
|
|
||
|
buffer[ bytesRead ] = NULL;
|
||
|
while ( *p != 'S' && bytesRead-- != 0 ) {
|
||
|
newp = CharNextExA( CP_UTF8, p, 0 );
|
||
|
if ( newp == p ) {
|
||
|
break;
|
||
|
}
|
||
|
p = newp;
|
||
|
}
|
||
|
|
||
|
if ( p != newp ) {
|
||
|
if ( strchr( p, "START" )) {
|
||
|
//
|
||
|
// set pointer to beginning of line
|
||
|
//
|
||
|
p = currentLine;
|
||
|
break;
|
||
|
}
|
||
|
} else {
|
||
|
//
|
||
|
// not found in this block; read in the next one
|
||
|
//
|
||
|
|
||
|
}
|
||
|
}
|
||
|
} while ( TRUE );
|
||
|
|
||
|
if ( *p == '\n' ) {
|
||
|
OmpLogFileLoWater = (DWORD)(currentPosition + ( p - buffer + 1 ));
|
||
|
}
|
||
|
|
||
|
if ( OmpLogFileLoWater == 0 ) {
|
||
|
//
|
||
|
// couldn't find any reasonable data. just set it to
|
||
|
// initial current position.
|
||
|
//
|
||
|
OmpLogFileLoWater = currentPosition + LOGBUF_SIZE;
|
||
|
}
|
||
|
}
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// finally, create the threadq that will handle the IO to the file
|
||
|
//
|
||
|
OmpLoggerWorkQueue = ClRtlCreateWorkQueue( 1, THREAD_PRIORITY_BELOW_NORMAL );
|
||
|
if ( OmpLoggerWorkQueue == NULL ) {
|
||
|
CloseHandle( OmpLogFileHandle );
|
||
|
|
||
|
ClRtlLogPrint(LOG_UNUSUAL,
|
||
|
"[OM] Unable to logger work queue. error %1!u!.\n",
|
||
|
GetLastError());
|
||
|
}
|
||
|
|
||
|
error_exit:
|
||
|
if ( logFileName != logFileBuffer && logFileName != NULL ) {
|
||
|
LocalFree( logFileName );
|
||
|
}
|
||
|
|
||
|
} // OmpOpenObjectLog
|
||
|
|
||
|
VOID
|
||
|
OmpLogStartRecord(
|
||
|
VOID
|
||
|
)
|
||
|
|
||
|
/*++
|
||
|
|
||
|
Routine Description:
|
||
|
|
||
|
Description
|
||
|
|
||
|
Arguments:
|
||
|
|
||
|
None
|
||
|
|
||
|
Return Value:
|
||
|
|
||
|
None
|
||
|
|
||
|
--*/
|
||
|
|
||
|
{
|
||
|
OSVERSIONINFOEXW version;
|
||
|
BOOL success;
|
||
|
PWCHAR suiteAbbrev;
|
||
|
SYSTEMTIME localTime;
|
||
|
|
||
|
GetLocalTime( &localTime );
|
||
|
|
||
|
version.dwOSVersionInfoSize = sizeof(version);
|
||
|
success = GetVersionExW((POSVERSIONINFOW)&version);
|
||
|
|
||
|
if ( success ) {
|
||
|
//
|
||
|
// Log the System version number
|
||
|
//
|
||
|
if ( version.wSuiteMask & VER_SUITE_DATACENTER ) {
|
||
|
suiteAbbrev = L"DTC";
|
||
|
} else if ( version.wSuiteMask & VER_SUITE_ENTERPRISE ) {
|
||
|
suiteAbbrev = L"ADS";
|
||
|
} else if ( version.wSuiteMask & VER_SUITE_EMBEDDEDNT ) {
|
||
|
suiteAbbrev = L"EMB";
|
||
|
} else if ( version.wProductType & VER_NT_WORKSTATION ) {
|
||
|
suiteAbbrev = L"WS";
|
||
|
} else if ( version.wProductType & VER_NT_DOMAIN_CONTROLLER ) {
|
||
|
suiteAbbrev = L"DC";
|
||
|
} else if ( version.wProductType & VER_NT_SERVER ) {
|
||
|
suiteAbbrev = L"SRV"; // otherwise - some non-descript Server
|
||
|
} else {
|
||
|
suiteAbbrev = L"";
|
||
|
}
|
||
|
|
||
|
OmpLogPrint(L"START %1!02d!/%2!02d!/%3!02d!-%4!02d!:%5!02d!:%6!02d!.%7!03d! %8!u! %9!u! "
|
||
|
L"%10!u! %11!u! %12!u! %13!u! \"%14!ws!\" "
|
||
|
L"%15!u! %16!u! %17!04X! (%18!ws!) %19!u!\n",
|
||
|
localTime.wYear,
|
||
|
localTime.wMonth,
|
||
|
localTime.wDay,
|
||
|
localTime.wHour,
|
||
|
localTime.wMinute,
|
||
|
localTime.wSecond,
|
||
|
localTime.wMilliseconds,
|
||
|
CLUSTER_GET_MAJOR_VERSION( CsMyHighestVersion ),
|
||
|
CLUSTER_GET_MINOR_VERSION( CsMyHighestVersion ),
|
||
|
version.dwMajorVersion, // param 10
|
||
|
version.dwMinorVersion,
|
||
|
version.dwBuildNumber,
|
||
|
version.dwPlatformId,
|
||
|
version.szCSDVersion,
|
||
|
version.wServicePackMajor, // param 15
|
||
|
version.wServicePackMinor,
|
||
|
version.wSuiteMask,
|
||
|
suiteAbbrev,
|
||
|
version.wProductType);
|
||
|
}
|
||
|
|
||
|
} // OmpLogStartRecord
|
||
|
|
||
|
VOID
|
||
|
OmpLogStopRecord(
|
||
|
VOID
|
||
|
)
|
||
|
|
||
|
/*++
|
||
|
|
||
|
Routine Description:
|
||
|
|
||
|
Description
|
||
|
|
||
|
Arguments:
|
||
|
|
||
|
None
|
||
|
|
||
|
Return Value:
|
||
|
|
||
|
None
|
||
|
|
||
|
--*/
|
||
|
|
||
|
{
|
||
|
OmpLogPrint( L"STOP\n" );
|
||
|
|
||
|
} // OmpLogStopRecord
|
||
|
|
||
|
|
||
|
/* end omlog.c */
|