[ashd.git] / src / callscgi.c

/*
    ashd - A Sane HTTP Daemon
    Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

/*
 * XXX: All the various ways to start a child process makes this
 * program quite ugly at the moment. It is unclear whether it is
 * meaningfully possible to unify them better than they currently are.
 */

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <ctype.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <netinet/in.h>
#include <netdb.h>
#include <signal.h>
#include <errno.h>

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <utils.h>
#include <req.h>
#include <resp.h>
#include <log.h>
#include <mt.h>
#include <mtio.h>

static char **progspec;
static char *sockid, *unspec, *inspec;
static int nolisten;
static struct sockaddr *curaddr;
static size_t caddrlen;
static int cafamily, isanon;
static pid_t child;

static struct addrinfo *resolv(int flags)
{
    int ret;
    struct addrinfo *ai, h;
    char *name, *srv, *p;
    
    if((p = strchr(inspec, ':')) != NULL) {
	name = smalloc(p - inspec + 1);
	memcpy(name, inspec, p - inspec);
	name[p - inspec] = 0;
	srv = p + 1;
    } else {
	name = sstrdup("localhost");
	srv = inspec;
    }
    memset(&h, 0, sizeof(h));
    h.ai_family = AF_UNSPEC;
    h.ai_socktype = SOCK_STREAM;
    h.ai_flags = flags;
    ret = getaddrinfo(name, srv, &h, &ai);
    free(name);
    if(ret != 0) {
	flog(LOG_ERR, "could not resolve TCP specification `%s': %s", inspec, gai_strerror(ret));
	exit(1);
    }
    return(ai);
}

static char *mksockid(char *sockid)
{
    char *home;
    
    home = getenv("HOME");
    if(home && !access(sprintf3("%s/.ashd/sockets/", home), X_OK))
	return(sprintf3("%s/.ashd/sockets/scgi-p-%s", home, sockid));
    return(sprintf3("/tmp/scgi-%i-%s", getuid(), sockid));
}

static char *mkanonid(void)
{
    char *home;
    char *tmpl;
    int fd;
    
    home = getenv("HOME");
    if(home && !access(sprintf3("%s/.ashd/sockets/", home), X_OK))
	tmpl = sprintf2("%s/.ashd/sockets/scgi-a-XXXXXX", home);
    else
	tmpl = sprintf2("/tmp/scgi-a-%i-XXXXXX", getuid());
    if((fd = mkstemp(tmpl)) < 0) {
	flog(LOG_ERR, "could not create anonymous socket `%s': %s", tmpl, strerror(errno));
	exit(1);
    }
    close(fd);
    unlink(tmpl);
    return(tmpl);
}

static void setupchild(void)
{
    /* PHP appears to not expect to inherit SIGCHLD set to SIG_IGN, so
     * reset it for it. */
    signal(SIGCHLD, SIG_DFL);
}

static void startlisten(void)
{
    int fd;
    struct addrinfo *ai, *cai;
    char *unpath;
    struct sockaddr_un unm;
    char *aname;
    
    isanon = 0;
    if(inspec != NULL) {
	fd = -1;
	for(cai = ai = resolv(AI_PASSIVE); cai != NULL; cai = cai->ai_next) {
	    if((fd = socket(cai->ai_family, cai->ai_socktype, cai->ai_protocol)) < 0)
		continue;
	    if(bind(fd, cai->ai_addr, cai->ai_addrlen)) {
		close(fd);
		fd = -1;
		continue;
	    }
	    if(listen(fd, 128)) {
		close(fd);
		fd = -1;
		continue;
	    }
	    break;
	}
	freeaddrinfo(ai);
	if(fd < 0) {
	    flog(LOG_ERR, "could not bind to specified TCP address: %s", strerror(errno));
	    exit(1);
	}
    } else if((unspec != NULL) || (sockid != NULL)) {
	if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
	    flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
	    exit(1);
	}
	if(unspec != NULL)
	    unpath = unspec;
	else
	    unpath = mksockid(sockid);
	unlink(unpath);
	unm.sun_family = AF_UNIX;
	strcpy(unm.sun_path, unpath);
	if(bind(fd, (struct sockaddr *)&unm, sizeof(unm))) {
	    flog(LOG_ERR, "could not bind Unix socket to `%s': %s", unspec, strerror(errno));
	    exit(1);
	}
	if(listen(fd, 128)) {
	    flog(LOG_ERR, "listen: %s", strerror(errno));
	    exit(1);
	}
    } else {
	if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
	    flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
	    exit(1);
	}
	memset(&unm, 0, sizeof(unm));
	aname = mkanonid();
	unm.sun_family = AF_UNIX;
	strcpy(unm.sun_path, aname);
	free(aname);
	if(bind(fd, (struct sockaddr *)&unm, sizeof(unm))) {
	    flog(LOG_ERR, "could not bind Unix socket to `%s': %s", unspec, strerror(errno));
	    exit(1);
	}
	if(listen(fd, 128)) {
	    flog(LOG_ERR, "listen: %s", strerror(errno));
	    exit(1);
	}
	
	curaddr = smalloc(caddrlen = sizeof(unm));
	memcpy(curaddr, &unm, sizeof(unm));
	cafamily = AF_UNIX;
	isanon = 1;
    }
    if((child = fork()) < 0) {
	flog(LOG_ERR, "could not fork: %s", strerror(errno));
	exit(1);
    }
    if(child == 0) {
	setupchild();
	dup2(fd, 0);
	close(fd);
	execvp(*progspec, progspec);
	flog(LOG_ERR, "callscgi: %s: %s", *progspec, strerror(errno));
	_exit(127);
    }
    close(fd);
}

static void startnolisten(void)
{
    int fd;
    
    if((child = fork()) < 0) {
	flog(LOG_ERR, "could not fork: %s", strerror(errno));
	exit(1);
    }
    if(child == 0) {
	setupchild();
	if((fd = open("/dev/null", O_RDONLY)) < 0) {
	    flog(LOG_ERR, "/dev/null: %s", strerror(errno));
	    _exit(127);
	}
	dup2(fd, 0);
	close(fd);
	execvp(*progspec, progspec);
	flog(LOG_ERR, "callscgi: %s: %s", *progspec, strerror(errno));
	_exit(127);
    }
}

static int sconnect(void)
{
    int fd;
    int err;
    socklen_t errlen;

    fd = socket(cafamily, SOCK_STREAM, 0);
    fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK);
    while(1) {
	if(connect(fd, curaddr, caddrlen)) {
	    if(errno == EINPROGRESS) {
		block(fd, EV_WRITE, 30);
		errlen = sizeof(err);
		if(getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &errlen) || ((errno = err) != 0)) {
		    close(fd);
		    return(-1);
		}
		return(fd);
	    }
	    close(fd);
	    return(-1);
	}
	return(fd);
    }
}

static int econnect(void)
{
    int fd;
    struct addrinfo *ai, *cai;
    int tries;
    char *unpath;
    struct sockaddr_un unm;
    
    tries = 0;
retry:
    if(inspec != NULL) {
	fd = -1;
	for(cai = ai = resolv(0); cai != NULL; cai = cai->ai_next) {
	    if((fd = socket(cai->ai_family, cai->ai_socktype, cai->ai_protocol)) < 0)
		continue;
	    if(connect(fd, cai->ai_addr, cai->ai_addrlen)) {
		close(fd);
		fd = -1;
		continue;
	    }
	    break;
	}
	if(fd < 0) {
	    if(tries++ < nolisten) {
		sleep(1);
		goto retry;
	    }
	    flog(LOG_ERR, "could not connect to specified TCP address: %s", strerror(errno));
	    exit(1);
	}
	curaddr = smalloc(caddrlen = cai->ai_addrlen);
	memcpy(curaddr, cai->ai_addr, caddrlen);
	cafamily = cai->ai_family;
	isanon = 0;
	freeaddrinfo(ai);
	return(fd);
    } else if((unspec != NULL) || (sockid != NULL)) {
	if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
	    flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
	    exit(1);
	}
	if(unspec != NULL)
	    unpath = unspec;
	else
	    unpath = mksockid(sockid);
	unlink(unpath);
	unm.sun_family = AF_UNIX;
	strcpy(unm.sun_path, unpath);
	if(connect(fd, (struct sockaddr *)&unm, sizeof(unm))) {
	    close(fd);
	    if(tries++ < nolisten) {
		sleep(1);
		goto retry;
	    }
	    flog(LOG_ERR, "could not connect to Unix socket `%s': %s", unspec, strerror(errno));
	    exit(1);
	}
	curaddr = smalloc(caddrlen = sizeof(unm));
	memcpy(curaddr, &unm, sizeof(unm));
	cafamily = AF_UNIX;
	isanon = 0;
	return(fd);
    } else {
	flog(LOG_ERR, "callscgi: cannot use an anonymous socket without a program to start");
	exit(1);
    }
}

static int startconn(void)
{
    if(*progspec) {
	if(nolisten == 0)
	    startlisten();
	else
	    startnolisten();
    }
    if(curaddr != NULL)
	return(sconnect());
    return(econnect());
}

static void killcuraddr(void)
{
    if(curaddr == NULL)
	return;
    if(isanon) {
	unlink(((struct sockaddr_un *)curaddr)->sun_path);
	if(child > 0)
	    kill(child, SIGTERM);
    }
    free(curaddr);
    curaddr = NULL;
}

static int reconn(void)
{
    int fd;
    
    if(curaddr != NULL) {
	if((fd = sconnect()) >= 0)
	    return(fd);
	killcuraddr();
    }
    return(startconn());
}

static off_t passdata(FILE *in, FILE *out)
{
    size_t read;
    off_t total;
    char buf[8192];
    
    total = 0;
    while(!feof(in)) {
	read = fread(buf, 1, sizeof(buf), in);
	if(ferror(in))
	    return(-1);
	if(fwrite(buf, 1, read, out) != read)
	    return(-1);
	total += read;
    }
    return(total);
}

static void bufaddenv(struct charbuf *dst, char *name, char *fmt, ...)
{
    va_list args;
    
    bufcatstr2(*dst, name);
    va_start(args, fmt);
    bvprintf(dst, fmt, args);
    va_end(args);
    bufadd(*dst, 0);
}

static char *absolutify(char *file)
{
    static int inited = 0;
    static char cwd[1024];
    
    if(*file != '/') {
	if(!inited) {
	    getcwd(cwd, sizeof(cwd));
	    inited = 1;
	}
	return(sprintf2("%s/%s", cwd, file));
    }
    return(sstrdup(file));
}

/* Mostly copied from callcgi. */
static void mkcgienv(struct hthead *req, struct charbuf *dst)
{
    int i;
    char *url, *pi, *tmp, *qp, *h, *p;
    
    bufaddenv(dst, "SERVER_SOFTWARE", "ashd/%s", VERSION);
    bufaddenv(dst, "GATEWAY_INTERFACE", "CGI/1.1");
    bufaddenv(dst, "SCGI", "1");
    bufaddenv(dst, "SERVER_PROTOCOL", "%s", req->ver);
    bufaddenv(dst, "REQUEST_METHOD", "%s", req->method);
    bufaddenv(dst, "REQUEST_URI", "%s", req->url);
    url = sstrdup(req->url);
    if((qp = strchr(url, '?')) != NULL)
	*(qp++) = 0;
    /* XXX: This is an ugly hack (I think), but though I can think of
     * several alternatives, none seem to be better. */
    if(*req->rest && (strlen(url) >= strlen(req->rest)) &&
       !strcmp(req->rest, url + strlen(url) - strlen(req->rest))) {
	url[strlen(url) - strlen(req->rest)] = 0;
    }
    if((pi = unquoteurl(req->rest)) == NULL)
	pi = sstrdup(req->rest);
    if(!strcmp(url, "/")) {
	/* This seems to be normal CGI behavior, but see callcgi.c for
	 * details. */
	url[0] = 0;
	pi = sprintf2("/%s", tmp = pi);
	free(tmp);
    }
    bufaddenv(dst, "PATH_INFO", "%s", pi);
    bufaddenv(dst, "SCRIPT_NAME", "%s", url);
    bufaddenv(dst, "QUERY_STRING", "%s", qp?qp:"");
    free(pi);
    free(url);
    if((h = getheader(req, "Host")) != NULL)
	bufaddenv(dst, "SERVER_NAME", "%s", h);
    if((h = getheader(req, "X-Ash-Server-Address")) != NULL)
	bufaddenv(dst, "SERVER_ADDR", "%s", h);
    if((h = getheader(req, "X-Ash-Server-Port")) != NULL)
	bufaddenv(dst, "SERVER_PORT", "%s", h);
    if((h = getheader(req, "X-Ash-Remote-User")) != NULL)
	bufaddenv(dst, "REMOTE_USER", "%s", h);
    if(((h = getheader(req, "X-Ash-Protocol")) != NULL) && !strcmp(h, "https"))
	bufaddenv(dst, "HTTPS", "on");
    if((h = getheader(req, "X-Ash-Address")) != NULL)
	bufaddenv(dst, "REMOTE_ADDR", "%s", h);
    if((h = getheader(req, "X-Ash-Port")) != NULL)
	bufaddenv(dst, "REMOTE_PORT", "%s", h);
    if((h = getheader(req, "Content-Type")) != NULL)
	bufaddenv(dst, "CONTENT_TYPE", "%s", h);
    if((h = getheader(req, "Content-Length")) != NULL)
	bufaddenv(dst, "CONTENT_LENGTH", "%s", h);
    else
	bufaddenv(dst, "CONTENT_LENGTH", "0");
    if((h = getheader(req, "X-Ash-File")) != NULL) {
	h = absolutify(h);
	bufaddenv(dst, "SCRIPT_FILENAME", "%s", h);
	free(h);
    }
    for(i = 0; i < req->noheaders; i++) {
	h = sprintf2("HTTP_%s", req->headers[i][0]);
	for(p = h; *p; p++) {
	    if(isalnum(*p))
		*p = toupper(*p);
	    else
		*p = '_';
	}
	bufcatstr2(*dst, h);
	free(h);
	bufcatstr2(*dst, req->headers[i][1]);
    }
}

static struct hthead *parseresp(FILE *in)
{
    struct hthead *resp;
    char *st, *p;
    
    omalloc(resp);
    resp->ver = sstrdup("HTTP/1.1");
    if(parseheaders(resp, in)) {
	freehthead(resp);
	return(NULL);
    }
    if((st = getheader(resp, "Status")) != NULL) {
	if((p = strchr(st, ' ')) != NULL) {
	    *(p++) = 0;
	    resp->code = atoi(st);
	    resp->msg = sstrdup(p);
	} else {
	    resp->code = atoi(st);
	    resp->msg = sstrdup(httpdefstatus(resp->code));
	}
	headrmheader(resp, "Status");
    } else if(getheader(resp, "Location")) {
	resp->code = 303;
	resp->msg = sstrdup("See Other");
    } else {
	resp->code = 200;
	resp->msg = sstrdup("OK");
    }
    return(resp);
}

static void serve(struct muth *muth, va_list args)
{
    vavar(struct hthead *, req);
    vavar(int, fd);
    vavar(int, sfd);
    FILE *is, *os;
    struct charbuf head;
    struct hthead *resp;
    
    sfd = reconn();
    is = mtstdopen(fd, 1, 60, "r+", NULL);
    os = mtstdopen(sfd, 1, 600, "r+", NULL);
    
    bufinit(head);
    mkcgienv(req, &head);
    fprintf(os, "%zi:", head.d);
    fwrite(head.b, head.d, 1, os);
    fputc(',', os);
    buffree(head);
    if(passdata(is, os) < 0)
	goto out;
    
    if((resp = parseresp(os)) == NULL)
	goto out;
    writeresp(is, resp);
    freehthead(resp);
    fputc('\n', is);
    if(passdata(os, is) < 0)
	goto out;
    
out:
    freehthead(req);
    fclose(is);
    fclose(os);
}

static void listenloop(struct muth *muth, va_list args)
{
    vavar(int, lfd);
    int fd;
    struct hthead *req;
    
    while(1) {
	block(0, EV_READ, 0);
	if((fd = recvreq(lfd, &req)) < 0) {
	    if(errno != 0)
		flog(LOG_ERR, "recvreq: %s", strerror(errno));
	    break;
	}
	mustart(serve, req, fd);
    }
}

static void sigign(int sig)
{
}

static void sigexit(int sig)
{
    shutdown(0, SHUT_RDWR);
}

static void usage(FILE *out)
{
    fprintf(out, "usage: callscgi [-h] [-N RETRIES] [-i ID] [-u UNIX-PATH] [-t [HOST:]TCP-PORT] [PROGRAM [ARGS...]]\n");
}

int main(int argc, char **argv)
{
    int c;
    
    while((c = getopt(argc, argv, "+hN:i:u:t:")) >= 0) {
	switch(c) {
	case 'h':
	    usage(stdout);
	    exit(0);
	case 'N':
	    nolisten = atoi(optarg);
	    break;
	case 'i':
	    sockid = optarg;
	    break;
	case 'u':
	    unspec = optarg;
	    break;
	case 't':
	    inspec = optarg;
	    break;
	default:
	    usage(stderr);
	    exit(1);
	}
    }
    progspec = argv + optind;
    if(((sockid != NULL) + (unspec != NULL) + (inspec != NULL)) > 1) {
	flog(LOG_ERR, "callscgi: at most one of -i, -u or -t may be given");
	exit(1);
    }
    signal(SIGCHLD, SIG_IGN);
    signal(SIGPIPE, sigign);
    signal(SIGINT, sigexit);
    signal(SIGTERM, sigexit);
    mustart(listenloop, 0);
    ioloop();
    killcuraddr();
    return(0);
}
Commit	Line	Data
f89ce57a FT	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	/*
	20	* XXX: All the various ways to start a child process makes this
	21	* program quite ugly at the moment. It is unclear whether it is
	22	* meaningfully possible to unify them better than they currently are.
	23	*/
	24
	25	#include <stdlib.h>
	26	#include <stdio.h>
	27	#include <unistd.h>
	28	#include <string.h>
	29	#include <fcntl.h>
	30	#include <ctype.h>
	31	#include <sys/socket.h>
	32	#include <sys/un.h>
	33	#include <netinet/in.h>
	34	#include <netdb.h>
cc5adcf9	35	#include <signal.h>
f89ce57a FT	36	#include <errno.h>
	37
	38	#ifdef HAVE_CONFIG_H
	39	#include <config.h>
	40	#endif
	41	#include <utils.h>
	42	#include <req.h>
62e76c42	43	#include <resp.h>
f89ce57a FT	44	#include <log.h>
	45	#include <mt.h>
	46	#include <mtio.h>
	47
	48	static char **progspec;
	49	static char sockid, unspec, *inspec;
	50	static int nolisten;
	51	static struct sockaddr *curaddr;
	52	static size_t caddrlen;
	53	static int cafamily, isanon;
	54	static pid_t child;
	55
	56	static struct addrinfo *resolv(int flags)
	57	{
	58	int ret;
	59	struct addrinfo *ai, h;
	60	char name, srv, *p;
	61
	62	if((p = strchr(inspec, ':')) != NULL) {
	63	name = smalloc(p - inspec + 1);
	64	memcpy(name, inspec, p - inspec);
	65	name[p - inspec] = 0;
	66	srv = p + 1;
	67	} else {
	68	name = sstrdup("localhost");
	69	srv = inspec;
	70	}
	71	memset(&h, 0, sizeof(h));
	72	h.ai_family = AF_UNSPEC;
	73	h.ai_socktype = SOCK_STREAM;
	74	h.ai_flags = flags;
	75	ret = getaddrinfo(name, srv, &h, &ai);
	76	free(name);
	77	if(ret != 0) {
	78	flog(LOG_ERR, "could not resolve TCP specification `%s': %s", inspec, gai_strerror(ret));
	79	exit(1);
	80	}
	81	return(ai);
	82	}
	83
	84	static char mksockid(char sockid)
	85	{
	86	char *home;
	87
	88	home = getenv("HOME");
	89	if(home && !access(sprintf3("%s/.ashd/sockets/", home), X_OK))
	90	return(sprintf3("%s/.ashd/sockets/scgi-p-%s", home, sockid));
	91	return(sprintf3("/tmp/scgi-%i-%s", getuid(), sockid));
	92	}
	93
	94	static char *mkanonid(void)
	95	{
	96	char *home;
	97	char *tmpl;
	98	int fd;
	99
	100	home = getenv("HOME");
	101	if(home && !access(sprintf3("%s/.ashd/sockets/", home), X_OK))
1376c0cd	102	tmpl = sprintf2("%s/.ashd/sockets/scgi-a-XXXXXX", home);
f89ce57a FT	103	else
	104	tmpl = sprintf2("/tmp/scgi-a-%i-XXXXXX", getuid());
	105	if((fd = mkstemp(tmpl)) < 0) {
	106	flog(LOG_ERR, "could not create anonymous socket `%s': %s", tmpl, strerror(errno));
	107	exit(1);
	108	}
1376c0cd	109	close(fd);
f89ce57a FT	110	unlink(tmpl);
	111	return(tmpl);
	112	}
	113
b1b0fe0f FT	114	static void setupchild(void)
	115	{
	116	/* PHP appears to not expect to inherit SIGCHLD set to SIG_IGN, so
	117	* reset it for it. */
	118	signal(SIGCHLD, SIG_DFL);
	119	}
	120
f89ce57a FT	121	static void startlisten(void)
f89ce57a FT	122	{
b1b0fe0f	123	int fd;
f89ce57a FT	124	struct addrinfo ai, cai;
	125	char *unpath;
	126	struct sockaddr_un unm;
	127	char *aname;
	128
	129	isanon = 0;
	130	if(inspec != NULL) {
	131	fd = -1;
	132	for(cai = ai = resolv(AI_PASSIVE); cai != NULL; cai = cai->ai_next) {
	133	if((fd = socket(cai->ai_family, cai->ai_socktype, cai->ai_protocol)) < 0)
	134	continue;
	135	if(bind(fd, cai->ai_addr, cai->ai_addrlen)) {
	136	close(fd);
	137	fd = -1;
	138	continue;
	139	}
	140	if(listen(fd, 128)) {
	141	close(fd);
	142	fd = -1;
	143	continue;
	144	}
	145	break;
	146	}
	147	freeaddrinfo(ai);
	148	if(fd < 0) {
	149	flog(LOG_ERR, "could not bind to specified TCP address: %s", strerror(errno));
	150	exit(1);
	151	}
	152	} else if((unspec != NULL) \|\| (sockid != NULL)) {
	153	if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
	154	flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
	155	exit(1);
	156	}
	157	if(unspec != NULL)
	158	unpath = unspec;
	159	else
	160	unpath = mksockid(sockid);
	161	unlink(unpath);
	162	unm.sun_family = AF_UNIX;
	163	strcpy(unm.sun_path, unpath);
	164	if(bind(fd, (struct sockaddr *)&unm, sizeof(unm))) {
	165	flog(LOG_ERR, "could not bind Unix socket to `%s': %s", unspec, strerror(errno));
	166	exit(1);
	167	}
	168	if(listen(fd, 128)) {
	169	flog(LOG_ERR, "listen: %s", strerror(errno));
	170	exit(1);
	171	}
	172	} else {
	173	if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
	174	flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
	175	exit(1);
	176	}
	177	memset(&unm, 0, sizeof(unm));
	178	aname = mkanonid();
	179	unm.sun_family = AF_UNIX;
	180	strcpy(unm.sun_path, aname);
	181	free(aname);
	182	if(bind(fd, (struct sockaddr *)&unm, sizeof(unm))) {
	183	flog(LOG_ERR, "could not bind Unix socket to `%s': %s", unspec, strerror(errno));
	184	exit(1);
	185	}
	186	if(listen(fd, 128)) {
	187	flog(LOG_ERR, "listen: %s", strerror(errno));
188	exit(1);
189	}
190
191	curaddr = smalloc(caddrlen = sizeof(unm));
192	memcpy(curaddr, &unm, sizeof(unm));
193	cafamily = AF_UNIX;
194	isanon = 1;
195	}
196	if((child = fork()) < 0) {
197	flog(LOG_ERR, "could not fork: %s", strerror(errno));
198	exit(1);
199	}
200	if(child == 0) {
b1b0fe0f	201	setupchild();
f89ce57a	202	dup2(fd, 0);
b1b0fe0f	203	close(fd);
f89ce57a	204	execvp(*progspec, progspec);
7b9c1cff	205	flog(LOG_ERR, "callscgi: %s: %s", *progspec, strerror(errno));
107aa2ef	206	_exit(127);
f89ce57a FT	207	}
	208	close(fd);
	209	}
	210
	211	static void startnolisten(void)
	212	{
b1b0fe0f	213	int fd;
f89ce57a FT	214
	215	if((child = fork()) < 0) {
	216	flog(LOG_ERR, "could not fork: %s", strerror(errno));
	217	exit(1);
	218	}
	219	if(child == 0) {
b1b0fe0f	220	setupchild();
02cb2a3d FT	221	if((fd = open("/dev/null", O_RDONLY)) < 0) {
02cb2a3d FT	222	flog(LOG_ERR, "/dev/null: %s", strerror(errno));
107aa2ef	223	_exit(127);
02cb2a3d FT	224	}
	225	dup2(fd, 0);
	226	close(fd);
f89ce57a	227	execvp(*progspec, progspec);
7b9c1cff	228	flog(LOG_ERR, "callscgi: %s: %s", *progspec, strerror(errno));
107aa2ef	229	_exit(127);
f89ce57a FT	230	}
	231	}
	232
	233	static int sconnect(void)
	234	{
	235	int fd;
	236	int err;
	237	socklen_t errlen;
	238
	239	fd = socket(cafamily, SOCK_STREAM, 0);
	240	fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) \| O_NONBLOCK);
	241	while(1) {
	242	if(connect(fd, curaddr, caddrlen)) {
	243	if(errno == EINPROGRESS) {
	244	block(fd, EV_WRITE, 30);
	245	errlen = sizeof(err);
	246	if(getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &errlen) \|\| ((errno = err) != 0)) {
	247	close(fd);
	248	return(-1);
	249	}
	250	return(fd);
	251	}
	252	close(fd);
	253	return(-1);
	254	}
	255	return(fd);
	256	}
	257	}
	258
	259	static int econnect(void)
	260	{
	261	int fd;
	262	struct addrinfo ai, cai;
	263	int tries;
	264	char *unpath;
	265	struct sockaddr_un unm;
	266
	267	tries = 0;
	268	retry:
	269	if(inspec != NULL) {
	270	fd = -1;
	271	for(cai = ai = resolv(0); cai != NULL; cai = cai->ai_next) {
	272	if((fd = socket(cai->ai_family, cai->ai_socktype, cai->ai_protocol)) < 0)
	273	continue;
	274	if(connect(fd, cai->ai_addr, cai->ai_addrlen)) {
	275	close(fd);
	276	fd = -1;
	277	continue;
	278	}
	279	break;
	280	}
f89ce57a	281	if(fd < 0) {
02cb2a3d FT	282	if(tries++ < nolisten) {
	283	sleep(1);
	284	goto retry;
	285	}
f89ce57a FT	286	flog(LOG_ERR, "could not connect to specified TCP address: %s", strerror(errno));
	287	exit(1);
	288	}
	289	curaddr = smalloc(caddrlen = cai->ai_addrlen);
	290	memcpy(curaddr, cai->ai_addr, caddrlen);
	291	cafamily = cai->ai_family;
	292	isanon = 0;
	293	freeaddrinfo(ai);
	294	return(fd);
	295	} else if((unspec != NULL) \|\| (sockid != NULL)) {
	296	if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
	297	flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
	298	exit(1);
	299	}
	300	if(unspec != NULL)
	301	unpath = unspec;
	302	else
	303	unpath = mksockid(sockid);
	304	unlink(unpath);
	305	unm.sun_family = AF_UNIX;
	306	strcpy(unm.sun_path, unpath);
	307	if(connect(fd, (struct sockaddr *)&unm, sizeof(unm))) {
	308	close(fd);
	309	if(tries++ < nolisten) {
	310	sleep(1);
	311	goto retry;
	312	}
	313	flog(LOG_ERR, "could not connect to Unix socket `%s': %s", unspec, strerror(errno));
	314	exit(1);
	315	}
	316	curaddr = smalloc(caddrlen = sizeof(unm));
	317	memcpy(curaddr, &unm, sizeof(unm));
	318	cafamily = AF_UNIX;
	319	isanon = 0;
	320	return(fd);
	321	} else {
425eccf6	322	flog(LOG_ERR, "callscgi: cannot use an anonymous socket without a program to start");
f89ce57a FT	323	exit(1);
	324	}
	325	}
	326
	327	static int startconn(void)
	328	{
	329	if(*progspec) {
	330	if(nolisten == 0)
	331	startlisten();
	332	else
	333	startnolisten();
	334	}
	335	if(curaddr != NULL)
	336	return(sconnect());
	337	return(econnect());
	338	}
	339
	340	static void killcuraddr(void)
	341	{
	342	if(curaddr == NULL)
	343	return;
	344	if(isanon) {
	345	unlink(((struct sockaddr_un *)curaddr)->sun_path);
	346	if(child > 0)
	347	kill(child, SIGTERM);
	348	}
	349	free(curaddr);
	350	curaddr = NULL;
	351	}
	352
	353	static int reconn(void)
	354	{
	355	int fd;
	356
	357	if(curaddr != NULL) {
	358	if((fd = sconnect()) >= 0)
	359	return(fd);
	360	killcuraddr();
	361	}
	362	return(startconn());
	363	}
	364
	365	static off_t passdata(FILE in, FILE out)
	366	{
	367	size_t read;
	368	off_t total;
	369	char buf[8192];
	370
	371	total = 0;
	372	while(!feof(in)) {
	373	read = fread(buf, 1, sizeof(buf), in);
	374	if(ferror(in))
	375	return(-1);
	376	if(fwrite(buf, 1, read, out) != read)
	377	return(-1);
	378	total += read;
	379	}
	380	return(total);
	381	}
	382
	383	static void bufaddenv(struct charbuf dst, char name, char *fmt, ...)
	384	{
	385	va_list args;
	386
387	bufcatstr2(*dst, name);
388	va_start(args, fmt);
389	bvprintf(dst, fmt, args);
390	va_end(args);
391	bufadd(*dst, 0);
392	}
393
394	static char absolutify(char file)
395	{
396	static int inited = 0;
397	static char cwd[1024];
398
399	if(*file != '/') {
400	if(!inited) {
401	getcwd(cwd, sizeof(cwd));
402	inited = 1;
403	}
404	return(sprintf2("%s/%s", cwd, file));
405	}
406	return(sstrdup(file));
407	}
408
409	/* Mostly copied from callcgi. */
410	static void mkcgienv(struct hthead req, struct charbuf dst)
411	{
412	int i;
53d666ca	413	char url, pi, tmp, qp, h, p;
f89ce57a FT	414
	415	bufaddenv(dst, "SERVER_SOFTWARE", "ashd/%s", VERSION);
	416	bufaddenv(dst, "GATEWAY_INTERFACE", "CGI/1.1");
	417	bufaddenv(dst, "SCGI", "1");
	418	bufaddenv(dst, "SERVER_PROTOCOL", "%s", req->ver);
	419	bufaddenv(dst, "REQUEST_METHOD", "%s", req->method);
	420	bufaddenv(dst, "REQUEST_URI", "%s", req->url);
f89ce57a FT	421	url = sstrdup(req->url);
	422	if((qp = strchr(url, '?')) != NULL)
	423	*(qp++) = 0;
	424	/* XXX: This is an ugly hack (I think), but though I can think of
	425	* several alternatives, none seem to be better. */
2d65add0 FT	426	if(*req->rest && (strlen(url) >= strlen(req->rest)) &&
2d65add0 FT	427	!strcmp(req->rest, url + strlen(url) - strlen(req->rest))) {
53d666ca FT	428	url[strlen(url) - strlen(req->rest)] = 0;
	429	}
	430	if((pi = unquoteurl(req->rest)) == NULL)
	431	pi = sstrdup(req->rest);
	432	if(!strcmp(url, "/")) {
	433	/* This seems to be normal CGI behavior, but see callcgi.c for
	434	* details. */
	435	url[0] = 0;
	436	pi = sprintf2("/%s", tmp = pi);
	437	free(tmp);
f89ce57a	438	}
a80c8f02 FT	439	bufaddenv(dst, "PATH_INFO", "%s", pi);
a80c8f02 FT	440	bufaddenv(dst, "SCRIPT_NAME", "%s", url);
f89ce57a	441	bufaddenv(dst, "QUERY_STRING", "%s", qp?qp:"");
53d666ca	442	free(pi);
b77076c4	443	free(url);
f89ce57a FT	444	if((h = getheader(req, "Host")) != NULL)
f89ce57a FT	445	bufaddenv(dst, "SERVER_NAME", "%s", h);
1af656d2 FT	446	if((h = getheader(req, "X-Ash-Server-Address")) != NULL)
1af656d2 FT	447	bufaddenv(dst, "SERVER_ADDR", "%s", h);
f89ce57a FT	448	if((h = getheader(req, "X-Ash-Server-Port")) != NULL)
f89ce57a FT	449	bufaddenv(dst, "SERVER_PORT", "%s", h);
381f9919 FT	450	if((h = getheader(req, "X-Ash-Remote-User")) != NULL)
381f9919 FT	451	bufaddenv(dst, "REMOTE_USER", "%s", h);
9230295c	452	if(((h = getheader(req, "X-Ash-Protocol")) != NULL) && !strcmp(h, "https"))
f89ce57a FT	453	bufaddenv(dst, "HTTPS", "on");
	454	if((h = getheader(req, "X-Ash-Address")) != NULL)
	455	bufaddenv(dst, "REMOTE_ADDR", "%s", h);
1af656d2 FT	456	if((h = getheader(req, "X-Ash-Port")) != NULL)
1af656d2 FT	457	bufaddenv(dst, "REMOTE_PORT", "%s", h);
f89ce57a FT	458	if((h = getheader(req, "Content-Type")) != NULL)
	459	bufaddenv(dst, "CONTENT_TYPE", "%s", h);
	460	if((h = getheader(req, "Content-Length")) != NULL)
	461	bufaddenv(dst, "CONTENT_LENGTH", "%s", h);
	462	else
	463	bufaddenv(dst, "CONTENT_LENGTH", "0");
e25bf4f9 FT	464	if((h = getheader(req, "X-Ash-File")) != NULL) {
	465	h = absolutify(h);
	466	bufaddenv(dst, "SCRIPT_FILENAME", "%s", h);
	467	free(h);
	468	}
f89ce57a	469	for(i = 0; i < req->noheaders; i++) {
02cb2a3d	470	h = sprintf2("HTTP_%s", req->headers[i][0]);
f89ce57a FT	471	for(p = h; *p; p++) {
	472	if(isalnum(*p))
	473	p = toupper(p);
	474	else
	475	*p = '_';
	476	}
	477	bufcatstr2(*dst, h);
	478	free(h);
	479	bufcatstr2(*dst, req->headers[i][1]);
	480	}
	481	}
	482
f89ce57a FT	483	static struct hthead parseresp(FILE in)
	484	{
	485	struct hthead *resp;
	486	char st, p;
	487
	488	omalloc(resp);
	489	resp->ver = sstrdup("HTTP/1.1");
	490	if(parseheaders(resp, in)) {
	491	freehthead(resp);
	492	return(NULL);
	493	}
	494	if((st = getheader(resp, "Status")) != NULL) {
	495	if((p = strchr(st, ' ')) != NULL) {
	496	*(p++) = 0;
	497	resp->code = atoi(st);
	498	resp->msg = sstrdup(p);
	499	} else {
	500	resp->code = atoi(st);
62e76c42	501	resp->msg = sstrdup(httpdefstatus(resp->code));
f89ce57a FT	502	}
	503	headrmheader(resp, "Status");
	504	} else if(getheader(resp, "Location")) {
	505	resp->code = 303;
	506	resp->msg = sstrdup("See Other");
	507	} else {
	508	resp->code = 200;
	509	resp->msg = sstrdup("OK");
	510	}
	511	return(resp);
	512	}
	513
	514	static void serve(struct muth *muth, va_list args)
	515	{
	516	vavar(struct hthead *, req);
	517	vavar(int, fd);
	518	vavar(int, sfd);
	519	FILE is, os;
	520	struct charbuf head;
	521	struct hthead *resp;
	522
	523	sfd = reconn();
b71ad67f FT	524	is = mtstdopen(fd, 1, 60, "r+", NULL);
b71ad67f FT	525	os = mtstdopen(sfd, 1, 600, "r+", NULL);
f89ce57a FT	526
	527	bufinit(head);
	528	mkcgienv(req, &head);
	529	fprintf(os, "%zi:", head.d);
	530	fwrite(head.b, head.d, 1, os);
	531	fputc(',', os);
	532	buffree(head);
	533	if(passdata(is, os) < 0)
	534	goto out;
	535
	536	if((resp = parseresp(os)) == NULL)
	537	goto out;
	538	writeresp(is, resp);
	539	freehthead(resp);
	540	fputc('\n', is);
	541	if(passdata(os, is) < 0)
	542	goto out;
	543
	544	out:
	545	freehthead(req);
	546	fclose(is);
	547	fclose(os);
	548	}
	549
	550	static void listenloop(struct muth *muth, va_list args)
	551	{
	552	vavar(int, lfd);
	553	int fd;
	554	struct hthead *req;
	555
	556	while(1) {
	557	block(0, EV_READ, 0);
	558	if((fd = recvreq(lfd, &req)) < 0) {
	559	if(errno != 0)
	560	flog(LOG_ERR, "recvreq: %s", strerror(errno));
	561	break;
	562	}
	563	mustart(serve, req, fd);
	564	}
	565	}
	566
fd735432 FT	567	static void sigign(int sig)
	568	{
	569	}
	570
9c86e336 FT	571	static void sigexit(int sig)
9c86e336 FT	572	{
254eb937	573	shutdown(0, SHUT_RDWR);
9c86e336 FT	574	}
9c86e336 FT	575
f89ce57a FT	576	static void usage(FILE *out)
f89ce57a FT	577	{
425eccf6	578	fprintf(out, "usage: callscgi [-h] [-N RETRIES] [-i ID] [-u UNIX-PATH] [-t [HOST:]TCP-PORT] [PROGRAM [ARGS...]]\n");
f89ce57a FT	579	}
	580
	581	int main(int argc, char **argv)
	582	{
	583	int c;
	584
	585	while((c = getopt(argc, argv, "+hN:i:u:t:")) >= 0) {
	586	switch(c) {
	587	case 'h':
	588	usage(stdout);
	589	exit(0);
	590	case 'N':
	591	nolisten = atoi(optarg);
	592	break;
	593	case 'i':
	594	sockid = optarg;
	595	break;
	596	case 'u':
	597	unspec = optarg;
	598	break;
	599	case 't':
	600	inspec = optarg;
	601	break;
	602	default:
	603	usage(stderr);
	604	exit(1);
	605	}
	606	}
	607	progspec = argv + optind;
	608	if(((sockid != NULL) + (unspec != NULL) + (inspec != NULL)) > 1) {
425eccf6	609	flog(LOG_ERR, "callscgi: at most one of -i, -u or -t may be given");
f89ce57a FT	610	exit(1);
	611	}
	612	signal(SIGCHLD, SIG_IGN);
fd735432	613	signal(SIGPIPE, sigign);
9c86e336 FT	614	signal(SIGINT, sigexit);
9c86e336 FT	615	signal(SIGTERM, sigexit);
f89ce57a	616	mustart(listenloop, 0);
f89ce57a	617	ioloop();
d06b6013	618	killcuraddr();
f89ce57a FT	619	return(0);
f89ce57a FT	620	}