X-Git-Url: http://www.dolda2000.com/gitweb/?p=ashd.git;a=blobdiff_plain;f=src%2Fhttrcall.c;fp=src%2Fhttrcall.c;h=c3e7e0da6c068250efae33d954d92d416675b07a;hp=0000000000000000000000000000000000000000;hb=bcad6b0c48d516ddc920b52f06083ceaa242e1ca;hpb=589987f8218c9aa61d65f582a3b3e1bbd32bda81

diff --git a/src/httrcall.c b/src/httrcall.c
new file mode 100644
index 0000000..c3e7e0d
--- /dev/null
+++ b/src/httrcall.c
@@ -0,0 +1,144 @@
+/*
+    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/>.
+*/
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+#include <signal.h>
+#include <sys/wait.h>
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+#include <utils.h>
+#include <log.h>
+#include <req.h>
+#include <proc.h>
+#include <resp.h>
+
+struct current {
+    struct current *next, *prev;
+    pid_t pid;
+};
+
+static char **prog;
+static struct current *running = NULL;
+static int nrunning = 0, limit = 0;
+static volatile int exited;
+
+static void checkexit(int block)
+{
+    pid_t pid;
+    int st;
+    struct current *rec;
+    
+    exited = 0;
+    while((pid = waitpid(-1, &st, block?0:WNOHANG)) > 0) {
+	if(WCOREDUMP(st))
+	    flog(LOG_WARNING, "child process %i dumped core", pid);
+	for(rec = running; rec != NULL; rec = rec->next) {
+	    if(rec->pid == pid) {
+		if(rec->next)
+		    rec->next->prev = rec->prev;
+		if(rec->prev)
+		    rec->prev->next = rec->next;
+		if(rec == running)
+		    running = rec->next;
+		free(rec);
+		nrunning--;
+		break;
+	    }
+	}
+    }
+}
+
+static void serve(struct hthead *req, int fd)
+{
+    pid_t new;
+    struct current *rec;
+    
+    while((limit > 0) && (nrunning >= limit))
+	checkexit(1);
+    if((new = stdforkserve(prog, req, fd, NULL, NULL)) < 0) {
+	simpleerror(fd, 500, "Server Error", "The server appears to be overloaded.");
+	return;
+    }
+    omalloc(rec);
+    rec->pid = new;
+    rec->next = running;
+    if(running != NULL)
+	running->prev = rec;
+    running = rec;
+    nrunning++;
+}
+
+static void chldhandler(int sig)
+{
+    exited = 1;
+}
+
+static void usage(FILE *out)
+{
+    fprintf(out, "usage: httrcall [-h] [-l LIMIT] PROGRAM [ARGS...]\n");
+}
+
+int main(int argc, char **argv)
+{
+    int c;
+    struct hthead *req;
+    int fd;
+    
+    while((c = getopt(argc, argv, "+hl:")) >= 0) {
+	switch(c) {
+	case 'h':
+	    usage(stdout);
+	    exit(0);
+	case 'l':
+	    limit = atoi(optarg);
+	    break;
+	default:
+	    usage(stderr);
+	    exit(1);
+	}
+    }
+    if(argc < optind - 1) {
+	usage(stderr);
+	exit(1);
+    }
+    prog = argv + optind;
+    sigaction(SIGCHLD, &(struct sigaction) {
+	    .sa_handler = chldhandler,
+        }, NULL);
+    while(1) {
+	if(exited)
+	    checkexit(0);
+	if((fd = recvreq(0, &req)) < 0) {
+	    if(errno == EINTR)
+		continue;
+	    if(errno != 0)
+		flog(LOG_ERR, "recvreq: %s", strerror(errno));
+	    break;
+	}
+	serve(req, fd);
+	freehthead(req);
+	close(fd);
+    }
+    return(0);
+}