[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 <sys/signal.h>
#include <errno.h>

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <utils.h>
#include <req.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 startlisten(void)
{
    int i, 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) {
	dup2(fd, 0);
	for(i = 3; i < FD_SETSIZE; i++)
	    close(i);
	execvp(*progspec, progspec);
	flog(LOG_ERR, "callfcgi: %s: %s", *progspec, strerror(errno));
	_exit(127);
    }
    close(fd);
}

static void startnolisten(void)
{
    int i, fd;
    
    if((child = fork()) < 0) {
	flog(LOG_ERR, "could not fork: %s", strerror(errno));
	exit(1);
    }
    if(child == 0) {
	for(i = 3; i < FD_SETSIZE; i++)
	    close(i);
	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, *unq, *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);
    if((unq = unquoteurl(req->rest)) != NULL) {
	bufaddenv(dst, "PATH_INFO", unq);
	free(unq);
    } else {
	bufaddenv(dst, "PATH_INFO", req->rest);
    }
    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))) {
	bufaddenv(dst, "SCRIPT_NAME", "%.*s", (int)(strlen(url) - strlen(req->rest)), url);
    } else {
	bufaddenv(dst, "SCRIPT_NAME", "%s", url);
    }
    free(url);
    bufaddenv(dst, "QUERY_STRING", "%s", qp?qp:"");
    if((h = getheader(req, "Host")) != NULL)
	bufaddenv(dst, "SERVER_NAME", "%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, "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 char *defstatus(int code)
{
    if(code == 200)
	return("OK");
    else if(code == 201)
	return("Created");
    else if(code == 202)
	return("Accepted");
    else if(code == 204)
	return("No Content");
    else if(code == 300)
	return("Multiple Choices");
    else if(code == 301)
	return("Moved Permanently");
    else if(code == 302)
	return("Found");
    else if(code == 303)
	return("See Other");
    else if(code == 304)
	return("Not Modified");
    else if(code == 307)
	return("Moved Temporarily");
    else if(code == 400)
	return("Bad Request");
    else if(code == 401)
	return("Unauthorized");
    else if(code == 403)
	return("Forbidden");
    else if(code == 404)
	return("Not Found");
    else if(code == 500)
	return("Internal Server Error");
    else if(code == 501)
	return("Not Implemented");
    else if(code == 503)
	return("Service Unavailable");
    else
	return("Unknown status");
}

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

#define fgetc2(f) ({int __c__ = fgetc(f); if(__c__ == EOF) return(-1); __c__;})

static int recvrec(FILE *in, int *type, int *rid, char **data, size_t *dlen)
{
    int b1, b2, pl;
    
    if(fgetc2(in) != 1)
	return(-1);
    *type = fgetc2(in);
    b1 = fgetc2(in);
    b2 = fgetc2(in);
    *rid = (b1 << 8) | b2;
    b1 = fgetc2(in);
    b2 = fgetc2(in);
    *dlen = (b1 << 8) | b2;
    pl = fgetc2(in);
    if(fgetc2(in) != 0)
	return(-1);
    *data = smalloc(max(*dlen, 1));
    if(fread(*data, 1, *dlen, in) != *dlen) {
	free(*data);
	return(-1);
    }
    for(; pl > 0; pl--) {
	if(fgetc(in) == EOF) {
	    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 mtiopipe(FILE **read, FILE **write)
{
    int fds[2];
    
    pipe(fds);
    *read = mtstdopen(fds[0], 0, 600, "r");
    *write = mtstdopen(fds[1], 0, 600, "w");
}

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+");
    os = mtstdopen(sfd, 1, 600, "r+");
    
    outi = NULL;
    mtiopipe(&outi, &outo); mtiopipe(&erri, &erro);
    mustart(outplex, mtstdopen(dup(sfd), 1, 600, "r+"), 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)
{
    exit(0);
}

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