ConsoleAppender.c

 
#include <winpr/config.h>
 
#include "ConsoleAppender.h"
#include "Message.h"
 
#ifdef ANDROID
#include <android/log.h>
#endif
 
#define WLOG_CONSOLE_DEFAULT 0
#define WLOG_CONSOLE_STDOUT 1
#define WLOG_CONSOLE_STDERR 2
#define WLOG_CONSOLE_DEBUG 4
 
typedef struct
{
  wLogAppender common;
 
  int outputStream;
} wLogConsoleAppender;
 
static BOOL WLog_ConsoleAppender_Open(WINPR_ATTR_UNUSED wLog* log,
                                      WINPR_ATTR_UNUSED wLogAppender* appender)
{
  return TRUE;
}
 
static BOOL WLog_ConsoleAppender_Close(WINPR_ATTR_UNUSED wLog* log,
                                       WINPR_ATTR_UNUSED wLogAppender* appender)
{
  return TRUE;
}
 
static BOOL WLog_ConsoleAppender_WriteMessage(wLog* log, wLogAppender* appender,
                                              const wLogMessage* cmessage)
{
  if (!appender)
    return FALSE;
 
  wLogConsoleAppender* consoleAppender = (wLogConsoleAppender*)appender;
 
  char prefix[WLOG_MAX_PREFIX_SIZE] = WINPR_C_ARRAY_INIT;
  WLog_Layout_GetMessagePrefix(log, appender->Layout, cmessage, prefix, sizeof(prefix));
 
#ifdef _WIN32
  if (consoleAppender->outputStream == WLOG_CONSOLE_DEBUG)
  {
    OutputDebugStringA(prefix);
    OutputDebugStringA(cmessage->TextString);
    OutputDebugStringA("\n");
 
    return TRUE;
  }
#endif
#ifdef ANDROID
  android_LogPriority level;
  switch (cmessage->Level)
  {
    case WLOG_TRACE:
      level = ANDROID_LOG_VERBOSE;
      break;
    case WLOG_DEBUG:
      level = ANDROID_LOG_DEBUG;
      break;
    case WLOG_INFO:
      level = ANDROID_LOG_INFO;
      break;
    case WLOG_WARN:
      level = ANDROID_LOG_WARN;
      break;
    case WLOG_ERROR:
      level = ANDROID_LOG_ERROR;
      break;
    case WLOG_FATAL:
      level = ANDROID_LOG_FATAL;
      break;
    case WLOG_OFF:
      level = ANDROID_LOG_SILENT;
      break;
    default:
      level = ANDROID_LOG_FATAL;
      break;
  }
 
  if (level != ANDROID_LOG_SILENT)
    __android_log_print(level, log->Name, "%s%s", prefix, cmessage->TextString);
 
#else
  FILE* fp = nullptr;
  switch (consoleAppender->outputStream)
  {
    case WLOG_CONSOLE_STDOUT:
      fp = stdout;
      break;
    case WLOG_CONSOLE_STDERR:
      fp = stderr;
      break;
    default:
      switch (cmessage->Level)
      {
        case WLOG_TRACE:
        case WLOG_DEBUG:
        case WLOG_INFO:
          fp = stdout;
          break;
        default:
          fp = stderr;
          break;
      }
      break;
  }
 
  if (cmessage->Level != WLOG_OFF)
    (void)fprintf(fp, "%s%s\n", prefix, cmessage->TextString);
#endif
  return TRUE;
}
 
static int g_DataId = 0;
 
static BOOL WLog_ConsoleAppender_WriteDataMessage(WINPR_ATTR_UNUSED wLog* log,
                                                  WINPR_ATTR_UNUSED wLogAppender* appender,
                                                  const wLogMessage* message)
{
#if defined(ANDROID)
  return FALSE;
#else
  const int DataId = g_DataId++;
  char* FullFileName = WLog_Message_GetOutputFileName(DataId, "dat");
 
  WLog_DataMessage_Write(FullFileName, message->Data, message->Length);
 
  free(FullFileName);
 
  return TRUE;
#endif
}
 
static int g_ImageId = 0;
 
static BOOL WLog_ConsoleAppender_WriteImageMessage(WINPR_ATTR_UNUSED wLog* log,
                                                   WINPR_ATTR_UNUSED wLogAppender* appender,
                                                   const wLogMessage* message)
{
#if defined(ANDROID)
  return FALSE;
#else
  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;
#endif
}
 
static int g_PacketId = 0;
 
static BOOL WLog_ConsoleAppender_WritePacketMessage(WINPR_ATTR_UNUSED wLog* log,
                                                    wLogAppender* appender,
                                                    const wLogMessage* message)
{
#if defined(ANDROID)
  return FALSE;
#else
  g_PacketId++;
 
  if (!appender->PacketMessageContext)
  {
    char* FullFileName = WLog_Message_GetOutputFileName(-1, "pcap");
    appender->PacketMessageContext = (void*)Pcap_Open(FullFileName, TRUE);
    free(FullFileName);
  }
 
  if (appender->PacketMessageContext)
    return WLog_PacketMessage_Write((wPcap*)appender->PacketMessageContext, message->PacketData,
                                    message->PacketLength, message->PacketFlags);
 
  return TRUE;
#endif
}
static BOOL WLog_ConsoleAppender_Set(wLogAppender* appender, const char* setting, void* value)
{
  wLogConsoleAppender* consoleAppender = (wLogConsoleAppender*)appender;
 
  /* Just check the value string is not empty */
  if (!value || (strnlen(value, 2) == 0))
    return FALSE;
 
  if (strcmp("outputstream", setting) != 0)
    return FALSE;
 
  if (!strcmp("stdout", value))
    consoleAppender->outputStream = WLOG_CONSOLE_STDOUT;
  else if (!strcmp("stderr", value))
    consoleAppender->outputStream = WLOG_CONSOLE_STDERR;
  else if (!strcmp("default", value))
    consoleAppender->outputStream = WLOG_CONSOLE_DEFAULT;
  else if (!strcmp("debug", value))
    consoleAppender->outputStream = WLOG_CONSOLE_DEBUG;
  else
    return FALSE;
 
  return TRUE;
}
 
static void WLog_ConsoleAppender_Free(wLogAppender* appender)
{
  if (appender)
  {
    if (appender->PacketMessageContext)
    {
      Pcap_Close((wPcap*)appender->PacketMessageContext);
    }
 
    free(appender);
  }
}
 
wLogAppender* WLog_ConsoleAppender_New(WINPR_ATTR_UNUSED wLog* log)
{
  wLogConsoleAppender* ConsoleAppender =
      (wLogConsoleAppender*)calloc(1, sizeof(wLogConsoleAppender));
 
  if (!ConsoleAppender)
    return nullptr;
 
  ConsoleAppender->common.Type = WLOG_APPENDER_CONSOLE;
  ConsoleAppender->common.Open = WLog_ConsoleAppender_Open;
  ConsoleAppender->common.Close = WLog_ConsoleAppender_Close;
  ConsoleAppender->common.WriteMessage = WLog_ConsoleAppender_WriteMessage;
  ConsoleAppender->common.WriteDataMessage = WLog_ConsoleAppender_WriteDataMessage;
  ConsoleAppender->common.WriteImageMessage = WLog_ConsoleAppender_WriteImageMessage;
  ConsoleAppender->common.WritePacketMessage = WLog_ConsoleAppender_WritePacketMessage;
  ConsoleAppender->common.Set = WLog_ConsoleAppender_Set;
  ConsoleAppender->common.Free = WLog_ConsoleAppender_Free;
 
  ConsoleAppender->outputStream = WLOG_CONSOLE_DEFAULT;
 
#ifdef _WIN32
  if (IsDebuggerPresent())
    ConsoleAppender->outputStream = WLOG_CONSOLE_DEBUG;
#endif
 
  return &ConsoleAppender->common;
}