[ashd.git] / src / callfcgi.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: This program is mostly copied from callscgi. It may be
 * reasonable to unify some of their shared code in a source file.
 */

/*
 * 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>

#define FCGI_BEGIN_REQUEST 1
#define FCGI_ABORT_REQUEST 2
#define FCGI_END_REQUEST 3
#define FCGI_PARAMS 4
#define FCGI_STDIN 5
#define FCGI_STDOUT 6
#define FCGI_STDERR 7

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/fcgi-p-%s", home, sockid));
    return(sprintf3("/tmp/fcgi-%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/fcgi-a-XXXXXX", home);
    else
	tmpl = sprintf2("/tmp/fcgi-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, "callfcgi: %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, "callfcgi: %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, "callfcgi: 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 bufcatkv(struct charbuf *dst, char *key, char *val)
{
    size_t kl, vl;
    
    if((kl = strlen(key)) < 128) {
	bufadd(*dst, kl);
    } else {
	bufadd(*dst, ((kl & 0x7f000000) >> 24) | 0x80);
	bufadd(*dst, (kl & 0x00ff0000) >> 16);
	bufadd(*dst, (kl & 0x0000ff00) >> 8);
	bufadd(*dst, kl & 0x000000ff);
    }
    if((vl = strlen(val)) < 128) {
	bufadd(*dst, vl);
    } else {
	bufadd(*dst, ((vl & 0x7f000000) >> 24) | 0x80);
	bufadd(*dst, (vl & 0x00ff0000) >> 16);
	bufadd(*dst, (vl & 0x0000ff00) >> 8);
	bufadd(*dst, vl & 0x000000ff);
    }
    bufcat(*dst, key, kl);
    bufcat(*dst, val, vl);
}

static void bufaddenv(struct charbuf *dst, char *name, char *fmt, ...)
{
    va_list args;
    char *val = NULL;
    
    va_start(args, fmt);
    val = vsprintf2(fmt, args);
    va_end(args);
    bufcatkv(dst, name, val);
    free(val);
}

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, "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 = '_';
	}
	bufcatkv(dst, h, req->headers[i][1]);
	free(h);
    }
}

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);
}

#define fputc2(b, f) if(fputc((b), (f)) == EOF) return(-1);

static int sendrec(FILE *out, int type, int rid, char *data, size_t dlen)
{
    off_t off;
    size_t cl;
    int p;
    
    off = 0;
    do {
	cl = min(dlen - off, 65535);
	p = (8 - (cl % 8)) % 8;
	fputc2(1, out);
	fputc2(type, out);
	fputc2((rid & 0xff00) >> 8, out);
	fputc2(rid & 0x00ff, out);
	fputc2((cl & 0xff00) >> 8, out);
	fputc2(cl & 0x00ff, out);
	fputc2(p, out);
	fputc2(0, out);
	if(fwrite(data + off, 1, cl, out) != cl)
	    return(-1);
	for(; p > 0; p--)
	    fputc2(0, out);
    } while((off += cl) < dlen);
    return(0);
}

static int recvrec(FILE *in, int *type, int *rid, char **data, size_t *dlen)
{
    unsigned char header[8];
    int tl;
    
    if(fread(header, 1, 8, in) != 8)
	return(-1);
    if(header[0] != 1)
	return(-1);
    *type = header[1];
    *rid  = (header[2] << 8) | header[3];
    *dlen = (header[4] << 8) | header[5];
    tl = *dlen + header[6];
    if(header[7] != 0)
	return(-1);
    *data = smalloc(max(tl, 1));
    if(fread(*data, 1, tl, in) != tl) {
	free(*data);
	return(-1);
    }
    return(0);
}

static int begreq(FILE *out, int rid)
{
    char rec[] = {0, 1, 0, 0, 0, 0, 0, 0};
    
    return(sendrec(out, FCGI_BEGIN_REQUEST, rid, rec, 8));
}

static void outplex(struct muth *muth, va_list args)
{
    vavar(FILE *, sk);
    struct {
	struct ch {
	    FILE *s;
	    int id;
	} *b;
	size_t s, d;
    } outs;
    int i;
    struct ch ch;
    int type, rid;
    char *data;
    size_t dlen;
    
    bufinit(outs);
    while((ch.s = va_arg(args, FILE *)) != NULL) {
	ch.id = va_arg(args, int);
	bufadd(outs, ch);
    }
    data = NULL;
    while(1) {
	if(recvrec(sk, &type, &rid, &data, &dlen))
	    goto out;
	if(rid != 1)
	    goto out;
	for(i = 0; i < outs.d; i++) {
	    if(outs.b[i].id == type) {
		if(outs.b[i].s != NULL) {
		    if(dlen == 0) {
			fclose(outs.b[i].s);
			outs.b[i].s = NULL;
		    } else {
			if(fwrite(data, 1, dlen, outs.b[i].s) != dlen)
			    goto out;
		    }
		}
		break;
	    }
	}
	free(data);
	data = NULL;
    }

out:
    if(data != NULL)
	free(data);
    for(i = 0; i < outs.d; i++) {
	if(outs.b[i].s != NULL)
	    fclose(outs.b[i].s);
    }
    buffree(outs);
    fclose(sk);
}

static void errhandler(struct muth *muth, va_list args)
{
    vavar(FILE *, in);
    char buf[1024];
    char *p;
    
    bufinit(buf);
    while(fgets(buf, sizeof(buf), in) != NULL) {
	p = buf + strlen(buf) - 1;
	while((p >= buf) && (*p == '\n'))
	    *(p--) = 0;
	if(buf[0])
	    flog(LOG_INFO, "child said: %s", buf);
    }
    fclose(in);
}

static void serve(struct muth *muth, va_list args)
{
    vavar(struct hthead *, req);
    vavar(int, fd);
    vavar(int, sfd);
    FILE *is, *os, *outi, *outo, *erri, *erro;
    struct charbuf head;
    struct hthead *resp;
    size_t read;
    char buf[8192];
    
    sfd = reconn();
    is = mtstdopen(fd, 1, 60, "r+", NULL);
    os = mtstdopen(sfd, 1, 600, "r+", NULL);
    
    outi = NULL;
    mtiopipe(&outi, &outo); mtiopipe(&erri, &erro);
    mustart(outplex, mtstdopen(dup(sfd), 1, 600, "r+", NULL), outo, FCGI_STDOUT, erro, FCGI_STDERR, NULL);
    mustart(errhandler, erri);
    
    if(begreq(os, 1))
	goto out;
    bufinit(head);
    mkcgienv(req, &head);
    if(sendrec(os, FCGI_PARAMS, 1, head.b, head.d))
	goto out;
    if(sendrec(os, FCGI_PARAMS, 1, NULL, 0))
	goto out;
    buffree(head);
    if(fflush(os))
	goto out;
    
    while(!feof(is)) {
	read = fread(buf, 1, sizeof(buf), is);
	if(ferror(is))
	    goto out;
	if(sendrec(os, FCGI_STDIN, 1, buf, read))
	    goto out;
    }
    if(sendrec(os, FCGI_STDIN, 1, NULL, 0))
	goto out;
    if(fflush(os))
	goto out;
    
    if((resp = parseresp(outi)) == NULL)
	goto out;
    writeresp(is, resp);
    freehthead(resp);
    fputc('\n', is);
    if(passdata(outi, is) < 0)
	goto out;
    
out:
    freehthead(req);
    buffree(head);
    shutdown(sfd, SHUT_RDWR);
    if(outi != NULL)
	fclose(outi);
    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: callfcgi [-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, "callfcgi: 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
5c2040da 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: This program is mostly copied from callscgi. It may be
	21	* reasonable to unify some of their shared code in a source file.
	22	*/
	23
	24	/*
	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.
	28	*/
	29
	30	#include <stdlib.h>
	31	#include <stdio.h>
	32	#include <unistd.h>
	33	#include <string.h>
	34	#include <fcntl.h>
	35	#include <ctype.h>
	36	#include <sys/socket.h>
	37	#include <sys/un.h>
	38	#include <netinet/in.h>
	39	#include <netdb.h>
cc5adcf9	40	#include <signal.h>
5c2040da FT	41	#include <errno.h>
	42
	43	#ifdef HAVE_CONFIG_H
	44	#include <config.h>
	45	#endif
	46	#include <utils.h>
	47	#include <req.h>
62e76c42	48	#include <resp.h>
5c2040da FT	49	#include <log.h>
	50	#include <mt.h>
	51	#include <mtio.h>
	52
	53	#define FCGI_BEGIN_REQUEST 1
	54	#define FCGI_ABORT_REQUEST 2
	55	#define FCGI_END_REQUEST 3
	56	#define FCGI_PARAMS 4
	57	#define FCGI_STDIN 5
	58	#define FCGI_STDOUT 6
	59	#define FCGI_STDERR 7
	60
	61	static char **progspec;
	62	static char sockid, unspec, *inspec;
	63	static int nolisten;
	64	static struct sockaddr *curaddr;
	65	static size_t caddrlen;
	66	static int cafamily, isanon;
	67	static pid_t child;
	68
	69	static struct addrinfo *resolv(int flags)
	70	{
	71	int ret;
	72	struct addrinfo *ai, h;
	73	char name, srv, *p;
	74
	75	if((p = strchr(inspec, ':')) != NULL) {
	76	name = smalloc(p - inspec + 1);
	77	memcpy(name, inspec, p - inspec);
	78	name[p - inspec] = 0;
	79	srv = p + 1;
	80	} else {
	81	name = sstrdup("localhost");
	82	srv = inspec;
	83	}
	84	memset(&h, 0, sizeof(h));
	85	h.ai_family = AF_UNSPEC;
	86	h.ai_socktype = SOCK_STREAM;
	87	h.ai_flags = flags;
	88	ret = getaddrinfo(name, srv, &h, &ai);
	89	free(name);
	90	if(ret != 0) {
	91	flog(LOG_ERR, "could not resolve TCP specification `%s': %s", inspec, gai_strerror(ret));
	92	exit(1);
	93	}
	94	return(ai);
	95	}
	96
	97	static char mksockid(char sockid)
	98	{
	99	char *home;
	100
	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));
	105	}
	106
	107	static char *mkanonid(void)
	108	{
	109	char *home;
	110	char *tmpl;
	111	int fd;
	112
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);
116	else
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));
120	exit(1);
121	}
122	close(fd);
123	unlink(tmpl);
124	return(tmpl);
125	}
126
b1b0fe0f FT	127	static void setupchild(void)
	128	{
	129	/* PHP appears to not expect to inherit SIGCHLD set to SIG_IGN, so
	130	* reset it for it. */
	131	signal(SIGCHLD, SIG_DFL);
	132	}
	133
5c2040da FT	134	static void startlisten(void)
5c2040da FT	135	{
b1b0fe0f	136	int fd;
5c2040da FT	137	struct addrinfo ai, cai;
	138	char *unpath;
	139	struct sockaddr_un unm;
	140	char *aname;
	141
	142	isanon = 0;
	143	if(inspec != NULL) {
	144	fd = -1;
	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)
	147	continue;
	148	if(bind(fd, cai->ai_addr, cai->ai_addrlen)) {
	149	close(fd);
	150	fd = -1;
	151	continue;
	152	}
	153	if(listen(fd, 128)) {
	154	close(fd);
	155	fd = -1;
	156	continue;
	157	}
	158	break;
	159	}
	160	freeaddrinfo(ai);
	161	if(fd < 0) {
	162	flog(LOG_ERR, "could not bind to specified TCP address: %s", strerror(errno));
	163	exit(1);
	164	}
	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));
	168	exit(1);
	169	}
	170	if(unspec != NULL)
	171	unpath = unspec;
	172	else
	173	unpath = mksockid(sockid);
	174	unlink(unpath);
	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));
	179	exit(1);
	180	}
	181	if(listen(fd, 128)) {
	182	flog(LOG_ERR, "listen: %s", strerror(errno));
	183	exit(1);
	184	}
	185	} else {
	186	if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
	187	flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
	188	exit(1);
	189	}
	190	memset(&unm, 0, sizeof(unm));
	191	aname = mkanonid();
	192	unm.sun_family = AF_UNIX;
	193	strcpy(unm.sun_path, aname);
	194	free(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));
	197	exit(1);
	198	}
	199	if(listen(fd, 128)) {
	200	flog(LOG_ERR, "listen: %s", strerror(errno));
201	exit(1);
202	}
203
204	curaddr = smalloc(caddrlen = sizeof(unm));
205	memcpy(curaddr, &unm, sizeof(unm));
206	cafamily = AF_UNIX;
207	isanon = 1;
208	}
209	if((child = fork()) < 0) {
210	flog(LOG_ERR, "could not fork: %s", strerror(errno));
211	exit(1);
212	}
213	if(child == 0) {
b1b0fe0f	214	setupchild();
5c2040da	215	dup2(fd, 0);
b1b0fe0f	216	close(fd);
5c2040da FT	217	execvp(*progspec, progspec);
5c2040da FT	218	flog(LOG_ERR, "callfcgi: %s: %s", *progspec, strerror(errno));
107aa2ef	219	_exit(127);
5c2040da FT	220	}
	221	close(fd);
	222	}
	223
	224	static void startnolisten(void)
	225	{
b1b0fe0f	226	int fd;
5c2040da FT	227
	228	if((child = fork()) < 0) {
	229	flog(LOG_ERR, "could not fork: %s", strerror(errno));
	230	exit(1);
	231	}
	232	if(child == 0) {
b1b0fe0f	233	setupchild();
5c2040da FT	234	if((fd = open("/dev/null", O_RDONLY)) < 0) {
5c2040da FT	235	flog(LOG_ERR, "/dev/null: %s", strerror(errno));
107aa2ef	236	_exit(127);
5c2040da FT	237	}
	238	dup2(fd, 0);
	239	close(fd);
	240	execvp(*progspec, progspec);
	241	flog(LOG_ERR, "callfcgi: %s: %s", *progspec, strerror(errno));
107aa2ef	242	_exit(127);
5c2040da FT	243	}
	244	}
	245
	246	static int sconnect(void)
	247	{
	248	int fd;
	249	int err;
	250	socklen_t errlen;
	251
	252	fd = socket(cafamily, SOCK_STREAM, 0);
	253	fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) \| O_NONBLOCK);
	254	while(1) {
	255	if(connect(fd, curaddr, caddrlen)) {
	256	if(errno == EINPROGRESS) {
	257	block(fd, EV_WRITE, 30);
	258	errlen = sizeof(err);
	259	if(getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &errlen) \|\| ((errno = err) != 0)) {
	260	close(fd);
	261	return(-1);
	262	}
	263	return(fd);
	264	}
	265	close(fd);
	266	return(-1);
	267	}
	268	return(fd);
	269	}
	270	}
	271
	272	static int econnect(void)
	273	{
	274	int fd;
	275	struct addrinfo ai, cai;
	276	int tries;
	277	char *unpath;
	278	struct sockaddr_un unm;
	279
	280	tries = 0;
	281	retry:
	282	if(inspec != NULL) {
	283	fd = -1;
	284	for(cai = ai = resolv(0); cai != NULL; cai = cai->ai_next) {
	285	if((fd = socket(cai->ai_family, cai->ai_socktype, cai->ai_protocol)) < 0)
	286	continue;
	287	if(connect(fd, cai->ai_addr, cai->ai_addrlen)) {
	288	close(fd);
	289	fd = -1;
	290	continue;
	291	}
	292	break;
	293	}
	294	if(fd < 0) {
	295	if(tries++ < nolisten) {
	296	sleep(1);
	297	goto retry;
	298	}
	299	flog(LOG_ERR, "could not connect to specified TCP address: %s", strerror(errno));
	300	exit(1);
	301	}
	302	curaddr = smalloc(caddrlen = cai->ai_addrlen);
	303	memcpy(curaddr, cai->ai_addr, caddrlen);
	304	cafamily = cai->ai_family;
	305	isanon = 0;
	306	freeaddrinfo(ai);
307	return(fd);
308	} else if((unspec != NULL) \|\| (sockid != NULL)) {
309	if((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
310	flog(LOG_ERR, "could not create Unix socket: %s", strerror(errno));
311	exit(1);
312	}
313	if(unspec != NULL)
314	unpath = unspec;
315	else
316	unpath = mksockid(sockid);
317	unlink(unpath);
318	unm.sun_family = AF_UNIX;
319	strcpy(unm.sun_path, unpath);
320	if(connect(fd, (struct sockaddr *)&unm, sizeof(unm))) {
321	close(fd);
322	if(tries++ < nolisten) {
323	sleep(1);
324	goto retry;
325	}
326	flog(LOG_ERR, "could not connect to Unix socket `%s': %s", unspec, strerror(errno));
327	exit(1);
328	}
329	curaddr = smalloc(caddrlen = sizeof(unm));
330	memcpy(curaddr, &unm, sizeof(unm));
331	cafamily = AF_UNIX;
332	isanon = 0;
333	return(fd);
334	} else {
335	flog(LOG_ERR, "callfcgi: cannot use an anonymous socket without a program to start");
336	exit(1);
337	}
338	}
339
340	static int startconn(void)
341	{
342	if(*progspec) {
343	if(nolisten == 0)
344	startlisten();
345	else
346	startnolisten();
347	}
348	if(curaddr != NULL)
349	return(sconnect());
350	return(econnect());
351	}
352
353	static void killcuraddr(void)
354	{
355	if(curaddr == NULL)
356	return;
357	if(isanon) {
358	unlink(((struct sockaddr_un *)curaddr)->sun_path);
359	if(child > 0)
360	kill(child, SIGTERM);
361	}
362	free(curaddr);
363	curaddr = NULL;
364	}
365
366	static int reconn(void)
367	{
368	int fd;
369
370	if(curaddr != NULL) {
371	if((fd = sconnect()) >= 0)
372	return(fd);
373	killcuraddr();
374	}
375	return(startconn());
376	}
377
378	static off_t passdata(FILE in, FILE out)
379	{
380	size_t read;
381	off_t total;
382	char buf[8192];
383
384	total = 0;
385	while(!feof(in)) {
386	read = fread(buf, 1, sizeof(buf), in);
387	if(ferror(in))
388	return(-1);
389	if(fwrite(buf, 1, read, out) != read)
390	return(-1);
391	total += read;
392	}
393	return(total);
394	}
395
396	static void bufcatkv(struct charbuf dst, char key, char *val)
397	{
398	size_t kl, vl;
399
400	if((kl = strlen(key)) < 128) {
401	bufadd(*dst, kl);
402	} else {
403	bufadd(*dst, ((kl & 0x7f000000) >> 24) \| 0x80);
404	bufadd(*dst, (kl & 0x00ff0000) >> 16);
405	bufadd(*dst, (kl & 0x0000ff00) >> 8);
406	bufadd(*dst, kl & 0x000000ff);
407	}
408	if((vl = strlen(val)) < 128) {
409	bufadd(*dst, vl);
410	} else {
411	bufadd(*dst, ((vl & 0x7f000000) >> 24) \| 0x80);
412	bufadd(*dst, (vl & 0x00ff0000) >> 16);
413	bufadd(*dst, (vl & 0x0000ff00) >> 8);
414	bufadd(*dst, vl & 0x000000ff);
415	}
416	bufcat(*dst, key, kl);
417	bufcat(*dst, val, vl);
418	}
419
420	static void bufaddenv(struct charbuf dst, char name, char *fmt, ...)
421	{
422	va_list args;
423	char *val = NULL;
424
425	va_start(args, fmt);
426	val = vsprintf2(fmt, args);
427	va_end(args);
428	bufcatkv(dst, name, val);
429	free(val);
430	}
431
432	static char absolutify(char file)
433	{
434	static int inited = 0;
435	static char cwd[1024];
436
437	if(*file != '/') {
438	if(!inited) {
439	getcwd(cwd, sizeof(cwd));
440	inited = 1;
441	}
442	return(sprintf2("%s/%s", cwd, file));
443	}
444	return(sstrdup(file));
445	}
446
447	/* Mostly copied from callcgi. */
448	static void mkcgienv(struct hthead req, struct charbuf dst)
449	{
450	int i;
53d666ca	451	char url, pi, tmp, qp, h, p;
5c2040da FT	452
	453	bufaddenv(dst, "SERVER_SOFTWARE", "ashd/%s", VERSION);
	454	bufaddenv(dst, "GATEWAY_INTERFACE", "CGI/1.1");
	455	bufaddenv(dst, "SERVER_PROTOCOL", "%s", req->ver);
	456	bufaddenv(dst, "REQUEST_METHOD", "%s", req->method);
	457	bufaddenv(dst, "REQUEST_URI", "%s", req->url);
5c2040da FT	458	url = sstrdup(req->url);
	459	if((qp = strchr(url, '?')) != NULL)
	460	*(qp++) = 0;
	461	/* XXX: This is an ugly hack (I think), but though I can think of
	462	* several alternatives, none seem to be better. */
	463	if(*req->rest && (strlen(url) >= strlen(req->rest)) &&
	464	!strcmp(req->rest, url + strlen(url) - strlen(req->rest))) {
53d666ca FT	465	url[strlen(url) - strlen(req->rest)] = 0;
	466	}
	467	if((pi = unquoteurl(req->rest)) == NULL)
	468	pi = sstrdup(req->rest);
	469	if(!strcmp(url, "/")) {
	470	/* This seems to be normal CGI behavior, but see callcgi.c for
	471	* details. */
	472	url[0] = 0;
	473	pi = sprintf2("/%s", tmp = pi);
	474	free(tmp);
5c2040da	475	}
a80c8f02 FT	476	bufaddenv(dst, "PATH_INFO", "%s", pi);
a80c8f02 FT	477	bufaddenv(dst, "SCRIPT_NAME", "%s", url);
5c2040da	478	bufaddenv(dst, "QUERY_STRING", "%s", qp?qp:"");
53d666ca	479	free(pi);
b77076c4	480	free(url);
5c2040da FT	481	if((h = getheader(req, "Host")) != NULL)
5c2040da FT	482	bufaddenv(dst, "SERVER_NAME", "%s", h);
1af656d2 FT	483	if((h = getheader(req, "X-Ash-Server-Address")) != NULL)
1af656d2 FT	484	bufaddenv(dst, "SERVER_ADDR", "%s", h);
5c2040da FT	485	if((h = getheader(req, "X-Ash-Server-Port")) != NULL)
5c2040da FT	486	bufaddenv(dst, "SERVER_PORT", "%s", h);
381f9919 FT	487	if((h = getheader(req, "X-Ash-Remote-User")) != NULL)
381f9919 FT	488	bufaddenv(dst, "REMOTE_USER", "%s", h);
5c2040da FT	489	if(((h = getheader(req, "X-Ash-Protocol")) != NULL) && !strcmp(h, "https"))
	490	bufaddenv(dst, "HTTPS", "on");
	491	if((h = getheader(req, "X-Ash-Address")) != NULL)
	492	bufaddenv(dst, "REMOTE_ADDR", "%s", h);
1af656d2 FT	493	if((h = getheader(req, "X-Ash-Port")) != NULL)
1af656d2 FT	494	bufaddenv(dst, "REMOTE_PORT", "%s", h);
5c2040da FT	495	if((h = getheader(req, "Content-Type")) != NULL)
	496	bufaddenv(dst, "CONTENT_TYPE", "%s", h);
	497	if((h = getheader(req, "Content-Length")) != NULL)
	498	bufaddenv(dst, "CONTENT_LENGTH", "%s", h);
	499	else
	500	bufaddenv(dst, "CONTENT_LENGTH", "0");
e25bf4f9 FT	501	if((h = getheader(req, "X-Ash-File")) != NULL) {
	502	h = absolutify(h);
	503	bufaddenv(dst, "SCRIPT_FILENAME", "%s", h);
	504	free(h);
	505	}
5c2040da FT	506	for(i = 0; i < req->noheaders; i++) {
	507	h = sprintf2("HTTP_%s", req->headers[i][0]);
	508	for(p = h; *p; p++) {
	509	if(isalnum(*p))
	510	p = toupper(p);
	511	else
	512	*p = '_';
	513	}
	514	bufcatkv(dst, h, req->headers[i][1]);
	515	free(h);
	516	}
	517	}
	518
5c2040da FT	519	static struct hthead parseresp(FILE in)
	520	{
	521	struct hthead *resp;
	522	char st, p;
	523
	524	omalloc(resp);
	525	resp->ver = sstrdup("HTTP/1.1");
	526	if(parseheaders(resp, in)) {
	527	freehthead(resp);
	528	return(NULL);
	529	}
	530	if((st = getheader(resp, "Status")) != NULL) {
	531	if((p = strchr(st, ' ')) != NULL) {
	532	*(p++) = 0;
	533	resp->code = atoi(st);
	534	resp->msg = sstrdup(p);
	535	} else {
	536	resp->code = atoi(st);
62e76c42	537	resp->msg = sstrdup(httpdefstatus(resp->code));
5c2040da FT	538	}
	539	headrmheader(resp, "Status");
	540	} else if(getheader(resp, "Location")) {
	541	resp->code = 303;
	542	resp->msg = sstrdup("See Other");
	543	} else {
	544	resp->code = 200;
	545	resp->msg = sstrdup("OK");
	546	}
	547	return(resp);
	548	}
	549
	550	#define fputc2(b, f) if(fputc((b), (f)) == EOF) return(-1);
	551
	552	static int sendrec(FILE out, int type, int rid, char data, size_t dlen)
	553	{
	554	off_t off;
	555	size_t cl;
	556	int p;
	557
	558	off = 0;
	559	do {
	560	cl = min(dlen - off, 65535);
	561	p = (8 - (cl % 8)) % 8;
	562	fputc2(1, out);
	563	fputc2(type, out);
	564	fputc2((rid & 0xff00) >> 8, out);
	565	fputc2(rid & 0x00ff, out);
	566	fputc2((cl & 0xff00) >> 8, out);
	567	fputc2(cl & 0x00ff, out);
	568	fputc2(p, out);
	569	fputc2(0, out);
	570	if(fwrite(data + off, 1, cl, out) != cl)
	571	return(-1);
	572	for(; p > 0; p--)
	573	fputc2(0, out);
	574	} while((off += cl) < dlen);
	575	return(0);
	576	}
	577
5c2040da FT	578	static int recvrec(FILE in, int type, int rid, char data, size_t dlen)
5c2040da FT	579	{
e6fbc5e5 FT	580	unsigned char header[8];
e6fbc5e5 FT	581	int tl;
5c2040da	582
e6fbc5e5 FT	583	if(fread(header, 1, 8, in) != 8)
	584	return(-1);
	585	if(header[0] != 1)
5c2040da	586	return(-1);
e6fbc5e5 FT	587	*type = header[1];
	588	*rid = (header[2] << 8) \| header[3];
	589	*dlen = (header[4] << 8) \| header[5];
	590	tl = *dlen + header[6];
	591	if(header[7] != 0)
5c2040da	592	return(-1);
e6fbc5e5 FT	593	*data = smalloc(max(tl, 1));
e6fbc5e5 FT	594	if(fread(*data, 1, tl, in) != tl) {
49b8b4d2	595	free(*data);
5c2040da FT	596	return(-1);
5c2040da FT	597	}
5c2040da FT	598	return(0);
	599	}
	600
	601	static int begreq(FILE *out, int rid)
	602	{
	603	char rec[] = {0, 1, 0, 0, 0, 0, 0, 0};
	604
	605	return(sendrec(out, FCGI_BEGIN_REQUEST, rid, rec, 8));
	606	}
	607
5c2040da FT	608	static void outplex(struct muth *muth, va_list args)
	609	{
	610	vavar(FILE *, sk);
	611	struct {
	612	struct ch {
	613	FILE *s;
	614	int id;
	615	} *b;
	616	size_t s, d;
	617	} outs;
	618	int i;
	619	struct ch ch;
	620	int type, rid;
	621	char *data;
	622	size_t dlen;
	623
	624	bufinit(outs);
	625	while((ch.s = va_arg(args, FILE *)) != NULL) {
	626	ch.id = va_arg(args, int);
	627	bufadd(outs, ch);
	628	}
	629	data = NULL;
	630	while(1) {
	631	if(recvrec(sk, &type, &rid, &data, &dlen))
	632	goto out;
	633	if(rid != 1)
	634	goto out;
	635	for(i = 0; i < outs.d; i++) {
	636	if(outs.b[i].id == type) {
	637	if(outs.b[i].s != NULL) {
	638	if(dlen == 0) {
	639	fclose(outs.b[i].s);
	640	outs.b[i].s = NULL;
	641	} else {
	642	if(fwrite(data, 1, dlen, outs.b[i].s) != dlen)
	643	goto out;
	644	}
	645	}
	646	break;
	647	}
	648	}
	649	free(data);
	650	data = NULL;
	651	}
	652
	653	out:
	654	if(data != NULL)
	655	free(data);
	656	for(i = 0; i < outs.d; i++) {
	657	if(outs.b[i].s != NULL)
	658	fclose(outs.b[i].s);
	659	}
	660	buffree(outs);
	661	fclose(sk);
	662	}
	663
	664	static void errhandler(struct muth *muth, va_list args)
	665	{
	666	vavar(FILE *, in);
	667	char buf[1024];
	668	char *p;
	669
	670	bufinit(buf);
	671	while(fgets(buf, sizeof(buf), in) != NULL) {
672	p = buf + strlen(buf) - 1;
673	while((p >= buf) && (*p == '\n'))
674	*(p--) = 0;
675	if(buf[0])
676	flog(LOG_INFO, "child said: %s", buf);
677	}
678	fclose(in);
679	}
680
681	static void serve(struct muth *muth, va_list args)
682	{
683	vavar(struct hthead *, req);
684	vavar(int, fd);
685	vavar(int, sfd);
686	FILE is, os, outi, outo, erri, erro;
687	struct charbuf head;
688	struct hthead *resp;
689	size_t read;
690	char buf[8192];
691
692	sfd = reconn();
b71ad67f FT	693	is = mtstdopen(fd, 1, 60, "r+", NULL);
b71ad67f FT	694	os = mtstdopen(sfd, 1, 600, "r+", NULL);
5c2040da FT	695
	696	outi = NULL;
	697	mtiopipe(&outi, &outo); mtiopipe(&erri, &erro);
b71ad67f	698	mustart(outplex, mtstdopen(dup(sfd), 1, 600, "r+", NULL), outo, FCGI_STDOUT, erro, FCGI_STDERR, NULL);
5c2040da FT	699	mustart(errhandler, erri);
	700
	701	if(begreq(os, 1))
	702	goto out;
	703	bufinit(head);
	704	mkcgienv(req, &head);
	705	if(sendrec(os, FCGI_PARAMS, 1, head.b, head.d))
	706	goto out;
	707	if(sendrec(os, FCGI_PARAMS, 1, NULL, 0))
	708	goto out;
	709	buffree(head);
	710	if(fflush(os))
	711	goto out;
	712
	713	while(!feof(is)) {
	714	read = fread(buf, 1, sizeof(buf), is);
	715	if(ferror(is))
	716	goto out;
	717	if(sendrec(os, FCGI_STDIN, 1, buf, read))
	718	goto out;
	719	}
	720	if(sendrec(os, FCGI_STDIN, 1, NULL, 0))
	721	goto out;
	722	if(fflush(os))
	723	goto out;
	724
	725	if((resp = parseresp(outi)) == NULL)
	726	goto out;
	727	writeresp(is, resp);
	728	freehthead(resp);
	729	fputc('\n', is);
	730	if(passdata(outi, is) < 0)
	731	goto out;
	732
	733	out:
	734	freehthead(req);
	735	buffree(head);
	736	shutdown(sfd, SHUT_RDWR);
	737	if(outi != NULL)
	738	fclose(outi);
	739	fclose(is);
	740	fclose(os);
	741	}
	742
	743	static void listenloop(struct muth *muth, va_list args)
	744	{
	745	vavar(int, lfd);
	746	int fd;
	747	struct hthead *req;
	748
	749	while(1) {
	750	block(0, EV_READ, 0);
	751	if((fd = recvreq(lfd, &req)) < 0) {
	752	if(errno != 0)
	753	flog(LOG_ERR, "recvreq: %s", strerror(errno));
	754	break;
	755	}
	756	mustart(serve, req, fd);
	757	}
	758	}
	759
	760	static void sigign(int sig)
	761	{
	762	}
763
764	static void sigexit(int sig)
765	{
254eb937	766	shutdown(0, SHUT_RDWR);
5c2040da FT	767	}
	768
	769	static void usage(FILE *out)
	770	{
	771	fprintf(out, "usage: callfcgi [-h] [-N RETRIES] [-i ID] [-u UNIX-PATH] [-t [HOST:]TCP-PORT] [PROGRAM [ARGS...]]\n");
	772	}
	773
	774	int main(int argc, char **argv)
	775	{
	776	int c;
	777
	778	while((c = getopt(argc, argv, "+hN:i:u:t:")) >= 0) {
	779	switch(c) {
	780	case 'h':
	781	usage(stdout);
	782	exit(0);
	783	case 'N':
	784	nolisten = atoi(optarg);
	785	break;
	786	case 'i':
	787	sockid = optarg;
	788	break;
	789	case 'u':
	790	unspec = optarg;
	791	break;
	792	case 't':
	793	inspec = optarg;
	794	break;
	795	default:
	796	usage(stderr);
	797	exit(1);
	798	}
	799	}
	800	progspec = argv + optind;
	801	if(((sockid != NULL) + (unspec != NULL) + (inspec != NULL)) > 1) {
	802	flog(LOG_ERR, "callfcgi: at most one of -i, -u or -t may be given");
	803	exit(1);
	804	}
	805	signal(SIGCHLD, SIG_IGN);
	806	signal(SIGPIPE, sigign);
	807	signal(SIGINT, sigexit);
	808	signal(SIGTERM, sigexit);
	809	mustart(listenloop, 0);
5c2040da	810	ioloop();
d06b6013	811	killcuraddr();
5c2040da FT	812	return(0);
5c2040da FT	813	}