log.c 3.69 KB
Newer Older
1
2
3
/* Copyright 2001,2002 Roger Dingledine, Matej Pfajfar. */
/* See LICENSE for licensing information */
/* $Id$ */
Roger Dingledine's avatar
Roger Dingledine committed
4

5
#include "../or/or.h"
6
#include "util.h"
7

8
9
10
11
12
13
14
15
16
17
18
19
struct logfile_t;

typedef struct logfile_t {
  struct logfile_t *next;
  const char *filename;
  FILE *file;
  int needs_close;
  int loglevel;
  int max_loglevel;
} logfile_t;

static INLINE const char *sev_to_string(int severity) {
20
  switch(severity) {
21
22
23
24
    case LOG_DEBUG:   return "debug";
    case LOG_INFO:    return "info";
    case LOG_WARNING: return "warn";
    case LOG_ERR:     return "err";
25
    default:          assert(0); return "UNKNOWN";
26
27
28
  }
}

29
static int loglevel = LOG_DEBUG;
30
static logfile_t *logfiles = NULL;
31

32
33
34
35
36
37
/* Format a log message into a fixed-sized buffer. (This is factored out
 * of 'logv' so that we never format a message more than once.
 */
static INLINE void format_msg(char *buf, size_t buf_len,
			      int severity, const char *funcname, 
			      const char *format, va_list ap) 
Roger Dingledine's avatar
Roger Dingledine committed
38
{
39
  time_t t;
40
  struct timeval now;
41
42
43
44
45
46
47
  int n;

  buf_len -= 2; /* subtract 2 characters so we have room for \n\0 */
  
  my_gettimeofday(&now);
  t = (time_t)now.tv_sec;
  
48
  n = strftime(buf, buf_len, "%b %d %H:%M:%S", localtime(&t));
49
  n += snprintf(buf+n, buf_len-n,
50
51
52
                ".%.3ld [%s] ", 
                (long)now.tv_usec / 1000, sev_to_string(severity));
  if(n > buf_len)
53
54
55
56
    n = buf_len-1; /* the *nprintf funcs return how many bytes they
                    * _would_ print, if the output is truncated.
                    * Subtract one because the count doesn't include the \0 */

57
  if (funcname) {
58
    n += snprintf(buf+n, buf_len-n, "%s(): ", funcname);
59
    if(n > buf_len)
60
      n = buf_len-1;
61
  }
62
63

  n += vsnprintf(buf+n,buf_len-n,format,ap);
64
  if(n > buf_len)
65
    n = buf_len-1;
66
67
68
69
70
71
72
  buf[n]='\n';
  buf[n+1]='\0';
}

static void 
logv(int severity, const char *funcname, const char *format, va_list ap)
{
73
  char buf[10024];
74
75
  int formatted = 0;
  logfile_t *lf;
76
  
77
  assert(format);
78
  if (severity < loglevel)
79
    return;
80
81
82
83
84
85
86
87
88
89
  if (!logfiles) {
    /* XXX This is a temporary measure until we get configuration support
       for logfiles. */
    add_stream_log(loglevel, "<stdout>", stdout);
  }
  for (lf = logfiles; lf; lf = lf->next) {
    if (severity < lf->loglevel || severity > lf->max_loglevel)
      continue;
    if (!lf->file)
      continue;
90

91
    if (!formatted) {
92
      format_msg(buf, 10024, severity, funcname, format, ap);
93
94
95
      formatted = 1;
    }
    fputs(buf, lf->file);
96
    /* XXX check for EOF */
97
  }
98
}
99

100
101
102
103
void 
log_set_severity(int severity)
{
  loglevel = severity;
104
105
106
107
108
109
110
111
112
}

/* Outputs a message to stdout */
void log(int severity, const char *format, ...)
{
  va_list ap;
  va_start(ap,format);
  logv(severity, NULL, format, ap);
  va_end(ap);
Roger Dingledine's avatar
Roger Dingledine committed
113
}
114

115
116
117
118
119
120
121
122
void _log_fn(int severity, const char *fn, const char *format, ...)
{
  va_list ap;
  va_start(ap,format);
  logv(severity, fn, format, ap);
  va_end(ap);
}

123
124
125
126
127
128
129
130
131
132
133
void close_logs()
{
  logfile_t *next, *lf;
  for (lf = logfiles; lf; lf = lf->next) {
    if (lf->needs_close)
      fclose(lf->file);
    next = lf->next;
    free(lf);
  }
  logfiles = NULL;
}
134

135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
void reset_logs()
{
  logfile_t *lf;
  for (lf = logfiles; lf; lf = lf->next) {
    if (lf->needs_close) {
      fclose(lf->file);
      lf->file = fopen(lf->filename, "a");
    }
  }
}

void add_stream_log(int loglevel, const char *name, FILE *stream)
{
  logfile_t *lf;
  lf = tor_malloc(sizeof(logfile_t));
  lf->filename = name;
  lf->needs_close = 0;
  lf->loglevel = loglevel;
  lf->max_loglevel = LOG_EMERG;
  lf->file = stream;
  lf->next = logfiles;
  logfiles = lf;
}

void add_file_log(int loglevel, const char *filename) 
{
  FILE *f;
  f = fopen(filename, "a");
  if (!f) return;
  add_stream_log(loglevel, filename, f);
  logfiles->needs_close = 1;
}
167