1*3f9f8982Sbenno /* $OpenBSD: slowcgi.c,v 1.19 2013/10/20 09:36:29 benno Exp $ */ 27ac15270Sflorian /* 37ac15270Sflorian * Copyright (c) 2013 David Gwynne <dlg@openbsd.org> 47ac15270Sflorian * Copyright (c) 2013 Florian Obser <florian@openbsd.org> 57ac15270Sflorian * 67ac15270Sflorian * Permission to use, copy, modify, and distribute this software for any 77ac15270Sflorian * purpose with or without fee is hereby granted, provided that the above 87ac15270Sflorian * copyright notice and this permission notice appear in all copies. 97ac15270Sflorian * 107ac15270Sflorian * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 117ac15270Sflorian * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 127ac15270Sflorian * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 137ac15270Sflorian * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 147ac15270Sflorian * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 157ac15270Sflorian * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 167ac15270Sflorian * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 177ac15270Sflorian */ 187ac15270Sflorian 197ac15270Sflorian #include <sys/types.h> 207ac15270Sflorian #include <sys/ioctl.h> 217ac15270Sflorian #include <sys/queue.h> 227ac15270Sflorian #include <sys/socket.h> 237ac15270Sflorian #include <sys/socketvar.h> 247ac15270Sflorian #include <sys/stat.h> 257ac15270Sflorian #include <sys/un.h> 267ac15270Sflorian #include <sys/wait.h> 277ac15270Sflorian #include <err.h> 2894ee156dSflorian #include <fcntl.h> 297ac15270Sflorian #include <ctype.h> 307ac15270Sflorian #include <errno.h> 317ac15270Sflorian #include <event.h> 327ac15270Sflorian #include <netdb.h> 337ac15270Sflorian #include <pwd.h> 347ac15270Sflorian #include <signal.h> 357ac15270Sflorian #include <stdio.h> 367ac15270Sflorian #include <stdlib.h> 377ac15270Sflorian #include <string.h> 387ac15270Sflorian #include <syslog.h> 397ac15270Sflorian #include <unistd.h> 407ac15270Sflorian 417ac15270Sflorian #define TIMEOUT_DEFAULT 120 427ac15270Sflorian #define SLOWCGI_USER "www" 430ee0284dSblambert 440ee0284dSblambert #define FCGI_CONTENT_SIZE 65535 450ee0284dSblambert #define FCGI_PADDING_SIZE 255 460ee0284dSblambert #define FCGI_RECORD_SIZE \ 470ee0284dSblambert (sizeof(struct fcgi_record_header) + FCGI_CONTENT_SIZE + FCGI_PADDING_SIZE) 480ee0284dSblambert 497ac15270Sflorian #define STDOUT_DONE 1 507ac15270Sflorian #define STDERR_DONE 2 517ac15270Sflorian #define SCRIPT_DONE 4 527ac15270Sflorian 537ac15270Sflorian #define FCGI_BEGIN_REQUEST 1 547ac15270Sflorian #define FCGI_ABORT_REQUEST 2 557ac15270Sflorian #define FCGI_END_REQUEST 3 567ac15270Sflorian #define FCGI_PARAMS 4 577ac15270Sflorian #define FCGI_STDIN 5 587ac15270Sflorian #define FCGI_STDOUT 6 597ac15270Sflorian #define FCGI_STDERR 7 607ac15270Sflorian #define FCGI_DATA 8 617ac15270Sflorian #define FCGI_GET_VALUES 9 627ac15270Sflorian #define FCGI_GET_VALUES_RESULT 10 637ac15270Sflorian #define FCGI_UNKNOWN_TYPE 11 647ac15270Sflorian #define FCGI_MAXTYPE (FCGI_UNKNOWN_TYPE) 657ac15270Sflorian 667ac15270Sflorian #define FCGI_REQUEST_COMPLETE 0 677ac15270Sflorian #define FCGI_CANT_MPX_CONN 1 687ac15270Sflorian #define FCGI_OVERLOADED 2 697ac15270Sflorian #define FCGI_UNKNOWN_ROLE 3 707ac15270Sflorian 71*3f9f8982Sbenno #define FD_RESERVE 5 72*3f9f8982Sbenno #define FD_NEEDED 6 73*3f9f8982Sbenno volatile int cgi_inflight = 0; 747ac15270Sflorian 757ac15270Sflorian struct listener { 767ac15270Sflorian struct event ev, pause; 777ac15270Sflorian }; 787ac15270Sflorian 797ac15270Sflorian struct env_val { 807ac15270Sflorian SLIST_ENTRY(env_val) entry; 817ac15270Sflorian char *val; 827ac15270Sflorian }; 837ac15270Sflorian SLIST_HEAD(env_head, env_val); 847ac15270Sflorian 857ac15270Sflorian struct fcgi_record_header { 867ac15270Sflorian uint8_t version; 877ac15270Sflorian uint8_t type; 887ac15270Sflorian uint16_t id; 897ac15270Sflorian uint16_t content_len; 907ac15270Sflorian uint8_t padding_len; 917ac15270Sflorian uint8_t reserved; 927ac15270Sflorian }__packed; 937ac15270Sflorian 947ac15270Sflorian struct fcgi_response { 957ac15270Sflorian TAILQ_ENTRY(fcgi_response) entry; 960ee0284dSblambert uint8_t data[FCGI_RECORD_SIZE]; 977ac15270Sflorian size_t data_pos; 987ac15270Sflorian size_t data_len; 997ac15270Sflorian }; 1007ac15270Sflorian TAILQ_HEAD(fcgi_response_head, fcgi_response); 1017ac15270Sflorian 1027ac15270Sflorian struct fcgi_stdin { 1037ac15270Sflorian TAILQ_ENTRY(fcgi_stdin) entry; 1047ac15270Sflorian uint8_t data[FCGI_RECORD_SIZE]; 1057ac15270Sflorian size_t data_pos; 1067ac15270Sflorian size_t data_len; 1077ac15270Sflorian }; 1087ac15270Sflorian TAILQ_HEAD(fcgi_stdin_head, fcgi_stdin); 1097ac15270Sflorian 110c3b13bc8Sblambert struct request { 1117ac15270Sflorian struct event ev; 1127ac15270Sflorian struct event resp_ev; 1137ac15270Sflorian struct event tmo; 1147ac15270Sflorian int fd; 1157ac15270Sflorian uint8_t buf[FCGI_RECORD_SIZE]; 1167ac15270Sflorian size_t buf_pos; 1177ac15270Sflorian size_t buf_len; 1187ac15270Sflorian struct fcgi_response_head response_head; 1197ac15270Sflorian struct fcgi_stdin_head stdin_head; 1207ac15270Sflorian uint16_t id; 1217ac15270Sflorian char script_name[MAXPATHLEN]; 1227ac15270Sflorian struct env_head env; 1237ac15270Sflorian int env_count; 1247ac15270Sflorian pid_t script_pid; 1257ac15270Sflorian int script_status; 1267ac15270Sflorian struct event script_ev; 1277ac15270Sflorian struct event script_err_ev; 1287ac15270Sflorian struct event script_stdin_ev; 1296bdc2311Sflorian int stdin_fd_closed; 1306bdc2311Sflorian int stdout_fd_closed; 1316bdc2311Sflorian int stderr_fd_closed; 1327ac15270Sflorian uint8_t script_flags; 1337ac15270Sflorian uint8_t request_started; 134*3f9f8982Sbenno int inflight_fds_accounted; 1357ac15270Sflorian }; 1367ac15270Sflorian 137c3b13bc8Sblambert struct requests { 138c3b13bc8Sblambert SLIST_ENTRY(requests) entry; 139c3b13bc8Sblambert struct request *request; 1407ac15270Sflorian }; 141c3b13bc8Sblambert SLIST_HEAD(requests_head, requests); 1427ac15270Sflorian 1437ac15270Sflorian struct slowcgi_proc { 144c3b13bc8Sblambert struct requests_head requests; 1457ac15270Sflorian struct event ev_sigchld; 1467ac15270Sflorian struct event ev_sigpipe; 1477ac15270Sflorian }; 1487ac15270Sflorian 1497ac15270Sflorian struct fcgi_begin_request_body { 1507ac15270Sflorian uint16_t role; 1517ac15270Sflorian uint8_t flags; 1527ac15270Sflorian uint8_t reserved[5]; 1537ac15270Sflorian }__packed; 1547ac15270Sflorian 15503e61f17Sblambert struct fcgi_end_request_body { 1567ac15270Sflorian uint32_t app_status; 1577ac15270Sflorian uint8_t protocol_status; 1587ac15270Sflorian uint8_t reserved[3]; 1597ac15270Sflorian }__packed; 16003e61f17Sblambert 1617ac15270Sflorian __dead void usage(void); 1620b5ee19eSblambert void slowcgi_listen(char *, gid_t); 1637ac15270Sflorian void slowcgi_paused(int, short, void *); 164*3f9f8982Sbenno int accept_reserve(int, struct sockaddr *, socklen_t *, int, 165*3f9f8982Sbenno volatile int *); 1667ac15270Sflorian void slowcgi_accept(int, short, void *); 1677ac15270Sflorian void slowcgi_request(int, short, void *); 1687ac15270Sflorian void slowcgi_response(int, short, void *); 1694404b4d5Sflorian void slowcgi_add_response(struct request *, struct fcgi_response *); 1707ac15270Sflorian void slowcgi_timeout(int, short, void *); 1717ac15270Sflorian void slowcgi_sig_handler(int, short, void *); 172c3b13bc8Sblambert size_t parse_record(uint8_t * , size_t, struct request *); 173c3b13bc8Sblambert void parse_begin_request(uint8_t *, uint16_t, struct request *, 1747ac15270Sflorian uint16_t); 175c3b13bc8Sblambert void parse_params(uint8_t *, uint16_t, struct request *, uint16_t); 176c3b13bc8Sblambert void parse_stdin(uint8_t *, uint16_t, struct request *, uint16_t); 177c3b13bc8Sblambert void exec_cgi(struct request *); 178c3b13bc8Sblambert void script_in(int, struct event *, struct request *, uint8_t); 1797ac15270Sflorian void script_std_in(int, short, void *); 1807ac15270Sflorian void script_err_in(int, short, void *); 1817ac15270Sflorian void script_out(int, short, void *); 182c3b13bc8Sblambert void create_end_record(struct request *); 18303e61f17Sblambert void dump_fcgi_record(const char *, 18403e61f17Sblambert struct fcgi_record_header *); 1857ac15270Sflorian void dump_fcgi_record_header(const char *, 1867ac15270Sflorian struct fcgi_record_header *); 18703e61f17Sblambert void dump_fcgi_begin_request_body(const char *, 18803e61f17Sblambert struct fcgi_begin_request_body *); 18903e61f17Sblambert void dump_fcgi_end_request_body(const char *, 19003e61f17Sblambert struct fcgi_end_request_body *); 191c3b13bc8Sblambert void cleanup_request(struct request *); 19203e61f17Sblambert 1937ac15270Sflorian struct loggers { 1947ac15270Sflorian void (*err)(int, const char *, ...); 1957ac15270Sflorian void (*errx)(int, const char *, ...); 1967ac15270Sflorian void (*warn)(const char *, ...); 1977ac15270Sflorian void (*warnx)(const char *, ...); 1987ac15270Sflorian void (*info)(const char *, ...); 1997ac15270Sflorian void (*debug)(const char *, ...); 2007ac15270Sflorian }; 2017ac15270Sflorian 2027ac15270Sflorian const struct loggers conslogger = { 2037ac15270Sflorian err, 2047ac15270Sflorian errx, 2057ac15270Sflorian warn, 2067ac15270Sflorian warnx, 2077ac15270Sflorian warnx, /* info */ 2087ac15270Sflorian warnx /* debug */ 2097ac15270Sflorian }; 2107ac15270Sflorian 2117ac15270Sflorian void syslog_err(int, const char *, ...); 2127ac15270Sflorian void syslog_errx(int, const char *, ...); 2137ac15270Sflorian void syslog_warn(const char *, ...); 2147ac15270Sflorian void syslog_warnx(const char *, ...); 2157ac15270Sflorian void syslog_info(const char *, ...); 2167ac15270Sflorian void syslog_debug(const char *, ...); 2177ac15270Sflorian void syslog_vstrerror(int, int, const char *, va_list); 2187ac15270Sflorian 2197ac15270Sflorian const struct loggers syslogger = { 2207ac15270Sflorian syslog_err, 2217ac15270Sflorian syslog_errx, 2227ac15270Sflorian syslog_warn, 2237ac15270Sflorian syslog_warnx, 2247ac15270Sflorian syslog_info, 2257ac15270Sflorian syslog_debug 2267ac15270Sflorian }; 2277ac15270Sflorian 2287ac15270Sflorian const struct loggers *logger = &conslogger; 2297ac15270Sflorian 2307ac15270Sflorian #define lerr(_e, _f...) logger->err((_e), _f) 2317ac15270Sflorian #define lerrx(_e, _f...) logger->errx((_e), _f) 2327ac15270Sflorian #define lwarn(_f...) logger->warn(_f) 2337ac15270Sflorian #define lwarnx(_f...) logger->warnx(_f) 2347ac15270Sflorian #define linfo(_f...) logger->info(_f) 2357ac15270Sflorian #define ldebug(_f...) logger->debug(_f) 2367ac15270Sflorian 2377ac15270Sflorian __dead void 2387ac15270Sflorian usage(void) 2397ac15270Sflorian { 2407ac15270Sflorian extern char *__progname; 2410b5ee19eSblambert fprintf(stderr, "usage: %s [-d] [-s socket]\n", __progname); 2427ac15270Sflorian exit(1); 2437ac15270Sflorian } 2447ac15270Sflorian 2457ac15270Sflorian struct timeval timeout = { TIMEOUT_DEFAULT, 0 }; 2467ac15270Sflorian struct slowcgi_proc slowcgi_proc; 2477ac15270Sflorian int debug = 0; 2487ac15270Sflorian int on = 1; 2490b5ee19eSblambert char *fcgi_socket = "/var/www/run/slowcgi.sock"; 2507ac15270Sflorian 2517ac15270Sflorian int 2527ac15270Sflorian main(int argc, char *argv[]) 2537ac15270Sflorian { 25406619349Sflorian extern char *__progname; 2557ac15270Sflorian struct passwd *pw; 2567ac15270Sflorian int c; 2577ac15270Sflorian 2580b5ee19eSblambert while ((c = getopt(argc, argv, "ds:")) != -1) { 2597ac15270Sflorian switch (c) { 2607ac15270Sflorian case 'd': 2617ac15270Sflorian debug = 1; 2627ac15270Sflorian break; 2630b5ee19eSblambert case 's': 2640b5ee19eSblambert fcgi_socket = optarg; 2650b5ee19eSblambert break; 2667ac15270Sflorian default: 2677ac15270Sflorian usage(); 2687ac15270Sflorian /* NOTREACHED */ 2697ac15270Sflorian } 2707ac15270Sflorian } 2717ac15270Sflorian 2727ac15270Sflorian if (geteuid() != 0) 2737ac15270Sflorian errx(1, "need root privileges"); 2747ac15270Sflorian 2757ac15270Sflorian pw = getpwnam(SLOWCGI_USER); 2767ac15270Sflorian if (pw == NULL) 2777ac15270Sflorian err(1, "no %s user", SLOWCGI_USER); 2787ac15270Sflorian 2797ac15270Sflorian if (!debug && daemon(1, 0) == -1) 2807ac15270Sflorian err(1, "daemon"); 2817ac15270Sflorian 28206619349Sflorian if (!debug) { 28306619349Sflorian openlog(__progname, LOG_PID|LOG_NDELAY, LOG_DAEMON); 28406619349Sflorian logger = &syslogger; 28506619349Sflorian } 28606619349Sflorian 2877ac15270Sflorian event_init(); 2887ac15270Sflorian 2890b5ee19eSblambert slowcgi_listen(fcgi_socket, pw->pw_gid); 2900b5ee19eSblambert 2917ac15270Sflorian if (chroot(pw->pw_dir) == -1) 2927ac15270Sflorian lerr(1, "chroot(%s)", pw->pw_dir); 2937ac15270Sflorian 2947ac15270Sflorian if (chdir("/") == -1) 2957ac15270Sflorian lerr(1, "chdir(%s)", pw->pw_dir); 2967ac15270Sflorian 297bde99fd8Sblambert ldebug("chroot: %s", pw->pw_dir); 298bde99fd8Sblambert 2997ac15270Sflorian if (setgroups(1, &pw->pw_gid) || 3007ac15270Sflorian setresgid(pw->pw_gid, pw->pw_gid, pw->pw_gid) || 3017ac15270Sflorian setresuid(pw->pw_uid, pw->pw_uid, pw->pw_uid)) 3027ac15270Sflorian lerr(1, "unable to revoke privs"); 3037ac15270Sflorian 304c3b13bc8Sblambert SLIST_INIT(&slowcgi_proc.requests); 3057ac15270Sflorian 3067ac15270Sflorian signal_set(&slowcgi_proc.ev_sigchld, SIGCHLD, slowcgi_sig_handler, 3077ac15270Sflorian &slowcgi_proc); 3087ac15270Sflorian signal_set(&slowcgi_proc.ev_sigpipe, SIGPIPE, slowcgi_sig_handler, 3097ac15270Sflorian &slowcgi_proc); 3107ac15270Sflorian 3117ac15270Sflorian signal_add(&slowcgi_proc.ev_sigchld, NULL); 3127ac15270Sflorian signal_add(&slowcgi_proc.ev_sigpipe, NULL); 3137ac15270Sflorian 3147ac15270Sflorian event_dispatch(); 3157ac15270Sflorian return (0); 3167ac15270Sflorian } 3177ac15270Sflorian void 3180b5ee19eSblambert slowcgi_listen(char *path, gid_t gid) 3197ac15270Sflorian { 3207ac15270Sflorian struct listener *l = NULL; 3217ac15270Sflorian struct sockaddr_un sun; 3227ac15270Sflorian mode_t old_umask, mode; 3237ac15270Sflorian int fd; 3247ac15270Sflorian 3257ac15270Sflorian if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) 3267ac15270Sflorian lerr(1, "slowcgi_listen: socket"); 32794ee156dSflorian fcntl(fd, F_SETFD, FD_CLOEXEC); 3287ac15270Sflorian 3297ac15270Sflorian bzero(&sun, sizeof(sun)); 3307ac15270Sflorian sun.sun_family = AF_UNIX; 3317ac15270Sflorian strlcpy(sun.sun_path, path, sizeof(sun.sun_path)); 3327ac15270Sflorian 3337ac15270Sflorian if (unlink(path) == -1) 3347ac15270Sflorian if (errno != ENOENT) 3357ac15270Sflorian lerr(1, "slowcgi_listen: unlink %s", path); 3367ac15270Sflorian 3377ac15270Sflorian old_umask = umask(S_IXUSR|S_IXGRP|S_IWOTH|S_IROTH|S_IXOTH); 3387ac15270Sflorian mode = S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP; 3397ac15270Sflorian 3407ac15270Sflorian if (bind(fd, (struct sockaddr *)&sun, sizeof(sun)) == -1) 3417ac15270Sflorian lerr(1,"slowcgi_listen: bind: %s", path); 3427ac15270Sflorian 3437ac15270Sflorian umask(old_umask); 3447ac15270Sflorian 3457ac15270Sflorian if (chmod(path, mode) == -1) 3467ac15270Sflorian lerr(1, "slowcgi_listen: chmod: %s", path); 3477ac15270Sflorian 3487ac15270Sflorian if (chown(path, 0, gid) == -1) 3497ac15270Sflorian lerr(1, "slowcgi_listen: chown: %s", path); 3507ac15270Sflorian 3517ac15270Sflorian if (ioctl(fd, FIONBIO, &on) == -1) 3527ac15270Sflorian lerr(1, "listener ioctl(FIONBIO)"); 3537ac15270Sflorian 3547ac15270Sflorian if (listen(fd, 5) == -1) 3557ac15270Sflorian lerr(1, "listen"); 3567ac15270Sflorian 3577ac15270Sflorian l = calloc(1, sizeof(*l)); 3587ac15270Sflorian if (l == NULL) 3597ac15270Sflorian lerr(1, "listener ev alloc"); 3607ac15270Sflorian 3617ac15270Sflorian event_set(&l->ev, fd, EV_READ | EV_PERSIST, slowcgi_accept, l); 3627ac15270Sflorian event_add(&l->ev, NULL); 3637ac15270Sflorian evtimer_set(&l->pause, slowcgi_paused, l); 364bde99fd8Sblambert 365bde99fd8Sblambert ldebug("socket: %s", path); 3667ac15270Sflorian } 3677ac15270Sflorian 3687ac15270Sflorian void 3697ac15270Sflorian slowcgi_paused(int fd, short events, void *arg) 3707ac15270Sflorian { 3717ac15270Sflorian struct listener *l = arg; 3727ac15270Sflorian event_add(&l->ev, NULL); 3737ac15270Sflorian } 3747ac15270Sflorian 375*3f9f8982Sbenno int 376*3f9f8982Sbenno accept_reserve(int sockfd, struct sockaddr *addr, socklen_t *addrlen, 377*3f9f8982Sbenno int reserve, volatile int *counter) 378*3f9f8982Sbenno { 379*3f9f8982Sbenno int ret; 380*3f9f8982Sbenno if (getdtablecount() + reserve + 381*3f9f8982Sbenno (*counter * FD_NEEDED) >= getdtablesize()) { 382*3f9f8982Sbenno ldebug("inflight fds exceded"); 383*3f9f8982Sbenno errno = EMFILE; 384*3f9f8982Sbenno return -1; 385*3f9f8982Sbenno } 386*3f9f8982Sbenno 387*3f9f8982Sbenno if ((ret = accept(sockfd, addr, addrlen)) > -1) { 388*3f9f8982Sbenno (*counter)++; 389*3f9f8982Sbenno ldebug("inflight incremented, now %d", *counter); 390*3f9f8982Sbenno } 391*3f9f8982Sbenno return ret; 392*3f9f8982Sbenno } 393*3f9f8982Sbenno 3947ac15270Sflorian void 3957ac15270Sflorian slowcgi_accept(int fd, short events, void *arg) 3967ac15270Sflorian { 3977ac15270Sflorian struct listener *l; 3987ac15270Sflorian struct sockaddr_storage ss; 3997ac15270Sflorian struct timeval pause; 400c3b13bc8Sblambert struct request *c; 401c3b13bc8Sblambert struct requests *requests; 4027ac15270Sflorian socklen_t len; 4037ac15270Sflorian int s; 4047ac15270Sflorian 4057ac15270Sflorian l = arg; 406965b2218Sblambert pause.tv_sec = 1; 407965b2218Sblambert pause.tv_usec = 0; 4087ac15270Sflorian c = NULL; 4097ac15270Sflorian 4107ac15270Sflorian len = sizeof(ss); 411*3f9f8982Sbenno if ((s = accept_reserve(fd, (struct sockaddr *)&ss, 412*3f9f8982Sbenno &len, FD_RESERVE, &cgi_inflight)) == -1) { 4137ac15270Sflorian switch (errno) { 4147ac15270Sflorian case EINTR: 4157ac15270Sflorian case EWOULDBLOCK: 4167ac15270Sflorian case ECONNABORTED: 4177ac15270Sflorian return; 4187ac15270Sflorian case EMFILE: 4197ac15270Sflorian case ENFILE: 4207ac15270Sflorian event_del(&l->ev); 4217ac15270Sflorian evtimer_add(&l->pause, &pause); 4227ac15270Sflorian return; 4237ac15270Sflorian default: 4247ac15270Sflorian lerr(1, "accept"); 4257ac15270Sflorian } 4267ac15270Sflorian } 4277ac15270Sflorian 42894ee156dSflorian fcntl(s, F_SETFD, FD_CLOEXEC); 4297ac15270Sflorian if (ioctl(s, FIONBIO, &on) == -1) 430c3b13bc8Sblambert lerr(1, "request ioctl(FIONBIO)"); 4317ac15270Sflorian 4327ac15270Sflorian c = calloc(1, sizeof(*c)); 4337ac15270Sflorian if (c == NULL) { 434c3b13bc8Sblambert lwarn("cannot calloc request"); 4357ac15270Sflorian close(s); 436*3f9f8982Sbenno cgi_inflight--; 4377ac15270Sflorian return; 4387ac15270Sflorian } 439c3b13bc8Sblambert requests = calloc(1, sizeof(*requests)); 440c3b13bc8Sblambert if (requests == NULL) { 441c3b13bc8Sblambert lwarn("cannot calloc requests"); 4427ac15270Sflorian close(s); 443*3f9f8982Sbenno cgi_inflight--; 4447ac15270Sflorian free(c); 4457ac15270Sflorian return; 4467ac15270Sflorian } 4477ac15270Sflorian c->fd = s; 4487ac15270Sflorian c->buf_pos = 0; 4497ac15270Sflorian c->buf_len = 0; 4507ac15270Sflorian c->request_started = 0; 4516bdc2311Sflorian c->stdin_fd_closed = c->stdout_fd_closed = c->stderr_fd_closed = 0; 452*3f9f8982Sbenno c->inflight_fds_accounted = 0; 4537ac15270Sflorian TAILQ_INIT(&c->response_head); 4547ac15270Sflorian TAILQ_INIT(&c->stdin_head); 4557ac15270Sflorian 4567ac15270Sflorian event_set(&c->ev, s, EV_READ | EV_PERSIST, slowcgi_request, c); 4577ac15270Sflorian event_add(&c->ev, NULL); 4584404b4d5Sflorian event_set(&c->resp_ev, s, EV_WRITE | EV_PERSIST, slowcgi_response, c); 4597ac15270Sflorian event_set(&c->tmo, s, 0, slowcgi_timeout, c); 4607ac15270Sflorian event_add(&c->tmo, &timeout); 461c3b13bc8Sblambert requests->request = c; 462c3b13bc8Sblambert SLIST_INSERT_HEAD(&slowcgi_proc.requests, requests, entry); 4637ac15270Sflorian } 4647ac15270Sflorian 4657ac15270Sflorian void 4667ac15270Sflorian slowcgi_timeout(int fd, short events, void *arg) 4677ac15270Sflorian { 468c3b13bc8Sblambert cleanup_request((struct request*) arg); 4697ac15270Sflorian } 4707ac15270Sflorian 4717ac15270Sflorian void 4727ac15270Sflorian slowcgi_sig_handler(int sig, short event, void *arg) 4737ac15270Sflorian { 474c3b13bc8Sblambert struct request *c; 475c3b13bc8Sblambert struct requests *ncs; 4767ac15270Sflorian struct slowcgi_proc *p; 4777ac15270Sflorian pid_t pid; 4787ac15270Sflorian int status; 4797ac15270Sflorian 4807ac15270Sflorian p = arg; 4817ac15270Sflorian c = NULL; 4827ac15270Sflorian 4837ac15270Sflorian switch (sig) { 4847ac15270Sflorian case SIGCHLD: 4857ac15270Sflorian pid = wait(&status); 486c3b13bc8Sblambert SLIST_FOREACH(ncs, &p->requests, entry) 487c3b13bc8Sblambert if (ncs->request->script_pid == pid) { 488c3b13bc8Sblambert c = ncs->request; 4897ac15270Sflorian break; 4907ac15270Sflorian } 491bde99fd8Sblambert if (c == NULL) { 492bde99fd8Sblambert lwarnx("caught exit of unknown child %i", pid); 493bde99fd8Sblambert break; 494bde99fd8Sblambert } 495bde99fd8Sblambert 496bde99fd8Sblambert if (WIFSIGNALED(status)) 497bde99fd8Sblambert c->script_status = WTERMSIG(status); 498bde99fd8Sblambert else 4997ac15270Sflorian c->script_status = WEXITSTATUS(status); 500bde99fd8Sblambert 5017ac15270Sflorian if (c->script_flags == (STDOUT_DONE | STDERR_DONE)) 502c3b13bc8Sblambert create_end_record(c); 5037ac15270Sflorian c->script_flags |= SCRIPT_DONE; 504bde99fd8Sblambert 505bde99fd8Sblambert ldebug("wait: %s", c->script_name); 506bde99fd8Sblambert break; 5077ac15270Sflorian case SIGPIPE: 5087ac15270Sflorian /* ignore */ 5097ac15270Sflorian break; 5107ac15270Sflorian default: 5117ac15270Sflorian lerr(1, "unexpected signal: %d", sig); 5127ac15270Sflorian 5137ac15270Sflorian } 5147ac15270Sflorian } 5157ac15270Sflorian 5167ac15270Sflorian void 5174404b4d5Sflorian slowcgi_add_response(struct request *c, struct fcgi_response *resp) 5184404b4d5Sflorian { 5194404b4d5Sflorian TAILQ_INSERT_TAIL(&c->response_head, resp, entry); 5204404b4d5Sflorian event_add(&c->resp_ev, NULL); 5214404b4d5Sflorian } 5224404b4d5Sflorian 5234404b4d5Sflorian void 5247ac15270Sflorian slowcgi_response(int fd, short events, void *arg) 5257ac15270Sflorian { 526c3b13bc8Sblambert struct request *c; 5277ac15270Sflorian struct fcgi_record_header *header; 5287ac15270Sflorian struct fcgi_response *resp; 5297ac15270Sflorian ssize_t n; 5307ac15270Sflorian 5317ac15270Sflorian c = arg; 5327ac15270Sflorian 5337ac15270Sflorian while ((resp = TAILQ_FIRST(&c->response_head))) { 5347ac15270Sflorian header = (struct fcgi_record_header*) resp->data; 5357ac15270Sflorian if (debug) 53603e61f17Sblambert dump_fcgi_record("resp ", header); 5377ac15270Sflorian 5387ac15270Sflorian n = write(fd, resp->data + resp->data_pos, resp->data_len); 5397ac15270Sflorian if (n == -1) { 5407ac15270Sflorian if (errno == EAGAIN) 5417ac15270Sflorian return; 542c3b13bc8Sblambert cleanup_request(c); 5437ac15270Sflorian return; 5447ac15270Sflorian } 5457ac15270Sflorian resp->data_pos += n; 5467ac15270Sflorian resp->data_len -= n; 5477ac15270Sflorian if (resp->data_len == 0) { 5487ac15270Sflorian TAILQ_REMOVE(&c->response_head, resp, entry); 5497ac15270Sflorian free(resp); 5507ac15270Sflorian } 5517ac15270Sflorian } 5527ac15270Sflorian 5537ac15270Sflorian if (TAILQ_EMPTY(&c->response_head)) { 5547ac15270Sflorian if (c->script_flags == (STDOUT_DONE | STDERR_DONE | 5557ac15270Sflorian SCRIPT_DONE)) 556c3b13bc8Sblambert cleanup_request(c); 5577ac15270Sflorian else 5587ac15270Sflorian event_del(&c->resp_ev); 5597ac15270Sflorian } 5607ac15270Sflorian } 5617ac15270Sflorian 5627ac15270Sflorian void 5637ac15270Sflorian slowcgi_request(int fd, short events, void *arg) 5647ac15270Sflorian { 565c3b13bc8Sblambert struct request *c; 5667ac15270Sflorian size_t n, parsed; 5677ac15270Sflorian 5687ac15270Sflorian c = arg; 5697ac15270Sflorian parsed = 0; 5707ac15270Sflorian 5717ac15270Sflorian n = read(fd, c->buf + c->buf_pos + c->buf_len, 5727ac15270Sflorian FCGI_RECORD_SIZE - c->buf_pos-c->buf_len); 5737ac15270Sflorian 5747ac15270Sflorian switch (n) { 5757ac15270Sflorian case -1: 5767ac15270Sflorian switch (errno) { 5777ac15270Sflorian case EINTR: 5787ac15270Sflorian case EAGAIN: 5797ac15270Sflorian return; 5807ac15270Sflorian default: 5817ac15270Sflorian goto fail; 5827ac15270Sflorian } 5837ac15270Sflorian break; 5847ac15270Sflorian 5857ac15270Sflorian case 0: 5867ac15270Sflorian ldebug("closed connection"); 5877ac15270Sflorian goto fail; 5887ac15270Sflorian default: 5897ac15270Sflorian break; 5907ac15270Sflorian } 5917ac15270Sflorian 5927ac15270Sflorian c->buf_len += n; 5937ac15270Sflorian 59402ad6491Sblambert /* 59502ad6491Sblambert * Parse the records as they are received. Per the FastCGI 59602ad6491Sblambert * specification, the server need only receive the FastCGI 59702ad6491Sblambert * parameter records in full; it is free to begin execution 59802ad6491Sblambert * at that point, which is what happens here. 59902ad6491Sblambert */ 6007ac15270Sflorian do { 601c3b13bc8Sblambert parsed = parse_record(c->buf + c->buf_pos, c->buf_len, c); 6027ac15270Sflorian c->buf_pos += parsed; 6037ac15270Sflorian c->buf_len -= parsed; 6047ac15270Sflorian } while (parsed > 0 && c->buf_len > 0); 6057ac15270Sflorian 606965b2218Sblambert /* Make space for further reads */ 6077ac15270Sflorian if (c->buf_len > 0) { 6087ac15270Sflorian bcopy(c->buf + c->buf_pos, c->buf, c->buf_len); 6097ac15270Sflorian c->buf_pos = 0; 6107ac15270Sflorian } 6117ac15270Sflorian return; 6127ac15270Sflorian fail: 613c3b13bc8Sblambert cleanup_request(c); 6147ac15270Sflorian } 6157ac15270Sflorian 6167ac15270Sflorian void 617c3b13bc8Sblambert parse_begin_request(uint8_t *buf, uint16_t n, struct request *c, uint16_t id) 6187ac15270Sflorian { 6197ac15270Sflorian struct fcgi_begin_request_body *b; 6207ac15270Sflorian 621965b2218Sblambert /* XXX -- FCGI_CANT_MPX_CONN */ 6227ac15270Sflorian if (c->request_started) { 6237ac15270Sflorian lwarnx("unexpected FCGI_BEGIN_REQUEST, ignoring"); 6247ac15270Sflorian return; 6257ac15270Sflorian } 6267ac15270Sflorian 6277ac15270Sflorian if (n != sizeof(struct fcgi_begin_request_body)) { 6287ac15270Sflorian lwarnx("wrong size %d != %d", n, 6297ac15270Sflorian sizeof(struct fcgi_begin_request_body)); 6307ac15270Sflorian return; 6317ac15270Sflorian } 6327ac15270Sflorian 6337ac15270Sflorian c->request_started = 1; 6347ac15270Sflorian b = (struct fcgi_begin_request_body*) buf; 6357ac15270Sflorian 6367ac15270Sflorian c->id = id; 6377ac15270Sflorian SLIST_INIT(&c->env); 6387ac15270Sflorian c->env_count = 0; 6397ac15270Sflorian } 6407ac15270Sflorian void 641c3b13bc8Sblambert parse_params(uint8_t *buf, uint16_t n, struct request *c, uint16_t id) 6427ac15270Sflorian { 6437ac15270Sflorian struct env_val *env_entry; 6447ac15270Sflorian uint32_t name_len, val_len; 6457ac15270Sflorian 6467ac15270Sflorian if (!c->request_started) { 6477ac15270Sflorian lwarnx("FCGI_PARAMS without FCGI_BEGIN_REQUEST, ignoring"); 6487ac15270Sflorian return; 6497ac15270Sflorian } 6507ac15270Sflorian 6517ac15270Sflorian if (c->id != id) { 6527ac15270Sflorian lwarnx("unexpected id, ignoring"); 6537ac15270Sflorian return; 6547ac15270Sflorian } 6557ac15270Sflorian 6567ac15270Sflorian name_len = val_len = 0; 6577ac15270Sflorian 65802ad6491Sblambert /* 65902ad6491Sblambert * If this is the last FastCGI parameter record, 66002ad6491Sblambert * begin execution of the CGI script. 66102ad6491Sblambert */ 6627ac15270Sflorian if (n == 0) { 6637ac15270Sflorian exec_cgi(c); 6647ac15270Sflorian return; 6657ac15270Sflorian } 66602ad6491Sblambert 6677ac15270Sflorian while (n > 0) { 6687ac15270Sflorian if (buf[0] >> 7 == 0) { 6697ac15270Sflorian name_len = buf[0]; 6707ac15270Sflorian n--; 6717ac15270Sflorian buf++; 6727ac15270Sflorian } else { 6737ac15270Sflorian if (n > 3) { 6747ac15270Sflorian name_len = ((buf[3] & 0x7f) << 24) + 6757ac15270Sflorian (buf[2] << 16) + (buf[1] << 8) + buf[0]; 6767ac15270Sflorian n -= 4; 6777ac15270Sflorian buf += 4; 6787ac15270Sflorian } else 6797ac15270Sflorian return; 6807ac15270Sflorian } 6817ac15270Sflorian 6827ac15270Sflorian if (n > 0) { 6837ac15270Sflorian if (buf[0] >> 7 == 0) { 6847ac15270Sflorian val_len = buf[0]; 6857ac15270Sflorian n--; 6867ac15270Sflorian buf++; 6877ac15270Sflorian } else { 6887ac15270Sflorian if (n > 3) { 6897ac15270Sflorian val_len = ((buf[3] & 0x7f) << 24) + 6907ac15270Sflorian (buf[2] << 16) + (buf[1] << 8) + 6917ac15270Sflorian buf[0]; 6927ac15270Sflorian n -= 4; 6937ac15270Sflorian buf += 4; 6947ac15270Sflorian } else 6957ac15270Sflorian return; 6967ac15270Sflorian } 6977ac15270Sflorian } 6987ac15270Sflorian if (n < name_len + val_len) 6997ac15270Sflorian return; 7007ac15270Sflorian 7017ac15270Sflorian if ((env_entry = malloc(sizeof(struct env_val))) == NULL) { 7027ac15270Sflorian lwarnx("cannot allocate env_entry"); 7037ac15270Sflorian return; 7047ac15270Sflorian } 7057ac15270Sflorian 7067ac15270Sflorian if ((env_entry->val = calloc(sizeof(char), name_len + val_len + 7077ac15270Sflorian 2)) == NULL) { 7087ac15270Sflorian lwarnx("cannot allocate env_entry->val"); 7097ac15270Sflorian free(env_entry); 7107ac15270Sflorian return; 7117ac15270Sflorian } 7127ac15270Sflorian 7137ac15270Sflorian bcopy(buf, env_entry->val, name_len); 714965b2218Sblambert buf += name_len; 715965b2218Sblambert n -= name_len; 7167ac15270Sflorian 7177ac15270Sflorian env_entry->val[name_len] = '\0'; 7187ac15270Sflorian if (val_len < MAXPATHLEN && strcmp(env_entry->val, 7197ac15270Sflorian "SCRIPT_NAME") == 0) { 7207ac15270Sflorian bcopy(buf, c->script_name, val_len); 72153847fa2Sflorian c->script_name[val_len] = '\0'; 7227ac15270Sflorian } 7237ac15270Sflorian env_entry->val[name_len] = '='; 7247ac15270Sflorian 7257ac15270Sflorian bcopy(buf, (env_entry->val) + name_len + 1, val_len); 726965b2218Sblambert buf += val_len; 727965b2218Sblambert n -= val_len; 7287ac15270Sflorian 7297ac15270Sflorian SLIST_INSERT_HEAD(&c->env, env_entry, entry); 7307ac15270Sflorian c->env_count++; 7317ac15270Sflorian } 7327ac15270Sflorian } 7337ac15270Sflorian 7347ac15270Sflorian void 735c3b13bc8Sblambert parse_stdin(uint8_t *buf, uint16_t n, struct request *c, uint16_t id) 7367ac15270Sflorian { 7377ac15270Sflorian struct fcgi_stdin *node; 7387ac15270Sflorian 7397ac15270Sflorian if (c->id != id) { 7407ac15270Sflorian lwarnx("unexpected id, ignoring"); 7417ac15270Sflorian return; 7427ac15270Sflorian } 7437ac15270Sflorian 7447ac15270Sflorian if ((node = calloc(1, sizeof(struct fcgi_stdin))) == NULL) { 7457ac15270Sflorian lwarnx("cannot calloc stdin node"); 7467ac15270Sflorian return; 7477ac15270Sflorian } 7487ac15270Sflorian 7497ac15270Sflorian bcopy(buf, node->data, n); 7507ac15270Sflorian node->data_pos = 0; 7517ac15270Sflorian node->data_len = n; 7527ac15270Sflorian 7537ac15270Sflorian TAILQ_INSERT_TAIL(&c->stdin_head, node, entry); 7547ac15270Sflorian 755a9461a24Sflorian if (event_initialized(&c->script_stdin_ev)) 7567ac15270Sflorian event_add(&c->script_stdin_ev, NULL); 7577ac15270Sflorian } 7587ac15270Sflorian 7597ac15270Sflorian size_t 760c3b13bc8Sblambert parse_record(uint8_t *buf, size_t n, struct request *c) 7617ac15270Sflorian { 7627ac15270Sflorian struct fcgi_record_header *h; 7637ac15270Sflorian 7647ac15270Sflorian if (n < sizeof(struct fcgi_record_header)) 7657ac15270Sflorian return (0); 7667ac15270Sflorian 7677ac15270Sflorian h = (struct fcgi_record_header*) buf; 7687ac15270Sflorian 7697ac15270Sflorian if (debug) 77003e61f17Sblambert dump_fcgi_record("", h); 7717ac15270Sflorian 7727ac15270Sflorian if (n < sizeof(struct fcgi_record_header) + ntohs(h->content_len) 7737ac15270Sflorian + h->padding_len) 7747ac15270Sflorian return (0); 7757ac15270Sflorian 7767ac15270Sflorian if (h->version != 1) 7777ac15270Sflorian lerrx(1, "wrong version"); 7787ac15270Sflorian 7797ac15270Sflorian switch (h->type) { 7807ac15270Sflorian case FCGI_BEGIN_REQUEST: 7817ac15270Sflorian parse_begin_request(buf + sizeof(struct fcgi_record_header), 7827ac15270Sflorian ntohs(h->content_len), c, ntohs(h->id)); 7837ac15270Sflorian break; 7847ac15270Sflorian case FCGI_PARAMS: 7857ac15270Sflorian parse_params(buf + sizeof(struct fcgi_record_header), 7867ac15270Sflorian ntohs(h->content_len), c, ntohs(h->id)); 7877ac15270Sflorian break; 7887ac15270Sflorian case FCGI_STDIN: 7897ac15270Sflorian parse_stdin(buf + sizeof(struct fcgi_record_header), 7907ac15270Sflorian ntohs(h->content_len), c, ntohs(h->id)); 7917ac15270Sflorian break; 7927ac15270Sflorian default: 7937ac15270Sflorian lwarnx("unimplemented type %d", h->type); 7947ac15270Sflorian } 7957ac15270Sflorian 7967ac15270Sflorian return (sizeof(struct fcgi_record_header) + ntohs(h->content_len) 7977ac15270Sflorian + h->padding_len); 7987ac15270Sflorian } 7997ac15270Sflorian 80002ad6491Sblambert /* 80102ad6491Sblambert * Fork a new CGI process to handle the request, translating 80202ad6491Sblambert * between FastCGI parameter records and CGI's environment variables, 80302ad6491Sblambert * as well as between the CGI process' stdin/stdout and the 80402ad6491Sblambert * corresponding FastCGI records. 80502ad6491Sblambert */ 8067ac15270Sflorian void 807c3b13bc8Sblambert exec_cgi(struct request *c) 8087ac15270Sflorian { 8097ac15270Sflorian struct env_val *env_entry; 810a9461a24Sflorian int s_in[2], s_out[2], s_err[2], i; 8117ac15270Sflorian pid_t pid; 8127ac15270Sflorian char *argv[2]; 8137ac15270Sflorian char **env; 8147ac15270Sflorian 8157ac15270Sflorian i = 0; 8167ac15270Sflorian 817a9461a24Sflorian if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, s_in) == -1) 818a9461a24Sflorian lerr(1, "socketpair"); 819a9461a24Sflorian if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, s_out) == -1) 8207ac15270Sflorian lerr(1, "socketpair"); 8217ac15270Sflorian if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, s_err) == -1) 8227ac15270Sflorian lerr(1, "socketpair"); 823*3f9f8982Sbenno cgi_inflight--; 824*3f9f8982Sbenno c->inflight_fds_accounted = 1; 8257ac15270Sflorian ldebug("fork: %s", c->script_name); 826965b2218Sblambert 8277ac15270Sflorian switch (pid = fork()) { 8287ac15270Sflorian case -1: 8297ac15270Sflorian lwarn("fork"); 8307ac15270Sflorian return; 8317ac15270Sflorian case 0: 8327ac15270Sflorian /* Child process */ 833a9461a24Sflorian close(s_in[0]); 834a9461a24Sflorian close(s_out[0]); 8357ac15270Sflorian close(s_err[0]); 836a9461a24Sflorian if (dup2(s_in[1], STDIN_FILENO) == -1) 8377ac15270Sflorian _exit(1); 838a9461a24Sflorian if (dup2(s_out[1], STDOUT_FILENO) == -1) 8397ac15270Sflorian _exit(1); 8407ac15270Sflorian if (dup2(s_err[1], STDERR_FILENO) == -1) 8417ac15270Sflorian _exit(1); 8427ac15270Sflorian argv[0] = c->script_name; 8437ac15270Sflorian argv[1] = NULL; 8447ac15270Sflorian if ((env = calloc(c->env_count + 1, sizeof(char*))) == NULL) 8457ac15270Sflorian _exit(1); 8467ac15270Sflorian SLIST_FOREACH(env_entry, &c->env, entry) 8477ac15270Sflorian env[i++] = env_entry->val; 8487ac15270Sflorian env[i++] = NULL; 8497ac15270Sflorian execve(c->script_name, argv, env); 8507ac15270Sflorian _exit(1); 8517ac15270Sflorian 8527ac15270Sflorian } 8537ac15270Sflorian /* Parent process*/ 854a9461a24Sflorian close(s_in[1]); 855a9461a24Sflorian close(s_out[1]); 8567ac15270Sflorian close(s_err[1]); 85794ee156dSflorian 858a9461a24Sflorian fcntl(s_in[0], F_SETFD, FD_CLOEXEC); 859a9461a24Sflorian fcntl(s_out[0], F_SETFD, FD_CLOEXEC); 86094ee156dSflorian fcntl(s_err[0], F_SETFD, FD_CLOEXEC); 86194ee156dSflorian 862a9461a24Sflorian if (ioctl(s_in[0], FIONBIO, &on) == -1) 863a9461a24Sflorian lerr(1, "script ioctl(FIONBIO)"); 864a9461a24Sflorian if (ioctl(s_out[0], FIONBIO, &on) == -1) 8657ac15270Sflorian lerr(1, "script ioctl(FIONBIO)"); 8667ac15270Sflorian if (ioctl(s_err[0], FIONBIO, &on) == -1) 8677ac15270Sflorian lerr(1, "script ioctl(FIONBIO)"); 8687ac15270Sflorian 8697ac15270Sflorian c->script_pid = pid; 870a9461a24Sflorian event_set(&c->script_stdin_ev, s_in[0], EV_WRITE | EV_PERSIST, 871a9461a24Sflorian script_out, c); 872a9461a24Sflorian event_add(&c->script_stdin_ev, NULL); 873a9461a24Sflorian event_set(&c->script_ev, s_out[0], EV_READ | EV_PERSIST, 874a9461a24Sflorian script_std_in, c); 8757ac15270Sflorian event_add(&c->script_ev, NULL); 8767ac15270Sflorian event_set(&c->script_err_ev, s_err[0], EV_READ | EV_PERSIST, 8777ac15270Sflorian script_err_in, c); 8787ac15270Sflorian event_add(&c->script_err_ev, NULL); 8797ac15270Sflorian } 8807ac15270Sflorian 8817ac15270Sflorian void 882c3b13bc8Sblambert create_end_record(struct request *c) 8837ac15270Sflorian { 8847ac15270Sflorian struct fcgi_response *resp; 8857ac15270Sflorian struct fcgi_record_header *header; 88603e61f17Sblambert struct fcgi_end_request_body *end_request; 8877ac15270Sflorian 8887ac15270Sflorian if ((resp = malloc(sizeof(struct fcgi_response))) == NULL) { 8897ac15270Sflorian lwarnx("cannot malloc fcgi_response"); 8907ac15270Sflorian return; 8917ac15270Sflorian } 8927ac15270Sflorian header = (struct fcgi_record_header*) resp->data; 8937ac15270Sflorian header->version = 1; 8947ac15270Sflorian header->type = FCGI_END_REQUEST; 8957ac15270Sflorian header->id = htons(c->id); 8967ac15270Sflorian header->content_len = htons(sizeof(struct 89703e61f17Sblambert fcgi_end_request_body)); 8987ac15270Sflorian header->padding_len = 0; 8997ac15270Sflorian header->reserved = 0; 9008c347892Sflorian end_request = (struct fcgi_end_request_body *) (resp->data + 9018c347892Sflorian sizeof(struct fcgi_record_header)); 9027ac15270Sflorian end_request->app_status = htonl(c->script_status); 9037ac15270Sflorian end_request->protocol_status = FCGI_REQUEST_COMPLETE; 9048c347892Sflorian end_request->reserved[0] = 0; 9058c347892Sflorian end_request->reserved[1] = 0; 9068c347892Sflorian end_request->reserved[2] = 0; 9077ac15270Sflorian resp->data_pos = 0; 90803e61f17Sblambert resp->data_len = sizeof(struct fcgi_end_request_body) + 9097ac15270Sflorian sizeof(struct fcgi_record_header); 9104404b4d5Sflorian slowcgi_add_response(c, resp); 9117ac15270Sflorian } 9127ac15270Sflorian 9137ac15270Sflorian void 914c3b13bc8Sblambert script_in(int fd, struct event *ev, struct request *c, uint8_t type) 9157ac15270Sflorian { 9167ac15270Sflorian struct fcgi_response *resp; 9177ac15270Sflorian struct fcgi_record_header *header; 9187ac15270Sflorian ssize_t n; 9197ac15270Sflorian 9207ac15270Sflorian if ((resp = malloc(sizeof(struct fcgi_response))) == NULL) { 9217ac15270Sflorian lwarnx("cannot malloc fcgi_response"); 9227ac15270Sflorian return; 9237ac15270Sflorian } 9247ac15270Sflorian header = (struct fcgi_record_header*) resp->data; 9257ac15270Sflorian header->version = 1; 9267ac15270Sflorian header->type = type; 9277ac15270Sflorian header->id = htons(c->id); 9287ac15270Sflorian header->padding_len = 0; 9297ac15270Sflorian header->reserved = 0; 9307ac15270Sflorian 9317ac15270Sflorian n = read(fd, resp->data + sizeof(struct fcgi_record_header), 9320ee0284dSblambert FCGI_CONTENT_SIZE); 9337ac15270Sflorian 9347ac15270Sflorian if (n == -1) { 9357ac15270Sflorian switch (errno) { 9367ac15270Sflorian case EINTR: 9377ac15270Sflorian case EAGAIN: 938aa48dbd7Sjasper free(resp); 9397ac15270Sflorian return; 9407ac15270Sflorian default: 9417ac15270Sflorian n = 0; /* fake empty FCGI_STD{OUT,ERR} response */ 9427ac15270Sflorian } 9437ac15270Sflorian } 9447ac15270Sflorian header->content_len = htons(n); 9457ac15270Sflorian resp->data_pos = 0; 9467ac15270Sflorian resp->data_len = n + sizeof(struct fcgi_record_header); 9474404b4d5Sflorian slowcgi_add_response(c, resp); 9487ac15270Sflorian 9497ac15270Sflorian if (n == 0) { 9507ac15270Sflorian if (type == FCGI_STDOUT) 9517ac15270Sflorian c->script_flags |= STDOUT_DONE; 9527ac15270Sflorian else 9537ac15270Sflorian c->script_flags |= STDERR_DONE; 9547ac15270Sflorian 9557ac15270Sflorian if (c->script_flags == (STDOUT_DONE | STDERR_DONE | 9567ac15270Sflorian SCRIPT_DONE)) { 957c3b13bc8Sblambert create_end_record(c); 9587ac15270Sflorian } 9597ac15270Sflorian event_del(ev); 9607ac15270Sflorian close(fd); 9616bdc2311Sflorian if (type == FCGI_STDOUT) 9626bdc2311Sflorian c->stdout_fd_closed = 1; 9636bdc2311Sflorian else 9646bdc2311Sflorian c->stderr_fd_closed = 1; 9657ac15270Sflorian } 9667ac15270Sflorian } 9677ac15270Sflorian 9687ac15270Sflorian void 9697ac15270Sflorian script_std_in(int fd, short events, void *arg) 9707ac15270Sflorian { 971c3b13bc8Sblambert struct request *c = arg; 9727ac15270Sflorian script_in(fd, &c->script_ev, c, FCGI_STDOUT); 9737ac15270Sflorian } 9747ac15270Sflorian 9757ac15270Sflorian void 9767ac15270Sflorian script_err_in(int fd, short events, void *arg) 9777ac15270Sflorian { 978c3b13bc8Sblambert struct request *c = arg; 9797ac15270Sflorian script_in(fd, &c->script_err_ev, c, FCGI_STDERR); 9807ac15270Sflorian } 9817ac15270Sflorian 9827ac15270Sflorian void 9837ac15270Sflorian script_out(int fd, short events, void *arg) 9847ac15270Sflorian { 985c3b13bc8Sblambert struct request *c; 9867ac15270Sflorian struct fcgi_stdin *node; 9877ac15270Sflorian ssize_t n; 9887ac15270Sflorian 9897ac15270Sflorian c = arg; 9907ac15270Sflorian 9917ac15270Sflorian while ((node = TAILQ_FIRST(&c->stdin_head))) { 9927ac15270Sflorian if (node->data_len == 0) { /* end of stdin marker */ 993a9461a24Sflorian close(fd); 9946bdc2311Sflorian c->stdin_fd_closed = 1; 9957ac15270Sflorian break; 9967ac15270Sflorian } 9977ac15270Sflorian n = write(fd, node->data + node->data_pos, node->data_len); 9987ac15270Sflorian if (n == -1) { 9997ac15270Sflorian if (errno == EAGAIN) 10007ac15270Sflorian return; 10017ac15270Sflorian event_del(&c->script_stdin_ev); 10027ac15270Sflorian return; 10037ac15270Sflorian } 10047ac15270Sflorian node->data_pos += n; 10057ac15270Sflorian node->data_len -= n; 10067ac15270Sflorian if (node->data_len == 0) { 10077ac15270Sflorian TAILQ_REMOVE(&c->stdin_head, node, entry); 10087ac15270Sflorian free(node); 10097ac15270Sflorian } 10107ac15270Sflorian } 1011a9461a24Sflorian event_del(&c->script_stdin_ev); 10127ac15270Sflorian } 10137ac15270Sflorian 10147ac15270Sflorian void 1015c3b13bc8Sblambert cleanup_request(struct request *c) 10167ac15270Sflorian { 10177ac15270Sflorian struct fcgi_response *resp; 10187ac15270Sflorian struct fcgi_stdin *stdin_node; 10197ac15270Sflorian struct env_val *env_entry; 1020c3b13bc8Sblambert struct requests *ncs, *tcs; 10217ac15270Sflorian 10227ac15270Sflorian evtimer_del(&c->tmo); 10237ac15270Sflorian if (event_initialized(&c->ev)) 10247ac15270Sflorian event_del(&c->ev); 10257ac15270Sflorian if (event_initialized(&c->resp_ev)) 10267ac15270Sflorian event_del(&c->resp_ev); 10277ac15270Sflorian if (event_initialized(&c->script_ev)) { 10286bdc2311Sflorian if (!c->stdout_fd_closed) 10297ac15270Sflorian close(EVENT_FD(&c->script_ev)); 10307ac15270Sflorian event_del(&c->script_ev); 10317ac15270Sflorian } 10327ac15270Sflorian if (event_initialized(&c->script_err_ev)) { 10336bdc2311Sflorian if (!c->stderr_fd_closed) 10347ac15270Sflorian close(EVENT_FD(&c->script_err_ev)); 10357ac15270Sflorian event_del(&c->script_err_ev); 10367ac15270Sflorian } 10377ac15270Sflorian if (event_initialized(&c->script_stdin_ev)) { 10386bdc2311Sflorian if (!c->stdin_fd_closed) 10397ac15270Sflorian close(EVENT_FD(&c->script_stdin_ev)); 10407ac15270Sflorian event_del(&c->script_stdin_ev); 10417ac15270Sflorian } 10427ac15270Sflorian close(c->fd); 10437ac15270Sflorian while (!SLIST_EMPTY(&c->env)) { 10447ac15270Sflorian env_entry = SLIST_FIRST(&c->env); 10457ac15270Sflorian SLIST_REMOVE_HEAD(&c->env, entry); 10467ac15270Sflorian free(env_entry->val); 10477ac15270Sflorian free(env_entry); 10487ac15270Sflorian } 10497ac15270Sflorian 10507ac15270Sflorian while ((resp = TAILQ_FIRST(&c->response_head))) { 10517ac15270Sflorian TAILQ_REMOVE(&c->response_head, resp, entry); 10527ac15270Sflorian free(resp); 10537ac15270Sflorian } 10547ac15270Sflorian while ((stdin_node = TAILQ_FIRST(&c->stdin_head))) { 10557ac15270Sflorian TAILQ_REMOVE(&c->stdin_head, stdin_node, entry); 10567ac15270Sflorian free(stdin_node); 10577ac15270Sflorian } 1058c3b13bc8Sblambert SLIST_FOREACH_SAFE(ncs, &slowcgi_proc.requests, entry, tcs) { 1059c3b13bc8Sblambert if (ncs->request == c) { 1060c3b13bc8Sblambert SLIST_REMOVE(&slowcgi_proc.requests, ncs, requests, 10617ac15270Sflorian entry); 10627ac15270Sflorian free(ncs); 10637ac15270Sflorian break; 10647ac15270Sflorian } 10657ac15270Sflorian } 1066*3f9f8982Sbenno if (! c->inflight_fds_accounted) 1067*3f9f8982Sbenno cgi_inflight--; 10687ac15270Sflorian free(c); 10697ac15270Sflorian } 10707ac15270Sflorian 1071965b2218Sblambert void 107203e61f17Sblambert dump_fcgi_record(const char *p, struct fcgi_record_header *h) 107303e61f17Sblambert { 107403e61f17Sblambert dump_fcgi_record_header(p, h); 107503e61f17Sblambert 107603e61f17Sblambert if (h->type == FCGI_BEGIN_REQUEST) 107703e61f17Sblambert dump_fcgi_begin_request_body(p, 1078d241f573Sflorian (struct fcgi_begin_request_body *)(h + 1)); 107903e61f17Sblambert else if (h->type == FCGI_END_REQUEST) 108003e61f17Sblambert dump_fcgi_end_request_body(p, 1081d241f573Sflorian (struct fcgi_end_request_body *)(h + 1)); 108203e61f17Sblambert } 108303e61f17Sblambert 108403e61f17Sblambert void 1085965b2218Sblambert dump_fcgi_record_header(const char* p, struct fcgi_record_header *h) 10867ac15270Sflorian { 10877ac15270Sflorian ldebug("%sversion: %d", p, h->version); 10887ac15270Sflorian ldebug("%stype: %d", p, h->type); 10897ac15270Sflorian ldebug("%srequestId: %d", p, ntohs(h->id)); 10907ac15270Sflorian ldebug("%scontentLength: %d", p, ntohs(h->content_len)); 10917ac15270Sflorian ldebug("%spaddingLength: %d", p, h->padding_len); 10927ac15270Sflorian ldebug("%sreserved: %d", p, h->reserved); 10937ac15270Sflorian } 10947ac15270Sflorian 10957ac15270Sflorian void 109603e61f17Sblambert dump_fcgi_begin_request_body(const char *p, struct fcgi_begin_request_body *b) 109703e61f17Sblambert { 109803e61f17Sblambert ldebug("%srole %d", p, ntohs(b->role)); 109903e61f17Sblambert ldebug("%sflags %d", p, b->flags); 110003e61f17Sblambert } 110103e61f17Sblambert 110203e61f17Sblambert void 110303e61f17Sblambert dump_fcgi_end_request_body(const char *p, struct fcgi_end_request_body *b) 110403e61f17Sblambert { 110503e61f17Sblambert ldebug("%sappStatus: %d", p, ntohl(b->app_status)); 110603e61f17Sblambert ldebug("%sprotocolStatus: %d", p, b->protocol_status); 110703e61f17Sblambert } 110803e61f17Sblambert 110903e61f17Sblambert void 11107ac15270Sflorian syslog_vstrerror(int e, int priority, const char *fmt, va_list ap) 11117ac15270Sflorian { 11127ac15270Sflorian char *s; 11137ac15270Sflorian 11147ac15270Sflorian if (vasprintf(&s, fmt, ap) == -1) { 11157ac15270Sflorian syslog(LOG_EMERG, "unable to alloc in syslog_vstrerror"); 11167ac15270Sflorian exit(1); 11177ac15270Sflorian } 11187ac15270Sflorian syslog(priority, "%s: %s", s, strerror(e)); 11197ac15270Sflorian free(s); 11207ac15270Sflorian } 11217ac15270Sflorian 11227ac15270Sflorian void 11237ac15270Sflorian syslog_err(int ecode, const char *fmt, ...) 11247ac15270Sflorian { 11257ac15270Sflorian va_list ap; 11267ac15270Sflorian 11277ac15270Sflorian va_start(ap, fmt); 11287ac15270Sflorian syslog_vstrerror(errno, LOG_EMERG, fmt, ap); 11297ac15270Sflorian va_end(ap); 11307ac15270Sflorian exit(ecode); 11317ac15270Sflorian } 11327ac15270Sflorian 11337ac15270Sflorian void 11347ac15270Sflorian syslog_errx(int ecode, const char *fmt, ...) 11357ac15270Sflorian { 11367ac15270Sflorian va_list ap; 11377ac15270Sflorian 11387ac15270Sflorian va_start(ap, fmt); 11397ac15270Sflorian vsyslog(LOG_WARNING, fmt, ap); 11407ac15270Sflorian va_end(ap); 11417ac15270Sflorian exit(ecode); 11427ac15270Sflorian } 11437ac15270Sflorian 11447ac15270Sflorian void 11457ac15270Sflorian syslog_warn(const char *fmt, ...) 11467ac15270Sflorian { 11477ac15270Sflorian va_list ap; 11487ac15270Sflorian 11497ac15270Sflorian va_start(ap, fmt); 11507ac15270Sflorian syslog_vstrerror(errno, LOG_WARNING, fmt, ap); 11517ac15270Sflorian va_end(ap); 11527ac15270Sflorian } 11537ac15270Sflorian 11547ac15270Sflorian void 11557ac15270Sflorian syslog_warnx(const char *fmt, ...) 11567ac15270Sflorian { 11577ac15270Sflorian va_list ap; 11587ac15270Sflorian 11597ac15270Sflorian va_start(ap, fmt); 11607ac15270Sflorian vsyslog(LOG_WARNING, fmt, ap); 11617ac15270Sflorian va_end(ap); 11627ac15270Sflorian } 11637ac15270Sflorian 11647ac15270Sflorian void 11657ac15270Sflorian syslog_info(const char *fmt, ...) 11667ac15270Sflorian { 11677ac15270Sflorian va_list ap; 11687ac15270Sflorian 11697ac15270Sflorian va_start(ap, fmt); 11707ac15270Sflorian vsyslog(LOG_INFO, fmt, ap); 11717ac15270Sflorian va_end(ap); 11727ac15270Sflorian } 11737ac15270Sflorian 11747ac15270Sflorian void 11757ac15270Sflorian syslog_debug(const char *fmt, ...) 11767ac15270Sflorian { 11777ac15270Sflorian va_list ap; 11787ac15270Sflorian 11797ac15270Sflorian if (!debug) 11807ac15270Sflorian return; 11817ac15270Sflorian 11827ac15270Sflorian va_start(ap, fmt); 11837ac15270Sflorian vsyslog(LOG_DEBUG, fmt, ap); 11847ac15270Sflorian va_end(ap); 11857ac15270Sflorian } 1186