FileAppender.c

 
#include <winpr/config.h>
 
#include "FileAppender.h"
#include "Message.h"
 
#include <winpr/crt.h>
#include <winpr/environment.h>
#include <winpr/file.h>
#include <winpr/path.h>
 
typedef struct
{
  wLogAppender common;
 
  char* FileName;
  char* FilePath;
  char* FullFileName;
  FILE* FileDescriptor;
} wLogFileAppender;
 
static BOOL WLog_FileAppender_SetOutputFileName(wLogFileAppender* appender, const char* filename)
{
  WINPR_ASSERT(appender);
  WINPR_ASSERT(filename);
 
  appender->FileName = _strdup(filename);
 
  return appender->FileName != nullptr;
}
 
static BOOL WLog_FileAppender_SetOutputFilePath(wLogFileAppender* appender, const char* filepath)
{
  appender->FilePath = _strdup(filepath);
 
  return appender->FilePath != nullptr;
}
 
static BOOL WLog_FileAppender_Open(wLog* log, wLogAppender* appender)
{
  wLogFileAppender* fileAppender = nullptr;
 
  if (!log || !appender)
    return FALSE;
 
  fileAppender = (wLogFileAppender*)appender;
 
  if (!fileAppender->FilePath)
  {
    fileAppender->FilePath = GetKnownSubPath(KNOWN_PATH_TEMP, "wlog");
 
    if (!fileAppender->FilePath)
      return FALSE;
  }
 
  if (!fileAppender->FileName)
  {
    fileAppender->FileName = (char*)malloc(MAX_PATH);
 
    if (!fileAppender->FileName)
      return FALSE;
 
    (void)sprintf_s(fileAppender->FileName, MAX_PATH, "%" PRIu32 ".log", GetCurrentProcessId());
  }
 
  if (!fileAppender->FullFileName)
  {
    fileAppender->FullFileName =
        GetCombinedPath(fileAppender->FilePath, fileAppender->FileName);
 
    if (!fileAppender->FullFileName)
      return FALSE;
  }
 
  if (!winpr_PathFileExists(fileAppender->FilePath))
  {
    if (!winpr_PathMakePath(fileAppender->FilePath, nullptr))
      return FALSE;
 
    UnixChangeFileMode(fileAppender->FilePath, 0xFFFF);
  }
 
  fileAppender->FileDescriptor = winpr_fopen(fileAppender->FullFileName, "a+");
 
  return fileAppender->FileDescriptor != nullptr;
}
 
static BOOL WLog_FileAppender_Close(wLog* log, wLogAppender* appender)
{
  wLogFileAppender* fileAppender = nullptr;
 
  if (!log || !appender)
    return FALSE;
 
  fileAppender = (wLogFileAppender*)appender;
 
  if (!fileAppender->FileDescriptor)
    return TRUE;
 
  (void)fclose(fileAppender->FileDescriptor);
  fileAppender->FileDescriptor = nullptr;
  return TRUE;
}
 
static BOOL WLog_FileAppender_WriteMessage(wLog* log, wLogAppender* appender,
                                           const wLogMessage* cmessage)
{
  if (!log || !appender || !cmessage)
    return FALSE;
 
  wLogFileAppender* fileAppender = (wLogFileAppender*)appender;
  FILE* fp = fileAppender->FileDescriptor;
 
  if (!fp)
    return FALSE;
 
  char prefix[WLOG_MAX_PREFIX_SIZE] = WINPR_C_ARRAY_INIT;
  WLog_Layout_GetMessagePrefix(log, appender->Layout, cmessage, prefix, sizeof(prefix));
  (void)fprintf(fp, "%s%s\n", prefix, cmessage->TextString);
  (void)fflush(fp); /* slow! */
  return TRUE;
}
 
static int g_DataId = 0;
 
static BOOL WLog_FileAppender_WriteDataMessage(wLog* log, wLogAppender* appender,
                                               const wLogMessage* message)
{
  if (!log || !appender || !message)
    return FALSE;
 
  const int DataId = g_DataId++;
  char* FullFileName = WLog_Message_GetOutputFileName(DataId, "dat");
  WLog_DataMessage_Write(FullFileName, message->Data, message->Length);
  free(FullFileName);
  return TRUE;
}
 
static int g_ImageId = 0;
 
static BOOL WLog_FileAppender_WriteImageMessage(wLog* log, wLogAppender* appender,
                                                const wLogMessage* message)
{
  if (!log || !appender || !message)
    return FALSE;
 
  const int ImageId = g_ImageId++;
  char* FullFileName = WLog_Message_GetOutputFileName(ImageId, "bmp");
  WLog_ImageMessage_Write(FullFileName, message->ImageData, message->ImageWidth,
                          message->ImageHeight, message->ImageBpp);
  free(FullFileName);
  return TRUE;
}
 
static BOOL WLog_FileAppender_Set(wLogAppender* appender, const char* setting, void* value)
{
  wLogFileAppender* fileAppender = (wLogFileAppender*)appender;
 
  /* Just check the value string is not empty */
  if (!value || (strnlen(value, 2) == 0))
    return FALSE;
 
  if (!strcmp("outputfilename", setting))
    return WLog_FileAppender_SetOutputFileName(fileAppender, (const char*)value);
 
  if (!strcmp("outputfilepath", setting))
    return WLog_FileAppender_SetOutputFilePath(fileAppender, (const char*)value);
 
  return FALSE;
}
 
static void WLog_FileAppender_Free(wLogAppender* appender)
{
  wLogFileAppender* fileAppender = nullptr;
 
  if (appender)
  {
    fileAppender = (wLogFileAppender*)appender;
    free(fileAppender->FileName);
    free(fileAppender->FilePath);
    free(fileAppender->FullFileName);
    free(fileAppender);
  }
}
 
wLogAppender* WLog_FileAppender_New(WINPR_ATTR_UNUSED wLog* log)
{
  LPSTR env = nullptr;
  LPCSTR name = nullptr;
  DWORD nSize = 0;
  wLogFileAppender* FileAppender = nullptr;
  FileAppender = (wLogFileAppender*)calloc(1, sizeof(wLogFileAppender));
 
  if (!FileAppender)
    return nullptr;
 
  FileAppender->common.Type = WLOG_APPENDER_FILE;
  FileAppender->common.Open = WLog_FileAppender_Open;
  FileAppender->common.Close = WLog_FileAppender_Close;
  FileAppender->common.WriteMessage = WLog_FileAppender_WriteMessage;
  FileAppender->common.WriteDataMessage = WLog_FileAppender_WriteDataMessage;
  FileAppender->common.WriteImageMessage = WLog_FileAppender_WriteImageMessage;
  FileAppender->common.Free = WLog_FileAppender_Free;
  FileAppender->common.Set = WLog_FileAppender_Set;
  name = "WLOG_FILEAPPENDER_OUTPUT_FILE_PATH";
  nSize = GetEnvironmentVariableA(name, nullptr, 0);
 
  if (nSize)
  {
    BOOL status = 0;
    env = (LPSTR)malloc(nSize);
 
    if (!env)
      goto error_free;
 
    if (GetEnvironmentVariableA(name, env, nSize) != nSize - 1)
    {
      free(env);
      goto error_free;
    }
 
    status = WLog_FileAppender_SetOutputFilePath(FileAppender, env);
    free(env);
 
    if (!status)
      goto error_free;
  }
 
  name = "WLOG_FILEAPPENDER_OUTPUT_FILE_NAME";
  nSize = GetEnvironmentVariableA(name, nullptr, 0);
 
  if (nSize)
  {
    BOOL status = FALSE;
    env = (LPSTR)malloc(nSize);
 
    if (!env)
      goto error_output_file_name;
 
    if (GetEnvironmentVariableA(name, env, nSize) == nSize - 1)
      status = WLog_FileAppender_SetOutputFileName(FileAppender, env);
    free(env);
 
    if (!status)
      goto error_output_file_name;
  }
 
  return (wLogAppender*)FileAppender;
error_output_file_name:
  free(FileAppender->FilePath);
error_free:
  free(FileAppender);
  return nullptr;
}