1 b0a6bcf7 2022-09-13 op /*
  2 b0a6bcf7 2022-09-13 op  * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
  3 b0a6bcf7 2022-09-13 op  * Copyright (c) 2006 - 2015 Reyk Floeter <reyk@openbsd.org>
  4 b0a6bcf7 2022-09-13 op  *
  5 b0a6bcf7 2022-09-13 op  * Permission to use, copy, modify, and distribute this software for any
  6 b0a6bcf7 2022-09-13 op  * purpose with or without fee is hereby granted, provided that the above
  7 b0a6bcf7 2022-09-13 op  * copyright notice and this permission notice appear in all copies.
  8 b0a6bcf7 2022-09-13 op  *
  9 b0a6bcf7 2022-09-13 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 10 b0a6bcf7 2022-09-13 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 11 b0a6bcf7 2022-09-13 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 12 b0a6bcf7 2022-09-13 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 13 b0a6bcf7 2022-09-13 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 14 b0a6bcf7 2022-09-13 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 15 b0a6bcf7 2022-09-13 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 16 b0a6bcf7 2022-09-13 op  */
 17 b0a6bcf7 2022-09-13 op 
 18 b0a6bcf7 2022-09-13 op #include <sys/types.h>
 19 b0a6bcf7 2022-09-13 op #include <sys/queue.h>
 20 b0a6bcf7 2022-09-13 op #include <sys/tree.h>
 21 b0a6bcf7 2022-09-13 op #include <sys/socket.h>
 22 b0a6bcf7 2022-09-13 op #include <sys/uio.h>
 23 b0a6bcf7 2022-09-13 op 
 24 b0a6bcf7 2022-09-13 op #include <netinet/in.h>
 25 b0a6bcf7 2022-09-13 op #include <arpa/inet.h>
 26 b0a6bcf7 2022-09-13 op #include <netdb.h>
 27 b0a6bcf7 2022-09-13 op 
 28 b0a6bcf7 2022-09-13 op #include <asr.h>
 29 b0a6bcf7 2022-09-13 op #include <ctype.h>
 30 b0a6bcf7 2022-09-13 op #include <errno.h>
 31 b0a6bcf7 2022-09-13 op #include <event.h>
 32 b0a6bcf7 2022-09-13 op #include <limits.h>
 33 b0a6bcf7 2022-09-13 op #include <stdlib.h>
 34 b0a6bcf7 2022-09-13 op #include <stdint.h>
 35 b0a6bcf7 2022-09-13 op #include <stdio.h>
 36 b0a6bcf7 2022-09-13 op #include <string.h>
 37 b0a6bcf7 2022-09-13 op #include <imsg.h>
 38 b0a6bcf7 2022-09-13 op #include <tls.h>
 39 b0a6bcf7 2022-09-13 op #include <unistd.h>
 40 b0a6bcf7 2022-09-13 op 
 41 b0a6bcf7 2022-09-13 op #include "log.h"
 42 b0a6bcf7 2022-09-13 op #include "proc.h"
 43 b0a6bcf7 2022-09-13 op 
 44 b0a6bcf7 2022-09-13 op #include "galileo.h"
 45 b0a6bcf7 2022-09-13 op 
 46 b0a6bcf7 2022-09-13 op #define MINIMUM(a, b)	((a) < (b) ? (a) : (b))
 47 b0a6bcf7 2022-09-13 op 
 48 b0a6bcf7 2022-09-13 op /* provided by OpenBSD' base libevent but not in any header? */
 49 b0a6bcf7 2022-09-13 op extern void	 bufferevent_read_pressure_cb(struct evbuffer *, size_t,
 50 b0a6bcf7 2022-09-13 op 		    size_t, void *);
 51 b0a6bcf7 2022-09-13 op 
 52 b0a6bcf7 2022-09-13 op void	proxy_init(struct privsep *, struct privsep_proc *, void *);
 53 b0a6bcf7 2022-09-13 op int	proxy_launch(struct galileo *);
 54 b0a6bcf7 2022-09-13 op void	proxy_inflight_dec(const char *);
 55 b0a6bcf7 2022-09-13 op int	proxy_dispatch_parent(int, struct privsep_proc *, struct imsg *);
 56 c9dc4bc6 2022-09-13 op void	proxy_translate_gemtext(struct client *);
 57 b0a6bcf7 2022-09-13 op void	proxy_resolved(struct asr_result *, void *);
 58 b0a6bcf7 2022-09-13 op void	proxy_connect(int, short, void *);
 59 b0a6bcf7 2022-09-13 op void	proxy_read(struct bufferevent *, void *);
 60 b0a6bcf7 2022-09-13 op void	proxy_write(struct bufferevent *, void *);
 61 b0a6bcf7 2022-09-13 op void	proxy_error(struct bufferevent *, short, void *);
 62 b0a6bcf7 2022-09-13 op int	proxy_bufferevent_add(struct event *, int);
 63 b0a6bcf7 2022-09-13 op void	proxy_tls_writecb(int, short, void *);
 64 b0a6bcf7 2022-09-13 op void	proxy_tls_readcb(int, short, void *);
 65 b0a6bcf7 2022-09-13 op 
 66 b0a6bcf7 2022-09-13 op static struct privsep_proc procs[] = {
 67 b0a6bcf7 2022-09-13 op 	{ "parent",	PROC_PARENT, proxy_dispatch_parent },
 68 b0a6bcf7 2022-09-13 op };
 69 b0a6bcf7 2022-09-13 op 
 70 b0a6bcf7 2022-09-13 op volatile int proxy_clients;
 71 b0a6bcf7 2022-09-13 op volatile int proxy_inflight;
 72 b0a6bcf7 2022-09-13 op uint32_t proxy_fcg_id;
 73 b0a6bcf7 2022-09-13 op 
 74 b0a6bcf7 2022-09-13 op void
 75 b0a6bcf7 2022-09-13 op proxy(struct privsep *ps, struct privsep_proc *p)
 76 b0a6bcf7 2022-09-13 op {
 77 b0a6bcf7 2022-09-13 op 	proc_run(ps, p, procs, nitems(procs), proxy_init, NULL);
 78 b0a6bcf7 2022-09-13 op }
 79 b0a6bcf7 2022-09-13 op 
 80 b0a6bcf7 2022-09-13 op void
 81 b0a6bcf7 2022-09-13 op proxy_init(struct privsep *ps, struct privsep_proc *p, void *arg)
 82 b0a6bcf7 2022-09-13 op {
 83 b0a6bcf7 2022-09-13 op 	if (config_init(ps->ps_env) == -1)
 84 b0a6bcf7 2022-09-13 op 		fatal("failed to initialize configuration");
 85 b0a6bcf7 2022-09-13 op 
 86 b0a6bcf7 2022-09-13 op 	/* We use a custom shutdown callback */
 87 b0a6bcf7 2022-09-13 op 	/* p->p_shutdown = proxy_shutdown */
 88 b0a6bcf7 2022-09-13 op 
 89 b0a6bcf7 2022-09-13 op 	if (pledge("stdio recvfd unix inet dns", NULL) == -1)
 90 b0a6bcf7 2022-09-13 op 		fatal("pledge");
 91 b0a6bcf7 2022-09-13 op }
 92 b0a6bcf7 2022-09-13 op 
 93 b0a6bcf7 2022-09-13 op int
 94 b0a6bcf7 2022-09-13 op proxy_launch(struct galileo *env)
 95 b0a6bcf7 2022-09-13 op {
 96 b0a6bcf7 2022-09-13 op 	event_add(&env->sc_evsock, NULL);
 97 b0a6bcf7 2022-09-13 op 	return (0);
 98 b0a6bcf7 2022-09-13 op }
 99 b0a6bcf7 2022-09-13 op 
100 b0a6bcf7 2022-09-13 op void
101 b0a6bcf7 2022-09-13 op proxy_purge(struct server *srv)
102 b0a6bcf7 2022-09-13 op {
103 b0a6bcf7 2022-09-13 op }
104 b0a6bcf7 2022-09-13 op 
105 b0a6bcf7 2022-09-13 op void
106 b0a6bcf7 2022-09-13 op proxy_inflight_dec(const char *why)
107 b0a6bcf7 2022-09-13 op {
108 b0a6bcf7 2022-09-13 op 	proxy_inflight--;
109 b0a6bcf7 2022-09-13 op 	log_debug("%s: inflight decremented, now %d, %s",
110 b0a6bcf7 2022-09-13 op 	    __func__, proxy_inflight, why);
111 b0a6bcf7 2022-09-13 op }
112 b0a6bcf7 2022-09-13 op 
113 b0a6bcf7 2022-09-13 op int
114 b0a6bcf7 2022-09-13 op proxy_dispatch_parent(int fd, struct privsep_proc *p, struct imsg *imsg)
115 b0a6bcf7 2022-09-13 op {
116 b0a6bcf7 2022-09-13 op 	struct privsep	*ps = p->p_ps;
117 b0a6bcf7 2022-09-13 op 	struct galileo	*env = ps->ps_env;
118 b0a6bcf7 2022-09-13 op 
119 b0a6bcf7 2022-09-13 op 	switch (imsg->hdr.type) {
120 b0a6bcf7 2022-09-13 op 	case IMSG_CFG_SRV:
121 b0a6bcf7 2022-09-13 op 		if (config_getserver(env, imsg) == -1)
122 b0a6bcf7 2022-09-13 op 			fatal("config_getproxy");
123 b0a6bcf7 2022-09-13 op 		break;
124 b0a6bcf7 2022-09-13 op 	case IMSG_CFG_SOCK:
125 b0a6bcf7 2022-09-13 op 		/* XXX: improve */
126 b0a6bcf7 2022-09-13 op 
127 b0a6bcf7 2022-09-13 op 		if (env->sc_sock_fd != -1) {
128 b0a6bcf7 2022-09-13 op 			event_del(&env->sc_evsock);
129 b0a6bcf7 2022-09-13 op 			close(env->sc_sock_fd);
130 b0a6bcf7 2022-09-13 op 		}
131 b0a6bcf7 2022-09-13 op 
132 b0a6bcf7 2022-09-13 op 		env->sc_sock_fd = config_getsock(env, imsg);
133 b0a6bcf7 2022-09-13 op 		if (env->sc_sock_fd == -1)
134 b0a6bcf7 2022-09-13 op 			fatal("config_getsock");
135 b0a6bcf7 2022-09-13 op 
136 b0a6bcf7 2022-09-13 op 		event_set(&env->sc_evsock, env->sc_sock_fd,
137 ea8874f1 2022-09-13 op 		    EV_READ | EV_PERSIST, fcgi_accept, env);
138 b0a6bcf7 2022-09-13 op 		event_add(&env->sc_evsock, NULL);
139 ea8874f1 2022-09-13 op 		evtimer_set(&env->sc_evpause, fcgi_accept, env);
140 b0a6bcf7 2022-09-13 op 		break;
141 b0a6bcf7 2022-09-13 op 	case IMSG_CFG_DONE:
142 b0a6bcf7 2022-09-13 op 		log_debug("config done!");
143 b0a6bcf7 2022-09-13 op 		break;
144 b0a6bcf7 2022-09-13 op 	case IMSG_CTL_START:
145 b0a6bcf7 2022-09-13 op 		proxy_launch(env);
146 b0a6bcf7 2022-09-13 op 		break;
147 b0a6bcf7 2022-09-13 op 	default:
148 b0a6bcf7 2022-09-13 op 		log_warnx("unknown message %d", imsg->hdr.type);
149 b0a6bcf7 2022-09-13 op 		return (-1);
150 c9dc4bc6 2022-09-13 op 	}
151 c9dc4bc6 2022-09-13 op 
152 c9dc4bc6 2022-09-13 op 	return (0);
153 c9dc4bc6 2022-09-13 op }
154 c9dc4bc6 2022-09-13 op 
155 c9dc4bc6 2022-09-13 op static inline int
156 c9dc4bc6 2022-09-13 op printurl(struct client *clt, const char *str)
157 c9dc4bc6 2022-09-13 op {
158 c9dc4bc6 2022-09-13 op 	for (; *str; ++str) {
159 c9dc4bc6 2022-09-13 op 		switch (*str) {
160 c9dc4bc6 2022-09-13 op 		case ' ':
161 c9dc4bc6 2022-09-13 op 		case '\t':
162 c9dc4bc6 2022-09-13 op 		case '\'':
163 c9dc4bc6 2022-09-13 op 		case '\\':
164 c9dc4bc6 2022-09-13 op 			if (clt_printf(clt, "%2X", (unsigned char)*str) == -1)
165 c9dc4bc6 2022-09-13 op 				return (-1);
166 c9dc4bc6 2022-09-13 op 			break;
167 c9dc4bc6 2022-09-13 op 		default:
168 c9dc4bc6 2022-09-13 op 			if (clt_putc(clt, *str) == -1)
169 c9dc4bc6 2022-09-13 op 				return (-1);
170 c9dc4bc6 2022-09-13 op 			break;
171 c9dc4bc6 2022-09-13 op 		}
172 c9dc4bc6 2022-09-13 op 	}
173 c9dc4bc6 2022-09-13 op 
174 c9dc4bc6 2022-09-13 op 	return (0);
175 c9dc4bc6 2022-09-13 op }
176 c9dc4bc6 2022-09-13 op 
177 c9dc4bc6 2022-09-13 op static inline int
178 c9dc4bc6 2022-09-13 op htmlescape(struct client *clt, const char *str)
179 c9dc4bc6 2022-09-13 op {
180 c9dc4bc6 2022-09-13 op 	int r;
181 c9dc4bc6 2022-09-13 op 
182 c9dc4bc6 2022-09-13 op 	for (; *str; ++str) {
183 c9dc4bc6 2022-09-13 op 		switch (*str) {
184 c9dc4bc6 2022-09-13 op 		case '<':
185 c9dc4bc6 2022-09-13 op 			r = clt_puts(clt, "&lt;");
186 c9dc4bc6 2022-09-13 op 			break;
187 c9dc4bc6 2022-09-13 op 		case '>':
188 c9dc4bc6 2022-09-13 op 			r = clt_puts(clt, "&gt;");
189 c9dc4bc6 2022-09-13 op 			break;
190 c9dc4bc6 2022-09-13 op 		case '&':
191 c9dc4bc6 2022-09-13 op 			r = clt_puts(clt, "&amp;");
192 c9dc4bc6 2022-09-13 op 			break;
193 c9dc4bc6 2022-09-13 op 		default:
194 c9dc4bc6 2022-09-13 op 			r = clt_putc(clt, *str);
195 c9dc4bc6 2022-09-13 op 			break;
196 c9dc4bc6 2022-09-13 op 		}
197 c9dc4bc6 2022-09-13 op 
198 c9dc4bc6 2022-09-13 op 		if (r == -1)
199 c9dc4bc6 2022-09-13 op 			return (-1);
200 c9dc4bc6 2022-09-13 op 	}
201 c9dc4bc6 2022-09-13 op 
202 c9dc4bc6 2022-09-13 op 	return (0);
203 c9dc4bc6 2022-09-13 op }
204 c9dc4bc6 2022-09-13 op 
205 c9dc4bc6 2022-09-13 op static int
206 c9dc4bc6 2022-09-13 op gemtext_translate_line(struct client *clt, char *line)
207 c9dc4bc6 2022-09-13 op {
208 c9dc4bc6 2022-09-13 op 	/* preformatted line / closing */
209 b9de0a74 2022-09-14 op 	if (clt->clt_translate & TR_PRE) {
210 c9dc4bc6 2022-09-13 op 		if (!strncmp(line, "```", 3)) {
211 b9de0a74 2022-09-14 op 			clt->clt_translate &= ~TR_PRE;
212 c9dc4bc6 2022-09-13 op 			return (clt_puts(clt, "</pre>"));
213 c9dc4bc6 2022-09-13 op 		}
214 c9dc4bc6 2022-09-13 op 
215 c9dc4bc6 2022-09-13 op 		if (htmlescape(clt, line) == -1)
216 c9dc4bc6 2022-09-13 op 			return (-1);
217 c9dc4bc6 2022-09-13 op 		return (clt_putc(clt, '\n'));
218 b9de0a74 2022-09-14 op 	}
219 b9de0a74 2022-09-14 op 
220 b9de0a74 2022-09-14 op 	/* bullet */
221 b9de0a74 2022-09-14 op 	if (!strncmp(line, "* ", 2)) {
222 b9de0a74 2022-09-14 op 		if (clt->clt_translate & TR_NAV) {
223 b9de0a74 2022-09-14 op 			if (clt_puts(clt, "</ul></nav>") == -1)
224 b9de0a74 2022-09-14 op 				return (-1);
225 b9de0a74 2022-09-14 op 			clt->clt_translate &= ~TR_NAV;
226 b9de0a74 2022-09-14 op 		}
227 b9de0a74 2022-09-14 op 
228 b9de0a74 2022-09-14 op 		if (!(clt->clt_translate & TR_LIST)) {
229 b9de0a74 2022-09-14 op 			if (clt_puts(clt, "<ul>") == -1)
230 b9de0a74 2022-09-14 op 				return (-1);
231 b9de0a74 2022-09-14 op 			clt->clt_translate |= TR_LIST;
232 b9de0a74 2022-09-14 op 		}
233 b9de0a74 2022-09-14 op 
234 b9de0a74 2022-09-14 op 		if (clt_puts(clt, "<li>") == -1 ||
235 b9de0a74 2022-09-14 op 		    htmlescape(clt, line + 2) == -1 ||
236 b9de0a74 2022-09-14 op 		    clt_puts(clt, "</li>") == -1)
237 b9de0a74 2022-09-14 op 			return (-1);
238 b9de0a74 2022-09-14 op 		return (0);
239 b9de0a74 2022-09-14 op 	}
240 b9de0a74 2022-09-14 op 
241 b9de0a74 2022-09-14 op 	if (clt->clt_translate & TR_LIST) {
242 b9de0a74 2022-09-14 op 		if (clt_puts(clt, "</ul>") == -1)
243 b9de0a74 2022-09-14 op 			return (-1);
244 b9de0a74 2022-09-14 op 		clt->clt_translate &= ~TR_LIST;
245 b9de0a74 2022-09-14 op 	}
246 b9de0a74 2022-09-14 op 
247 b9de0a74 2022-09-14 op 	/* link -- TODO: relativify from SCRIPT_NAME */
248 b9de0a74 2022-09-14 op 	if (!strncmp(line, "=>", 2)) {
249 b9de0a74 2022-09-14 op 		char *label;
250 b9de0a74 2022-09-14 op 
251 b9de0a74 2022-09-14 op 		if (!(clt->clt_translate & TR_NAV)) {
252 b9de0a74 2022-09-14 op 			if (clt_puts(clt, "<nav><ul>") == -1)
253 b9de0a74 2022-09-14 op 				return (-1);
254 b9de0a74 2022-09-14 op 			clt->clt_translate |= TR_NAV;
255 b9de0a74 2022-09-14 op 		}
256 b9de0a74 2022-09-14 op 
257 b9de0a74 2022-09-14 op 		line += 2;
258 b9de0a74 2022-09-14 op 		line += strspn(line, " \t");
259 b9de0a74 2022-09-14 op 
260 b9de0a74 2022-09-14 op 		label = line + strcspn(line, " \t");
261 b9de0a74 2022-09-14 op 		if (*label == '\0')
262 b9de0a74 2022-09-14 op 			label = line;
263 b9de0a74 2022-09-14 op 		else
264 b9de0a74 2022-09-14 op 			*label++ = '\0';
265 b9de0a74 2022-09-14 op 
266 b9de0a74 2022-09-14 op 		if (clt_puts(clt, "<li><a href='") == -1 ||
267 b9de0a74 2022-09-14 op 		    printurl(clt, line) == -1 ||
268 b9de0a74 2022-09-14 op 		    clt_puts(clt, "'>") == -1 ||
269 b9de0a74 2022-09-14 op 		    htmlescape(clt, label) == -1 ||
270 b9de0a74 2022-09-14 op 		    clt_puts(clt, "</a></li>") == -1)
271 b9de0a74 2022-09-14 op 			return (-1);
272 b9de0a74 2022-09-14 op 
273 b9de0a74 2022-09-14 op 		return (0);
274 c9dc4bc6 2022-09-13 op 	}
275 c9dc4bc6 2022-09-13 op 
276 b9de0a74 2022-09-14 op 	if (clt->clt_translate & TR_NAV) {
277 b9de0a74 2022-09-14 op 		if (clt_puts(clt, "</ul></nav>") == -1)
278 b9de0a74 2022-09-14 op 			return (-1);
279 b9de0a74 2022-09-14 op 		clt->clt_translate &= ~TR_NAV;
280 b9de0a74 2022-09-14 op 	}
281 b9de0a74 2022-09-14 op 
282 c9dc4bc6 2022-09-13 op 	/* pre opening */
283 c9dc4bc6 2022-09-13 op 	if (!strncmp(line, "```", 3)) {
284 b9de0a74 2022-09-14 op 		clt->clt_translate |= TR_PRE;
285 c9dc4bc6 2022-09-13 op 		return (clt_puts(clt, "<pre>"));
286 c9dc4bc6 2022-09-13 op 	}
287 c9dc4bc6 2022-09-13 op 
288 c9dc4bc6 2022-09-13 op 	/* citation block */
289 c9dc4bc6 2022-09-13 op 	if (*line == '>') {
290 c9dc4bc6 2022-09-13 op 		if (clt_puts(clt, "<blockquote>") == -1 ||
291 c9dc4bc6 2022-09-13 op 		    htmlescape(clt, line + 1) == -1 ||
292 c9dc4bc6 2022-09-13 op 		    clt_puts(clt, "</blockquote>") == -1)
293 c9dc4bc6 2022-09-13 op 			return (-1);
294 c9dc4bc6 2022-09-13 op 		return (0);
295 c9dc4bc6 2022-09-13 op 	}
296 c9dc4bc6 2022-09-13 op 
297 c9dc4bc6 2022-09-13 op 	/* headings */
298 c9dc4bc6 2022-09-13 op 	if (!strncmp(line, "###", 3)) {
299 c9dc4bc6 2022-09-13 op 		if (clt_puts(clt, "<h3>") == -1 ||
300 c9dc4bc6 2022-09-13 op 		    htmlescape(clt, line + 3) == -1 ||
301 c9dc4bc6 2022-09-13 op 		    clt_puts(clt, "</h3>") == -1)
302 c9dc4bc6 2022-09-13 op 			return (-1);
303 c9dc4bc6 2022-09-13 op 		return (0);
304 c9dc4bc6 2022-09-13 op 	}
305 c9dc4bc6 2022-09-13 op 	if (!strncmp(line, "##", 2)) {
306 c9dc4bc6 2022-09-13 op 		if (clt_puts(clt, "<h2>") == -1 ||
307 c9dc4bc6 2022-09-13 op 		    htmlescape(clt, line + 2) == -1 ||
308 c9dc4bc6 2022-09-13 op 		    clt_puts(clt, "</h2>") == -1)
309 c9dc4bc6 2022-09-13 op 			return (-1);
310 c9dc4bc6 2022-09-13 op 		return (0);
311 b0a6bcf7 2022-09-13 op 	}
312 c9dc4bc6 2022-09-13 op 	if (!strncmp(line, "#", 1)) {
313 c9dc4bc6 2022-09-13 op 		if (clt_puts(clt, "<h1>") == -1 ||
314 c9dc4bc6 2022-09-13 op 		    htmlescape(clt, line + 1) == -1 ||
315 c9dc4bc6 2022-09-13 op 		    clt_puts(clt, "</h1>") == -1)
316 c9dc4bc6 2022-09-13 op 			return (-1);
317 c9dc4bc6 2022-09-13 op 		return (0);
318 c9dc4bc6 2022-09-13 op 	}
319 df5279ca 2022-09-14 op 
320 df5279ca 2022-09-14 op 	/* Not following strictly the gemini specification... */
321 df5279ca 2022-09-14 op 	if (*line == '\0')
322 df5279ca 2022-09-14 op 		return (0);
323 b0a6bcf7 2022-09-13 op 
324 c9dc4bc6 2022-09-13 op 	/* paragraph */
325 c9dc4bc6 2022-09-13 op 	if (clt_puts(clt, "<p>") == -1 ||
326 c9dc4bc6 2022-09-13 op 	    htmlescape(clt, line) == -1 ||
327 c9dc4bc6 2022-09-13 op 	    clt_puts(clt, "</p>") == -1)
328 c9dc4bc6 2022-09-13 op 		return (-1);
329 b9de0a74 2022-09-14 op 
330 b0a6bcf7 2022-09-13 op 	return (0);
331 b0a6bcf7 2022-09-13 op }
332 b0a6bcf7 2022-09-13 op 
333 c9dc4bc6 2022-09-13 op void
334 c9dc4bc6 2022-09-13 op proxy_translate_gemtext(struct client *clt)
335 c9dc4bc6 2022-09-13 op {
336 c9dc4bc6 2022-09-13 op 	struct bufferevent	*bev = clt->clt_bev;
337 c9dc4bc6 2022-09-13 op 	struct evbuffer		*src = EVBUFFER_INPUT(bev);
338 c9dc4bc6 2022-09-13 op 	char			*line;
339 c9dc4bc6 2022-09-13 op 	size_t			 len;
340 c9dc4bc6 2022-09-13 op 	int			 r;
341 c9dc4bc6 2022-09-13 op 
342 c9dc4bc6 2022-09-13 op 	for (;;) {
343 b9de0a74 2022-09-14 op 		line = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF);
344 c9dc4bc6 2022-09-13 op 		if (line == NULL)
345 c9dc4bc6 2022-09-13 op 			return;
346 c9dc4bc6 2022-09-13 op 
347 c9dc4bc6 2022-09-13 op 		r = gemtext_translate_line(clt, line);
348 c9dc4bc6 2022-09-13 op 		free(line);
349 c9dc4bc6 2022-09-13 op 		if (r == -1)
350 c9dc4bc6 2022-09-13 op 			return;
351 c9dc4bc6 2022-09-13 op 	}
352 c9dc4bc6 2022-09-13 op }
353 c9dc4bc6 2022-09-13 op 
354 b0a6bcf7 2022-09-13 op static struct proxy_config *
355 b0a6bcf7 2022-09-13 op proxy_server_match(struct galileo *env, struct client *clt)
356 b0a6bcf7 2022-09-13 op {
357 b0a6bcf7 2022-09-13 op 	struct server		*srv;
358 b0a6bcf7 2022-09-13 op 
359 b0a6bcf7 2022-09-13 op 	if (clt->clt_server_name == NULL)
360 b0a6bcf7 2022-09-13 op 		return NULL;
361 b0a6bcf7 2022-09-13 op 
362 b0a6bcf7 2022-09-13 op 	TAILQ_FOREACH(srv, &env->sc_servers, srv_entry) {
363 b0a6bcf7 2022-09-13 op 		if (!strcmp(clt->clt_server_name, srv->srv_conf.host))
364 b0a6bcf7 2022-09-13 op 			return &srv->srv_conf;
365 b0a6bcf7 2022-09-13 op 	}
366 b0a6bcf7 2022-09-13 op 
367 b0a6bcf7 2022-09-13 op 	return NULL;
368 b0a6bcf7 2022-09-13 op }
369 b0a6bcf7 2022-09-13 op 
370 b0a6bcf7 2022-09-13 op void
371 b0a6bcf7 2022-09-13 op proxy_start_request(struct galileo *env, struct client *clt)
372 b0a6bcf7 2022-09-13 op {
373 b0a6bcf7 2022-09-13 op 	struct addrinfo		 hints;
374 b0a6bcf7 2022-09-13 op 	struct asr_query	*query;
375 b0a6bcf7 2022-09-13 op 	char			 port[32];
376 b0a6bcf7 2022-09-13 op 
377 b0a6bcf7 2022-09-13 op 	if ((clt->clt_pc = proxy_server_match(env, clt)) == NULL) {
378 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Status: 501\r\n\r\n") == -1)
379 b0a6bcf7 2022-09-13 op 			return;
380 b0a6bcf7 2022-09-13 op 		fcgi_end_request(clt, 1);
381 b0a6bcf7 2022-09-13 op 		return;
382 b0a6bcf7 2022-09-13 op 	}
383 b0a6bcf7 2022-09-13 op 
384 b0a6bcf7 2022-09-13 op 	(void)snprintf(port, sizeof(port), "%d", clt->clt_pc->proxy_port);
385 b0a6bcf7 2022-09-13 op 
386 b0a6bcf7 2022-09-13 op 	memset(&hints, 0, sizeof(hints));
387 b0a6bcf7 2022-09-13 op 	hints.ai_family = AF_UNSPEC;
388 b0a6bcf7 2022-09-13 op 	hints.ai_socktype = SOCK_STREAM;
389 b0a6bcf7 2022-09-13 op 
390 b0a6bcf7 2022-09-13 op 	query = getaddrinfo_async(clt->clt_pc->proxy_addr, port, &hints, NULL);
391 b0a6bcf7 2022-09-13 op 	if (query == NULL) {
392 b0a6bcf7 2022-09-13 op 		log_warn("getaddrinfo_async");
393 b0a6bcf7 2022-09-13 op 		fcgi_abort_request(clt);
394 b0a6bcf7 2022-09-13 op 		return;
395 b0a6bcf7 2022-09-13 op 	}
396 b0a6bcf7 2022-09-13 op 
397 b0a6bcf7 2022-09-13 op 	clt->clt_evasr = event_asr_run(query, proxy_resolved, clt);
398 b0a6bcf7 2022-09-13 op 	if (clt->clt_evasr == NULL) {
399 b0a6bcf7 2022-09-13 op 		log_warn("event_asr_run");
400 b0a6bcf7 2022-09-13 op 		asr_abort(query);
401 b0a6bcf7 2022-09-13 op 		fcgi_abort_request(clt);
402 b0a6bcf7 2022-09-13 op 		return;
403 b0a6bcf7 2022-09-13 op 	}
404 b0a6bcf7 2022-09-13 op }
405 b0a6bcf7 2022-09-13 op 
406 b0a6bcf7 2022-09-13 op void
407 b0a6bcf7 2022-09-13 op proxy_resolved(struct asr_result *res, void *d)
408 b0a6bcf7 2022-09-13 op {
409 b0a6bcf7 2022-09-13 op 	struct client		*clt = d;
410 b0a6bcf7 2022-09-13 op 	struct proxy_config	*pc = clt->clt_pc;
411 b0a6bcf7 2022-09-13 op 
412 b0a6bcf7 2022-09-13 op 	clt->clt_evasr = NULL;
413 b0a6bcf7 2022-09-13 op 
414 b0a6bcf7 2022-09-13 op 	if (res->ar_gai_errno != 0) {
415 b0a6bcf7 2022-09-13 op 		log_warnx("failed to resolve %s:%d: %s",
416 b0a6bcf7 2022-09-13 op 		    pc->proxy_addr, pc->proxy_port,
417 b0a6bcf7 2022-09-13 op 		    gai_strerror(res->ar_gai_errno));
418 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Status: 501\r\n") == -1)
419 b0a6bcf7 2022-09-13 op 			return;
420 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
421 b0a6bcf7 2022-09-13 op 			return;
422 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Proxy error; connection failed") == -1)
423 b0a6bcf7 2022-09-13 op 			return;
424 b0a6bcf7 2022-09-13 op 		fcgi_end_request(clt, 1);
425 b0a6bcf7 2022-09-13 op 		return;
426 b0a6bcf7 2022-09-13 op 	}
427 b0a6bcf7 2022-09-13 op 
428 b0a6bcf7 2022-09-13 op 	clt->clt_addrinfo = res->ar_addrinfo;
429 b0a6bcf7 2022-09-13 op 	clt->clt_p = clt->clt_addrinfo;
430 b0a6bcf7 2022-09-13 op 	proxy_connect(-1, 0, clt);
431 b0a6bcf7 2022-09-13 op }
432 b0a6bcf7 2022-09-13 op 
433 b0a6bcf7 2022-09-13 op void
434 b0a6bcf7 2022-09-13 op proxy_connect(int fd, short ev, void *d)
435 b0a6bcf7 2022-09-13 op {
436 b0a6bcf7 2022-09-13 op 	struct client		*clt = d;
437 b0a6bcf7 2022-09-13 op 	struct evbuffer		*out;
438 b0a6bcf7 2022-09-13 op 	struct addrinfo		*p;
439 b0a6bcf7 2022-09-13 op 	struct tls_config	*conf;
440 b0a6bcf7 2022-09-13 op 	struct timeval		 conntv = {5, 0};
441 b0a6bcf7 2022-09-13 op 	int			 err = 0;
442 b0a6bcf7 2022-09-13 op 	socklen_t		 len = sizeof(err);
443 b0a6bcf7 2022-09-13 op 
444 b0a6bcf7 2022-09-13 op again:
445 b0a6bcf7 2022-09-13 op 	if (clt->clt_p == NULL)
446 b0a6bcf7 2022-09-13 op 		goto err;
447 b0a6bcf7 2022-09-13 op 
448 b0a6bcf7 2022-09-13 op 	if (clt->clt_fd != -1) {
449 b0a6bcf7 2022-09-13 op 		if (getsockopt(clt->clt_fd, SOL_SOCKET, SO_ERROR, &err, &len)
450 b0a6bcf7 2022-09-13 op 		    == -1)
451 b0a6bcf7 2022-09-13 op 			goto err;
452 b0a6bcf7 2022-09-13 op 		if (err != 0) {
453 b0a6bcf7 2022-09-13 op 			errno = err;
454 b0a6bcf7 2022-09-13 op 			goto err;
455 b0a6bcf7 2022-09-13 op 		}
456 b0a6bcf7 2022-09-13 op 		goto done;
457 b0a6bcf7 2022-09-13 op 	}
458 b0a6bcf7 2022-09-13 op 
459 b0a6bcf7 2022-09-13 op 	p = clt->clt_p;
460 b0a6bcf7 2022-09-13 op 	clt->clt_fd = socket(p->ai_family, p->ai_socktype | SOCK_NONBLOCK,
461 b0a6bcf7 2022-09-13 op 	    p->ai_protocol);
462 b0a6bcf7 2022-09-13 op 	if (clt->clt_fd == -1) {
463 b0a6bcf7 2022-09-13 op 		clt->clt_p = clt->clt_p->ai_next;
464 b0a6bcf7 2022-09-13 op 		goto again;
465 b0a6bcf7 2022-09-13 op 	}
466 b0a6bcf7 2022-09-13 op 
467 b0a6bcf7 2022-09-13 op 	if (connect(clt->clt_fd, p->ai_addr, p->ai_addrlen) == 0)
468 b0a6bcf7 2022-09-13 op 		goto done;
469 b0a6bcf7 2022-09-13 op 
470 b0a6bcf7 2022-09-13 op 	clt->clt_evconn_live = 1;
471 b0a6bcf7 2022-09-13 op 	event_set(&clt->clt_evconn, clt->clt_fd, EV_WRITE, proxy_connect, clt);
472 b0a6bcf7 2022-09-13 op 	event_add(&clt->clt_evconn, &conntv);
473 b0a6bcf7 2022-09-13 op 	return;
474 b0a6bcf7 2022-09-13 op 
475 b0a6bcf7 2022-09-13 op done:
476 b0a6bcf7 2022-09-13 op 	clt->clt_evconn_live = 0;
477 b0a6bcf7 2022-09-13 op 	freeaddrinfo(clt->clt_addrinfo);
478 b0a6bcf7 2022-09-13 op 	clt->clt_addrinfo = clt->clt_p = NULL;
479 b0a6bcf7 2022-09-13 op 
480 b0a6bcf7 2022-09-13 op 	/* initialize TLS for Gemini */
481 b0a6bcf7 2022-09-13 op 	if ((conf = tls_config_new()) == NULL) {
482 b0a6bcf7 2022-09-13 op 		log_warn("tls_config_new failed");
483 b0a6bcf7 2022-09-13 op 		goto err;
484 b0a6bcf7 2022-09-13 op 	}
485 b0a6bcf7 2022-09-13 op 
486 b0a6bcf7 2022-09-13 op 	tls_config_insecure_noverifycert(conf);
487 b0a6bcf7 2022-09-13 op 
488 b0a6bcf7 2022-09-13 op 	if ((clt->clt_ctx = tls_client()) == NULL) {
489 b0a6bcf7 2022-09-13 op 		log_warnx("tls_client failed");
490 b0a6bcf7 2022-09-13 op 		tls_config_free(conf);
491 b0a6bcf7 2022-09-13 op 		goto err;
492 b0a6bcf7 2022-09-13 op 	}
493 b0a6bcf7 2022-09-13 op 
494 b0a6bcf7 2022-09-13 op 	if (tls_configure(clt->clt_ctx, conf) == -1) {
495 b0a6bcf7 2022-09-13 op 		log_warnx("tls_configure failed");
496 b0a6bcf7 2022-09-13 op 		tls_config_free(conf);
497 b0a6bcf7 2022-09-13 op 		goto err;
498 b0a6bcf7 2022-09-13 op 	}
499 b0a6bcf7 2022-09-13 op 
500 b0a6bcf7 2022-09-13 op 	tls_config_free(conf);
501 b0a6bcf7 2022-09-13 op 
502 b0a6bcf7 2022-09-13 op 	if (tls_connect_socket(clt->clt_ctx, clt->clt_fd,
503 b0a6bcf7 2022-09-13 op 	    clt->clt_pc->proxy_name) == -1) {
504 b0a6bcf7 2022-09-13 op 		log_warnx("tls_connect_socket failed");
505 b0a6bcf7 2022-09-13 op 		goto err;
506 b0a6bcf7 2022-09-13 op 	}
507 b0a6bcf7 2022-09-13 op 
508 b0a6bcf7 2022-09-13 op 	clt->clt_bev = bufferevent_new(clt->clt_fd, proxy_read, proxy_write,
509 b0a6bcf7 2022-09-13 op 	    proxy_error, clt);
510 b0a6bcf7 2022-09-13 op 	if (clt->clt_bev == NULL) {
511 b0a6bcf7 2022-09-13 op 		log_warn("bufferevent_new");
512 b0a6bcf7 2022-09-13 op 		goto err;
513 b0a6bcf7 2022-09-13 op 	}
514 b0a6bcf7 2022-09-13 op 	out = EVBUFFER_OUTPUT(clt->clt_bev);
515 b0a6bcf7 2022-09-13 op 
516 b0a6bcf7 2022-09-13 op 	event_set(&clt->clt_bev->ev_read, clt->clt_fd, EV_READ,
517 b0a6bcf7 2022-09-13 op 	    proxy_tls_readcb, clt->clt_bev);
518 b0a6bcf7 2022-09-13 op 	event_set(&clt->clt_bev->ev_write, clt->clt_fd, EV_WRITE,
519 b0a6bcf7 2022-09-13 op 	    proxy_tls_writecb, clt->clt_bev);
520 b0a6bcf7 2022-09-13 op 
521 b0a6bcf7 2022-09-13 op 	/* bufferevent_settimeout(); */
522 b0a6bcf7 2022-09-13 op 	bufferevent_enable(clt->clt_bev, EV_READ|EV_WRITE);
523 b0a6bcf7 2022-09-13 op 
524 b0a6bcf7 2022-09-13 op 	/* TODO: compute the URL */
525 ee80262d 2022-09-13 op 	if (evbuffer_add_printf(out, "gemini://%s/%s\r\n",
526 ee80262d 2022-09-13 op 	    clt->clt_pc->proxy_name, clt->clt_path_info) == -1) {
527 b0a6bcf7 2022-09-13 op 		log_warn("bufferevent_printf failed");
528 b0a6bcf7 2022-09-13 op 		goto err;
529 b0a6bcf7 2022-09-13 op 	}
530 b0a6bcf7 2022-09-13 op 
531 b0a6bcf7 2022-09-13 op 	return;
532 b0a6bcf7 2022-09-13 op 
533 b0a6bcf7 2022-09-13 op err:
534 b0a6bcf7 2022-09-13 op 	log_warn("failed to connect to %s:%d",
535 b0a6bcf7 2022-09-13 op 	    clt->clt_pc->proxy_addr, clt->clt_pc->proxy_port);
536 b0a6bcf7 2022-09-13 op 	if (clt_printf(clt, "Status: 501\r\n") == -1)
537 b0a6bcf7 2022-09-13 op 		return;
538 b0a6bcf7 2022-09-13 op 	if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
539 b0a6bcf7 2022-09-13 op 		return;
540 b0a6bcf7 2022-09-13 op 	if (clt_printf(clt, "Proxy error; connection failed") == -1)
541 b0a6bcf7 2022-09-13 op 		return;
542 b0a6bcf7 2022-09-13 op 	fcgi_end_request(clt, 1);
543 b0a6bcf7 2022-09-13 op }
544 b0a6bcf7 2022-09-13 op 
545 b0a6bcf7 2022-09-13 op void
546 b0a6bcf7 2022-09-13 op proxy_read(struct bufferevent *bev, void *d)
547 b0a6bcf7 2022-09-13 op {
548 b0a6bcf7 2022-09-13 op 	struct client		*clt = d;
549 fc1cb9ea 2022-09-14 op 	struct proxy_config	*pc = clt->clt_pc;
550 b0a6bcf7 2022-09-13 op 	struct evbuffer		*src = EVBUFFER_INPUT(bev);
551 c9dc4bc6 2022-09-13 op 	const char		*ctype;
552 b0a6bcf7 2022-09-13 op 	char			*hdr;
553 b0a6bcf7 2022-09-13 op 	size_t			 len;
554 b0a6bcf7 2022-09-13 op 	int			 code;
555 b0a6bcf7 2022-09-13 op 
556 b0a6bcf7 2022-09-13 op 	if (clt->clt_headersdone) {
557 b0a6bcf7 2022-09-13 op 	copy:
558 c9dc4bc6 2022-09-13 op 		if (clt->clt_translate)
559 c9dc4bc6 2022-09-13 op 			proxy_translate_gemtext(clt);
560 c9dc4bc6 2022-09-13 op 		else
561 c9dc4bc6 2022-09-13 op 			clt_write_bufferevent(clt, bev);
562 b0a6bcf7 2022-09-13 op 		return;
563 b0a6bcf7 2022-09-13 op 	}
564 b0a6bcf7 2022-09-13 op 
565 b0a6bcf7 2022-09-13 op 	hdr = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF_STRICT);
566 b0a6bcf7 2022-09-13 op 	if (hdr == NULL) {
567 b0a6bcf7 2022-09-13 op 		if (EVBUFFER_LENGTH(src) >= 1026)
568 b0a6bcf7 2022-09-13 op 			proxy_error(bev, EV_READ, clt);
569 b0a6bcf7 2022-09-13 op 		return;
570 b0a6bcf7 2022-09-13 op 	}
571 b0a6bcf7 2022-09-13 op 
572 b0a6bcf7 2022-09-13 op 	if (len < 4 ||
573 b0a6bcf7 2022-09-13 op 	    !isdigit((unsigned char)hdr[0]) ||
574 b0a6bcf7 2022-09-13 op 	    !isdigit((unsigned char)hdr[1]) ||
575 b0a6bcf7 2022-09-13 op 	    hdr[2] != ' ') {
576 b0a6bcf7 2022-09-13 op 		log_warnx("invalid ");
577 b0a6bcf7 2022-09-13 op 		free(hdr);
578 b0a6bcf7 2022-09-13 op 		proxy_error(bev, EV_READ, clt);
579 b0a6bcf7 2022-09-13 op 		return;
580 b0a6bcf7 2022-09-13 op 	}
581 b0a6bcf7 2022-09-13 op 
582 b0a6bcf7 2022-09-13 op 	code = (hdr[0] - '0') * 10 + (hdr[1] - '0');
583 b0a6bcf7 2022-09-13 op 	if (code != 20) {
584 b0a6bcf7 2022-09-13 op 		log_warnx("un-handled gemini reply status %d", code);
585 b0a6bcf7 2022-09-13 op 		free(hdr);
586 b0a6bcf7 2022-09-13 op 		proxy_error(bev, EV_READ, clt);
587 b0a6bcf7 2022-09-13 op 		return;
588 b0a6bcf7 2022-09-13 op 	}
589 b0a6bcf7 2022-09-13 op 
590 c9dc4bc6 2022-09-13 op 	if (!strncmp(&hdr[3], "text/gemini", 11)) {
591 c9dc4bc6 2022-09-13 op 		ctype = "text/html; charset=utf8";
592 b9de0a74 2022-09-14 op 		clt->clt_translate = TR_ENABLED;
593 c9dc4bc6 2022-09-13 op 	} else
594 c9dc4bc6 2022-09-13 op 		ctype = &hdr[3];
595 c9dc4bc6 2022-09-13 op 
596 c9dc4bc6 2022-09-13 op 	if (clt_printf(clt, "Content-Type: %s\r\n", ctype) == -1)
597 b0a6bcf7 2022-09-13 op 		return;
598 b0a6bcf7 2022-09-13 op 	if (clt_printf(clt, "\r\n") == -1)
599 b0a6bcf7 2022-09-13 op 		return;
600 b0a6bcf7 2022-09-13 op 
601 b0a6bcf7 2022-09-13 op 	clt->clt_headersdone = 1;
602 c9dc4bc6 2022-09-13 op 
603 c9dc4bc6 2022-09-13 op 	if (clt->clt_translate) {
604 fc1cb9ea 2022-09-14 op 		if (clt_puts(clt, "<!doctype html><html><head>") == -1)
605 c9dc4bc6 2022-09-13 op 			return;
606 fc1cb9ea 2022-09-14 op 		if (*pc->stylesheet != '\0' &&
607 fc1cb9ea 2022-09-14 op 		    clt_printf(clt, "<link rel='stylesheet' href='%s' />",
608 fc1cb9ea 2022-09-14 op 		    pc->stylesheet) == -1)
609 fc1cb9ea 2022-09-14 op 			return;
610 fc1cb9ea 2022-09-14 op 		if (clt_puts(clt, "</head><body>") == -1)
611 fc1cb9ea 2022-09-14 op 			return;
612 c9dc4bc6 2022-09-13 op 	}
613 c9dc4bc6 2022-09-13 op 
614 b0a6bcf7 2022-09-13 op 	goto copy;
615 b0a6bcf7 2022-09-13 op }
616 b0a6bcf7 2022-09-13 op 
617 b0a6bcf7 2022-09-13 op void
618 b0a6bcf7 2022-09-13 op proxy_write(struct bufferevent *bev, void *d)
619 b0a6bcf7 2022-09-13 op {
620 b0a6bcf7 2022-09-13 op 	return;
621 b0a6bcf7 2022-09-13 op }
622 b0a6bcf7 2022-09-13 op 
623 b0a6bcf7 2022-09-13 op void
624 b0a6bcf7 2022-09-13 op proxy_error(struct bufferevent *bev, short err, void *d)
625 b0a6bcf7 2022-09-13 op {
626 b0a6bcf7 2022-09-13 op 	struct client		*clt = d;
627 b0a6bcf7 2022-09-13 op 	int			 status = !(err & EVBUFFER_EOF);
628 b0a6bcf7 2022-09-13 op 
629 b0a6bcf7 2022-09-13 op 	log_debug("proxy error, shutting down the connection (err: %x)",
630 b0a6bcf7 2022-09-13 op 	    err);
631 b0a6bcf7 2022-09-13 op 
632 b0a6bcf7 2022-09-13 op 	if (!clt->clt_headersdone) {
633 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Status: 501\r\n") == -1)
634 b0a6bcf7 2022-09-13 op 			return;
635 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
636 b0a6bcf7 2022-09-13 op 			return;
637 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Proxy error\n") == -1)
638 b0a6bcf7 2022-09-13 op 			return;
639 c9dc4bc6 2022-09-13 op 	} else if (status == 0) {
640 c9dc4bc6 2022-09-13 op 		if (clt_puts(clt, "</body></html>") == -1)
641 c9dc4bc6 2022-09-13 op 			return;
642 b0a6bcf7 2022-09-13 op 	}
643 b0a6bcf7 2022-09-13 op 
644 b0a6bcf7 2022-09-13 op 	fcgi_end_request(clt, status);
645 b0a6bcf7 2022-09-13 op }
646 b0a6bcf7 2022-09-13 op 
647 b0a6bcf7 2022-09-13 op void
648 b0a6bcf7 2022-09-13 op proxy_tls_readcb(int fd, short event, void *arg)
649 b0a6bcf7 2022-09-13 op {
650 b0a6bcf7 2022-09-13 op 	struct bufferevent	*bufev = arg;
651 b0a6bcf7 2022-09-13 op 	struct client		*clt = bufev->cbarg;
652 b0a6bcf7 2022-09-13 op 	char			 rbuf[IBUF_READ_SIZE];
653 b0a6bcf7 2022-09-13 op 	int			 what = EVBUFFER_READ;
654 b0a6bcf7 2022-09-13 op 	int			 howmuch = IBUF_READ_SIZE;
655 b0a6bcf7 2022-09-13 op 	ssize_t			 ret;
656 b0a6bcf7 2022-09-13 op 	size_t			 len;
657 b0a6bcf7 2022-09-13 op 
658 b0a6bcf7 2022-09-13 op 	if (event == EV_TIMEOUT) {
659 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_TIMEOUT;
660 b0a6bcf7 2022-09-13 op 		goto err;
661 b0a6bcf7 2022-09-13 op 	}
662 b0a6bcf7 2022-09-13 op 
663 b0a6bcf7 2022-09-13 op 	if (bufev->wm_read.high != 0)
664 b0a6bcf7 2022-09-13 op 		howmuch = MINIMUM(sizeof(rbuf), bufev->wm_read.high);
665 b0a6bcf7 2022-09-13 op 
666 b0a6bcf7 2022-09-13 op 	ret = tls_read(clt->clt_ctx, rbuf, howmuch);
667 b0a6bcf7 2022-09-13 op 	if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
668 b0a6bcf7 2022-09-13 op 		goto retry;
669 b0a6bcf7 2022-09-13 op 	} else if (ret == -1) {
670 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_ERROR;
671 b0a6bcf7 2022-09-13 op 		goto err;
672 b0a6bcf7 2022-09-13 op 	}
673 b0a6bcf7 2022-09-13 op 	len = ret;
674 b0a6bcf7 2022-09-13 op 
675 b0a6bcf7 2022-09-13 op 	if (len == 0) {
676 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_EOF;
677 b0a6bcf7 2022-09-13 op 		goto err;
678 b0a6bcf7 2022-09-13 op 	}
679 b0a6bcf7 2022-09-13 op 
680 b0a6bcf7 2022-09-13 op 	if (evbuffer_add(bufev->input, rbuf, len) == -1) {
681 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_ERROR;
682 b0a6bcf7 2022-09-13 op 		goto err;
683 b0a6bcf7 2022-09-13 op 	}
684 b0a6bcf7 2022-09-13 op 
685 b0a6bcf7 2022-09-13 op 	proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
686 b0a6bcf7 2022-09-13 op 
687 b0a6bcf7 2022-09-13 op 	len = EVBUFFER_LENGTH(bufev->input);
688 b0a6bcf7 2022-09-13 op 	if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
689 b0a6bcf7 2022-09-13 op 		return;
690 b0a6bcf7 2022-09-13 op 	if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
691 b0a6bcf7 2022-09-13 op 		struct evbuffer *buf = bufev->input;
692 b0a6bcf7 2022-09-13 op 		event_del(&bufev->ev_read);
693 b0a6bcf7 2022-09-13 op 		evbuffer_setcb(buf, bufferevent_read_pressure_cb, bufev);
694 b0a6bcf7 2022-09-13 op 		return;
695 b0a6bcf7 2022-09-13 op 	}
696 b0a6bcf7 2022-09-13 op 
697 b0a6bcf7 2022-09-13 op 	if (bufev->readcb != NULL)
698 b0a6bcf7 2022-09-13 op 		(*bufev->readcb)(bufev, bufev->cbarg);
699 b0a6bcf7 2022-09-13 op 	return;
700 b0a6bcf7 2022-09-13 op 
701 b0a6bcf7 2022-09-13 op retry:
702 b0a6bcf7 2022-09-13 op 	proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
703 b0a6bcf7 2022-09-13 op 	return;
704 b0a6bcf7 2022-09-13 op 
705 b0a6bcf7 2022-09-13 op err:
706 b0a6bcf7 2022-09-13 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
707 b0a6bcf7 2022-09-13 op }
708 b0a6bcf7 2022-09-13 op 
709 b0a6bcf7 2022-09-13 op int
710 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(struct event *ev, int timeout)
711 b0a6bcf7 2022-09-13 op {
712 b0a6bcf7 2022-09-13 op 	struct timeval tv, *ptv = NULL;
713 b0a6bcf7 2022-09-13 op 
714 b0a6bcf7 2022-09-13 op 	if (timeout) {
715 b0a6bcf7 2022-09-13 op 		timerclear(&tv);
716 b0a6bcf7 2022-09-13 op 		tv.tv_sec = timeout;
717 b0a6bcf7 2022-09-13 op 		ptv = &tv;
718 b0a6bcf7 2022-09-13 op 	}
719 b0a6bcf7 2022-09-13 op 
720 b0a6bcf7 2022-09-13 op 	return (event_add(ev, ptv));
721 b0a6bcf7 2022-09-13 op }
722 b0a6bcf7 2022-09-13 op 
723 b0a6bcf7 2022-09-13 op void
724 b0a6bcf7 2022-09-13 op proxy_tls_writecb(int fd, short event, void *arg)
725 b0a6bcf7 2022-09-13 op {
726 b0a6bcf7 2022-09-13 op 	struct bufferevent	*bufev = arg;
727 b0a6bcf7 2022-09-13 op 	struct client		*clt = bufev->cbarg;
728 b0a6bcf7 2022-09-13 op 	ssize_t			 ret;
729 b0a6bcf7 2022-09-13 op 	short			 what = EVBUFFER_WRITE;
730 b0a6bcf7 2022-09-13 op 	size_t			 len;
731 b0a6bcf7 2022-09-13 op 
732 b0a6bcf7 2022-09-13 op 	if (event == EV_TIMEOUT) {
733 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_TIMEOUT;
734 b0a6bcf7 2022-09-13 op 		goto err;
735 b0a6bcf7 2022-09-13 op 	}
736 b0a6bcf7 2022-09-13 op 
737 b0a6bcf7 2022-09-13 op 	if (EVBUFFER_LENGTH(bufev->output)) {
738 b0a6bcf7 2022-09-13 op 		ret = tls_write(clt->clt_ctx,
739 b0a6bcf7 2022-09-13 op 		    EVBUFFER_DATA(bufev->output),
740 b0a6bcf7 2022-09-13 op 		    EVBUFFER_LENGTH(bufev->output));
741 b0a6bcf7 2022-09-13 op 		if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
742 b0a6bcf7 2022-09-13 op 			goto retry;
743 b0a6bcf7 2022-09-13 op 		} else if (ret == -1) {
744 b0a6bcf7 2022-09-13 op 			what |= EVBUFFER_ERROR;
745 b0a6bcf7 2022-09-13 op 			goto err;
746 b0a6bcf7 2022-09-13 op 		}
747 b0a6bcf7 2022-09-13 op 		len = ret;
748 b0a6bcf7 2022-09-13 op 		evbuffer_drain(bufev->output, len);
749 b0a6bcf7 2022-09-13 op 	}
750 b0a6bcf7 2022-09-13 op 
751 b0a6bcf7 2022-09-13 op 	if (EVBUFFER_LENGTH(bufev->output) != 0)
752 b0a6bcf7 2022-09-13 op 		proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
753 b0a6bcf7 2022-09-13 op 
754 b0a6bcf7 2022-09-13 op 	if (bufev->writecb != NULL &&
755 b0a6bcf7 2022-09-13 op 	    EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
756 b0a6bcf7 2022-09-13 op 		(*bufev->writecb)(bufev, bufev->cbarg);
757 b0a6bcf7 2022-09-13 op 	return;
758 b0a6bcf7 2022-09-13 op 
759 b0a6bcf7 2022-09-13 op retry:
760 b0a6bcf7 2022-09-13 op 	proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
761 b0a6bcf7 2022-09-13 op 	return;
762 b0a6bcf7 2022-09-13 op 
763 b0a6bcf7 2022-09-13 op err:
764 b0a6bcf7 2022-09-13 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
765 b0a6bcf7 2022-09-13 op }
766 b0a6bcf7 2022-09-13 op 
767 b0a6bcf7 2022-09-13 op void
768 b0a6bcf7 2022-09-13 op proxy_client_free(struct client *clt)
769 b0a6bcf7 2022-09-13 op {
770 b0a6bcf7 2022-09-13 op 	if (clt->clt_evasr)
771 b0a6bcf7 2022-09-13 op 		event_asr_abort(clt->clt_evasr);
772 b0a6bcf7 2022-09-13 op 
773 b0a6bcf7 2022-09-13 op 	if (clt->clt_addrinfo)
774 b0a6bcf7 2022-09-13 op 		freeaddrinfo(clt->clt_addrinfo);
775 b0a6bcf7 2022-09-13 op 
776 b0a6bcf7 2022-09-13 op 	if (clt->clt_evconn_live)
777 b0a6bcf7 2022-09-13 op 		event_del(&clt->clt_evconn);
778 b0a6bcf7 2022-09-13 op 
779 b0a6bcf7 2022-09-13 op 	if (clt->clt_fd != -1)
780 b0a6bcf7 2022-09-13 op 		close(clt->clt_fd);
781 b0a6bcf7 2022-09-13 op 
782 b0a6bcf7 2022-09-13 op 	if (clt->clt_ctx)
783 b0a6bcf7 2022-09-13 op 		tls_free(clt->clt_ctx);
784 b0a6bcf7 2022-09-13 op 
785 b0a6bcf7 2022-09-13 op 	if (clt->clt_bev)
786 b0a6bcf7 2022-09-13 op 		bufferevent_free(clt->clt_bev);
787 b0a6bcf7 2022-09-13 op 
788 b0a6bcf7 2022-09-13 op 	free(clt->clt_server_name);
789 b0a6bcf7 2022-09-13 op 	free(clt->clt_script_name);
790 b0a6bcf7 2022-09-13 op 	free(clt->clt_path_info);
791 b0a6bcf7 2022-09-13 op 	free(clt);
792 b0a6bcf7 2022-09-13 op }