src/accesslog.c

   1 /*
   2     ashd - A Sane HTTP Daemon
   3     Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>
   4
   5     This program is free software: you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation, either version 3 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #include <stdlib.h>
  20 #include <stdio.h>
  21 #include <unistd.h>
  22 #include <string.h>
  23 #include <errno.h>
  24 #include <sys/poll.h>
  25 #include <time.h>
  26 #include <sys/time.h>
  27 #include <signal.h>
  28 #include <fcntl.h>
  29 #include <sys/stat.h>
  30
  31 #ifdef HAVE_CONFIG_H
  32 #include <config.h>
  33 #endif
  34 #include <utils.h>
  35 #include <log.h>
  36 #include <req.h>
  37 #include <proc.h>
  38
  39 #define DEFFORMAT "%{%Y-%m-%d %H:%M:%S}t %m %u %A \"%G\""
  40
  41 static int ch;
  42 static char *outname = NULL;
  43 static FILE *out;
  44 static int flush = 1, locklog = 1;
  45 static char *format;
  46 static struct timeval now;
  47 static volatile int reopen = 0;
  48
  49 static void qputs(char *sp, FILE *o)
  50 {
  51     unsigned char *s = (unsigned char *)sp;
  52
  53     for(; *s; s++) {
  54         if(*s == '\"') {
  55             fputs("\\\"", o);
  56         } else if(*s == '\\') {
  57             fputs("\\\\", o);
  58         } else if(*s == '\n') {
  59             fputs("\\n", o);
  60         } else if(*s == '\t') {
  61             fputs("\\t", o);
  62         } else if((*s < 32) || (*s >= 128)) {
  63             fprintf(o, "\\x%02x", (int)*s);
  64         } else {
  65             fputc(*s, o);
  66         }
  67     }
  68 }
  69
  70 static void logitem(struct hthead *req, char o, char *d)
  71 {
  72     char *h, *p;
  73     char buf[1024];
  74
  75     switch(o) {
  76     case '%':
  77         putc('%', out);
  78         break;
  79     case 'h':
  80         if((h = getheader(req, d)) == NULL) {
  81             putc('-', out);
  82         } else {
  83             qputs(h, out);
  84         }
  85         break;
  86     case 'u':
  87         qputs(req->url, out);
  88         break;
  89     case 'U':
  90         strcpy(buf, req->url);
  91         if((p = strchr(buf, '?')) != NULL)
  92             *p = 0;
  93         qputs(buf, out);
  94         break;
  95     case 'm':
  96         qputs(req->method, out);
  97         break;
  98     case 'r':
  99         qputs(req->rest, out);
 100         break;
 101     case 'v':
 102         qputs(req->ver, out);
 103         break;
 104     case 't':
 105         if(!*d)
 106             d = "%a, %d %b %Y %H:%M:%S %z";
 107         strftime(buf, sizeof(buf), d, localtime(&now.tv_sec));
 108         qputs(buf, out);
 109         break;
 110     case 'T':
 111         if(!*d)
 112             d = "%a, %d %b %Y %H:%M:%S %z";
 113         strftime(buf, sizeof(buf), d, gmtime(&now.tv_sec));
 114         qputs(buf, out);
 115         break;
 116     case 's':
 117         fprintf(out, "%06i", (int)now.tv_usec);
 118         break;
 119     case 'A':
 120         logitem(req, 'h', "X-Ash-Address");
 121         break;
 122     case 'H':
 123         logitem(req, 'h', "Host");
 124         break;
 125     case 'R':
 126         logitem(req, 'h', "Referer");
 127         break;
 128     case 'G':
 129         logitem(req, 'h', "User-Agent");
 130         break;
 131     }
 132 }
 133
 134 static void logreq(struct hthead *req)
 135 {
 136     char *p, *p2;
 137     char d[strlen(format)];
 138     char o;
 139
 140     p = format;
 141     while(*p) {
 142         if(*p == '%') {
 143             p++;
 144             if(*p == '{') {
 145                 p++;
 146                 if((p2 = strchr(p, '}')) == NULL)
 147                     continue;
 148                 memcpy(d, p, p2 - p);
 149                 d[p2 - p] = 0;
 150                 p = p2 + 1;
 151             } else {
 152                 d[0] = 0;
 153             }
 154             o = *p++;
 155             if(o == 0)
 156                 break;
 157             logitem(req, o, d);
 158         } else {
 159             fputc(*p++, out);
 160         }
 161     }
 162     fputc('\n', out);
 163     if(flush)
 164         fflush(out);
 165 }
 166
 167 static void serve(struct hthead *req, int fd)
 168 {
 169     gettimeofday(&now, NULL);
 170     if(sendreq(ch, req, fd)) {
 171         flog(LOG_ERR, "accesslog: could not pass request to child: %s", strerror(errno));
 172         exit(1);
 173     }
 174     logreq(req);
 175 }
 176
 177 static void sighandler(int sig)
 178 {
 179     if(sig == SIGHUP)
 180         reopen = 1;
 181 }
 182
 183 static int lockfile(FILE *file)
 184 {
 185     struct flock ld;
 186
 187     memset(&ld, 0, sizeof(ld));
 188     ld.l_type = F_WRLCK;
 189     ld.l_whence = SEEK_SET;
 190     ld.l_start = 0;
 191     ld.l_len = 0;
 192     return(fcntl(fileno(file), F_SETLK, &ld));
 193 }
 194
 195 static void fetchpid(char *filename)
 196 {
 197     int fd, ret;
 198     struct flock ld;
 199
 200     if((fd = open(filename, O_WRONLY)) < 0) {
 201         fprintf(stderr, "accesslog: %s: %s\n", filename, strerror(errno));
 202         exit(1);
 203     }
 204     memset(&ld, 0, sizeof(ld));
 205     ld.l_type = F_WRLCK;
 206     ld.l_whence = SEEK_SET;
 207     ld.l_start = 0;
 208     ld.l_len = 0;
 209     ret = fcntl(fd, F_GETLK, &ld);
 210     close(fd);
 211     if(ret) {
 212         fprintf(stderr, "accesslog: %s: %s\n", filename, strerror(errno));
 213         exit(1);
 214     }
 215     if(ld.l_type == F_UNLCK) {
 216         fprintf(stderr, "accesslog: %s: not locked\n", filename);
 217         exit(1);
 218     }
 219     printf("%i\n", (int)ld.l_pid);
 220 }
 221
 222 static void reopenlog(void)
 223 {
 224     FILE *new;
 225     struct stat olds, news;
 226
 227     if(outname == NULL) {
 228         flog(LOG_WARNING, "accesslog: received SIGHUP but logging to stdout, so ignoring");
 229         return;
 230     }
 231     if(locklog) {
 232         if(fstat(fileno(out), &olds)) {
 233             flog(LOG_ERR, "accesslog: could not stat current logfile(?!): %s", strerror(errno));
 234             return;
 235         }
 236         if(!stat(outname, &news)) {
 237             if((olds.st_dev == news.st_dev) && (olds.st_ino == news.st_ino)) {
 238                 /*
 239                  * This needs to be ignored, because if the same logfile
 240                  * is opened and then closed, the lock is lost. To quote
 241                  * the Linux fcntl(2) manpage: "This is bad." No kidding.
 242                  *
 243                  * Technically, there is a race condition here when the
 244                  * file has been stat'ed but not yet opened, where the old
 245                  * log file, having been previously renamed, changes name
 246                  * back to the name accesslog knows and is thus reopened
 247                  * regardlessly, but I think that might fit under the
 248                  * idiom "pathological case". It should, at least, not be
 249                  * a security problem.
 250                  */
 251                 flog(LOG_INFO, "accesslog: received SIGHUP, but logfile has not changed, so ignoring");
 252                 return;
 253             }
 254         }
 255     }
 256     if((new = fopen(outname, "a")) == NULL) {
 257         flog(LOG_WARNING, "accesslog: could not reopen log file `%s' on SIGHUP: %s", outname, strerror(errno));
 258         return;
 259     }
 260     fcntl(fileno(new), F_SETFD, FD_CLOEXEC);
 261     if(locklog) {
 262         if(lockfile(new)) {
 263             if((errno == EAGAIN) || (errno == EACCES)) {
 264                 flog(LOG_ERR, "accesslog: logfile is already locked; reverting to current log", strerror(errno));
 265                 fclose(new);
 266                 return;
 267             } else {
 268                 flog(LOG_WARNING, "accesslog: could not lock logfile, so no lock will be held: %s", strerror(errno));
 269             }
 270         }
 271     }
 272     fclose(out);
 273     out = new;
 274 }
 275
 276 static void usage(FILE *out)
 277 {
 278     fprintf(out, "usage: accesslog [-hFaL] [-f FORMAT] [-p PIDFILE] OUTFILE CHILD [ARGS...]\n");
 279     fprintf(out, "       accesslog -P LOGFILE\n");
 280 }
 281
 282 int main(int argc, char **argv)
 283 {
 284     int c, ret;
 285     struct hthead *req;
 286     int fd;
 287     struct pollfd pfd[2];
 288     char *pidfile;
 289     FILE *pidout;
 290
 291     pidfile = NULL;
 292     while((c = getopt(argc, argv, "+hFaLf:p:P:")) >= 0) {
 293         switch(c) {
 294         case 'h':
 295             usage(stdout);
 296             exit(0);
 297         case 'F':
 298             flush = 0;
 299             break;
 300         case 'L':
 301             locklog = 0;
 302             break;
 303         case 'f':
 304             format = optarg;
 305             break;
 306         case 'P':
 307             fetchpid(optarg);
 308             exit(0);
 309         case 'p':
 310             pidfile = optarg;
 311             break;
 312         case 'a':
 313             format = "%A - - [%{%d/%b/%Y:%H:%M:%S %z}t] \"%m %u %v\" - - \"%R\" \"%G\"";
 314             break;
 315         default:
 316             usage(stderr);
 317             exit(1);
 318         }
 319     }
 320     if(argc - optind < 2) {
 321         usage(stderr);
 322         exit(1);
 323     }
 324     if(format == NULL)
 325         format = DEFFORMAT;
 326     if(!strcmp(argv[optind], "-"))
 327         outname = NULL;
 328     else
 329         outname = argv[optind];
 330     if(outname == NULL) {
 331         out = stdout;
 332         locklog = 0;
 333     } else {
 334         if((out = fopen(argv[optind], "a")) == NULL) {
 335             flog(LOG_ERR, "accesslog: could not open %s for logging: %s", argv[optind], strerror(errno));
 336             exit(1);
 337         }
 338         fcntl(fileno(out), F_SETFD, FD_CLOEXEC);
 339     }
 340     if(locklog) {
 341         if(lockfile(out)) {
 342             if((errno == EAGAIN) || (errno == EACCES)) {
 343                 flog(LOG_ERR, "accesslog: logfile is already locked", strerror(errno));
 344                 exit(1);
 345             } else {
 346                 flog(LOG_WARNING, "accesslog: could not lock logfile: %s", strerror(errno));
 347             }
 348         }
 349     }
 350     if((ch = stdmkchild(argv + optind + 1, NULL, NULL)) < 0) {
 351         flog(LOG_ERR, "accesslog: could not fork child: %s", strerror(errno));
 352         exit(1);
 353     }
 354     signal(SIGHUP, sighandler);
 355     if(pidfile) {
 356         if(!strcmp(pidfile, "-")) {
 357             if(!outname) {
 358                 flog(LOG_ERR, "accesslog: cannot derive PID file name without an output file");
 359                 exit(1);
 360             }
 361             pidfile = sprintf2("%s.pid", outname);
 362         }
 363         if((pidout = fopen(pidfile, "w")) == NULL) {
 364             flog(LOG_ERR, "accesslog: could not open PID file %s for writing: %s", pidfile);
 365             exit(1);
 366         }
 367         fprintf(pidout, "%i\n", (int)getpid());
 368         fclose(pidout);
 369     }
 370     while(1) {
 371         if(reopen) {
 372             reopenlog();
 373             reopen = 0;
 374         }
 375         memset(pfd, 0, sizeof(pfd));
 376         pfd[0].fd = 0;
 377         pfd[0].events = POLLIN;
 378         pfd[1].fd = ch;
 379         pfd[1].events = POLLHUP;
 380         if((ret = poll(pfd, 2, -1)) < 0) {
 381             if(errno != EINTR) {
 382                 flog(LOG_ERR, "accesslog: error in poll: %s", strerror(errno));
 383                 exit(1);
 384             }
 385         }
 386         if(pfd[0].revents) {
 387             if((fd = recvreq(0, &req)) < 0) {
 388                 if(errno == 0)
 389                     break;
 390                 flog(LOG_ERR, "accesslog: error in recvreq: %s", strerror(errno));
 391                 exit(1);
 392             }
 393             serve(req, fd);
 394             freehthead(req);
 395             close(fd);
 396         }
 397         if(pfd[1].revents & POLLHUP)
 398             break;
 399     }
 400     fclose(out);
 401     if(pidfile != NULL)
 402         unlink(pidfile);
 403     return(0);
 404 }