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/>.
20 * XXX: This program is mostly copied from callscgi. It may be
21 * reasonable to unify some of their shared code in a source file.
25 * XXX: All the various ways to start a child process makes this
26 * program quite ugly at the moment. It is unclear whether it is
27 * meaningfully possible to unify them better than they currently are.
36 #include <sys/socket.h>
38 #include <netinet/in.h>
53 #define FCGI_BEGIN_REQUEST 1
54 #define FCGI_ABORT_REQUEST 2
55 #define FCGI_END_REQUEST 3
61 static char **progspec;
62 static char *sockid, *unspec, *inspec;
64 static struct sockaddr *curaddr;
65 static size_t caddrlen;
66 static int cafamily, isanon;
69 static struct addrinfo *resolv(int flags)
72 struct addrinfo *ai, h;
75 if((p = strchr(inspec, ':')) != NULL) {
76 name = smalloc(p - inspec + 1);
77 memcpy(name, inspec, p - inspec);
81 name = sstrdup("localhost");
84 memset(&h, 0, sizeof(h));
85 h.ai_family = AF_UNSPEC;
86 h.ai_socktype = SOCK_STREAM;
88 ret = getaddrinfo(name, srv, &h, &ai);
91 flog(LOG_ERR, "could not resolve TCP specification `%s': %s", inspec, gai_strerror(ret));
97 static char *mksockid(char *sockid)
101 home = getenv("HOME");
102 if(home && !access(sprintf3("%s/.ashd/sockets/", home), X_OK))
103 return(sprintf3("%s/.ashd/sockets/fcgi-p-%s", home, sockid));
104 return(sprintf3("/tmp/fcgi-%i-%s", getuid(), sockid));
107 static char *mkanonid(void)
113 home = getenv("HOME");
114 if(home && !access(sprintf3("%s/.ashd/sockets/", home), X_OK))
115 tmpl = sprintf2("%s/.ashd/sockets/fcgi-a-XXXXXX", home);
117 tmpl = sprintf2("/tmp/fcgi-a-%i-XXXXXX", getuid());
118 if((fd = mkstemp(tmpl)) < 0) {
119 flog(LOG_ERR, "could not create anonymous socket `%s': %s", tmpl, strerror(errno));
127 static void setupchild(void)
129 /* PHP appears to not expect to inherit SIGCHLD set to SIG_IGN, so
130 * reset it for it. */
131 signal(SIGCHLD, SIG_DFL);
134 static void startlisten(void)
137 struct addrinfo *ai, *cai;
139 struct sockaddr_un unm;
145 for(cai = ai = resolv(AI_PASSIVE); cai != NULL; cai = cai->ai_next) {
146 if((fd = socket(cai->ai_family, cai->ai_socktype, cai->ai_protocol)) < 0)
148 if(bind(fd, cai->ai_addr, cai->ai_addrlen)) {
153 if(listen(fd, 128)) {
162 flog(LOG_ERR, "could not bind to specified TCP address: %s", strerror(errno));
165 } else if((unspec != NULL) || (sockid != NULL)) {
166 if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
167 flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
173 unpath = mksockid(sockid);
175 unm.sun_family = AF_UNIX;
176 strcpy(unm.sun_path, unpath);
177 if(bind(fd, (struct sockaddr *)&unm, sizeof(unm))) {
178 flog(LOG_ERR, "could not bind Unix socket to `%s': %s", unspec, strerror(errno));
181 if(listen(fd, 128)) {
182 flog(LOG_ERR, "listen: %s", strerror(errno));
186 if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
187 flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
190 memset(&unm, 0, sizeof(unm));
192 unm.sun_family = AF_UNIX;
193 strcpy(unm.sun_path, aname);
195 if(bind(fd, (struct sockaddr *)&unm, sizeof(unm))) {
196 flog(LOG_ERR, "could not bind Unix socket to `%s': %s", unspec, strerror(errno));
199 if(listen(fd, 128)) {
200 flog(LOG_ERR, "listen: %s", strerror(errno));
204 curaddr = smalloc(caddrlen = sizeof(unm));
205 memcpy(curaddr, &unm, sizeof(unm));
209 if((child = fork()) < 0) {
210 flog(LOG_ERR, "could not fork: %s", strerror(errno));
217 execvp(*progspec, progspec);
218 flog(LOG_ERR, "callfcgi: %s: %s", *progspec, strerror(errno));
224 static void startnolisten(void)
228 if((child = fork()) < 0) {
229 flog(LOG_ERR, "could not fork: %s", strerror(errno));
234 if((fd = open("/dev/null", O_RDONLY)) < 0) {
235 flog(LOG_ERR, "/dev/null: %s", strerror(errno));
240 execvp(*progspec, progspec);
241 flog(LOG_ERR, "callfcgi: %s: %s", *progspec, strerror(errno));
246 static int sconnect(void)
252 fd = socket(cafamily, SOCK_STREAM, 0);
253 fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK);
255 if(!connect(fd, curaddr, caddrlen))
257 if(errno == EAGAIN) {
261 if(errno == EINPROGRESS) {
262 block(fd, EV_WRITE, 30);
263 errlen = sizeof(err);
264 if(getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &errlen) || ((errno = err) != 0)) {
275 static int econnect(void)
278 struct addrinfo *ai, *cai;
281 struct sockaddr_un unm;
287 for(cai = ai = resolv(0); cai != NULL; cai = cai->ai_next) {
288 if((fd = socket(cai->ai_family, cai->ai_socktype, cai->ai_protocol)) < 0)
290 if(connect(fd, cai->ai_addr, cai->ai_addrlen)) {
298 if(tries++ < nolisten) {
302 flog(LOG_ERR, "could not connect to specified TCP address: %s", strerror(errno));
305 curaddr = smalloc(caddrlen = cai->ai_addrlen);
306 memcpy(curaddr, cai->ai_addr, caddrlen);
307 cafamily = cai->ai_family;
311 } else if((unspec != NULL) || (sockid != NULL)) {
312 if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
313 flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
319 unpath = mksockid(sockid);
321 unm.sun_family = AF_UNIX;
322 strcpy(unm.sun_path, unpath);
323 if(connect(fd, (struct sockaddr *)&unm, sizeof(unm))) {
325 if(tries++ < nolisten) {
329 flog(LOG_ERR, "could not connect to Unix socket `%s': %s", unspec, strerror(errno));
332 curaddr = smalloc(caddrlen = sizeof(unm));
333 memcpy(curaddr, &unm, sizeof(unm));
338 flog(LOG_ERR, "callfcgi: cannot use an anonymous socket without a program to start");
343 static int startconn(void)
356 static void killcuraddr(void)
361 unlink(((struct sockaddr_un *)curaddr)->sun_path);
363 kill(child, SIGTERM);
369 static int reconn(void)
373 if(curaddr != NULL) {
374 if((fd = sconnect()) >= 0)
381 static off_t passdata(FILE *in, FILE *out)
389 read = fread(buf, 1, sizeof(buf), in);
392 if(fwrite(buf, 1, read, out) != read)
399 static void bufcatkv(struct charbuf *dst, char *key, char *val)
403 if((kl = strlen(key)) < 128) {
406 bufadd(*dst, ((kl & 0x7f000000) >> 24) | 0x80);
407 bufadd(*dst, (kl & 0x00ff0000) >> 16);
408 bufadd(*dst, (kl & 0x0000ff00) >> 8);
409 bufadd(*dst, kl & 0x000000ff);
411 if((vl = strlen(val)) < 128) {
414 bufadd(*dst, ((vl & 0x7f000000) >> 24) | 0x80);
415 bufadd(*dst, (vl & 0x00ff0000) >> 16);
416 bufadd(*dst, (vl & 0x0000ff00) >> 8);
417 bufadd(*dst, vl & 0x000000ff);
419 bufcat(*dst, key, kl);
420 bufcat(*dst, val, vl);
423 static void bufaddenv(struct charbuf *dst, char *name, char *fmt, ...)
429 val = vsprintf2(fmt, args);
431 bufcatkv(dst, name, val);
435 static char *absolutify(char *file)
437 static int inited = 0;
438 static char cwd[1024];
442 getcwd(cwd, sizeof(cwd));
445 return(sprintf2("%s/%s", cwd, file));
447 return(sstrdup(file));
450 /* Mostly copied from callcgi. */
451 static void mkcgienv(struct hthead *req, struct charbuf *dst)
454 char *url, *pi, *tmp, *qp, *h, *p;
456 bufaddenv(dst, "SERVER_SOFTWARE", "ashd/%s", VERSION);
457 bufaddenv(dst, "GATEWAY_INTERFACE", "CGI/1.1");
458 bufaddenv(dst, "SERVER_PROTOCOL", "%s", req->ver);
459 bufaddenv(dst, "REQUEST_METHOD", "%s", req->method);
460 bufaddenv(dst, "REQUEST_URI", "%s", req->url);
461 url = sstrdup(req->url);
462 if((qp = strchr(url, '?')) != NULL)
464 /* XXX: This is an ugly hack (I think), but though I can think of
465 * several alternatives, none seem to be better. */
466 if(*req->rest && (strlen(url) >= strlen(req->rest)) &&
467 !strcmp(req->rest, url + strlen(url) - strlen(req->rest))) {
468 url[strlen(url) - strlen(req->rest)] = 0;
470 if((pi = unquoteurl(req->rest)) == NULL)
471 pi = sstrdup(req->rest);
472 if(!strcmp(url, "/")) {
473 /* This seems to be normal CGI behavior, but see callcgi.c for
476 pi = sprintf2("/%s", tmp = pi);
479 bufaddenv(dst, "PATH_INFO", "%s", pi);
480 bufaddenv(dst, "SCRIPT_NAME", "%s", url);
481 bufaddenv(dst, "QUERY_STRING", "%s", qp?qp:"");
484 if((h = getheader(req, "Host")) != NULL)
485 bufaddenv(dst, "SERVER_NAME", "%s", h);
486 if((h = getheader(req, "X-Ash-Server-Address")) != NULL)
487 bufaddenv(dst, "SERVER_ADDR", "%s", h);
488 if((h = getheader(req, "X-Ash-Server-Port")) != NULL)
489 bufaddenv(dst, "SERVER_PORT", "%s", h);
490 if((h = getheader(req, "X-Ash-Remote-User")) != NULL)
491 bufaddenv(dst, "REMOTE_USER", "%s", h);
492 if(((h = getheader(req, "X-Ash-Protocol")) != NULL) && !strcmp(h, "https"))
493 bufaddenv(dst, "HTTPS", "on");
494 if((h = getheader(req, "X-Ash-Address")) != NULL)
495 bufaddenv(dst, "REMOTE_ADDR", "%s", h);
496 if((h = getheader(req, "X-Ash-Port")) != NULL)
497 bufaddenv(dst, "REMOTE_PORT", "%s", h);
498 if((h = getheader(req, "Content-Type")) != NULL)
499 bufaddenv(dst, "CONTENT_TYPE", "%s", h);
500 if((h = getheader(req, "Content-Length")) != NULL)
501 bufaddenv(dst, "CONTENT_LENGTH", "%s", h);
503 bufaddenv(dst, "CONTENT_LENGTH", "0");
504 if((h = getheader(req, "X-Ash-File")) != NULL) {
506 bufaddenv(dst, "SCRIPT_FILENAME", "%s", h);
509 for(i = 0; i < req->noheaders; i++) {
510 h = sprintf2("HTTP_%s", req->headers[i][0]);
511 for(p = h; *p; p++) {
517 bufcatkv(dst, h, req->headers[i][1]);
522 static struct hthead *parseresp(FILE *in)
528 resp->ver = sstrdup("HTTP/1.1");
529 if(parseheaders(resp, in)) {
533 if((st = getheader(resp, "Status")) != NULL) {
534 if((p = strchr(st, ' ')) != NULL) {
536 resp->code = atoi(st);
537 resp->msg = sstrdup(p);
539 resp->code = atoi(st);
540 resp->msg = sstrdup(httpdefstatus(resp->code));
542 headrmheader(resp, "Status");
543 } else if(getheader(resp, "Location")) {
545 resp->msg = sstrdup("See Other");
548 resp->msg = sstrdup("OK");
553 #define fputc2(b, f) if(fputc((b), (f)) == EOF) return(-1);
555 static int sendrec(FILE *out, int type, int rid, char *data, size_t dlen)
563 cl = min(dlen - off, 65535);
564 p = (8 - (cl % 8)) % 8;
567 fputc2((rid & 0xff00) >> 8, out);
568 fputc2(rid & 0x00ff, out);
569 fputc2((cl & 0xff00) >> 8, out);
570 fputc2(cl & 0x00ff, out);
573 if(fwrite(data + off, 1, cl, out) != cl)
577 } while((off += cl) < dlen);
581 static int recvrec(FILE *in, int *type, int *rid, char **data, size_t *dlen)
583 unsigned char header[8];
586 if(fread(header, 1, 8, in) != 8)
591 *rid = (header[2] << 8) | header[3];
592 *dlen = (header[4] << 8) | header[5];
593 tl = *dlen + header[6];
596 *data = smalloc(max(tl, 1));
597 if(fread(*data, 1, tl, in) != tl) {
604 static int begreq(FILE *out, int rid)
606 char rec[] = {0, 1, 0, 0, 0, 0, 0, 0};
608 return(sendrec(out, FCGI_BEGIN_REQUEST, rid, rec, 8));
611 static void outplex(struct muth *muth, va_list args)
628 while((ch.s = va_arg(args, FILE *)) != NULL) {
629 ch.id = va_arg(args, int);
634 if(recvrec(sk, &type, &rid, &data, &dlen))
638 for(i = 0; i < outs.d; i++) {
639 if(outs.b[i].id == type) {
640 if(outs.b[i].s != NULL) {
645 if(fwrite(data, 1, dlen, outs.b[i].s) != dlen)
659 for(i = 0; i < outs.d; i++) {
660 if(outs.b[i].s != NULL)
667 static void errhandler(struct muth *muth, va_list args)
674 while(fgets(buf, sizeof(buf), in) != NULL) {
675 p = buf + strlen(buf) - 1;
676 while((p >= buf) && (*p == '\n'))
679 flog(LOG_INFO, "child said: %s", buf);
684 static void serve(struct muth *muth, va_list args)
686 vavar(struct hthead *, req);
689 FILE *is, *os, *outi, *outo, *erri, *erro;
695 is = mtstdopen(fd, 1, 60, "r+", NULL);
696 os = mtstdopen(sfd, 1, 600, "r+", NULL);
699 mtiopipe(&outi, &outo); mtiopipe(&erri, &erro);
700 mustart(outplex, mtstdopen(dup(sfd), 1, 600, "r+", NULL), outo, FCGI_STDOUT, erro, FCGI_STDERR, NULL);
701 mustart(errhandler, erri);
706 mkcgienv(req, &head);
707 if(sendrec(os, FCGI_PARAMS, 1, head.b, head.d))
709 if(sendrec(os, FCGI_PARAMS, 1, NULL, 0))
716 read = fread(buf, 1, sizeof(buf), is);
719 if(sendrec(os, FCGI_STDIN, 1, buf, read))
722 if(sendrec(os, FCGI_STDIN, 1, NULL, 0))
727 if((resp = parseresp(outi)) == NULL)
732 if(passdata(outi, is) < 0)
738 shutdown(sfd, SHUT_RDWR);
745 static void listenloop(struct muth *muth, va_list args)
752 block(0, EV_READ, 0);
753 if((fd = recvreq(lfd, &req)) < 0) {
755 flog(LOG_ERR, "recvreq: %s", strerror(errno));
758 if((sfd = reconn()) < 0) {
763 mustart(serve, req, fd, sfd);
767 static void sigign(int sig)
771 static void sigexit(int sig)
773 shutdown(0, SHUT_RDWR);
776 static void usage(FILE *out)
778 fprintf(out, "usage: callfcgi [-h] [-N RETRIES] [-i ID] [-u UNIX-PATH] [-t [HOST:]TCP-PORT] [PROGRAM [ARGS...]]\n");
781 int main(int argc, char **argv)
785 while((c = getopt(argc, argv, "+hN:i:u:t:")) >= 0) {
791 nolisten = atoi(optarg);
807 progspec = argv + optind;
808 if(((sockid != NULL) + (unspec != NULL) + (inspec != NULL)) > 1) {
809 flog(LOG_ERR, "callfcgi: at most one of -i, -u or -t may be given");
812 signal(SIGCHLD, SIG_IGN);
813 signal(SIGPIPE, sigign);
814 signal(SIGINT, sigexit);
815 signal(SIGTERM, sigexit);
816 mustart(listenloop, 0);