op public repos

Blame

Date:: Wed Sep 22 16:48:26 2021 UTC
Message:: proposal: show custom message when the capsule is not reachable As discussed in the mailing list, this is a quick draft for a (imho) neat functionality: if there's an error connecting to a capsule, try to do a TXT query at _offline.$hostname (this is subject to change) and show the message to the user. It's a nice way to communicate to the user that a certain capsule is, for example, supposed to be offline in certain situations.
Actions:: History | Blob | Raw File
  1 c8dba1e6 2021-07-24 op /*
  2 c8dba1e6 2021-07-24 op  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
  3 c8dba1e6 2021-07-24 op  *
  4 c8dba1e6 2021-07-24 op  * Permission to use, copy, modify, and distribute this software for any
  5 c8dba1e6 2021-07-24 op  * purpose with or without fee is hereby granted, provided that the above
  6 c8dba1e6 2021-07-24 op  * copyright notice and this permission notice appear in all copies.
  7 c8dba1e6 2021-07-24 op  *
  8 c8dba1e6 2021-07-24 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 c8dba1e6 2021-07-24 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 c8dba1e6 2021-07-24 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 c8dba1e6 2021-07-24 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 c8dba1e6 2021-07-24 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 c8dba1e6 2021-07-24 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 c8dba1e6 2021-07-24 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 c8dba1e6 2021-07-24 op  */
 16 c8dba1e6 2021-07-24 op 
 17 c8dba1e6 2021-07-24 op #include "compat.h"
 18 c8dba1e6 2021-07-24 op 
 19 c8dba1e6 2021-07-24 op #include <sys/types.h>
 20 c8dba1e6 2021-07-24 op #include <sys/socket.h>
 21 c8dba1e6 2021-07-24 op 
 22 c8dba1e6 2021-07-24 op #include <netinet/in.h>
 23 c8dba1e6 2021-07-24 op 
 24 c8dba1e6 2021-07-24 op #include <assert.h>
 25 c8dba1e6 2021-07-24 op #include <ctype.h>
 26 c8dba1e6 2021-07-24 op #include <errno.h>
 27 c8dba1e6 2021-07-24 op #include <netdb.h>
 28 c8dba1e6 2021-07-24 op #include <stdarg.h>
 29 c8dba1e6 2021-07-24 op #include <stdio.h>
 30 c8dba1e6 2021-07-24 op #include <stdlib.h>
 31 c8dba1e6 2021-07-24 op #include <string.h>
 32 c8dba1e6 2021-07-24 op #include <tls.h>
 33 c8dba1e6 2021-07-24 op #include <unistd.h>
 34 c8dba1e6 2021-07-24 op 
 35 c8dba1e6 2021-07-24 op #if HAVE_ASR_RUN
 36 c8dba1e6 2021-07-24 op # include <asr.h>
 37 c8dba1e6 2021-07-24 op #endif
 38 c8dba1e6 2021-07-24 op 
 39 35a31606 2021-09-22 op #define DNS_RDATACLASS_IN	1
 40 35a31606 2021-09-22 op #define DNS_RDATATYPE_TXT	16
 41 35a31606 2021-09-22 op 
 42 c8dba1e6 2021-07-24 op #include "telescope.h"
 43 c8dba1e6 2021-07-24 op 
 44 c8dba1e6 2021-07-24 op static struct imsgev		*iev_ui;
 45 c8dba1e6 2021-07-24 op static struct tls_config	*tlsconf;
 46 c8dba1e6 2021-07-24 op 
 47 edfe87be 2021-07-24 op /* a pending request */
 48 edfe87be 2021-07-24 op struct req {
 49 edfe87be 2021-07-24 op 	struct phos_uri		 url;
 50 edfe87be 2021-07-24 op 	uint32_t		 id;
 51 5fb52fc0 2021-07-25 op 	int			 proto;
 52 edfe87be 2021-07-24 op 	int			 fd;
 53 edfe87be 2021-07-24 op 	struct tls		*ctx;
 54 edfe87be 2021-07-24 op 	char			 req[1024];
 55 edfe87be 2021-07-24 op 	size_t			 len;
 56 edfe87be 2021-07-24 op 	int			 done_header;
 57 edfe87be 2021-07-24 op 	struct bufferevent	*bev;
 58 c8dba1e6 2021-07-24 op 
 59 edfe87be 2021-07-24 op 	struct addrinfo		*servinfo, *p;
 60 edfe87be 2021-07-24 op #if HAVE_ASR_RUN
 61 edfe87be 2021-07-24 op 	struct addrinfo		 hints;
 62 edfe87be 2021-07-24 op 	struct event_asr	*asrev;
 63 edfe87be 2021-07-24 op #endif
 64 edfe87be 2021-07-24 op 
 65 edfe87be 2021-07-24 op 	TAILQ_ENTRY(req)	 reqs;
 66 edfe87be 2021-07-24 op };
 67 edfe87be 2021-07-24 op 
 68 e2a349f7 2021-07-24 op static struct req	*req_by_id(uint32_t);
 69 e2a349f7 2021-07-24 op 
 70 77ae5e91 2021-07-24 op static void	 die(void) __attribute__((__noreturn__));
 71 c8dba1e6 2021-07-24 op 
 72 77ae5e91 2021-07-24 op static void	 try_to_connect(int, short, void*);
 73 c8dba1e6 2021-07-24 op 
 74 35a31606 2021-09-22 op static void	 offline_dns_done(struct rrsetinfo *, struct req *);
 75 35a31606 2021-09-22 op static void	 offline_dns_query(struct req *);
 76 35a31606 2021-09-22 op 
 77 c8dba1e6 2021-07-24 op #if HAVE_ASR_RUN
 78 77ae5e91 2021-07-24 op static void	 query_done(struct asr_result*, void*);
 79 77ae5e91 2021-07-24 op static void	 async_conn_towards(struct req*);
 80 35a31606 2021-09-22 op static void	 offline_query_done(struct asr_result *, void *);
 81 c8dba1e6 2021-07-24 op #else
 82 77ae5e91 2021-07-24 op static void	 blocking_conn_towards(struct req*);
 83 c8dba1e6 2021-07-24 op #endif
 84 c8dba1e6 2021-07-24 op 
 85 77ae5e91 2021-07-24 op static void	 close_with_err(struct req*, const char*);
 86 77ae5e91 2021-07-24 op static void	 close_with_errf(struct req*, const char*, ...)
 87 bfb9acb0 2021-07-24 op     __attribute__((format(printf, 2, 3)));
 88 c8dba1e6 2021-07-24 op 
 89 77ae5e91 2021-07-24 op static void	 net_tls_handshake(int, short, void *);
 90 77ae5e91 2021-07-24 op static void	 net_tls_readcb(int, short, void *);
 91 77ae5e91 2021-07-24 op static void	 net_tls_writecb(int, short, void *);
 92 5b762f01 2021-07-24 op 
 93 77ae5e91 2021-07-24 op static int	 gemini_parse_reply(struct req *, const char *, size_t);
 94 5b762f01 2021-07-24 op 
 95 77ae5e91 2021-07-24 op static void	 net_ready(struct req *req);
 96 77ae5e91 2021-07-24 op static void	 net_read(struct bufferevent *, void *);
 97 77ae5e91 2021-07-24 op static void	 net_write(struct bufferevent *, void *);
 98 77ae5e91 2021-07-24 op static void	 net_error(struct bufferevent *, short, void *);
 99 c8dba1e6 2021-07-24 op 
100 77ae5e91 2021-07-24 op static void	 handle_get_raw(struct imsg *, size_t);
101 77ae5e91 2021-07-24 op static void	 handle_cert_status(struct imsg*, size_t);
102 77ae5e91 2021-07-24 op static void	 handle_proceed(struct imsg*, size_t);
103 77ae5e91 2021-07-24 op static void	 handle_stop(struct imsg*, size_t);
104 77ae5e91 2021-07-24 op static void	 handle_quit(struct imsg*, size_t);
105 77ae5e91 2021-07-24 op static void	 handle_dispatch_imsg(int, short, void*);
106 77ae5e91 2021-07-24 op 
107 77ae5e91 2021-07-24 op static int	 net_send_ui(int, uint32_t, const void *, uint16_t);
108 c8dba1e6 2021-07-24 op 
109 c8dba1e6 2021-07-24 op /* TODO: making this customizable */
110 c8dba1e6 2021-07-24 op struct timeval timeout_for_handshake = { 5, 0 };
111 c8dba1e6 2021-07-24 op 
112 c8dba1e6 2021-07-24 op static imsg_handlerfn *handlers[] = {
113 c8dba1e6 2021-07-24 op 	[IMSG_GET_RAW]		= handle_get_raw,
114 c8dba1e6 2021-07-24 op 	[IMSG_CERT_STATUS]	= handle_cert_status,
115 c8dba1e6 2021-07-24 op 	[IMSG_PROCEED]		= handle_proceed,
116 c8dba1e6 2021-07-24 op 	[IMSG_STOP]		= handle_stop,
117 c8dba1e6 2021-07-24 op 	[IMSG_QUIT]		= handle_quit,
118 c8dba1e6 2021-07-24 op };
119 c8dba1e6 2021-07-24 op 
120 c8dba1e6 2021-07-24 op typedef void (*statefn)(int, short, void*);
121 c8dba1e6 2021-07-24 op 
122 c8dba1e6 2021-07-24 op TAILQ_HEAD(, req) reqhead;
123 c8dba1e6 2021-07-24 op 
124 c8dba1e6 2021-07-24 op static inline void
125 c8dba1e6 2021-07-24 op yield_r(struct req *req, statefn fn, struct timeval *tv)
126 c8dba1e6 2021-07-24 op {
127 c8dba1e6 2021-07-24 op 	event_once(req->fd, EV_READ, fn, req, tv);
128 c8dba1e6 2021-07-24 op }
129 c8dba1e6 2021-07-24 op 
130 c8dba1e6 2021-07-24 op static inline void
131 c8dba1e6 2021-07-24 op yield_w(struct req *req, statefn fn, struct timeval *tv)
132 c8dba1e6 2021-07-24 op {
133 c8dba1e6 2021-07-24 op 	event_once(req->fd, EV_WRITE, fn, req, tv);
134 c8dba1e6 2021-07-24 op }
135 c8dba1e6 2021-07-24 op 
136 e2a349f7 2021-07-24 op static struct req *
137 e2a349f7 2021-07-24 op req_by_id(uint32_t id)
138 e2a349f7 2021-07-24 op {
139 e2a349f7 2021-07-24 op 	struct req *r;
140 e2a349f7 2021-07-24 op 
141 e2a349f7 2021-07-24 op 	TAILQ_FOREACH(r, &reqhead, reqs) {
142 e2a349f7 2021-07-24 op 		if (r->id == id)
143 e2a349f7 2021-07-24 op 			return r;
144 e2a349f7 2021-07-24 op 	}
145 e2a349f7 2021-07-24 op 
146 e2a349f7 2021-07-24 op 	return NULL;
147 e2a349f7 2021-07-24 op }
148 e2a349f7 2021-07-24 op 
149 c8dba1e6 2021-07-24 op static void __attribute__((__noreturn__))
150 c8dba1e6 2021-07-24 op die(void)
151 c8dba1e6 2021-07-24 op {
152 c8dba1e6 2021-07-24 op 	abort(); 		/* TODO */
153 c8dba1e6 2021-07-24 op }
154 c8dba1e6 2021-07-24 op 
155 c8dba1e6 2021-07-24 op static void
156 c8dba1e6 2021-07-24 op try_to_connect(int fd, short ev, void *d)
157 c8dba1e6 2021-07-24 op {
158 c8dba1e6 2021-07-24 op 	struct req	*req = d;
159 c8dba1e6 2021-07-24 op 	int		 error = 0;
160 c8dba1e6 2021-07-24 op 	socklen_t	 len = sizeof(error);
161 c8dba1e6 2021-07-24 op 
162 c8dba1e6 2021-07-24 op again:
163 c8dba1e6 2021-07-24 op 	if (req->p == NULL)
164 c8dba1e6 2021-07-24 op 		goto err;
165 c8dba1e6 2021-07-24 op 
166 c8dba1e6 2021-07-24 op 	if (req->fd != -1) {
167 bfb9acb0 2021-07-24 op 		if (getsockopt(req->fd, SOL_SOCKET, SO_ERROR, &error,
168 bfb9acb0 2021-07-24 op 		    &len) == -1)
169 c8dba1e6 2021-07-24 op 			goto err;
170 c8dba1e6 2021-07-24 op 		if (error != 0) {
171 95a8c791 2021-08-26 op 			errno = error;
172 95a8c791 2021-08-26 op 			goto err;
173 c8dba1e6 2021-07-24 op 		}
174 c8dba1e6 2021-07-24 op 		goto done;
175 c8dba1e6 2021-07-24 op 	}
176 c8dba1e6 2021-07-24 op 
177 bfb9acb0 2021-07-24 op 	req->fd = socket(req->p->ai_family, req->p->ai_socktype,
178 bfb9acb0 2021-07-24 op 	    req->p->ai_protocol);
179 c8dba1e6 2021-07-24 op 	if (req->fd == -1) {
180 c8dba1e6 2021-07-24 op 		req->p = req->p->ai_next;
181 c8dba1e6 2021-07-24 op 		goto again;
182 c8dba1e6 2021-07-24 op 	} else {
183 c8dba1e6 2021-07-24 op 		mark_nonblock(req->fd);
184 c8dba1e6 2021-07-24 op 		if (connect(req->fd, req->p->ai_addr, req->p->ai_addrlen) == 0)
185 c8dba1e6 2021-07-24 op 			goto done;
186 c8dba1e6 2021-07-24 op 		yield_w(req, try_to_connect, NULL);
187 c8dba1e6 2021-07-24 op 	}
188 c8dba1e6 2021-07-24 op 	return;
189 c8dba1e6 2021-07-24 op 
190 c8dba1e6 2021-07-24 op err:
191 c8dba1e6 2021-07-24 op 	freeaddrinfo(req->servinfo);
192 35a31606 2021-09-22 op 	offline_dns_query(req);
193 c8dba1e6 2021-07-24 op 	return;
194 c8dba1e6 2021-07-24 op 
195 c8dba1e6 2021-07-24 op done:
196 c8dba1e6 2021-07-24 op 	freeaddrinfo(req->servinfo);
197 0d40acb1 2021-07-24 op 
198 5fb52fc0 2021-07-25 op 	switch (req->proto) {
199 5fb52fc0 2021-07-25 op 	case PROTO_FINGER:
200 1663bf1e 2021-07-25 op 	case PROTO_GOPHER:
201 1663bf1e 2021-07-25 op 		/* finger and gopher don't have a header nor TLS */
202 5fb52fc0 2021-07-25 op 		req->done_header = 1;
203 5fb52fc0 2021-07-25 op 		net_ready(req);
204 5fb52fc0 2021-07-25 op 		break;
205 5fb52fc0 2021-07-25 op 
206 5fb52fc0 2021-07-25 op 	case PROTO_GEMINI:
207 5fb52fc0 2021-07-25 op 		/* prepare tls */
208 5fb52fc0 2021-07-25 op 		if ((req->ctx = tls_client()) == NULL) {
209 5fb52fc0 2021-07-25 op 			close_with_errf(req, "tls_client: %s",
210 5fb52fc0 2021-07-25 op 			    strerror(errno));
211 5fb52fc0 2021-07-25 op 			return;
212 5fb52fc0 2021-07-25 op 		}
213 5fb52fc0 2021-07-25 op 		if (tls_configure(req->ctx, tlsconf) == -1) {
214 5fb52fc0 2021-07-25 op 			close_with_errf(req, "tls_configure: %s",
215 5fb52fc0 2021-07-25 op 			    tls_error(req->ctx));
216 5fb52fc0 2021-07-25 op 			return;
217 5fb52fc0 2021-07-25 op 		}
218 5fb52fc0 2021-07-25 op 		if (tls_connect_socket(req->ctx, req->fd, req->url.host)
219 5fb52fc0 2021-07-25 op 		    == -1) {
220 5fb52fc0 2021-07-25 op 			close_with_errf(req, "tls_connect_socket: %s",
221 5fb52fc0 2021-07-25 op 			    tls_error(req->ctx));
222 5fb52fc0 2021-07-25 op 			return;
223 5fb52fc0 2021-07-25 op 		}
224 5fb52fc0 2021-07-25 op 		yield_w(req, net_tls_handshake, &timeout_for_handshake);
225 5fb52fc0 2021-07-25 op 		break;
226 5fb52fc0 2021-07-25 op 
227 5fb52fc0 2021-07-25 op 	default:
228 5fb52fc0 2021-07-25 op 		die();
229 35a31606 2021-09-22 op 	}
230 35a31606 2021-09-22 op }
231 35a31606 2021-09-22 op 
232 35a31606 2021-09-22 op static void
233 35a31606 2021-09-22 op offline_dns_done(struct rrsetinfo *res, struct req *req)
234 35a31606 2021-09-22 op {
235 35a31606 2021-09-22 op 	size_t		 i, len;
236 35a31606 2021-09-22 op 	const char	*reason = NULL;
237 35a31606 2021-09-22 op 
238 35a31606 2021-09-22 op 	for (i = 0; i < res->rri_nrdatas; ++i) {
239 35a31606 2021-09-22 op 		if (res->rri_rdatas[i].rdi_length <= 1)
240 35a31606 2021-09-22 op 			continue;
241 35a31606 2021-09-22 op 
242 35a31606 2021-09-22 op 		len = *(uint8_t*)res->rri_rdatas[i].rdi_data;
243 35a31606 2021-09-22 op 		reason = res->rri_rdatas[i].rdi_data+1;
244 35a31606 2021-09-22 op 		break;
245 35a31606 2021-09-22 op 	}
246 35a31606 2021-09-22 op 
247 35a31606 2021-09-22 op 	if (reason == NULL)
248 35a31606 2021-09-22 op 		close_with_errf(req, "failed to connect to %s", req->url.host);
249 35a31606 2021-09-22 op 	else
250 35a31606 2021-09-22 op 		close_with_errf(req,
251 35a31606 2021-09-22 op 		    "failed to connect to %s\n\nThe site says: %*s\n",
252 35a31606 2021-09-22 op 		    req->url.host, (int)len, reason);
253 35a31606 2021-09-22 op 
254 35a31606 2021-09-22 op 	freerrset(res);
255 35a31606 2021-09-22 op }
256 35a31606 2021-09-22 op 
257 35a31606 2021-09-22 op static void
258 35a31606 2021-09-22 op offline_dns_query(struct req *req)
259 35a31606 2021-09-22 op {
260 35a31606 2021-09-22 op 	char		hostname[254];
261 35a31606 2021-09-22 op 
262 35a31606 2021-09-22 op 	strlcpy(hostname, "_offline.", sizeof(hostname));
263 35a31606 2021-09-22 op 	strlcat(hostname, req->url.host, sizeof(hostname));
264 35a31606 2021-09-22 op 
265 35a31606 2021-09-22 op #if HAVE_ASR_RUN
266 35a31606 2021-09-22 op 	{
267 35a31606 2021-09-22 op 		struct asr_query *q;
268 35a31606 2021-09-22 op 
269 35a31606 2021-09-22 op 		q = getrrsetbyname_async(hostname, DNS_RDATACLASS_IN,
270 35a31606 2021-09-22 op 		    DNS_RDATATYPE_TXT, 0, NULL);
271 35a31606 2021-09-22 op 		req->asrev = event_asr_run(q, offline_query_done, req);
272 0d40acb1 2021-07-24 op 	}
273 35a31606 2021-09-22 op #else
274 35a31606 2021-09-22 op 	{
275 35a31606 2021-09-22 op 		struct rrsetinfo *res;
276 35a31606 2021-09-22 op 
277 35a31606 2021-09-22 op 		if (getrrsetbyname(hostname, DNS_RDATACLASS_IN,
278 35a31606 2021-09-22 op 		    DNS_RDATATYPE_TXT, 0, &res))
279 35a31606 2021-09-22 op 			close_with_errf(req, "failed to connect to %s",
280 35a31606 2021-09-22 op 			    req->url.host);
281 35a31606 2021-09-22 op 		else
282 35a31606 2021-09-22 op 			offline_dns_done(res, req);
283 35a31606 2021-09-22 op 	}
284 35a31606 2021-09-22 op #endif
285 c8dba1e6 2021-07-24 op }
286 c8dba1e6 2021-07-24 op 
287 c8dba1e6 2021-07-24 op #if HAVE_ASR_RUN
288 c8dba1e6 2021-07-24 op static void
289 c8dba1e6 2021-07-24 op query_done(struct asr_result *res, void *d)
290 c8dba1e6 2021-07-24 op {
291 c8dba1e6 2021-07-24 op 	struct req	*req = d;
292 c8dba1e6 2021-07-24 op 
293 c8dba1e6 2021-07-24 op 	req->asrev = NULL;
294 c8dba1e6 2021-07-24 op 	if (res->ar_gai_errno != 0) {
295 c8dba1e6 2021-07-24 op 		close_with_errf(req, "failed to resolve %s: %s",
296 c8dba1e6 2021-07-24 op 		    req->url.host, gai_strerror(res->ar_gai_errno));
297 c8dba1e6 2021-07-24 op 		return;
298 c8dba1e6 2021-07-24 op 	}
299 c8dba1e6 2021-07-24 op 
300 c8dba1e6 2021-07-24 op 	req->fd = -1;
301 c8dba1e6 2021-07-24 op 	req->servinfo = res->ar_addrinfo;
302 c8dba1e6 2021-07-24 op 	req->p = res->ar_addrinfo;
303 c8dba1e6 2021-07-24 op 	try_to_connect(0, 0, req);
304 c8dba1e6 2021-07-24 op }
305 c8dba1e6 2021-07-24 op 
306 c8dba1e6 2021-07-24 op static void
307 c8dba1e6 2021-07-24 op async_conn_towards(struct req *req)
308 c8dba1e6 2021-07-24 op {
309 c8dba1e6 2021-07-24 op 	struct asr_query	*q;
310 c8dba1e6 2021-07-24 op 	const char		*proto = "1965";
311 c8dba1e6 2021-07-24 op 
312 c8dba1e6 2021-07-24 op 	if (*req->url.port != '\0')
313 c8dba1e6 2021-07-24 op 		proto = req->url.port;
314 c8dba1e6 2021-07-24 op 
315 c8dba1e6 2021-07-24 op 	req->hints.ai_family = AF_UNSPEC;
316 c8dba1e6 2021-07-24 op 	req->hints.ai_socktype = SOCK_STREAM;
317 c8dba1e6 2021-07-24 op 	q = getaddrinfo_async(req->url.host, proto, &req->hints, NULL);
318 c8dba1e6 2021-07-24 op 	req->asrev = event_asr_run(q, query_done, req);
319 c8dba1e6 2021-07-24 op }
320 35a31606 2021-09-22 op 
321 35a31606 2021-09-22 op static void
322 35a31606 2021-09-22 op offline_query_done(struct asr_result *res, void *d)
323 35a31606 2021-09-22 op {
324 35a31606 2021-09-22 op 	struct req		*req = d;
325 35a31606 2021-09-22 op 
326 35a31606 2021-09-22 op 	req->asrev = NULL;
327 35a31606 2021-09-22 op 	if (res->ar_rrset_errno != 0) {
328 35a31606 2021-09-22 op 		close_with_errf(req, "failed to resolve %s", req->url.host);
329 35a31606 2021-09-22 op 		return;
330 35a31606 2021-09-22 op 	}
331 35a31606 2021-09-22 op 
332 35a31606 2021-09-22 op 	offline_dns_done(res->ar_rrsetinfo, req);
333 35a31606 2021-09-22 op }
334 35a31606 2021-09-22 op 
335 c8dba1e6 2021-07-24 op #else
336 c8dba1e6 2021-07-24 op static void
337 c8dba1e6 2021-07-24 op blocking_conn_towards(struct req *req)
338 c8dba1e6 2021-07-24 op {
339 c8dba1e6 2021-07-24 op 	struct addrinfo	 hints;
340 c8dba1e6 2021-07-24 op 	struct phos_uri	*url = &req->url;
341 c8dba1e6 2021-07-24 op 	int		 status;
342 c8dba1e6 2021-07-24 op 	const char	*proto = "1965";
343 c8dba1e6 2021-07-24 op 
344 c8dba1e6 2021-07-24 op 	if (*url->port != '\0')
345 c8dba1e6 2021-07-24 op 		proto = url->port;
346 c8dba1e6 2021-07-24 op 
347 c8dba1e6 2021-07-24 op 	memset(&hints, 0, sizeof(hints));
348 c8dba1e6 2021-07-24 op 	hints.ai_family = AF_UNSPEC;
349 c8dba1e6 2021-07-24 op 	hints.ai_socktype = SOCK_STREAM;
350 c8dba1e6 2021-07-24 op 
351 c8dba1e6 2021-07-24 op 	if ((status = getaddrinfo(url->host, proto, &hints, &req->servinfo))) {
352 c8dba1e6 2021-07-24 op 		close_with_errf(req, "failed to resolve %s: %s",
353 c8dba1e6 2021-07-24 op 		    url->host, gai_strerror(status));
354 c8dba1e6 2021-07-24 op 		return;
355 c8dba1e6 2021-07-24 op 	}
356 c8dba1e6 2021-07-24 op 
357 c8dba1e6 2021-07-24 op 	req->fd = -1;
358 c8dba1e6 2021-07-24 op 	req->p = req->servinfo;
359 c8dba1e6 2021-07-24 op 	try_to_connect(0, 0, req);
360 c8dba1e6 2021-07-24 op }
361 c8dba1e6 2021-07-24 op #endif
362 c8dba1e6 2021-07-24 op 
363 c8dba1e6 2021-07-24 op static void
364 c8dba1e6 2021-07-24 op close_conn(int fd, short ev, void *d)
365 c8dba1e6 2021-07-24 op {
366 c8dba1e6 2021-07-24 op 	struct req	*req = d;
367 c8dba1e6 2021-07-24 op 
368 c8dba1e6 2021-07-24 op #if HAVE_ASR_RUN
369 c8dba1e6 2021-07-24 op 	if (req->asrev != NULL)
370 c8dba1e6 2021-07-24 op 		event_asr_abort(req->asrev);
371 c8dba1e6 2021-07-24 op #endif
372 c8dba1e6 2021-07-24 op 
373 5b762f01 2021-07-24 op 	if (req->bev != NULL) {
374 5b762f01 2021-07-24 op 		bufferevent_free(req->bev);
375 5b762f01 2021-07-24 op 		req->bev = NULL;
376 5b762f01 2021-07-24 op 	}
377 5b762f01 2021-07-24 op 
378 c8dba1e6 2021-07-24 op 	if (req->ctx != NULL) {
379 c8dba1e6 2021-07-24 op 		switch (tls_close(req->ctx)) {
380 c8dba1e6 2021-07-24 op 		case TLS_WANT_POLLIN:
381 c8dba1e6 2021-07-24 op 			yield_r(req, close_conn, NULL);
382 c8dba1e6 2021-07-24 op 			return;
383 c8dba1e6 2021-07-24 op 		case TLS_WANT_POLLOUT:
384 c8dba1e6 2021-07-24 op 			yield_w(req, close_conn, NULL);
385 c8dba1e6 2021-07-24 op 			return;
386 c8dba1e6 2021-07-24 op 		}
387 c8dba1e6 2021-07-24 op 
388 c8dba1e6 2021-07-24 op 		tls_free(req->ctx);
389 5b762f01 2021-07-24 op 		req->ctx = NULL;
390 c8dba1e6 2021-07-24 op 	}
391 c8dba1e6 2021-07-24 op 
392 c8dba1e6 2021-07-24 op 	TAILQ_REMOVE(&reqhead, req, reqs);
393 c8dba1e6 2021-07-24 op 	if (req->fd != -1)
394 c8dba1e6 2021-07-24 op 		close(req->fd);
395 c8dba1e6 2021-07-24 op 	free(req);
396 c8dba1e6 2021-07-24 op }
397 c8dba1e6 2021-07-24 op 
398 c8dba1e6 2021-07-24 op static void
399 c8dba1e6 2021-07-24 op close_with_err(struct req *req, const char *err)
400 c8dba1e6 2021-07-24 op {
401 c8dba1e6 2021-07-24 op 	net_send_ui(IMSG_ERR, req->id, err, strlen(err)+1);
402 c8dba1e6 2021-07-24 op 	close_conn(0, 0, req);
403 c8dba1e6 2021-07-24 op }
404 c8dba1e6 2021-07-24 op 
405 c8dba1e6 2021-07-24 op static void
406 c8dba1e6 2021-07-24 op close_with_errf(struct req *req, const char *fmt, ...)
407 c8dba1e6 2021-07-24 op {
408 c8dba1e6 2021-07-24 op 	va_list	 ap;
409 c8dba1e6 2021-07-24 op 	char	*s;
410 c8dba1e6 2021-07-24 op 
411 c8dba1e6 2021-07-24 op 	va_start(ap, fmt);
412 c8dba1e6 2021-07-24 op 	if (vasprintf(&s, fmt, ap) == -1)
413 c8dba1e6 2021-07-24 op 		abort();
414 c8dba1e6 2021-07-24 op 	va_end(ap);
415 c8dba1e6 2021-07-24 op 
416 c8dba1e6 2021-07-24 op 	close_with_err(req, s);
417 c8dba1e6 2021-07-24 op 	free(s);
418 c8dba1e6 2021-07-24 op }
419 c8dba1e6 2021-07-24 op 
420 c8dba1e6 2021-07-24 op static void
421 5b762f01 2021-07-24 op net_tls_handshake(int fd, short event, void *d)
422 c8dba1e6 2021-07-24 op {
423 c8dba1e6 2021-07-24 op 	struct req	*req = d;
424 c8dba1e6 2021-07-24 op 	const char	*hash;
425 c8dba1e6 2021-07-24 op 
426 5b762f01 2021-07-24 op 	if (event == EV_TIMEOUT) {
427 c8dba1e6 2021-07-24 op 		close_with_err(req, "Timeout loading page");
428 c8dba1e6 2021-07-24 op 		return;
429 c8dba1e6 2021-07-24 op 	}
430 c8dba1e6 2021-07-24 op 
431 c8dba1e6 2021-07-24 op 	switch (tls_handshake(req->ctx)) {
432 c8dba1e6 2021-07-24 op 	case TLS_WANT_POLLIN:
433 5b762f01 2021-07-24 op 		yield_r(req, net_tls_handshake, NULL);
434 c8dba1e6 2021-07-24 op 		return;
435 c8dba1e6 2021-07-24 op 	case TLS_WANT_POLLOUT:
436 5b762f01 2021-07-24 op 		yield_w(req, net_tls_handshake, NULL);
437 c8dba1e6 2021-07-24 op 		return;
438 c8dba1e6 2021-07-24 op 	}
439 c8dba1e6 2021-07-24 op 
440 c8dba1e6 2021-07-24 op 	hash = tls_peer_cert_hash(req->ctx);
441 c8dba1e6 2021-07-24 op 	if (hash == NULL) {
442 bfb9acb0 2021-07-24 op 		close_with_errf(req, "handshake failed: %s",
443 bfb9acb0 2021-07-24 op 		    tls_error(req->ctx));
444 c8dba1e6 2021-07-24 op 		return;
445 c8dba1e6 2021-07-24 op 	}
446 c8dba1e6 2021-07-24 op 	net_send_ui(IMSG_CHECK_CERT, req->id, hash, strlen(hash)+1);
447 c8dba1e6 2021-07-24 op }
448 c8dba1e6 2021-07-24 op 
449 c8dba1e6 2021-07-24 op static void
450 5b762f01 2021-07-24 op net_tls_readcb(int fd, short event, void *d)
451 c8dba1e6 2021-07-24 op {
452 5b762f01 2021-07-24 op 	struct bufferevent	*bufev = d;
453 5b762f01 2021-07-24 op 	struct req		*req = bufev->cbarg;
454 69f631d9 2021-07-25 op 	char			 buf[IBUF_READ_SIZE];
455 5b762f01 2021-07-24 op 	int			 what = EVBUFFER_READ;
456 5b762f01 2021-07-24 op 	int			 howmuch = IBUF_READ_SIZE;
457 1470eab3 2021-07-27 op 	int			 res;
458 5b762f01 2021-07-24 op 	ssize_t			 ret;
459 5b762f01 2021-07-24 op 	size_t			 len;
460 c8dba1e6 2021-07-24 op 
461 5b762f01 2021-07-24 op 	if (event == EV_TIMEOUT) {
462 5b762f01 2021-07-24 op 		what |= EVBUFFER_TIMEOUT;
463 5b762f01 2021-07-24 op 		goto err;
464 5b762f01 2021-07-24 op 	}
465 5b762f01 2021-07-24 op 
466 5b762f01 2021-07-24 op 	if (bufev->wm_read.high != 0)
467 5b762f01 2021-07-24 op 		howmuch = MIN(sizeof(buf), bufev->wm_read.high);
468 5b762f01 2021-07-24 op 
469 5b762f01 2021-07-24 op 	switch (ret = tls_read(req->ctx, buf, howmuch)) {
470 c8dba1e6 2021-07-24 op 	case TLS_WANT_POLLIN:
471 c8dba1e6 2021-07-24 op 	case TLS_WANT_POLLOUT:
472 5b762f01 2021-07-24 op 		goto retry;
473 5b762f01 2021-07-24 op 	case -1:
474 5b762f01 2021-07-24 op 		what |= EVBUFFER_ERROR;
475 5b762f01 2021-07-24 op 		goto err;
476 c8dba1e6 2021-07-24 op 	}
477 5b762f01 2021-07-24 op 	len = ret;
478 c8dba1e6 2021-07-24 op 
479 5b762f01 2021-07-24 op 	if (len == 0) {
480 5b762f01 2021-07-24 op 		what |= EVBUFFER_EOF;
481 5b762f01 2021-07-24 op 		goto err;
482 5b762f01 2021-07-24 op 	}
483 c8dba1e6 2021-07-24 op 
484 1470eab3 2021-07-27 op 	res = evbuffer_add(bufev->input, buf, len);
485 1470eab3 2021-07-27 op 	if (res == -1) {
486 5b762f01 2021-07-24 op 		what |= EVBUFFER_ERROR;
487 5b762f01 2021-07-24 op 		goto err;
488 c8dba1e6 2021-07-24 op 	}
489 5b762f01 2021-07-24 op 
490 5b762f01 2021-07-24 op 	event_add(&bufev->ev_read, NULL);
491 5b762f01 2021-07-24 op 
492 5b762f01 2021-07-24 op 	len = EVBUFFER_LENGTH(bufev->input);
493 5b762f01 2021-07-24 op 	if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
494 5b762f01 2021-07-24 op 		return;
495 5b762f01 2021-07-24 op 
496 5b762f01 2021-07-24 op 	if (bufev->readcb != NULL)
497 5b762f01 2021-07-24 op 		(*bufev->readcb)(bufev, bufev->cbarg);
498 5b762f01 2021-07-24 op 	return;
499 5b762f01 2021-07-24 op 
500 5b762f01 2021-07-24 op retry:
501 5b762f01 2021-07-24 op 	event_add(&bufev->ev_read, NULL);
502 5b762f01 2021-07-24 op 	return;
503 5b762f01 2021-07-24 op 
504 5b762f01 2021-07-24 op err:
505 5b762f01 2021-07-24 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
506 c8dba1e6 2021-07-24 op }
507 c8dba1e6 2021-07-24 op 
508 c8dba1e6 2021-07-24 op static void
509 5b762f01 2021-07-24 op net_tls_writecb(int fd, short event, void *d)
510 c8dba1e6 2021-07-24 op {
511 5b762f01 2021-07-24 op 	struct bufferevent	*bufev = d;
512 5b762f01 2021-07-24 op 	struct req		*req = bufev->cbarg;
513 5b762f01 2021-07-24 op 	ssize_t			 ret;
514 5b762f01 2021-07-24 op 	size_t			 len;
515 5b762f01 2021-07-24 op 	short			 what = EVBUFFER_WRITE;
516 c8dba1e6 2021-07-24 op 
517 5b762f01 2021-07-24 op 	if (event & EV_TIMEOUT) {
518 5b762f01 2021-07-24 op 		what |= EVBUFFER_TIMEOUT;
519 c8dba1e6 2021-07-24 op 		goto err;
520 5b762f01 2021-07-24 op 	}
521 c8dba1e6 2021-07-24 op 
522 5b762f01 2021-07-24 op 	if (EVBUFFER_LENGTH(bufev->output) != 0) {
523 5b762f01 2021-07-24 op 		ret = tls_write(req->ctx, EVBUFFER_DATA(bufev->output),
524 5b762f01 2021-07-24 op 		    EVBUFFER_LENGTH(bufev->output));
525 5b762f01 2021-07-24 op 		switch (ret) {
526 5b762f01 2021-07-24 op 		case TLS_WANT_POLLIN:
527 5b762f01 2021-07-24 op 		case TLS_WANT_POLLOUT:
528 5b762f01 2021-07-24 op 			goto retry;
529 5b762f01 2021-07-24 op 		case -1:
530 5b762f01 2021-07-24 op 			what |= EVBUFFER_ERROR;
531 5b762f01 2021-07-24 op 			goto err;
532 5b762f01 2021-07-24 op 		}
533 5b762f01 2021-07-24 op 		len = ret;
534 4c7d2b8f 2021-07-27 op 
535 5b762f01 2021-07-24 op 		evbuffer_drain(bufev->output, len);
536 5b762f01 2021-07-24 op 	}
537 c8dba1e6 2021-07-24 op 
538 5b762f01 2021-07-24 op 	if (EVBUFFER_LENGTH(bufev->output) != 0)
539 5b762f01 2021-07-24 op 		event_add(&bufev->ev_write, NULL);
540 c8dba1e6 2021-07-24 op 
541 5b762f01 2021-07-24 op 	if (bufev->writecb != NULL &&
542 5b762f01 2021-07-24 op 	    EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
543 5b762f01 2021-07-24 op 		(*bufev->writecb)(bufev, bufev->cbarg);
544 5b762f01 2021-07-24 op 	return;
545 c8dba1e6 2021-07-24 op 
546 5b762f01 2021-07-24 op retry:
547 5b762f01 2021-07-24 op 	event_add(&bufev->ev_write, NULL);
548 5b762f01 2021-07-24 op 	return;
549 5b762f01 2021-07-24 op 
550 5b762f01 2021-07-24 op err:
551 5b762f01 2021-07-24 op 	(*bufev->errorcb)(bufev, what, bufev->cbarg);
552 5b762f01 2021-07-24 op }
553 5b762f01 2021-07-24 op 
554 5b762f01 2021-07-24 op static int
555 5b762f01 2021-07-24 op gemini_parse_reply(struct req *req, const char *header, size_t len)
556 5b762f01 2021-07-24 op {
557 5b762f01 2021-07-24 op 	int		 code;
558 5b762f01 2021-07-24 op 	const char	*t;
559 5b762f01 2021-07-24 op 
560 5b762f01 2021-07-24 op 	if (len < 4)
561 5b762f01 2021-07-24 op 		return 0;
562 5b762f01 2021-07-24 op 
563 5b762f01 2021-07-24 op 	if (!isdigit(header[0]) || !isdigit(header[1]))
564 5b762f01 2021-07-24 op 		return 0;
565 5b762f01 2021-07-24 op 
566 5b762f01 2021-07-24 op 	code = (header[0] - '0')*10 + (header[1] - '0');
567 5b762f01 2021-07-24 op 	if (header[2] != ' ')
568 5b762f01 2021-07-24 op 		return 0;
569 5b762f01 2021-07-24 op 
570 5b762f01 2021-07-24 op 	t = header + 3;
571 5b762f01 2021-07-24 op 
572 5b762f01 2021-07-24 op 	net_send_ui(IMSG_GOT_CODE, req->id, &code, sizeof(code));
573 5b762f01 2021-07-24 op 	net_send_ui(IMSG_GOT_META, req->id, t, strlen(t)+1);
574 5b762f01 2021-07-24 op 
575 5b762f01 2021-07-24 op 	bufferevent_disable(req->bev, EV_READ|EV_WRITE);
576 5b762f01 2021-07-24 op 
577 5b762f01 2021-07-24 op 	if (code < 20 || code >= 30)
578 c8dba1e6 2021-07-24 op 		close_conn(0, 0, req);
579 5b762f01 2021-07-24 op 	return 1;
580 5b762f01 2021-07-24 op }
581 c8dba1e6 2021-07-24 op 
582 5b762f01 2021-07-24 op /* called when we're ready to read/write */
583 5b762f01 2021-07-24 op static void
584 5b762f01 2021-07-24 op net_ready(struct req *req)
585 5b762f01 2021-07-24 op {
586 5b762f01 2021-07-24 op 	req->bev = bufferevent_new(req->fd, net_read, net_write, net_error,
587 5b762f01 2021-07-24 op 	    req);
588 5b762f01 2021-07-24 op 	if (req->bev == NULL)
589 5b762f01 2021-07-24 op 		die();
590 5b762f01 2021-07-24 op 
591 b5c5e56d 2021-07-28 op #if HAVE_EVENT2
592 73bb7969 2021-07-27 op 	evbuffer_unfreeze(req->bev->input, 0);
593 73bb7969 2021-07-27 op 	evbuffer_unfreeze(req->bev->output, 1);
594 b5c5e56d 2021-07-28 op #endif
595 73bb7969 2021-07-27 op 
596 5b762f01 2021-07-24 op 	/* setup tls i/o layer */
597 5b762f01 2021-07-24 op 	if (req->ctx != NULL) {
598 5b762f01 2021-07-24 op 		event_set(&req->bev->ev_read, req->fd, EV_READ,
599 5b762f01 2021-07-24 op 		    net_tls_readcb, req->bev);
600 5b762f01 2021-07-24 op 		event_set(&req->bev->ev_write, req->fd, EV_WRITE,
601 5b762f01 2021-07-24 op 		    net_tls_writecb, req->bev);
602 5b762f01 2021-07-24 op 	}
603 5b762f01 2021-07-24 op 
604 5b762f01 2021-07-24 op 	/* TODO: adjust watermarks */
605 5b762f01 2021-07-24 op 	bufferevent_setwatermark(req->bev, EV_WRITE, 1, 0);
606 5b762f01 2021-07-24 op 	bufferevent_setwatermark(req->bev, EV_READ,  1, 0);
607 5b762f01 2021-07-24 op 
608 5b762f01 2021-07-24 op 	bufferevent_enable(req->bev, EV_READ|EV_WRITE);
609 5b762f01 2021-07-24 op 
610 5b762f01 2021-07-24 op 	bufferevent_write(req->bev, req->req, req->len);
611 5b762f01 2021-07-24 op }
612 5b762f01 2021-07-24 op 
613 5b762f01 2021-07-24 op /* called after a read has been done */
614 5b762f01 2021-07-24 op static void
615 5b762f01 2021-07-24 op net_read(struct bufferevent *bev, void *d)
616 5b762f01 2021-07-24 op {
617 5b762f01 2021-07-24 op 	struct req	*req = d;
618 5b762f01 2021-07-24 op 	struct evbuffer	*src = EVBUFFER_INPUT(bev);
619 5b762f01 2021-07-24 op 	void		*data;
620 0e8f22ed 2021-09-12 op 	size_t		 len, chunk;
621 5b762f01 2021-07-24 op 	int		 r;
622 5b762f01 2021-07-24 op 	char		*header;
623 5b762f01 2021-07-24 op 
624 5b762f01 2021-07-24 op 	if (!req->done_header) {
625 5b762f01 2021-07-24 op 		header = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF_STRICT);
626 5b762f01 2021-07-24 op 		if (header == NULL && EVBUFFER_LENGTH(src) >= 1024)
627 5b762f01 2021-07-24 op 			goto err;
628 5b762f01 2021-07-24 op 		if (header == NULL)
629 5b762f01 2021-07-24 op 			return;
630 5b762f01 2021-07-24 op 		r = gemini_parse_reply(req, header, len);
631 5b762f01 2021-07-24 op 		free(header);
632 5b762f01 2021-07-24 op 		if (!r)
633 5b762f01 2021-07-24 op 			goto err;
634 5b762f01 2021-07-24 op 		req->done_header = 1;
635 5b762f01 2021-07-24 op 		return;
636 5b762f01 2021-07-24 op 	}
637 5b762f01 2021-07-24 op 
638 5b762f01 2021-07-24 op 	if ((len = EVBUFFER_LENGTH(src)) == 0)
639 5b762f01 2021-07-24 op 		return;
640 5b762f01 2021-07-24 op 	data = EVBUFFER_DATA(src);
641 0e8f22ed 2021-09-12 op 
642 0e8f22ed 2021-09-12 op 	/*
643 0e8f22ed 2021-09-12 op 	 * Split data into chunks before sending.  imsg can't handle
644 0e8f22ed 2021-09-12 op 	 * message that are "too big".
645 0e8f22ed 2021-09-12 op 	 */
646 0e8f22ed 2021-09-12 op 	while (len != 0) {
647 0e8f22ed 2021-09-12 op 		chunk = MIN(len, 4096);
648 0e8f22ed 2021-09-12 op 		net_send_ui(IMSG_BUF, req->id, data, chunk);
649 0e8f22ed 2021-09-12 op 		data += chunk;
650 0e8f22ed 2021-09-12 op 		len -= chunk;
651 0e8f22ed 2021-09-12 op 	}
652 0e8f22ed 2021-09-12 op 
653 0e8f22ed 2021-09-12 op 	evbuffer_drain(src, EVBUFFER_LENGTH(src));
654 c8dba1e6 2021-07-24 op 	return;
655 c8dba1e6 2021-07-24 op 
656 c8dba1e6 2021-07-24 op err:
657 5b762f01 2021-07-24 op 	(*bev->errorcb)(bev, EVBUFFER_READ, bev->cbarg);
658 c8dba1e6 2021-07-24 op }
659 c8dba1e6 2021-07-24 op 
660 5b762f01 2021-07-24 op /* called after a write has been done */
661 c8dba1e6 2021-07-24 op static void
662 5b762f01 2021-07-24 op net_write(struct bufferevent *bev, void *d)
663 c8dba1e6 2021-07-24 op {
664 5b762f01 2021-07-24 op 	struct evbuffer	*dst = EVBUFFER_OUTPUT(bev);
665 5b762f01 2021-07-24 op 
666 5b762f01 2021-07-24 op 	if (EVBUFFER_LENGTH(dst) == 0)
667 5b762f01 2021-07-24 op 		(*bev->errorcb)(bev, EVBUFFER_WRITE, bev->cbarg);
668 5b762f01 2021-07-24 op }
669 5b762f01 2021-07-24 op 
670 5b762f01 2021-07-24 op static void
671 5b762f01 2021-07-24 op net_error(struct bufferevent *bev, short error, void *d)
672 5b762f01 2021-07-24 op {
673 c8dba1e6 2021-07-24 op 	struct req	*req = d;
674 590c00b7 2021-07-25 op 	struct evbuffer	*src;
675 c8dba1e6 2021-07-24 op 
676 5b762f01 2021-07-24 op 	if (error & EVBUFFER_TIMEOUT) {
677 5b762f01 2021-07-24 op 		close_with_err(req, "Timeout loading page");
678 5b762f01 2021-07-24 op 		return;
679 5b762f01 2021-07-24 op 	}
680 c8dba1e6 2021-07-24 op 
681 5b762f01 2021-07-24 op 	if (error & EVBUFFER_ERROR) {
682 5b762f01 2021-07-24 op 		close_with_err(req, "buffer event error");
683 5b762f01 2021-07-24 op 		return;
684 c8dba1e6 2021-07-24 op 	}
685 5b762f01 2021-07-24 op 
686 5b762f01 2021-07-24 op 	if (error & EVBUFFER_EOF) {
687 95a8c791 2021-08-26 op 		src = EVBUFFER_INPUT(req->bev);
688 590c00b7 2021-07-25 op 		if (EVBUFFER_LENGTH(src) != 0)
689 590c00b7 2021-07-25 op 			net_send_ui(IMSG_BUF, req->id, EVBUFFER_DATA(src),
690 590c00b7 2021-07-25 op 			    EVBUFFER_LENGTH(src));
691 5b762f01 2021-07-24 op 		net_send_ui(IMSG_EOF, req->id, NULL, 0);
692 5b762f01 2021-07-24 op 		close_conn(0, 0, req);
693 5b762f01 2021-07-24 op 		return;
694 5b762f01 2021-07-24 op 	}
695 5b762f01 2021-07-24 op 
696 5b762f01 2021-07-24 op 	if (error & EVBUFFER_WRITE) {
697 5b762f01 2021-07-24 op 		/* finished sending request */
698 5b762f01 2021-07-24 op 		bufferevent_disable(bev, EV_WRITE);
699 5b762f01 2021-07-24 op 		return;
700 5b762f01 2021-07-24 op 	}
701 5b762f01 2021-07-24 op 
702 5b762f01 2021-07-24 op 	if (error & EVBUFFER_READ) {
703 5b762f01 2021-07-24 op 		close_with_err(req, "protocol error");
704 5b762f01 2021-07-24 op 		return;
705 5b762f01 2021-07-24 op 	}
706 5b762f01 2021-07-24 op 
707 5b762f01 2021-07-24 op 	close_with_errf(req, "unknown event error %x", error);
708 c8dba1e6 2021-07-24 op }
709 c8dba1e6 2021-07-24 op 
710 c8dba1e6 2021-07-24 op static void
711 c8dba1e6 2021-07-24 op handle_get_raw(struct imsg *imsg, size_t datalen)
712 c8dba1e6 2021-07-24 op {
713 c8dba1e6 2021-07-24 op 	struct req	*req;
714 c8dba1e6 2021-07-24 op 	struct get_req	*r;
715 c8dba1e6 2021-07-24 op 
716 c8dba1e6 2021-07-24 op 	r = imsg->data;
717 c8dba1e6 2021-07-24 op 
718 c8dba1e6 2021-07-24 op 	if (datalen != sizeof(*r))
719 c8dba1e6 2021-07-24 op 		die();
720 c8dba1e6 2021-07-24 op 
721 c8dba1e6 2021-07-24 op 	if ((req = calloc(1, sizeof(*req))) == NULL)
722 c8dba1e6 2021-07-24 op 		die();
723 c8dba1e6 2021-07-24 op 
724 c8dba1e6 2021-07-24 op 	req->id = imsg->hdr.peerid;
725 c8dba1e6 2021-07-24 op 	TAILQ_INSERT_HEAD(&reqhead, req, reqs);
726 c8dba1e6 2021-07-24 op 
727 c8dba1e6 2021-07-24 op 	strlcpy(req->url.host, r->host, sizeof(req->url.host));
728 c8dba1e6 2021-07-24 op 	strlcpy(req->url.port, r->port, sizeof(req->url.port));
729 c8dba1e6 2021-07-24 op 
730 5b762f01 2021-07-24 op 	strlcpy(req->req, r->req, sizeof(req->req));
731 5b762f01 2021-07-24 op 	req->len = strlen(r->req);
732 5b762f01 2021-07-24 op 
733 5fb52fc0 2021-07-25 op 	req->proto = r->proto;
734 5fb52fc0 2021-07-25 op 
735 c8dba1e6 2021-07-24 op #if HAVE_ASR_RUN
736 95a8c791 2021-08-26 op 	async_conn_towards(req);
737 c8dba1e6 2021-07-24 op #else
738 c8dba1e6 2021-07-24 op 	blocking_conn_towards(req);
739 c8dba1e6 2021-07-24 op #endif
740 c8dba1e6 2021-07-24 op }
741 c8dba1e6 2021-07-24 op 
742 c8dba1e6 2021-07-24 op static void
743 c8dba1e6 2021-07-24 op handle_cert_status(struct imsg *imsg, size_t datalen)
744 c8dba1e6 2021-07-24 op {
745 c8dba1e6 2021-07-24 op 	struct req	*req;
746 c8dba1e6 2021-07-24 op 	int		 is_ok;
747 c8dba1e6 2021-07-24 op 
748 c8dba1e6 2021-07-24 op 	req = req_by_id(imsg->hdr.peerid);
749 c8dba1e6 2021-07-24 op 
750 c8dba1e6 2021-07-24 op 	if (datalen < sizeof(is_ok))
751 c8dba1e6 2021-07-24 op 		die();
752 c8dba1e6 2021-07-24 op 	memcpy(&is_ok, imsg->data, sizeof(is_ok));
753 c8dba1e6 2021-07-24 op 
754 c8dba1e6 2021-07-24 op 	if (is_ok)
755 5b762f01 2021-07-24 op 		net_ready(req);
756 c8dba1e6 2021-07-24 op 	else
757 c8dba1e6 2021-07-24 op 		close_conn(0, 0, req);
758 c8dba1e6 2021-07-24 op }
759 c8dba1e6 2021-07-24 op 
760 c8dba1e6 2021-07-24 op static void
761 c8dba1e6 2021-07-24 op handle_proceed(struct imsg *imsg, size_t datalen)
762 c8dba1e6 2021-07-24 op {
763 5b762f01 2021-07-24 op 	struct req *req;
764 5b762f01 2021-07-24 op 
765 318a6c1a 2021-07-24 op 	if ((req = req_by_id(imsg->hdr.peerid)) == NULL)
766 5b762f01 2021-07-24 op 		return;
767 5b762f01 2021-07-24 op 
768 5b762f01 2021-07-24 op 	bufferevent_enable(req->bev, EV_READ);
769 c8dba1e6 2021-07-24 op }
770 c8dba1e6 2021-07-24 op 
771 c8dba1e6 2021-07-24 op static void
772 c8dba1e6 2021-07-24 op handle_stop(struct imsg *imsg, size_t datalen)
773 c8dba1e6 2021-07-24 op {
774 c8dba1e6 2021-07-24 op 	struct req	*req;
775 c8dba1e6 2021-07-24 op 
776 318a6c1a 2021-07-24 op 	if ((req = req_by_id(imsg->hdr.peerid)) == NULL)
777 c8dba1e6 2021-07-24 op 		return;
778 c8dba1e6 2021-07-24 op 	close_conn(0, 0, req);
779 c8dba1e6 2021-07-24 op }
780 c8dba1e6 2021-07-24 op 
781 c8dba1e6 2021-07-24 op static void
782 c8dba1e6 2021-07-24 op handle_quit(struct imsg *imsg, size_t datalen)
783 c8dba1e6 2021-07-24 op {
784 c8dba1e6 2021-07-24 op 	event_loopbreak();
785 c8dba1e6 2021-07-24 op }
786 c8dba1e6 2021-07-24 op 
787 c8dba1e6 2021-07-24 op static void
788 c8dba1e6 2021-07-24 op handle_dispatch_imsg(int fd, short ev, void *d)
789 c8dba1e6 2021-07-24 op {
790 c8dba1e6 2021-07-24 op 	struct imsgev	*iev = d;
791 c8dba1e6 2021-07-24 op 
792 c8dba1e6 2021-07-24 op 	if (dispatch_imsg(iev, ev, handlers, sizeof(handlers)) == -1)
793 c8dba1e6 2021-07-24 op 		err(1, "connection closed");
794 c8dba1e6 2021-07-24 op }
795 c8dba1e6 2021-07-24 op 
796 c8dba1e6 2021-07-24 op static int
797 c8dba1e6 2021-07-24 op net_send_ui(int type, uint32_t peerid, const void *data,
798 c8dba1e6 2021-07-24 op     uint16_t datalen)
799 c8dba1e6 2021-07-24 op {
800 c8dba1e6 2021-07-24 op 	return imsg_compose_event(iev_ui, type, peerid, 0, -1,
801 c8dba1e6 2021-07-24 op 	    data, datalen);
802 c8dba1e6 2021-07-24 op }
803 c8dba1e6 2021-07-24 op 
804 c8dba1e6 2021-07-24 op int
805 f45bd2e3 2021-07-24 op net_main(void)
806 c8dba1e6 2021-07-24 op {
807 c8dba1e6 2021-07-24 op 	setproctitle("net");
808 c8dba1e6 2021-07-24 op 
809 c8dba1e6 2021-07-24 op 	TAILQ_INIT(&reqhead);
810 c8dba1e6 2021-07-24 op 
811 c8dba1e6 2021-07-24 op 	if ((tlsconf = tls_config_new()) == NULL)
812 c8dba1e6 2021-07-24 op 		die();
813 c8dba1e6 2021-07-24 op 	tls_config_insecure_noverifycert(tlsconf);
814 c8dba1e6 2021-07-24 op 	tls_config_insecure_noverifyname(tlsconf);
815 c8dba1e6 2021-07-24 op 
816 c8dba1e6 2021-07-24 op 	event_init();
817 c8dba1e6 2021-07-24 op 
818 c8dba1e6 2021-07-24 op 	/* Setup pipe and event handler to the main process */
819 c8dba1e6 2021-07-24 op 	if ((iev_ui = malloc(sizeof(*iev_ui))) == NULL)
820 c8dba1e6 2021-07-24 op 		die();
821 c8dba1e6 2021-07-24 op 	imsg_init(&iev_ui->ibuf, 3);
822 c8dba1e6 2021-07-24 op 	iev_ui->handler = handle_dispatch_imsg;
823 c8dba1e6 2021-07-24 op 	iev_ui->events = EV_READ;
824 c8dba1e6 2021-07-24 op 	event_set(&iev_ui->ev, iev_ui->ibuf.fd, iev_ui->events,
825 c8dba1e6 2021-07-24 op 	    iev_ui->handler, iev_ui);
826 c8dba1e6 2021-07-24 op 	event_add(&iev_ui->ev, NULL);
827 c8dba1e6 2021-07-24 op 
828 c8dba1e6 2021-07-24 op 	sandbox_net_process();
829 c8dba1e6 2021-07-24 op 
830 c8dba1e6 2021-07-24 op 	event_dispatch();
831 74c0f6ba 2021-07-24 op 
832 74c0f6ba 2021-07-24 op 	tls_config_free(tlsconf);
833 74c0f6ba 2021-07-24 op 	msgbuf_clear(&iev_ui->ibuf.w);
834 74c0f6ba 2021-07-24 op 	close(iev_ui->ibuf.fd);
835 74c0f6ba 2021-07-24 op 	free(iev_ui);
836 74c0f6ba 2021-07-24 op 
837 c8dba1e6 2021-07-24 op 	return 0;
838 c8dba1e6 2021-07-24 op }