2 ashd - A Sane HTTP Daemon
3 Copyright (C) 2008 Fredrik Tolf <fredrik@dolda2000.com>
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.
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.
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/>.
39 #define DEFFORMAT "%{%Y-%m-%d %H:%M:%S}t %m %u %A \"%G\""
42 static char *outname = NULL;
44 static int flush = 1, locklog = 1;
46 static struct timeval now;
47 static volatile int reopen = 0;
49 static void qputs(char *s, FILE *o)
54 } else if(*s == '\\') {
56 } else if(*s == '\n') {
58 } else if(*s == '\t') {
60 } else if((*s < 32) || (*s >= 128)) {
61 fprintf(o, "\\x%02x", *s);
68 static void logitem(struct hthead *req, char o, char *d)
78 if((h = getheader(req, d)) == NULL) {
88 strcpy(buf, req->url);
89 if((p = strchr(buf, '?')) != NULL)
94 qputs(req->method, out);
97 qputs(req->rest, out);
100 qputs(req->ver, out);
104 d = "%a, %d %b %Y %H:%M:%S %z";
105 strftime(buf, sizeof(buf), d, localtime(&now.tv_sec));
110 d = "%a, %d %b %Y %H:%M:%S %z";
111 strftime(buf, sizeof(buf), d, gmtime(&now.tv_sec));
115 fprintf(out, "%06i", (int)now.tv_usec);
118 logitem(req, 'h', "X-Ash-Address");
121 logitem(req, 'h', "Host");
124 logitem(req, 'h', "Referer");
127 logitem(req, 'h', "User-Agent");
132 static void logreq(struct hthead *req)
135 char d[strlen(format)];
144 if((p2 = strchr(p, '}')) == NULL)
146 memcpy(d, p, p2 - p);
165 static void serve(struct hthead *req, int fd)
167 gettimeofday(&now, NULL);
168 if(sendreq(ch, req, fd)) {
169 flog(LOG_ERR, "accesslog: could not pass request to child: %s", strerror(errno));
175 static void sighandler(int sig)
181 static int lockfile(FILE *file)
185 memset(&ld, 0, sizeof(ld));
187 ld.l_whence = SEEK_SET;
190 return(fcntl(fileno(file), F_SETLK, &ld));
193 static void fetchpid(char *filename)
198 if((fd = open(filename, O_WRONLY)) < 0) {
199 fprintf(stderr, "accesslog: %s: %s\n", filename, strerror(errno));
202 memset(&ld, 0, sizeof(ld));
204 ld.l_whence = SEEK_SET;
207 ret = fcntl(fd, F_GETLK, &ld);
210 fprintf(stderr, "accesslog: %s: %s\n", filename, strerror(errno));
213 if(ld.l_type == F_UNLCK) {
214 fprintf(stderr, "accesslog: %s: not locked\n", filename);
217 printf("%i\n", (int)ld.l_pid);
220 static void reopenlog(void)
223 struct stat olds, news;
225 if(outname == NULL) {
226 flog(LOG_WARNING, "accesslog: received SIGHUP but logging to stdout, so ignoring");
230 if(fstat(fileno(out), &olds)) {
231 flog(LOG_ERR, "accesslog: could not stat current logfile(?!): %s", strerror(errno));
234 if(!stat(outname, &news)) {
235 if((olds.st_dev == news.st_dev) && (olds.st_ino == news.st_ino)) {
237 * This needs to be ignored, because if the same logfile
238 * is opened and then closed, the lock is lost. To quote
239 * the Linux fcntl(2) manpage: "This is bad." No kidding.
241 * Technically, there is a race condition here when the
242 * file has been stat'ed but not yet opened, where the old
243 * log file, having been previously renamed, changes name
244 * back to the name accesslog knows and is thus reopened
245 * regardlessly, but I think that might fit under the
246 * idiom "pathological case". It should, at least, not be
247 * a security problem.
249 flog(LOG_INFO, "accesslog: received SIGHUP, but logfile has not changed, so ignoring");
254 if((new = fopen(outname, "a")) == NULL) {
255 flog(LOG_WARNING, "accesslog: could not reopen log file `%s' on SIGHUP: %s", outname, strerror(errno));
260 if((errno == EAGAIN) || (errno == EACCES)) {
261 flog(LOG_ERR, "accesslog: logfile is already locked; reverting to current log", strerror(errno));
265 flog(LOG_WARNING, "accesslog: could not lock logfile, so no lock will be held: %s", strerror(errno));
273 static void usage(FILE *out)
275 fprintf(out, "usage: accesslog [-hFaL] [-f FORMAT] [-p PIDFILE] OUTFILE CHILD [ARGS...]\n");
276 fprintf(out, " accesslog -P LOGFILE\n");
279 int main(int argc, char **argv)
284 struct pollfd pfd[2];
289 while((c = getopt(argc, argv, "+hFaLf:p:P:")) >= 0) {
310 format = "%A - - [%{%d/%b/%Y:%H:%M:%S %z}t] \"%m %u %v\" - - \"%R\" \"%G\"";
317 if(argc - optind < 2) {
323 if(!strcmp(argv[optind], "-"))
326 outname = argv[optind];
327 if(outname == NULL) {
330 if((out = fopen(argv[optind], "a")) == NULL) {
331 flog(LOG_ERR, "accesslog: could not open %s for logging: %s", argv[optind], strerror(errno));
337 if((errno == EAGAIN) || (errno == EACCES)) {
338 flog(LOG_ERR, "accesslog: logfile is already locked", strerror(errno));
341 flog(LOG_WARNING, "accesslog: could not lock logfile: %s", strerror(errno));
345 if((ch = stdmkchild(argv + optind + 1, NULL, NULL)) < 0) {
346 flog(LOG_ERR, "accesslog: could not fork child: %s", strerror(errno));
349 signal(SIGHUP, sighandler);
351 if(!strcmp(pidfile, "-")) {
353 flog(LOG_ERR, "accesslog: cannot derive PID file name without an output file");
356 pidfile = sprintf2("%s.pid", outname);
358 if((pidout = fopen(pidfile, "w")) == NULL) {
359 flog(LOG_ERR, "accesslog: could not open PID file %s for writing: %s", pidfile);
362 fprintf(pidout, "%i\n", (int)getpid());
370 memset(pfd, 0, sizeof(pfd));
372 pfd[0].events = POLLIN;
374 pfd[1].events = POLLHUP;
375 if((ret = poll(pfd, 2, -1)) < 0) {
377 flog(LOG_ERR, "accesslog: error in poll: %s", strerror(errno));
382 if((fd = recvreq(0, &req)) < 0) {
385 flog(LOG_ERR, "accesslog: error in recvreq: %s", strerror(errno));
392 if(pfd[1].revents & POLLHUP)