Home | History | Annotate | Download | only in racoon
      1 /*	$NetBSD: logger.c,v 1.4 2006/09/09 16:22:09 manu Exp $	*/
      2 
      3 /*	$KAME: logger.c,v 1.9 2002/09/03 14:37:03 itojun Exp $	*/
      4 
      5 /*
      6  * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
      7  * All rights reserved.
      8  *
      9  * Redistribution and use in source and binary forms, with or without
     10  * modification, are permitted provided that the following conditions
     11  * are met:
     12  * 1. Redistributions of source code must retain the above copyright
     13  *    notice, this list of conditions and the following disclaimer.
     14  * 2. Redistributions in binary form must reproduce the above copyright
     15  *    notice, this list of conditions and the following disclaimer in the
     16  *    documentation and/or other materials provided with the distribution.
     17  * 3. Neither the name of the project 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 PROJECT 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 PROJECT 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 "config.h"
     35 
     36 #include <sys/types.h>
     37 #include <sys/param.h>
     38 
     39 #include <stdlib.h>
     40 #include <stdio.h>
     41 #include <string.h>
     42 #include <errno.h>
     43 #ifdef HAVE_STDARG_H
     44 #include <stdarg.h>
     45 #else
     46 #include <varargs.h>
     47 #endif
     48 #if TIME_WITH_SYS_TIME
     49 # include <sys/time.h>
     50 # include <time.h>
     51 #else
     52 # if HAVE_SYS_TIME_H
     53 #  include <sys/time.h>
     54 # else
     55 #  include <time.h>
     56 # endif
     57 #endif
     58 
     59 #include "logger.h"
     60 #include "var.h"
     61 #include "gcmalloc.h"
     62 
     63 struct log *
     64 log_open(siz, fname)
     65 	size_t siz;
     66 	char *fname;
     67 {
     68 	struct log *p;
     69 
     70 	p = (struct log *)racoon_malloc(sizeof(*p));
     71 	if (p == NULL)
     72 		return NULL;
     73 	memset(p, 0, sizeof(*p));
     74 
     75 	p->buf = (char **)racoon_malloc(sizeof(char *) * siz);
     76 	if (p->buf == NULL) {
     77 		racoon_free(p);
     78 		return NULL;
     79 	}
     80 	memset(p->buf, 0, sizeof(char *) * siz);
     81 
     82 	p->tbuf = (time_t *)racoon_malloc(sizeof(time_t *) * siz);
     83 	if (p->tbuf == NULL) {
     84 		racoon_free(p->buf);
     85 		racoon_free(p);
     86 		return NULL;
     87 	}
     88 	memset(p->tbuf, 0, sizeof(time_t *) * siz);
     89 
     90 	p->siz = siz;
     91 	if (fname)
     92 		p->fname = racoon_strdup(fname);
     93 
     94 	return p;
     95 }
     96 
     97 /*
     98  * append string to ring buffer.
     99  * string must be \n-terminated (since we add timestamps).
    100  * even if not, we'll add \n to avoid formatting mistake (see log_close()).
    101  */
    102 void
    103 log_add(p, str)
    104 	struct log *p;
    105 	char *str;
    106 {
    107 	/* syslog if p->fname == NULL? */
    108 	if (p->buf[p->head])
    109 		racoon_free(p->buf[p->head]);
    110 	p->buf[p->head] = racoon_strdup(str);
    111 	p->tbuf[p->head] = time(NULL);
    112 	p->head++;
    113 	p->head %= p->siz;
    114 }
    115 
    116 /*
    117  * write out string to the log file, as is.
    118  * \n-termination is up to the caller.  if you don't add \n, the file
    119  * format may be broken.
    120  */
    121 int
    122 log_print(p, str)
    123 	struct log *p;
    124 	char *str;
    125 {
    126 	FILE *fp;
    127 
    128 	if (p->fname == NULL)
    129 		return -1;	/*XXX syslog?*/
    130 	fp = fopen(p->fname, "a");
    131 	if (fp == NULL)
    132 		return -1;
    133 	fprintf(fp, "%s", str);
    134 	fclose(fp);
    135 
    136 	return 0;
    137 }
    138 
    139 int
    140 log_vprint(struct log *p, const char *fmt, ...)
    141 {
    142 	va_list ap;
    143 
    144 	FILE *fp;
    145 
    146 	if (p->fname == NULL)
    147 		return -1;	/*XXX syslog?*/
    148 	fp = fopen(p->fname, "a");
    149 	if (fp == NULL)
    150 		return -1;
    151 	va_start(ap, fmt);
    152 	vfprintf(fp, fmt, ap);
    153 	va_end(ap);
    154 
    155 	fclose(fp);
    156 
    157 	return 0;
    158 }
    159 
    160 int
    161 log_vaprint(struct log *p, const char *fmt, va_list ap)
    162 {
    163 	FILE *fp;
    164 
    165 	if (p->fname == NULL)
    166 		return -1;	/*XXX syslog?*/
    167 	fp = fopen(p->fname, "a");
    168 	if (fp == NULL)
    169 		return -1;
    170 	vfprintf(fp, fmt, ap);
    171 	fclose(fp);
    172 
    173 	return 0;
    174 }
    175 
    176 /*
    177  * write out content of ring buffer, and reclaim the log structure
    178  */
    179 int
    180 log_close(p)
    181 	struct log *p;
    182 {
    183 	FILE *fp;
    184 	int i, j;
    185 	char ts[256];
    186 	struct tm *tm;
    187 
    188 	if (p->fname == NULL)
    189 		goto nowrite;
    190 	fp = fopen(p->fname, "a");
    191 	if (fp == NULL)
    192 		goto nowrite;
    193 
    194 	for (i = 0; i < p->siz; i++) {
    195 		j = (p->head + i) % p->siz;
    196 		if (p->buf[j]) {
    197 			tm = localtime(&p->tbuf[j]);
    198 			strftime(ts, sizeof(ts), "%B %d %T", tm);
    199 			fprintf(fp, "%s: %s\n", ts, p->buf[j]);
    200 			if (*(p->buf[j] + strlen(p->buf[j]) - 1) != '\n')
    201 				fprintf(fp, "\n");
    202 		}
    203 	}
    204 	fclose(fp);
    205 
    206 nowrite:
    207 	log_free(p);
    208 	return 0;
    209 }
    210 
    211 void
    212 log_free(p)
    213 	struct log *p;
    214 {
    215 	int i;
    216 
    217 	for (i = 0; i < p->siz; i++)
    218 		racoon_free(p->buf[i]);
    219 	racoon_free(p->buf);
    220 	racoon_free(p->tbuf);
    221 	if (p->fname)
    222 		racoon_free(p->fname);
    223 	racoon_free(p);
    224 }
    225 
    226 #ifdef TEST
    227 struct log *l;
    228 
    229 void
    230 vatest(const char *fmt, ...)
    231 {
    232 	va_list ap;
    233 	va_start(ap, fmt);
    234 	log_vaprint(l, fmt, ap);
    235 	va_end(ap);
    236 }
    237 
    238 int
    239 main(argc, argv)
    240 	int argc;
    241 	char **argv;
    242 {
    243 	int i;
    244 
    245 	l = log_open(30, "/tmp/hoge");
    246 	if (l == NULL)
    247 		errx(1, "hoge");
    248 
    249 	for (i = 0; i < 50; i++) {
    250 		log_add(l, "foo");
    251 		log_add(l, "baa");
    252 		log_add(l, "baz");
    253 	}
    254 	log_print(l, "hoge\n");
    255 	log_vprint(l, "hoge %s\n", "this is test");
    256 	vatest("%s %s\n", "this is", "vprint test");
    257 	abort();
    258 	log_free(l);
    259 }
    260 
    261 #endif
    262 
    263