lib: Added mblock support for kqueue.
[ashd.git] / src / httrcall.c
CommitLineData
6caaee99
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#include <stdlib.h>
20#include <stdio.h>
21#include <string.h>
22#include <unistd.h>
23#include <errno.h>
24#include <signal.h>
25#include <sys/wait.h>
26
27#ifdef HAVE_CONFIG_H
28#include <config.h>
29#endif
30#include <utils.h>
31#include <log.h>
32#include <req.h>
33#include <proc.h>
34#include <resp.h>
35
547a4189
FT
36struct current {
37 struct current *next, *prev;
38 pid_t pid;
39};
6caaee99 40
547a4189
FT
41static char **prog;
42static struct current *running = NULL;
43static int nrunning = 0, limit = 0;
44static volatile int exited;
6caaee99 45
547a4189 46static void checkexit(int block)
6caaee99
FT
47{
48 pid_t pid;
49 int st;
547a4189 50 struct current *rec;
6caaee99 51
547a4189
FT
52 exited = 0;
53 while((pid = waitpid(-1, &st, block?0:WNOHANG)) > 0) {
6caaee99
FT
54 if(WCOREDUMP(st))
55 flog(LOG_WARNING, "child process %i dumped core", pid);
547a4189
FT
56 for(rec = running; rec != NULL; rec = rec->next) {
57 if(rec->pid == pid) {
58 if(rec->next)
59 rec->next->prev = rec->prev;
60 if(rec->prev)
61 rec->prev->next = rec->next;
62 if(rec == running)
63 running = rec->next;
64 free(rec);
65 nrunning--;
66 break;
67 }
68 }
6caaee99
FT
69 }
70}
71
547a4189
FT
72static void serve(struct hthead *req, int fd)
73{
74 pid_t new;
75 struct current *rec;
76
77 while((limit > 0) && (nrunning >= limit))
78 checkexit(1);
79 if((new = stdforkserve(prog, req, fd, NULL, NULL)) < 0) {
80 simpleerror(fd, 500, "Server Error", "The server appears to be overloaded.");
81 return;
82 }
83 omalloc(rec);
84 rec->pid = new;
85 rec->next = running;
86 if(running != NULL)
87 running->prev = rec;
88 running = rec;
89 nrunning++;
90}
91
92static void chldhandler(int sig)
93{
94 exited = 1;
95}
96
6caaee99
FT
97static void usage(FILE *out)
98{
547a4189 99 fprintf(out, "usage: httrcall [-h] [-l LIMIT] PROGRAM [ARGS...]\n");
6caaee99
FT
100}
101
102int main(int argc, char **argv)
103{
104 int c;
105 struct hthead *req;
106 int fd;
107
547a4189 108 while((c = getopt(argc, argv, "+hl:")) >= 0) {
6caaee99
FT
109 switch(c) {
110 case 'h':
111 usage(stdout);
112 exit(0);
547a4189
FT
113 case 'l':
114 limit = atoi(optarg);
115 break;
6caaee99
FT
116 default:
117 usage(stderr);
118 exit(1);
119 }
120 }
121 if(argc < optind - 1) {
122 usage(stderr);
123 exit(1);
124 }
125 prog = argv + optind;
547a4189
FT
126 sigaction(SIGCHLD, &(struct sigaction) {
127 .sa_handler = chldhandler,
128 }, NULL);
6caaee99 129 while(1) {
547a4189
FT
130 if(exited)
131 checkexit(0);
6caaee99
FT
132 if((fd = recvreq(0, &req)) < 0) {
133 if(errno == EINTR)
134 continue;
135 if(errno != 0)
136 flog(LOG_ERR, "recvreq: %s", strerror(errno));
137 break;
138 }
139 serve(req, fd);
140 freehthead(req);
141 close(fd);
142 }
143 return(0);
144}