2 b0a6bcf7 2022-09-13 op * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
3 b0a6bcf7 2022-09-13 op * Copyright (c) 2006 - 2015 Reyk Floeter <reyk@openbsd.org>
5 b0a6bcf7 2022-09-13 op * Permission to use, copy, modify, and distribute this software for any
6 b0a6bcf7 2022-09-13 op * purpose with or without fee is hereby granted, provided that the above
7 b0a6bcf7 2022-09-13 op * copyright notice and this permission notice appear in all copies.
9 b0a6bcf7 2022-09-13 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 b0a6bcf7 2022-09-13 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 b0a6bcf7 2022-09-13 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 b0a6bcf7 2022-09-13 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 b0a6bcf7 2022-09-13 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 b0a6bcf7 2022-09-13 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 b0a6bcf7 2022-09-13 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 b0a6bcf7 2022-09-13 op #include <sys/types.h>
19 b0a6bcf7 2022-09-13 op #include <sys/queue.h>
20 b0a6bcf7 2022-09-13 op #include <sys/tree.h>
21 b0a6bcf7 2022-09-13 op #include <sys/socket.h>
22 b0a6bcf7 2022-09-13 op #include <sys/uio.h>
24 b0a6bcf7 2022-09-13 op #include <netinet/in.h>
25 b0a6bcf7 2022-09-13 op #include <arpa/inet.h>
26 b0a6bcf7 2022-09-13 op #include <netdb.h>
28 b0a6bcf7 2022-09-13 op #include <asr.h>
29 b0a6bcf7 2022-09-13 op #include <ctype.h>
30 b0a6bcf7 2022-09-13 op #include <errno.h>
31 b0a6bcf7 2022-09-13 op #include <event.h>
32 33f4bf26 2022-09-14 op #include <fnmatch.h>
33 b0a6bcf7 2022-09-13 op #include <limits.h>
34 b0a6bcf7 2022-09-13 op #include <stdlib.h>
35 b0a6bcf7 2022-09-13 op #include <stdint.h>
36 b0a6bcf7 2022-09-13 op #include <stdio.h>
37 b0a6bcf7 2022-09-13 op #include <string.h>
38 b0a6bcf7 2022-09-13 op #include <imsg.h>
39 b0a6bcf7 2022-09-13 op #include <tls.h>
40 b0a6bcf7 2022-09-13 op #include <unistd.h>
42 b0a6bcf7 2022-09-13 op #include "log.h"
43 b0a6bcf7 2022-09-13 op #include "proc.h"
45 b0a6bcf7 2022-09-13 op #include "galileo.h"
47 b0a6bcf7 2022-09-13 op #define MINIMUM(a, b) ((a) < (b) ? (a) : (b))
49 b0a6bcf7 2022-09-13 op /* provided by OpenBSD' base libevent but not in any header? */
50 b0a6bcf7 2022-09-13 op extern void bufferevent_read_pressure_cb(struct evbuffer *, size_t,
51 b0a6bcf7 2022-09-13 op size_t, void *);
53 b0a6bcf7 2022-09-13 op void proxy_init(struct privsep *, struct privsep_proc *, void *);
54 b0a6bcf7 2022-09-13 op int proxy_launch(struct galileo *);
55 b0a6bcf7 2022-09-13 op void proxy_inflight_dec(const char *);
56 b0a6bcf7 2022-09-13 op int proxy_dispatch_parent(int, struct privsep_proc *, struct imsg *);
57 c9dc4bc6 2022-09-13 op void proxy_translate_gemtext(struct client *);
58 b0a6bcf7 2022-09-13 op void proxy_resolved(struct asr_result *, void *);
59 b0a6bcf7 2022-09-13 op void proxy_connect(int, short, void *);
60 b0a6bcf7 2022-09-13 op void proxy_read(struct bufferevent *, void *);
61 b0a6bcf7 2022-09-13 op void proxy_write(struct bufferevent *, void *);
62 b0a6bcf7 2022-09-13 op void proxy_error(struct bufferevent *, short, void *);
63 b0a6bcf7 2022-09-13 op int proxy_bufferevent_add(struct event *, int);
64 b0a6bcf7 2022-09-13 op void proxy_tls_writecb(int, short, void *);
65 b0a6bcf7 2022-09-13 op void proxy_tls_readcb(int, short, void *);
67 b0a6bcf7 2022-09-13 op static struct privsep_proc procs[] = {
68 b0a6bcf7 2022-09-13 op { "parent", PROC_PARENT, proxy_dispatch_parent },
71 b0a6bcf7 2022-09-13 op volatile int proxy_clients;
72 b0a6bcf7 2022-09-13 op volatile int proxy_inflight;
73 b0a6bcf7 2022-09-13 op uint32_t proxy_fcg_id;
76 b0a6bcf7 2022-09-13 op proxy(struct privsep *ps, struct privsep_proc *p)
78 b0a6bcf7 2022-09-13 op proc_run(ps, p, procs, nitems(procs), proxy_init, NULL);
82 b0a6bcf7 2022-09-13 op proxy_init(struct privsep *ps, struct privsep_proc *p, void *arg)
84 b0a6bcf7 2022-09-13 op if (config_init(ps->ps_env) == -1)
85 b0a6bcf7 2022-09-13 op fatal("failed to initialize configuration");
87 b0a6bcf7 2022-09-13 op /* We use a custom shutdown callback */
88 b0a6bcf7 2022-09-13 op /* p->p_shutdown = proxy_shutdown */
90 b0a6bcf7 2022-09-13 op if (pledge("stdio recvfd unix inet dns", NULL) == -1)
91 b0a6bcf7 2022-09-13 op fatal("pledge");
95 b0a6bcf7 2022-09-13 op proxy_launch(struct galileo *env)
97 b0a6bcf7 2022-09-13 op event_add(&env->sc_evsock, NULL);
102 b0a6bcf7 2022-09-13 op proxy_purge(struct server *srv)
107 b0a6bcf7 2022-09-13 op proxy_inflight_dec(const char *why)
109 b0a6bcf7 2022-09-13 op proxy_inflight--;
110 b0a6bcf7 2022-09-13 op log_debug("%s: inflight decremented, now %d, %s",
111 b0a6bcf7 2022-09-13 op __func__, proxy_inflight, why);
115 b0a6bcf7 2022-09-13 op proxy_dispatch_parent(int fd, struct privsep_proc *p, struct imsg *imsg)
117 b0a6bcf7 2022-09-13 op struct privsep *ps = p->p_ps;
118 b0a6bcf7 2022-09-13 op struct galileo *env = ps->ps_env;
120 b0a6bcf7 2022-09-13 op switch (imsg->hdr.type) {
121 b0a6bcf7 2022-09-13 op case IMSG_CFG_SRV:
122 b0a6bcf7 2022-09-13 op if (config_getserver(env, imsg) == -1)
123 b0a6bcf7 2022-09-13 op fatal("config_getproxy");
125 b0a6bcf7 2022-09-13 op case IMSG_CFG_SOCK:
126 b0a6bcf7 2022-09-13 op /* XXX: improve */
128 b0a6bcf7 2022-09-13 op if (env->sc_sock_fd != -1) {
129 b0a6bcf7 2022-09-13 op event_del(&env->sc_evsock);
130 b0a6bcf7 2022-09-13 op close(env->sc_sock_fd);
133 b0a6bcf7 2022-09-13 op env->sc_sock_fd = config_getsock(env, imsg);
134 b0a6bcf7 2022-09-13 op if (env->sc_sock_fd == -1)
135 b0a6bcf7 2022-09-13 op fatal("config_getsock");
137 b0a6bcf7 2022-09-13 op event_set(&env->sc_evsock, env->sc_sock_fd,
138 ea8874f1 2022-09-13 op EV_READ | EV_PERSIST, fcgi_accept, env);
139 b0a6bcf7 2022-09-13 op event_add(&env->sc_evsock, NULL);
140 ea8874f1 2022-09-13 op evtimer_set(&env->sc_evpause, fcgi_accept, env);
142 b0a6bcf7 2022-09-13 op case IMSG_CFG_DONE:
143 b0a6bcf7 2022-09-13 op log_debug("config done!");
145 b0a6bcf7 2022-09-13 op case IMSG_CTL_START:
146 b0a6bcf7 2022-09-13 op proxy_launch(env);
149 b0a6bcf7 2022-09-13 op log_warnx("unknown message %d", imsg->hdr.type);
156 c9dc4bc6 2022-09-13 op static inline int
157 c9dc4bc6 2022-09-13 op printurl(struct client *clt, const char *str)
159 c9dc4bc6 2022-09-13 op for (; *str; ++str) {
160 c9dc4bc6 2022-09-13 op switch (*str) {
165 c9dc4bc6 2022-09-13 op if (clt_printf(clt, "%2X", (unsigned char)*str) == -1)
169 c9dc4bc6 2022-09-13 op if (clt_putc(clt, *str) == -1)
178 c9dc4bc6 2022-09-13 op static inline int
179 c9dc4bc6 2022-09-13 op htmlescape(struct client *clt, const char *str)
183 c9dc4bc6 2022-09-13 op for (; *str; ++str) {
184 c9dc4bc6 2022-09-13 op switch (*str) {
186 c9dc4bc6 2022-09-13 op r = clt_puts(clt, "<");
189 c9dc4bc6 2022-09-13 op r = clt_puts(clt, ">");
192 c9dc4bc6 2022-09-13 op r = clt_puts(clt, "&");
195 c9dc4bc6 2022-09-13 op r = clt_putc(clt, *str);
207 c9dc4bc6 2022-09-13 op gemtext_translate_line(struct client *clt, char *line)
209 c9dc4bc6 2022-09-13 op /* preformatted line / closing */
210 b9de0a74 2022-09-14 op if (clt->clt_translate & TR_PRE) {
211 c9dc4bc6 2022-09-13 op if (!strncmp(line, "```", 3)) {
212 b9de0a74 2022-09-14 op clt->clt_translate &= ~TR_PRE;
213 c9dc4bc6 2022-09-13 op return (clt_puts(clt, "</pre>"));
216 c9dc4bc6 2022-09-13 op if (htmlescape(clt, line) == -1)
218 c9dc4bc6 2022-09-13 op return (clt_putc(clt, '\n'));
222 b9de0a74 2022-09-14 op if (!strncmp(line, "* ", 2)) {
223 b9de0a74 2022-09-14 op if (clt->clt_translate & TR_NAV) {
224 b9de0a74 2022-09-14 op if (clt_puts(clt, "</ul></nav>") == -1)
226 b9de0a74 2022-09-14 op clt->clt_translate &= ~TR_NAV;
229 b9de0a74 2022-09-14 op if (!(clt->clt_translate & TR_LIST)) {
230 b9de0a74 2022-09-14 op if (clt_puts(clt, "<ul>") == -1)
232 b9de0a74 2022-09-14 op clt->clt_translate |= TR_LIST;
235 b9de0a74 2022-09-14 op if (clt_puts(clt, "<li>") == -1 ||
236 b9de0a74 2022-09-14 op htmlescape(clt, line + 2) == -1 ||
237 b9de0a74 2022-09-14 op clt_puts(clt, "</li>") == -1)
242 b9de0a74 2022-09-14 op if (clt->clt_translate & TR_LIST) {
243 b9de0a74 2022-09-14 op if (clt_puts(clt, "</ul>") == -1)
245 b9de0a74 2022-09-14 op clt->clt_translate &= ~TR_LIST;
248 b9de0a74 2022-09-14 op /* link -- TODO: relativify from SCRIPT_NAME */
249 b9de0a74 2022-09-14 op if (!strncmp(line, "=>", 2)) {
253 b9de0a74 2022-09-14 op line += strspn(line, " \t");
255 b9de0a74 2022-09-14 op label = line + strcspn(line, " \t");
256 b9de0a74 2022-09-14 op if (*label == '\0')
257 b9de0a74 2022-09-14 op label = line;
259 b9de0a74 2022-09-14 op *label++ = '\0';
261 33f4bf26 2022-09-14 op if (fnmatch("*.jpg", line, 0) == 0 ||
262 33f4bf26 2022-09-14 op fnmatch("*.jpeg", line, 0) == 0 ||
263 33f4bf26 2022-09-14 op fnmatch("*.gif", line, 0) == 0 ||
264 33f4bf26 2022-09-14 op fnmatch("*.png", line, 0) == 0 ||
265 33f4bf26 2022-09-14 op fnmatch("*.svg", line, 0) == 0 ||
266 33f4bf26 2022-09-14 op fnmatch("*.webp", line, 0) == 0) {
267 33f4bf26 2022-09-14 op if (clt->clt_translate & TR_NAV) {
268 33f4bf26 2022-09-14 op if (clt_puts(clt, "</ul></nav>") == -1)
270 33f4bf26 2022-09-14 op clt->clt_translate &= ~TR_NAV;
273 33f4bf26 2022-09-14 op if (clt_puts(clt, "<figure><a href='") == -1 ||
274 33f4bf26 2022-09-14 op printurl(clt, line) == -1 ||
275 33f4bf26 2022-09-14 op clt_puts(clt, "'><img src='") == -1 ||
276 33f4bf26 2022-09-14 op printurl(clt, line) == -1 ||
277 33f4bf26 2022-09-14 op clt_puts(clt, "' /></a><figcaption>") == -1 ||
278 33f4bf26 2022-09-14 op htmlescape(clt, label) == -1 ||
279 33f4bf26 2022-09-14 op clt_puts(clt, "</figcaption></figure>") == -1)
285 33f4bf26 2022-09-14 op if (!(clt->clt_translate & TR_NAV)) {
286 33f4bf26 2022-09-14 op if (clt_puts(clt, "<nav><ul>") == -1)
288 33f4bf26 2022-09-14 op clt->clt_translate |= TR_NAV;
291 b9de0a74 2022-09-14 op if (clt_puts(clt, "<li><a href='") == -1 ||
292 b9de0a74 2022-09-14 op printurl(clt, line) == -1 ||
293 b9de0a74 2022-09-14 op clt_puts(clt, "'>") == -1 ||
294 b9de0a74 2022-09-14 op htmlescape(clt, label) == -1 ||
295 b9de0a74 2022-09-14 op clt_puts(clt, "</a></li>") == -1)
301 b9de0a74 2022-09-14 op if (clt->clt_translate & TR_NAV) {
302 b9de0a74 2022-09-14 op if (clt_puts(clt, "</ul></nav>") == -1)
304 b9de0a74 2022-09-14 op clt->clt_translate &= ~TR_NAV;
307 c9dc4bc6 2022-09-13 op /* pre opening */
308 c9dc4bc6 2022-09-13 op if (!strncmp(line, "```", 3)) {
309 b9de0a74 2022-09-14 op clt->clt_translate |= TR_PRE;
310 c9dc4bc6 2022-09-13 op return (clt_puts(clt, "<pre>"));
313 c9dc4bc6 2022-09-13 op /* citation block */
314 c9dc4bc6 2022-09-13 op if (*line == '>') {
315 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<blockquote>") == -1 ||
316 c9dc4bc6 2022-09-13 op htmlescape(clt, line + 1) == -1 ||
317 c9dc4bc6 2022-09-13 op clt_puts(clt, "</blockquote>") == -1)
322 c9dc4bc6 2022-09-13 op /* headings */
323 c9dc4bc6 2022-09-13 op if (!strncmp(line, "###", 3)) {
324 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<h3>") == -1 ||
325 c9dc4bc6 2022-09-13 op htmlescape(clt, line + 3) == -1 ||
326 c9dc4bc6 2022-09-13 op clt_puts(clt, "</h3>") == -1)
330 c9dc4bc6 2022-09-13 op if (!strncmp(line, "##", 2)) {
331 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<h2>") == -1 ||
332 c9dc4bc6 2022-09-13 op htmlescape(clt, line + 2) == -1 ||
333 c9dc4bc6 2022-09-13 op clt_puts(clt, "</h2>") == -1)
337 c9dc4bc6 2022-09-13 op if (!strncmp(line, "#", 1)) {
338 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<h1>") == -1 ||
339 c9dc4bc6 2022-09-13 op htmlescape(clt, line + 1) == -1 ||
340 c9dc4bc6 2022-09-13 op clt_puts(clt, "</h1>") == -1)
345 df5279ca 2022-09-14 op /* Not following strictly the gemini specification... */
346 df5279ca 2022-09-14 op if (*line == '\0')
349 c9dc4bc6 2022-09-13 op /* paragraph */
350 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<p>") == -1 ||
351 c9dc4bc6 2022-09-13 op htmlescape(clt, line) == -1 ||
352 c9dc4bc6 2022-09-13 op clt_puts(clt, "</p>") == -1)
359 c9dc4bc6 2022-09-13 op proxy_translate_gemtext(struct client *clt)
361 c9dc4bc6 2022-09-13 op struct bufferevent *bev = clt->clt_bev;
362 c9dc4bc6 2022-09-13 op struct evbuffer *src = EVBUFFER_INPUT(bev);
368 b9de0a74 2022-09-14 op line = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF);
369 c9dc4bc6 2022-09-13 op if (line == NULL)
372 c9dc4bc6 2022-09-13 op r = gemtext_translate_line(clt, line);
379 b0a6bcf7 2022-09-13 op static struct proxy_config *
380 b0a6bcf7 2022-09-13 op proxy_server_match(struct galileo *env, struct client *clt)
382 b0a6bcf7 2022-09-13 op struct server *srv;
384 b0a6bcf7 2022-09-13 op if (clt->clt_server_name == NULL)
387 b0a6bcf7 2022-09-13 op TAILQ_FOREACH(srv, &env->sc_servers, srv_entry) {
388 b0a6bcf7 2022-09-13 op if (!strcmp(clt->clt_server_name, srv->srv_conf.host))
389 b0a6bcf7 2022-09-13 op return &srv->srv_conf;
396 b0a6bcf7 2022-09-13 op proxy_start_request(struct galileo *env, struct client *clt)
398 b0a6bcf7 2022-09-13 op struct addrinfo hints;
399 b0a6bcf7 2022-09-13 op struct asr_query *query;
400 b0a6bcf7 2022-09-13 op char port[32];
402 b0a6bcf7 2022-09-13 op if ((clt->clt_pc = proxy_server_match(env, clt)) == NULL) {
403 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Status: 501\r\n\r\n") == -1)
405 b0a6bcf7 2022-09-13 op fcgi_end_request(clt, 1);
409 b0a6bcf7 2022-09-13 op (void)snprintf(port, sizeof(port), "%d", clt->clt_pc->proxy_port);
411 b0a6bcf7 2022-09-13 op memset(&hints, 0, sizeof(hints));
412 b0a6bcf7 2022-09-13 op hints.ai_family = AF_UNSPEC;
413 b0a6bcf7 2022-09-13 op hints.ai_socktype = SOCK_STREAM;
415 b0a6bcf7 2022-09-13 op query = getaddrinfo_async(clt->clt_pc->proxy_addr, port, &hints, NULL);
416 b0a6bcf7 2022-09-13 op if (query == NULL) {
417 b0a6bcf7 2022-09-13 op log_warn("getaddrinfo_async");
418 b0a6bcf7 2022-09-13 op fcgi_abort_request(clt);
422 b0a6bcf7 2022-09-13 op clt->clt_evasr = event_asr_run(query, proxy_resolved, clt);
423 b0a6bcf7 2022-09-13 op if (clt->clt_evasr == NULL) {
424 b0a6bcf7 2022-09-13 op log_warn("event_asr_run");
425 b0a6bcf7 2022-09-13 op asr_abort(query);
426 b0a6bcf7 2022-09-13 op fcgi_abort_request(clt);
432 b0a6bcf7 2022-09-13 op proxy_resolved(struct asr_result *res, void *d)
434 b0a6bcf7 2022-09-13 op struct client *clt = d;
435 b0a6bcf7 2022-09-13 op struct proxy_config *pc = clt->clt_pc;
437 b0a6bcf7 2022-09-13 op clt->clt_evasr = NULL;
439 b0a6bcf7 2022-09-13 op if (res->ar_gai_errno != 0) {
440 b0a6bcf7 2022-09-13 op log_warnx("failed to resolve %s:%d: %s",
441 b0a6bcf7 2022-09-13 op pc->proxy_addr, pc->proxy_port,
442 b0a6bcf7 2022-09-13 op gai_strerror(res->ar_gai_errno));
443 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Status: 501\r\n") == -1)
445 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
447 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Proxy error; connection failed") == -1)
449 b0a6bcf7 2022-09-13 op fcgi_end_request(clt, 1);
453 b0a6bcf7 2022-09-13 op clt->clt_addrinfo = res->ar_addrinfo;
454 b0a6bcf7 2022-09-13 op clt->clt_p = clt->clt_addrinfo;
455 b0a6bcf7 2022-09-13 op proxy_connect(-1, 0, clt);
459 b0a6bcf7 2022-09-13 op proxy_connect(int fd, short ev, void *d)
461 b0a6bcf7 2022-09-13 op struct client *clt = d;
462 b0a6bcf7 2022-09-13 op struct evbuffer *out;
463 b0a6bcf7 2022-09-13 op struct addrinfo *p;
464 b0a6bcf7 2022-09-13 op struct tls_config *conf;
465 b0a6bcf7 2022-09-13 op struct timeval conntv = {5, 0};
467 b0a6bcf7 2022-09-13 op socklen_t len = sizeof(err);
470 b0a6bcf7 2022-09-13 op if (clt->clt_p == NULL)
473 b0a6bcf7 2022-09-13 op if (clt->clt_fd != -1) {
474 b0a6bcf7 2022-09-13 op if (getsockopt(clt->clt_fd, SOL_SOCKET, SO_ERROR, &err, &len)
477 b0a6bcf7 2022-09-13 op if (err != 0) {
484 b0a6bcf7 2022-09-13 op p = clt->clt_p;
485 b0a6bcf7 2022-09-13 op clt->clt_fd = socket(p->ai_family, p->ai_socktype | SOCK_NONBLOCK,
486 b0a6bcf7 2022-09-13 op p->ai_protocol);
487 b0a6bcf7 2022-09-13 op if (clt->clt_fd == -1) {
488 b0a6bcf7 2022-09-13 op clt->clt_p = clt->clt_p->ai_next;
492 b0a6bcf7 2022-09-13 op if (connect(clt->clt_fd, p->ai_addr, p->ai_addrlen) == 0)
495 b0a6bcf7 2022-09-13 op clt->clt_evconn_live = 1;
496 b0a6bcf7 2022-09-13 op event_set(&clt->clt_evconn, clt->clt_fd, EV_WRITE, proxy_connect, clt);
497 b0a6bcf7 2022-09-13 op event_add(&clt->clt_evconn, &conntv);
501 b0a6bcf7 2022-09-13 op clt->clt_evconn_live = 0;
502 b0a6bcf7 2022-09-13 op freeaddrinfo(clt->clt_addrinfo);
503 b0a6bcf7 2022-09-13 op clt->clt_addrinfo = clt->clt_p = NULL;
505 b0a6bcf7 2022-09-13 op /* initialize TLS for Gemini */
506 b0a6bcf7 2022-09-13 op if ((conf = tls_config_new()) == NULL) {
507 b0a6bcf7 2022-09-13 op log_warn("tls_config_new failed");
511 b0a6bcf7 2022-09-13 op tls_config_insecure_noverifycert(conf);
513 b0a6bcf7 2022-09-13 op if ((clt->clt_ctx = tls_client()) == NULL) {
514 b0a6bcf7 2022-09-13 op log_warnx("tls_client failed");
515 b0a6bcf7 2022-09-13 op tls_config_free(conf);
519 b0a6bcf7 2022-09-13 op if (tls_configure(clt->clt_ctx, conf) == -1) {
520 b0a6bcf7 2022-09-13 op log_warnx("tls_configure failed");
521 b0a6bcf7 2022-09-13 op tls_config_free(conf);
525 b0a6bcf7 2022-09-13 op tls_config_free(conf);
527 b0a6bcf7 2022-09-13 op if (tls_connect_socket(clt->clt_ctx, clt->clt_fd,
528 b0a6bcf7 2022-09-13 op clt->clt_pc->proxy_name) == -1) {
529 b0a6bcf7 2022-09-13 op log_warnx("tls_connect_socket failed");
533 b0a6bcf7 2022-09-13 op clt->clt_bev = bufferevent_new(clt->clt_fd, proxy_read, proxy_write,
534 b0a6bcf7 2022-09-13 op proxy_error, clt);
535 b0a6bcf7 2022-09-13 op if (clt->clt_bev == NULL) {
536 b0a6bcf7 2022-09-13 op log_warn("bufferevent_new");
539 b0a6bcf7 2022-09-13 op out = EVBUFFER_OUTPUT(clt->clt_bev);
541 b0a6bcf7 2022-09-13 op event_set(&clt->clt_bev->ev_read, clt->clt_fd, EV_READ,
542 b0a6bcf7 2022-09-13 op proxy_tls_readcb, clt->clt_bev);
543 b0a6bcf7 2022-09-13 op event_set(&clt->clt_bev->ev_write, clt->clt_fd, EV_WRITE,
544 b0a6bcf7 2022-09-13 op proxy_tls_writecb, clt->clt_bev);
546 b0a6bcf7 2022-09-13 op /* bufferevent_settimeout(); */
547 b0a6bcf7 2022-09-13 op bufferevent_enable(clt->clt_bev, EV_READ|EV_WRITE);
549 b0a6bcf7 2022-09-13 op /* TODO: compute the URL */
550 ee80262d 2022-09-13 op if (evbuffer_add_printf(out, "gemini://%s/%s\r\n",
551 ee80262d 2022-09-13 op clt->clt_pc->proxy_name, clt->clt_path_info) == -1) {
552 b0a6bcf7 2022-09-13 op log_warn("bufferevent_printf failed");
559 b0a6bcf7 2022-09-13 op log_warn("failed to connect to %s:%d",
560 b0a6bcf7 2022-09-13 op clt->clt_pc->proxy_addr, clt->clt_pc->proxy_port);
561 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Status: 501\r\n") == -1)
563 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
565 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Proxy error; connection failed") == -1)
567 b0a6bcf7 2022-09-13 op fcgi_end_request(clt, 1);
570 6edd9a00 2022-09-14 op static inline int
571 6edd9a00 2022-09-14 op parse_mime(struct client *clt, char *mime, char *lang, size_t len)
573 6edd9a00 2022-09-14 op char *t, *semi;
575 6edd9a00 2022-09-14 op if (strncmp(mime, "text/gemini", 11) != 0)
578 6edd9a00 2022-09-14 op clt->clt_translate = TR_ENABLED;
580 6edd9a00 2022-09-14 op if ((mime = strchr(mime, ';')) == NULL)
583 6edd9a00 2022-09-14 op *mime++ = '\0';
584 6edd9a00 2022-09-14 op while ((t = strsep(&mime, ";")) != NULL) {
585 6edd9a00 2022-09-14 op if (!strncmp(t, "charset=", 8)) {
587 6edd9a00 2022-09-14 op if (!strncasecmp(t, "utf8", 4) ||
588 6edd9a00 2022-09-14 op !strncasecmp(t, "ascii", 5)) {
589 6edd9a00 2022-09-14 op log_debug("unknown charset %s", t);
595 6edd9a00 2022-09-14 op if (!strncmp(t, "lang=", 5)) {
597 6edd9a00 2022-09-14 op if ((semi = strchr(t, ';')) != NULL)
598 6edd9a00 2022-09-14 op *semi = '\0';
600 6edd9a00 2022-09-14 op if (strlcpy(lang, t, len) >= len) {
601 6edd9a00 2022-09-14 op log_debug("lang too long: %s", t);
602 6edd9a00 2022-09-14 op *lang = '\0';
615 b0a6bcf7 2022-09-13 op proxy_read(struct bufferevent *bev, void *d)
617 b0a6bcf7 2022-09-13 op struct client *clt = d;
618 fc1cb9ea 2022-09-14 op struct proxy_config *pc = clt->clt_pc;
619 b0a6bcf7 2022-09-13 op struct evbuffer *src = EVBUFFER_INPUT(bev);
620 c9dc4bc6 2022-09-13 op const char *ctype;
621 6edd9a00 2022-09-14 op char lang[16];
622 6edd9a00 2022-09-14 op char *hdr, *mime;
625 b0a6bcf7 2022-09-13 op if (clt->clt_headersdone) {
626 c9dc4bc6 2022-09-13 op if (clt->clt_translate)
627 c9dc4bc6 2022-09-13 op proxy_translate_gemtext(clt);
629 c9dc4bc6 2022-09-13 op clt_write_bufferevent(clt, bev);
633 b0a6bcf7 2022-09-13 op hdr = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF_STRICT);
634 b0a6bcf7 2022-09-13 op if (hdr == NULL) {
635 b0a6bcf7 2022-09-13 op if (EVBUFFER_LENGTH(src) >= 1026)
636 b0a6bcf7 2022-09-13 op proxy_error(bev, EV_READ, clt);
640 b0a6bcf7 2022-09-13 op if (len < 4 ||
641 b0a6bcf7 2022-09-13 op !isdigit((unsigned char)hdr[0]) ||
642 b0a6bcf7 2022-09-13 op !isdigit((unsigned char)hdr[1]) ||
643 b0a6bcf7 2022-09-13 op hdr[2] != ' ') {
644 b0a6bcf7 2022-09-13 op log_warnx("invalid ");
645 b0a6bcf7 2022-09-13 op proxy_error(bev, EV_READ, clt);
649 6e8f426e 2022-09-14 op switch (hdr[0]) {
651 6e8f426e 2022-09-14 op /* handled below */
654 6e8f426e 2022-09-14 op if (clt_puts(clt, "Status: 501\r\n") == -1)
656 6e8f426e 2022-09-14 op if (clt_puts(clt,
657 6e8f426e 2022-09-14 op "Content-Type: text/plain;charset=utf8\r\n") == -1)
659 6e8f426e 2022-09-14 op if (clt_puts(clt, "\r\n") == -1)
661 6e8f426e 2022-09-14 op if (clt_printf(clt, "Request failed with code %c%c\n\n",
662 6e8f426e 2022-09-14 op hdr[0], hdr[1]) == -1)
664 6e8f426e 2022-09-14 op if (clt_printf(clt, "The server says: %s\n", &hdr[3]) == -1)
666 6e8f426e 2022-09-14 op fcgi_end_request(clt, 1);
670 6edd9a00 2022-09-14 op mime = hdr + 2 + strspn(hdr + 2, " \t");
671 6edd9a00 2022-09-14 op if (parse_mime(clt, mime, lang, sizeof(lang)) == -1) {
672 6edd9a00 2022-09-14 op if (clt_puts(clt, "Status: 501\r\n") == -1)
674 6edd9a00 2022-09-14 op if (clt_puts(clt,
675 6edd9a00 2022-09-14 op "Content-Type: text/plain;charset=utf8\r\n") == -1)
677 6edd9a00 2022-09-14 op if (clt_puts(clt, "\r\n") == -1)
679 6edd9a00 2022-09-14 op if (clt_printf(clt, "Failed to parse the Gemini response\n")
681 6edd9a00 2022-09-14 op fcgi_end_request(clt, 1);
685 6edd9a00 2022-09-14 op if (clt->clt_translate)
686 6edd9a00 2022-09-14 op ctype = "text/html;charset=utf8";
688 6edd9a00 2022-09-14 op ctype = mime;
690 6edd9a00 2022-09-14 op if (clt_printf(clt, "Content-Type: %s\r\n\r\n", ctype) == -1)
693 b0a6bcf7 2022-09-13 op clt->clt_headersdone = 1;
695 c9dc4bc6 2022-09-13 op if (clt->clt_translate) {
696 6edd9a00 2022-09-14 op if (clt_puts(clt, "<!doctype html><html") == -1)
698 6edd9a00 2022-09-14 op if (*lang != '\0') {
699 6edd9a00 2022-09-14 op if (clt_puts(clt, " lang='") == -1 ||
700 6edd9a00 2022-09-14 op printurl(clt, lang) == -1 ||
701 6edd9a00 2022-09-14 op clt_puts(clt, "'") == -1)
704 6edd9a00 2022-09-14 op if (clt_puts(clt, "><head>") == -1)
706 fc1cb9ea 2022-09-14 op if (*pc->stylesheet != '\0' &&
707 fc1cb9ea 2022-09-14 op clt_printf(clt, "<link rel='stylesheet' href='%s' />",
708 fc1cb9ea 2022-09-14 op pc->stylesheet) == -1)
710 fc1cb9ea 2022-09-14 op if (clt_puts(clt, "</head><body>") == -1)
715 1337dc2b 2022-09-14 op * Trigger the read again so we proceed with the response
716 1337dc2b 2022-09-14 op * body, if any.
719 1337dc2b 2022-09-14 op proxy_read(bev, d);
727 b0a6bcf7 2022-09-13 op proxy_write(struct bufferevent *bev, void *d)
733 b0a6bcf7 2022-09-13 op proxy_error(struct bufferevent *bev, short err, void *d)
735 b0a6bcf7 2022-09-13 op struct client *clt = d;
736 b0a6bcf7 2022-09-13 op int status = !(err & EVBUFFER_EOF);
738 b0a6bcf7 2022-09-13 op log_debug("proxy error, shutting down the connection (err: %x)",
741 b0a6bcf7 2022-09-13 op if (!clt->clt_headersdone) {
742 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Status: 501\r\n") == -1)
744 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
746 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Proxy error\n") == -1)
748 c9dc4bc6 2022-09-13 op } else if (status == 0) {
749 8fa3006a 2022-09-14 op if (clt_puts(clt, "<footer>") == -1 ||
750 8fa3006a 2022-09-14 op clt_puts(clt, "<hr />") == -1 ||
751 8fa3006a 2022-09-14 op clt_puts(clt, "<dl>") == -1 ||
752 8fa3006a 2022-09-14 op clt_puts(clt, "<dt>Original URL:</dt>") == -1 ||
753 8fa3006a 2022-09-14 op clt_puts(clt, "<dd><a href='gemini://") == -1 ||
754 8fa3006a 2022-09-14 op printurl(clt, clt->clt_pc->proxy_name) == -1 ||
755 8fa3006a 2022-09-14 op printurl(clt, clt->clt_path_info) == -1 ||
756 8fa3006a 2022-09-14 op clt_puts(clt, "'>gemini://") == -1 ||
757 8fa3006a 2022-09-14 op htmlescape(clt, clt->clt_pc->proxy_name) == -1 ||
758 8fa3006a 2022-09-14 op htmlescape(clt, clt->clt_path_info) == -1 ||
759 8fa3006a 2022-09-14 op clt_puts(clt, "</a></dd>") == -1 ||
760 8fa3006a 2022-09-14 op clt_puts(clt, "</footer>") == -1)
762 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "</body></html>") == -1)
766 b0a6bcf7 2022-09-13 op fcgi_end_request(clt, status);
770 b0a6bcf7 2022-09-13 op proxy_tls_readcb(int fd, short event, void *arg)
772 b0a6bcf7 2022-09-13 op struct bufferevent *bufev = arg;
773 b0a6bcf7 2022-09-13 op struct client *clt = bufev->cbarg;
774 b0a6bcf7 2022-09-13 op char rbuf[IBUF_READ_SIZE];
775 b0a6bcf7 2022-09-13 op int what = EVBUFFER_READ;
776 b0a6bcf7 2022-09-13 op int howmuch = IBUF_READ_SIZE;
780 b0a6bcf7 2022-09-13 op if (event == EV_TIMEOUT) {
781 b0a6bcf7 2022-09-13 op what |= EVBUFFER_TIMEOUT;
785 b0a6bcf7 2022-09-13 op if (bufev->wm_read.high != 0)
786 b0a6bcf7 2022-09-13 op howmuch = MINIMUM(sizeof(rbuf), bufev->wm_read.high);
788 b0a6bcf7 2022-09-13 op ret = tls_read(clt->clt_ctx, rbuf, howmuch);
789 b0a6bcf7 2022-09-13 op if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
791 b0a6bcf7 2022-09-13 op } else if (ret == -1) {
792 b0a6bcf7 2022-09-13 op what |= EVBUFFER_ERROR;
797 b0a6bcf7 2022-09-13 op if (len == 0) {
798 b0a6bcf7 2022-09-13 op what |= EVBUFFER_EOF;
802 b0a6bcf7 2022-09-13 op if (evbuffer_add(bufev->input, rbuf, len) == -1) {
803 b0a6bcf7 2022-09-13 op what |= EVBUFFER_ERROR;
807 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
809 b0a6bcf7 2022-09-13 op len = EVBUFFER_LENGTH(bufev->input);
810 b0a6bcf7 2022-09-13 op if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
812 b0a6bcf7 2022-09-13 op if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
813 b0a6bcf7 2022-09-13 op struct evbuffer *buf = bufev->input;
814 b0a6bcf7 2022-09-13 op event_del(&bufev->ev_read);
815 b0a6bcf7 2022-09-13 op evbuffer_setcb(buf, bufferevent_read_pressure_cb, bufev);
819 b0a6bcf7 2022-09-13 op if (bufev->readcb != NULL)
820 b0a6bcf7 2022-09-13 op (*bufev->readcb)(bufev, bufev->cbarg);
824 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
828 b0a6bcf7 2022-09-13 op (*bufev->errorcb)(bufev, what, bufev->cbarg);
832 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(struct event *ev, int timeout)
834 b0a6bcf7 2022-09-13 op struct timeval tv, *ptv = NULL;
836 b0a6bcf7 2022-09-13 op if (timeout) {
837 b0a6bcf7 2022-09-13 op timerclear(&tv);
838 b0a6bcf7 2022-09-13 op tv.tv_sec = timeout;
842 b0a6bcf7 2022-09-13 op return (event_add(ev, ptv));
846 b0a6bcf7 2022-09-13 op proxy_tls_writecb(int fd, short event, void *arg)
848 b0a6bcf7 2022-09-13 op struct bufferevent *bufev = arg;
849 b0a6bcf7 2022-09-13 op struct client *clt = bufev->cbarg;
851 b0a6bcf7 2022-09-13 op short what = EVBUFFER_WRITE;
854 b0a6bcf7 2022-09-13 op if (event == EV_TIMEOUT) {
855 b0a6bcf7 2022-09-13 op what |= EVBUFFER_TIMEOUT;
859 b0a6bcf7 2022-09-13 op if (EVBUFFER_LENGTH(bufev->output)) {
860 b0a6bcf7 2022-09-13 op ret = tls_write(clt->clt_ctx,
861 b0a6bcf7 2022-09-13 op EVBUFFER_DATA(bufev->output),
862 b0a6bcf7 2022-09-13 op EVBUFFER_LENGTH(bufev->output));
863 b0a6bcf7 2022-09-13 op if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
865 b0a6bcf7 2022-09-13 op } else if (ret == -1) {
866 b0a6bcf7 2022-09-13 op what |= EVBUFFER_ERROR;
870 b0a6bcf7 2022-09-13 op evbuffer_drain(bufev->output, len);
873 b0a6bcf7 2022-09-13 op if (EVBUFFER_LENGTH(bufev->output) != 0)
874 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
876 b0a6bcf7 2022-09-13 op if (bufev->writecb != NULL &&
877 b0a6bcf7 2022-09-13 op EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
878 b0a6bcf7 2022-09-13 op (*bufev->writecb)(bufev, bufev->cbarg);
882 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
886 b0a6bcf7 2022-09-13 op (*bufev->errorcb)(bufev, what, bufev->cbarg);
890 b0a6bcf7 2022-09-13 op proxy_client_free(struct client *clt)
892 b0a6bcf7 2022-09-13 op if (clt->clt_evasr)
893 b0a6bcf7 2022-09-13 op event_asr_abort(clt->clt_evasr);
895 b0a6bcf7 2022-09-13 op if (clt->clt_addrinfo)
896 b0a6bcf7 2022-09-13 op freeaddrinfo(clt->clt_addrinfo);
898 b0a6bcf7 2022-09-13 op if (clt->clt_evconn_live)
899 b0a6bcf7 2022-09-13 op event_del(&clt->clt_evconn);
901 b0a6bcf7 2022-09-13 op if (clt->clt_fd != -1)
902 b0a6bcf7 2022-09-13 op close(clt->clt_fd);
904 b0a6bcf7 2022-09-13 op if (clt->clt_ctx)
905 b0a6bcf7 2022-09-13 op tls_free(clt->clt_ctx);
907 b0a6bcf7 2022-09-13 op if (clt->clt_bev)
908 b0a6bcf7 2022-09-13 op bufferevent_free(clt->clt_bev);
910 b0a6bcf7 2022-09-13 op free(clt->clt_server_name);
911 b0a6bcf7 2022-09-13 op free(clt->clt_script_name);
912 b0a6bcf7 2022-09-13 op free(clt->clt_path_info);