Slightly more sensible argument parsing for htparser.
[ashd.git] / src / callcgi.c
CommitLineData
8cc51634
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 <string.h>
21#include <stdio.h>
22#include <unistd.h>
23#include <errno.h>
24#include <ctype.h>
25#include <signal.h>
26
27#ifdef HAVE_CONFIG_H
28#include <config.h>
29#endif
30#include <utils.h>
31#include <log.h>
32
33static char **environ;
34
35static void passdata(FILE *in, FILE *out)
36{
37 int ret;
38 char *buf;
39
40 buf = smalloc(65536);
41 while(!feof(in)) {
42 ret = fread(buf, 1, 65536, in);
43 if(ferror(in)) {
44 flog(LOG_ERR, "sendfile: could not read input: %s", strerror(errno));
45 break;
46 }
47 if(fwrite(buf, 1, ret, out) != ret) {
48 flog(LOG_ERR, "sendfile: could not write output: %s", strerror(errno));
49 break;
50 }
51 }
52 free(buf);
53}
54
bac8c1f9
FT
55static char *absolutify(char *file)
56{
57 char cwd[1024];
58
59 if(*file != '/') {
60 getcwd(cwd, sizeof(cwd));
61 return(sprintf2("%s/%s", cwd, file));
62 }
63 return(sstrdup(file));
64}
65
341b3f0b 66static void forkchild(int inpath, char *prog, char *file, char *method, char *url, char *rest, int *infd, int *outfd)
8cc51634
FT
67{
68 int i;
69 char *qp, **env;
70 int inp[2], outp[2];
71 pid_t pid;
72
73 pipe(inp);
74 pipe(outp);
75 if((pid = fork()) < 0) {
76 flog(LOG_ERR, "callcgi: could not fork");
77 exit(1);
78 }
79 if(pid == 0) {
80 close(inp[1]);
81 close(outp[0]);
82 dup2(inp[0], 0);
83 dup2(outp[1], 1);
84 for(i = 3; i < FD_SETSIZE; i++)
85 close(i);
86 if((qp = strchr(url, '?')) != NULL)
87 *(qp++) = 0;
88 /*
89 * XXX: Currently missing:
90 * SERVER_NAME (Partially)
91 * SERVER_PORT
92 */
93 putenv(sprintf2("SERVER_SOFTWARE=ashd/%s", VERSION));
94 putenv("GATEWAY_INTERFACE=CGI/1.1");
95 if(getenv("HTTP_VERSION"))
96 putenv(sprintf2("SERVER_PROTOCOL=HTTP/%s", getenv("HTTP_VERSION")));
97 putenv(sprintf2("REQUEST_METHOD=%s", method));
98 putenv(sprintf2("PATH_INFO=%s", rest));
99 putenv(sprintf2("SCRIPT_NAME=%s", url));
100 putenv(sprintf2("QUERY_STRING=%s", qp?qp:""));
101 if(getenv("REQ_HOST"))
102 putenv(sprintf2("SERVER_NAME=%s", getenv("REQ_HOST")));
103 if(getenv("REQ_X_ASH_ADDRESS"))
104 putenv(sprintf2("REMOTE_ADDR=%s", getenv("REQ_X_ASH_ADDRESS")));
105 if(getenv("REQ_CONTENT_TYPE"))
106 putenv(sprintf2("CONTENT_TYPE=%s", getenv("REQ_CONTENT_TYPE")));
107 if(getenv("REQ_CONTENT_LENGTH"))
108 putenv(sprintf2("CONTENT_LENGTH=%s", getenv("REQ_CONTENT_LENGTH")));
109 for(env = environ; *env; env++) {
110 if(!strncmp(*env, "REQ_", 4))
111 putenv(sprintf2("HTTP_%s", (*env) + 4));
112 }
113 /*
114 * This is (understandably) missing from the CGI
115 * specification, but PHP seems to require it.
116 */
bac8c1f9 117 putenv(sprintf2("SCRIPT_FILENAME=%s", absolutify(file)));
341b3f0b
FT
118 if(inpath)
119 execlp(prog, prog, file, NULL);
120 else
121 execl(prog, prog, file, NULL);
8cc51634
FT
122 exit(127);
123 }
124 close(inp[0]);
125 close(outp[1]);
126 *infd = inp[1];
127 *outfd = outp[0];
128}
129
130static void trim(struct charbuf *buf)
131{
132 char *p;
133
134 for(p = buf->b; (p - buf->b < buf->d) && isspace(*p); p++);
135 memmove(buf->b, p, buf->d -= (p - buf->b));
136 for(p = buf->b + buf->d - 1; (p > buf->b) && isspace(*p); p--, buf->d--);
137}
138
139static char **parseheaders(FILE *s)
140{
141 int c, state;
142 struct charvbuf hbuf;
143 struct charbuf buf;
144
145 bufinit(hbuf);
146 bufinit(buf);
147 state = 0;
148 while(1) {
149 c = fgetc(s);
150 again:
151 if(state == 0) {
152 if(c == '\r') {
153 } else if(c == '\n') {
154 break;
155 } else if(c == EOF) {
156 goto fail;
157 } else {
158 state = 1;
159 goto again;
160 }
161 } else if(state == 1) {
162 if(c == ':') {
163 trim(&buf);
164 bufadd(buf, 0);
165 bufadd(hbuf, buf.b);
166 bufinit(buf);
167 state = 2;
168 } else if(c == '\r') {
169 } else if(c == '\n') {
170 goto fail;
171 } else if(c == EOF) {
172 goto fail;
173 } else {
174 bufadd(buf, c);
175 }
176 } else if(state == 2) {
177 if(c == '\r') {
178 } else if(c == '\n') {
179 trim(&buf);
180 bufadd(buf, 0);
181 bufadd(hbuf, buf.b);
182 bufinit(buf);
183 state = 0;
184 } else if(c == EOF) {
185 goto fail;
186 } else {
187 bufadd(buf, c);
188 }
189 }
190 }
191 bufadd(hbuf, NULL);
192 return(hbuf.b);
193
194fail:
195 buffree(hbuf);
196 buffree(buf);
197 return(NULL);
198}
199
2e535ab0
FT
200static char *defstatus(int code)
201{
202 if(code == 200)
203 return("OK");
204 else if(code == 201)
205 return("Created");
206 else if(code == 202)
207 return("Accepted");
208 else if(code == 204)
209 return("No Content");
210 else if(code == 300)
211 return("Multiple Choices");
212 else if(code == 301)
213 return("Moved Permanently");
214 else if(code == 302)
215 return("Found");
216 else if(code == 303)
217 return("See Other");
218 else if(code == 304)
219 return("Not Modified");
220 else if(code == 307)
221 return("Moved Temporarily");
222 else if(code == 400)
223 return("Bad Request");
224 else if(code == 401)
225 return("Unauthorized");
226 else if(code == 403)
227 return("Forbidden");
228 else if(code == 404)
229 return("Not Found");
230 else if(code == 500)
231 return("Internal Server Error");
232 else if(code == 501)
233 return("Not Implemented");
234 else if(code == 503)
235 return("Service Unavailable");
236 else
237 return("Unknown status");
238}
239
8cc51634
FT
240static void sendstatus(char **headers, FILE *out)
241{
242 char **hp;
243 char *status, *location;
244
245 hp = headers;
246 status = location = NULL;
247 while(*hp) {
248 if(!strcasecmp(hp[0], "status")) {
249 status = hp[1];
250 /* Clear this header, so that it is not transmitted by sendheaders. */
251 **hp = 0;
252 } else if(!strcasecmp(hp[0], "location")) {
253 location = hp[1];
f9255ddd 254 hp += 2;
8cc51634
FT
255 } else {
256 hp += 2;
257 }
258 }
259 if(status) {
2e535ab0 260 if(strchr(status, ' '))
81cfca6c 261 fprintf(out, "HTTP/1.1 %s\n", status);
2e535ab0 262 else
81cfca6c 263 fprintf(out, "HTTP/1.1 %i %s\n", atoi(status), defstatus(atoi(status)));
8cc51634 264 } else if(location) {
81cfca6c 265 fprintf(out, "HTTP/1.1 303 See Other\n");
8cc51634 266 } else {
81cfca6c 267 fprintf(out, "HTTP/1.1 200 OK\n");
8cc51634
FT
268 }
269}
270
271static void sendheaders(char **headers, FILE *out)
272{
273 while(*headers) {
274 if(**headers)
81cfca6c 275 fprintf(out, "%s: %s\n", headers[0], headers[1]);
8cc51634
FT
276 headers += 2;
277 }
278}
279
341b3f0b
FT
280static void usage(void)
281{
282 flog(LOG_ERR, "usage: callcgi [-p PROGRAM] METHOD URL REST");
283}
284
8cc51634
FT
285int main(int argc, char **argv, char **envp)
286{
341b3f0b
FT
287 int c;
288 char *file, *prog;
289 int inpath;
290 int infd, outfd;
291 FILE *in, *out;
8cc51634
FT
292 char **headers;
293
294 environ = envp;
295 signal(SIGPIPE, SIG_IGN);
341b3f0b
FT
296
297 prog = NULL;
298 inpath = 0;
299 while((c = getopt(argc, argv, "p:")) >= 0) {
300 switch(c) {
301 case 'p':
302 prog = optarg;
303 inpath = 1;
304 break;
305 default:
306 usage();
307 exit(1);
308 }
309 }
310
311 if(argc - optind < 3) {
312 usage();
8cc51634
FT
313 exit(1);
314 }
315 if((file = getenv("REQ_X_ASH_FILE")) == NULL) {
316 flog(LOG_ERR, "callcgi: needs to be called with the X-Ash-File header");
317 exit(1);
318 }
341b3f0b
FT
319 if(prog == NULL)
320 prog = file;
321 forkchild(inpath, prog, file, argv[optind], argv[optind + 1], argv[optind + 2], &infd, &outfd);
322 in = fdopen(infd, "w");
323 passdata(stdin, in);
324 fclose(in);
325 out = fdopen(outfd, "r");
326 if((headers = parseheaders(out)) == NULL) {
8cc51634
FT
327 flog(LOG_WARNING, "CGI handler returned invalid headers");
328 exit(1);
329 }
330 sendstatus(headers, stdout);
331 sendheaders(headers, stdout);
81cfca6c 332 printf("\n");
341b3f0b 333 passdata(out, stdout);
8cc51634
FT
334 return(0);
335}