op public repos

Blame

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