op public repos

Blob

Date:: Mon Sep 19 14:21:51 2022 UTC
Message:: import the template compiler i've been working on
Actions:: History | Blame | Raw File
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  */
17 
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>
23 
24 #include <netinet/in.h>
25 #include <arpa/inet.h>
26 #include <netdb.h>
27 
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>
41 
42 #include "log.h"
43 #include "proc.h"
44 #include "tmpl.h"
45 
46 #include "galileo.h"
47 
48 #define MINIMUM(a, b)	((a) < (b) ? (a) : (b))
49 
50 /* provided by OpenBSD' base libevent but not in any header? */
51 extern void	 bufferevent_read_pressure_cb(struct evbuffer *, size_t,
52 		    size_t, void *);
53 
54 void	proxy_init(struct privsep *, struct privsep_proc *, void *);
55 int	proxy_launch(struct galileo *);
56 void	proxy_inflight_dec(const char *);
57 int	proxy_dispatch_parent(int, struct privsep_proc *, struct imsg *);
58 void	proxy_translate_gemtext(struct client *);
59 void	proxy_resolved(struct asr_result *, void *);
60 void	proxy_connect(int, short, void *);
61 void	proxy_read(struct bufferevent *, void *);
62 void	proxy_write(struct bufferevent *, void *);
63 void	proxy_error(struct bufferevent *, short, void *);
64 int	proxy_bufferevent_add(struct event *, int);
65 void	proxy_tls_writecb(int, short, void *);
66 void	proxy_tls_readcb(int, short, void *);
67 
68 static struct privsep_proc procs[] = {
69 	{ "parent",	PROC_PARENT, proxy_dispatch_parent },
70 };
71 
72 volatile int proxy_clients;
73 volatile int proxy_inflight;
74 uint32_t proxy_fcg_id;
75 
76 void
77 proxy(struct privsep *ps, struct privsep_proc *p)
78 {
79 	proc_run(ps, p, procs, nitems(procs), proxy_init, NULL);
80 }
81 
82 void
83 proxy_init(struct privsep *ps, struct privsep_proc *p, void *arg)
84 {
85 	if (config_init(ps->ps_env) == -1)
86 		fatal("failed to initialize configuration");
87 
88 	/* We use a custom shutdown callback */
89 	/* p->p_shutdown = proxy_shutdown */
90 
91 	if (pledge("stdio recvfd unix inet dns", NULL) == -1)
92 		fatal("pledge");
93 }
94 
95 int
96 proxy_launch(struct galileo *env)
97 {
98 	event_add(&env->sc_evsock, NULL);
99 	return (0);
100 }
101 
102 void
103 proxy_purge(struct server *srv)
104 {
105 }
106 
107 void
108 proxy_inflight_dec(const char *why)
109 {
110 	proxy_inflight--;
111 	log_debug("%s: inflight decremented, now %d, %s",
112 	    __func__, proxy_inflight, why);
113 }
114 
115 int
116 proxy_dispatch_parent(int fd, struct privsep_proc *p, struct imsg *imsg)
117 {
118 	struct privsep	*ps = p->p_ps;
119 	struct galileo	*env = ps->ps_env;
120 
121 	switch (imsg->hdr.type) {
122 	case IMSG_CFG_SRV:
123 		if (config_getserver(env, imsg) == -1)
124 			fatal("config_getproxy");
125 		break;
126 	case IMSG_CFG_SOCK:
127 		/* XXX: improve */
128 
129 		if (env->sc_sock_fd != -1) {
130 			event_del(&env->sc_evsock);
131 			close(env->sc_sock_fd);
132 		}
133 
134 		env->sc_sock_fd = config_getsock(env, imsg);
135 		if (env->sc_sock_fd == -1)
136 			fatal("config_getsock");
137 
138 		event_set(&env->sc_evsock, env->sc_sock_fd,
139 		    EV_READ | EV_PERSIST, fcgi_accept, env);
140 		event_add(&env->sc_evsock, NULL);
141 		evtimer_set(&env->sc_evpause, fcgi_accept, env);
142 		break;
143 	case IMSG_CFG_DONE:
144 		log_debug("config done!");
145 		break;
146 	case IMSG_CTL_START:
147 		proxy_launch(env);
148 		break;
149 	default:
150 		log_warnx("unknown message %d", imsg->hdr.type);
151 		return (-1);
152 	}
153 
154 	return (0);
155 }
156 
157 static inline int
158 printurl(struct client *clt, const char *str)
159 {
160 	for (; *str; ++str) {
161 		switch (*str) {
162 		case ' ':
163 		case '\t':
164 		case '\'':
165 		case '\\':
166 			if (clt_printf(clt, "%2X", (unsigned char)*str) == -1)
167 				return (-1);
168 			break;
169 		default:
170 			if (clt_putc(clt, *str) == -1)
171 				return (-1);
172 			break;
173 		}
174 	}
175 
176 	return (0);
177 }
178 
179 static inline int
180 htmlescape(struct client *clt, const char *str)
181 {
182 	int r;
183 
184 	for (; *str; ++str) {
185 		switch (*str) {
186 		case '<':
187 			r = clt_puts(clt, "&lt;");
188 			break;
189 		case '>':
190 			r = clt_puts(clt, "&gt;");
191 			break;
192 		case '&':
193 			r = clt_puts(clt, "&amp;");
194 			break;
195 		default:
196 			r = clt_putc(clt, *str);
197 			break;
198 		}
199 
200 		if (r == -1)
201 			return (-1);
202 	}
203 
204 	return (0);
205 }
206 
207 static int
208 gemtext_translate_line(struct client *clt, char *line)
209 {
210 	/* preformatted line / closing */
211 	if (clt->clt_translate & TR_PRE) {
212 		if (!strncmp(line, "```", 3)) {
213 			clt->clt_translate &= ~TR_PRE;
214 			return (clt_puts(clt, "</pre>"));
215 		}
216 
217 		if (htmlescape(clt, line) == -1)
218 			return (-1);
219 		return (clt_putc(clt, '\n'));
220 	}
221 
222 	/* bullet */
223 	if (!strncmp(line, "* ", 2)) {
224 		if (clt->clt_translate & TR_NAV) {
225 			if (clt_puts(clt, "</ul></nav>") == -1)
226 				return (-1);
227 			clt->clt_translate &= ~TR_NAV;
228 		}
229 
230 		if (!(clt->clt_translate & TR_LIST)) {
231 			if (clt_puts(clt, "<ul>") == -1)
232 				return (-1);
233 			clt->clt_translate |= TR_LIST;
234 		}
235 
236 		if (clt_puts(clt, "<li>") == -1 ||
237 		    htmlescape(clt, line + 2) == -1 ||
238 		    clt_puts(clt, "</li>") == -1)
239 			return (-1);
240 		return (0);
241 	}
242 
243 	if (clt->clt_translate & TR_LIST) {
244 		if (clt_puts(clt, "</ul>") == -1)
245 			return (-1);
246 		clt->clt_translate &= ~TR_LIST;
247 	}
248 
249 	/* link -- TODO: relativify from SCRIPT_NAME */
250 	if (!strncmp(line, "=>", 2)) {
251 		char *label;
252 
253 		line += 2;
254 		line += strspn(line, " \t");
255 
256 		label = line + strcspn(line, " \t");
257 		if (*label == '\0')
258 			label = line;
259 		else
260 			*label++ = '\0';
261 
262 		if (fnmatch("*.jpg", line, 0) == 0 ||
263 		    fnmatch("*.jpeg", line, 0) == 0 ||
264 		    fnmatch("*.gif", line, 0) == 0 ||
265 		    fnmatch("*.png", line, 0) == 0 ||
266 		    fnmatch("*.svg", line, 0) == 0 ||
267 		    fnmatch("*.webp", line, 0) == 0) {
268 			if (clt->clt_translate & TR_NAV) {
269 				if (clt_puts(clt, "</ul></nav>") == -1)
270 					return (-1);
271 				clt->clt_translate &= ~TR_NAV;
272 			}
273 
274 			if (tp_figure(clt->clt_tp, line, label) == -1)
275 				return (-1);
276 
277 			return (0);
278 		}
279 
280 		if (!(clt->clt_translate & TR_NAV)) {
281 			if (clt_puts(clt, "<nav><ul>") == -1)
282 				return (-1);
283 			clt->clt_translate |= TR_NAV;
284 		}
285 
286 		if (clt_puts(clt, "<li><a href='") == -1 ||
287 		    printurl(clt, line) == -1 ||
288 		    clt_puts(clt, "'>") == -1 ||
289 		    htmlescape(clt, label) == -1 ||
290 		    clt_puts(clt, "</a></li>") == -1)
291 			return (-1);
292 
293 		return (0);
294 	}
295 
296 	if (clt->clt_translate & TR_NAV) {
297 		if (clt_puts(clt, "</ul></nav>") == -1)
298 			return (-1);
299 		clt->clt_translate &= ~TR_NAV;
300 	}
301 
302 	/* pre opening */
303 	if (!strncmp(line, "```", 3)) {
304 		clt->clt_translate |= TR_PRE;
305 		return (clt_puts(clt, "<pre>"));
306 	}
307 
308 	/* citation block */
309 	if (*line == '>') {
310 		if (clt_puts(clt, "<blockquote>") == -1 ||
311 		    htmlescape(clt, line + 1) == -1 ||
312 		    clt_puts(clt, "</blockquote>") == -1)
313 			return (-1);
314 		return (0);
315 	}
316 
317 	/* headings */
318 	if (!strncmp(line, "###", 3)) {
319 		if (clt_puts(clt, "<h3>") == -1 ||
320 		    htmlescape(clt, line + 3) == -1 ||
321 		    clt_puts(clt, "</h3>") == -1)
322 			return (-1);
323 		return (0);
324 	}
325 	if (!strncmp(line, "##", 2)) {
326 		if (clt_puts(clt, "<h2>") == -1 ||
327 		    htmlescape(clt, line + 2) == -1 ||
328 		    clt_puts(clt, "</h2>") == -1)
329 			return (-1);
330 		return (0);
331 	}
332 	if (!strncmp(line, "#", 1)) {
333 		if (clt_puts(clt, "<h1>") == -1 ||
334 		    htmlescape(clt, line + 1) == -1 ||
335 		    clt_puts(clt, "</h1>") == -1)
336 			return (-1);
337 		return (0);
338 	}
339 
340 	/* Not following strictly the gemini specification... */
341 	if (*line == '\0')
342 		return (0);
343 
344 	/* paragraph */
345 	if (clt_puts(clt, "<p>") == -1 ||
346 	    htmlescape(clt, line) == -1 ||
347 	    clt_puts(clt, "</p>") == -1)
348 		return (-1);
349 
350 	return (0);
351 }
352 
353 void
354 proxy_translate_gemtext(struct client *clt)
355 {
356 	struct bufferevent	*bev = clt->clt_bev;
357 	struct evbuffer		*src = EVBUFFER_INPUT(bev);
358 	char			*line;
359 	size_t			 len;
360 	int			 r;
361 
362 	for (;;) {
363 		line = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF);
364 		if (line == NULL)
365 			return;
366 
367 		r = gemtext_translate_line(clt, line);
368 		free(line);
369 		if (r == -1)
370 			return;
371 	}
372 }
373 
374 static struct proxy_config *
375 proxy_server_match(struct galileo *env, struct client *clt)
376 {
377 	struct server		*srv;
378 
379 	if (clt->clt_server_name == NULL)
380 		return NULL;
381 
382 	TAILQ_FOREACH(srv, &env->sc_servers, srv_entry) {
383 		if (!strcmp(clt->clt_server_name, srv->srv_conf.host))
384 			return &srv->srv_conf;
385 	}
386 
387 	return NULL;
388 }
389 
390 void
391 proxy_start_request(struct galileo *env, struct client *clt)
392 {
393 	struct addrinfo		 hints;
394 	struct asr_query	*query;
395 	char			 port[32];
396 
397 	if ((clt->clt_pc = proxy_server_match(env, clt)) == NULL) {
398 		if (clt_printf(clt, "Status: 501\r\n\r\n") == -1)
399 			return;
400 		fcgi_end_request(clt, 1);
401 		return;
402 	}
403 
404 	(void)snprintf(port, sizeof(port), "%d", clt->clt_pc->proxy_port);
405 
406 	memset(&hints, 0, sizeof(hints));
407 	hints.ai_family = AF_UNSPEC;
408 	hints.ai_socktype = SOCK_STREAM;
409 
410 	query = getaddrinfo_async(clt->clt_pc->proxy_addr, port, &hints, NULL);
411 	if (query == NULL) {
412 		log_warn("getaddrinfo_async");
413 		fcgi_abort_request(clt);
414 		return;
415 	}
416 
417 	clt->clt_evasr = event_asr_run(query, proxy_resolved, clt);
418 	if (clt->clt_evasr == NULL) {
419 		log_warn("event_asr_run");
420 		asr_abort(query);
421 		fcgi_abort_request(clt);
422 		return;
423 	}
424 }
425 
426 void
427 proxy_resolved(struct asr_result *res, void *d)
428 {
429 	struct client		*clt = d;
430 	struct proxy_config	*pc = clt->clt_pc;
431 
432 	clt->clt_evasr = NULL;
433 
434 	if (res->ar_gai_errno != 0) {
435 		log_warnx("failed to resolve %s:%d: %s",
436 		    pc->proxy_addr, pc->proxy_port,
437 		    gai_strerror(res->ar_gai_errno));
438 		if (clt_printf(clt, "Status: 501\r\n") == -1)
439 			return;
440 		if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
441 			return;
442 		if (clt_printf(clt, "Proxy error; connection failed") == -1)
443 			return;
444 		fcgi_end_request(clt, 1);
445 		return;
446 	}
447 
448 	clt->clt_addrinfo = res->ar_addrinfo;
449 	clt->clt_p = clt->clt_addrinfo;
450 	proxy_connect(-1, 0, clt);
451 }
452 
453 void
454 proxy_connect(int fd, short ev, void *d)
455 {
456 	struct client		*clt = d;
457 	struct evbuffer		*out;
458 	struct addrinfo		*p;
459 	struct tls_config	*conf;
460 	struct timeval		 conntv = {5, 0};
461 	int			 err = 0;
462 	socklen_t		 len = sizeof(err);
463 
464 again:
465 	if (clt->clt_p == NULL)
466 		goto err;
467 
468 	if (clt->clt_fd != -1) {
469 		if (getsockopt(clt->clt_fd, SOL_SOCKET, SO_ERROR, &err, &len)
470 		    == -1)
471 			goto err;
472 		if (err != 0) {
473 			errno = err;
474 			goto err;
475 		}
476 		goto done;
477 	}
478 
479 	p = clt->clt_p;
480 	clt->clt_fd = socket(p->ai_family, p->ai_socktype | SOCK_NONBLOCK,
481 	    p->ai_protocol);
482 	if (clt->clt_fd == -1) {
483 		clt->clt_p = clt->clt_p->ai_next;
484 		goto again;
485 	}
486 
487 	if (connect(clt->clt_fd, p->ai_addr, p->ai_addrlen) == 0)
488 		goto done;
489 
490 	clt->clt_evconn_live = 1;
491 	event_set(&clt->clt_evconn, clt->clt_fd, EV_WRITE, proxy_connect, clt);
492 	event_add(&clt->clt_evconn, &conntv);
493 	return;
494 
495 done:
496 	clt->clt_evconn_live = 0;
497 	freeaddrinfo(clt->clt_addrinfo);
498 	clt->clt_addrinfo = clt->clt_p = NULL;
499 
500 	/* initialize TLS for Gemini */
501 	if ((conf = tls_config_new()) == NULL) {
502 		log_warn("tls_config_new failed");
503 		goto err;
504 	}
505 
506 	tls_config_insecure_noverifycert(conf);
507 
508 	if ((clt->clt_ctx = tls_client()) == NULL) {
509 		log_warnx("tls_client failed");
510 		tls_config_free(conf);
511 		goto err;
512 	}
513 
514 	if (tls_configure(clt->clt_ctx, conf) == -1) {
515 		log_warnx("tls_configure failed");
516 		tls_config_free(conf);
517 		goto err;
518 	}
519 
520 	tls_config_free(conf);
521 
522 	if (tls_connect_socket(clt->clt_ctx, clt->clt_fd,
523 	    clt->clt_pc->proxy_name) == -1) {
524 		log_warnx("tls_connect_socket failed");
525 		goto err;
526 	}
527 
528 	clt->clt_bev = bufferevent_new(clt->clt_fd, proxy_read, proxy_write,
529 	    proxy_error, clt);
530 	if (clt->clt_bev == NULL) {
531 		log_warn("bufferevent_new");
532 		goto err;
533 	}
534 	out = EVBUFFER_OUTPUT(clt->clt_bev);
535 
536 	event_set(&clt->clt_bev->ev_read, clt->clt_fd, EV_READ,
537 	    proxy_tls_readcb, clt->clt_bev);
538 	event_set(&clt->clt_bev->ev_write, clt->clt_fd, EV_WRITE,
539 	    proxy_tls_writecb, clt->clt_bev);
540 
541 	/* bufferevent_settimeout(); */
542 	bufferevent_enable(clt->clt_bev, EV_READ|EV_WRITE);
543 
544 	/* TODO: compute the URL */
545 	if (evbuffer_add_printf(out, "gemini://%s/%s\r\n",
546 	    clt->clt_pc->proxy_name, clt->clt_path_info) == -1) {
547 		log_warn("bufferevent_printf failed");
548 		goto err;
549 	}
550 
551 	return;
552 
553 err:
554 	log_warn("failed to connect to %s:%d",
555 	    clt->clt_pc->proxy_addr, clt->clt_pc->proxy_port);
556 	if (clt_printf(clt, "Status: 501\r\n") == -1)
557 		return;
558 	if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
559 		return;
560 	if (clt_printf(clt, "Proxy error; connection failed") == -1)
561 		return;
562 	fcgi_end_request(clt, 1);
563 }
564 
565 static inline int
566 parse_mime(struct client *clt, char *mime, char *lang, size_t len)
567 {
568 	char			*t, *semi;
569 
570 	if (strncmp(mime, "text/gemini", 11) != 0)
571 		return (0);
572 
573 	clt->clt_translate = TR_ENABLED;
574 
575 	if ((mime = strchr(mime, ';')) == NULL)
576 		return (0);
577 
578 	*mime++ = '\0';
579 	while ((t = strsep(&mime, ";")) != NULL) {
580 		if (!strncmp(t, "charset=", 8)) {
581 			t += 8;
582 			if (!strncasecmp(t, "utf8", 4) ||
583 			    !strncasecmp(t, "utf-8", 5) ||
584 			    !strncasecmp(t, "ascii", 5)) {
585 				log_debug("unknown charset %s", t);
586 				return (-1);
587 			}
588 			continue;
589 		}
590 
591 		if (!strncmp(t, "lang=", 5)) {
592 			t += 5;
593 			if ((semi = strchr(t, ';')) != NULL)
594 				*semi = '\0';
595 
596 			if (strlcpy(lang, t, len) >= len) {
597 				log_debug("lang too long: %s", t);
598 				*lang = '\0';
599 			}
600 
601 			if (semi)
602 				*semi = ';';
603 			continue;
604 		}
605 	}
606 
607 	return (0);
608 }
609 
610 static int
611 proxy_tp_puts(struct template *tp, const char *c)
612 {
613 	struct client		*clt = tp->tp_arg;
614 
615 	if (clt_puts(clt, c) == -1) {
616 		tp->tp_ret = -1;
617 		return (-1);
618 	}
619 	return (0);
620 }
621 
622 static int
623 proxy_tp_putc(struct template *tp, int c)
624 {
625 	struct client		*clt = tp->tp_arg;
626 
627 	if (clt_putc(clt, c) == -1) {
628 		tp->tp_ret = -1;
629 		return (-1);
630 	}
631 	return (0);
632 }
633 
634 void
635 proxy_read(struct bufferevent *bev, void *d)
636 {
637 	struct client		*clt = d;
638 	struct proxy_config	*pc = clt->clt_pc;
639 	struct evbuffer		*src = EVBUFFER_INPUT(bev);
640 	const char		*ctype;
641 	char			 lang[16];
642 	char			*hdr, *mime;
643 	size_t			 len;
644 
645 	if (clt->clt_headersdone) {
646 		if (clt->clt_translate)
647 			proxy_translate_gemtext(clt);
648 		else
649 			clt_write_bufferevent(clt, bev);
650 		return;
651 	}
652 
653 	hdr = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF_STRICT);
654 	if (hdr == NULL) {
655 		if (EVBUFFER_LENGTH(src) >= 1026)
656 			proxy_error(bev, EV_READ, clt);
657 		return;
658 	}
659 
660 	if (len < 4 ||
661 	    !isdigit((unsigned char)hdr[0]) ||
662 	    !isdigit((unsigned char)hdr[1]) ||
663 	    hdr[2] != ' ') {
664 		log_warnx("invalid ");
665 		proxy_error(bev, EV_READ, clt);
666 		goto err;
667 	}
668 
669 	switch (hdr[0]) {
670 	case '2':
671 		/* handled below */
672 		break;
673 	default:
674 		if (clt_puts(clt, "Status: 501\r\n") == -1)
675 			goto err;
676 		if (clt_puts(clt,
677 		    "Content-Type: text/plain;charset=utf-8\r\n") == -1)
678 			goto err;
679 		if (clt_puts(clt, "\r\n") == -1)
680 			goto err;
681 		if (clt_printf(clt, "Request failed with code %c%c\n\n",
682 		    hdr[0], hdr[1]) == -1)
683 			goto err;
684 		if (clt_printf(clt, "The server says: %s\n", &hdr[3]) == -1)
685 			goto err;
686 		fcgi_end_request(clt, 1);
687 		goto err;
688 	}
689 
690 	mime = hdr + 2 + strspn(hdr + 2, " \t");
691 	if (parse_mime(clt, mime, lang, sizeof(lang)) == -1) {
692 		if (clt_puts(clt, "Status: 501\r\n") == -1)
693 			goto err;
694 		if (clt_puts(clt,
695 		    "Content-Type: text/plain;charset=utf-8\r\n") == -1)
696 			goto err;
697 		if (clt_puts(clt, "\r\n") == -1)
698 			goto err;
699 		if (clt_printf(clt, "Failed to parse the Gemini response\n")
700 		    == -1)
701 		fcgi_end_request(clt, 1);
702 		goto err;
703 	}
704 
705 	if (clt->clt_translate) {
706 		clt->clt_tp = template(clt, proxy_tp_puts, proxy_tp_putc);
707 
708 		ctype = "text/html;charset=utf-8";
709 	} else
710 		ctype = mime;
711 
712 	if (clt_printf(clt, "Content-Type: %s\r\n\r\n", ctype) == -1)
713 		goto err;
714 
715 	clt->clt_headersdone = 1;
716 
717 	if (clt->clt_translate &&
718 	    tp_head(clt->clt_tp, lang, pc->stylesheet) == -1)
719 		goto err;
720 
721 	/*
722 	 * Trigger the read again so we proceed with the response
723 	 * body, if any.
724 	 */
725 	free(hdr);
726 	proxy_read(bev, d);
727 	return;
728 
729 err:
730 	free(hdr);
731 }
732 
733 void
734 proxy_write(struct bufferevent *bev, void *d)
735 {
736 	return;
737 }
738 
739 void
740 proxy_error(struct bufferevent *bev, short err, void *d)
741 {
742 	struct client		*clt = d;
743 	int			 status = !(err & EVBUFFER_EOF);
744 
745 	log_debug("proxy error, shutting down the connection (err: %x)",
746 	    err);
747 
748 	if (!clt->clt_headersdone) {
749 		if (clt_printf(clt, "Status: 501\r\n") == -1)
750 			return;
751 		if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
752 			return;
753 		if (clt_printf(clt, "Proxy error\n") == -1)
754 			return;
755 	} else if (status == 0) {
756 		if (clt->clt_translate & TR_PRE) {
757 			if (clt_puts(clt, "</pre>"))
758 				return;
759 			clt->clt_translate &= ~TR_PRE;
760 		}
761 
762 		if (clt->clt_translate & TR_LIST) {
763 			if (clt_puts(clt, "</ul>") == -1)
764 				return;
765 			clt->clt_translate &= ~TR_LIST;
766 		}
767 
768 		if (clt->clt_translate & TR_NAV) {
769 			if (clt_puts(clt, "</ul></nav>") == -1)
770 				return;
771 			clt->clt_translate &= ~TR_NAV;
772 		}
773 
774 		if (tp_foot(clt->clt_tp) == -1)
775 			return;
776 	}
777 
778 	fcgi_end_request(clt, status);
779 }
780 
781 void
782 proxy_tls_readcb(int fd, short event, void *arg)
783 {
784 	struct bufferevent	*bufev = arg;
785 	struct client		*clt = bufev->cbarg;
786 	char			 rbuf[IBUF_READ_SIZE];
787 	int			 what = EVBUFFER_READ;
788 	int			 howmuch = IBUF_READ_SIZE;
789 	ssize_t			 ret;
790 	size_t			 len;
791 
792 	if (event == EV_TIMEOUT) {
793 		what |= EVBUFFER_TIMEOUT;
794 		goto err;
795 	}
796 
797 	if (bufev->wm_read.high != 0)
798 		howmuch = MINIMUM(sizeof(rbuf), bufev->wm_read.high);
799 
800 	ret = tls_read(clt->clt_ctx, rbuf, howmuch);
801 	if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
802 		goto retry;
803 	} else if (ret == -1) {
804 		what |= EVBUFFER_ERROR;
805 		goto err;
806 	}
807 	len = ret;
808 
809 	if (len == 0) {
810 		what |= EVBUFFER_EOF;
811 		goto err;
812 	}
813 
814 	if (evbuffer_add(bufev->input, rbuf, len) == -1) {
815 		what |= EVBUFFER_ERROR;
816 		goto err;
817 	}
818 
819 	proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
820 
821 	len = EVBUFFER_LENGTH(bufev->input);
822 	if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
823 		return;
824 	if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
825 		struct evbuffer *buf = bufev->input;
826 		event_del(&bufev->ev_read);
827 		evbuffer_setcb(buf, bufferevent_read_pressure_cb, bufev);
828 		return;
829 	}
830 
831 	if (bufev->readcb != NULL)
832 		(*bufev->readcb)(bufev, bufev->cbarg);
833 	return;
834 
835 retry:
836 	proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
837 	return;
838 
839 err:
840 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
841 }
842 
843 int
844 proxy_bufferevent_add(struct event *ev, int timeout)
845 {
846 	struct timeval tv, *ptv = NULL;
847 
848 	if (timeout) {
849 		timerclear(&tv);
850 		tv.tv_sec = timeout;
851 		ptv = &tv;
852 	}
853 
854 	return (event_add(ev, ptv));
855 }
856 
857 void
858 proxy_tls_writecb(int fd, short event, void *arg)
859 {
860 	struct bufferevent	*bufev = arg;
861 	struct client		*clt = bufev->cbarg;
862 	ssize_t			 ret;
863 	short			 what = EVBUFFER_WRITE;
864 	size_t			 len;
865 
866 	if (event == EV_TIMEOUT) {
867 		what |= EVBUFFER_TIMEOUT;
868 		goto err;
869 	}
870 
871 	if (EVBUFFER_LENGTH(bufev->output)) {
872 		ret = tls_write(clt->clt_ctx,
873 		    EVBUFFER_DATA(bufev->output),
874 		    EVBUFFER_LENGTH(bufev->output));
875 		if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
876 			goto retry;
877 		} else if (ret == -1) {
878 			what |= EVBUFFER_ERROR;
879 			goto err;
880 		}
881 		len = ret;
882 		evbuffer_drain(bufev->output, len);
883 	}
884 
885 	if (EVBUFFER_LENGTH(bufev->output) != 0)
886 		proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
887 
888 	if (bufev->writecb != NULL &&
889 	    EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
890 		(*bufev->writecb)(bufev, bufev->cbarg);
891 	return;
892 
893 retry:
894 	proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
895 	return;
896 
897 err:
898 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
899 }
900 
901 void
902 proxy_client_free(struct client *clt)
903 {
904 	if (clt->clt_evasr)
905 		event_asr_abort(clt->clt_evasr);
906 
907 	if (clt->clt_addrinfo)
908 		freeaddrinfo(clt->clt_addrinfo);
909 
910 	if (clt->clt_evconn_live)
911 		event_del(&clt->clt_evconn);
912 
913 	if (clt->clt_fd != -1)
914 		close(clt->clt_fd);
915 
916 	if (clt->clt_ctx)
917 		tls_free(clt->clt_ctx);
918 
919 	if (clt->clt_bev)
920 		bufferevent_free(clt->clt_bev);
921 
922 	free(clt->clt_tp);
923 	free(clt->clt_server_name);
924 	free(clt->clt_script_name);
925 	free(clt->clt_path_info);
926 	free(clt);
927 }