op public repos

Blob

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