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 b0a6bcf7 2022-09-13 op void	proxy_resolved(struct asr_result *, void *);
 57 b0a6bcf7 2022-09-13 op void	proxy_connect(int, short, void *);
 58 b0a6bcf7 2022-09-13 op void	proxy_read(struct bufferevent *, void *);
 59 b0a6bcf7 2022-09-13 op void	proxy_write(struct bufferevent *, void *);
 60 b0a6bcf7 2022-09-13 op void	proxy_error(struct bufferevent *, short, void *);
 61 b0a6bcf7 2022-09-13 op int	proxy_bufferevent_add(struct event *, int);
 62 b0a6bcf7 2022-09-13 op void	proxy_tls_writecb(int, short, void *);
 63 b0a6bcf7 2022-09-13 op void	proxy_tls_readcb(int, short, void *);
 64 b0a6bcf7 2022-09-13 op 
 65 b0a6bcf7 2022-09-13 op static struct privsep_proc procs[] = {
 66 b0a6bcf7 2022-09-13 op 	{ "parent",	PROC_PARENT, proxy_dispatch_parent },
 67 b0a6bcf7 2022-09-13 op };
 68 b0a6bcf7 2022-09-13 op 
 69 b0a6bcf7 2022-09-13 op volatile int proxy_clients;
 70 b0a6bcf7 2022-09-13 op volatile int proxy_inflight;
 71 b0a6bcf7 2022-09-13 op uint32_t proxy_fcg_id;
 72 b0a6bcf7 2022-09-13 op 
 73 b0a6bcf7 2022-09-13 op void
 74 b0a6bcf7 2022-09-13 op proxy(struct privsep *ps, struct privsep_proc *p)
 75 b0a6bcf7 2022-09-13 op {
 76 b0a6bcf7 2022-09-13 op 	proc_run(ps, p, procs, nitems(procs), proxy_init, NULL);
 77 b0a6bcf7 2022-09-13 op }
 78 b0a6bcf7 2022-09-13 op 
 79 b0a6bcf7 2022-09-13 op void
 80 b0a6bcf7 2022-09-13 op proxy_init(struct privsep *ps, struct privsep_proc *p, void *arg)
 81 b0a6bcf7 2022-09-13 op {
 82 b0a6bcf7 2022-09-13 op 	if (config_init(ps->ps_env) == -1)
 83 b0a6bcf7 2022-09-13 op 		fatal("failed to initialize configuration");
 84 b0a6bcf7 2022-09-13 op 
 85 b0a6bcf7 2022-09-13 op 	/* We use a custom shutdown callback */
 86 b0a6bcf7 2022-09-13 op 	/* p->p_shutdown = proxy_shutdown */
 87 b0a6bcf7 2022-09-13 op 
 88 b0a6bcf7 2022-09-13 op 	if (pledge("stdio recvfd unix inet dns", NULL) == -1)
 89 b0a6bcf7 2022-09-13 op 		fatal("pledge");
 90 b0a6bcf7 2022-09-13 op }
 91 b0a6bcf7 2022-09-13 op 
 92 b0a6bcf7 2022-09-13 op int
 93 b0a6bcf7 2022-09-13 op proxy_launch(struct galileo *env)
 94 b0a6bcf7 2022-09-13 op {
 95 b0a6bcf7 2022-09-13 op 	event_add(&env->sc_evsock, NULL);
 96 b0a6bcf7 2022-09-13 op 	return (0);
 97 b0a6bcf7 2022-09-13 op }
 98 b0a6bcf7 2022-09-13 op 
 99 b0a6bcf7 2022-09-13 op void
100 b0a6bcf7 2022-09-13 op proxy_purge(struct server *srv)
101 b0a6bcf7 2022-09-13 op {
102 b0a6bcf7 2022-09-13 op }
103 b0a6bcf7 2022-09-13 op 
104 b0a6bcf7 2022-09-13 op void
105 b0a6bcf7 2022-09-13 op proxy_inflight_dec(const char *why)
106 b0a6bcf7 2022-09-13 op {
107 b0a6bcf7 2022-09-13 op 	proxy_inflight--;
108 b0a6bcf7 2022-09-13 op 	log_debug("%s: inflight decremented, now %d, %s",
109 b0a6bcf7 2022-09-13 op 	    __func__, proxy_inflight, why);
110 b0a6bcf7 2022-09-13 op }
111 b0a6bcf7 2022-09-13 op 
112 b0a6bcf7 2022-09-13 op int
113 b0a6bcf7 2022-09-13 op proxy_dispatch_parent(int fd, struct privsep_proc *p, struct imsg *imsg)
114 b0a6bcf7 2022-09-13 op {
115 b0a6bcf7 2022-09-13 op 	struct privsep	*ps = p->p_ps;
116 b0a6bcf7 2022-09-13 op 	struct galileo	*env = ps->ps_env;
117 b0a6bcf7 2022-09-13 op 
118 b0a6bcf7 2022-09-13 op 	switch (imsg->hdr.type) {
119 b0a6bcf7 2022-09-13 op 	case IMSG_CFG_SRV:
120 b0a6bcf7 2022-09-13 op 		if (config_getserver(env, imsg) == -1)
121 b0a6bcf7 2022-09-13 op 			fatal("config_getproxy");
122 b0a6bcf7 2022-09-13 op 		break;
123 b0a6bcf7 2022-09-13 op 	case IMSG_CFG_SOCK:
124 b0a6bcf7 2022-09-13 op 		/* XXX: improve */
125 b0a6bcf7 2022-09-13 op 
126 b0a6bcf7 2022-09-13 op 		if (env->sc_sock_fd != -1) {
127 b0a6bcf7 2022-09-13 op 			event_del(&env->sc_evsock);
128 b0a6bcf7 2022-09-13 op 			close(env->sc_sock_fd);
129 b0a6bcf7 2022-09-13 op 		}
130 b0a6bcf7 2022-09-13 op 
131 b0a6bcf7 2022-09-13 op 		env->sc_sock_fd = config_getsock(env, imsg);
132 b0a6bcf7 2022-09-13 op 		if (env->sc_sock_fd == -1)
133 b0a6bcf7 2022-09-13 op 			fatal("config_getsock");
134 b0a6bcf7 2022-09-13 op 
135 b0a6bcf7 2022-09-13 op 		event_set(&env->sc_evsock, env->sc_sock_fd,
136 ea8874f1 2022-09-13 op 		    EV_READ | EV_PERSIST, fcgi_accept, env);
137 b0a6bcf7 2022-09-13 op 		event_add(&env->sc_evsock, NULL);
138 ea8874f1 2022-09-13 op 		evtimer_set(&env->sc_evpause, fcgi_accept, env);
139 b0a6bcf7 2022-09-13 op 		break;
140 b0a6bcf7 2022-09-13 op 	case IMSG_CFG_DONE:
141 b0a6bcf7 2022-09-13 op 		log_debug("config done!");
142 b0a6bcf7 2022-09-13 op 		break;
143 b0a6bcf7 2022-09-13 op 	case IMSG_CTL_START:
144 b0a6bcf7 2022-09-13 op 		proxy_launch(env);
145 b0a6bcf7 2022-09-13 op 		break;
146 b0a6bcf7 2022-09-13 op 	default:
147 b0a6bcf7 2022-09-13 op 		log_warnx("unknown message %d", imsg->hdr.type);
148 b0a6bcf7 2022-09-13 op 		return (-1);
149 b0a6bcf7 2022-09-13 op 	}
150 b0a6bcf7 2022-09-13 op 
151 b0a6bcf7 2022-09-13 op 	return (0);
152 b0a6bcf7 2022-09-13 op }
153 b0a6bcf7 2022-09-13 op 
154 b0a6bcf7 2022-09-13 op static struct proxy_config *
155 b0a6bcf7 2022-09-13 op proxy_server_match(struct galileo *env, struct client *clt)
156 b0a6bcf7 2022-09-13 op {
157 b0a6bcf7 2022-09-13 op 	struct server		*srv;
158 b0a6bcf7 2022-09-13 op 
159 b0a6bcf7 2022-09-13 op 	if (clt->clt_server_name == NULL)
160 b0a6bcf7 2022-09-13 op 		return NULL;
161 b0a6bcf7 2022-09-13 op 
162 b0a6bcf7 2022-09-13 op 	TAILQ_FOREACH(srv, &env->sc_servers, srv_entry) {
163 b0a6bcf7 2022-09-13 op 		if (!strcmp(clt->clt_server_name, srv->srv_conf.host))
164 b0a6bcf7 2022-09-13 op 			return &srv->srv_conf;
165 b0a6bcf7 2022-09-13 op 	}
166 b0a6bcf7 2022-09-13 op 
167 b0a6bcf7 2022-09-13 op 	return NULL;
168 b0a6bcf7 2022-09-13 op }
169 b0a6bcf7 2022-09-13 op 
170 b0a6bcf7 2022-09-13 op void
171 b0a6bcf7 2022-09-13 op proxy_start_request(struct galileo *env, struct client *clt)
172 b0a6bcf7 2022-09-13 op {
173 b0a6bcf7 2022-09-13 op 	struct addrinfo		 hints;
174 b0a6bcf7 2022-09-13 op 	struct asr_query	*query;
175 b0a6bcf7 2022-09-13 op 	char			 port[32];
176 b0a6bcf7 2022-09-13 op 
177 b0a6bcf7 2022-09-13 op 	if ((clt->clt_pc = proxy_server_match(env, clt)) == NULL) {
178 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Status: 501\r\n\r\n") == -1)
179 b0a6bcf7 2022-09-13 op 			return;
180 b0a6bcf7 2022-09-13 op 		fcgi_end_request(clt, 1);
181 b0a6bcf7 2022-09-13 op 		return;
182 b0a6bcf7 2022-09-13 op 	}
183 b0a6bcf7 2022-09-13 op 
184 b0a6bcf7 2022-09-13 op 	(void)snprintf(port, sizeof(port), "%d", clt->clt_pc->proxy_port);
185 b0a6bcf7 2022-09-13 op 
186 b0a6bcf7 2022-09-13 op 	memset(&hints, 0, sizeof(hints));
187 b0a6bcf7 2022-09-13 op 	hints.ai_family = AF_UNSPEC;
188 b0a6bcf7 2022-09-13 op 	hints.ai_socktype = SOCK_STREAM;
189 b0a6bcf7 2022-09-13 op 
190 b0a6bcf7 2022-09-13 op 	query = getaddrinfo_async(clt->clt_pc->proxy_addr, port, &hints, NULL);
191 b0a6bcf7 2022-09-13 op 	if (query == NULL) {
192 b0a6bcf7 2022-09-13 op 		log_warn("getaddrinfo_async");
193 b0a6bcf7 2022-09-13 op 		fcgi_abort_request(clt);
194 b0a6bcf7 2022-09-13 op 		return;
195 b0a6bcf7 2022-09-13 op 	}
196 b0a6bcf7 2022-09-13 op 
197 b0a6bcf7 2022-09-13 op 	clt->clt_evasr = event_asr_run(query, proxy_resolved, clt);
198 b0a6bcf7 2022-09-13 op 	if (clt->clt_evasr == NULL) {
199 b0a6bcf7 2022-09-13 op 		log_warn("event_asr_run");
200 b0a6bcf7 2022-09-13 op 		asr_abort(query);
201 b0a6bcf7 2022-09-13 op 		fcgi_abort_request(clt);
202 b0a6bcf7 2022-09-13 op 		return;
203 b0a6bcf7 2022-09-13 op 	}
204 b0a6bcf7 2022-09-13 op }
205 b0a6bcf7 2022-09-13 op 
206 b0a6bcf7 2022-09-13 op void
207 b0a6bcf7 2022-09-13 op proxy_resolved(struct asr_result *res, void *d)
208 b0a6bcf7 2022-09-13 op {
209 b0a6bcf7 2022-09-13 op 	struct client		*clt = d;
210 b0a6bcf7 2022-09-13 op 	struct proxy_config	*pc = clt->clt_pc;
211 b0a6bcf7 2022-09-13 op 
212 b0a6bcf7 2022-09-13 op 	clt->clt_evasr = NULL;
213 b0a6bcf7 2022-09-13 op 
214 b0a6bcf7 2022-09-13 op 	if (res->ar_gai_errno != 0) {
215 b0a6bcf7 2022-09-13 op 		log_warnx("failed to resolve %s:%d: %s",
216 b0a6bcf7 2022-09-13 op 		    pc->proxy_addr, pc->proxy_port,
217 b0a6bcf7 2022-09-13 op 		    gai_strerror(res->ar_gai_errno));
218 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Status: 501\r\n") == -1)
219 b0a6bcf7 2022-09-13 op 			return;
220 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
221 b0a6bcf7 2022-09-13 op 			return;
222 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Proxy error; connection failed") == -1)
223 b0a6bcf7 2022-09-13 op 			return;
224 b0a6bcf7 2022-09-13 op 		fcgi_end_request(clt, 1);
225 b0a6bcf7 2022-09-13 op 		return;
226 b0a6bcf7 2022-09-13 op 	}
227 b0a6bcf7 2022-09-13 op 
228 b0a6bcf7 2022-09-13 op 	clt->clt_addrinfo = res->ar_addrinfo;
229 b0a6bcf7 2022-09-13 op 	clt->clt_p = clt->clt_addrinfo;
230 b0a6bcf7 2022-09-13 op 	proxy_connect(-1, 0, clt);
231 b0a6bcf7 2022-09-13 op }
232 b0a6bcf7 2022-09-13 op 
233 b0a6bcf7 2022-09-13 op void
234 b0a6bcf7 2022-09-13 op proxy_connect(int fd, short ev, void *d)
235 b0a6bcf7 2022-09-13 op {
236 b0a6bcf7 2022-09-13 op 	struct client		*clt = d;
237 b0a6bcf7 2022-09-13 op 	struct evbuffer		*out;
238 b0a6bcf7 2022-09-13 op 	struct addrinfo		*p;
239 b0a6bcf7 2022-09-13 op 	struct tls_config	*conf;
240 b0a6bcf7 2022-09-13 op 	struct timeval		 conntv = {5, 0};
241 b0a6bcf7 2022-09-13 op 	int			 err = 0;
242 b0a6bcf7 2022-09-13 op 	socklen_t		 len = sizeof(err);
243 b0a6bcf7 2022-09-13 op 
244 b0a6bcf7 2022-09-13 op again:
245 b0a6bcf7 2022-09-13 op 	if (clt->clt_p == NULL)
246 b0a6bcf7 2022-09-13 op 		goto err;
247 b0a6bcf7 2022-09-13 op 
248 b0a6bcf7 2022-09-13 op 	if (clt->clt_fd != -1) {
249 b0a6bcf7 2022-09-13 op 		if (getsockopt(clt->clt_fd, SOL_SOCKET, SO_ERROR, &err, &len)
250 b0a6bcf7 2022-09-13 op 		    == -1)
251 b0a6bcf7 2022-09-13 op 			goto err;
252 b0a6bcf7 2022-09-13 op 		if (err != 0) {
253 b0a6bcf7 2022-09-13 op 			errno = err;
254 b0a6bcf7 2022-09-13 op 			goto err;
255 b0a6bcf7 2022-09-13 op 		}
256 b0a6bcf7 2022-09-13 op 		goto done;
257 b0a6bcf7 2022-09-13 op 	}
258 b0a6bcf7 2022-09-13 op 
259 b0a6bcf7 2022-09-13 op 	p = clt->clt_p;
260 b0a6bcf7 2022-09-13 op 	clt->clt_fd = socket(p->ai_family, p->ai_socktype | SOCK_NONBLOCK,
261 b0a6bcf7 2022-09-13 op 	    p->ai_protocol);
262 b0a6bcf7 2022-09-13 op 	if (clt->clt_fd == -1) {
263 b0a6bcf7 2022-09-13 op 		clt->clt_p = clt->clt_p->ai_next;
264 b0a6bcf7 2022-09-13 op 		goto again;
265 b0a6bcf7 2022-09-13 op 	}
266 b0a6bcf7 2022-09-13 op 
267 b0a6bcf7 2022-09-13 op 	if (connect(clt->clt_fd, p->ai_addr, p->ai_addrlen) == 0)
268 b0a6bcf7 2022-09-13 op 		goto done;
269 b0a6bcf7 2022-09-13 op 
270 b0a6bcf7 2022-09-13 op 	clt->clt_evconn_live = 1;
271 b0a6bcf7 2022-09-13 op 	event_set(&clt->clt_evconn, clt->clt_fd, EV_WRITE, proxy_connect, clt);
272 b0a6bcf7 2022-09-13 op 	event_add(&clt->clt_evconn, &conntv);
273 b0a6bcf7 2022-09-13 op 	return;
274 b0a6bcf7 2022-09-13 op 
275 b0a6bcf7 2022-09-13 op done:
276 b0a6bcf7 2022-09-13 op 	clt->clt_evconn_live = 0;
277 b0a6bcf7 2022-09-13 op 	freeaddrinfo(clt->clt_addrinfo);
278 b0a6bcf7 2022-09-13 op 	clt->clt_addrinfo = clt->clt_p = NULL;
279 b0a6bcf7 2022-09-13 op 
280 b0a6bcf7 2022-09-13 op 	/* initialize TLS for Gemini */
281 b0a6bcf7 2022-09-13 op 	if ((conf = tls_config_new()) == NULL) {
282 b0a6bcf7 2022-09-13 op 		log_warn("tls_config_new failed");
283 b0a6bcf7 2022-09-13 op 		goto err;
284 b0a6bcf7 2022-09-13 op 	}
285 b0a6bcf7 2022-09-13 op 
286 b0a6bcf7 2022-09-13 op 	tls_config_insecure_noverifycert(conf);
287 b0a6bcf7 2022-09-13 op 
288 b0a6bcf7 2022-09-13 op 	if ((clt->clt_ctx = tls_client()) == NULL) {
289 b0a6bcf7 2022-09-13 op 		log_warnx("tls_client failed");
290 b0a6bcf7 2022-09-13 op 		tls_config_free(conf);
291 b0a6bcf7 2022-09-13 op 		goto err;
292 b0a6bcf7 2022-09-13 op 	}
293 b0a6bcf7 2022-09-13 op 
294 b0a6bcf7 2022-09-13 op 	if (tls_configure(clt->clt_ctx, conf) == -1) {
295 b0a6bcf7 2022-09-13 op 		log_warnx("tls_configure failed");
296 b0a6bcf7 2022-09-13 op 		tls_config_free(conf);
297 b0a6bcf7 2022-09-13 op 		goto err;
298 b0a6bcf7 2022-09-13 op 	}
299 b0a6bcf7 2022-09-13 op 
300 b0a6bcf7 2022-09-13 op 	tls_config_free(conf);
301 b0a6bcf7 2022-09-13 op 
302 b0a6bcf7 2022-09-13 op 	if (tls_connect_socket(clt->clt_ctx, clt->clt_fd,
303 b0a6bcf7 2022-09-13 op 	    clt->clt_pc->proxy_name) == -1) {
304 b0a6bcf7 2022-09-13 op 		log_warnx("tls_connect_socket failed");
305 b0a6bcf7 2022-09-13 op 		goto err;
306 b0a6bcf7 2022-09-13 op 	}
307 b0a6bcf7 2022-09-13 op 
308 b0a6bcf7 2022-09-13 op 	clt->clt_bev = bufferevent_new(clt->clt_fd, proxy_read, proxy_write,
309 b0a6bcf7 2022-09-13 op 	    proxy_error, clt);
310 b0a6bcf7 2022-09-13 op 	if (clt->clt_bev == NULL) {
311 b0a6bcf7 2022-09-13 op 		log_warn("bufferevent_new");
312 b0a6bcf7 2022-09-13 op 		goto err;
313 b0a6bcf7 2022-09-13 op 	}
314 b0a6bcf7 2022-09-13 op 	out = EVBUFFER_OUTPUT(clt->clt_bev);
315 b0a6bcf7 2022-09-13 op 
316 b0a6bcf7 2022-09-13 op 	event_set(&clt->clt_bev->ev_read, clt->clt_fd, EV_READ,
317 b0a6bcf7 2022-09-13 op 	    proxy_tls_readcb, clt->clt_bev);
318 b0a6bcf7 2022-09-13 op 	event_set(&clt->clt_bev->ev_write, clt->clt_fd, EV_WRITE,
319 b0a6bcf7 2022-09-13 op 	    proxy_tls_writecb, clt->clt_bev);
320 b0a6bcf7 2022-09-13 op 
321 b0a6bcf7 2022-09-13 op 	/* bufferevent_settimeout(); */
322 b0a6bcf7 2022-09-13 op 	bufferevent_enable(clt->clt_bev, EV_READ|EV_WRITE);
323 b0a6bcf7 2022-09-13 op 
324 b0a6bcf7 2022-09-13 op 	/* TODO: compute the URL */
325 b0a6bcf7 2022-09-13 op 	if (evbuffer_add_printf(out, "gemini://localhost/\r\n") == -1) {
326 b0a6bcf7 2022-09-13 op 		log_warn("bufferevent_printf failed");
327 b0a6bcf7 2022-09-13 op 		goto err;
328 b0a6bcf7 2022-09-13 op 	}
329 b0a6bcf7 2022-09-13 op 
330 b0a6bcf7 2022-09-13 op 	return;
331 b0a6bcf7 2022-09-13 op 
332 b0a6bcf7 2022-09-13 op err:
333 b0a6bcf7 2022-09-13 op 	log_warn("failed to connect to %s:%d",
334 b0a6bcf7 2022-09-13 op 	    clt->clt_pc->proxy_addr, clt->clt_pc->proxy_port);
335 b0a6bcf7 2022-09-13 op 	if (clt_printf(clt, "Status: 501\r\n") == -1)
336 b0a6bcf7 2022-09-13 op 		return;
337 b0a6bcf7 2022-09-13 op 	if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
338 b0a6bcf7 2022-09-13 op 		return;
339 b0a6bcf7 2022-09-13 op 	if (clt_printf(clt, "Proxy error; connection failed") == -1)
340 b0a6bcf7 2022-09-13 op 		return;
341 b0a6bcf7 2022-09-13 op 	fcgi_end_request(clt, 1);
342 b0a6bcf7 2022-09-13 op }
343 b0a6bcf7 2022-09-13 op 
344 b0a6bcf7 2022-09-13 op void
345 b0a6bcf7 2022-09-13 op proxy_read(struct bufferevent *bev, void *d)
346 b0a6bcf7 2022-09-13 op {
347 b0a6bcf7 2022-09-13 op 	struct client		*clt = d;
348 b0a6bcf7 2022-09-13 op 	struct evbuffer		*src = EVBUFFER_INPUT(bev);
349 b0a6bcf7 2022-09-13 op 	char			*hdr;
350 b0a6bcf7 2022-09-13 op 	size_t			 len;
351 b0a6bcf7 2022-09-13 op 	int			 code;
352 b0a6bcf7 2022-09-13 op 
353 b0a6bcf7 2022-09-13 op 	if (clt->clt_headersdone) {
354 b0a6bcf7 2022-09-13 op 	copy:
355 b0a6bcf7 2022-09-13 op 		clt_write_bufferevent(clt, bev);
356 b0a6bcf7 2022-09-13 op 		return;
357 b0a6bcf7 2022-09-13 op 	}
358 b0a6bcf7 2022-09-13 op 
359 b0a6bcf7 2022-09-13 op 	hdr = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF_STRICT);
360 b0a6bcf7 2022-09-13 op 	if (hdr == NULL) {
361 b0a6bcf7 2022-09-13 op 		if (EVBUFFER_LENGTH(src) >= 1026)
362 b0a6bcf7 2022-09-13 op 			proxy_error(bev, EV_READ, clt);
363 b0a6bcf7 2022-09-13 op 		return;
364 b0a6bcf7 2022-09-13 op 	}
365 b0a6bcf7 2022-09-13 op 
366 b0a6bcf7 2022-09-13 op 	if (len < 4 ||
367 b0a6bcf7 2022-09-13 op 	    !isdigit((unsigned char)hdr[0]) ||
368 b0a6bcf7 2022-09-13 op 	    !isdigit((unsigned char)hdr[1]) ||
369 b0a6bcf7 2022-09-13 op 	    hdr[2] != ' ') {
370 b0a6bcf7 2022-09-13 op 		log_warnx("invalid ");
371 b0a6bcf7 2022-09-13 op 		free(hdr);
372 b0a6bcf7 2022-09-13 op 		proxy_error(bev, EV_READ, clt);
373 b0a6bcf7 2022-09-13 op 		return;
374 b0a6bcf7 2022-09-13 op 	}
375 b0a6bcf7 2022-09-13 op 
376 b0a6bcf7 2022-09-13 op 	code = (hdr[0] - '0') * 10 + (hdr[1] - '0');
377 b0a6bcf7 2022-09-13 op 	if (code != 20) {
378 b0a6bcf7 2022-09-13 op 		log_warnx("un-handled gemini reply status %d", code);
379 b0a6bcf7 2022-09-13 op 		free(hdr);
380 b0a6bcf7 2022-09-13 op 		proxy_error(bev, EV_READ, clt);
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 	if (clt_printf(clt, "Content-Type: %s\r\n", &hdr[4]) == -1)
385 b0a6bcf7 2022-09-13 op 		return;
386 b0a6bcf7 2022-09-13 op 	if (clt_printf(clt, "\r\n") == -1)
387 b0a6bcf7 2022-09-13 op 		return;
388 b0a6bcf7 2022-09-13 op 
389 b0a6bcf7 2022-09-13 op 	clt->clt_headersdone = 1;
390 b0a6bcf7 2022-09-13 op 	goto copy;
391 b0a6bcf7 2022-09-13 op }
392 b0a6bcf7 2022-09-13 op 
393 b0a6bcf7 2022-09-13 op void
394 b0a6bcf7 2022-09-13 op proxy_write(struct bufferevent *bev, void *d)
395 b0a6bcf7 2022-09-13 op {
396 b0a6bcf7 2022-09-13 op 	return;
397 b0a6bcf7 2022-09-13 op }
398 b0a6bcf7 2022-09-13 op 
399 b0a6bcf7 2022-09-13 op void
400 b0a6bcf7 2022-09-13 op proxy_error(struct bufferevent *bev, short err, void *d)
401 b0a6bcf7 2022-09-13 op {
402 b0a6bcf7 2022-09-13 op 	struct client		*clt = d;
403 b0a6bcf7 2022-09-13 op 	int			 status = !(err & EVBUFFER_EOF);
404 b0a6bcf7 2022-09-13 op 
405 b0a6bcf7 2022-09-13 op 	log_debug("proxy error, shutting down the connection (err: %x)",
406 b0a6bcf7 2022-09-13 op 	    err);
407 b0a6bcf7 2022-09-13 op 
408 b0a6bcf7 2022-09-13 op 	if (!clt->clt_headersdone) {
409 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Status: 501\r\n") == -1)
410 b0a6bcf7 2022-09-13 op 			return;
411 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
412 b0a6bcf7 2022-09-13 op 			return;
413 b0a6bcf7 2022-09-13 op 		if (clt_printf(clt, "Proxy error\n") == -1)
414 b0a6bcf7 2022-09-13 op 			return;
415 b0a6bcf7 2022-09-13 op 	}
416 b0a6bcf7 2022-09-13 op 
417 b0a6bcf7 2022-09-13 op 	fcgi_end_request(clt, status);
418 b0a6bcf7 2022-09-13 op }
419 b0a6bcf7 2022-09-13 op 
420 b0a6bcf7 2022-09-13 op void
421 b0a6bcf7 2022-09-13 op proxy_tls_readcb(int fd, short event, void *arg)
422 b0a6bcf7 2022-09-13 op {
423 b0a6bcf7 2022-09-13 op 	struct bufferevent	*bufev = arg;
424 b0a6bcf7 2022-09-13 op 	struct client		*clt = bufev->cbarg;
425 b0a6bcf7 2022-09-13 op 	char			 rbuf[IBUF_READ_SIZE];
426 b0a6bcf7 2022-09-13 op 	int			 what = EVBUFFER_READ;
427 b0a6bcf7 2022-09-13 op 	int			 howmuch = IBUF_READ_SIZE;
428 b0a6bcf7 2022-09-13 op 	ssize_t			 ret;
429 b0a6bcf7 2022-09-13 op 	size_t			 len;
430 b0a6bcf7 2022-09-13 op 
431 b0a6bcf7 2022-09-13 op 	if (event == EV_TIMEOUT) {
432 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_TIMEOUT;
433 b0a6bcf7 2022-09-13 op 		goto err;
434 b0a6bcf7 2022-09-13 op 	}
435 b0a6bcf7 2022-09-13 op 
436 b0a6bcf7 2022-09-13 op 	if (bufev->wm_read.high != 0)
437 b0a6bcf7 2022-09-13 op 		howmuch = MINIMUM(sizeof(rbuf), bufev->wm_read.high);
438 b0a6bcf7 2022-09-13 op 
439 b0a6bcf7 2022-09-13 op 	ret = tls_read(clt->clt_ctx, rbuf, howmuch);
440 b0a6bcf7 2022-09-13 op 	if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
441 b0a6bcf7 2022-09-13 op 		goto retry;
442 b0a6bcf7 2022-09-13 op 	} else if (ret == -1) {
443 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_ERROR;
444 b0a6bcf7 2022-09-13 op 		goto err;
445 b0a6bcf7 2022-09-13 op 	}
446 b0a6bcf7 2022-09-13 op 	len = ret;
447 b0a6bcf7 2022-09-13 op 
448 b0a6bcf7 2022-09-13 op 	if (len == 0) {
449 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_EOF;
450 b0a6bcf7 2022-09-13 op 		goto err;
451 b0a6bcf7 2022-09-13 op 	}
452 b0a6bcf7 2022-09-13 op 
453 b0a6bcf7 2022-09-13 op 	if (evbuffer_add(bufev->input, rbuf, len) == -1) {
454 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_ERROR;
455 b0a6bcf7 2022-09-13 op 		goto err;
456 b0a6bcf7 2022-09-13 op 	}
457 b0a6bcf7 2022-09-13 op 
458 b0a6bcf7 2022-09-13 op 	proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
459 b0a6bcf7 2022-09-13 op 
460 b0a6bcf7 2022-09-13 op 	len = EVBUFFER_LENGTH(bufev->input);
461 b0a6bcf7 2022-09-13 op 	if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
462 b0a6bcf7 2022-09-13 op 		return;
463 b0a6bcf7 2022-09-13 op 	if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
464 b0a6bcf7 2022-09-13 op 		struct evbuffer *buf = bufev->input;
465 b0a6bcf7 2022-09-13 op 		event_del(&bufev->ev_read);
466 b0a6bcf7 2022-09-13 op 		evbuffer_setcb(buf, bufferevent_read_pressure_cb, bufev);
467 b0a6bcf7 2022-09-13 op 		return;
468 b0a6bcf7 2022-09-13 op 	}
469 b0a6bcf7 2022-09-13 op 
470 b0a6bcf7 2022-09-13 op 	if (bufev->readcb != NULL)
471 b0a6bcf7 2022-09-13 op 		(*bufev->readcb)(bufev, bufev->cbarg);
472 b0a6bcf7 2022-09-13 op 	return;
473 b0a6bcf7 2022-09-13 op 
474 b0a6bcf7 2022-09-13 op retry:
475 b0a6bcf7 2022-09-13 op 	proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
476 b0a6bcf7 2022-09-13 op 	return;
477 b0a6bcf7 2022-09-13 op 
478 b0a6bcf7 2022-09-13 op err:
479 b0a6bcf7 2022-09-13 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
480 b0a6bcf7 2022-09-13 op }
481 b0a6bcf7 2022-09-13 op 
482 b0a6bcf7 2022-09-13 op int
483 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(struct event *ev, int timeout)
484 b0a6bcf7 2022-09-13 op {
485 b0a6bcf7 2022-09-13 op 	struct timeval tv, *ptv = NULL;
486 b0a6bcf7 2022-09-13 op 
487 b0a6bcf7 2022-09-13 op 	if (timeout) {
488 b0a6bcf7 2022-09-13 op 		timerclear(&tv);
489 b0a6bcf7 2022-09-13 op 		tv.tv_sec = timeout;
490 b0a6bcf7 2022-09-13 op 		ptv = &tv;
491 b0a6bcf7 2022-09-13 op 	}
492 b0a6bcf7 2022-09-13 op 
493 b0a6bcf7 2022-09-13 op 	return (event_add(ev, ptv));
494 b0a6bcf7 2022-09-13 op }
495 b0a6bcf7 2022-09-13 op 
496 b0a6bcf7 2022-09-13 op void
497 b0a6bcf7 2022-09-13 op proxy_tls_writecb(int fd, short event, void *arg)
498 b0a6bcf7 2022-09-13 op {
499 b0a6bcf7 2022-09-13 op 	struct bufferevent	*bufev = arg;
500 b0a6bcf7 2022-09-13 op 	struct client		*clt = bufev->cbarg;
501 b0a6bcf7 2022-09-13 op 	ssize_t			 ret;
502 b0a6bcf7 2022-09-13 op 	short			 what = EVBUFFER_WRITE;
503 b0a6bcf7 2022-09-13 op 	size_t			 len;
504 b0a6bcf7 2022-09-13 op 
505 b0a6bcf7 2022-09-13 op 	if (event == EV_TIMEOUT) {
506 b0a6bcf7 2022-09-13 op 		what |= EVBUFFER_TIMEOUT;
507 b0a6bcf7 2022-09-13 op 		goto err;
508 b0a6bcf7 2022-09-13 op 	}
509 b0a6bcf7 2022-09-13 op 
510 b0a6bcf7 2022-09-13 op 	if (EVBUFFER_LENGTH(bufev->output)) {
511 b0a6bcf7 2022-09-13 op 		ret = tls_write(clt->clt_ctx,
512 b0a6bcf7 2022-09-13 op 		    EVBUFFER_DATA(bufev->output),
513 b0a6bcf7 2022-09-13 op 		    EVBUFFER_LENGTH(bufev->output));
514 b0a6bcf7 2022-09-13 op 		if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
515 b0a6bcf7 2022-09-13 op 			goto retry;
516 b0a6bcf7 2022-09-13 op 		} else if (ret == -1) {
517 b0a6bcf7 2022-09-13 op 			what |= EVBUFFER_ERROR;
518 b0a6bcf7 2022-09-13 op 			goto err;
519 b0a6bcf7 2022-09-13 op 		}
520 b0a6bcf7 2022-09-13 op 		len = ret;
521 b0a6bcf7 2022-09-13 op 		evbuffer_drain(bufev->output, len);
522 b0a6bcf7 2022-09-13 op 	}
523 b0a6bcf7 2022-09-13 op 
524 b0a6bcf7 2022-09-13 op 	if (EVBUFFER_LENGTH(bufev->output) != 0)
525 b0a6bcf7 2022-09-13 op 		proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
526 b0a6bcf7 2022-09-13 op 
527 b0a6bcf7 2022-09-13 op 	if (bufev->writecb != NULL &&
528 b0a6bcf7 2022-09-13 op 	    EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
529 b0a6bcf7 2022-09-13 op 		(*bufev->writecb)(bufev, bufev->cbarg);
530 b0a6bcf7 2022-09-13 op 	return;
531 b0a6bcf7 2022-09-13 op 
532 b0a6bcf7 2022-09-13 op retry:
533 b0a6bcf7 2022-09-13 op 	proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
534 b0a6bcf7 2022-09-13 op 	return;
535 b0a6bcf7 2022-09-13 op 
536 b0a6bcf7 2022-09-13 op err:
537 b0a6bcf7 2022-09-13 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
538 b0a6bcf7 2022-09-13 op }
539 b0a6bcf7 2022-09-13 op 
540 b0a6bcf7 2022-09-13 op void
541 b0a6bcf7 2022-09-13 op proxy_client_free(struct client *clt)
542 b0a6bcf7 2022-09-13 op {
543 b0a6bcf7 2022-09-13 op 	if (clt->clt_evasr)
544 b0a6bcf7 2022-09-13 op 		event_asr_abort(clt->clt_evasr);
545 b0a6bcf7 2022-09-13 op 
546 b0a6bcf7 2022-09-13 op 	if (clt->clt_addrinfo)
547 b0a6bcf7 2022-09-13 op 		freeaddrinfo(clt->clt_addrinfo);
548 b0a6bcf7 2022-09-13 op 
549 b0a6bcf7 2022-09-13 op 	if (clt->clt_evconn_live)
550 b0a6bcf7 2022-09-13 op 		event_del(&clt->clt_evconn);
551 b0a6bcf7 2022-09-13 op 
552 b0a6bcf7 2022-09-13 op 	if (clt->clt_fd != -1)
553 b0a6bcf7 2022-09-13 op 		close(clt->clt_fd);
554 b0a6bcf7 2022-09-13 op 
555 b0a6bcf7 2022-09-13 op 	if (clt->clt_ctx)
556 b0a6bcf7 2022-09-13 op 		tls_free(clt->clt_ctx);
557 b0a6bcf7 2022-09-13 op 
558 b0a6bcf7 2022-09-13 op 	if (clt->clt_bev)
559 b0a6bcf7 2022-09-13 op 		bufferevent_free(clt->clt_bev);
560 b0a6bcf7 2022-09-13 op 
561 b0a6bcf7 2022-09-13 op 	free(clt->clt_server_name);
562 b0a6bcf7 2022-09-13 op 	free(clt->clt_script_name);
563 b0a6bcf7 2022-09-13 op 	free(clt->clt_path_info);
564 b0a6bcf7 2022-09-13 op 	free(clt);
565 b0a6bcf7 2022-09-13 op }