FreeRDP/winpr/libwinpr/utils/wlog/BinaryAppender.c

239 lines
6.2 KiB
C
Raw Normal View History

/**
* WinPR: Windows Portable Runtime
* WinPR Logger
*
* Copyright 2013 Marc-Andre Moreau <marcandre.moreau@gmail.com>
* Copyright 2015 Thincast Technologies GmbH
* Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "BinaryAppender.h"
#include <winpr/crt.h>
#include <winpr/file.h>
#include <winpr/path.h>
#include <winpr/stream.h>
struct _wLogBinaryAppender
{
WLOG_APPENDER_COMMON();
char* FileName;
char* FilePath;
char* FullFileName;
FILE* FileDescriptor;
};
typedef struct _wLogBinaryAppender wLogBinaryAppender;
static BOOL WLog_BinaryAppender_Open(wLog* log, wLogAppender* appender)
{
wLogBinaryAppender* binaryAppender;
if (!log || !appender)
return FALSE;
2019-11-06 17:24:51 +03:00
binaryAppender = (wLogBinaryAppender*)appender;
if (!binaryAppender->FileName)
{
2019-11-06 17:24:51 +03:00
binaryAppender->FileName = (char*)malloc(MAX_PATH);
if (!binaryAppender->FileName)
return FALSE;
2019-11-06 17:24:51 +03:00
sprintf_s(binaryAppender->FileName, MAX_PATH, "%" PRIu32 ".wlog", GetCurrentProcessId());
}
if (!binaryAppender->FilePath)
{
binaryAppender->FilePath = GetKnownSubPath(KNOWN_PATH_TEMP, "wlog");
if (!binaryAppender->FilePath)
return FALSE;
}
if (!binaryAppender->FullFileName)
{
2019-11-06 17:24:51 +03:00
binaryAppender->FullFileName =
GetCombinedPath(binaryAppender->FilePath, binaryAppender->FileName);
if (!binaryAppender->FullFileName)
return FALSE;
}
if (!winpr_PathFileExists(binaryAppender->FilePath))
{
if (!winpr_PathMakePath(binaryAppender->FilePath, 0))
return FALSE;
UnixChangeFileMode(binaryAppender->FilePath, 0xFFFF);
}
binaryAppender->FileDescriptor = winpr_fopen(binaryAppender->FullFileName, "a+");
if (!binaryAppender->FileDescriptor)
return FALSE;
return TRUE;
}
static BOOL WLog_BinaryAppender_Close(wLog* log, wLogAppender* appender)
{
wLogBinaryAppender* binaryAppender;
if (!appender)
return FALSE;
2019-11-06 17:24:51 +03:00
binaryAppender = (wLogBinaryAppender*)appender;
if (!binaryAppender->FileDescriptor)
return TRUE;
fclose(binaryAppender->FileDescriptor);
binaryAppender->FileDescriptor = NULL;
return TRUE;
}
2019-11-06 17:24:51 +03:00
static BOOL WLog_BinaryAppender_WriteMessage(wLog* log, wLogAppender* appender,
wLogMessage* message)
{
FILE* fp;
wStream* s;
size_t MessageLength;
size_t FileNameLength;
size_t FunctionNameLength;
size_t TextStringLength;
BOOL ret = TRUE;
wLogBinaryAppender* binaryAppender;
if (!log || !appender || !message)
return FALSE;
2019-11-06 17:24:51 +03:00
binaryAppender = (wLogBinaryAppender*)appender;
fp = binaryAppender->FileDescriptor;
if (!fp)
return FALSE;
FileNameLength = strnlen(message->FileName, INT_MAX);
FunctionNameLength = strnlen(message->FunctionName, INT_MAX);
TextStringLength = strnlen(message->TextString, INT_MAX);
2019-11-06 17:24:51 +03:00
MessageLength =
16 + (4 + FileNameLength + 1) + (4 + FunctionNameLength + 1) + (4 + TextStringLength + 1);
2019-11-06 17:24:51 +03:00
if ((MessageLength > UINT32_MAX) || (FileNameLength > UINT32_MAX) ||
(FunctionNameLength > UINT32_MAX) || (TextStringLength > UINT32_MAX))
return FALSE;
s = Stream_New(NULL, MessageLength);
if (!s)
return FALSE;
Stream_Write_UINT32(s, (UINT32)MessageLength);
Stream_Write_UINT32(s, message->Type);
Stream_Write_UINT32(s, message->Level);
Stream_Write_UINT32(s, message->LineNumber);
Stream_Write_UINT32(s, (UINT32)FileNameLength);
Stream_Write(s, message->FileName, FileNameLength + 1);
Stream_Write_UINT32(s, (UINT32)FunctionNameLength);
Stream_Write(s, message->FunctionName, FunctionNameLength + 1);
Stream_Write_UINT32(s, (UINT32)TextStringLength);
Stream_Write(s, message->TextString, TextStringLength + 1);
Stream_SealLength(s);
if (fwrite(Stream_Buffer(s), MessageLength, 1, fp) != 1)
ret = FALSE;
Stream_Free(s, TRUE);
return ret;
}
2019-11-06 17:24:51 +03:00
static BOOL WLog_BinaryAppender_WriteDataMessage(wLog* log, wLogAppender* appender,
wLogMessage* message)
{
return TRUE;
}
2019-11-06 17:24:51 +03:00
static BOOL WLog_BinaryAppender_WriteImageMessage(wLog* log, wLogAppender* appender,
wLogMessage* message)
{
return TRUE;
}
2019-11-06 17:24:51 +03:00
static BOOL WLog_BinaryAppender_Set(wLogAppender* appender, const char* setting, void* value)
{
2019-11-06 17:24:51 +03:00
wLogBinaryAppender* binaryAppender = (wLogBinaryAppender*)appender;
/* Just check if the value string is longer than 0 */
if (!value || (strnlen(value, 2) == 0))
return FALSE;
if (!strcmp("outputfilename", setting))
{
2019-11-06 17:24:51 +03:00
binaryAppender->FileName = _strdup((const char*)value);
if (!binaryAppender->FileName)
return FALSE;
}
else if (!strcmp("outputfilepath", setting))
{
2019-11-06 17:24:51 +03:00
binaryAppender->FilePath = _strdup((const char*)value);
if (!binaryAppender->FilePath)
return FALSE;
}
else
return FALSE;
return TRUE;
}
static void WLog_BinaryAppender_Free(wLogAppender* appender)
{
2019-11-06 17:24:51 +03:00
wLogBinaryAppender* binaryAppender;
if (appender)
{
2019-11-06 17:24:51 +03:00
binaryAppender = (wLogBinaryAppender*)appender;
free(binaryAppender->FileName);
free(binaryAppender->FilePath);
free(binaryAppender->FullFileName);
free(binaryAppender);
}
}
wLogAppender* WLog_BinaryAppender_New(wLog* log)
{
wLogBinaryAppender* BinaryAppender;
2019-11-06 17:24:51 +03:00
BinaryAppender = (wLogBinaryAppender*)calloc(1, sizeof(wLogBinaryAppender));
if (!BinaryAppender)
return NULL;
BinaryAppender->Type = WLOG_APPENDER_BINARY;
BinaryAppender->Open = WLog_BinaryAppender_Open;
BinaryAppender->Close = WLog_BinaryAppender_Close;
BinaryAppender->WriteMessage = WLog_BinaryAppender_WriteMessage;
BinaryAppender->WriteDataMessage = WLog_BinaryAppender_WriteDataMessage;
BinaryAppender->WriteImageMessage = WLog_BinaryAppender_WriteImageMessage;
BinaryAppender->Free = WLog_BinaryAppender_Free;
BinaryAppender->Set = WLog_BinaryAppender_Set;
2019-11-06 17:24:51 +03:00
return (wLogAppender*)BinaryAppender;
}