xref: /openbsd/usr.sbin/slowcgi/slowcgi.c (revision 8c347892)
1*8c347892Sflorian /*	$OpenBSD: slowcgi.c,v 1.13 2013/10/18 14:43:21 florian 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 
717ac15270Sflorian 
727ac15270Sflorian struct listener {
737ac15270Sflorian 	struct event	ev, pause;
747ac15270Sflorian };
757ac15270Sflorian 
767ac15270Sflorian struct env_val {
777ac15270Sflorian 	SLIST_ENTRY(env_val)	 entry;
787ac15270Sflorian 	char			*val;
797ac15270Sflorian };
807ac15270Sflorian SLIST_HEAD(env_head, env_val);
817ac15270Sflorian 
827ac15270Sflorian struct fcgi_record_header {
837ac15270Sflorian 	uint8_t		version;
847ac15270Sflorian 	uint8_t		type;
857ac15270Sflorian 	uint16_t	id;
867ac15270Sflorian 	uint16_t	content_len;
877ac15270Sflorian 	uint8_t		padding_len;
887ac15270Sflorian 	uint8_t		reserved;
897ac15270Sflorian }__packed;
907ac15270Sflorian 
917ac15270Sflorian struct fcgi_response {
927ac15270Sflorian 	TAILQ_ENTRY(fcgi_response)	entry;
930ee0284dSblambert 	uint8_t				data[FCGI_RECORD_SIZE];
947ac15270Sflorian 	size_t				data_pos;
957ac15270Sflorian 	size_t				data_len;
967ac15270Sflorian };
977ac15270Sflorian TAILQ_HEAD(fcgi_response_head, fcgi_response);
987ac15270Sflorian 
997ac15270Sflorian struct fcgi_stdin {
1007ac15270Sflorian 	TAILQ_ENTRY(fcgi_stdin)	entry;
1017ac15270Sflorian 	uint8_t			data[FCGI_RECORD_SIZE];
1027ac15270Sflorian 	size_t			data_pos;
1037ac15270Sflorian 	size_t			data_len;
1047ac15270Sflorian };
1057ac15270Sflorian TAILQ_HEAD(fcgi_stdin_head, fcgi_stdin);
1067ac15270Sflorian 
107c3b13bc8Sblambert struct request {
1087ac15270Sflorian 	struct event			ev;
1097ac15270Sflorian 	struct event			resp_ev;
1107ac15270Sflorian 	struct event			tmo;
1117ac15270Sflorian 	int				fd;
1127ac15270Sflorian 	uint8_t				buf[FCGI_RECORD_SIZE];
1137ac15270Sflorian 	size_t				buf_pos;
1147ac15270Sflorian 	size_t				buf_len;
1157ac15270Sflorian 	struct fcgi_response_head	response_head;
1167ac15270Sflorian 	struct fcgi_stdin_head		stdin_head;
1177ac15270Sflorian 	uint16_t			id;
1187ac15270Sflorian 	char				script_name[MAXPATHLEN];
1197ac15270Sflorian 	struct env_head			env;
1207ac15270Sflorian 	int				env_count;
1217ac15270Sflorian 	pid_t				script_pid;
1227ac15270Sflorian 	int				script_status;
1237ac15270Sflorian 	struct event			script_ev;
1247ac15270Sflorian 	struct event			script_err_ev;
1257ac15270Sflorian 	struct event			script_stdin_ev;
1267ac15270Sflorian 	uint8_t				script_flags;
1277ac15270Sflorian 	uint8_t				request_started;
1287ac15270Sflorian };
1297ac15270Sflorian 
130c3b13bc8Sblambert struct requests {
131c3b13bc8Sblambert 	SLIST_ENTRY(requests)	 entry;
132c3b13bc8Sblambert 	struct request		*request;
1337ac15270Sflorian };
134c3b13bc8Sblambert SLIST_HEAD(requests_head, requests);
1357ac15270Sflorian 
1367ac15270Sflorian struct slowcgi_proc {
137c3b13bc8Sblambert 	struct requests_head	requests;
1387ac15270Sflorian 	struct event		ev_sigchld;
1397ac15270Sflorian 	struct event		ev_sigpipe;
1407ac15270Sflorian };
1417ac15270Sflorian 
1427ac15270Sflorian struct fcgi_begin_request_body {
1437ac15270Sflorian 	uint16_t	role;
1447ac15270Sflorian 	uint8_t		flags;
1457ac15270Sflorian 	uint8_t		reserved[5];
1467ac15270Sflorian }__packed;
1477ac15270Sflorian 
14803e61f17Sblambert struct fcgi_end_request_body {
1497ac15270Sflorian 	uint32_t	app_status;
1507ac15270Sflorian 	uint8_t		protocol_status;
1517ac15270Sflorian 	uint8_t		reserved[3];
1527ac15270Sflorian }__packed;
15303e61f17Sblambert 
1547ac15270Sflorian __dead void	usage(void);
1550b5ee19eSblambert void		slowcgi_listen(char *, gid_t);
1567ac15270Sflorian void		slowcgi_paused(int, short, void *);
1577ac15270Sflorian void		slowcgi_accept(int, short, void *);
1587ac15270Sflorian void		slowcgi_request(int, short, void *);
1597ac15270Sflorian void		slowcgi_response(int, short, void *);
1607ac15270Sflorian void		slowcgi_timeout(int, short, void *);
1617ac15270Sflorian void		slowcgi_sig_handler(int, short, void *);
162c3b13bc8Sblambert size_t		parse_record(uint8_t * , size_t, struct request *);
163c3b13bc8Sblambert void		parse_begin_request(uint8_t *, uint16_t, struct request *,
1647ac15270Sflorian 		    uint16_t);
165c3b13bc8Sblambert void		parse_params(uint8_t *, uint16_t, struct request *, uint16_t);
166c3b13bc8Sblambert void		parse_stdin(uint8_t *, uint16_t, struct request *, uint16_t);
167c3b13bc8Sblambert void		exec_cgi(struct request *);
168c3b13bc8Sblambert void		script_in(int, struct event *, struct request *, uint8_t);
1697ac15270Sflorian void		script_std_in(int, short, void *);
1707ac15270Sflorian void		script_err_in(int, short, void *);
1717ac15270Sflorian void		script_out(int, short, void *);
172c3b13bc8Sblambert void		create_end_record(struct request *);
17303e61f17Sblambert void		dump_fcgi_record(const char *,
17403e61f17Sblambert 		    struct fcgi_record_header *);
1757ac15270Sflorian void		dump_fcgi_record_header(const char *,
1767ac15270Sflorian 		    struct fcgi_record_header *);
17703e61f17Sblambert void		dump_fcgi_begin_request_body(const char *,
17803e61f17Sblambert 		    struct fcgi_begin_request_body *);
17903e61f17Sblambert void		dump_fcgi_end_request_body(const char *,
18003e61f17Sblambert 		    struct fcgi_end_request_body *);
181c3b13bc8Sblambert void		cleanup_request(struct request *);
18203e61f17Sblambert 
1837ac15270Sflorian struct loggers {
1847ac15270Sflorian 	void (*err)(int, const char *, ...);
1857ac15270Sflorian 	void (*errx)(int, const char *, ...);
1867ac15270Sflorian 	void (*warn)(const char *, ...);
1877ac15270Sflorian 	void (*warnx)(const char *, ...);
1887ac15270Sflorian 	void (*info)(const char *, ...);
1897ac15270Sflorian 	void (*debug)(const char *, ...);
1907ac15270Sflorian };
1917ac15270Sflorian 
1927ac15270Sflorian const struct loggers conslogger = {
1937ac15270Sflorian 	err,
1947ac15270Sflorian 	errx,
1957ac15270Sflorian 	warn,
1967ac15270Sflorian 	warnx,
1977ac15270Sflorian 	warnx, /* info */
1987ac15270Sflorian 	warnx /* debug */
1997ac15270Sflorian };
2007ac15270Sflorian 
2017ac15270Sflorian void	syslog_err(int, const char *, ...);
2027ac15270Sflorian void	syslog_errx(int, const char *, ...);
2037ac15270Sflorian void	syslog_warn(const char *, ...);
2047ac15270Sflorian void	syslog_warnx(const char *, ...);
2057ac15270Sflorian void	syslog_info(const char *, ...);
2067ac15270Sflorian void	syslog_debug(const char *, ...);
2077ac15270Sflorian void	syslog_vstrerror(int, int, const char *, va_list);
2087ac15270Sflorian 
2097ac15270Sflorian const struct loggers syslogger = {
2107ac15270Sflorian 	syslog_err,
2117ac15270Sflorian 	syslog_errx,
2127ac15270Sflorian 	syslog_warn,
2137ac15270Sflorian 	syslog_warnx,
2147ac15270Sflorian 	syslog_info,
2157ac15270Sflorian 	syslog_debug
2167ac15270Sflorian };
2177ac15270Sflorian 
2187ac15270Sflorian const struct loggers *logger = &conslogger;
2197ac15270Sflorian 
2207ac15270Sflorian #define lerr(_e, _f...) logger->err((_e), _f)
2217ac15270Sflorian #define lerrx(_e, _f...) logger->errx((_e), _f)
2227ac15270Sflorian #define lwarn(_f...) logger->warn(_f)
2237ac15270Sflorian #define lwarnx(_f...) logger->warnx(_f)
2247ac15270Sflorian #define linfo(_f...) logger->info(_f)
2257ac15270Sflorian #define ldebug(_f...) logger->debug(_f)
2267ac15270Sflorian 
2277ac15270Sflorian __dead void
2287ac15270Sflorian usage(void)
2297ac15270Sflorian {
2307ac15270Sflorian 	extern char *__progname;
2310b5ee19eSblambert 	fprintf(stderr, "usage: %s [-d] [-s socket]\n", __progname);
2327ac15270Sflorian 	exit(1);
2337ac15270Sflorian }
2347ac15270Sflorian 
2357ac15270Sflorian struct timeval		timeout = { TIMEOUT_DEFAULT, 0 };
2367ac15270Sflorian struct slowcgi_proc	slowcgi_proc;
2377ac15270Sflorian int			debug = 0;
2387ac15270Sflorian int			on = 1;
2390b5ee19eSblambert char 			*fcgi_socket = "/var/www/run/slowcgi.sock";
2407ac15270Sflorian 
2417ac15270Sflorian int
2427ac15270Sflorian main(int argc, char *argv[])
2437ac15270Sflorian {
2447ac15270Sflorian 	struct passwd	*pw;
2457ac15270Sflorian 	int		 c;
2467ac15270Sflorian 
2470b5ee19eSblambert 	while ((c = getopt(argc, argv, "ds:")) != -1) {
2487ac15270Sflorian 		switch (c) {
2497ac15270Sflorian 		case 'd':
2507ac15270Sflorian 			debug = 1;
2517ac15270Sflorian 			break;
2520b5ee19eSblambert 		case 's':
2530b5ee19eSblambert 			fcgi_socket = optarg;
2540b5ee19eSblambert 			break;
2557ac15270Sflorian 		default:
2567ac15270Sflorian 			usage();
2577ac15270Sflorian 			/* NOTREACHED */
2587ac15270Sflorian 		}
2597ac15270Sflorian 	}
2607ac15270Sflorian 
2617ac15270Sflorian 	if (geteuid() != 0)
2627ac15270Sflorian 		errx(1, "need root privileges");
2637ac15270Sflorian 
2647ac15270Sflorian 	pw = getpwnam(SLOWCGI_USER);
2657ac15270Sflorian 	if (pw == NULL)
2667ac15270Sflorian 		err(1, "no %s user", SLOWCGI_USER);
2677ac15270Sflorian 
2687ac15270Sflorian 	if (!debug && daemon(1, 0) == -1)
2697ac15270Sflorian 		err(1, "daemon");
2707ac15270Sflorian 
2717ac15270Sflorian 	event_init();
2727ac15270Sflorian 
2730b5ee19eSblambert 	slowcgi_listen(fcgi_socket, pw->pw_gid);
2740b5ee19eSblambert 
2757ac15270Sflorian 	if (chroot(pw->pw_dir) == -1)
2767ac15270Sflorian 		lerr(1, "chroot(%s)", pw->pw_dir);
2777ac15270Sflorian 
2787ac15270Sflorian 	if (chdir("/") == -1)
2797ac15270Sflorian 		lerr(1, "chdir(%s)", pw->pw_dir);
2807ac15270Sflorian 
281bde99fd8Sblambert 	ldebug("chroot: %s", pw->pw_dir);
282bde99fd8Sblambert 
2837ac15270Sflorian 	if (setgroups(1, &pw->pw_gid) ||
2847ac15270Sflorian 	    setresgid(pw->pw_gid, pw->pw_gid, pw->pw_gid) ||
2857ac15270Sflorian 	    setresuid(pw->pw_uid, pw->pw_uid, pw->pw_uid))
2867ac15270Sflorian 		lerr(1, "unable to revoke privs");
2877ac15270Sflorian 
288c3b13bc8Sblambert 	SLIST_INIT(&slowcgi_proc.requests);
2897ac15270Sflorian 
2907ac15270Sflorian 	signal_set(&slowcgi_proc.ev_sigchld, SIGCHLD, slowcgi_sig_handler,
2917ac15270Sflorian 	    &slowcgi_proc);
2927ac15270Sflorian 	signal_set(&slowcgi_proc.ev_sigpipe, SIGPIPE, slowcgi_sig_handler,
2937ac15270Sflorian 	    &slowcgi_proc);
2947ac15270Sflorian 
2957ac15270Sflorian 	signal_add(&slowcgi_proc.ev_sigchld, NULL);
2967ac15270Sflorian 	signal_add(&slowcgi_proc.ev_sigpipe, NULL);
2977ac15270Sflorian 
2987ac15270Sflorian 	event_dispatch();
2997ac15270Sflorian 	return (0);
3007ac15270Sflorian }
3017ac15270Sflorian void
3020b5ee19eSblambert slowcgi_listen(char *path, gid_t gid)
3037ac15270Sflorian {
3047ac15270Sflorian 	struct listener		 *l = NULL;
3057ac15270Sflorian 	struct sockaddr_un	 sun;
3067ac15270Sflorian 	mode_t			 old_umask, mode;
3077ac15270Sflorian 	int			 fd;
3087ac15270Sflorian 
3097ac15270Sflorian 	if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
3107ac15270Sflorian 		lerr(1, "slowcgi_listen: socket");
31194ee156dSflorian 	fcntl(fd, F_SETFD, FD_CLOEXEC);
3127ac15270Sflorian 
3137ac15270Sflorian 	bzero(&sun, sizeof(sun));
3147ac15270Sflorian 	sun.sun_family = AF_UNIX;
3157ac15270Sflorian 	strlcpy(sun.sun_path, path, sizeof(sun.sun_path));
3167ac15270Sflorian 
3177ac15270Sflorian 	if (unlink(path) == -1)
3187ac15270Sflorian 		if (errno != ENOENT)
3197ac15270Sflorian 			lerr(1, "slowcgi_listen: unlink %s", path);
3207ac15270Sflorian 
3217ac15270Sflorian 	old_umask = umask(S_IXUSR|S_IXGRP|S_IWOTH|S_IROTH|S_IXOTH);
3227ac15270Sflorian 	mode = S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP;
3237ac15270Sflorian 
3247ac15270Sflorian 	if (bind(fd, (struct sockaddr *)&sun, sizeof(sun)) == -1)
3257ac15270Sflorian 		lerr(1,"slowcgi_listen: bind: %s", path);
3267ac15270Sflorian 
3277ac15270Sflorian 	umask(old_umask);
3287ac15270Sflorian 
3297ac15270Sflorian 	if (chmod(path, mode) == -1)
3307ac15270Sflorian 		lerr(1, "slowcgi_listen: chmod: %s", path);
3317ac15270Sflorian 
3327ac15270Sflorian 	if (chown(path, 0, gid) == -1)
3337ac15270Sflorian 		lerr(1, "slowcgi_listen: chown: %s", path);
3347ac15270Sflorian 
3357ac15270Sflorian 	if (ioctl(fd, FIONBIO, &on) == -1)
3367ac15270Sflorian 		lerr(1, "listener ioctl(FIONBIO)");
3377ac15270Sflorian 
3387ac15270Sflorian 	if (listen(fd, 5) == -1)
3397ac15270Sflorian 		lerr(1, "listen");
3407ac15270Sflorian 
3417ac15270Sflorian 	l = calloc(1, sizeof(*l));
3427ac15270Sflorian 	if (l == NULL)
3437ac15270Sflorian 		lerr(1, "listener ev alloc");
3447ac15270Sflorian 
3457ac15270Sflorian 	event_set(&l->ev, fd, EV_READ | EV_PERSIST, slowcgi_accept, l);
3467ac15270Sflorian 	event_add(&l->ev, NULL);
3477ac15270Sflorian 	evtimer_set(&l->pause, slowcgi_paused, l);
348bde99fd8Sblambert 
349bde99fd8Sblambert 	ldebug("socket: %s", path);
3507ac15270Sflorian }
3517ac15270Sflorian 
3527ac15270Sflorian void
3537ac15270Sflorian slowcgi_paused(int fd, short events, void *arg)
3547ac15270Sflorian {
3557ac15270Sflorian 	struct listener	*l = arg;
3567ac15270Sflorian 	event_add(&l->ev, NULL);
3577ac15270Sflorian }
3587ac15270Sflorian 
3597ac15270Sflorian void
3607ac15270Sflorian slowcgi_accept(int fd, short events, void *arg)
3617ac15270Sflorian {
3627ac15270Sflorian 	struct listener		*l;
3637ac15270Sflorian 	struct sockaddr_storage	 ss;
3647ac15270Sflorian 	struct timeval		 pause;
365c3b13bc8Sblambert 	struct request		*c;
366c3b13bc8Sblambert 	struct requests		*requests;
3677ac15270Sflorian 	socklen_t		 len;
3687ac15270Sflorian 	int			 s;
3697ac15270Sflorian 
3707ac15270Sflorian 	l = arg;
371965b2218Sblambert 	pause.tv_sec = 1;
372965b2218Sblambert 	pause.tv_usec = 0;
3737ac15270Sflorian 	c = NULL;
3747ac15270Sflorian 
3757ac15270Sflorian 	len = sizeof(ss);
3767ac15270Sflorian 	s = accept(fd, (struct sockaddr *)&ss, &len);
3777ac15270Sflorian 	if (s == -1) {
3787ac15270Sflorian 		switch (errno) {
3797ac15270Sflorian 		case EINTR:
3807ac15270Sflorian 		case EWOULDBLOCK:
3817ac15270Sflorian 		case ECONNABORTED:
3827ac15270Sflorian 			return;
3837ac15270Sflorian 		case EMFILE:
3847ac15270Sflorian 		case ENFILE:
3857ac15270Sflorian 			event_del(&l->ev);
3867ac15270Sflorian 			evtimer_add(&l->pause, &pause);
3877ac15270Sflorian 			return;
3887ac15270Sflorian 		default:
3897ac15270Sflorian 			lerr(1, "accept");
3907ac15270Sflorian 		}
3917ac15270Sflorian 	}
3927ac15270Sflorian 
39394ee156dSflorian 	fcntl(s, F_SETFD, FD_CLOEXEC);
3947ac15270Sflorian 	if (ioctl(s, FIONBIO, &on) == -1)
395c3b13bc8Sblambert 		lerr(1, "request ioctl(FIONBIO)");
3967ac15270Sflorian 
3977ac15270Sflorian 	c = calloc(1, sizeof(*c));
3987ac15270Sflorian 	if (c == NULL) {
399c3b13bc8Sblambert 		lwarn("cannot calloc request");
4007ac15270Sflorian 		close(s);
4017ac15270Sflorian 		return;
4027ac15270Sflorian 	}
403c3b13bc8Sblambert 	requests = calloc(1, sizeof(*requests));
404c3b13bc8Sblambert 	if (requests == NULL) {
405c3b13bc8Sblambert 		lwarn("cannot calloc requests");
4067ac15270Sflorian 		close(s);
4077ac15270Sflorian 		free(c);
4087ac15270Sflorian 		return;
4097ac15270Sflorian 	}
4107ac15270Sflorian 	c->fd = s;
4117ac15270Sflorian 	c->buf_pos = 0;
4127ac15270Sflorian 	c->buf_len = 0;
4137ac15270Sflorian 	c->request_started = 0;
4147ac15270Sflorian 	TAILQ_INIT(&c->response_head);
4157ac15270Sflorian 	TAILQ_INIT(&c->stdin_head);
4167ac15270Sflorian 
4177ac15270Sflorian 	event_set(&c->ev, s, EV_READ | EV_PERSIST, slowcgi_request, c);
4187ac15270Sflorian 	event_add(&c->ev, NULL);
4197ac15270Sflorian 
4207ac15270Sflorian 	event_set(&c->tmo, s, 0, slowcgi_timeout, c);
4217ac15270Sflorian 	event_add(&c->tmo, &timeout);
422c3b13bc8Sblambert 	requests->request = c;
423c3b13bc8Sblambert 	SLIST_INSERT_HEAD(&slowcgi_proc.requests, requests, entry);
4247ac15270Sflorian }
4257ac15270Sflorian 
4267ac15270Sflorian void
4277ac15270Sflorian slowcgi_timeout(int fd, short events, void *arg)
4287ac15270Sflorian {
429c3b13bc8Sblambert 	cleanup_request((struct request*) arg);
4307ac15270Sflorian }
4317ac15270Sflorian 
4327ac15270Sflorian void
4337ac15270Sflorian slowcgi_sig_handler(int sig, short event, void *arg)
4347ac15270Sflorian {
435c3b13bc8Sblambert 	struct request		*c;
436c3b13bc8Sblambert 	struct requests		*ncs;
4377ac15270Sflorian 	struct slowcgi_proc 	*p;
4387ac15270Sflorian 	pid_t			 pid;
4397ac15270Sflorian 	int			 status;
4407ac15270Sflorian 
4417ac15270Sflorian 	p = arg;
4427ac15270Sflorian 	c = NULL;
4437ac15270Sflorian 
4447ac15270Sflorian 	switch (sig) {
4457ac15270Sflorian 	case SIGCHLD:
4467ac15270Sflorian 		pid = wait(&status);
447c3b13bc8Sblambert 		SLIST_FOREACH(ncs, &p->requests, entry)
448c3b13bc8Sblambert 			if (ncs->request->script_pid == pid) {
449c3b13bc8Sblambert 				c = ncs->request;
4507ac15270Sflorian 				break;
4517ac15270Sflorian 			}
452bde99fd8Sblambert 		if (c == NULL) {
453bde99fd8Sblambert 			lwarnx("caught exit of unknown child %i", pid);
454bde99fd8Sblambert 			break;
455bde99fd8Sblambert 		}
456bde99fd8Sblambert 
457bde99fd8Sblambert 		if (WIFSIGNALED(status))
458bde99fd8Sblambert 			c->script_status = WTERMSIG(status);
459bde99fd8Sblambert 		else
4607ac15270Sflorian 			c->script_status = WEXITSTATUS(status);
461bde99fd8Sblambert 
4627ac15270Sflorian 		if (c->script_flags == (STDOUT_DONE | STDERR_DONE))
463c3b13bc8Sblambert 			create_end_record(c);
4647ac15270Sflorian 		c->script_flags |= SCRIPT_DONE;
465bde99fd8Sblambert 
466bde99fd8Sblambert 		ldebug("wait: %s", c->script_name);
467bde99fd8Sblambert 		break;
4687ac15270Sflorian 	case SIGPIPE:
4697ac15270Sflorian 		/* ignore */
4707ac15270Sflorian 		break;
4717ac15270Sflorian 	default:
4727ac15270Sflorian 		lerr(1, "unexpected signal: %d", sig);
4737ac15270Sflorian 
4747ac15270Sflorian 	}
4757ac15270Sflorian }
4767ac15270Sflorian 
4777ac15270Sflorian void
4787ac15270Sflorian slowcgi_response(int fd, short events, void *arg)
4797ac15270Sflorian {
480c3b13bc8Sblambert 	struct request		 	*c;
4817ac15270Sflorian 	struct fcgi_record_header	*header;
4827ac15270Sflorian 	struct fcgi_response		*resp;
4837ac15270Sflorian 	ssize_t 			 n;
4847ac15270Sflorian 
4857ac15270Sflorian 	c = arg;
4867ac15270Sflorian 
4877ac15270Sflorian 	while ((resp = TAILQ_FIRST(&c->response_head))) {
4887ac15270Sflorian 		header = (struct fcgi_record_header*) resp->data;
4897ac15270Sflorian 		if (debug)
49003e61f17Sblambert 			dump_fcgi_record("resp ", header);
4917ac15270Sflorian 
4927ac15270Sflorian 		n = write(fd, resp->data + resp->data_pos, resp->data_len);
4937ac15270Sflorian 		if (n == -1) {
4947ac15270Sflorian 			if (errno == EAGAIN)
4957ac15270Sflorian 				return;
496c3b13bc8Sblambert 			cleanup_request(c);
4977ac15270Sflorian 			return;
4987ac15270Sflorian 		}
4997ac15270Sflorian 		resp->data_pos += n;
5007ac15270Sflorian 		resp->data_len -= n;
5017ac15270Sflorian 		if (resp->data_len == 0) {
5027ac15270Sflorian 			TAILQ_REMOVE(&c->response_head, resp, entry);
5037ac15270Sflorian 			free(resp);
5047ac15270Sflorian 		}
5057ac15270Sflorian 	}
5067ac15270Sflorian 
5077ac15270Sflorian 	if (TAILQ_EMPTY(&c->response_head)) {
5087ac15270Sflorian 		if (c->script_flags == (STDOUT_DONE | STDERR_DONE |
5097ac15270Sflorian 		    SCRIPT_DONE))
510c3b13bc8Sblambert 			cleanup_request(c);
5117ac15270Sflorian 		else
5127ac15270Sflorian 			event_del(&c->resp_ev);
5137ac15270Sflorian 	}
5147ac15270Sflorian }
5157ac15270Sflorian 
5167ac15270Sflorian void
5177ac15270Sflorian slowcgi_request(int fd, short events, void *arg)
5187ac15270Sflorian {
519c3b13bc8Sblambert 	struct request	*c;
5207ac15270Sflorian 	size_t 		 n, parsed;
5217ac15270Sflorian 
5227ac15270Sflorian 	c = arg;
5237ac15270Sflorian 	parsed = 0;
5247ac15270Sflorian 
5257ac15270Sflorian 	n = read(fd, c->buf + c->buf_pos + c->buf_len,
5267ac15270Sflorian 	    FCGI_RECORD_SIZE - c->buf_pos-c->buf_len);
5277ac15270Sflorian 
5287ac15270Sflorian 	switch (n) {
5297ac15270Sflorian 	case -1:
5307ac15270Sflorian 		switch (errno) {
5317ac15270Sflorian 		case EINTR:
5327ac15270Sflorian 		case EAGAIN:
5337ac15270Sflorian 			return;
5347ac15270Sflorian 		default:
5357ac15270Sflorian 			goto fail;
5367ac15270Sflorian 		}
5377ac15270Sflorian 		break;
5387ac15270Sflorian 
5397ac15270Sflorian 	case 0:
5407ac15270Sflorian 		ldebug("closed connection");
5417ac15270Sflorian 		goto fail;
5427ac15270Sflorian 	default:
5437ac15270Sflorian 		break;
5447ac15270Sflorian 	}
5457ac15270Sflorian 
5467ac15270Sflorian 	c->buf_len += n;
5477ac15270Sflorian 
54802ad6491Sblambert 	/*
54902ad6491Sblambert 	 * Parse the records as they are received. Per the FastCGI
55002ad6491Sblambert 	 * specification, the server need only receive the FastCGI
55102ad6491Sblambert 	 * parameter records in full; it is free to begin execution
55202ad6491Sblambert 	 * at that point, which is what happens here.
55302ad6491Sblambert 	 */
5547ac15270Sflorian 	do {
555c3b13bc8Sblambert 		parsed = parse_record(c->buf + c->buf_pos, c->buf_len, c);
5567ac15270Sflorian 		c->buf_pos += parsed;
5577ac15270Sflorian 		c->buf_len -= parsed;
5587ac15270Sflorian 	} while (parsed > 0 && c->buf_len > 0);
5597ac15270Sflorian 
560965b2218Sblambert 	/* Make space for further reads */
5617ac15270Sflorian 	if (c->buf_len > 0) {
5627ac15270Sflorian 		bcopy(c->buf + c->buf_pos, c->buf, c->buf_len);
5637ac15270Sflorian 		c->buf_pos = 0;
5647ac15270Sflorian 	}
5657ac15270Sflorian 	return;
5667ac15270Sflorian fail:
567c3b13bc8Sblambert 	cleanup_request(c);
5687ac15270Sflorian }
5697ac15270Sflorian 
5707ac15270Sflorian void
571c3b13bc8Sblambert parse_begin_request(uint8_t *buf, uint16_t n, struct request *c, uint16_t id)
5727ac15270Sflorian {
5737ac15270Sflorian 	struct fcgi_begin_request_body	*b;
5747ac15270Sflorian 
575965b2218Sblambert 	/* XXX -- FCGI_CANT_MPX_CONN */
5767ac15270Sflorian 	if (c->request_started) {
5777ac15270Sflorian 		lwarnx("unexpected FCGI_BEGIN_REQUEST, ignoring");
5787ac15270Sflorian 		return;
5797ac15270Sflorian 	}
5807ac15270Sflorian 
5817ac15270Sflorian 	if (n != sizeof(struct fcgi_begin_request_body)) {
5827ac15270Sflorian 		lwarnx("wrong size %d != %d", n,
5837ac15270Sflorian 		    sizeof(struct fcgi_begin_request_body));
5847ac15270Sflorian 		return;
5857ac15270Sflorian 	}
5867ac15270Sflorian 
5877ac15270Sflorian 	c->request_started = 1;
5887ac15270Sflorian 	b = (struct fcgi_begin_request_body*) buf;
5897ac15270Sflorian 
5907ac15270Sflorian 	c->id = id;
5917ac15270Sflorian 	SLIST_INIT(&c->env);
5927ac15270Sflorian 	c->env_count = 0;
5937ac15270Sflorian }
5947ac15270Sflorian void
595c3b13bc8Sblambert parse_params(uint8_t *buf, uint16_t n, struct request *c, uint16_t id)
5967ac15270Sflorian {
5977ac15270Sflorian 	struct env_val			*env_entry;
5987ac15270Sflorian 	uint32_t			 name_len, val_len;
5997ac15270Sflorian 
6007ac15270Sflorian 	if (!c->request_started) {
6017ac15270Sflorian 		lwarnx("FCGI_PARAMS without FCGI_BEGIN_REQUEST, ignoring");
6027ac15270Sflorian 		return;
6037ac15270Sflorian 	}
6047ac15270Sflorian 
6057ac15270Sflorian 	if (c->id != id) {
6067ac15270Sflorian 		lwarnx("unexpected id, ignoring");
6077ac15270Sflorian 		return;
6087ac15270Sflorian 	}
6097ac15270Sflorian 
6107ac15270Sflorian 	name_len = val_len = 0;
6117ac15270Sflorian 
61202ad6491Sblambert 	/*
61302ad6491Sblambert 	 * If this is the last FastCGI parameter record,
61402ad6491Sblambert 	 * begin execution of the CGI script.
61502ad6491Sblambert 	 */
6167ac15270Sflorian 	if (n == 0) {
6177ac15270Sflorian 		exec_cgi(c);
6187ac15270Sflorian 		return;
6197ac15270Sflorian 	}
62002ad6491Sblambert 
6217ac15270Sflorian 	while (n > 0) {
6227ac15270Sflorian 		if (buf[0] >> 7 == 0) {
6237ac15270Sflorian 			name_len = buf[0];
6247ac15270Sflorian 			n--;
6257ac15270Sflorian 			buf++;
6267ac15270Sflorian 		} else {
6277ac15270Sflorian 			if (n > 3) {
6287ac15270Sflorian 				name_len = ((buf[3] & 0x7f) << 24) +
6297ac15270Sflorian 				    (buf[2] << 16) + (buf[1] << 8) + buf[0];
6307ac15270Sflorian 				n -= 4;
6317ac15270Sflorian 				buf += 4;
6327ac15270Sflorian 			} else
6337ac15270Sflorian 				return;
6347ac15270Sflorian 		}
6357ac15270Sflorian 
6367ac15270Sflorian 		if (n > 0) {
6377ac15270Sflorian 			if (buf[0] >> 7 == 0) {
6387ac15270Sflorian 				val_len = buf[0];
6397ac15270Sflorian 				n--;
6407ac15270Sflorian 				buf++;
6417ac15270Sflorian 			} else {
6427ac15270Sflorian 				if (n > 3) {
6437ac15270Sflorian 					val_len = ((buf[3] & 0x7f) << 24) +
6447ac15270Sflorian 					    (buf[2] << 16) + (buf[1] << 8) +
6457ac15270Sflorian 					     buf[0];
6467ac15270Sflorian 					n -= 4;
6477ac15270Sflorian 					buf += 4;
6487ac15270Sflorian 				} else
6497ac15270Sflorian 					return;
6507ac15270Sflorian 			}
6517ac15270Sflorian 		}
6527ac15270Sflorian 		if (n < name_len + val_len)
6537ac15270Sflorian 			return;
6547ac15270Sflorian 
6557ac15270Sflorian 		if ((env_entry = malloc(sizeof(struct env_val))) == NULL) {
6567ac15270Sflorian 			lwarnx("cannot allocate env_entry");
6577ac15270Sflorian 			return;
6587ac15270Sflorian 		}
6597ac15270Sflorian 
6607ac15270Sflorian 		if ((env_entry->val = calloc(sizeof(char), name_len + val_len +
6617ac15270Sflorian 		    2)) == NULL) {
6627ac15270Sflorian 			lwarnx("cannot allocate env_entry->val");
6637ac15270Sflorian 			free(env_entry);
6647ac15270Sflorian 			return;
6657ac15270Sflorian 		}
6667ac15270Sflorian 
6677ac15270Sflorian 		bcopy(buf, env_entry->val, name_len);
668965b2218Sblambert 		buf += name_len;
669965b2218Sblambert 		n -= name_len;
6707ac15270Sflorian 
6717ac15270Sflorian 		env_entry->val[name_len] = '\0';
6727ac15270Sflorian 		if (val_len < MAXPATHLEN && strcmp(env_entry->val,
6737ac15270Sflorian 		    "SCRIPT_NAME") == 0) {
6747ac15270Sflorian 			bcopy(buf, c->script_name, val_len);
67553847fa2Sflorian 			c->script_name[val_len] = '\0';
6767ac15270Sflorian 		}
6777ac15270Sflorian 		env_entry->val[name_len] = '=';
6787ac15270Sflorian 
6797ac15270Sflorian 		bcopy(buf, (env_entry->val) + name_len + 1, val_len);
680965b2218Sblambert 		buf += val_len;
681965b2218Sblambert 		n -= val_len;
6827ac15270Sflorian 
6837ac15270Sflorian 		SLIST_INSERT_HEAD(&c->env, env_entry, entry);
6847ac15270Sflorian 		c->env_count++;
6857ac15270Sflorian 	}
6867ac15270Sflorian }
6877ac15270Sflorian 
6887ac15270Sflorian void
689c3b13bc8Sblambert parse_stdin(uint8_t *buf, uint16_t n, struct request *c, uint16_t id)
6907ac15270Sflorian {
6917ac15270Sflorian 	struct fcgi_stdin	*node;
6927ac15270Sflorian 
6937ac15270Sflorian 	if (c->id != id) {
6947ac15270Sflorian 		lwarnx("unexpected id, ignoring");
6957ac15270Sflorian 		return;
6967ac15270Sflorian 	}
6977ac15270Sflorian 
6987ac15270Sflorian 	if ((node = calloc(1, sizeof(struct fcgi_stdin))) == NULL) {
6997ac15270Sflorian 		lwarnx("cannot calloc stdin node");
7007ac15270Sflorian 		return;
7017ac15270Sflorian 	}
7027ac15270Sflorian 
7037ac15270Sflorian 	bcopy(buf, node->data, n);
7047ac15270Sflorian 	node->data_pos = 0;
7057ac15270Sflorian 	node->data_len = n;
7067ac15270Sflorian 
7077ac15270Sflorian 	TAILQ_INSERT_TAIL(&c->stdin_head, node, entry);
7087ac15270Sflorian 
7097ac15270Sflorian 	event_del(&c->script_stdin_ev);
7107ac15270Sflorian 	event_set(&c->script_stdin_ev, EVENT_FD(&c->script_ev), EV_WRITE |
7117ac15270Sflorian 	    EV_PERSIST, script_out, c);
7127ac15270Sflorian 	event_add(&c->script_stdin_ev, NULL);
7137ac15270Sflorian }
7147ac15270Sflorian 
7157ac15270Sflorian size_t
716c3b13bc8Sblambert parse_record(uint8_t *buf, size_t n, struct request *c)
7177ac15270Sflorian {
7187ac15270Sflorian 	struct fcgi_record_header 	*h;
7197ac15270Sflorian 
7207ac15270Sflorian 	if (n < sizeof(struct fcgi_record_header))
7217ac15270Sflorian 		return (0);
7227ac15270Sflorian 
7237ac15270Sflorian 	h = (struct fcgi_record_header*) buf;
7247ac15270Sflorian 
7257ac15270Sflorian 	if (debug)
72603e61f17Sblambert 		dump_fcgi_record("", h);
7277ac15270Sflorian 
7287ac15270Sflorian 	if (n < sizeof(struct fcgi_record_header) + ntohs(h->content_len)
7297ac15270Sflorian 	    + h->padding_len)
7307ac15270Sflorian 		return (0);
7317ac15270Sflorian 
7327ac15270Sflorian 	if (h->version != 1)
7337ac15270Sflorian 		lerrx(1, "wrong version");
7347ac15270Sflorian 
7357ac15270Sflorian 	switch (h->type) {
7367ac15270Sflorian 	case FCGI_BEGIN_REQUEST:
7377ac15270Sflorian 		parse_begin_request(buf + sizeof(struct fcgi_record_header),
7387ac15270Sflorian 		    ntohs(h->content_len), c, ntohs(h->id));
7397ac15270Sflorian 		break;
7407ac15270Sflorian 	case FCGI_PARAMS:
7417ac15270Sflorian 		parse_params(buf + sizeof(struct fcgi_record_header),
7427ac15270Sflorian 		    ntohs(h->content_len), c, ntohs(h->id));
7437ac15270Sflorian 		break;
7447ac15270Sflorian 	case FCGI_STDIN:
7457ac15270Sflorian 		parse_stdin(buf + sizeof(struct fcgi_record_header),
7467ac15270Sflorian 		    ntohs(h->content_len), c, ntohs(h->id));
7477ac15270Sflorian 		break;
7487ac15270Sflorian 	default:
7497ac15270Sflorian 		lwarnx("unimplemented type %d", h->type);
7507ac15270Sflorian 	}
7517ac15270Sflorian 
7527ac15270Sflorian 	return (sizeof(struct fcgi_record_header) + ntohs(h->content_len)
7537ac15270Sflorian 	    + h->padding_len);
7547ac15270Sflorian }
7557ac15270Sflorian 
75602ad6491Sblambert /*
75702ad6491Sblambert  * Fork a new CGI process to handle the request, translating
75802ad6491Sblambert  * between FastCGI parameter records and CGI's environment variables,
75902ad6491Sblambert  * as well as between the CGI process' stdin/stdout and the
76002ad6491Sblambert  * corresponding FastCGI records.
76102ad6491Sblambert  */
7627ac15270Sflorian void
763c3b13bc8Sblambert exec_cgi(struct request *c)
7647ac15270Sflorian {
7657ac15270Sflorian 	struct env_val	*env_entry;
7667ac15270Sflorian 	int		 s[2], s_err[2], i;
7677ac15270Sflorian 	pid_t		 pid;
7687ac15270Sflorian 	char		*argv[2];
7697ac15270Sflorian 	char		**env;
7707ac15270Sflorian 
7717ac15270Sflorian 	i = 0;
7727ac15270Sflorian 
7737ac15270Sflorian 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, s) == -1)
7747ac15270Sflorian 		lerr(1, "socketpair");
7757ac15270Sflorian 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, s_err) == -1)
7767ac15270Sflorian 		lerr(1, "socketpair");
7777ac15270Sflorian 	ldebug("fork: %s", c->script_name);
778965b2218Sblambert 
7797ac15270Sflorian 	switch (pid = fork()) {
7807ac15270Sflorian 	case -1:
7817ac15270Sflorian 		lwarn("fork");
7827ac15270Sflorian 		return;
7837ac15270Sflorian 	case 0:
7847ac15270Sflorian 		/* Child process */
7857ac15270Sflorian 		close(s[0]);
7867ac15270Sflorian 		close(s_err[0]);
7877ac15270Sflorian 		if (dup2(s[1], STDIN_FILENO) == -1)
7887ac15270Sflorian 			_exit(1);
7897ac15270Sflorian 		if (dup2(s[1], STDOUT_FILENO) == -1)
7907ac15270Sflorian 			_exit(1);
7917ac15270Sflorian 		if (dup2(s_err[1], STDERR_FILENO) == -1)
7927ac15270Sflorian 			_exit(1);
7937ac15270Sflorian 		argv[0] = c->script_name;
7947ac15270Sflorian 		argv[1] = NULL;
7957ac15270Sflorian 		if ((env = calloc(c->env_count + 1, sizeof(char*))) == NULL)
7967ac15270Sflorian 			_exit(1);
7977ac15270Sflorian 		SLIST_FOREACH(env_entry, &c->env, entry)
7987ac15270Sflorian 			env[i++] = env_entry->val;
7997ac15270Sflorian 		env[i++] = NULL;
8007ac15270Sflorian 		execve(c->script_name, argv, env);
8017ac15270Sflorian 		_exit(1);
8027ac15270Sflorian 
8037ac15270Sflorian 	}
8047ac15270Sflorian 	/* Parent process*/
8057ac15270Sflorian 	close(s[1]);
8067ac15270Sflorian 	close(s_err[1]);
80794ee156dSflorian 
80894ee156dSflorian 	fcntl(s[0], F_SETFD, FD_CLOEXEC);
80994ee156dSflorian 	fcntl(s_err[0], F_SETFD, FD_CLOEXEC);
81094ee156dSflorian 
8117ac15270Sflorian 	if (ioctl(s[0], FIONBIO, &on) == -1)
8127ac15270Sflorian 		lerr(1, "script ioctl(FIONBIO)");
8137ac15270Sflorian 	if (ioctl(s_err[0], FIONBIO, &on) == -1)
8147ac15270Sflorian 		lerr(1, "script ioctl(FIONBIO)");
8157ac15270Sflorian 
8167ac15270Sflorian 	c->script_pid = pid;
8177ac15270Sflorian 	event_set(&c->script_ev, s[0], EV_READ | EV_PERSIST, script_std_in, c);
8187ac15270Sflorian 	event_add(&c->script_ev, NULL);
8197ac15270Sflorian 	event_set(&c->script_err_ev, s_err[0], EV_READ | EV_PERSIST,
8207ac15270Sflorian 	    script_err_in, c);
8217ac15270Sflorian 	event_add(&c->script_err_ev, NULL);
8227ac15270Sflorian }
8237ac15270Sflorian 
8247ac15270Sflorian void
825c3b13bc8Sblambert create_end_record(struct request *c)
8267ac15270Sflorian {
8277ac15270Sflorian 	struct fcgi_response		*resp;
8287ac15270Sflorian 	struct fcgi_record_header	*header;
82903e61f17Sblambert 	struct fcgi_end_request_body	*end_request;
8307ac15270Sflorian 
8317ac15270Sflorian 	if ((resp = malloc(sizeof(struct fcgi_response))) == NULL) {
8327ac15270Sflorian 		lwarnx("cannot malloc fcgi_response");
8337ac15270Sflorian 		return;
8347ac15270Sflorian 	}
8357ac15270Sflorian 	header = (struct fcgi_record_header*) resp->data;
8367ac15270Sflorian 	header->version = 1;
8377ac15270Sflorian 	header->type = FCGI_END_REQUEST;
8387ac15270Sflorian 	header->id = htons(c->id);
8397ac15270Sflorian 	header->content_len = htons(sizeof(struct
84003e61f17Sblambert 	    fcgi_end_request_body));
8417ac15270Sflorian 	header->padding_len = 0;
8427ac15270Sflorian 	header->reserved = 0;
843*8c347892Sflorian 	end_request = (struct fcgi_end_request_body *) (resp->data +
844*8c347892Sflorian 	    sizeof(struct fcgi_record_header));
8457ac15270Sflorian 	end_request->app_status = htonl(c->script_status);
8467ac15270Sflorian 	end_request->protocol_status = FCGI_REQUEST_COMPLETE;
847*8c347892Sflorian 	end_request->reserved[0] = 0;
848*8c347892Sflorian 	end_request->reserved[1] = 0;
849*8c347892Sflorian 	end_request->reserved[2] = 0;
8507ac15270Sflorian 	resp->data_pos = 0;
85103e61f17Sblambert 	resp->data_len = sizeof(struct fcgi_end_request_body) +
8527ac15270Sflorian 	    sizeof(struct fcgi_record_header);
8537ac15270Sflorian 	TAILQ_INSERT_TAIL(&c->response_head, resp, entry);
8547ac15270Sflorian }
8557ac15270Sflorian 
8567ac15270Sflorian void
857c3b13bc8Sblambert script_in(int fd, struct event *ev, struct request *c, uint8_t type)
8587ac15270Sflorian {
8597ac15270Sflorian 	struct fcgi_response		*resp;
8607ac15270Sflorian 	struct fcgi_record_header	*header;
8617ac15270Sflorian 	ssize_t 			 n;
8627ac15270Sflorian 
8637ac15270Sflorian 	if ((resp = malloc(sizeof(struct fcgi_response))) == NULL) {
8647ac15270Sflorian 		lwarnx("cannot malloc fcgi_response");
8657ac15270Sflorian 		return;
8667ac15270Sflorian 	}
8677ac15270Sflorian 	header = (struct fcgi_record_header*) resp->data;
8687ac15270Sflorian 	header->version = 1;
8697ac15270Sflorian 	header->type = type;
8707ac15270Sflorian 	header->id = htons(c->id);
8717ac15270Sflorian 	header->padding_len = 0;
8727ac15270Sflorian 	header->reserved = 0;
8737ac15270Sflorian 
8747ac15270Sflorian 	n = read(fd, resp->data + sizeof(struct fcgi_record_header),
8750ee0284dSblambert 	     FCGI_CONTENT_SIZE);
8767ac15270Sflorian 
8777ac15270Sflorian 	if (n == -1) {
8787ac15270Sflorian 		switch (errno) {
8797ac15270Sflorian 		case EINTR:
8807ac15270Sflorian 		case EAGAIN:
881aa48dbd7Sjasper 			free(resp);
8827ac15270Sflorian 			return;
8837ac15270Sflorian 		default:
8847ac15270Sflorian 			n = 0; /* fake empty FCGI_STD{OUT,ERR} response */
8857ac15270Sflorian 		}
8867ac15270Sflorian 	}
8877ac15270Sflorian 	header->content_len = htons(n);
8887ac15270Sflorian 	resp->data_pos = 0;
8897ac15270Sflorian 	resp->data_len = n + sizeof(struct fcgi_record_header);
8907ac15270Sflorian 	TAILQ_INSERT_TAIL(&c->response_head, resp, entry);
8917ac15270Sflorian 
8927ac15270Sflorian 	event_del(&c->resp_ev);
8937ac15270Sflorian 	event_set(&c->resp_ev, EVENT_FD(&c->ev), EV_WRITE | EV_PERSIST,
8947ac15270Sflorian 	    slowcgi_response, c);
8957ac15270Sflorian 	event_add(&c->resp_ev, NULL);
8967ac15270Sflorian 
8977ac15270Sflorian 	if (n == 0) {
8987ac15270Sflorian 		if (type == FCGI_STDOUT)
8997ac15270Sflorian 			c->script_flags |= STDOUT_DONE;
9007ac15270Sflorian 		else
9017ac15270Sflorian 			c->script_flags |= STDERR_DONE;
9027ac15270Sflorian 
9037ac15270Sflorian 		if (c->script_flags == (STDOUT_DONE | STDERR_DONE |
9047ac15270Sflorian 		    SCRIPT_DONE)) {
905c3b13bc8Sblambert 			create_end_record(c);
9067ac15270Sflorian 		}
9077ac15270Sflorian 		event_del(ev);
9087ac15270Sflorian 		close(fd);
9097ac15270Sflorian 	}
9107ac15270Sflorian }
9117ac15270Sflorian 
9127ac15270Sflorian void
9137ac15270Sflorian script_std_in(int fd, short events, void *arg)
9147ac15270Sflorian {
915c3b13bc8Sblambert 	struct request *c = arg;
9167ac15270Sflorian 	script_in(fd, &c->script_ev, c, FCGI_STDOUT);
9177ac15270Sflorian }
9187ac15270Sflorian 
9197ac15270Sflorian void
9207ac15270Sflorian script_err_in(int fd, short events, void *arg)
9217ac15270Sflorian {
922c3b13bc8Sblambert 	struct request *c = arg;
9237ac15270Sflorian 	script_in(fd, &c->script_err_ev, c, FCGI_STDERR);
9247ac15270Sflorian }
9257ac15270Sflorian 
9267ac15270Sflorian void
9277ac15270Sflorian script_out(int fd, short events, void *arg)
9287ac15270Sflorian {
929c3b13bc8Sblambert 	struct request		*c;
9307ac15270Sflorian 	struct fcgi_stdin	*node;
9317ac15270Sflorian 	ssize_t 		 n;
9327ac15270Sflorian 
9337ac15270Sflorian 	c = arg;
9347ac15270Sflorian 
9357ac15270Sflorian 	while ((node = TAILQ_FIRST(&c->stdin_head))) {
9367ac15270Sflorian 		if (node->data_len == 0) { /* end of stdin marker */
9377ac15270Sflorian 			shutdown(fd, SHUT_WR);
9387ac15270Sflorian 			break;
9397ac15270Sflorian 		}
9407ac15270Sflorian 		n = write(fd, node->data + node->data_pos, node->data_len);
9417ac15270Sflorian 		if (n == -1) {
9427ac15270Sflorian 			if (errno == EAGAIN)
9437ac15270Sflorian 				return;
9447ac15270Sflorian 			event_del(&c->script_stdin_ev);
9457ac15270Sflorian 			return;
9467ac15270Sflorian 		}
9477ac15270Sflorian 		node->data_pos += n;
9487ac15270Sflorian 		node->data_len -= n;
9497ac15270Sflorian 		if (node->data_len == 0) {
9507ac15270Sflorian 			TAILQ_REMOVE(&c->stdin_head, node, entry);
9517ac15270Sflorian 			free(node);
9527ac15270Sflorian 		}
9537ac15270Sflorian 	}
9547ac15270Sflorian }
9557ac15270Sflorian 
9567ac15270Sflorian void
957c3b13bc8Sblambert cleanup_request(struct request *c)
9587ac15270Sflorian {
9597ac15270Sflorian 	struct fcgi_response	*resp;
9607ac15270Sflorian 	struct fcgi_stdin	*stdin_node;
9617ac15270Sflorian 	struct env_val		*env_entry;
962c3b13bc8Sblambert 	struct requests		*ncs, *tcs;
9637ac15270Sflorian 
9647ac15270Sflorian 	evtimer_del(&c->tmo);
9657ac15270Sflorian 	if (event_initialized(&c->ev))
9667ac15270Sflorian 		event_del(&c->ev);
9677ac15270Sflorian 	if (event_initialized(&c->resp_ev))
9687ac15270Sflorian 		event_del(&c->resp_ev);
9697ac15270Sflorian 	if (event_initialized(&c->script_ev)) {
9707ac15270Sflorian 		close(EVENT_FD(&c->script_ev));
9717ac15270Sflorian 		event_del(&c->script_ev);
9727ac15270Sflorian 	}
9737ac15270Sflorian 	if (event_initialized(&c->script_err_ev)) {
9747ac15270Sflorian 		close(EVENT_FD(&c->script_err_ev));
9757ac15270Sflorian 		event_del(&c->script_err_ev);
9767ac15270Sflorian 	}
9777ac15270Sflorian 	if (event_initialized(&c->script_stdin_ev)) {
9787ac15270Sflorian 		close(EVENT_FD(&c->script_stdin_ev));
9797ac15270Sflorian 		event_del(&c->script_stdin_ev);
9807ac15270Sflorian 	}
9817ac15270Sflorian 	close(c->fd);
9827ac15270Sflorian 	while (!SLIST_EMPTY(&c->env)) {
9837ac15270Sflorian 		env_entry = SLIST_FIRST(&c->env);
9847ac15270Sflorian 		SLIST_REMOVE_HEAD(&c->env, entry);
9857ac15270Sflorian 		free(env_entry->val);
9867ac15270Sflorian 		free(env_entry);
9877ac15270Sflorian 	}
9887ac15270Sflorian 
9897ac15270Sflorian 	while ((resp = TAILQ_FIRST(&c->response_head))) {
9907ac15270Sflorian 		TAILQ_REMOVE(&c->response_head, resp, entry);
9917ac15270Sflorian 		free(resp);
9927ac15270Sflorian 	}
9937ac15270Sflorian 	while ((stdin_node = TAILQ_FIRST(&c->stdin_head))) {
9947ac15270Sflorian 		TAILQ_REMOVE(&c->stdin_head, stdin_node, entry);
9957ac15270Sflorian 		free(stdin_node);
9967ac15270Sflorian 	}
997c3b13bc8Sblambert 	SLIST_FOREACH_SAFE(ncs, &slowcgi_proc.requests, entry, tcs) {
998c3b13bc8Sblambert 		if (ncs->request == c) {
999c3b13bc8Sblambert 			SLIST_REMOVE(&slowcgi_proc.requests, ncs, requests,
10007ac15270Sflorian 			    entry);
10017ac15270Sflorian 			free(ncs);
10027ac15270Sflorian 			break;
10037ac15270Sflorian 		}
10047ac15270Sflorian 	}
10057ac15270Sflorian 	free(c);
10067ac15270Sflorian }
10077ac15270Sflorian 
1008965b2218Sblambert void
100903e61f17Sblambert dump_fcgi_record(const char *p, struct fcgi_record_header *h)
101003e61f17Sblambert {
101103e61f17Sblambert 	dump_fcgi_record_header(p, h);
101203e61f17Sblambert 
101303e61f17Sblambert 	if (h->type == FCGI_BEGIN_REQUEST)
101403e61f17Sblambert 		dump_fcgi_begin_request_body(p,
101503e61f17Sblambert 		    (struct fcgi_begin_request_body *)((char *)h) + sizeof(*h));
101603e61f17Sblambert 	else if (h->type == FCGI_END_REQUEST)
101703e61f17Sblambert 		dump_fcgi_end_request_body(p,
101803e61f17Sblambert 		    (struct fcgi_end_request_body *)((char *)h) + sizeof(*h));
101903e61f17Sblambert }
102003e61f17Sblambert 
102103e61f17Sblambert void
1022965b2218Sblambert dump_fcgi_record_header(const char* p, struct fcgi_record_header *h)
10237ac15270Sflorian {
10247ac15270Sflorian 	ldebug("%sversion:         %d", p, h->version);
10257ac15270Sflorian 	ldebug("%stype:            %d", p, h->type);
10267ac15270Sflorian 	ldebug("%srequestId:       %d", p, ntohs(h->id));
10277ac15270Sflorian 	ldebug("%scontentLength:   %d", p, ntohs(h->content_len));
10287ac15270Sflorian 	ldebug("%spaddingLength:   %d", p, h->padding_len);
10297ac15270Sflorian 	ldebug("%sreserved:        %d", p, h->reserved);
10307ac15270Sflorian }
10317ac15270Sflorian 
10327ac15270Sflorian void
103303e61f17Sblambert dump_fcgi_begin_request_body(const char *p, struct fcgi_begin_request_body *b)
103403e61f17Sblambert {
103503e61f17Sblambert 	ldebug("%srole             %d", p, ntohs(b->role));
103603e61f17Sblambert 	ldebug("%sflags            %d", p, b->flags);
103703e61f17Sblambert }
103803e61f17Sblambert 
103903e61f17Sblambert void
104003e61f17Sblambert dump_fcgi_end_request_body(const char *p, struct fcgi_end_request_body *b)
104103e61f17Sblambert {
104203e61f17Sblambert 	ldebug("%sappStatus:       %d", p, ntohl(b->app_status));
104303e61f17Sblambert 	ldebug("%sprotocolStatus:  %d", p, b->protocol_status);
104403e61f17Sblambert }
104503e61f17Sblambert 
104603e61f17Sblambert void
10477ac15270Sflorian syslog_vstrerror(int e, int priority, const char *fmt, va_list ap)
10487ac15270Sflorian {
10497ac15270Sflorian 	char *s;
10507ac15270Sflorian 
10517ac15270Sflorian 	if (vasprintf(&s, fmt, ap) == -1) {
10527ac15270Sflorian 		syslog(LOG_EMERG, "unable to alloc in syslog_vstrerror");
10537ac15270Sflorian 		exit(1);
10547ac15270Sflorian 	}
10557ac15270Sflorian 	syslog(priority, "%s: %s", s, strerror(e));
10567ac15270Sflorian 	free(s);
10577ac15270Sflorian }
10587ac15270Sflorian 
10597ac15270Sflorian void
10607ac15270Sflorian syslog_err(int ecode, const char *fmt, ...)
10617ac15270Sflorian {
10627ac15270Sflorian 	va_list ap;
10637ac15270Sflorian 
10647ac15270Sflorian 	va_start(ap, fmt);
10657ac15270Sflorian 	syslog_vstrerror(errno, LOG_EMERG, fmt, ap);
10667ac15270Sflorian 	va_end(ap);
10677ac15270Sflorian 	exit(ecode);
10687ac15270Sflorian }
10697ac15270Sflorian 
10707ac15270Sflorian void
10717ac15270Sflorian syslog_errx(int ecode, const char *fmt, ...)
10727ac15270Sflorian {
10737ac15270Sflorian 	va_list ap;
10747ac15270Sflorian 
10757ac15270Sflorian 	va_start(ap, fmt);
10767ac15270Sflorian 	vsyslog(LOG_WARNING, fmt, ap);
10777ac15270Sflorian 	va_end(ap);
10787ac15270Sflorian 	exit(ecode);
10797ac15270Sflorian }
10807ac15270Sflorian 
10817ac15270Sflorian void
10827ac15270Sflorian syslog_warn(const char *fmt, ...)
10837ac15270Sflorian {
10847ac15270Sflorian 	va_list ap;
10857ac15270Sflorian 
10867ac15270Sflorian 	va_start(ap, fmt);
10877ac15270Sflorian 	syslog_vstrerror(errno, LOG_WARNING, fmt, ap);
10887ac15270Sflorian 	va_end(ap);
10897ac15270Sflorian }
10907ac15270Sflorian 
10917ac15270Sflorian void
10927ac15270Sflorian syslog_warnx(const char *fmt, ...)
10937ac15270Sflorian {
10947ac15270Sflorian 	va_list ap;
10957ac15270Sflorian 
10967ac15270Sflorian 	va_start(ap, fmt);
10977ac15270Sflorian 	vsyslog(LOG_WARNING, fmt, ap);
10987ac15270Sflorian 	va_end(ap);
10997ac15270Sflorian }
11007ac15270Sflorian 
11017ac15270Sflorian void
11027ac15270Sflorian syslog_info(const char *fmt, ...)
11037ac15270Sflorian {
11047ac15270Sflorian 	va_list ap;
11057ac15270Sflorian 
11067ac15270Sflorian 	va_start(ap, fmt);
11077ac15270Sflorian 	vsyslog(LOG_INFO, fmt, ap);
11087ac15270Sflorian 	va_end(ap);
11097ac15270Sflorian }
11107ac15270Sflorian 
11117ac15270Sflorian void
11127ac15270Sflorian syslog_debug(const char *fmt, ...)
11137ac15270Sflorian {
11147ac15270Sflorian 	va_list ap;
11157ac15270Sflorian 
11167ac15270Sflorian 	if (!debug)
11177ac15270Sflorian 		return;
11187ac15270Sflorian 
11197ac15270Sflorian 	va_start(ap, fmt);
11207ac15270Sflorian 	vsyslog(LOG_DEBUG, fmt, ap);
11217ac15270Sflorian 	va_end(ap);
11227ac15270Sflorian }
1123