/*
 * Configurable logging and traceing
 *
 * See the README file for copyright information and how to reach the author.
 *
 * $Id: setup-softlog.c,v 1.10 2011-04-13 19:56:58 lucke Exp $
 */

#include "setup-softlog.h"

#ifdef HAVE_CONFIG
# include "config.h"
#endif
#include <stdarg.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/syscall.h>
#include <sys/time.h>

#define MAXSYSLOGBUF 256

/* ---------------------------------------------------------------------------
 */
cSetupSoftlog::cSetupSoftlog()
{
  logFile = stderr;
  logFileName = strdup("stderr");
  appendPID = true;
  enabledPriorities = SOFT_LOG_ERROR | SOFT_LOG_INFO | SOFT_LOG_DEBUG;
  traceFlags = 0;
}

/* ---------------------------------------------------------------------------
 */
cSetupSoftlog::~cSetupSoftlog()
{
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::IsEnabled(int priority)
{
  return priority & enabledPriorities;
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::LogPriority(int priority)
{
  // no syslog loging for trace messages
  if (priority & SOFT_LOG_TRACE)
    return NO_LOG;

  if (IsEnabled(priority) & SOFT_LOG_ERROR)
    return LOG_ERR;
  if (IsEnabled(priority) & SOFT_LOG_INFO)
    return LOG_INFO;
  if (IsEnabled(priority) & SOFT_LOG_DEBUG)
    return LOG_DEBUG;

  // unknown logtype -> return invalid priority
  return NO_LOG;
}

/* ---------------------------------------------------------------------------
 */
char cSetupSoftlog::Priority2Char(int priority)
{
  if (priority & SOFT_LOG_TRACE)
    return 'T';
  if (priority & SOFT_LOG_ERROR)
    return 'E';
  if (priority & SOFT_LOG_INFO)
    return 'I';
  if (priority & SOFT_LOG_DEBUG)
    return 'D';
  return '?';
}

/* ---------------------------------------------------------------------------
 */
void cSetupSoftlog::CloseLogFile()
{
  if (logFile && logFile != stderr && logFile != stdout)
    fclose(logFile);

  free(logFileName);
  logFileName = NULL;
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::SetLogPriorities(int newPriorities)
{
    int oldenabledPriorities = enabledPriorities;

  /* -------------------------------------------------------------------------
   * set new priority flags, but error messages are allways enabled
   */
  enabledPriorities = (newPriorities | SOFT_LOG_ERROR);
  return oldenabledPriorities;
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::GetLogPriorities()
{
  return enabledPriorities;
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::SetTraceFlags(int newFlags)
{
    int oldTraceFlags = traceFlags;

  traceFlags = newFlags;
  return oldTraceFlags;
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::GetTraceFlags()
{
  return traceFlags;
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::SetAppendMode(int newMode)
{
    int oldMode = appendPID;

  if (newMode != appendPID) {
      char *oldName = strdup (logFileName);

    appendPID = newMode;
    SetLogFile(oldName);
    free (oldName);
  }

  return oldMode;
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::GetAppendMode()
{
  return appendPID;
}

/* ---------------------------------------------------------------------------
 */
int cSetupSoftlog::SetLogFile(const char *fileName)
{
  /* -------------------------------------------------------------------------
   * Ignore zero length file names
   */
  if (!strlen (fileName)) {
    Log (SOFT_LOG_ERROR, 0, "[softlog] ignoring zero length fileName\n");
    return 0;
  }

  /* -------------------------------------------------------------------------
   * Don't close stderr. Handle fileName "stderr" special so that it is
   * possible to switch back from file logging to stderr.
   */
  CloseLogFile();
  if (!strcmp (fileName, "stderr")) {
    logFile = stderr;
  } else if (!strcmp (fileName, "stdout")) {
    logFile = stdout;
  } else {
      char realName [256];

    if (appendPID)
      snprintf (realName, sizeof(realName), "%s-%d", fileName, getpid());
    else
      snprintf (realName, sizeof(realName), "%s", fileName);

    if (!(logFile = fopen (realName, "w+"))) {
      logFile = stderr;
      logFileName = strdup("stderr");

      Log (SOFT_LOG_ERROR, 0,
              "[softlog] X could not open logfile (%s) len %d %s, using stderr\n",
              realName, strlen (fileName), fileName);
      return 0;
    }
  }

  logFileName = strdup(fileName);
  return 1;
}

/* ---------------------------------------------------------------------------
 */
char *cSetupSoftlog::GetLogFileName()
{
  return logFileName;
}

/* ---------------------------------------------------------------------------
 */
void cSetupSoftlog::DisableLog2File()
{
  CloseLogFile();
  logFile = NULL;
}

/* ---------------------------------------------------------------------------
 */
int SoftdeviceGetTid()
{
#ifdef __APPLE__
  return getpid();
#else
  return syscall(__NR_gettid);
#endif
};

/* ---------------------------------------------------------------------------
 */
void cSetupSoftlog::Log(int currPriority,
                        int traceFlags,
                        const char *format, ...)
{
    va_list argList;
    int     priority;
    char    fmt[MAXSYSLOGBUF];

  if (!IsEnabled(currPriority))
    return;

  priority = LogPriority(currPriority);

  if (priority != NO_LOG) {
    snprintf(fmt, sizeof(fmt), "[%d] %s",SoftdeviceGetTid(), format);
    va_start(argList, format);
    vsyslog(priority, fmt, argList);
    va_end(argList);
  }

  if (logFile) {
      struct timeval now;
      struct tm *tmp;

    gettimeofday(&now, NULL);
    tmp = localtime(&now.tv_sec);
    snprintf(fmt, sizeof(fmt), "%02d:%02d:%02d.%04d %c [%d] %s",
             tmp->tm_hour, tmp->tm_min, tmp->tm_sec, (int) now.tv_usec/1000,
             Priority2Char(currPriority),
             SoftdeviceGetTid(), format);
    va_start(argList, format);
    vfprintf(logFile, fmt, argList);
    va_end(argList);
    fflush(logFile);
  }
}

/* ---------------------------------------------------------------------------
 */
bool cSetupSoftlog::Parse(const char *name, const char *value)
{
  if (!strcasecmp(name,"softlog-priorities"))
    SetLogPriorities(atoi(value));
  else if (!strcasecmp(name,"softlog-file"))
    SetLogFile(value);
  else if (!strcasecmp(name,"softlog-appendpid"))
    SetAppendMode(atoi(value));
  else
    return false;

  return true;
}