source4/heimdal/lib/krb5/log.c

   1 /*
   2  * Copyright (c) 1997-2006 Kungliga Tekniska Högskolan
   3  * (Royal Institute of Technology, Stockholm, Sweden).
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  *
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  *
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * 3. Neither the name of the Institute nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #include "krb5_locl.h"
  35
  36 RCSID("$Id$");
  37
  38 struct facility {
  39     int min;
  40     int max;
  41     krb5_log_log_func_t log_func;
  42     krb5_log_close_func_t close_func;
  43     void *data;
  44 };
  45
  46 static struct facility*
  47 log_realloc(krb5_log_facility *f)
  48 {
  49     struct facility *fp;
  50     fp = realloc(f->val, (f->len + 1) * sizeof(*f->val));
  51     if(fp == NULL)
  52         return NULL;
  53     f->len++;
  54     f->val = fp;
  55     fp += f->len - 1;
  56     return fp;
  57 }
  58
  59 struct s2i {
  60     const char *s;
  61     int val;
  62 };
  63
  64 #define L(X) { #X, LOG_ ## X }
  65
  66 static struct s2i syslogvals[] = {
  67     L(EMERG),
  68     L(ALERT),
  69     L(CRIT),
  70     L(ERR),
  71     L(WARNING),
  72     L(NOTICE),
  73     L(INFO),
  74     L(DEBUG),
  75
  76     L(AUTH),
  77 #ifdef LOG_AUTHPRIV
  78     L(AUTHPRIV),
  79 #endif
  80 #ifdef LOG_CRON
  81     L(CRON),
  82 #endif
  83     L(DAEMON),
  84 #ifdef LOG_FTP
  85     L(FTP),
  86 #endif
  87     L(KERN),
  88     L(LPR),
  89     L(MAIL),
  90 #ifdef LOG_NEWS
  91     L(NEWS),
  92 #endif
  93     L(SYSLOG),
  94     L(USER),
  95 #ifdef LOG_UUCP
  96     L(UUCP),
  97 #endif
  98     L(LOCAL0),
  99     L(LOCAL1),
 100     L(LOCAL2),
 101     L(LOCAL3),
 102     L(LOCAL4),
 103     L(LOCAL5),
 104     L(LOCAL6),
 105     L(LOCAL7),
 106     { NULL, -1 }
 107 };
 108
 109 static int
 110 find_value(const char *s, struct s2i *table)
 111 {
 112     while(table->s && strcasecmp(table->s, s))
 113         table++;
 114     return table->val;
 115 }
 116
 117 krb5_error_code KRB5_LIB_FUNCTION
 118 krb5_initlog(krb5_context context,
 119              const char *program,
 120              krb5_log_facility **fac)
 121 {
 122     krb5_log_facility *f = calloc(1, sizeof(*f));
 123     if(f == NULL) {
 124         krb5_set_error_message(context, ENOMEM, "malloc: out of memory");
 125         return ENOMEM;
 126     }
 127     f->program = strdup(program);
 128     if(f->program == NULL){
 129         free(f);
 130         krb5_set_error_message(context, ENOMEM, "malloc: out of memory");
 131         return ENOMEM;
 132     }
 133     *fac = f;
 134     return 0;
 135 }
 136
 137 krb5_error_code KRB5_LIB_FUNCTION
 138 krb5_addlog_func(krb5_context context,
 139                  krb5_log_facility *fac,
 140                  int min,
 141                  int max,
 142                  krb5_log_log_func_t log_func,
 143                  krb5_log_close_func_t close_func,
 144                  void *data)
 145 {
 146     struct facility *fp = log_realloc(fac);
 147     if(fp == NULL) {
 148         krb5_set_error_message(context, ENOMEM, "malloc: out of memory");
 149         return ENOMEM;
 150     }
 151     fp->min = min;
 152     fp->max = max;
 153     fp->log_func = log_func;
 154     fp->close_func = close_func;
 155     fp->data = data;
 156     return 0;
 157 }
 158
 159
 160 struct _heimdal_syslog_data{
 161     int priority;
 162 };
 163
 164 static void
 165 log_syslog(const char *timestr,
 166            const char *msg,
 167            void *data)
 168
 169 {
 170     struct _heimdal_syslog_data *s = data;
 171     syslog(s->priority, "%s", msg);
 172 }
 173
 174 static void
 175 close_syslog(void *data)
 176 {
 177     free(data);
 178     closelog();
 179 }
 180
 181 static krb5_error_code
 182 open_syslog(krb5_context context,
 183             krb5_log_facility *facility, int min, int max,
 184             const char *sev, const char *fac)
 185 {
 186     struct _heimdal_syslog_data *sd = malloc(sizeof(*sd));
 187     int i;
 188
 189     if(sd == NULL) {
 190         krb5_set_error_message(context, ENOMEM, "malloc: out of memory");
 191         return ENOMEM;
 192     }
 193     i = find_value(sev, syslogvals);
 194     if(i == -1)
 195         i = LOG_ERR;
 196     sd->priority = i;
 197     i = find_value(fac, syslogvals);
 198     if(i == -1)
 199         i = LOG_AUTH;
 200     sd->priority |= i;
 201     roken_openlog(facility->program, LOG_PID | LOG_NDELAY, i);
 202     return krb5_addlog_func(context, facility, min, max,
 203                             log_syslog, close_syslog, sd);
 204 }
 205
 206 struct file_data{
 207     const char *filename;
 208     const char *mode;
 209     FILE *fd;
 210     int keep_open;
 211 };
 212
 213 static void
 214 log_file(const char *timestr,
 215          const char *msg,
 216          void *data)
 217 {
 218     struct file_data *f = data;
 219     if(f->keep_open == 0)
 220         f->fd = fopen(f->filename, f->mode);
 221     if(f->fd == NULL)
 222         return;
 223     fprintf(f->fd, "%s %s\n", timestr, msg);
 224     if(f->keep_open == 0) {
 225         fclose(f->fd);
 226         f->fd = NULL;
 227     }
 228 }
 229
 230 static void
 231 close_file(void *data)
 232 {
 233     struct file_data *f = data;
 234     if(f->keep_open && f->filename)
 235         fclose(f->fd);
 236     free(data);
 237 }
 238
 239 static krb5_error_code
 240 open_file(krb5_context context, krb5_log_facility *fac, int min, int max,
 241           const char *filename, const char *mode, FILE *f, int keep_open)
 242 {
 243     struct file_data *fd = malloc(sizeof(*fd));
 244     if(fd == NULL) {
 245         krb5_set_error_message(context, ENOMEM, "malloc: out of memory");
 246         return ENOMEM;
 247     }
 248     fd->filename = filename;
 249     fd->mode = mode;
 250     fd->fd = f;
 251     fd->keep_open = keep_open;
 252
 253     return krb5_addlog_func(context, fac, min, max, log_file, close_file, fd);
 254 }
 255
 256
 257
 258 krb5_error_code KRB5_LIB_FUNCTION
 259 krb5_addlog_dest(krb5_context context, krb5_log_facility *f, const char *orig)
 260 {
 261     krb5_error_code ret = 0;
 262     int min = 0, max = -1, n;
 263     char c;
 264     const char *p = orig;
 265
 266     n = sscanf(p, "%d%c%d/", &min, &c, &max);
 267     if(n == 2){
 268         if(c == '/') {
 269             if(min < 0){
 270                 max = -min;
 271                 min = 0;
 272             }else{
 273                 max = min;
 274             }
 275         }
 276     }
 277     if(n){
 278         p = strchr(p, '/');
 279         if(p == NULL) {
 280             krb5_set_error_message(context, HEIM_ERR_LOG_PARSE,
 281                                    "failed to parse \"%s\"", orig);
 282             return HEIM_ERR_LOG_PARSE;
 283         }
 284         p++;
 285     }
 286     if(strcmp(p, "STDERR") == 0){
 287         ret = open_file(context, f, min, max, NULL, NULL, stderr, 1);
 288     }else if(strcmp(p, "CONSOLE") == 0){
 289         ret = open_file(context, f, min, max, "/dev/console", "w", NULL, 0);
 290     }else if(strncmp(p, "FILE", 4) == 0 && (p[4] == ':' || p[4] == '=')){
 291         char *fn;
 292         FILE *file = NULL;
 293         int keep_open = 0;
 294         fn = strdup(p + 5);
 295         if(fn == NULL) {
 296             krb5_set_error_message(context, ENOMEM, "malloc: out of memory");
 297             return ENOMEM;
 298         }
 299         if(p[4] == '='){
 300             int i = open(fn, O_WRONLY | O_CREAT |
 301                          O_TRUNC | O_APPEND, 0666);
 302             if(i < 0) {
 303                 ret = errno;
 304                 krb5_set_error_message(context, ret, "open(%s): %s", fn,
 305                                        strerror(ret));
 306                 free(fn);
 307                 return ret;
 308             }
 309             rk_cloexec(i);
 310             file = fdopen(i, "a");
 311             if(file == NULL){
 312                 ret = errno;
 313                 close(i);
 314                 krb5_set_error_message(context, ret, "fdopen(%s): %s", fn,
 315                                        strerror(ret));
 316                 free(fn);
 317                 return ret;
 318             }
 319             keep_open = 1;
 320         }
 321         ret = open_file(context, f, min, max, fn, "a", file, keep_open);
 322     }else if(strncmp(p, "DEVICE", 6) == 0 && (p[6] == ':' || p[6] == '=')){
 323         ret = open_file(context, f, min, max, strdup(p + 7), "w", NULL, 0);
 324     }else if(strncmp(p, "SYSLOG", 6) == 0 && (p[6] == '\0' || p[6] == ':')){
 325         char severity[128] = "";
 326         char facility[128] = "";
 327         p += 6;
 328         if(*p != '\0')
 329             p++;
 330         if(strsep_copy(&p, ":", severity, sizeof(severity)) != -1)
 331             strsep_copy(&p, ":", facility, sizeof(facility));
 332         if(*severity == '\0')
 333             strlcpy(severity, "ERR", sizeof(severity));
 334         if(*facility == '\0')
 335             strlcpy(facility, "AUTH", sizeof(facility));
 336         ret = open_syslog(context, f, min, max, severity, facility);
 337     }else{
 338         ret = HEIM_ERR_LOG_PARSE; /* XXX */
 339         krb5_set_error_message (context, ret, "unknown log type: %s", p);
 340     }
 341     return ret;
 342 }
 343
 344
 345 krb5_error_code KRB5_LIB_FUNCTION
 346 krb5_openlog(krb5_context context,
 347              const char *program,
 348              krb5_log_facility **fac)
 349 {
 350     krb5_error_code ret;
 351     char **p, **q;
 352
 353     ret = krb5_initlog(context, program, fac);
 354     if(ret)
 355         return ret;
 356
 357     p = krb5_config_get_strings(context, NULL, "logging", program, NULL);
 358     if(p == NULL)
 359         p = krb5_config_get_strings(context, NULL, "logging", "default", NULL);
 360     if(p){
 361         for(q = p; *q && ret == 0; q++)
 362             ret = krb5_addlog_dest(context, *fac, *q);
 363         krb5_config_free_strings(p);
 364     }else
 365         ret = krb5_addlog_dest(context, *fac, "SYSLOG");
 366     return ret;
 367 }
 368
 369 krb5_error_code KRB5_LIB_FUNCTION
 370 krb5_closelog(krb5_context context,
 371               krb5_log_facility *fac)
 372 {
 373     int i;
 374     for(i = 0; i < fac->len; i++)
 375         (*fac->val[i].close_func)(fac->val[i].data);
 376     free(fac->val);
 377     free(fac->program);
 378     fac->val = NULL;
 379     fac->len = 0;
 380     fac->program = NULL;
 381     free(fac);
 382     return 0;
 383 }
 384
 385 #undef __attribute__
 386 #define __attribute__(X)
 387
 388 krb5_error_code KRB5_LIB_FUNCTION
 389 krb5_vlog_msg(krb5_context context,
 390               krb5_log_facility *fac,
 391               char **reply,
 392               int level,
 393               const char *fmt,
 394               va_list ap)
 395      __attribute__((format (printf, 5, 0)))
 396 {
 397
 398     char *msg = NULL;
 399     const char *actual = NULL;
 400     char buf[64];
 401     time_t t = 0;
 402     int i;
 403
 404     for(i = 0; fac && i < fac->len; i++)
 405         if(fac->val[i].min <= level &&
 406            (fac->val[i].max < 0 || fac->val[i].max >= level)) {
 407             if(t == 0) {
 408                 t = time(NULL);
 409                 krb5_format_time(context, t, buf, sizeof(buf), TRUE);
 410             }
 411             if(actual == NULL) {
 412                 vasprintf(&msg, fmt, ap);
 413                 if(msg == NULL)
 414                     actual = fmt;
 415                 else
 416                     actual = msg;
 417             }
 418             (*fac->val[i].log_func)(buf, actual, fac->val[i].data);
 419         }
 420     if(reply == NULL)
 421         free(msg);
 422     else
 423         *reply = msg;
 424     return 0;
 425 }
 426
 427 krb5_error_code KRB5_LIB_FUNCTION
 428 krb5_vlog(krb5_context context,
 429           krb5_log_facility *fac,
 430           int level,
 431           const char *fmt,
 432           va_list ap)
 433      __attribute__((format (printf, 4, 0)))
 434 {
 435     return krb5_vlog_msg(context, fac, NULL, level, fmt, ap);
 436 }
 437
 438 krb5_error_code KRB5_LIB_FUNCTION
 439 krb5_log_msg(krb5_context context,
 440              krb5_log_facility *fac,
 441              int level,
 442              char **reply,
 443              const char *fmt,
 444              ...)
 445      __attribute__((format (printf, 5, 6)))
 446 {
 447     va_list ap;
 448     krb5_error_code ret;
 449
 450     va_start(ap, fmt);
 451     ret = krb5_vlog_msg(context, fac, reply, level, fmt, ap);
 452     va_end(ap);
 453     return ret;
 454 }
 455
 456
 457 krb5_error_code KRB5_LIB_FUNCTION
 458 krb5_log(krb5_context context,
 459          krb5_log_facility *fac,
 460          int level,
 461          const char *fmt,
 462          ...)
 463      __attribute__((format (printf, 4, 5)))
 464 {
 465     va_list ap;
 466     krb5_error_code ret;
 467
 468     va_start(ap, fmt);
 469     ret = krb5_vlog(context, fac, level, fmt, ap);
 470     va_end(ap);
 471     return ret;
 472 }
 473