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> | |
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)); | |
107aa2ef | 211 | _exit(127); |
5c2040da FT |
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)); | |
107aa2ef | 229 | _exit(127); |
5c2040da FT |
230 | } |
231 | dup2(fd, 0); | |
232 | close(fd); | |
233 | execvp(*progspec, progspec); | |
234 | flog(LOG_ERR, "callfcgi: %s: %s", *progspec, strerror(errno)); | |
107aa2ef | 235 | _exit(127); |
5c2040da FT |
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; | |
09c82f9c | 444 | char *url, *unq, *qp, *h, *p; |
5c2040da FT |
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); | |
09c82f9c FT |
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 | } | |
5c2040da FT |
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 | bufaddenv(dst, "QUERY_STRING", "%s", qp?qp:""); | |
b77076c4 | 469 | free(url); |
5c2040da FT |
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); | |
381f9919 FT |
474 | if((h = getheader(req, "X-Ash-Remote-User")) != NULL) |
475 | bufaddenv(dst, "REMOTE_USER", "%s", h); | |
5c2040da FT |
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"); | |
e25bf4f9 FT |
486 | if((h = getheader(req, "X-Ash-File")) != NULL) { |
487 | h = absolutify(h); | |
488 | bufaddenv(dst, "SCRIPT_FILENAME", "%s", h); | |
489 | free(h); | |
490 | } | |
5c2040da FT |
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) { | |
49b8b4d2 | 623 | free(*data); |
5c2040da FT |
624 | return(-1); |
625 | } | |
626 | for(; pl > 0; pl--) { | |
627 | if(fgetc(in) == EOF) { | |
49b8b4d2 | 628 | free(*data); |
5c2040da FT |
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 | } |