aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorOmar Polo <op@omarpolo.com>2021-02-08 10:01:45 +0000
committerOmar Polo <op@omarpolo.com>2021-02-08 10:01:45 +0000
commitabc007d2b32f597054399bd8fe72c1fbffdfb6a9 (patch)
tree1380b46057b39b60df96c3a32883f80967325df4
parentb63e30ff449ee1cf0091d5431f9b72cdd3b1e7e0 (diff)
rewrite main loop using libevent
-rw-r--r--gmid.h13
-rw-r--r--server.c472
2 files changed, 236 insertions, 249 deletions
diff --git a/gmid.h b/gmid.h
index f6567ce..5067b98 100644
--- a/gmid.h
+++ b/gmid.h
@@ -24,7 +24,6 @@
#include <netinet/in.h>
#include <dirent.h>
-#include <poll.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
@@ -34,10 +33,6 @@
#include "config.h"
-#ifndef INFTIM
-# define INFTIM -1
-#endif
-
#define GEMINI_URL_LEN (1024+3) /* URL max len + \r\n + \0 */
#define SUCCESS 20
@@ -134,7 +129,7 @@ struct parser {
struct client;
-typedef void (*statefn)(struct pollfd*, struct client*);
+typedef void (*statefn)(int, short, void*);
/*
* DFA: handle_handshake is the initial state, close_conn the final.
@@ -165,14 +160,14 @@ struct client {
char req[GEMINI_URL_LEN];
struct iri iri;
char domain[DOMAIN_NAME_LEN];
- statefn state, next;
+ statefn next;
int code;
const char *meta;
- int fd, waiting_on_child;
+ int fd, pfd;
+ DIR *dir;
char sbuf[1024]; /* static buffer */
void *buf, *i; /* mmap buffer */
ssize_t len, off; /* mmap/static buffer */
- DIR *dir;
struct sockaddr_storage addr;
struct vhost *host; /* host she's talking to */
};
diff --git a/server.c b/server.c
index 6feb7b5..daeeb93 100644
--- a/server.c
+++ b/server.c
@@ -21,6 +21,7 @@
#include <assert.h>
#include <errno.h>
+#include <event.h>
#include <fcntl.h>
#include <fnmatch.h>
#include <limits.h>
@@ -28,31 +29,57 @@
#include "gmid.h"
+struct server {
+ struct client clients[MAX_USERS];
+ struct tls *ctx;
+};
+
+struct server_events {
+ struct event e4;
+ int has_ipv6;
+ struct event e6;
+ struct event sighup;
+};
+
int connected_clients;
+static inline void reschedule_read(int, struct client*, statefn);
+static inline void reschedule_write(int, struct client*, statefn);
+
static int check_path(struct client*, const char*, int*);
-static void open_file(struct pollfd*, struct client*);
-static void load_file(struct pollfd*, struct client*);
-static void check_for_cgi(struct pollfd*, struct client*);
-static void handle_handshake(struct pollfd*, struct client*);
-static int apply_block_return(struct pollfd*, struct client*);
-static void handle_open_conn(struct pollfd*, struct client*);
-static void start_reply(struct pollfd*, struct client*, int, const char*);
-static void handle_start_reply(struct pollfd*, struct client*);
-static void start_cgi(const char*, const char*, struct pollfd*, struct client*);
-static void send_file(struct pollfd*, struct client*);
-static void open_dir(struct pollfd*, struct client*);
-static void redirect_canonical_dir(struct pollfd*, struct client*);
-static void enter_handle_dirlist(struct pollfd*, struct client*);
-static void handle_dirlist(struct pollfd*, struct client*);
+static void open_file(struct client*);
+static void load_file(struct client*);
+static void check_for_cgi(struct client*);
+static void handle_handshake(int, short, void*);
+static int apply_block_return(struct client*);
+static void handle_open_conn(int, short, void*);
+static void start_reply(struct client*, int, const char*);
+static void handle_start_reply(int, short, void*);
+static void start_cgi(const char*, const char*, struct client*);
+static void send_file(int, short, void*);
+static void open_dir(struct client*);
+static void redirect_canonical_dir(struct client*);
+static void enter_handle_dirlist(int, short, void*);
+static void handle_dirlist(int, short, void*);
static int read_next_dir_entry(struct client*);
-static void send_directory_listing(struct pollfd*, struct client*);
-static void cgi_poll_on_child(struct pollfd*, struct client*);
-static void cgi_poll_on_client(struct pollfd*, struct client*);
-static void handle_cgi_reply(struct pollfd*, struct client*);
-static void handle_cgi(struct pollfd*, struct client*);
-static void close_conn(struct pollfd*, struct client*);
-static void do_accept(int, struct tls*, struct pollfd*, struct client*);
+static void send_directory_listing(int, short, void*);
+static void handle_cgi_reply(int, short, void*);
+static void handle_cgi(int, short, void*);
+static void close_conn(int, short, void*);
+static void do_accept(int, short, void*);
+static void handle_sighup(int, short, void*);
+
+static inline void
+reschedule_read(int fd, struct client *c, statefn fn)
+{
+ event_once(fd, EV_READ, fn, c, NULL);
+}
+
+void
+reschedule_write(int fd, struct client *c, statefn fn)
+{
+ event_once(fd, EV_WRITE, fn, c, NULL);
+}
const char *
vhost_lang(struct vhost *v, const char *path)
@@ -211,31 +238,31 @@ check_path(struct client *c, const char *path, int *fd)
}
static void
-open_file(struct pollfd *fds, struct client *c)
+open_file(struct client *c)
{
- switch (check_path(c, c->iri.path, &c->fd)) {
+ switch (check_path(c, c->iri.path, &c->pfd)) {
case FILE_EXECUTABLE:
if (c->host->cgi != NULL && !fnmatch(c->host->cgi, c->iri.path, 0)) {
- start_cgi(c->iri.path, "", fds, c);
+ start_cgi(c->iri.path, "", c);
return;
}
/* fallthrough */
case FILE_EXISTS:
- load_file(fds, c);
+ load_file(c);
return;
case FILE_DIRECTORY:
- open_dir(fds, c);
+ open_dir(c);
return;
case FILE_MISSING:
if (c->host->cgi != NULL && !fnmatch(c->host->cgi, c->iri.path, 0)) {
- check_for_cgi(fds, c);
+ check_for_cgi(c);
return;
}
- start_reply(fds, c, NOT_FOUND, "not found");
+ start_reply(c, NOT_FOUND, "not found");
return;
default:
@@ -245,24 +272,24 @@ open_file(struct pollfd *fds, struct client *c)
}
static void
-load_file(struct pollfd *fds, struct client *c)
+load_file(struct client *c)
{
- if ((c->len = filesize(c->fd)) == -1) {
+ if ((c->len = filesize(c->pfd)) == -1) {
log_err(c, "failed to get file size for %s: %s",
c->iri.path, strerror(errno));
- start_reply(fds, c, TEMP_FAILURE, "internal server error");
+ start_reply(c, TEMP_FAILURE, "internal server error");
return;
}
if ((c->buf = mmap(NULL, c->len, PROT_READ, MAP_PRIVATE,
- c->fd, 0)) == MAP_FAILED) {
+ c->pfd, 0)) == MAP_FAILED) {
log_err(c, "mmap: %s: %s", c->iri.path, strerror(errno));
- start_reply(fds, c, TEMP_FAILURE, "internal server error");
+ start_reply(c, TEMP_FAILURE, "internal server error");
return;
}
c->i = c->buf;
c->next = send_file;
- start_reply(fds, c, SUCCESS, mime(c->host, c->iri.path));
+ start_reply(c, SUCCESS, mime(c->host, c->iri.path));
}
/*
@@ -273,7 +300,7 @@ load_file(struct pollfd *fds, struct client *c)
* executable is found or we emptied the path.
*/
static void
-check_for_cgi(struct pollfd *fds, struct client *c)
+check_for_cgi(struct client *c)
{
char path[PATH_MAX];
char *end;
@@ -281,8 +308,6 @@ check_for_cgi(struct pollfd *fds, struct client *c)
strlcpy(path, c->iri.path, sizeof(path));
end = strchr(path, '\0');
- /* NB: assume CGI is enabled and path matches cgi */
-
while (end > path) {
/* go up one level. UNIX paths are simple and POSIX
* dirname, with its ambiguities on if the given path
@@ -291,9 +316,9 @@ check_for_cgi(struct pollfd *fds, struct client *c)
end--;
*end = '\0';
- switch (check_path(c, path, &c->fd)) {
+ switch (check_path(c, path, &c->pfd)) {
case FILE_EXECUTABLE:
- start_cgi(path, end+1, fds, c);
+ start_cgi(path, end+1, c);
return;
case FILE_MISSING:
break;
@@ -306,7 +331,7 @@ check_for_cgi(struct pollfd *fds, struct client *c)
}
err:
- start_reply(fds, c, NOT_FOUND, "not found");
+ start_reply(c, NOT_FOUND, "not found");
return;
}
@@ -322,8 +347,9 @@ mark_nonblock(int fd)
}
static void
-handle_handshake(struct pollfd *fds, struct client *c)
+handle_handshake(int fd, short ev, void *d)
{
+ struct client *c = d;
struct vhost *h;
const char *servname;
const char *parse_err = "unknown error";
@@ -333,10 +359,10 @@ handle_handshake(struct pollfd *fds, struct client *c)
case -1: /* already handshaked */
break;
case TLS_WANT_POLLIN:
- fds->events = POLLIN;
+ reschedule_read(fd, c, &handle_handshake);
return;
case TLS_WANT_POLLOUT:
- fds->events = POLLOUT;
+ reschedule_write(fd, c, &handle_handshake);
return;
default:
/* unreachable */
@@ -361,8 +387,7 @@ handle_handshake(struct pollfd *fds, struct client *c)
if (h->domain != NULL) {
c->host = h;
- c->state = handle_open_conn;
- c->state(fds, c);
+ handle_open_conn(fd, ev, c);
return;
}
@@ -372,12 +397,12 @@ err:
else
strncpy(c->req, "null", sizeof(c->req));
- start_reply(fds, c, BAD_REQUEST, "Wrong/malformed host or missing SNI");
+ start_reply(c, BAD_REQUEST, "Wrong/malformed host or missing SNI");
}
/* 1 if a matching `block return' (and apply it), 0 otherwise */
static int
-apply_block_return(struct pollfd *fds, struct client *c)
+apply_block_return(struct client *c)
{
char *t, *path, buf[32];
const char *fmt;
@@ -438,13 +463,14 @@ apply_block_return(struct pollfd *fds, struct client *c)
if (i != 0)
strlcat(c->sbuf, buf, sizeof(c->sbuf));
- start_reply(fds, c, code, c->sbuf);
+ start_reply(c, code, c->sbuf);
return 1;
}
static void
-handle_open_conn(struct pollfd *fds, struct client *c)
+handle_open_conn(int fd, short ev, void *d)
{
+ struct client *c = d;
const char *parse_err = "invalid request";
char decoded[DOMAIN_NAME_LEN];
@@ -454,15 +480,15 @@ handle_open_conn(struct pollfd *fds, struct client *c)
switch (tls_read(c->ctx, c->req, sizeof(c->req)-1)) {
case -1:
log_err(c, "tls_read: %s", tls_error(c->ctx));
- close_conn(fds, c);
+ close_conn(fd, ev, c);
return;
case TLS_WANT_POLLIN:
- fds->events = POLLIN;
+ reschedule_read(fd, c, &handle_open_conn);
return;
case TLS_WANT_POLLOUT:
- fds->events = POLLOUT;
+ reschedule_write(fd, c, &handle_open_conn);
return;
}
@@ -470,40 +496,40 @@ handle_open_conn(struct pollfd *fds, struct client *c)
|| !parse_iri(c->req, &c->iri, &parse_err)
|| !puny_decode(c->iri.host, decoded, sizeof(decoded), &parse_err)) {
log_info(c, "iri parse error: %s", parse_err);
- start_reply(fds, c, BAD_REQUEST, "invalid request");
+ start_reply(c, BAD_REQUEST, "invalid request");
return;
}
if (c->iri.port_no != conf.port
|| strcmp(c->iri.schema, "gemini")
|| strcmp(decoded, c->domain)) {
- start_reply(fds, c, PROXY_REFUSED, "won't proxy request");
+ start_reply(c, PROXY_REFUSED, "won't proxy request");
return;
}
- if (apply_block_return(fds, c))
+ if (apply_block_return(c))
return;
if (c->host->entrypoint != NULL) {
- start_cgi(c->host->entrypoint, c->iri.path, fds, c);
+ start_cgi(c->host->entrypoint, c->iri.path, c);
return;
}
- open_file(fds, c);
+ open_file(c);
}
static void
-start_reply(struct pollfd *pfd, struct client *c, int code, const char *meta)
+start_reply(struct client *c, int code, const char *meta)
{
c->code = code;
c->meta = meta;
- c->state = handle_start_reply;
- handle_start_reply(pfd, c);
+ handle_start_reply(c->fd, 0, c);
}
static void
-handle_start_reply(struct pollfd *pfd, struct client *c)
+handle_start_reply(int fd, short ev, void *d)
{
+ struct client *c = d;
char buf[1030]; /* status + ' ' + max reply len + \r\n\0 */
const char *lang;
size_t len;
@@ -522,32 +548,26 @@ handle_start_reply(struct pollfd *pfd, struct client *c)
switch (tls_write(c->ctx, buf, len)) {
case -1:
- close_conn(pfd, c);
+ close_conn(fd, ev, c);
return;
case TLS_WANT_POLLIN:
- pfd->events = POLLIN;
+ reschedule_read(fd, c, &handle_start_reply);
return;
case TLS_WANT_POLLOUT:
- pfd->events = POLLOUT;
+ reschedule_write(fd, c, &handle_start_reply);
return;
}
log_request(c, buf, sizeof(buf));
- /* we don't need a body */
- if (c->code != SUCCESS) {
- close_conn(pfd, c);
- return;
- }
-
- /* advance the state machine */
- c->state = c->next;
- c->state(pfd, c);
+ if (c->code != SUCCESS)
+ close_conn(fd, ev, c);
+ else
+ c->next(fd, ev, c);
}
static void
-start_cgi(const char *spath, const char *relpath,
- struct pollfd *fds, struct client *c)
+start_cgi(const char *spath, const char *relpath, struct client *c)
{
char addr[NI_MAXHOST];
int e;
@@ -571,14 +591,13 @@ start_cgi(const char *spath, const char *relpath,
|| !send_vhost(exfd, c->host))
goto err;
- close(c->fd);
- if ((c->fd = recv_fd(exfd)) == -1) {
- start_reply(fds, c, TEMP_FAILURE, "internal server error");
+ close(c->pfd);
+ if ((c->pfd = recv_fd(exfd)) == -1) {
+ start_reply(c, TEMP_FAILURE, "internal server error");
return;
}
- cgi_poll_on_child(fds, c);
- c->state = handle_cgi_reply;
+ reschedule_read(c->pfd, c, &handle_cgi_reply);
return;
err:
@@ -587,8 +606,9 @@ err:
}
static void
-send_file(struct pollfd *fds, struct client *c)
+send_file(int fd, short ev, void *d)
{
+ struct client *c = d;
ssize_t ret, len;
len = (c->buf + c->len) - c->i;
@@ -597,15 +617,15 @@ send_file(struct pollfd *fds, struct client *c)
switch (ret = tls_write(c->ctx, c->i, len)) {
case -1:
log_err(c, "tls_write: %s", tls_error(c->ctx));
- close_conn(fds, c);
+ close_conn(fd, ev, c);
return;
case TLS_WANT_POLLIN:
- fds->events = POLLIN;
+ reschedule_read(fd, c, &send_file);
return;
case TLS_WANT_POLLOUT:
- fds->events = POLLOUT;
+ reschedule_write(fd, c, &send_file);
return;
default:
@@ -615,11 +635,11 @@ send_file(struct pollfd *fds, struct client *c)
}
}
- close_conn(fds, c);
+ close_conn(fd, ev, c);
}
static void
-open_dir(struct pollfd *fds, struct client *c)
+open_dir(struct client *c)
{
size_t len;
int dirfd;
@@ -627,7 +647,7 @@ open_dir(struct pollfd *fds, struct client *c)
len = strlen(c->iri.path);
if (len > 0 && !ends_with(c->iri.path, "/")) {
- redirect_canonical_dir(fds, c);
+ redirect_canonical_dir(c);
return;
}
@@ -639,53 +659,53 @@ open_dir(struct pollfd *fds, struct client *c)
len = strlcat(c->sbuf, vhost_index(c->host, c->iri.path),
sizeof(c->sbuf));
if (len >= sizeof(c->sbuf)) {
- start_reply(fds, c, TEMP_FAILURE, "internal server error");
+ start_reply(c, TEMP_FAILURE, "internal server error");
return;
}
c->iri.path = c->sbuf;
/* close later unless we have to generate the dir listing */
- dirfd = c->fd;
- c->fd = -1;
+ dirfd = c->pfd;
+ c->pfd = -1;
- switch (check_path(c, c->iri.path, &c->fd)) {
+ switch (check_path(c, c->iri.path, &c->pfd)) {
case FILE_EXECUTABLE:
if (c->host->cgi != NULL && !fnmatch(c->host->cgi, c->iri.path, 0)) {
- start_cgi(c->iri.path, "", fds, c);
+ start_cgi(c->iri.path, "", c);
break;
}
/* fallthrough */
case FILE_EXISTS:
- load_file(fds, c);
+ load_file(c);
break;
case FILE_DIRECTORY:
- start_reply(fds, c, TEMP_REDIRECT, c->sbuf);
+ start_reply(c, TEMP_REDIRECT, c->sbuf);
break;
case FILE_MISSING:
*before_file = '\0';
if (!vhost_auto_index(c->host, c->iri.path)) {
- start_reply(fds, c, NOT_FOUND, "not found");
+ start_reply(c, NOT_FOUND, "not found");
break;
}
- c->fd = dirfd;
+ c->pfd = dirfd;
c->next = enter_handle_dirlist;
- if ((c->dir = fdopendir(c->fd)) == NULL) {
+ if ((c->dir = fdopendir(c->pfd)) == NULL) {
log_err(c, "fdopendir(%d) (vhost:%s) %s: %s",
- c->fd, c->host->domain, c->iri.path, strerror(errno));
- start_reply(fds, c, TEMP_FAILURE, "internal server error");
+ c->pfd, c->host->domain, c->iri.path, strerror(errno));
+ start_reply(c, TEMP_FAILURE, "internal server error");
return;
}
c->off = 0;
- start_reply(fds, c, SUCCESS, "text/gemini");
+ start_reply(c, SUCCESS, "text/gemini");
return;
default:
@@ -697,7 +717,7 @@ open_dir(struct pollfd *fds, struct client *c)
}
static void
-redirect_canonical_dir(struct pollfd *fds, struct client *c)
+redirect_canonical_dir(struct client *c)
{
size_t len;
@@ -706,16 +726,17 @@ redirect_canonical_dir(struct pollfd *fds, struct client *c)
len = strlcat(c->sbuf, "/", sizeof(c->sbuf));
if (len >= sizeof(c->sbuf)) {
- start_reply(fds, c, TEMP_FAILURE, "internal server error");
+ start_reply(c, TEMP_FAILURE, "internal server error");
return;
}
- start_reply(fds, c, TEMP_REDIRECT, c->sbuf);
+ start_reply(c, TEMP_REDIRECT, c->sbuf);
}
static void
-enter_handle_dirlist(struct pollfd *fds, struct client *c)
+enter_handle_dirlist(int fd, short ev, void *d)
{
+ struct client *c = d;
char b[PATH_MAX];
size_t l;
@@ -727,30 +748,30 @@ enter_handle_dirlist(struct pollfd *fds, struct client *c)
* in c->sbuf to hold the ancilliary string plus the
* full path; but it wouldn't read nice without some
* error checking, and I'd like to avoid a strlen. */
- close_conn(fds, c);
+ close_conn(fd, ev, c);
return;
}
- c->len = l;
- c->state = handle_dirlist;
- handle_dirlist(fds, c);
+ c->len = l;
+ handle_dirlist(fd, ev, c);
}
static void
-handle_dirlist(struct pollfd *fds, struct client *c)
+handle_dirlist(int fd, short ev, void *d)
{
+ struct client *c = d;
ssize_t r;
while (c->len > 0) {
switch (r = tls_write(c->ctx, c->sbuf + c->off, c->len)) {
case -1:
- close_conn(fds, c);
+ close_conn(fd, ev, c);
return;
case TLS_WANT_POLLOUT:
- fds->events = POLLOUT;
+ reschedule_read(fd, c, &handle_dirlist);
return;
case TLS_WANT_POLLIN:
- fds->events = POLLIN;
+ reschedule_write(fd, c, &handle_dirlist);
return;
default:
c->off += r;
@@ -758,8 +779,7 @@ handle_dirlist(struct pollfd *fds, struct client *c)
}
}
- c->state = send_directory_listing;
- send_directory_listing(fds, c);
+ send_directory_listing(fd, ev, c);
}
static int
@@ -786,8 +806,9 @@ read_next_dir_entry(struct client *c)
}
static void
-send_directory_listing(struct pollfd *fds, struct client *c)
+send_directory_listing(int fd, short ev, void *d)
{
+ struct client *c = d;
ssize_t r;
while (1) {
@@ -802,11 +823,11 @@ send_directory_listing(struct pollfd *fds, struct client *c)
goto end;
case TLS_WANT_POLLOUT:
- fds->events = POLLOUT;
+ reschedule_read(fd, c, &send_directory_listing);
return;
case TLS_WANT_POLLIN:
- fds->events = POLLIN;
+ reschedule_write(fd, c, &send_directory_listing);
return;
default:
@@ -818,55 +839,25 @@ send_directory_listing(struct pollfd *fds, struct client *c)
}
end:
- close_conn(fds, c);
-}
-
-static inline void
-cgi_poll_on_child(struct pollfd *fds, struct client *c)
-{
- int fd;
-
- if (c->waiting_on_child)
- return;
- c->waiting_on_child = 1;
-
- fds->events = POLLIN;
-
- fd = fds->fd;
- fds->fd = c->fd;
- c->fd = fd;
-}
-
-static inline void
-cgi_poll_on_client(struct pollfd *fds, struct client *c)
-{
- int fd;
-
- if (!c->waiting_on_child)
- return;
- c->waiting_on_child = 0;
-
- fd = fds->fd;
- fds->fd = c->fd;
- c->fd = fd;
+ close_conn(fd, ev, d);
}
/* accumulate the meta line from the cgi script. */
static void
-handle_cgi_reply(struct pollfd *fds, struct client *c)
+handle_cgi_reply(int fd, short ev, void *d)
{
+ struct client *c = d;
void *buf, *e;
size_t len;
ssize_t r;
+
buf = c->sbuf + c->len;
len = sizeof(c->sbuf) - c->len;
- /* we're polling on the child! */
- r = read(fds->fd, buf, len);
+ r = read(c->pfd, buf, len);
if (r == 0 || r == -1) {
- cgi_poll_on_client(fds, c);
- start_reply(fds, c, CGI_ERROR, "CGI error");
+ start_reply(c, CGI_ERROR, "CGI error");
return;
}
@@ -878,28 +869,27 @@ handle_cgi_reply(struct pollfd *fds, struct client *c)
log_request(c, c->sbuf, c->len);
c->off = 0;
- c->state = handle_cgi;
- c->state(fds, c);
+ handle_cgi(fd, ev, c);
return;
}
+
+ reschedule_read(fd, c, &handle_cgi_reply);
}
static void
-handle_cgi(struct pollfd *fds, struct client *c)
+handle_cgi(int fd, short ev, void *d)
{
+ struct client *c = d;
ssize_t r;
- /* ensure c->fd is the child and fds->fd the client */
- cgi_poll_on_client(fds, c);
-
while (1) {
if (c->len == 0) {
- switch (r = read(c->fd, c->sbuf, sizeof(c->sbuf))) {
+ switch (r = read(c->pfd, c->sbuf, sizeof(c->sbuf))) {
case 0:
goto end;
case -1:
if (errno == EAGAIN || errno == EWOULDBLOCK) {
- cgi_poll_on_child(fds, c);
+ reschedule_read(c->pfd, c, &handle_cgi);
return;
}
goto end;
@@ -915,11 +905,11 @@ handle_cgi(struct pollfd *fds, struct client *c)
goto end;
case TLS_WANT_POLLOUT:
- fds->events = POLLOUT;
+ reschedule_read(c->fd, c, &handle_cgi);
return;
case TLS_WANT_POLLIN:
- fds->events = POLLIN;
+ reschedule_write(c->fd, c, &handle_cgi);
return;
default:
@@ -931,20 +921,20 @@ handle_cgi(struct pollfd *fds, struct client *c)
}
end:
- close_conn(fds, c);
+ close_conn(c->fd, ev, d);
}
static void
-close_conn(struct pollfd *pfd, struct client *c)
+close_conn(int fd, short ev, void *d)
{
- c->state = close_conn;
+ struct client *c = d;
switch (tls_close(c->ctx)) {
case TLS_WANT_POLLIN:
- pfd->events = POLLIN;
+ reschedule_read(fd, c, &close_conn);
return;
case TLS_WANT_POLLOUT:
- pfd->events = POLLOUT;
+ reschedule_read(fd, c, &close_conn);
return;
}
@@ -956,49 +946,51 @@ close_conn(struct pollfd *pfd, struct client *c)
if (c->buf != MAP_FAILED)
munmap(c->buf, c->len);
- if (c->fd != -1)
- close(c->fd);
+ if (c->pfd != -1)
+ close(c->pfd);
if (c->dir != NULL)
closedir(c->dir);
- close(pfd->fd);
- pfd->fd = -1;
+ close(c->fd);
+ c->fd = -1;
}
static void
-do_accept(int sock, struct tls *ctx, struct pollfd *fds, struct client *clients)
+do_accept(int sock, short et, void *d)
{
- int i, fd;
+ struct client *c;
+ struct server *s = d;
struct sockaddr_storage addr;
+ struct sockaddr *saddr;
socklen_t len;
+ int i, fd;
+
+ (void)et;
+
+ saddr = (struct sockaddr*)&addr;
len = sizeof(addr);
- if ((fd = accept(sock, (struct sockaddr*)&addr, &len)) == -1) {
+ if ((fd = accept4(sock, saddr, &len, SOCK_NONBLOCK)) == -1) {
if (errno == EWOULDBLOCK || errno == EAGAIN)
return;
fatal("accept: %s", strerror(errno));
}
- mark_nonblock(fd);
-
for (i = 0; i < MAX_USERS; ++i) {
- if (fds[i].fd == -1) {
- bzero(&clients[i], sizeof(struct client));
- if (tls_accept_socket(ctx, &clients[i].ctx, fd) == -1)
+ c = &s->clients[i];
+ if (c->fd == -1) {
+ memset(c, 0, sizeof(*c));
+ if (tls_accept_socket(s->ctx, &c->ctx, fd) == -1)
break; /* goodbye fd! */
- fds[i].fd = fd;
- fds[i].events = POLLIN;
-
- clients[i].state = handle_handshake;
- clients[i].next = send_file;
- clients[i].fd = -1;
- clients[i].waiting_on_child = 0;
- clients[i].buf = MAP_FAILED;
- clients[i].dir = NULL;
- clients[i].addr = addr;
+ c->fd = fd;
+ c->pfd = -1;
+ c->buf = MAP_FAILED;
+ c->dir = NULL;
+ c->addr = addr;
+ reschedule_read(fd, c, &handle_handshake);
connected_clients++;
return;
}
@@ -1007,65 +999,65 @@ do_accept(int sock, struct tls *ctx, struct pollfd *fds, struct client *clients)
close(fd);
}
-void
-loop(struct tls *ctx, int sock4, int sock6)
+static void
+handle_sighup(int fd, short ev, void *d)
{
- int i, n;
- struct client clients[MAX_USERS];
- struct pollfd fds[MAX_USERS];
+ struct server_events *events = d;
- for (i = 0; i < MAX_USERS; ++i) {
- fds[i].fd = -1;
- fds[i].events = POLLIN;
- bzero(&clients[i], sizeof(struct client));
- }
+ (void)fd;
+ (void)ev;
- fds[0].fd = sock4;
- fds[1].fd = sock6;
-
- for (;;) {
- if ((n = poll(fds, MAX_USERS, INFTIM)) == -1) {
- if (errno == EINTR) {
- log_info(NULL, "%d connected clients",
- connected_clients);
- } else
- fatal("poll: %s", strerror(errno));
- }
+ event_del(&events->e4);
+ if (events->has_ipv6)
+ event_del(&events->e6);
+ signal_del(&events->sighup);
+}
- for (i = 0; i < MAX_USERS && n > 0; i++) {
- if (fds[i].revents == 0)
- continue;
+static void
+handle_siginfo(int fd, short ev, void *d)
+{
+ (void)fd;
+ (void)ev;
+ (void)d;
- if (fds[i].revents & (POLLERR|POLLNVAL))
- fatal("bad fd %d: %s", fds[i].fd,
- strerror(errno));
+ log_info(NULL, "%d connected clients", connected_clients);
+}
- n--;
+void
+loop(struct tls *ctx, int sock4, int sock6)
+{
+ struct server_events events;
+ struct server server;
+ struct event info;
+ size_t i;
- if (fds[i].revents & POLLHUP) {
- /* fds[i] may be the fd of the stdin
- * of a cgi script that has exited. */
- if (!clients[i].waiting_on_child) {
- close_conn(&fds[i], &clients[i]);
- continue;
- }
- }
+ event_init();
- if (fds[i].fd == sock4)
- do_accept(sock4, ctx, fds, clients);
- else if (fds[i].fd == sock6)
- do_accept(sock6, ctx, fds, clients);
- else
- clients[i].state(&fds[i], &clients[i]);
- }
+ memset(&events, 0, sizeof(events));
+ memset(&server, 0, sizeof(server));
+ for (i = 0; i < MAX_USERS; ++i)
+ server.clients[i].fd = -1;
- if (hupped) {
- if (connected_clients == 0)
- return;
+ event_set(&events.e4, sock4, EV_READ | EV_PERSIST, &do_accept, &server);
+ event_add(&events.e4, NULL);
- fds[0].fd = -1;
- if (sock6 != -1)
- fds[1].fd = -1;
- }
+ if (sock6 != -1) {
+ events.has_ipv6 = 1;
+ event_set(&events.e6, sock6, EV_READ | EV_PERSIST, &do_accept, &server);
+ event_add(&events.e6, NULL);
}
+
+ signal_set(&events.sighup, SIGHUP, &handle_sighup, &events);
+ signal_add(&events.sighup, NULL);
+
+#ifdef SIGINFO
+ signal_set(&info, SIGINFO, &handle_siginfo, NULL);
+ signal_add(&info, NULL);
+#endif
+ signal_set(&info, SIGUSR2, &handle_siginfo, NULL);
+ signal_add(&info, NULL);
+
+ server.ctx = ctx;
+
+ event_dispatch();
}