src/libdbg/dbglog.c

   1 /* Understanding is not required. Only obedience.
   2  *
   3  * This program is free software. It comes without any warranty, to
   4  * the extent permitted by applicable law. You can redistribute it
   5  * and/or modify it under the terms of the Do What The Fuck You Want
   6  * To Public License, Version 2, as published by Sam Hocevar. See
   7  * http://sam.zoy.org/wtfpl/COPYING for more details. */
   8 /* debug logs */
   9 #include "dbglog.h"
  10
  11 #define DBG_TMP_BUF_SIZE  (65536)
  12
  13 #ifdef WIN32
  14 # include <windows.h>
  15 #endif
  16
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19 #include <stdarg.h>
  20 #include <string.h>
  21 #include <time.h>
  22 #include <unistd.h>
  23
  24 #include <sys/types.h>
  25
  26
  27 ////////////////////////////////////////////////////////////////////////////////
  28 DbgLogUpdateFn dbgLogPreUpdate = NULL;
  29 DbgLogUpdateFn dbgLogPostUpdate = NULL;
  30
  31
  32 ////////////////////////////////////////////////////////////////////////////////
  33 static char ccLock = 0;
  34
  35
  36 static void dbgLock (void) {
  37   while (__sync_bool_compare_and_swap(&ccLock, 0, 1)) {
  38 #ifndef WIN32
  39     usleep(1);
  40 #else
  41     Sleep(1); // i don't care
  42 #endif
  43   }
  44 }
  45
  46
  47 static void dbgUnlock (void) {
  48   //__sync_synchronize();
  49   //ccLock = 0;
  50   __sync_fetch_and_and(&ccLock, 0);
  51 }
  52
  53
  54 ////////////////////////////////////////////////////////////////////////////////
  55 #ifndef NO_DEBUG_LOG
  56 static char dbgLogFileName[8192] = {0};
  57 static int dbgLogWriteToFile = 1;
  58 static int dbgLogWriteToScr = 0;
  59 #endif
  60
  61
  62 #ifndef NO_DEBUG_LOG
  63 static __attribute__((constructor(DBGLOG_CONSTRUCTOR_PRIO))) void dbgLogInit (void) {
  64   if (!dbgLogFileName[0]) dbglog_set_filename("dlog.log");
  65 }
  66 #endif
  67
  68
  69 void dbglog_set_filename (const char *fname) {
  70 #ifndef NO_DEBUG_LOG
  71   dbgLock();
  72   if (fname && fname[0]) {
  73     strcpy(dbgLogFileName, fname);
  74   } else {
  75     dbgLogFileName[0] = '\0';
  76   }
  77   dbgUnlock();
  78 #endif
  79 }
  80
  81
  82 int dbglog_set_fileout (int doIt) {
  83 #ifndef NO_DEBUG_LOG
  84   int old;
  85   dbgLock();
  86   old = dbgLogWriteToFile;
  87   dbgLogWriteToFile = (doIt ? 1 : 0);
  88   dbgUnlock();
  89   return old;
  90 #else
  91   return 0;
  92 #endif
  93 }
  94
  95
  96 int dbglog_set_screenout (int doIt) {
  97 #ifndef NO_DEBUG_LOG
  98   int old;
  99   dbgLock();
 100   old = dbgLogWriteToScr;
 101   dbgLogWriteToScr = (doIt ? 1 : 0);
 102   dbgUnlock();
 103   return old;
 104 #else
 105   return 0;
 106 #endif
 107 }
 108
 109
 110 #ifndef NO_DEBUG_LOG
 111 __attribute__((format(printf, 1, 2))) void dlogf (const char *fmt, ...) {
 112   va_list ap;
 113   va_start(ap, fmt);
 114   dlogfVA(fmt, ap);
 115   va_end(ap);
 116 }
 117
 118
 119 static void writebuf (FILE *fo, const char *sbuf, const char *pfx) {
 120   int plen = (pfx != NULL ? strlen(pfx) : 0);
 121   if (!sbuf[0]) {
 122     if (plen > 0) fwrite(pfx, plen, 1, fo);
 123     fwrite("\n", 1, 1, fo);
 124     return;
 125   }
 126   while (*sbuf) {
 127     const char *nlp;
 128     int olen;
 129     if (plen > 0) fwrite(pfx, plen, 1, fo);
 130     nlp = strchr(sbuf, '\n');
 131     if (nlp == NULL) {
 132       olen = strlen(sbuf);
 133       nlp = sbuf+olen;
 134     } else {
 135       olen = nlp-sbuf;
 136       if (olen > 1 && nlp[-1] == '\r') --olen;
 137       ++nlp; // skip 'newline'
 138     }
 139     if (olen > 0) fwrite(sbuf, olen, 1, fo);
 140     fwrite("\n", 1, 1, fo);
 141     sbuf = nlp;
 142   }
 143 }
 144
 145
 146 void dlogfVA (const char *fmt, va_list inap) {
 147   static char timebuf[128];
 148   static char buf[DBG_TMP_BUF_SIZE];
 149   time_t t;
 150   struct tm *bt;
 151   if (fmt == NULL && !fmt[0]) return;
 152   dbgLock();
 153   if (dbgLogWriteToFile || dbgLogWriteToScr) {
 154     vsnprintf(buf, sizeof(buf), fmt, inap);
 155     if (dbgLogPreUpdate) dbgLogPreUpdate();
 156     t = time(NULL);
 157     bt = localtime(&t);
 158     snprintf(timebuf, sizeof(timebuf), "[%05u] %04d/%02d/%02d %02d:%02d:%02d: ",
 159       (unsigned int)getpid(),
 160       bt->tm_year+1900, bt->tm_mon+1, bt->tm_mday,
 161       bt->tm_hour, bt->tm_min, bt->tm_sec
 162     );
 163     if (dbgLogWriteToFile && dbgLogFileName[0]) {
 164       FILE *outfile = fopen(dbgLogFileName, "ab");
 165       if (outfile) {
 166         writebuf(outfile, buf, timebuf);
 167         fclose(outfile);
 168       }
 169     }
 170     if (dbgLogWriteToScr && stderr != NULL) writebuf(stderr, buf, timebuf+13); // skip PID and year
 171     if (dbgLogPostUpdate) dbgLogPostUpdate();
 172   }
 173   dbgUnlock();
 174 }
 175 #endif