aboutsummaryrefslogtreecommitdiffstats
path: root/slowcgi.c
diff options
context:
space:
mode:
authorflorian2013-05-23 12:03:24 +0000
committerWynn Wolf Arbor2020-05-24 12:33:55 +0200
commit95abc4e350e418fec9b5fa73551e4f9bf07808d5 (patch)
tree67f8b77dca8af272281bbaffcf56e92fd7984116 /slowcgi.c
downloadslowcgi-95abc4e350e418fec9b5fa73551e4f9bf07808d5.tar.gz
Put slowcgi(8) a FastCGI to CGI wrapper in to work on it in tree. Not hooked up to the build yet. OK sthen@, deraadt@ agrees
Diffstat (limited to 'slowcgi.c')
-rw-r--r--slowcgi.c1030
1 files changed, 1030 insertions, 0 deletions
diff --git a/slowcgi.c b/slowcgi.c
new file mode 100644
index 0000000..2bdc72b
--- /dev/null
+++ b/slowcgi.c
@@ -0,0 +1,1030 @@
+/* $OpenBSD: slowcgi.c,v 1.1 2013/05/23 12:03:24 florian Exp $ */
+/*
+ * Copyright (c) 2013 David Gwynne <dlg@openbsd.org>
+ * Copyright (c) 2013 Florian Obser <florian@openbsd.org>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include <sys/types.h>
+#include <sys/ioctl.h>
+#include <sys/queue.h>
+#include <sys/socket.h>
+#include <sys/socketvar.h>
+#include <sys/stat.h>
+#include <sys/un.h>
+#include <sys/wait.h>
+#include <err.h>
+#include <ctype.h>
+#include <errno.h>
+#include <event.h>
+#include <netdb.h>
+#include <pwd.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <syslog.h>
+#include <unistd.h>
+
+#define TIMEOUT_DEFAULT 120
+#define SLOWCGI_USER "www"
+#define FCGI_RECORD_SIZE 64*1024
+#define STDOUT_DONE 1
+#define STDERR_DONE 2
+#define SCRIPT_DONE 4
+
+#define FCGI_BEGIN_REQUEST 1
+#define FCGI_ABORT_REQUEST 2
+#define FCGI_END_REQUEST 3
+#define FCGI_PARAMS 4
+#define FCGI_STDIN 5
+#define FCGI_STDOUT 6
+#define FCGI_STDERR 7
+#define FCGI_DATA 8
+#define FCGI_GET_VALUES 9
+#define FCGI_GET_VALUES_RESULT 10
+#define FCGI_UNKNOWN_TYPE 11
+#define FCGI_MAXTYPE (FCGI_UNKNOWN_TYPE)
+
+#define FCGI_REQUEST_COMPLETE 0
+#define FCGI_CANT_MPX_CONN 1
+#define FCGI_OVERLOADED 2
+#define FCGI_UNKNOWN_ROLE 3
+
+
+struct listener {
+ struct event ev, pause;
+};
+
+struct env_val {
+ SLIST_ENTRY(env_val) entry;
+ char *val;
+};
+SLIST_HEAD(env_head, env_val);
+
+struct fcgi_record_header {
+ uint8_t version;
+ uint8_t type;
+ uint16_t id;
+ uint16_t content_len;
+ uint8_t padding_len;
+ uint8_t reserved;
+}__packed;
+
+struct fcgi_response {
+ TAILQ_ENTRY(fcgi_response) entry;
+ uint8_t data[FCGI_RECORD_SIZE + sizeof(struct
+ fcgi_record_header)];
+ size_t data_pos;
+ size_t data_len;
+};
+TAILQ_HEAD(fcgi_response_head, fcgi_response);
+
+struct fcgi_stdin {
+ TAILQ_ENTRY(fcgi_stdin) entry;
+ uint8_t data[FCGI_RECORD_SIZE];
+ size_t data_pos;
+ size_t data_len;
+};
+TAILQ_HEAD(fcgi_stdin_head, fcgi_stdin);
+
+struct client {
+ struct event ev;
+ struct event resp_ev;
+ struct event tmo;
+ int fd;
+ uint8_t buf[FCGI_RECORD_SIZE];
+ size_t buf_pos;
+ size_t buf_len;
+ struct fcgi_response_head response_head;
+ struct fcgi_stdin_head stdin_head;
+ uint16_t id;
+ char script_name[MAXPATHLEN];
+ struct env_head env;
+ int env_count;
+ pid_t script_pid;
+ int script_status;
+ struct event script_ev;
+ struct event script_err_ev;
+ struct event script_stdin_ev;
+ uint8_t script_flags;
+ uint8_t request_started;
+};
+
+struct clients {
+ SLIST_ENTRY(clients) entry;
+ struct client *client;
+};
+SLIST_HEAD(clients_head, clients);
+
+struct slowcgi_proc {
+ struct clients_head clients;
+ struct event ev_sigchld;
+ struct event ev_sigpipe;
+};
+
+struct fcgi_begin_request_body {
+ uint16_t role;
+ uint8_t flags;
+ uint8_t reserved[5];
+}__packed;
+
+struct fcgi_end_request {
+ uint32_t app_status;
+ uint8_t protocol_status;
+ uint8_t reserved[3];
+}__packed;
+__dead void usage(void);
+void slowcgi_listen(const char *, gid_t);
+void slowcgi_paused(int, short, void*);
+void slowcgi_accept(int, short, void*);
+void slowcgi_request(int, short, void*);
+void slowcgi_response(int, short, void*);
+void slowcgi_timeout(int, short, void*);
+void slowcgi_sig_handler(int, short, void*);
+size_t parse_request(uint8_t* , size_t, struct client*);
+void parse_begin_request(uint8_t*, uint16_t, struct client*,
+ uint16_t);
+void parse_params(uint8_t*, uint16_t, struct client*, uint16_t);
+void parse_stdin(uint8_t*, uint16_t, struct client*, uint16_t);
+void exec_cgi(struct client*);
+void script_in(int, struct event*, struct client*, uint8_t);
+void script_std_in(int, short, void*);
+void script_err_in(int, short, void*);
+void script_out(int, short, void*);
+void create_end_request(struct client*);
+void dump_fcgi_record_header(const char*,
+ struct fcgi_record_header*);
+void cleanup_client(struct client*);
+struct loggers {
+ void (*err)(int, const char *, ...);
+ void (*errx)(int, const char *, ...);
+ void (*warn)(const char *, ...);
+ void (*warnx)(const char *, ...);
+ void (*info)(const char *, ...);
+ void (*debug)(const char *, ...);
+};
+
+const struct loggers conslogger = {
+ err,
+ errx,
+ warn,
+ warnx,
+ warnx, /* info */
+ warnx /* debug */
+};
+
+void syslog_err(int, const char *, ...);
+void syslog_errx(int, const char *, ...);
+void syslog_warn(const char *, ...);
+void syslog_warnx(const char *, ...);
+void syslog_info(const char *, ...);
+void syslog_debug(const char *, ...);
+void syslog_vstrerror(int, int, const char *, va_list);
+
+const struct loggers syslogger = {
+ syslog_err,
+ syslog_errx,
+ syslog_warn,
+ syslog_warnx,
+ syslog_info,
+ syslog_debug
+};
+
+const struct loggers *logger = &conslogger;
+
+#define lerr(_e, _f...) logger->err((_e), _f)
+#define lerrx(_e, _f...) logger->errx((_e), _f)
+#define lwarn(_f...) logger->warn(_f)
+#define lwarnx(_f...) logger->warnx(_f)
+#define linfo(_f...) logger->info(_f)
+#define ldebug(_f...) logger->debug(_f)
+
+__dead void
+usage(void)
+{
+ extern char *__progname;
+ fprintf(stderr, "usage: %s [-d]\n", __progname);
+ exit(1);
+}
+
+struct timeval timeout = { TIMEOUT_DEFAULT, 0 };
+struct slowcgi_proc slowcgi_proc;
+int debug = 0;
+int on = 1;
+
+int
+main(int argc, char *argv[])
+{
+ struct passwd *pw;
+ int c;
+
+ while ((c = getopt(argc, argv, "d")) != -1) {
+ switch (c) {
+ case 'd':
+ debug = 1;
+ break;
+ default:
+ usage();
+ /* NOTREACHED */
+ }
+ }
+
+ if (geteuid() != 0)
+ errx(1, "need root privileges");
+
+ pw = getpwnam(SLOWCGI_USER);
+ if (pw == NULL)
+ err(1, "no %s user", SLOWCGI_USER);
+
+ if (!debug && daemon(1, 0) == -1)
+ err(1, "daemon");
+
+ event_init();
+
+ slowcgi_listen("/var/www/logs/slowcgi.sock", pw->pw_gid);
+ if (chroot(pw->pw_dir) == -1)
+ lerr(1, "chroot(%s)", pw->pw_dir);
+
+ if (chdir("/") == -1)
+ lerr(1, "chdir(%s)", pw->pw_dir);
+
+ if (setgroups(1, &pw->pw_gid) ||
+ setresgid(pw->pw_gid, pw->pw_gid, pw->pw_gid) ||
+ setresuid(pw->pw_uid, pw->pw_uid, pw->pw_uid))
+ lerr(1, "unable to revoke privs");
+
+ SLIST_INIT(&slowcgi_proc.clients);
+
+ signal_set(&slowcgi_proc.ev_sigchld, SIGCHLD, slowcgi_sig_handler,
+ &slowcgi_proc);
+ signal_set(&slowcgi_proc.ev_sigpipe, SIGPIPE, slowcgi_sig_handler,
+ &slowcgi_proc);
+
+ signal_add(&slowcgi_proc.ev_sigchld, NULL);
+ signal_add(&slowcgi_proc.ev_sigpipe, NULL);
+
+ event_dispatch();
+ return (0);
+}
+void
+slowcgi_listen(const char *path, gid_t gid)
+{
+ struct listener *l = NULL;
+ struct sockaddr_un sun;
+ mode_t old_umask, mode;
+ int fd;
+
+ if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
+ lerr(1, "slowcgi_listen: socket");
+
+ bzero(&sun, sizeof(sun));
+ sun.sun_family = AF_UNIX;
+ strlcpy(sun.sun_path, path, sizeof(sun.sun_path));
+
+ if (unlink(path) == -1)
+ if (errno != ENOENT)
+ lerr(1, "slowcgi_listen: unlink %s", path);
+
+ old_umask = umask(S_IXUSR|S_IXGRP|S_IWOTH|S_IROTH|S_IXOTH);
+ mode = S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP;
+
+ if (bind(fd, (struct sockaddr *)&sun, sizeof(sun)) == -1)
+ lerr(1,"slowcgi_listen: bind: %s", path);
+
+ umask(old_umask);
+
+ if (chmod(path, mode) == -1)
+ lerr(1, "slowcgi_listen: chmod: %s", path);
+
+ if (chown(path, 0, gid) == -1)
+ lerr(1, "slowcgi_listen: chown: %s", path);
+
+ if (ioctl(fd, FIONBIO, &on) == -1)
+ lerr(1, "listener ioctl(FIONBIO)");
+
+ if (listen(fd, 5) == -1)
+ lerr(1, "listen");
+
+ l = calloc(1, sizeof(*l));
+ if (l == NULL)
+ lerr(1, "listener ev alloc");
+
+ event_set(&l->ev, fd, EV_READ | EV_PERSIST, slowcgi_accept, l);
+ event_add(&l->ev, NULL);
+ evtimer_set(&l->pause, slowcgi_paused, l);
+}
+
+void
+slowcgi_paused(int fd, short events, void *arg)
+{
+ struct listener *l = arg;
+ event_add(&l->ev, NULL);
+}
+
+void
+slowcgi_accept(int fd, short events, void *arg)
+{
+ struct listener *l;
+ struct sockaddr_storage ss;
+ struct timeval pause;
+ struct client *c;
+ struct clients *clients;
+ socklen_t len;
+ int s;
+
+ l = arg;
+ pause.tv_sec = 1; pause.tv_usec = 0;
+ c = NULL;
+
+ len = sizeof(ss);
+ s = accept(fd, (struct sockaddr *)&ss, &len);
+ if (s == -1) {
+ switch (errno) {
+ case EINTR:
+ case EWOULDBLOCK:
+ case ECONNABORTED:
+ return;
+ case EMFILE:
+ case ENFILE:
+ event_del(&l->ev);
+ evtimer_add(&l->pause, &pause);
+ return;
+ default:
+ lerr(1, "accept");
+ }
+ }
+
+ if (ioctl(s, FIONBIO, &on) == -1)
+ lerr(1, "client ioctl(FIONBIO)");
+
+ c = calloc(1, sizeof(*c));
+ if (c == NULL) {
+ lwarn("cannot calloc client");
+ close(s);
+ return;
+ }
+ clients = calloc(1, sizeof(*clients));
+ if (c == NULL) {
+ lwarn("cannot calloc clients");
+ close(s);
+ free(c);
+ return;
+ }
+ c->fd = s;
+ c->buf_pos = 0;
+ c->buf_len = 0;
+ c->request_started = 0;
+ TAILQ_INIT(&c->response_head);
+ TAILQ_INIT(&c->stdin_head);
+
+ event_set(&c->ev, s, EV_READ | EV_PERSIST, slowcgi_request, c);
+ event_add(&c->ev, NULL);
+
+ event_set(&c->tmo, s, 0, slowcgi_timeout, c);
+ event_add(&c->tmo, &timeout);
+ clients->client = c;
+ SLIST_INSERT_HEAD(&slowcgi_proc.clients, clients, entry);
+}
+
+void
+slowcgi_timeout(int fd, short events, void *arg)
+{
+ cleanup_client((struct client*) arg);
+}
+
+void
+slowcgi_sig_handler(int sig, short event, void *arg)
+{
+ struct client *c;
+ struct clients *ncs;
+ struct slowcgi_proc *p;
+ pid_t pid;
+ int status;
+
+ p = arg;
+ c = NULL;
+
+ switch (sig) {
+ case SIGCHLD:
+ pid = wait(&status);
+ SLIST_FOREACH(ncs, &p->clients, entry)
+ if (ncs->client->script_pid == pid) {
+ c = ncs->client;
+ break;
+ }
+ if (c != NULL) {
+ c->script_status = WEXITSTATUS(status);
+ if (c->script_flags == (STDOUT_DONE | STDERR_DONE))
+ create_end_request(c);
+ c->script_flags |= SCRIPT_DONE;
+ }
+ case SIGPIPE:
+ /* ignore */
+ break;
+ default:
+ lerr(1, "unexpected signal: %d", sig);
+
+ }
+}
+
+void
+slowcgi_response(int fd, short events, void *arg)
+{
+ struct client *c;
+ struct fcgi_record_header *header;
+ struct fcgi_response *resp;
+ ssize_t n;
+
+ c = arg;
+
+ while ((resp = TAILQ_FIRST(&c->response_head))) {
+ header = (struct fcgi_record_header*) resp->data;
+ if (debug)
+ dump_fcgi_record_header("resp ", header);
+
+ n = write(fd, resp->data + resp->data_pos, resp->data_len);
+ if ( n == -1) {
+ if (errno == EAGAIN)
+ return;
+ cleanup_client(c);
+ return;
+ }
+ resp->data_pos += n;
+ resp->data_len -= n;
+ if(resp->data_len == 0) {
+ TAILQ_REMOVE(&c->response_head, resp, entry);
+ free(resp);
+ }
+ }
+
+ if (TAILQ_EMPTY(&c->response_head)) {
+ if (c->script_flags == (STDOUT_DONE | STDERR_DONE |
+ SCRIPT_DONE))
+ cleanup_client(c);
+ else
+ event_del(&c->resp_ev);
+ }
+}
+
+void
+slowcgi_request(int fd, short events, void *arg)
+{
+ struct client *c;
+ size_t n, parsed;
+
+ c = arg;
+ parsed = 0;
+
+ n = read(fd, c->buf+c->buf_pos+c->buf_len,
+ FCGI_RECORD_SIZE - c->buf_pos-c->buf_len);
+
+ switch (n) {
+ case -1:
+ switch (errno) {
+ case EINTR:
+ case EAGAIN:
+ return;
+ default:
+ goto fail;
+ }
+ break;
+
+ case 0:
+ ldebug("closed connection");
+ goto fail;
+ default:
+ break;
+ }
+
+ c->buf_len+=n;
+
+ do {
+ parsed = parse_request(c->buf+c->buf_pos, c->buf_len, c);
+ c->buf_pos += parsed;
+ c->buf_len -= parsed;
+ } while (parsed > 0 && c->buf_len > 0);
+
+ if (c->buf_len > 0) {
+ bcopy(c->buf+c->buf_pos, c->buf, c->buf_len);
+ c->buf_pos = 0;
+ }
+ return;
+fail:
+ cleanup_client(c);
+}
+
+void
+parse_begin_request(uint8_t *buf, uint16_t n, struct client *c, uint16_t id)
+{
+ struct fcgi_begin_request_body *b;
+
+ if (c->request_started) {
+ lwarnx("unexpected FCGI_BEGIN_REQUEST, ignoring");
+ return;
+ }
+
+ if (n != sizeof(struct fcgi_begin_request_body)) {
+ lwarnx("wrong size %d != %d", n,
+ sizeof(struct fcgi_begin_request_body));
+ return;
+ }
+
+ c->request_started = 1;
+ b = (struct fcgi_begin_request_body*) buf;
+
+ c->id = id;
+ SLIST_INIT(&c->env);
+ c->env_count = 0;
+}
+void
+parse_params(uint8_t *buf, uint16_t n, struct client *c, uint16_t id)
+{
+ struct env_val *env_entry;
+ uint32_t name_len, val_len;
+
+ if (!c->request_started) {
+ lwarnx("FCGI_PARAMS without FCGI_BEGIN_REQUEST, ignoring");
+ return;
+ }
+
+ if (c->id != id) {
+ lwarnx("unexpected id, ignoring");
+ return;
+ }
+
+ name_len = val_len = 0;
+
+ if (n == 0) {
+ exec_cgi(c);
+ return;
+ }
+ while (n > 0) {
+ if (buf[0] >> 7 == 0) {
+ name_len = buf[0];
+ n--;
+ buf++;
+ } else {
+ if (n > 3) {
+ name_len = ((buf[3] & 0x7f) << 24) +
+ (buf[2] << 16) + (buf[1] << 8) + buf[0];
+ n -= 4;
+ buf += 4;
+ } else
+ return;
+ }
+
+ if (n > 0) {
+ if (buf[0] >> 7 == 0) {
+ val_len = buf[0];
+ n--;
+ buf++;
+ } else {
+ if (n > 3) {
+ val_len = ((buf[3] & 0x7f) << 24) +
+ (buf[2] << 16) + (buf[1] << 8) +
+ buf[0];
+ n -= 4;
+ buf += 4;
+ } else
+ return;
+ }
+ }
+ if (n < name_len + val_len)
+ return;
+
+ if ((env_entry = malloc(sizeof(struct env_val))) == NULL) {
+ lwarnx("cannot allocate env_entry");
+ return;
+ }
+
+ if ((env_entry->val = calloc(sizeof(char), name_len + val_len +
+ 2)) == NULL) {
+ lwarnx("cannot allocate env_entry->val");
+ free(env_entry);
+ return;
+ }
+
+ bcopy(buf, env_entry->val, name_len);
+ buf += name_len; n -= name_len;
+
+ env_entry->val[name_len] = '\0';
+ if (val_len < MAXPATHLEN && strcmp(env_entry->val,
+ "SCRIPT_NAME") == 0) {
+ bcopy(buf, c->script_name, val_len);
+ c->script_name[val_len+1] = '\0';
+ }
+ env_entry->val[name_len] = '=';
+
+ bcopy(buf, (env_entry->val)+name_len+1, val_len);
+ buf += val_len; n -= val_len;
+
+ SLIST_INSERT_HEAD(&c->env, env_entry, entry);
+ c->env_count++;
+ }
+}
+
+void
+parse_stdin(uint8_t *buf, uint16_t n, struct client *c, uint16_t id)
+{
+ struct fcgi_stdin *node;
+
+ if (c->id != id) {
+ lwarnx("unexpected id, ignoring");
+ return;
+ }
+
+ if ((node = calloc(1, sizeof(struct fcgi_stdin))) == NULL) {
+ lwarnx("cannot calloc stdin node");
+ return;
+ }
+
+ bcopy(buf, node->data, n);
+ node->data_pos = 0;
+ node->data_len = n;
+
+ TAILQ_INSERT_TAIL(&c->stdin_head, node, entry);
+
+ event_del(&c->script_stdin_ev);
+ event_set(&c->script_stdin_ev, EVENT_FD(&c->script_ev), EV_WRITE |
+ EV_PERSIST, script_out, c);
+ event_add(&c->script_stdin_ev, NULL);
+
+}
+
+size_t
+parse_request(uint8_t *buf, size_t n, struct client *c)
+{
+ struct fcgi_record_header *h;
+
+ if (n < sizeof(struct fcgi_record_header))
+ return (0);
+
+ h = (struct fcgi_record_header*) buf;
+
+ if (debug)
+ dump_fcgi_record_header("", h);
+
+ if (n < sizeof(struct fcgi_record_header) + ntohs(h->content_len)
+ + h->padding_len)
+ return (0);
+
+ if (h->version != 1)
+ lerrx(1, "wrong version");
+
+ switch (h->type) {
+ case FCGI_BEGIN_REQUEST:
+ parse_begin_request(buf+sizeof(struct fcgi_record_header),
+ ntohs(h->content_len), c, ntohs(h->id));
+ break;
+ case FCGI_PARAMS:
+ parse_params(buf+sizeof(struct fcgi_record_header),
+ ntohs(h->content_len), c, ntohs(h->id));
+ break;
+ case FCGI_STDIN:
+ parse_stdin(buf+sizeof(struct fcgi_record_header),
+ ntohs(h->content_len), c, ntohs(h->id));
+ break;
+ default:
+ lwarnx("unimplemented type %d", h->type);
+ }
+
+ return (sizeof(struct fcgi_record_header) + ntohs(h->content_len)
+ + h->padding_len);
+}
+
+void
+exec_cgi(struct client *c)
+{
+ struct env_val *env_entry;
+ int s[2], s_err[2], i;
+ pid_t pid;
+ char *argv[2];
+ char **env;
+
+ i = 0;
+
+ if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, s) == -1)
+ lerr(1, "socketpair");
+ if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, s_err) == -1)
+ lerr(1, "socketpair");
+ ldebug("fork: %s", c->script_name);
+ switch(pid = fork()) {
+ case -1:
+ lwarn("fork");
+ return;
+ case 0:
+ /* Child process */
+ close(s[0]);
+ close(s_err[0]);
+ if (dup2(s[1], STDIN_FILENO) == -1)
+ _exit(1);
+ if (dup2(s[1], STDOUT_FILENO) == -1)
+ _exit(1);
+ if (dup2(s_err[1], STDERR_FILENO) == -1)
+ _exit(1);
+ argv[0] = c->script_name;
+ argv[1] = NULL;
+ if((env = calloc(c->env_count+1, sizeof(char*)))==NULL)
+ _exit(1);
+ SLIST_FOREACH(env_entry, &c->env, entry)
+ env[i++] = env_entry->val;
+ env[i++] = NULL;
+ execve(c->script_name, argv, env);
+ _exit(1);
+
+ }
+ /* Parent process*/
+ close(s[1]);
+ close(s_err[1]);
+ if (ioctl(s[0], FIONBIO, &on) == -1)
+ lerr(1, "script ioctl(FIONBIO)");
+ if (ioctl(s_err[0], FIONBIO, &on) == -1)
+ lerr(1, "script ioctl(FIONBIO)");
+
+ c->script_pid = pid;
+ event_set(&c->script_ev, s[0], EV_READ | EV_PERSIST, script_std_in, c);
+ event_add(&c->script_ev, NULL);
+ event_set(&c->script_err_ev, s_err[0], EV_READ | EV_PERSIST,
+ script_err_in, c);
+ event_add(&c->script_err_ev, NULL);
+}
+
+void
+create_end_request(struct client *c)
+{
+ struct fcgi_response *resp;
+ struct fcgi_record_header *header;
+ struct fcgi_end_request *end_request;
+
+ if ((resp = malloc(sizeof(struct fcgi_response))) == NULL) {
+ lwarnx("cannot malloc fcgi_response");
+ return;
+ }
+ header = (struct fcgi_record_header*) resp->data;
+ header->version = 1;
+ header->type = FCGI_END_REQUEST;
+ header->id = htons(c->id);
+ header->content_len = htons(sizeof(struct
+ fcgi_end_request));
+ header->padding_len = 0;
+ header->reserved = 0;
+ end_request = (struct fcgi_end_request *) resp->data +
+ sizeof(struct fcgi_record_header);
+ end_request->app_status = htonl(c->script_status);
+ end_request->protocol_status = FCGI_REQUEST_COMPLETE;
+ resp->data_pos = 0;
+ resp->data_len = sizeof(struct fcgi_end_request) +
+ sizeof(struct fcgi_record_header);
+ TAILQ_INSERT_TAIL(&c->response_head, resp, entry);
+}
+
+void
+script_in(int fd, struct event *ev, struct client *c, uint8_t type)
+{
+ struct fcgi_response *resp;
+ struct fcgi_record_header *header;
+ ssize_t n;
+
+ if ((resp = malloc(sizeof(struct fcgi_response))) == NULL) {
+ lwarnx("cannot malloc fcgi_response");
+ return;
+ }
+ header = (struct fcgi_record_header*) resp->data;
+ header->version = 1;
+ header->type = type;
+ header->id = htons(c->id);
+ header->padding_len = 0;
+ header->reserved = 0;
+
+ n = read(fd, resp->data + sizeof(struct fcgi_record_header),
+ FCGI_RECORD_SIZE);
+
+ if (n == -1) {
+ switch (errno) {
+ case EINTR:
+ case EAGAIN:
+ return;
+ default:
+ n = 0; /* fake empty FCGI_STD{OUT,ERR} response */
+ }
+ }
+ header->content_len = htons(n);
+ resp->data_pos = 0;
+ resp->data_len = n + sizeof(struct fcgi_record_header);
+ TAILQ_INSERT_TAIL(&c->response_head, resp, entry);
+
+ event_del(&c->resp_ev);
+ event_set(&c->resp_ev, EVENT_FD(&c->ev), EV_WRITE | EV_PERSIST,
+ slowcgi_response, c);
+ event_add(&c->resp_ev, NULL);
+
+ if (n == 0) {
+ if (type == FCGI_STDOUT)
+ c->script_flags |= STDOUT_DONE;
+ else
+ c->script_flags |= STDERR_DONE;
+
+ if (c->script_flags == (STDOUT_DONE | STDERR_DONE |
+ SCRIPT_DONE)) {
+ create_end_request(c);
+ }
+ event_del(ev);
+ close(fd);
+ }
+}
+
+void
+script_std_in(int fd, short events, void *arg)
+{
+ struct client *c = arg;
+ script_in(fd, &c->script_ev, c, FCGI_STDOUT);
+}
+
+void
+script_err_in(int fd, short events, void *arg)
+{
+ struct client *c = arg;
+ script_in(fd, &c->script_err_ev, c, FCGI_STDERR);
+}
+
+void
+script_out(int fd, short events, void *arg)
+{
+ struct client *c;
+ struct fcgi_stdin *node;
+ ssize_t n;
+
+ c = arg;
+
+ while ((node = TAILQ_FIRST(&c->stdin_head))) {
+ if (node->data_len == 0) { /* end of stdin marker */
+ shutdown(fd, SHUT_WR);
+ break;
+ }
+ n = write(fd, node->data + node->data_pos, node->data_len);
+ if ( n == -1) {
+ if (errno == EAGAIN)
+ return;
+ event_del(&c->script_stdin_ev);
+ return;
+ }
+ node->data_pos += n;
+ node->data_len -= n;
+ if(node->data_len == 0) {
+ TAILQ_REMOVE(&c->stdin_head, node, entry);
+ free(node);
+ }
+ }
+}
+
+void
+cleanup_client(struct client *c)
+{
+ struct fcgi_response *resp;
+ struct fcgi_stdin *stdin_node;
+ struct env_val *env_entry;
+ struct clients *ncs, *tcs;
+
+ evtimer_del(&c->tmo);
+ if (event_initialized(&c->ev))
+ event_del(&c->ev);
+ if (event_initialized(&c->resp_ev))
+ event_del(&c->resp_ev);
+ if (event_initialized(&c->script_ev)) {
+ close(EVENT_FD(&c->script_ev));
+ event_del(&c->script_ev);
+ }
+ if (event_initialized(&c->script_err_ev)) {
+ close(EVENT_FD(&c->script_err_ev));
+ event_del(&c->script_err_ev);
+ }
+ if (event_initialized(&c->script_stdin_ev)) {
+ close(EVENT_FD(&c->script_stdin_ev));
+ event_del(&c->script_stdin_ev);
+ }
+ close(c->fd);
+ while (!SLIST_EMPTY(&c->env)) {
+ env_entry = SLIST_FIRST(&c->env);
+ SLIST_REMOVE_HEAD(&c->env, entry);
+ free(env_entry->val);
+ free(env_entry);
+ }
+
+ while ((resp = TAILQ_FIRST(&c->response_head))) {
+ TAILQ_REMOVE(&c->response_head, resp, entry);
+ free(resp);
+ }
+ while ((stdin_node = TAILQ_FIRST(&c->stdin_head))) {
+ TAILQ_REMOVE(&c->stdin_head, stdin_node, entry);
+ free(stdin_node);
+ }
+ SLIST_FOREACH_SAFE(ncs, &slowcgi_proc.clients, entry, tcs) {
+ if(ncs->client == c) {
+ SLIST_REMOVE(&slowcgi_proc.clients, ncs, clients,
+ entry);
+ free(ncs);
+ break;
+ }
+ }
+ free(c);
+}
+
+void dump_fcgi_record_header(const char* p, struct fcgi_record_header *h)
+{
+ ldebug("%sversion: %d", p, h->version);
+ ldebug("%stype: %d", p, h->type);
+ ldebug("%srequestId: %d", p, ntohs(h->id));
+ ldebug("%scontentLength: %d", p, ntohs(h->content_len));
+ ldebug("%spaddingLength: %d", p, h->padding_len);
+ ldebug("%sreserved: %d", p, h->reserved);
+}
+
+void
+syslog_vstrerror(int e, int priority, const char *fmt, va_list ap)
+{
+ char *s;
+
+ if (vasprintf(&s, fmt, ap) == -1) {
+ syslog(LOG_EMERG, "unable to alloc in syslog_vstrerror");
+ exit(1);
+ }
+ syslog(priority, "%s: %s", s, strerror(e));
+ free(s);
+}
+
+void
+syslog_err(int ecode, const char *fmt, ...)
+{
+ va_list ap;
+
+ va_start(ap, fmt);
+ syslog_vstrerror(errno, LOG_EMERG, fmt, ap);
+ va_end(ap);
+ exit(ecode);
+}
+
+void
+syslog_errx(int ecode, const char *fmt, ...)
+{
+ va_list ap;
+
+ va_start(ap, fmt);
+ vsyslog(LOG_WARNING, fmt, ap);
+ va_end(ap);
+ exit(ecode);
+}
+
+void
+syslog_warn(const char *fmt, ...)
+{
+ va_list ap;
+
+ va_start(ap, fmt);
+ syslog_vstrerror(errno, LOG_WARNING, fmt, ap);
+ va_end(ap);
+}
+
+void
+syslog_warnx(const char *fmt, ...)
+{
+ va_list ap;
+
+ va_start(ap, fmt);
+ vsyslog(LOG_WARNING, fmt, ap);
+ va_end(ap);
+}
+
+void
+syslog_info(const char *fmt, ...)
+{
+ va_list ap;
+
+ va_start(ap, fmt);
+ vsyslog(LOG_INFO, fmt, ap);
+ va_end(ap);
+}
+
+void
+syslog_debug(const char *fmt, ...)
+{
+ va_list ap;
+
+ if (!debug)
+ return;
+
+ va_start(ap, fmt);
+ vsyslog(LOG_DEBUG, fmt, ap);
+ va_end(ap);
+}