doc: Added manpage for htextauth.
[ashd.git] / 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
29 #ifdef HAVE_CONFIG_H
30 #include <config.h>
31 #endif
32 #include <utils.h>
33 #include <log.h>
34 #include <req.h>
35 #include <proc.h>
36
37 #define DEFFORMAT "%{%Y-%m-%d %H:%M:%S}t %m %u %A \"%G\""
38
39 static int ch;
40 static char *outname = NULL;
41 static FILE *out;
42 static int flush = 1;
43 static char *format;
44 static struct timeval now;
45 static volatile int reopen = 0;
46
47 static void qputs(char *s, FILE *o)
48 {
49     for(; *s; s++) {
50         if(*s == '\"') {
51             fputs("\\\"", o);
52         } else if(*s == '\\') {
53             fputs("\\\\", o);
54         } else if(*s == '\n') {
55             fputs("\\n", o);
56         } else if(*s == '\t') {
57             fputs("\\t", o);
58         } else if((*s < 32) || (*s >= 128)) {
59             fprintf(o, "\\x%02x", *s);
60         } else {
61             fputc(*s, o);
62         }
63     }
64 }
65
66 static void logitem(struct hthead *req, char o, char *d)
67 {
68     char *h, *p;
69     char buf[1024];
70     
71     switch(o) {
72     case '%':
73         putc('%', out);
74         break;
75     case 'h':
76         if((h = getheader(req, d)) == NULL) {
77             putc('-', out);
78         } else {
79             qputs(h, out);
80         }
81         break;
82     case 'u':
83         qputs(req->url, out);
84         break;
85     case 'U':
86         strcpy(buf, req->url);
87         if((p = strchr(buf, '?')) != NULL)
88             *p = 0;
89         qputs(buf, out);
90         break;
91     case 'm':
92         qputs(req->method, out);
93         break;
94     case 'r':
95         qputs(req->rest, out);
96         break;
97     case 'v':
98         qputs(req->ver, out);
99         break;
100     case 't':
101         if(!*d)
102             d = "%a, %d %b %Y %H:%M:%S %z";
103         strftime(buf, sizeof(buf), d, localtime(&now.tv_sec));
104         qputs(buf, out);
105         break;
106     case 'T':
107         if(!*d)
108             d = "%a, %d %b %Y %H:%M:%S %z";
109         strftime(buf, sizeof(buf), d, gmtime(&now.tv_sec));
110         qputs(buf, out);
111         break;
112     case 's':
113         fprintf(out, "%06i", (int)now.tv_usec);
114         break;
115     case 'A':
116         logitem(req, 'h', "X-Ash-Address");
117         break;
118     case 'H':
119         logitem(req, 'h', "Host");
120         break;
121     case 'R':
122         logitem(req, 'h', "Referer");
123         break;
124     case 'G':
125         logitem(req, 'h', "User-Agent");
126         break;
127     }
128 }
129
130 static void logreq(struct hthead *req)
131 {
132     char *p, *p2;
133     char d[strlen(format)];
134     char o;
135     
136     p = format;
137     while(*p) {
138         if(*p == '%') {
139             p++;
140             if(*p == '{') {
141                 p++;
142                 if((p2 = strchr(p, '}')) == NULL)
143                     continue;
144                 memcpy(d, p, p2 - p);
145                 d[p2 - p] = 0;
146                 p = p2 + 1;
147             } else {
148                 d[0] = 0;
149             }
150             o = *p++;
151             if(o == 0)
152                 break;
153             logitem(req, o, d);
154         } else {
155             fputc(*p++, out);
156         }
157     }
158     fputc('\n', out);
159     if(flush)
160         fflush(out);
161 }
162
163 static void serve(struct hthead *req, int fd)
164 {
165     gettimeofday(&now, NULL);
166     if(sendreq(ch, req, fd)) {
167         flog(LOG_ERR, "accesslog: could not pass request to child: %s", strerror(errno));
168         exit(1);
169     }
170     logreq(req);
171 }
172
173 static void sighandler(int sig)
174 {
175     if(sig == SIGHUP)
176         reopen = 1;
177 }
178
179 static void reopenlog(void)
180 {
181     FILE *new;
182     
183     if(outname == NULL) {
184         flog(LOG_WARNING, "accesslog: received SIGHUP but logging to stdout, so ignoring");
185         return;
186     }
187     if((new = fopen(outname, "a")) == NULL) {
188         flog(LOG_WARNING, "accesslog: could not reopen log file `%s' on SIGHUP: %s", outname, strerror(errno));
189         return;
190     }
191     fclose(out);
192     out = new;
193 }
194
195 static void usage(FILE *out)
196 {
197     fprintf(out, "usage: accesslog [-hFa] [-f FORMAT] OUTFILE CHILD [ARGS...]\n");
198 }
199
200 int main(int argc, char **argv)
201 {
202     int c, ret;
203     struct hthead *req;
204     int fd;
205     struct pollfd pfd[2];
206     
207     while((c = getopt(argc, argv, "+hFaf:")) >= 0) {
208         switch(c) {
209         case 'h':
210             usage(stdout);
211             exit(0);
212         case 'F':
213             flush = 0;
214             break;
215         case 'f':
216             format = optarg;
217             break;
218         case 'a':
219             format = "%A - - [%{%d/%b/%Y:%H:%M:%S %z}t] \"%m %u %v\" - - \"%R\" \"%G\"";
220             break;
221         default:
222             usage(stderr);
223             exit(1);
224         }
225     }
226     if(argc - optind < 2) {
227         usage(stderr);
228         exit(1);
229     }
230     if(format == NULL)
231         format = DEFFORMAT;
232     if(!strcmp(argv[optind], "-"))
233         outname = NULL;
234     else
235         outname = argv[optind];
236     if(outname == NULL) {
237         out = stdout;
238     } else {
239         if((out = fopen(argv[optind], "a")) == NULL) {
240             flog(LOG_ERR, "accesslog: could not open %s for logging: %s", argv[optind], strerror(errno));
241             exit(1);
242         }
243     }
244     if((ch = stdmkchild(argv + optind + 1, NULL, NULL)) < 0) {
245         flog(LOG_ERR, "accesslog: could not fork child: %s", strerror(errno));
246         exit(1);
247     }
248     signal(SIGHUP, sighandler);
249     while(1) {
250         if(reopen) {
251             reopenlog();
252             reopen = 0;
253         }
254         memset(pfd, 0, sizeof(pfd));
255         pfd[0].fd = 0;
256         pfd[0].events = POLLIN;
257         pfd[1].fd = ch;
258         pfd[1].events = POLLHUP;
259         if((ret = poll(pfd, 2, -1)) < 0) {
260             if(errno != EINTR) {
261                 flog(LOG_ERR, "accesslog: error in poll: %s", strerror(errno));
262                 exit(1);
263             }
264         }
265         if(pfd[0].revents) {
266             if((fd = recvreq(0, &req)) < 0) {
267                 if(errno == 0)
268                     break;
269                 flog(LOG_ERR, "accesslog: error in recvreq: %s", strerror(errno));
270                 exit(1);
271             }
272             serve(req, fd);
273             freehthead(req);
274             close(fd);
275         }
276         if(pfd[1].revents & POLLHUP)
277             break;
278     }
279     return(0);
280 }