op public repos

Blob

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