Blob


1 /*
2 * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
3 * Copyright (c) 2006 - 2015 Reyk Floeter <reyk@openbsd.org>
4 *
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 */
18 #include <sys/types.h>
19 #include <sys/queue.h>
20 #include <sys/tree.h>
21 #include <sys/socket.h>
22 #include <sys/uio.h>
24 #include <netinet/in.h>
25 #include <arpa/inet.h>
26 #include <netdb.h>
28 #include <asr.h>
29 #include <ctype.h>
30 #include <errno.h>
31 #include <event.h>
32 #include <fnmatch.h>
33 #include <limits.h>
34 #include <stdlib.h>
35 #include <stdint.h>
36 #include <stdio.h>
37 #include <string.h>
38 #include <imsg.h>
39 #include <tls.h>
40 #include <unistd.h>
42 #include "log.h"
43 #include "proc.h"
44 #include "tmpl.h"
46 #include "galileo.h"
48 #define MINIMUM(a, b) ((a) < (b) ? (a) : (b))
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 *);
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 *);
68 static struct privsep_proc procs[] = {
69 { "parent", PROC_PARENT, proxy_dispatch_parent },
70 };
72 volatile int proxy_clients;
73 volatile int proxy_inflight;
74 uint32_t proxy_fcg_id;
76 void
77 proxy(struct privsep *ps, struct privsep_proc *p)
78 {
79 proc_run(ps, p, procs, nitems(procs), proxy_init, NULL);
80 }
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");
88 /* We use a custom shutdown callback */
89 /* p->p_shutdown = proxy_shutdown */
91 if (pledge("stdio recvfd unix inet dns", NULL) == -1)
92 fatal("pledge");
93 }
95 int
96 proxy_launch(struct galileo *env)
97 {
98 event_add(&env->sc_evsock, NULL);
99 return (0);
102 void
103 proxy_purge(struct server *srv)
107 void
108 proxy_inflight_dec(const char *why)
110 proxy_inflight--;
111 log_debug("%s: inflight decremented, now %d, %s",
112 __func__, proxy_inflight, why);
115 int
116 proxy_dispatch_parent(int fd, struct privsep_proc *p, struct imsg *imsg)
118 struct privsep *ps = p->p_ps;
119 struct galileo *env = ps->ps_env;
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 */
129 if (env->sc_sock_fd != -1) {
130 event_del(&env->sc_evsock);
131 close(env->sc_sock_fd);
134 env->sc_sock_fd = config_getsock(env, imsg);
135 if (env->sc_sock_fd == -1)
136 fatal("config_getsock");
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);
154 return (0);
157 static int
158 gemtext_translate_line(struct client *clt, char *line)
160 /* preformatted line / closing */
161 if (clt->clt_translate & TR_PRE) {
162 if (!strncmp(line, "```", 3)) {
163 clt->clt_translate &= ~TR_PRE;
164 return (clt_puts(clt, "</pre>"));
167 if (tp_htmlescape(clt->clt_tp, line) == -1)
168 return (-1);
169 return (clt_putc(clt, '\n'));
172 /* bullet */
173 if (!strncmp(line, "* ", 2)) {
174 if (clt->clt_translate & TR_NAV) {
175 if (clt_puts(clt, "</ul></nav>") == -1)
176 return (-1);
177 clt->clt_translate &= ~TR_NAV;
180 if (!(clt->clt_translate & TR_LIST)) {
181 if (clt_puts(clt, "<ul>") == -1)
182 return (-1);
183 clt->clt_translate |= TR_LIST;
186 if (clt_puts(clt, "<li>") == -1 ||
187 tp_htmlescape(clt->clt_tp, line + 2) == -1 ||
188 clt_puts(clt, "</li>") == -1)
189 return (-1);
190 return (0);
193 if (clt->clt_translate & TR_LIST) {
194 if (clt_puts(clt, "</ul>") == -1)
195 return (-1);
196 clt->clt_translate &= ~TR_LIST;
199 /* link -- TODO: relativify from SCRIPT_NAME */
200 if (!strncmp(line, "=>", 2)) {
201 char *label;
203 line += 2;
204 line += strspn(line, " \t");
206 label = line + strcspn(line, " \t");
207 if (*label == '\0')
208 label = line;
209 else
210 *label++ = '\0';
212 if (fnmatch("*.jpg", line, 0) == 0 ||
213 fnmatch("*.jpeg", line, 0) == 0 ||
214 fnmatch("*.gif", line, 0) == 0 ||
215 fnmatch("*.png", line, 0) == 0 ||
216 fnmatch("*.svg", line, 0) == 0 ||
217 fnmatch("*.webp", line, 0) == 0) {
218 if (clt->clt_translate & TR_NAV) {
219 if (clt_puts(clt, "</ul></nav>") == -1)
220 return (-1);
221 clt->clt_translate &= ~TR_NAV;
224 if (tp_figure(clt->clt_tp, line, label) == -1)
225 return (-1);
227 return (0);
230 if (!(clt->clt_translate & TR_NAV)) {
231 if (clt_puts(clt, "<nav><ul>") == -1)
232 return (-1);
233 clt->clt_translate |= TR_NAV;
236 if (clt_puts(clt, "<li><a href='") == -1)
237 return (-1);
239 /* XXX: do proper parsing */
240 if (*line == '/' || strstr(line, "//") == NULL) {
241 if (tp_urlescape(clt->clt_tp,
242 clt->clt_script_name) == -1)
243 return (-1);
245 /* skip the first / */
246 line++;
249 if (tp_urlescape(clt->clt_tp, line) == -1 ||
250 clt_puts(clt, "'>") == -1 ||
251 tp_htmlescape(clt->clt_tp, label) == -1 ||
252 clt_puts(clt, "</a></li>") == -1)
253 return (-1);
255 return (0);
258 if (clt->clt_translate & TR_NAV) {
259 if (clt_puts(clt, "</ul></nav>") == -1)
260 return (-1);
261 clt->clt_translate &= ~TR_NAV;
264 /* pre opening */
265 if (!strncmp(line, "```", 3)) {
266 clt->clt_translate |= TR_PRE;
267 return (clt_puts(clt, "<pre>"));
270 /* citation block */
271 if (*line == '>') {
272 if (clt_puts(clt, "<blockquote>") == -1 ||
273 tp_htmlescape(clt->clt_tp, line + 1) == -1 ||
274 clt_puts(clt, "</blockquote>") == -1)
275 return (-1);
276 return (0);
279 /* headings */
280 if (!strncmp(line, "###", 3)) {
281 if (clt_puts(clt, "<h3>") == -1 ||
282 tp_htmlescape(clt->clt_tp, line + 3) == -1 ||
283 clt_puts(clt, "</h3>") == -1)
284 return (-1);
285 return (0);
287 if (!strncmp(line, "##", 2)) {
288 if (clt_puts(clt, "<h2>") == -1 ||
289 tp_htmlescape(clt->clt_tp, line + 2) == -1 ||
290 clt_puts(clt, "</h2>") == -1)
291 return (-1);
292 return (0);
294 if (!strncmp(line, "#", 1)) {
295 if (clt_puts(clt, "<h1>") == -1 ||
296 tp_htmlescape(clt->clt_tp, line + 1) == -1 ||
297 clt_puts(clt, "</h1>") == -1)
298 return (-1);
299 return (0);
302 /* Not following strictly the gemini specification... */
303 if (*line == '\0')
304 return (0);
306 /* paragraph */
307 if (clt_puts(clt, "<p>") == -1 ||
308 tp_htmlescape(clt->clt_tp, line) == -1 ||
309 clt_puts(clt, "</p>") == -1)
310 return (-1);
312 return (0);
315 void
316 proxy_translate_gemtext(struct client *clt)
318 struct bufferevent *bev = clt->clt_bev;
319 struct evbuffer *src = EVBUFFER_INPUT(bev);
320 char *line;
321 size_t len;
322 int r;
324 for (;;) {
325 line = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF);
326 if (line == NULL)
327 return;
329 r = gemtext_translate_line(clt, line);
330 free(line);
331 if (r == -1)
332 return;
336 static struct proxy_config *
337 proxy_server_match(struct galileo *env, struct client *clt)
339 struct server *srv;
341 if (clt->clt_server_name == NULL)
342 return NULL;
344 TAILQ_FOREACH(srv, &env->sc_servers, srv_entry) {
345 if (!strcmp(clt->clt_server_name, srv->srv_conf.host))
346 return &srv->srv_conf;
349 return NULL;
352 void
353 proxy_start_request(struct galileo *env, struct client *clt)
355 struct addrinfo hints;
356 struct asr_query *query;
357 char port[32];
359 if ((clt->clt_pc = proxy_server_match(env, clt)) == NULL) {
360 if (clt_printf(clt, "Status: 501\r\n\r\n") == -1)
361 return;
362 fcgi_end_request(clt, 1);
363 return;
366 (void)snprintf(port, sizeof(port), "%d", clt->clt_pc->proxy_port);
368 memset(&hints, 0, sizeof(hints));
369 hints.ai_family = AF_UNSPEC;
370 hints.ai_socktype = SOCK_STREAM;
372 query = getaddrinfo_async(clt->clt_pc->proxy_addr, port, &hints, NULL);
373 if (query == NULL) {
374 log_warn("getaddrinfo_async");
375 fcgi_abort_request(clt);
376 return;
379 clt->clt_evasr = event_asr_run(query, proxy_resolved, clt);
380 if (clt->clt_evasr == NULL) {
381 log_warn("event_asr_run");
382 asr_abort(query);
383 fcgi_abort_request(clt);
384 return;
388 void
389 proxy_resolved(struct asr_result *res, void *d)
391 struct client *clt = d;
392 struct proxy_config *pc = clt->clt_pc;
394 clt->clt_evasr = NULL;
396 if (res->ar_gai_errno != 0) {
397 log_warnx("failed to resolve %s:%d: %s",
398 pc->proxy_addr, pc->proxy_port,
399 gai_strerror(res->ar_gai_errno));
400 if (clt_printf(clt, "Status: 501\r\n") == -1)
401 return;
402 if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
403 return;
404 if (clt_printf(clt, "Proxy error; connection failed") == -1)
405 return;
406 fcgi_end_request(clt, 1);
407 return;
410 clt->clt_addrinfo = res->ar_addrinfo;
411 clt->clt_p = clt->clt_addrinfo;
412 proxy_connect(-1, 0, clt);
415 void
416 proxy_connect(int fd, short ev, void *d)
418 struct client *clt = d;
419 struct evbuffer *out;
420 struct addrinfo *p;
421 struct tls_config *conf;
422 struct timeval conntv = {5, 0};
423 int err = 0;
424 socklen_t len = sizeof(err);
426 again:
427 if (clt->clt_p == NULL)
428 goto err;
430 if (clt->clt_fd != -1) {
431 if (getsockopt(clt->clt_fd, SOL_SOCKET, SO_ERROR, &err, &len)
432 == -1)
433 goto err;
434 if (err != 0) {
435 errno = err;
436 goto err;
438 goto done;
441 p = clt->clt_p;
442 clt->clt_fd = socket(p->ai_family, p->ai_socktype | SOCK_NONBLOCK,
443 p->ai_protocol);
444 if (clt->clt_fd == -1) {
445 clt->clt_p = clt->clt_p->ai_next;
446 goto again;
449 if (connect(clt->clt_fd, p->ai_addr, p->ai_addrlen) == 0)
450 goto done;
452 clt->clt_evconn_live = 1;
453 event_set(&clt->clt_evconn, clt->clt_fd, EV_WRITE, proxy_connect, clt);
454 event_add(&clt->clt_evconn, &conntv);
455 return;
457 done:
458 clt->clt_evconn_live = 0;
459 freeaddrinfo(clt->clt_addrinfo);
460 clt->clt_addrinfo = clt->clt_p = NULL;
462 /* initialize TLS for Gemini */
463 if ((conf = tls_config_new()) == NULL) {
464 log_warn("tls_config_new failed");
465 goto err;
468 tls_config_insecure_noverifycert(conf);
470 if ((clt->clt_ctx = tls_client()) == NULL) {
471 log_warnx("tls_client failed");
472 tls_config_free(conf);
473 goto err;
476 if (tls_configure(clt->clt_ctx, conf) == -1) {
477 log_warnx("tls_configure failed");
478 tls_config_free(conf);
479 goto err;
482 tls_config_free(conf);
484 if (tls_connect_socket(clt->clt_ctx, clt->clt_fd,
485 clt->clt_pc->proxy_name) == -1) {
486 log_warnx("tls_connect_socket failed");
487 goto err;
490 clt->clt_bev = bufferevent_new(clt->clt_fd, proxy_read, proxy_write,
491 proxy_error, clt);
492 if (clt->clt_bev == NULL) {
493 log_warn("bufferevent_new");
494 goto err;
496 out = EVBUFFER_OUTPUT(clt->clt_bev);
498 event_set(&clt->clt_bev->ev_read, clt->clt_fd, EV_READ,
499 proxy_tls_readcb, clt->clt_bev);
500 event_set(&clt->clt_bev->ev_write, clt->clt_fd, EV_WRITE,
501 proxy_tls_writecb, clt->clt_bev);
503 /* bufferevent_settimeout(); */
504 bufferevent_enable(clt->clt_bev, EV_READ|EV_WRITE);
506 /* TODO: compute the URL */
507 if (evbuffer_add_printf(out, "gemini://%s/%s\r\n",
508 clt->clt_pc->proxy_name, clt->clt_path_info) == -1) {
509 log_warn("bufferevent_printf failed");
510 goto err;
513 return;
515 err:
516 log_warn("failed to connect to %s:%d",
517 clt->clt_pc->proxy_addr, clt->clt_pc->proxy_port);
518 if (clt_printf(clt, "Status: 501\r\n") == -1)
519 return;
520 if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
521 return;
522 if (clt_printf(clt, "Proxy error; connection failed") == -1)
523 return;
524 fcgi_end_request(clt, 1);
527 static inline int
528 parse_mime(struct client *clt, char *mime, char *lang, size_t len)
530 char *t, *semi;
532 if (strncmp(mime, "text/gemini", 11) != 0)
533 return (0);
535 clt->clt_translate = TR_ENABLED;
537 if ((mime = strchr(mime, ';')) == NULL)
538 return (0);
540 *mime++ = '\0';
541 while ((t = strsep(&mime, ";")) != NULL) {
542 if (!strncmp(t, "charset=", 8)) {
543 t += 8;
544 if (!strncasecmp(t, "utf8", 4) ||
545 !strncasecmp(t, "utf-8", 5) ||
546 !strncasecmp(t, "ascii", 5)) {
547 log_debug("unknown charset %s", t);
548 return (-1);
550 continue;
553 if (!strncmp(t, "lang=", 5)) {
554 t += 5;
555 if ((semi = strchr(t, ';')) != NULL)
556 *semi = '\0';
558 if (strlcpy(lang, t, len) >= len) {
559 log_debug("lang too long: %s", t);
560 *lang = '\0';
563 if (semi)
564 *semi = ';';
565 continue;
569 return (0);
572 static int
573 proxy_tp_puts(struct template *tp, const char *c)
575 struct client *clt = tp->tp_arg;
577 if (clt_puts(clt, c) == -1) {
578 tp->tp_ret = -1;
579 return (-1);
581 return (0);
584 static int
585 proxy_tp_putc(struct template *tp, int c)
587 struct client *clt = tp->tp_arg;
589 if (clt_putc(clt, c) == -1) {
590 tp->tp_ret = -1;
591 return (-1);
593 return (0);
596 void
597 proxy_read(struct bufferevent *bev, void *d)
599 struct client *clt = d;
600 struct proxy_config *pc = clt->clt_pc;
601 struct evbuffer *src = EVBUFFER_INPUT(bev);
602 const char *ctype;
603 char lang[16];
604 char *hdr, *mime;
605 size_t len;
607 if (clt->clt_headersdone) {
608 if (clt->clt_translate)
609 proxy_translate_gemtext(clt);
610 else
611 clt_write_bufferevent(clt, bev);
612 return;
615 hdr = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF_STRICT);
616 if (hdr == NULL) {
617 if (EVBUFFER_LENGTH(src) >= 1026)
618 proxy_error(bev, EV_READ, clt);
619 return;
622 if (len < 4 ||
623 !isdigit((unsigned char)hdr[0]) ||
624 !isdigit((unsigned char)hdr[1]) ||
625 hdr[2] != ' ') {
626 log_warnx("invalid ");
627 proxy_error(bev, EV_READ, clt);
628 goto err;
631 switch (hdr[0]) {
632 case '2':
633 /* handled below */
634 break;
635 default:
636 if (clt_puts(clt, "Status: 501\r\n") == -1)
637 goto err;
638 if (clt_puts(clt,
639 "Content-Type: text/plain;charset=utf-8\r\n") == -1)
640 goto err;
641 if (clt_puts(clt, "\r\n") == -1)
642 goto err;
643 if (clt_printf(clt, "Request failed with code %c%c\n\n",
644 hdr[0], hdr[1]) == -1)
645 goto err;
646 if (clt_printf(clt, "The server says: %s\n", &hdr[3]) == -1)
647 goto err;
648 fcgi_end_request(clt, 1);
649 goto err;
652 mime = hdr + 2 + strspn(hdr + 2, " \t");
653 if (parse_mime(clt, mime, lang, sizeof(lang)) == -1) {
654 if (clt_puts(clt, "Status: 501\r\n") == -1)
655 goto err;
656 if (clt_puts(clt,
657 "Content-Type: text/plain;charset=utf-8\r\n") == -1)
658 goto err;
659 if (clt_puts(clt, "\r\n") == -1)
660 goto err;
661 if (clt_printf(clt, "Failed to parse the Gemini response\n")
662 == -1)
663 fcgi_end_request(clt, 1);
664 goto err;
667 if (clt->clt_translate) {
668 clt->clt_tp = template(clt, proxy_tp_puts, proxy_tp_putc);
670 ctype = "text/html;charset=utf-8";
671 } else
672 ctype = mime;
674 if (clt_printf(clt, "Content-Type: %s\r\n\r\n", ctype) == -1)
675 goto err;
677 clt->clt_headersdone = 1;
679 if (clt->clt_translate &&
680 tp_head(clt->clt_tp, lang, pc->stylesheet) == -1)
681 goto err;
683 /*
684 * Trigger the read again so we proceed with the response
685 * body, if any.
686 */
687 free(hdr);
688 proxy_read(bev, d);
689 return;
691 err:
692 free(hdr);
695 void
696 proxy_write(struct bufferevent *bev, void *d)
698 return;
701 void
702 proxy_error(struct bufferevent *bev, short err, void *d)
704 struct client *clt = d;
705 int status = !(err & EVBUFFER_EOF);
707 log_debug("proxy error, shutting down the connection (err: %x)",
708 err);
710 if (!clt->clt_headersdone) {
711 if (clt_printf(clt, "Status: 501\r\n") == -1)
712 return;
713 if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
714 return;
715 if (clt_printf(clt, "Proxy error\n") == -1)
716 return;
717 } else if (status == 0) {
718 if (clt->clt_translate & TR_PRE) {
719 if (clt_puts(clt, "</pre>"))
720 return;
721 clt->clt_translate &= ~TR_PRE;
724 if (clt->clt_translate & TR_LIST) {
725 if (clt_puts(clt, "</ul>") == -1)
726 return;
727 clt->clt_translate &= ~TR_LIST;
730 if (clt->clt_translate & TR_NAV) {
731 if (clt_puts(clt, "</ul></nav>") == -1)
732 return;
733 clt->clt_translate &= ~TR_NAV;
736 if (tp_foot(clt->clt_tp) == -1)
737 return;
740 fcgi_end_request(clt, status);
743 void
744 proxy_tls_readcb(int fd, short event, void *arg)
746 struct bufferevent *bufev = arg;
747 struct client *clt = bufev->cbarg;
748 char rbuf[IBUF_READ_SIZE];
749 int what = EVBUFFER_READ;
750 int howmuch = IBUF_READ_SIZE;
751 ssize_t ret;
752 size_t len;
754 if (event == EV_TIMEOUT) {
755 what |= EVBUFFER_TIMEOUT;
756 goto err;
759 if (bufev->wm_read.high != 0)
760 howmuch = MINIMUM(sizeof(rbuf), bufev->wm_read.high);
762 ret = tls_read(clt->clt_ctx, rbuf, howmuch);
763 if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
764 goto retry;
765 } else if (ret == -1) {
766 what |= EVBUFFER_ERROR;
767 goto err;
769 len = ret;
771 if (len == 0) {
772 what |= EVBUFFER_EOF;
773 goto err;
776 if (evbuffer_add(bufev->input, rbuf, len) == -1) {
777 what |= EVBUFFER_ERROR;
778 goto err;
781 proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
783 len = EVBUFFER_LENGTH(bufev->input);
784 if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
785 return;
786 if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
787 struct evbuffer *buf = bufev->input;
788 event_del(&bufev->ev_read);
789 evbuffer_setcb(buf, bufferevent_read_pressure_cb, bufev);
790 return;
793 if (bufev->readcb != NULL)
794 (*bufev->readcb)(bufev, bufev->cbarg);
795 return;
797 retry:
798 proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
799 return;
801 err:
802 (*bufev->errorcb)(bufev, what, bufev->cbarg);
805 int
806 proxy_bufferevent_add(struct event *ev, int timeout)
808 struct timeval tv, *ptv = NULL;
810 if (timeout) {
811 timerclear(&tv);
812 tv.tv_sec = timeout;
813 ptv = &tv;
816 return (event_add(ev, ptv));
819 void
820 proxy_tls_writecb(int fd, short event, void *arg)
822 struct bufferevent *bufev = arg;
823 struct client *clt = bufev->cbarg;
824 ssize_t ret;
825 short what = EVBUFFER_WRITE;
826 size_t len;
828 if (event == EV_TIMEOUT) {
829 what |= EVBUFFER_TIMEOUT;
830 goto err;
833 if (EVBUFFER_LENGTH(bufev->output)) {
834 ret = tls_write(clt->clt_ctx,
835 EVBUFFER_DATA(bufev->output),
836 EVBUFFER_LENGTH(bufev->output));
837 if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
838 goto retry;
839 } else if (ret == -1) {
840 what |= EVBUFFER_ERROR;
841 goto err;
843 len = ret;
844 evbuffer_drain(bufev->output, len);
847 if (EVBUFFER_LENGTH(bufev->output) != 0)
848 proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
850 if (bufev->writecb != NULL &&
851 EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
852 (*bufev->writecb)(bufev, bufev->cbarg);
853 return;
855 retry:
856 proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
857 return;
859 err:
860 (*bufev->errorcb)(bufev, what, bufev->cbarg);
863 void
864 proxy_client_free(struct client *clt)
866 if (clt->clt_evasr)
867 event_asr_abort(clt->clt_evasr);
869 if (clt->clt_addrinfo)
870 freeaddrinfo(clt->clt_addrinfo);
872 if (clt->clt_evconn_live)
873 event_del(&clt->clt_evconn);
875 if (clt->clt_fd != -1)
876 close(clt->clt_fd);
878 if (clt->clt_ctx)
879 tls_free(clt->clt_ctx);
881 if (clt->clt_bev)
882 bufferevent_free(clt->clt_bev);
884 free(clt->clt_tp);
885 free(clt->clt_server_name);
886 free(clt->clt_script_name);
887 free(clt->clt_path_info);
888 free(clt);