2 d3a08f4d 2021-01-17 op * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
4 d3a08f4d 2021-01-17 op * Permission to use, copy, modify, and distribute this software for any
5 d3a08f4d 2021-01-17 op * purpose with or without fee is hereby granted, provided that the above
6 d3a08f4d 2021-01-17 op * copyright notice and this permission notice appear in all copies.
8 d3a08f4d 2021-01-17 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 d3a08f4d 2021-01-17 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 d3a08f4d 2021-01-17 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 d3a08f4d 2021-01-17 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 d3a08f4d 2021-01-17 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 d3a08f4d 2021-01-17 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 d3a08f4d 2021-01-17 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 52418c8d 2021-02-12 op #include "gmid.h"
19 d3a08f4d 2021-01-17 op #include <sys/stat.h>
21 d3a08f4d 2021-01-17 op #include <assert.h>
22 d3a08f4d 2021-01-17 op #include <errno.h>
23 abc007d2 2021-02-08 op #include <event.h>
24 d3a08f4d 2021-01-17 op #include <fcntl.h>
25 b4d409cf 2021-01-21 op #include <fnmatch.h>
26 2fafa2d2 2021-02-01 op #include <limits.h>
27 d3a08f4d 2021-01-17 op #include <string.h>
29 bc99d868 2021-03-19 op static struct client clients[MAX_USERS];
30 bc99d868 2021-03-19 op static struct tls *ctx;
32 bc99d868 2021-03-19 op static struct event e4, e6, imsgev, siginfo, sigusr2;
33 6b191ed5 2021-02-23 op static int has_ipv6, has_siginfo;
35 d3a08f4d 2021-01-17 op int connected_clients;
37 49b73ba1 2021-02-10 op static inline int matches(const char*, const char*);
39 abc007d2 2021-02-08 op static inline void reschedule_read(int, struct client*, statefn);
40 abc007d2 2021-02-08 op static inline void reschedule_write(int, struct client*, statefn);
42 fe406389 2021-02-02 op static int check_path(struct client*, const char*, int*);
43 abc007d2 2021-02-08 op static void open_file(struct client*);
44 abc007d2 2021-02-08 op static void check_for_cgi(struct client*);
45 abc007d2 2021-02-08 op static void handle_handshake(int, short, void*);
46 57ec3e77 2021-02-08 op static char *strip_path(char*, int);
47 57ec3e77 2021-02-08 op static void fmt_sbuf(const char*, struct client*, const char*);
48 abc007d2 2021-02-08 op static int apply_block_return(struct client*);
49 02be96c6 2021-02-09 op static int apply_require_ca(struct client*);
50 abc007d2 2021-02-08 op static void handle_open_conn(int, short, void*);
51 abc007d2 2021-02-08 op static void start_reply(struct client*, int, const char*);
52 abc007d2 2021-02-08 op static void handle_start_reply(int, short, void*);
53 abc007d2 2021-02-08 op static void start_cgi(const char*, const char*, struct client*);
54 abc007d2 2021-02-08 op static void open_dir(struct client*);
55 abc007d2 2021-02-08 op static void redirect_canonical_dir(struct client*);
56 abc007d2 2021-02-08 op static void enter_handle_dirlist(int, short, void*);
57 abc007d2 2021-02-08 op static void handle_dirlist(int, short, void*);
58 fe406389 2021-02-02 op static int read_next_dir_entry(struct client*);
59 abc007d2 2021-02-08 op static void send_directory_listing(int, short, void*);
60 abc007d2 2021-02-08 op static void handle_cgi_reply(int, short, void*);
61 27b2fa9a 2021-02-12 op static void handle_copy(int, short, void*);
62 abc007d2 2021-02-08 op static void close_conn(int, short, void*);
63 abc007d2 2021-02-08 op static void do_accept(int, short, void*);
64 bc99d868 2021-03-19 op struct client *client_by_id(int);
65 bc99d868 2021-03-19 op static void handle_imsg_cgi_res(struct imsgbuf*, struct imsg*, size_t);
66 bc99d868 2021-03-19 op static void handle_imsg_quit(struct imsgbuf*, struct imsg*, size_t);
67 bc99d868 2021-03-19 op static void handle_siginfo(int, short, void*);
69 bc99d868 2021-03-19 op static imsg_handlerfn *handlers[] = {
70 bc99d868 2021-03-19 op [IMSG_QUIT] = handle_imsg_quit,
71 bc99d868 2021-03-19 op [IMSG_CGI_RES] = handle_imsg_cgi_res,
74 49b73ba1 2021-02-10 op static inline int
75 49b73ba1 2021-02-10 op matches(const char *pattern, const char *path)
77 49b73ba1 2021-02-10 op if (*path == '/')
79 49b73ba1 2021-02-10 op return !fnmatch(pattern, path, 0);
82 abc007d2 2021-02-08 op static inline void
83 abc007d2 2021-02-08 op reschedule_read(int fd, struct client *c, statefn fn)
85 abc007d2 2021-02-08 op event_once(fd, EV_READ, fn, c, NULL);
88 c39b26d3 2021-02-12 op static inline void
89 abc007d2 2021-02-08 op reschedule_write(int fd, struct client *c, statefn fn)
91 abc007d2 2021-02-08 op event_once(fd, EV_WRITE, fn, c, NULL);
95 c8b74339 2021-01-24 op vhost_lang(struct vhost *v, const char *path)
97 c8b74339 2021-01-24 op struct location *loc;
99 caad0308 2021-01-27 op if (v == NULL || path == NULL)
102 6016a593 2021-01-30 op for (loc = &v->locations[1]; loc->match != NULL; ++loc) {
103 cd761624 2021-02-06 op if (loc->lang != NULL) {
104 49b73ba1 2021-02-10 op if (matches(loc->match, path))
105 6016a593 2021-01-30 op return loc->lang;
109 6016a593 2021-01-30 op return v->locations[0].lang;
113 c8b74339 2021-01-24 op vhost_default_mime(struct vhost *v, const char *path)
115 c8b74339 2021-01-24 op struct location *loc;
116 c8b74339 2021-01-24 op const char *default_mime = "application/octet-stream";
118 caad0308 2021-01-27 op if (v == NULL || path == NULL)
119 caad0308 2021-01-27 op return default_mime;
121 6016a593 2021-01-30 op for (loc = &v->locations[1]; loc->match != NULL; ++loc) {
122 cd761624 2021-02-06 op if (loc->default_mime != NULL) {
123 49b73ba1 2021-02-10 op if (matches(loc->match, path))
124 6016a593 2021-01-30 op return loc->default_mime;
128 6016a593 2021-01-30 op if (v->locations[0].default_mime != NULL)
129 6016a593 2021-01-30 op return v->locations[0].default_mime;
130 c8b74339 2021-01-24 op return default_mime;
134 c8b74339 2021-01-24 op vhost_index(struct vhost *v, const char *path)
136 c8b74339 2021-01-24 op struct location *loc;
137 c8b74339 2021-01-24 op const char *index = "index.gmi";
139 caad0308 2021-01-27 op if (v == NULL || path == NULL)
140 caad0308 2021-01-27 op return index;
142 6016a593 2021-01-30 op for (loc = &v->locations[1]; loc->match != NULL; ++loc) {
143 cd761624 2021-02-06 op if (loc->index != NULL) {
144 49b73ba1 2021-02-10 op if (matches(loc->match, path))
145 6016a593 2021-01-30 op return loc->index;
149 b59f3cdd 2021-01-30 op if (v->locations[0].index != NULL)
150 b59f3cdd 2021-01-30 op return v->locations[0].index;
151 c8b74339 2021-01-24 op return index;
155 252908e6 2021-01-24 op vhost_auto_index(struct vhost *v, const char *path)
157 252908e6 2021-01-24 op struct location *loc;
159 6016a593 2021-01-30 op if (v == NULL || path == NULL)
162 daac4a94 2021-02-06 op for (loc = &v->locations[1]; loc->match != NULL; ++loc) {
163 cd761624 2021-02-06 op if (loc->auto_index != 0) {
164 49b73ba1 2021-02-10 op if (matches(loc->match, path))
165 6016a593 2021-01-30 op return loc->auto_index == 1;
169 6016a593 2021-01-30 op return v->locations[0].auto_index == 1;
173 6abda252 2021-02-06 op vhost_block_return(struct vhost *v, const char *path, int *code, const char **fmt)
175 6abda252 2021-02-06 op struct location *loc;
177 6abda252 2021-02-06 op if (v == NULL || path == NULL)
180 6abda252 2021-02-06 op for (loc = &v->locations[1]; loc->match != NULL; ++loc) {
181 cd761624 2021-02-06 op if (loc->block_code != 0) {
182 49b73ba1 2021-02-10 op if (matches(loc->match, path)) {
183 6abda252 2021-02-06 op *code = loc->block_code;
184 6abda252 2021-02-06 op *fmt = loc->block_fmt;
190 6abda252 2021-02-06 op *code = v->locations[0].block_code;
191 6abda252 2021-02-06 op *fmt = v->locations[0].block_fmt;
192 6abda252 2021-02-06 op return v->locations[0].block_code != 0;
196 6abda252 2021-02-06 op vhost_strip(struct vhost *v, const char *path)
198 6abda252 2021-02-06 op struct location *loc;
200 6abda252 2021-02-06 op if (v == NULL || path == NULL)
203 6abda252 2021-02-06 op for (loc = &v->locations[1]; loc->match != NULL; ++loc) {
204 cd761624 2021-02-06 op if (loc->strip != 0) {
205 49b73ba1 2021-02-10 op if (matches(loc->match, path))
206 6abda252 2021-02-06 op return loc->strip;
210 6abda252 2021-02-06 op return v->locations[0].strip;
214 02be96c6 2021-02-09 op vhost_require_ca(struct vhost *v, const char *path)
216 02be96c6 2021-02-09 op struct location *loc;
218 02be96c6 2021-02-09 op if (v == NULL || path == NULL)
221 02be96c6 2021-02-09 op for (loc = &v->locations[1]; loc->match != NULL; ++loc) {
222 02be96c6 2021-02-09 op if (loc->reqca != NULL) {
223 49b73ba1 2021-02-10 op if (matches(loc->match, path))
224 02be96c6 2021-02-09 op return loc->reqca;
228 02be96c6 2021-02-09 op return v->locations[0].reqca;
232 793835cb 2021-02-23 op vhost_disable_log(struct vhost *v, const char *path)
234 793835cb 2021-02-23 op struct location *loc;
236 793835cb 2021-02-23 op if (v == NULL || path == NULL)
239 793835cb 2021-02-23 op for (loc = &v->locations[1]; loc->match != NULL; ++loc) {
240 793835cb 2021-02-23 op if (loc->disable_log && matches(loc->match, path))
244 793835cb 2021-02-23 op return v->locations[0].disable_log;
248 d3a08f4d 2021-01-17 op check_path(struct client *c, const char *path, int *fd)
250 d3a08f4d 2021-01-17 op struct stat sb;
251 d1ca3911 2021-01-21 op const char *p;
254 d3a08f4d 2021-01-17 op assert(path != NULL);
256 d1ca3911 2021-01-21 op if (*path == '\0')
258 d1ca3911 2021-01-21 op else if (*path == '/')
259 d1ca3911 2021-01-21 op /* in send_dir we add an initial / (to be
260 d1ca3911 2021-01-21 op * redirect-friendly), but here we want to skip it */
265 252908e6 2021-01-24 op flags = O_RDONLY | O_NOFOLLOW;
267 252908e6 2021-01-24 op if (*fd == -1 && (*fd = openat(c->host->dirfd, p, flags)) == -1)
268 d3a08f4d 2021-01-17 op return FILE_MISSING;
270 d3a08f4d 2021-01-17 op if (fstat(*fd, &sb) == -1) {
271 3abf91b0 2021-02-07 op log_notice(c, "failed stat for %s: %s", path, strerror(errno));
272 d3a08f4d 2021-01-17 op return FILE_MISSING;
275 d3a08f4d 2021-01-17 op if (S_ISDIR(sb.st_mode))
276 d3a08f4d 2021-01-17 op return FILE_DIRECTORY;
278 d3a08f4d 2021-01-17 op if (sb.st_mode & S_IXUSR)
279 d3a08f4d 2021-01-17 op return FILE_EXECUTABLE;
281 d3a08f4d 2021-01-17 op return FILE_EXISTS;
285 abc007d2 2021-02-08 op open_file(struct client *c)
287 abc007d2 2021-02-08 op switch (check_path(c, c->iri.path, &c->pfd)) {
288 d3a08f4d 2021-01-17 op case FILE_EXECUTABLE:
289 eecad7a3 2021-02-12 op if (c->host->cgi != NULL && matches(c->host->cgi, c->iri.path)) {
290 abc007d2 2021-02-08 op start_cgi(c->iri.path, "", c);
294 d3a08f4d 2021-01-17 op /* fallthrough */
296 d3a08f4d 2021-01-17 op case FILE_EXISTS:
297 27b2fa9a 2021-02-12 op c->next = handle_copy;
298 27b2fa9a 2021-02-12 op start_reply(c, SUCCESS, mime(c->host, c->iri.path));
301 d3a08f4d 2021-01-17 op case FILE_DIRECTORY:
305 d3a08f4d 2021-01-17 op case FILE_MISSING:
306 eecad7a3 2021-02-12 op if (c->host->cgi != NULL && matches(c->host->cgi, c->iri.path)) {
307 abc007d2 2021-02-08 op check_for_cgi(c);
310 abc007d2 2021-02-08 op start_reply(c, NOT_FOUND, "not found");
314 d3a08f4d 2021-01-17 op /* unreachable */
320 d3a08f4d 2021-01-17 op * the inverse of this algorithm, i.e. starting from the start of the
321 d3a08f4d 2021-01-17 op * path + strlen(cgi), and checking if each component, should be
322 d3a08f4d 2021-01-17 op * faster. But it's tedious to write. This does the opposite: starts
323 d3a08f4d 2021-01-17 op * from the end and strip one component at a time, until either an
324 d3a08f4d 2021-01-17 op * executable is found or we emptied the path.
327 abc007d2 2021-02-08 op check_for_cgi(struct client *c)
329 2fafa2d2 2021-02-01 op char path[PATH_MAX];
332 2fafa2d2 2021-02-01 op strlcpy(path, c->iri.path, sizeof(path));
333 d3a08f4d 2021-01-17 op end = strchr(path, '\0');
335 d3a08f4d 2021-01-17 op while (end > path) {
336 d3a08f4d 2021-01-17 op /* go up one level. UNIX paths are simple and POSIX
337 d3a08f4d 2021-01-17 op * dirname, with its ambiguities on if the given path
338 d3a08f4d 2021-01-17 op * is changed or not, gives me headaches. */
339 d3a08f4d 2021-01-17 op while (*end != '/')
343 abc007d2 2021-02-08 op switch (check_path(c, path, &c->pfd)) {
344 d3a08f4d 2021-01-17 op case FILE_EXECUTABLE:
345 abc007d2 2021-02-08 op start_cgi(path, end+1, c);
347 d3a08f4d 2021-01-17 op case FILE_MISSING:
358 abc007d2 2021-02-08 op start_reply(c, NOT_FOUND, "not found");
363 9b8f5ed2 2021-02-03 op mark_nonblock(int fd)
367 9b8f5ed2 2021-02-03 op if ((flags = fcntl(fd, F_GETFL)) == -1)
368 9b8f5ed2 2021-02-03 op fatal("fcntl(F_GETFL): %s", strerror(errno));
369 9b8f5ed2 2021-02-03 op if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1)
370 9b8f5ed2 2021-02-03 op fatal("fcntl(F_SETFL): %s", strerror(errno));
374 abc007d2 2021-02-08 op handle_handshake(int fd, short ev, void *d)
376 abc007d2 2021-02-08 op struct client *c = d;
377 d3a08f4d 2021-01-17 op struct vhost *h;
378 d3a08f4d 2021-01-17 op const char *servname;
379 a8d4a897 2021-01-29 op const char *parse_err = "unknown error";
381 d3a08f4d 2021-01-17 op switch (tls_handshake(c->ctx)) {
382 d3a08f4d 2021-01-17 op case 0: /* success */
383 d3a08f4d 2021-01-17 op case -1: /* already handshaked */
385 d3a08f4d 2021-01-17 op case TLS_WANT_POLLIN:
386 abc007d2 2021-02-08 op reschedule_read(fd, c, &handle_handshake);
388 d3a08f4d 2021-01-17 op case TLS_WANT_POLLOUT:
389 abc007d2 2021-02-08 op reschedule_write(fd, c, &handle_handshake);
392 d3a08f4d 2021-01-17 op /* unreachable */
396 d3a08f4d 2021-01-17 op servname = tls_conn_servername(c->ctx);
397 a8d4a897 2021-01-29 op if (!puny_decode(servname, c->domain, sizeof(c->domain), &parse_err)) {
398 3abf91b0 2021-02-07 op log_info(c, "puny_decode: %s", parse_err);
402 d3a08f4d 2021-01-17 op for (h = hosts; h->domain != NULL; ++h) {
403 eecad7a3 2021-02-12 op if (matches(h->domain, c->domain))
407 3abf91b0 2021-02-07 op log_debug(c, "handshake: SNI: \"%s\"; decoded: \"%s\"; matched: \"%s\"",
408 3abf91b0 2021-02-07 op servname != NULL ? servname : "(null)",
410 3abf91b0 2021-02-07 op h->domain != NULL ? h->domain : "(null)");
412 d3a08f4d 2021-01-17 op if (h->domain != NULL) {
414 abc007d2 2021-02-08 op handle_open_conn(fd, ev, c);
419 0ab65593 2021-01-21 op if (servname != NULL)
420 0ab65593 2021-01-21 op strncpy(c->req, servname, sizeof(c->req));
422 0ab65593 2021-01-21 op strncpy(c->req, "null", sizeof(c->req));
424 abc007d2 2021-02-08 op start_reply(c, BAD_REQUEST, "Wrong/malformed host or missing SNI");
427 57ec3e77 2021-02-08 op static char *
428 57ec3e77 2021-02-08 op strip_path(char *path, int strip)
432 6abda252 2021-02-06 op while (strip > 0) {
433 6abda252 2021-02-06 op if ((t = strchr(path, '/')) == NULL) {
434 6abda252 2021-02-06 op path = strchr(path, '\0');
445 57ec3e77 2021-02-08 op fmt_sbuf(const char *fmt, struct client *c, const char *path)
448 57ec3e77 2021-02-08 op char buf[32];
450 6abda252 2021-02-06 op memset(buf, 0, sizeof(buf));
451 6abda252 2021-02-06 op for (i = 0; *fmt; ++fmt) {
452 6abda252 2021-02-06 op if (i == sizeof(buf)-1 || *fmt == '%') {
453 6abda252 2021-02-06 op strlcat(c->sbuf, buf, sizeof(c->sbuf));
454 6abda252 2021-02-06 op memset(buf, 0, sizeof(buf));
458 6abda252 2021-02-06 op if (*fmt != '%') {
459 6abda252 2021-02-06 op buf[i++] = *fmt;
463 6abda252 2021-02-06 op switch (*++fmt) {
465 6abda252 2021-02-06 op strlcat(c->sbuf, "%", sizeof(c->sbuf));
468 6abda252 2021-02-06 op strlcat(c->sbuf, path, sizeof(c->sbuf));
471 6abda252 2021-02-06 op strlcat(c->sbuf, c->iri.query, sizeof(c->sbuf));
474 6abda252 2021-02-06 op snprintf(buf, sizeof(buf), "%d", conf.port);
475 6abda252 2021-02-06 op strlcat(c->sbuf, buf, sizeof(c->sbuf));
476 6abda252 2021-02-06 op memset(buf, 0, sizeof(buf));
479 6abda252 2021-02-06 op strlcat(c->sbuf, c->domain, sizeof(c->sbuf));
482 6abda252 2021-02-06 op fatal("%s: unknown fmt specifier %c",
483 6abda252 2021-02-06 op __func__, *fmt);
488 6abda252 2021-02-06 op strlcat(c->sbuf, buf, sizeof(c->sbuf));
491 57ec3e77 2021-02-08 op /* 1 if a matching `block return' (and apply it), 0 otherwise */
493 57ec3e77 2021-02-08 op apply_block_return(struct client *c)
495 57ec3e77 2021-02-08 op const char *fmt, *path;
498 57ec3e77 2021-02-08 op if (!vhost_block_return(c->host, c->iri.path, &code, &fmt))
501 57ec3e77 2021-02-08 op path = strip_path(c->iri.path, vhost_strip(c->host, c->iri.path));
502 57ec3e77 2021-02-08 op fmt_sbuf(fmt, c, path);
504 abc007d2 2021-02-08 op start_reply(c, code, c->sbuf);
508 02be96c6 2021-02-09 op /* 1 if matching `require client ca' fails (and apply it), 0 otherwise */
510 02be96c6 2021-02-09 op apply_require_ca(struct client *c)
512 02be96c6 2021-02-09 op X509_STORE *store;
513 02be96c6 2021-02-09 op const uint8_t *cert;
516 02be96c6 2021-02-09 op if ((store = vhost_require_ca(c->host, c->iri.path)) == NULL)
519 02be96c6 2021-02-09 op if (!tls_peer_cert_provided(c->ctx)) {
520 02be96c6 2021-02-09 op start_reply(c, CLIENT_CERT_REQ, "client certificate required");
524 02be96c6 2021-02-09 op cert = tls_peer_cert_chain_pem(c->ctx, &len);
525 02be96c6 2021-02-09 op if (!validate_against_ca(store, cert, len)) {
526 02be96c6 2021-02-09 op start_reply(c, CERT_NOT_AUTH, "certificate not authorised");
534 abc007d2 2021-02-08 op handle_open_conn(int fd, short ev, void *d)
536 abc007d2 2021-02-08 op struct client *c = d;
537 d3a08f4d 2021-01-17 op const char *parse_err = "invalid request";
538 3300cbe0 2021-01-27 op char decoded[DOMAIN_NAME_LEN];
540 0be51733 2021-01-20 op bzero(c->req, sizeof(c->req));
541 0be51733 2021-01-20 op bzero(&c->iri, sizeof(c->iri));
543 0be51733 2021-01-20 op switch (tls_read(c->ctx, c->req, sizeof(c->req)-1)) {
545 3abf91b0 2021-02-07 op log_err(c, "tls_read: %s", tls_error(c->ctx));
546 abc007d2 2021-02-08 op close_conn(fd, ev, c);
549 d3a08f4d 2021-01-17 op case TLS_WANT_POLLIN:
550 abc007d2 2021-02-08 op reschedule_read(fd, c, &handle_open_conn);
553 d3a08f4d 2021-01-17 op case TLS_WANT_POLLOUT:
554 abc007d2 2021-02-08 op reschedule_write(fd, c, &handle_open_conn);
558 c4f682f8 2021-01-27 op if (!trim_req_iri(c->req, &parse_err)
559 a2fd8013 2021-01-29 op || !parse_iri(c->req, &c->iri, &parse_err)
560 a2fd8013 2021-01-29 op || !puny_decode(c->iri.host, decoded, sizeof(decoded), &parse_err)) {
561 3abf91b0 2021-02-07 op log_info(c, "iri parse error: %s", parse_err);
562 abc007d2 2021-02-08 op start_reply(c, BAD_REQUEST, "invalid request");
566 3300cbe0 2021-01-27 op if (c->iri.port_no != conf.port
567 3300cbe0 2021-01-27 op || strcmp(c->iri.schema, "gemini")
568 3300cbe0 2021-01-27 op || strcmp(decoded, c->domain)) {
569 abc007d2 2021-02-08 op start_reply(c, PROXY_REFUSED, "won't proxy request");
573 02be96c6 2021-02-09 op if (apply_require_ca(c))
576 abc007d2 2021-02-08 op if (apply_block_return(c))
579 e3ddf390 2021-02-06 op if (c->host->entrypoint != NULL) {
580 abc007d2 2021-02-08 op start_cgi(c->host->entrypoint, c->iri.path, c);
584 abc007d2 2021-02-08 op open_file(c);
588 abc007d2 2021-02-08 op start_reply(struct client *c, int code, const char *meta)
590 112802ea 2021-02-01 op c->code = code;
591 112802ea 2021-02-01 op c->meta = meta;
592 abc007d2 2021-02-08 op handle_start_reply(c->fd, 0, c);
596 abc007d2 2021-02-08 op handle_start_reply(int fd, short ev, void *d)
598 abc007d2 2021-02-08 op struct client *c = d;
599 05c23a54 2021-01-19 op char buf[1030]; /* status + ' ' + max reply len + \r\n\0 */
600 c8b74339 2021-01-24 op const char *lang;
603 c8b74339 2021-01-24 op lang = vhost_lang(c->host, c->iri.path);
605 112802ea 2021-02-01 op snprintf(buf, sizeof(buf), "%d ", c->code);
606 112802ea 2021-02-01 op strlcat(buf, c->meta, sizeof(buf));
607 112802ea 2021-02-01 op if (!strcmp(c->meta, "text/gemini") && lang != NULL) {
608 05c23a54 2021-01-19 op strlcat(buf, "; lang=", sizeof(buf));
609 c8b74339 2021-01-24 op strlcat(buf, lang, sizeof(buf));
612 05c23a54 2021-01-19 op len = strlcat(buf, "\r\n", sizeof(buf));
613 0be51733 2021-01-20 op assert(len < sizeof(buf));
615 df79b4c1 2021-01-19 op switch (tls_write(c->ctx, buf, len)) {
617 abc007d2 2021-02-08 op close_conn(fd, ev, c);
619 d3a08f4d 2021-01-17 op case TLS_WANT_POLLIN:
620 abc007d2 2021-02-08 op reschedule_read(fd, c, &handle_start_reply);
622 d3a08f4d 2021-01-17 op case TLS_WANT_POLLOUT:
623 abc007d2 2021-02-08 op reschedule_write(fd, c, &handle_start_reply);
627 4604dc96 2021-02-23 op if (!vhost_disable_log(c->host, c->iri.path))
628 4604dc96 2021-02-23 op log_request(c, buf, sizeof(buf));
630 abc007d2 2021-02-08 op if (c->code != SUCCESS)
631 abc007d2 2021-02-08 op close_conn(fd, ev, c);
633 abc007d2 2021-02-08 op c->next(fd, ev, c);
637 abc007d2 2021-02-08 op start_cgi(const char *spath, const char *relpath, struct client *c)
639 d3a08f4d 2021-01-17 op char addr[NI_MAXHOST];
640 bc99d868 2021-03-19 op const char *t;
641 bc99d868 2021-03-19 op struct cgireq req;
644 d3a08f4d 2021-01-17 op e = getnameinfo((struct sockaddr*)&c->addr, sizeof(c->addr),
645 d3a08f4d 2021-01-17 op addr, sizeof(addr),
647 d3a08f4d 2021-01-17 op NI_NUMERICHOST);
649 bc99d868 2021-03-19 op fatal("getnameinfo failed");
651 bc99d868 2021-03-19 op memset(&req, 0, sizeof(req));
653 bc99d868 2021-03-19 op memcpy(req.buf, c->req, sizeof(req.buf));
655 bc99d868 2021-03-19 op req.iri_schema_off = c->iri.schema - c->req;
656 bc99d868 2021-03-19 op req.iri_host_off = c->iri.host - c->req;
657 bc99d868 2021-03-19 op req.iri_port_off = c->iri.port - c->req;
658 bc99d868 2021-03-19 op req.iri_path_off = c->iri.path - c->req;
659 bc99d868 2021-03-19 op req.iri_query_off = c->iri.query - c->req;
660 bc99d868 2021-03-19 op req.iri_fragment_off = c->iri.fragment - c->req;
662 bc99d868 2021-03-19 op req.iri_portno = c->iri.port_no;
664 bc99d868 2021-03-19 op strlcpy(req.spath, spath, sizeof(req.spath));
665 bc99d868 2021-03-19 op strlcpy(req.relpath, relpath, sizeof(req.relpath));
666 bc99d868 2021-03-19 op strlcpy(req.addr, addr, sizeof(req.addr));
668 bc99d868 2021-03-19 op if ((t = tls_peer_cert_subject(c->ctx)) != NULL)
669 bc99d868 2021-03-19 op strlcpy(req.subject, t, sizeof(req.subject));
670 bc99d868 2021-03-19 op if ((t = tls_peer_cert_issuer(c->ctx)) != NULL)
671 bc99d868 2021-03-19 op strlcpy(req.issuer, t, sizeof(req.issuer));
672 bc99d868 2021-03-19 op if ((t = tls_peer_cert_hash(c->ctx)) != NULL)
673 bc99d868 2021-03-19 op strlcpy(req.hash, t, sizeof(req.hash));
675 bc99d868 2021-03-19 op req.notbefore = tls_peer_cert_notbefore(c->ctx);
676 bc99d868 2021-03-19 op req.notafter = tls_peer_cert_notafter(c->ctx);
678 bc99d868 2021-03-19 op req.host_off = c->host - hosts;
680 bc99d868 2021-03-19 op imsg_compose(&exibuf, IMSG_CGI_REQ, c->id, 0, -1, &req, sizeof(req));
681 bc99d868 2021-03-19 op imsg_flush(&exibuf);
683 bc99d868 2021-03-19 op close(c->pfd);
687 abc007d2 2021-02-08 op open_dir(struct client *c)
691 252908e6 2021-01-24 op char *before_file;
693 0be51733 2021-01-20 op len = strlen(c->iri.path);
694 252908e6 2021-01-24 op if (len > 0 && !ends_with(c->iri.path, "/")) {
695 abc007d2 2021-02-08 op redirect_canonical_dir(c);
699 252908e6 2021-01-24 op strlcpy(c->sbuf, "/", sizeof(c->sbuf));
700 d1ca3911 2021-01-21 op strlcat(c->sbuf, c->iri.path, sizeof(c->sbuf));
701 d1ca3911 2021-01-21 op if (!ends_with(c->sbuf, "/"))
702 0be51733 2021-01-20 op strlcat(c->sbuf, "/", sizeof(c->sbuf));
703 252908e6 2021-01-24 op before_file = strchr(c->sbuf, '\0');
704 c8b74339 2021-01-24 op len = strlcat(c->sbuf, vhost_index(c->host, c->iri.path),
705 c8b74339 2021-01-24 op sizeof(c->sbuf));
706 252908e6 2021-01-24 op if (len >= sizeof(c->sbuf)) {
707 abc007d2 2021-02-08 op start_reply(c, TEMP_FAILURE, "internal server error");
711 252908e6 2021-01-24 op c->iri.path = c->sbuf;
713 252908e6 2021-01-24 op /* close later unless we have to generate the dir listing */
714 abc007d2 2021-02-08 op dirfd = c->pfd;
717 abc007d2 2021-02-08 op switch (check_path(c, c->iri.path, &c->pfd)) {
718 252908e6 2021-01-24 op case FILE_EXECUTABLE:
719 eecad7a3 2021-02-12 op if (c->host->cgi != NULL && matches(c->host->cgi, c->iri.path)) {
720 abc007d2 2021-02-08 op start_cgi(c->iri.path, "", c);
724 252908e6 2021-01-24 op /* fallthrough */
726 252908e6 2021-01-24 op case FILE_EXISTS:
727 98ee8406 2021-02-12 op c->next = handle_copy;
728 98ee8406 2021-02-12 op start_reply(c, SUCCESS, mime(c->host, c->iri.path));
731 252908e6 2021-01-24 op case FILE_DIRECTORY:
732 abc007d2 2021-02-08 op start_reply(c, TEMP_REDIRECT, c->sbuf);
735 252908e6 2021-01-24 op case FILE_MISSING:
736 252908e6 2021-01-24 op *before_file = '\0';
738 252908e6 2021-01-24 op if (!vhost_auto_index(c->host, c->iri.path)) {
739 abc007d2 2021-02-08 op start_reply(c, NOT_FOUND, "not found");
743 abc007d2 2021-02-08 op c->pfd = dirfd;
744 5f715ce4 2021-02-02 op c->next = enter_handle_dirlist;
746 abc007d2 2021-02-08 op if ((c->dir = fdopendir(c->pfd)) == NULL) {
747 3abf91b0 2021-02-07 op log_err(c, "fdopendir(%d) (vhost:%s) %s: %s",
748 abc007d2 2021-02-08 op c->pfd, c->host->domain, c->iri.path, strerror(errno));
749 abc007d2 2021-02-08 op start_reply(c, TEMP_FAILURE, "internal server error");
754 abc007d2 2021-02-08 op start_reply(c, SUCCESS, "text/gemini");
758 252908e6 2021-01-24 op /* unreachable */
762 252908e6 2021-01-24 op close(dirfd);
766 abc007d2 2021-02-08 op redirect_canonical_dir(struct client *c)
770 252908e6 2021-01-24 op strlcpy(c->sbuf, "/", sizeof(c->sbuf));
771 252908e6 2021-01-24 op strlcat(c->sbuf, c->iri.path, sizeof(c->sbuf));
772 252908e6 2021-01-24 op len = strlcat(c->sbuf, "/", sizeof(c->sbuf));
774 0be51733 2021-01-20 op if (len >= sizeof(c->sbuf)) {
775 abc007d2 2021-02-08 op start_reply(c, TEMP_FAILURE, "internal server error");
779 abc007d2 2021-02-08 op start_reply(c, TEMP_REDIRECT, c->sbuf);
783 abc007d2 2021-02-08 op enter_handle_dirlist(int fd, short ev, void *d)
785 abc007d2 2021-02-08 op struct client *c = d;
786 5f715ce4 2021-02-02 op char b[PATH_MAX];
789 5f715ce4 2021-02-02 op strlcpy(b, c->iri.path, sizeof(b));
790 5f715ce4 2021-02-02 op l = snprintf(c->sbuf, sizeof(c->sbuf),
791 5f715ce4 2021-02-02 op "# Index of %s\n\n", b);
792 5f715ce4 2021-02-02 op if (l >= sizeof(c->sbuf)) {
793 5f715ce4 2021-02-02 op /* this is impossible, given that we have enough space
794 5f715ce4 2021-02-02 op * in c->sbuf to hold the ancilliary string plus the
795 5f715ce4 2021-02-02 op * full path; but it wouldn't read nice without some
796 5f715ce4 2021-02-02 op * error checking, and I'd like to avoid a strlen. */
797 abc007d2 2021-02-08 op close_conn(fd, ev, c);
802 abc007d2 2021-02-08 op handle_dirlist(fd, ev, c);
806 abc007d2 2021-02-08 op handle_dirlist(int fd, short ev, void *d)
808 abc007d2 2021-02-08 op struct client *c = d;
811 5f715ce4 2021-02-02 op while (c->len > 0) {
812 5f715ce4 2021-02-02 op switch (r = tls_write(c->ctx, c->sbuf + c->off, c->len)) {
814 abc007d2 2021-02-08 op close_conn(fd, ev, c);
816 5f715ce4 2021-02-02 op case TLS_WANT_POLLOUT:
817 abc007d2 2021-02-08 op reschedule_read(fd, c, &handle_dirlist);
819 5f715ce4 2021-02-02 op case TLS_WANT_POLLIN:
820 abc007d2 2021-02-08 op reschedule_write(fd, c, &handle_dirlist);
828 abc007d2 2021-02-08 op send_directory_listing(fd, ev, c);
832 252908e6 2021-01-24 op read_next_dir_entry(struct client *c)
834 252908e6 2021-01-24 op struct dirent *d;
838 252908e6 2021-01-24 op if ((d = readdir(c->dir)) == NULL) {
839 252908e6 2021-01-24 op if (errno != 0)
840 3abf91b0 2021-02-07 op log_err(c, "readdir: %s", strerror(errno));
843 252908e6 2021-01-24 op } while (!strcmp(d->d_name, "."));
845 252908e6 2021-01-24 op /* XXX: url escape */
846 252908e6 2021-01-24 op snprintf(c->sbuf, sizeof(c->sbuf), "=> %s %s\n",
847 252908e6 2021-01-24 op d->d_name, d->d_name);
848 252908e6 2021-01-24 op c->len = strlen(c->sbuf);
855 abc007d2 2021-02-08 op send_directory_listing(int fd, short ev, void *d)
857 abc007d2 2021-02-08 op struct client *c = d;
861 252908e6 2021-01-24 op if (c->len == 0) {
862 252908e6 2021-01-24 op if (!read_next_dir_entry(c))
866 252908e6 2021-01-24 op while (c->len > 0) {
867 252908e6 2021-01-24 op switch (r = tls_write(c->ctx, c->sbuf + c->off, c->len)) {
871 252908e6 2021-01-24 op case TLS_WANT_POLLOUT:
872 abc007d2 2021-02-08 op reschedule_read(fd, c, &send_directory_listing);
875 252908e6 2021-01-24 op case TLS_WANT_POLLIN:
876 abc007d2 2021-02-08 op reschedule_write(fd, c, &send_directory_listing);
888 abc007d2 2021-02-08 op close_conn(fd, ev, d);
891 35744950 2021-02-01 op /* accumulate the meta line from the cgi script. */
893 abc007d2 2021-02-08 op handle_cgi_reply(int fd, short ev, void *d)
895 abc007d2 2021-02-08 op struct client *c = d;
896 35744950 2021-02-01 op void *buf, *e;
901 35744950 2021-02-01 op buf = c->sbuf + c->len;
902 35744950 2021-02-01 op len = sizeof(c->sbuf) - c->len;
904 abc007d2 2021-02-08 op r = read(c->pfd, buf, len);
905 35744950 2021-02-01 op if (r == 0 || r == -1) {
906 abc007d2 2021-02-08 op start_reply(c, CGI_ERROR, "CGI error");
912 35744950 2021-02-01 op /* TODO: error if the CGI script don't reply correctly */
913 35744950 2021-02-01 op e = strchr(c->sbuf, '\n');
914 35744950 2021-02-01 op if (e != NULL || c->len == sizeof(c->sbuf)) {
915 3309ef97 2021-01-23 op log_request(c, c->sbuf, c->len);
918 27b2fa9a 2021-02-12 op handle_copy(fd, ev, c);
922 abc007d2 2021-02-08 op reschedule_read(fd, c, &handle_cgi_reply);
926 27b2fa9a 2021-02-12 op handle_copy(int fd, short ev, void *d)
928 abc007d2 2021-02-08 op struct client *c = d;
932 d3a08f4d 2021-01-17 op while (c->len > 0) {
933 d3a08f4d 2021-01-17 op switch (r = tls_write(c->ctx, c->sbuf + c->off, c->len)) {
937 d3a08f4d 2021-01-17 op case TLS_WANT_POLLOUT:
938 27b2fa9a 2021-02-12 op reschedule_write(c->fd, c, &handle_copy);
941 d3a08f4d 2021-01-17 op case TLS_WANT_POLLIN:
942 27b2fa9a 2021-02-12 op reschedule_read(c->fd, c, &handle_copy);
952 27b2fa9a 2021-02-12 op switch (r = read(c->pfd, c->sbuf, sizeof(c->sbuf))) {
956 27b2fa9a 2021-02-12 op if (errno == EAGAIN || errno == EWOULDBLOCK) {
957 27b2fa9a 2021-02-12 op reschedule_read(c->pfd, c, &handle_copy);
968 abc007d2 2021-02-08 op close_conn(c->fd, ev, d);
972 abc007d2 2021-02-08 op close_conn(int fd, short ev, void *d)
974 abc007d2 2021-02-08 op struct client *c = d;
976 d3a08f4d 2021-01-17 op switch (tls_close(c->ctx)) {
977 d3a08f4d 2021-01-17 op case TLS_WANT_POLLIN:
978 abc007d2 2021-02-08 op reschedule_read(fd, c, &close_conn);
980 d3a08f4d 2021-01-17 op case TLS_WANT_POLLOUT:
981 abc007d2 2021-02-08 op reschedule_read(fd, c, &close_conn);
985 d3a08f4d 2021-01-17 op connected_clients--;
987 d3a08f4d 2021-01-17 op tls_free(c->ctx);
988 d3a08f4d 2021-01-17 op c->ctx = NULL;
990 abc007d2 2021-02-08 op if (c->pfd != -1)
991 abc007d2 2021-02-08 op close(c->pfd);
993 252908e6 2021-01-24 op if (c->dir != NULL)
994 252908e6 2021-01-24 op closedir(c->dir);
996 abc007d2 2021-02-08 op close(c->fd);
1001 abc007d2 2021-02-08 op do_accept(int sock, short et, void *d)
1003 abc007d2 2021-02-08 op struct client *c;
1004 d3a08f4d 2021-01-17 op struct sockaddr_storage addr;
1005 abc007d2 2021-02-08 op struct sockaddr *saddr;
1006 d3a08f4d 2021-01-17 op socklen_t len;
1011 abc007d2 2021-02-08 op saddr = (struct sockaddr*)&addr;
1012 d3a08f4d 2021-01-17 op len = sizeof(addr);
1013 3cb3dd4d 2021-02-12 op if ((fd = accept(sock, saddr, &len)) == -1) {
1014 cfb8a77f 2021-02-07 op if (errno == EWOULDBLOCK || errno == EAGAIN)
1016 d3a08f4d 2021-01-17 op fatal("accept: %s", strerror(errno));
1019 3cb3dd4d 2021-02-12 op mark_nonblock(fd);
1021 d3a08f4d 2021-01-17 op for (i = 0; i < MAX_USERS; ++i) {
1022 bc99d868 2021-03-19 op c = &clients[i];
1023 abc007d2 2021-02-08 op if (c->fd == -1) {
1024 abc007d2 2021-02-08 op memset(c, 0, sizeof(*c));
1026 bc99d868 2021-03-19 op if (tls_accept_socket(ctx, &c->ctx, fd) == -1)
1027 d3a08f4d 2021-01-17 op break; /* goodbye fd! */
1030 abc007d2 2021-02-08 op c->pfd = -1;
1031 abc007d2 2021-02-08 op c->dir = NULL;
1032 abc007d2 2021-02-08 op c->addr = addr;
1034 abc007d2 2021-02-08 op reschedule_read(fd, c, &handle_handshake);
1035 d3a08f4d 2021-01-17 op connected_clients++;
1043 bc99d868 2021-03-19 op struct client *
1044 bc99d868 2021-03-19 op client_by_id(int id)
1046 bc99d868 2021-03-19 op if ((size_t)id > sizeof(clients)/sizeof(clients[0]))
1047 bc99d868 2021-03-19 op fatal("in client_by_id: invalid id %d", id);
1048 bc99d868 2021-03-19 op return &clients[id];
1052 bc99d868 2021-03-19 op handle_imsg_cgi_res(struct imsgbuf *ibuf, struct imsg *imsg, size_t len)
1054 bc99d868 2021-03-19 op struct client *c;
1056 bc99d868 2021-03-19 op c = client_by_id(imsg->hdr.peerid);
1058 bc99d868 2021-03-19 op if ((c->pfd = imsg->fd) == -1)
1059 bc99d868 2021-03-19 op start_reply(c, TEMP_FAILURE, "internal server error");
1061 bc99d868 2021-03-19 op reschedule_read(c->pfd, c, &handle_cgi_reply);
1065 bc99d868 2021-03-19 op handle_imsg_quit(struct imsgbuf *ibuf, struct imsg *imsg, size_t len)
1070 bc99d868 2021-03-19 op /* don't call event_loopbreak since we want to finish to
1071 bc99d868 2021-03-19 op * handle the ongoing connections. */
1073 a6e689d7 2021-02-12 op event_del(&e4);
1074 a6e689d7 2021-02-12 op if (has_ipv6)
1075 a6e689d7 2021-02-12 op event_del(&e6);
1076 a6e689d7 2021-02-12 op if (has_siginfo)
1077 a6e689d7 2021-02-12 op signal_del(&siginfo);
1078 bc99d868 2021-03-19 op event_del(&imsgev);
1079 a6e689d7 2021-02-12 op signal_del(&sigusr2);
1083 bc99d868 2021-03-19 op handle_dispatch_imsg(int fd, short ev, void *d)
1085 bc99d868 2021-03-19 op struct imsgbuf *ibuf = d;
1086 bc99d868 2021-03-19 op dispatch_imsg(ibuf, handlers, sizeof(handlers));
1090 abc007d2 2021-02-08 op handle_siginfo(int fd, short ev, void *d)
1096 abc007d2 2021-02-08 op log_info(NULL, "%d connected clients", connected_clients);
1100 bc99d868 2021-03-19 op loop(struct tls *ctx_, int sock4, int sock6, struct imsgbuf *ibuf)
1106 abc007d2 2021-02-08 op event_init();
1108 bc99d868 2021-03-19 op memset(&clients, 0, sizeof(clients));
1109 abc007d2 2021-02-08 op for (i = 0; i < MAX_USERS; ++i)
1110 bc99d868 2021-03-19 op clients[i].fd = -1;
1112 bc99d868 2021-03-19 op event_set(&e4, sock4, EV_READ | EV_PERSIST, &do_accept, NULL);
1113 a6e689d7 2021-02-12 op event_add(&e4, NULL);
1115 abc007d2 2021-02-08 op if (sock6 != -1) {
1116 a6e689d7 2021-02-12 op has_ipv6 = 1;
1117 bc99d868 2021-03-19 op event_set(&e6, sock6, EV_READ | EV_PERSIST, &do_accept, NULL);
1118 a6e689d7 2021-02-12 op event_add(&e6, NULL);
1121 bc99d868 2021-03-19 op event_set(&imsgev, ibuf->fd, EV_READ | EV_PERSIST, handle_dispatch_imsg, ibuf);
1122 bc99d868 2021-03-19 op event_add(&imsgev, NULL);
1124 abc007d2 2021-02-08 op #ifdef SIGINFO
1125 a6e689d7 2021-02-12 op has_siginfo = 1;
1126 a6e689d7 2021-02-12 op signal_set(&siginfo, SIGINFO, &handle_siginfo, NULL);
1127 a6e689d7 2021-02-12 op signal_add(&siginfo, NULL);
1129 5e3285d5 2021-02-12 op signal_set(&sigusr2, SIGUSR2, &handle_siginfo, NULL);
1130 a6e689d7 2021-02-12 op signal_add(&sigusr2, NULL);
1133 abc007d2 2021-02-08 op event_dispatch();
