Blob


1 /*
2 * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
3 * Copyright (c) 2006 - 2015 Reyk Floeter <reyk@openbsd.org>
4 *
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 */
18 #include <sys/types.h>
19 #include <sys/queue.h>
20 #include <sys/tree.h>
21 #include <sys/socket.h>
22 #include <sys/uio.h>
24 #include <netinet/in.h>
25 #include <arpa/inet.h>
26 #include <netdb.h>
28 #include <asr.h>
29 #include <ctype.h>
30 #include <errno.h>
31 #include <event.h>
32 #include <fnmatch.h>
33 #include <limits.h>
34 #include <stdlib.h>
35 #include <stdint.h>
36 #include <stdio.h>
37 #include <string.h>
38 #include <imsg.h>
39 #include <tls.h>
40 #include <unistd.h>
42 #include "config.h"
43 #include "log.h"
44 #include "proc.h"
45 #include "tmpl.h"
47 #include "galileo.h"
49 #define MINIMUM(a, b) ((a) < (b) ? (a) : (b))
51 #if HAVE_LIBEVENT2
52 # define G_TOUT(t) ((t).tv_sec)
53 #else
54 # define G_TOUT(t) (t)
55 #endif
57 /* provided by OpenBSD' base libevent but not in any header? */
58 extern void bufferevent_read_pressure_cb(struct evbuffer *, size_t,
59 size_t, void *);
61 void proxy_init(struct privsep *, struct privsep_proc *, void *);
62 int proxy_launch(struct galileo *);
63 void proxy_inflight_dec(const char *);
64 int proxy_dispatch_parent(int, struct privsep_proc *, struct imsg *);
65 int proxy_resurl(struct client *, const char *, char *, size_t);
66 void proxy_translate_gemtext(struct client *);
67 void proxy_resolved(struct asr_result *, void *);
68 void proxy_connect(int, short, void *);
69 int proxy_start_reply(struct client *, int, const char *);
70 void proxy_read(struct bufferevent *, void *);
71 void proxy_write(struct bufferevent *, void *);
72 void proxy_error(struct bufferevent *, short, void *);
73 int proxy_bufferevent_add(struct event *, int);
74 void proxy_tls_writecb(int, short, void *);
75 void proxy_tls_readcb(int, short, void *);
77 static struct privsep_proc procs[] = {
78 { "parent", PROC_PARENT, proxy_dispatch_parent },
79 };
81 volatile int proxy_clients;
82 volatile int proxy_inflight;
83 uint32_t proxy_fcg_id;
85 void
86 proxy(struct privsep *ps, struct privsep_proc *p)
87 {
88 proc_run(ps, p, procs, nitems(procs), proxy_init, NULL);
89 }
91 void
92 proxy_init(struct privsep *ps, struct privsep_proc *p, void *arg)
93 {
94 if (config_init(ps->ps_env) == -1)
95 fatal("failed to initialize configuration");
97 /* We use a custom shutdown callback */
98 /* p->p_shutdown = proxy_shutdown */
100 if (pledge("stdio recvfd unix inet dns", NULL) == -1)
101 fatal("pledge");
104 int
105 proxy_launch(struct galileo *env)
107 event_add(&env->sc_evsock, NULL);
108 return (0);
111 void
112 proxy_purge(struct proxy *pr)
114 free(pr);
117 void
118 proxy_inflight_dec(const char *why)
120 proxy_inflight--;
121 log_debug("%s: inflight decremented, now %d, %s",
122 __func__, proxy_inflight, why);
125 int
126 proxy_dispatch_parent(int fd, struct privsep_proc *p, struct imsg *imsg)
128 struct privsep *ps = p->p_ps;
129 struct galileo *env = ps->ps_env;
131 switch (imsg->hdr.type) {
132 case IMSG_CFG_SRV:
133 if (config_getproxy(env, imsg) == -1)
134 fatal("config_getproxy");
135 break;
136 case IMSG_CFG_SOCK:
137 /* XXX: improve */
139 env->sc_sock_fd = config_getsock(env, imsg);
140 if (env->sc_sock_fd == -1)
141 fatal("config_getsock");
143 event_set(&env->sc_evsock, env->sc_sock_fd,
144 EV_READ | EV_PERSIST, fcgi_accept, env);
145 evtimer_set(&env->sc_evpause, fcgi_accept, env);
146 break;
147 case IMSG_CFG_DONE:
148 config_getcfg(env, imsg);
149 proxy_launch(env);
150 break;
151 case IMSG_CTL_START:
152 break;
153 case IMSG_CTL_RESET:
154 config_getreset(env, imsg);
155 break;
156 default:
157 return (-1);
160 return (0);
163 static int
164 portok(const char **url, struct client *clt)
166 const char *u = *url;
167 const char *port = clt->clt_pc->proxy_port;
168 size_t len;
170 len = strlen(port);
172 if (*u == '\0' || *u == '/' || *u == '?' || *u == '#')
173 return (1);
174 if (*u != ':')
175 return (0);
176 u++;
177 if (strncmp(u, port, len) != 0)
178 return (0);
179 u += len;
180 if (*u != '\0' && *u != '/' && *u != '?' && *u != '#')
181 return (0);
182 *url = u;
183 return (1);
186 int
187 proxy_resurl(struct client *clt, const char *url, char *buf, size_t len)
189 const char *tmp;
190 size_t l;
192 if (clt->clt_server_name == NULL)
193 return (-1);
195 l = strlen(clt->clt_server_name);
197 if (len == 0) {
198 log_warn("%s: zero-sized buffer!", __func__);
199 return (-1);
202 /* look if it's an absolute URI */
203 if (strncmp(url, "//", 2) == 0) {
204 tmp = url + 2;
205 if (strncmp(tmp, clt->clt_server_name, l) != 0)
206 goto done;
208 tmp += l;
209 if (!portok(&tmp, clt))
210 goto done;
211 url = tmp;
212 } else if (strncmp(url, "gemini://", 9) == 0) {
213 tmp = url + 9;
214 if (strncmp(tmp, clt->clt_server_name, l) != 0)
215 goto done;
217 tmp += l;
218 if (!portok(&tmp, clt))
219 goto done;
220 url = tmp;
221 } else {
222 tmp = url;
223 while (isalpha((unsigned char)*tmp) || *tmp == '+')
224 tmp++;
225 if (strncmp(tmp, "://", 3) == 0)
226 goto done;
229 /* maybe it's an absolute path? */
230 if (*url == '\0' || *url == '/') {
231 if (strlcpy(buf, clt->clt_script_name, len) >= len)
232 return (-1);
233 if (strlcat(buf, url + 1, len) >= len)
234 return (-1);
235 return (0);
238 /* otherwise, leave it as it is */
239 done:
240 if (strlcpy(buf, url, len) >= len)
241 return (-1);
242 return (0);
245 static inline int
246 match_image_heur(const char *url)
248 return (fnmatch("*.jpg", url, 0) == 0 ||
249 fnmatch("*.jpeg", url, 0) == 0 ||
250 fnmatch("*.gif", url, 0) == 0 ||
251 fnmatch("*.png", url, 0) == 0 ||
252 fnmatch("*.svg", url, 0) == 0 ||
253 fnmatch("*.webp", url, 0) == 0);
256 static int
257 gemtext_translate_line(struct client *clt, char *line)
259 char buf[1025];
260 char *url;
262 /* preformatted line / closing */
263 if (clt->clt_translate & TR_PRE) {
264 if (!strncmp(line, "```", 3)) {
265 clt->clt_translate &= ~TR_PRE;
266 return (tp_pre_close(clt->clt_tp));
269 if (tp_htmlescape(clt->clt_tp, line) == -1)
270 return (-1);
271 return (clt_putc(clt, '\n'));
274 /* bullet */
275 if (!strncmp(line, "* ", 2)) {
276 if (clt->clt_translate & TR_NAV) {
277 if (clt_puts(clt, "</ul></nav>") == -1)
278 return (-1);
279 clt->clt_translate &= ~TR_NAV;
282 if (!(clt->clt_translate & TR_LIST)) {
283 if (clt_puts(clt, "<ul>") == -1)
284 return (-1);
285 clt->clt_translate |= TR_LIST;
288 if (clt_puts(clt, "<li>") == -1 ||
289 tp_htmlescape(clt->clt_tp, line + 2) == -1 ||
290 clt_puts(clt, "</li>") == -1)
291 return (-1);
292 return (0);
295 if (clt->clt_translate & TR_LIST) {
296 if (clt_puts(clt, "</ul>") == -1)
297 return (-1);
298 clt->clt_translate &= ~TR_LIST;
301 /* link */
302 if (!strncmp(line, "=>", 2)) {
303 char *label;
305 line += 2;
306 line += strspn(line, " \t");
308 label = line + strcspn(line, " \t");
309 if (*label == '\0')
310 label = line;
311 else {
312 *label++ = '\0';
313 label += strspn(label, " \t");
314 if (*label == '\0')
315 label = line;
318 if (proxy_resurl(clt, line, buf, sizeof(buf)) == 0)
319 url = buf;
320 else
321 url = line; /* leave the URL as it is */
323 if (!(clt->clt_pc->flags & PROXY_NO_IMGPRV) &&
324 match_image_heur(url)) {
325 if (clt->clt_translate & TR_NAV) {
326 if (clt_puts(clt, "</ul></nav>") == -1)
327 return (-1);
328 clt->clt_translate &= ~TR_NAV;
331 if (tp_figure(clt->clt_tp, url, label) == -1)
332 return (-1);
334 return (0);
337 if (!(clt->clt_translate & TR_NAV)) {
338 if (clt_puts(clt, "<nav><ul>") == -1)
339 return (-1);
340 clt->clt_translate |= TR_NAV;
343 if (clt_puts(clt, "<li><a href='") == -1)
344 return (-1);
346 if (tp_urlescape(clt->clt_tp, url) == -1 ||
347 clt_puts(clt, "'>") == -1 ||
348 tp_htmlescape(clt->clt_tp, label) == -1 ||
349 clt_puts(clt, "</a></li>") == -1)
350 return (-1);
352 return (0);
355 if (clt->clt_translate & TR_NAV) {
356 if (clt_puts(clt, "</ul></nav>") == -1)
357 return (-1);
358 clt->clt_translate &= ~TR_NAV;
361 /* pre opening */
362 if (!strncmp(line, "```", 3)) {
363 line += 3;
364 line += strspn(line, " \t");
366 clt->clt_translate |= TR_PRE;
367 return (tp_pre_open(clt->clt_tp, line));
370 /* citation block */
371 if (*line == '>') {
372 if (clt_puts(clt, "<blockquote>") == -1 ||
373 tp_htmlescape(clt->clt_tp, line + 1) == -1 ||
374 clt_puts(clt, "</blockquote>") == -1)
375 return (-1);
376 return (0);
379 /* headings */
380 if (!strncmp(line, "###", 3)) {
381 if (clt_puts(clt, "<h3>") == -1 ||
382 tp_htmlescape(clt->clt_tp, line + 3) == -1 ||
383 clt_puts(clt, "</h3>") == -1)
384 return (-1);
385 return (0);
387 if (!strncmp(line, "##", 2)) {
388 if (clt_puts(clt, "<h2>") == -1 ||
389 tp_htmlescape(clt->clt_tp, line + 2) == -1 ||
390 clt_puts(clt, "</h2>") == -1)
391 return (-1);
392 return (0);
394 if (!strncmp(line, "#", 1)) {
395 if (clt_puts(clt, "<h1>") == -1 ||
396 tp_htmlescape(clt->clt_tp, line + 1) == -1 ||
397 clt_puts(clt, "</h1>") == -1)
398 return (-1);
399 return (0);
402 /* Not following strictly the gemini specification... */
403 if (*line == '\0')
404 return (0);
406 /* paragraph */
407 if (clt_puts(clt, "<p>") == -1 ||
408 tp_htmlescape(clt->clt_tp, line) == -1 ||
409 clt_puts(clt, "</p>") == -1)
410 return (-1);
412 return (0);
415 void
416 proxy_translate_gemtext(struct client *clt)
418 struct bufferevent *bev = clt->clt_bev;
419 struct evbuffer *src = EVBUFFER_INPUT(bev);
420 char *line;
421 size_t len;
422 int r;
424 for (;;) {
425 line = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF);
426 if (line == NULL)
427 return;
429 r = gemtext_translate_line(clt, line);
430 free(line);
431 if (r == -1)
432 return;
436 struct proxy_config *
437 proxy_match(struct galileo *env, const char *name)
439 struct proxy *pr;
441 if (name == NULL)
442 return (NULL);
444 TAILQ_FOREACH(pr, &env->sc_proxies, pr_entry) {
445 if (!strcmp(name, pr->pr_conf.host))
446 return (&pr->pr_conf);
449 return (NULL);
452 int
453 proxy_start_request(struct galileo *env, struct client *clt)
455 struct addrinfo hints;
456 struct asr_query *query;
457 int r;
458 char *url;
460 if (clt->clt_path_info == NULL) {
461 log_warnx("PATH_INFO not defined!");
462 if (proxy_start_reply(clt, 501, "text/html") == -1)
463 return (-1);
464 if (tp_error(clt->clt_tp, -1, "internal server error") == -1)
465 return (-1);
466 return (fcgi_end_request(clt, 1));
469 if ((clt->clt_pc = proxy_match(env, clt->clt_server_name)) == NULL) {
470 if (proxy_start_reply(clt, 501, "text/html") == -1)
471 return (-1);
472 if (tp_error(clt->clt_tp, -1, "unknown server") == -1)
473 return (-1);
474 return (fcgi_end_request(clt, 1));
477 if (clt->clt_bodylen != 0 && clt->clt_body == NULL) {
478 if (proxy_start_reply(clt, 400, "text/html") == -1)
479 return (-1);
480 if (tp_error(clt->clt_tp, -1, "bad request") == -1)
481 return (-1);
482 return (fcgi_end_request(clt, 1));
485 if (clt->clt_body) {
486 r = asprintf(&url, "%s%s?%s", clt->clt_script_name,
487 clt->clt_path_info + 1, clt->clt_body);
488 if (r == -1)
489 return (fcgi_end_request(clt, 1));
491 if (proxy_start_reply(clt, 302, url) == -1 ||
492 fcgi_end_request(clt, 1) == -1) {
493 free(url);
494 return (-1);
496 free(url);
497 return (0);
500 memset(&hints, 0, sizeof(hints));
501 hints.ai_family = AF_UNSPEC;
502 hints.ai_socktype = SOCK_STREAM;
504 query = getaddrinfo_async(clt->clt_pc->proxy_addr,
505 clt->clt_pc->proxy_port, &hints, NULL);
506 if (query == NULL) {
507 log_warn("getaddrinfo_async");
508 return (fcgi_abort_request(clt));
511 clt->clt_evasr = event_asr_run(query, proxy_resolved, clt);
512 if (clt->clt_evasr == NULL) {
513 log_warn("event_asr_run");
514 asr_abort(query);
515 return (fcgi_abort_request(clt));
518 return (0);
521 void
522 proxy_resolved(struct asr_result *res, void *d)
524 struct client *clt = d;
525 struct proxy_config *pc = clt->clt_pc;
527 clt->clt_evasr = NULL;
529 if (res->ar_gai_errno != 0) {
530 log_warnx("failed to resolve %s:%s: %s",
531 pc->proxy_addr, pc->proxy_port,
532 gai_strerror(res->ar_gai_errno));
533 if (proxy_start_reply(clt, 501, "text/html") == -1)
534 return;
535 if (tp_error(clt->clt_tp, -1, "Can't resolve host") == -1)
536 return;
537 fcgi_end_request(clt, 1);
538 return;
541 clt->clt_addrinfo = res->ar_addrinfo;
542 clt->clt_p = clt->clt_addrinfo;
543 proxy_connect(-1, 0, clt);
546 void
547 proxy_connect(int fd, short ev, void *d)
549 struct client *clt = d;
550 struct evbuffer *out;
551 struct addrinfo *p;
552 struct tls_config *conf;
553 struct timeval conntv = {5, 0};
554 int err = 0;
555 socklen_t len = sizeof(err);
557 again:
558 if (clt->clt_p == NULL)
559 goto err;
561 if (clt->clt_fd != -1) {
562 if (getsockopt(clt->clt_fd, SOL_SOCKET, SO_ERROR, &err, &len)
563 == -1)
564 goto err;
565 if (err != 0) {
566 errno = err;
567 goto err;
569 goto done;
572 p = clt->clt_p;
573 clt->clt_fd = socket(p->ai_family, p->ai_socktype | SOCK_NONBLOCK,
574 p->ai_protocol);
575 if (clt->clt_fd == -1) {
576 clt->clt_p = clt->clt_p->ai_next;
577 goto again;
580 if (connect(clt->clt_fd, p->ai_addr, p->ai_addrlen) == 0)
581 goto done;
583 clt->clt_evconn_live = 1;
584 event_set(&clt->clt_evconn, clt->clt_fd, EV_WRITE, proxy_connect, clt);
585 event_add(&clt->clt_evconn, &conntv);
586 return;
588 done:
589 clt->clt_evconn_live = 0;
590 freeaddrinfo(clt->clt_addrinfo);
591 clt->clt_addrinfo = clt->clt_p = NULL;
593 clt->clt_bev = bufferevent_new(clt->clt_fd, proxy_read, proxy_write,
594 proxy_error, clt);
595 if (clt->clt_bev == NULL) {
596 log_warn("bufferevent_new");
597 goto err;
600 if (!(clt->clt_pc->flags & PROXY_NO_TLS)) {
601 /* initialize TLS for Gemini */
602 if ((conf = tls_config_new()) == NULL) {
603 log_warn("tls_config_new failed");
604 goto err;
607 tls_config_insecure_noverifycert(conf);
609 if ((clt->clt_ctx = tls_client()) == NULL) {
610 log_warnx("tls_client failed");
611 tls_config_free(conf);
612 goto err;
615 if (tls_configure(clt->clt_ctx, conf) == -1) {
616 log_warnx("tls_configure failed");
617 tls_config_free(conf);
618 goto err;
621 tls_config_free(conf);
623 if (tls_connect_socket(clt->clt_ctx, clt->clt_fd,
624 clt->clt_pc->proxy_name) == -1) {
625 log_warnx("tls_connect_socket failed");
626 goto err;
629 event_set(&clt->clt_bev->ev_read, clt->clt_fd, EV_READ,
630 proxy_tls_readcb, clt->clt_bev);
631 event_set(&clt->clt_bev->ev_write, clt->clt_fd, EV_WRITE,
632 proxy_tls_writecb, clt->clt_bev);
634 #if HAVE_LIBEVENT2
635 evbuffer_unfreeze(clt->clt_bev->input, 0);
636 evbuffer_unfreeze(clt->clt_bev->output, 1);
637 #endif
640 /* bufferevent_settimeout(); */
641 bufferevent_enable(clt->clt_bev, EV_READ|EV_WRITE);
643 /* TODO: compute the URL */
644 out = EVBUFFER_OUTPUT(clt->clt_bev);
645 if (evbuffer_add_printf(out, "gemini://%s/%s",
646 clt->clt_pc->proxy_name, clt->clt_path_info) == -1) {
647 log_warn("bufferevent_printf failed");
648 goto err;
651 if (clt->clt_query &&
652 evbuffer_add_printf(out, "?%s", clt->clt_query) == -1) {
653 log_warn("bufferevent_printf failed");
654 goto err;
657 if (evbuffer_add(out, "\r\n", 2) == -1) {
658 log_warn("bufferevent_add failed");
659 goto err;
662 return;
664 err:
665 log_warn("failed to connect to %s:%s",
666 clt->clt_pc->proxy_addr, clt->clt_pc->proxy_port);
667 if (proxy_start_reply(clt, 501, "text/html") == -1)
668 return;
669 if (tp_error(clt->clt_tp, -1, "Can't connect") == -1)
670 return;
671 fcgi_end_request(clt, 1);
674 static inline int
675 parse_mime(struct client *clt, char *mime, char *lang, size_t len)
677 char *t, *semi;
679 if (strncmp(mime, "text/gemini", 11) != 0)
680 return (0);
682 clt->clt_translate = TR_ENABLED;
684 if ((mime = strchr(mime, ';')) == NULL)
685 return (0);
687 *mime++ = '\0';
688 while ((t = strsep(&mime, ";")) != NULL) {
689 if (!strncmp(t, "charset=", 8)) {
690 t += 8;
691 if (!strncasecmp(t, "utf8", 4) ||
692 !strncasecmp(t, "utf-8", 5) ||
693 !strncasecmp(t, "ascii", 5)) {
694 log_debug("unknown charset %s", t);
695 return (-1);
697 continue;
700 if (!strncmp(t, "lang=", 5)) {
701 t += 5;
702 if ((semi = strchr(t, ';')) != NULL)
703 *semi = '\0';
705 if (strlcpy(lang, t, len) >= len) {
706 log_debug("lang too long: %s", t);
707 *lang = '\0';
710 if (semi)
711 *semi = ';';
712 continue;
716 return (0);
719 int
720 proxy_start_reply(struct client *clt, int status, const char *ctype)
722 const char *csp;
724 csp = "Content-Security-Policy: default-src 'self'; "
725 "script-src 'none'; object-src 'none';\r\n";
727 if (status != 200 &&
728 clt_printf(clt, "Status: %d\r\n", status) == -1)
729 return (-1);
731 if (clt_puts(clt, csp) == -1)
732 return (-1);
734 if (status == 302) {
735 /* use "ctype" as redirect target */
736 if (clt_printf(clt, "Location: %s\r\n", ctype) == -1)
737 return (-1);
738 if (clt_puts(clt, "\r\n") == -1)
739 return (-1);
740 return (0);
743 if (ctype != NULL) {
744 if (!strcmp(ctype, "text/html"))
745 ctype = "text/html;charset=utf-8";
746 if (clt_printf(clt, "Content-Type: %s\r\n", ctype)
747 == -1)
748 return (-1);
751 if (clt_puts(clt, "\r\n") == -1)
752 return (-1);
754 return (0);
757 void
758 proxy_read(struct bufferevent *bev, void *d)
760 struct client *clt = d;
761 struct evbuffer *src = EVBUFFER_INPUT(bev);
762 const char *ctype;
763 char buf[1025];
764 char lang[16];
765 char *hdr, *mime;
766 size_t len;
767 int code;
769 if (clt->clt_headersdone) {
770 if (clt->clt_translate)
771 proxy_translate_gemtext(clt);
772 else
773 clt_write_bufferevent(clt, bev);
774 return;
777 hdr = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF_STRICT);
778 if (hdr == NULL) {
779 if (EVBUFFER_LENGTH(src) >= 1026)
780 proxy_error(bev, EV_READ, clt);
781 return;
784 if (len < 4 ||
785 !isdigit((unsigned char)hdr[0]) ||
786 !isdigit((unsigned char)hdr[1]) ||
787 hdr[2] != ' ') {
788 log_warnx("invalid ");
789 proxy_error(bev, EV_READ, clt);
790 goto err;
793 code = (hdr[0] - '0') * 10 + (hdr[1] - '0');
795 switch (hdr[0]) {
796 case '1':
797 if (proxy_start_reply(clt, 200, "text/html") == -1)
798 goto err;
799 if (tp_inputpage(clt->clt_tp, &hdr[3]) == -1)
800 goto err;
801 fcgi_end_request(clt, 0);
802 goto err;
803 case '2':
804 /* handled below */
805 break;
806 case '3':
807 if (proxy_resurl(clt, &hdr[3], buf, sizeof(buf)) == 0) {
808 if (proxy_start_reply(clt, 302, buf) == -1)
809 goto err;
810 fcgi_end_request(clt, 0);
811 goto err;
813 /* fallthrough */
814 default:
815 if (proxy_start_reply(clt, 501, "text/html") == -1)
816 goto err;
817 if (tp_error(clt->clt_tp, code, &hdr[3]) == -1)
818 goto err;
819 fcgi_end_request(clt, 1);
820 goto err;
823 mime = hdr + 2 + strspn(hdr + 2, " \t");
824 if (parse_mime(clt, mime, lang, sizeof(lang)) == -1) {
825 if (proxy_start_reply(clt, 501, "text/html") == -1)
826 goto err;
827 if (tp_error(clt->clt_tp, -1, "Bad response") == -1)
828 goto err;
829 fcgi_end_request(clt, 1);
830 goto err;
833 if (clt->clt_translate)
834 ctype = "text/html;charset=utf-8";
835 else
836 ctype = mime;
838 if (clt_printf(clt, "Content-Type: %s\r\n\r\n", ctype) == -1)
839 goto err;
841 clt->clt_headersdone = 1;
843 if (clt->clt_translate &&
844 tp_head(clt->clt_tp, lang, NULL) == -1)
845 goto err;
847 /*
848 * Trigger the read again so we proceed with the response
849 * body, if any.
850 */
851 free(hdr);
852 proxy_read(bev, d);
853 return;
855 err:
856 free(hdr);
859 void
860 proxy_write(struct bufferevent *bev, void *d)
862 return;
865 void
866 proxy_error(struct bufferevent *bev, short err, void *d)
868 struct client *clt = d;
869 int status = !(err & EVBUFFER_EOF);
871 log_debug("proxy error, shutting down the connection (err: %x)",
872 err);
874 if (!clt->clt_headersdone) {
875 if (proxy_start_reply(clt, 501, "text/html") == -1)
876 return;
877 if (tp_error(clt->clt_tp, -1, "Proxy error") == -1)
878 return;
879 } else if (status == 0) {
880 if (clt->clt_translate & TR_PRE) {
881 if (tp_pre_close(clt->clt_tp))
882 return;
883 clt->clt_translate &= ~TR_PRE;
886 if (clt->clt_translate & TR_LIST) {
887 if (clt_puts(clt, "</ul>") == -1)
888 return;
889 clt->clt_translate &= ~TR_LIST;
892 if (clt->clt_translate & TR_NAV) {
893 if (clt_puts(clt, "</ul></nav>") == -1)
894 return;
895 clt->clt_translate &= ~TR_NAV;
898 if (clt->clt_translate &&
899 tp_foot(clt->clt_tp) == -1)
900 return;
903 fcgi_end_request(clt, status);
906 void
907 proxy_tls_readcb(int fd, short event, void *arg)
909 struct bufferevent *bufev = arg;
910 struct client *clt = bufev->cbarg;
911 char rbuf[IBUF_READ_SIZE];
912 int what = EVBUFFER_READ;
913 int howmuch = IBUF_READ_SIZE;
914 ssize_t ret;
915 size_t len;
917 if (event == EV_TIMEOUT) {
918 what |= EVBUFFER_TIMEOUT;
919 goto err;
922 if (bufev->wm_read.high != 0)
923 howmuch = MINIMUM(sizeof(rbuf), bufev->wm_read.high);
925 ret = tls_read(clt->clt_ctx, rbuf, howmuch);
926 if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
927 goto retry;
928 } else if (ret == -1) {
929 what |= EVBUFFER_ERROR;
930 goto err;
932 len = ret;
934 if (len == 0) {
935 what |= EVBUFFER_EOF;
936 goto err;
939 if (evbuffer_add(bufev->input, rbuf, len) == -1) {
940 what |= EVBUFFER_ERROR;
941 goto err;
944 proxy_bufferevent_add(&bufev->ev_read, G_TOUT(bufev->timeout_read));
946 len = EVBUFFER_LENGTH(bufev->input);
947 if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
948 return;
949 if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
950 struct evbuffer *buf = bufev->input;
951 event_del(&bufev->ev_read);
952 evbuffer_setcb(buf, bufferevent_read_pressure_cb, bufev);
953 return;
956 if (bufev->readcb != NULL)
957 (*bufev->readcb)(bufev, bufev->cbarg);
958 return;
960 retry:
961 proxy_bufferevent_add(&bufev->ev_read, G_TOUT(bufev->timeout_read));
962 return;
964 err:
965 (*bufev->errorcb)(bufev, what, bufev->cbarg);
968 int
969 proxy_bufferevent_add(struct event *ev, int timeout)
971 struct timeval tv, *ptv = NULL;
973 if (timeout) {
974 timerclear(&tv);
975 tv.tv_sec = timeout;
976 ptv = &tv;
979 return (event_add(ev, ptv));
982 void
983 proxy_tls_writecb(int fd, short event, void *arg)
985 struct bufferevent *bufev = arg;
986 struct client *clt = bufev->cbarg;
987 ssize_t ret;
988 short what = EVBUFFER_WRITE;
989 size_t len;
991 if (event == EV_TIMEOUT) {
992 what |= EVBUFFER_TIMEOUT;
993 goto err;
996 if (EVBUFFER_LENGTH(bufev->output)) {
997 ret = tls_write(clt->clt_ctx,
998 EVBUFFER_DATA(bufev->output),
999 EVBUFFER_LENGTH(bufev->output));
1000 if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
1001 goto retry;
1002 } else if (ret == -1) {
1003 what |= EVBUFFER_ERROR;
1004 goto err;
1006 len = ret;
1007 evbuffer_drain(bufev->output, len);
1010 if (EVBUFFER_LENGTH(bufev->output) != 0)
1011 proxy_bufferevent_add(&bufev->ev_write,
1012 G_TOUT(bufev->timeout_write));
1014 if (bufev->writecb != NULL &&
1015 EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
1016 (*bufev->writecb)(bufev, bufev->cbarg);
1017 return;
1019 retry:
1020 proxy_bufferevent_add(&bufev->ev_write, G_TOUT(bufev->timeout_write));
1021 return;
1023 err:
1024 (*bufev->errorcb)(bufev, what, bufev->cbarg);
1027 void
1028 proxy_client_free(struct client *clt)
1030 if (clt->clt_evasr)
1031 event_asr_abort(clt->clt_evasr);
1033 if (clt->clt_addrinfo)
1034 freeaddrinfo(clt->clt_addrinfo);
1036 if (clt->clt_evconn_live)
1037 event_del(&clt->clt_evconn);
1039 if (clt->clt_fd != -1)
1040 close(clt->clt_fd);
1042 if (clt->clt_ctx)
1043 tls_free(clt->clt_ctx);
1045 if (clt->clt_bev)
1046 bufferevent_free(clt->clt_bev);
1048 template_free(clt->clt_tp);
1050 free(clt->clt_body);
1051 free(clt->clt_server_name);
1052 free(clt->clt_script_name);
1053 free(clt->clt_path_info);
1054 free(clt->clt_query);
1055 free(clt);