debug.c

   1 /*
   2  * Copyright (C) 2007 Stig Venaas <venaas@uninett.no>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  */
   8
   9 #ifndef SYS_SOLARIS9
  10 #include <stdint.h>
  11 #endif
  12 #include <stdio.h>
  13 #include <stdlib.h>
  14 #include <stdarg.h>
  15 #include <string.h>
  16 #include <strings.h>
  17 #include <time.h>
  18 #include <sys/time.h>
  19 #include <syslog.h>
  20 #include <errno.h>
  21 #include <assert.h>
  22 #include "debug.h"
  23 #include "util.h"
  24
  25 static char *debug_ident = NULL;
  26 static uint8_t debug_level = DBG_INFO;
  27 static char *debug_filepath = NULL;
  28 static FILE *debug_file = NULL;
  29 static int debug_syslogfacility = 0;
  30 static uint8_t debug_timestamp = 0;
  31
  32 void debug_init(char *ident) {
  33     debug_file = stderr;
  34     setvbuf(debug_file, NULL, _IONBF, 0);
  35     debug_ident = ident;
  36 }
  37
  38 void debug_set_level(uint8_t level) {
  39     switch (level) {
  40     case 1:
  41         debug_level = DBG_ERR;
  42         return;
  43     case 2:
  44         debug_level = DBG_WARN;
  45         return;
  46     case 3:
  47         debug_level = DBG_INFO;
  48         return;
  49     case 4:
  50         debug_level = DBG_DBG;
  51         return;
  52     }
  53 }
  54
  55 void debug_timestamp_on() {
  56     debug_timestamp = 1;
  57 }
  58
  59 uint8_t debug_get_level() {
  60     return debug_level;
  61 }
  62
  63 int debug_set_destination(char *dest) {
  64     static const char *facstrings[] = { "LOG_DAEMON", "LOG_MAIL", "LOG_USER", "LOG_LOCAL0",
  65                                         "LOG_LOCAL1", "LOG_LOCAL2", "LOG_LOCAL3", "LOG_LOCAL4",
  66                                         "LOG_LOCAL5", "LOG_LOCAL6", "LOG_LOCAL7", NULL };
  67     static const int facvals[] = { LOG_DAEMON, LOG_MAIL, LOG_USER, LOG_LOCAL0,
  68                                    LOG_LOCAL1, LOG_LOCAL2, LOG_LOCAL3, LOG_LOCAL4,
  69                                    LOG_LOCAL5, LOG_LOCAL6, LOG_LOCAL7 };
  70     extern int errno;
  71     int i;
  72
  73     if (!strncasecmp(dest, "file:///", 8)) {
  74         debug_filepath = stringcopy(dest + 7, 0);
  75         debug_file = fopen(debug_filepath, "a");
  76         if (!debug_file) {
  77             debug_file = stderr;
  78             debugx(1, DBG_ERR, "Failed to open logfile %s\n%s",
  79                    debug_filepath, strerror(errno));
  80         }
  81         setvbuf(debug_file, NULL, _IONBF, 0);
  82         return 1;
  83     }
  84     if (!strncasecmp(dest, "x-syslog://", 11)) {
  85         dest += 11;
  86         if (*dest == '/')
  87             dest++;
  88         if (*dest) {
  89             for (i = 0; facstrings[i]; i++)
  90                 if (!strcasecmp(dest, facstrings[i]))
  91                     break;
  92             if (!facstrings[i])
  93                 debugx(1, DBG_ERR, "Unknown syslog facility %s", dest);
  94             debug_syslogfacility = facvals[i];
  95         } else
  96             debug_syslogfacility = LOG_DAEMON;
  97         openlog(debug_ident, LOG_PID, debug_syslogfacility);
  98         return 1;
  99     }
 100     debug(DBG_ERR, "Unknown log destination, exiting %s", dest);
 101     exit(1);
 102 }
 103
 104 void debug_reopen_log() {
 105     extern int errno;
 106
 107     /* not a file, noop, return success */
 108     if (!debug_filepath) {
 109         debug(DBG_ERR, "skipping reopen");
 110         return;
 111     }
 112
 113     if (debug_file != stderr)
 114         fclose(debug_file);
 115
 116     debug_file = fopen(debug_filepath, "a");
 117     if (debug_file)
 118         debug(DBG_ERR, "Reopened logfile %s", debug_filepath);
 119     else {
 120         debug_file = stderr;
 121         debug(DBG_ERR, "Failed to open logfile %s, using stderr\n%s",
 122               debug_filepath, strerror(errno));
 123     }
 124     setvbuf(debug_file, NULL, _IONBF, 0);
 125 }
 126
 127 void debug_logit(uint8_t level, const char *format, va_list ap) {
 128     struct timeval now;
 129     char *timebuf;
 130     int priority;
 131
 132     if (debug_syslogfacility) {
 133         switch (level) {
 134         case DBG_DBG:
 135             priority = LOG_DEBUG;
 136             break;
 137         case DBG_INFO:
 138             priority = LOG_INFO;
 139             break;
 140         case DBG_WARN:
 141             priority = LOG_WARNING;
 142             break;
 143         case DBG_ERR:
 144             priority = LOG_ERR;
 145             break;
 146         default:
 147             priority = LOG_DEBUG;
 148         }
 149         vsyslog(priority, format, ap);
 150     } else {
 151         if (debug_timestamp && (timebuf = malloc(256))) {
 152             gettimeofday(&now, NULL);
 153             ctime_r(&now.tv_sec, timebuf);
 154             timebuf[strlen(timebuf) - 1] = '\0';
 155             fprintf(debug_file, "%s: ", timebuf + 4);
 156             free(timebuf);
 157         }
 158         vfprintf(debug_file, format, ap);
 159         fprintf(debug_file, "\n");
 160     }
 161 }
 162
 163 void debug(uint8_t level, char *format, ...) {
 164     va_list ap;
 165     if (level < debug_level)
 166         return;
 167     va_start(ap, format);
 168     debug_logit(level, format, ap);
 169     va_end(ap);
 170 }
 171
 172 void debugx(int status, uint8_t level, char *format, ...) {
 173     if (level >= debug_level) {
 174         va_list ap;
 175         va_start(ap, format);
 176         debug_logit(level, format, ap);
 177         va_end(ap);
 178     }
 179     exit(status);
 180 }
 181
 182 void debugerrno(int err, uint8_t level, char *format, ...) {
 183     if (level >= debug_level) {
 184         va_list ap;
 185         size_t len = strlen(format);
 186         char *tmp = malloc(len + 1024 + 2);
 187         assert(tmp);
 188         strcpy(tmp, format);
 189         tmp[len++] = ':';
 190         tmp[len++] = ' ';
 191         if (strerror_r(err, tmp + len, 1024))
 192             tmp = format;
 193         va_start(ap, format);
 194         debug_logit(level, tmp, ap);
 195         va_end(ap);
 196     }
 197 }
 198
 199 void debugerrnox(int err, uint8_t level, char *format, ...) {
 200     if (level >= debug_level) {
 201         va_list ap;
 202         va_start(ap, format);
 203         debugerrno(err, level, format, ap);
 204         va_end(ap);
 205     }
 206     exit(err);
 207 }
 208
 209 /* Local Variables: */
 210 /* c-file-style: "stroustrup" */
 211 /* End: */