op public repos

Blob

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