1 /*
2  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include "compat.h"
18 
19 #include <sys/socket.h>
20 #include <sys/un.h>
21 #include <sys/wait.h>
22 
23 #include <errno.h>
24 #include <getopt.h>
25 #include <signal.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <unistd.h>
30 
31 #include "control.h"
32 #include "defaults.h"
33 #include "fs.h"
34 #include "mcache.h"
35 #include "minibuffer.h"
36 #include "parser.h"
37 #include "session.h"
38 #include "telescope.h"
39 #include "ui.h"
40 #include "utils.h"
41 
42 static const struct option longopts[] = {
43 	{"colors",      no_argument,    NULL,   'C'},
44 	{"colours",	no_argument,    NULL,   'C'},
45 	{"help",	no_argument,	NULL,	'h'},
46 	{"safe",	no_argument,	NULL,	'S'},
47 	{"version",	no_argument,	NULL,	'v'},
48 	{NULL,		0,		NULL,	0},
49 };
50 
51 static const char *opts = "Cc:hnST:v";
52 
53 static int	has_url;
54 static char	url[GEMINI_URL_LEN];
55 
56 /*
57  * Used to know when we're finished loading.
58  */
59 int			 operating;
60 
61 /*
62  * "Safe" (or "sandobox") mode.  If enabled, Telescope shouldn't write
63  * anything to the filesystem or execute external programs.
64  */
65 int			safe_mode;
66 
67 static struct imsgev	*iev_fs, *iev_net;
68 
69 struct tabshead		 tabshead = TAILQ_HEAD_INITIALIZER(tabshead);
70 struct tabshead		 ktabshead = TAILQ_HEAD_INITIALIZER(ktabshead);
71 struct proxylist	 proxies = TAILQ_HEAD_INITIALIZER(proxies);
72 
73 enum telescope_process {
74 	PROC_UI,
75 	PROC_FS,
76 	PROC_NET,
77 };
78 
79 #define CANNOT_FETCH		0
80 #define TOO_MUCH_REDIRECTS	1
81 #define MALFORMED_RESPONSE	2
82 #define UNKNOWN_TYPE_OR_CSET	3
83 #define UNKNOWN_PROTOCOL	4
84 
85 /* XXX: keep in sync with normalize_code */
86 const char *err_pages[70] = {
87 	[CANNOT_FETCH]		= "# Couldn't load the page\n",
88 	[TOO_MUCH_REDIRECTS]	= "# Too much redirects\n",
89 	[MALFORMED_RESPONSE]	= "# Got a malformed response\n",
90 	[UNKNOWN_TYPE_OR_CSET]	= "# Unsupported type or charset\n",
91 	[UNKNOWN_PROTOCOL]	= "# Unknown protocol\n",
92 
93 	[10] = "# Input required\n",
94 	[11] = "# Input required\n",
95 	[40] = "# Temporary failure\n",
96 	[41] = "# Server unavailable\n",
97 	[42] = "# CGI error\n",
98 	[43] = "# Proxy error\n",
99 	[44] = "# Slow down\n",
100 	[50] = "# Permanent failure\n",
101 	[51] = "# Not found\n",
102 	[52] = "# Gone\n",
103 	[53] = "# Proxy request refused\n",
104 	[59] = "# Bad request\n",
105 	[60] = "# Client certificate required\n",
106 	[61] = "# Certificate not authorised\n",
107 	[62] = "# Certificate not valid\n"
108 };
109 
110 static void		 die(void) __attribute__((__noreturn__));
111 static struct tab	*tab_by_id(uint32_t);
112 static void		 handle_imsg_err(struct imsg *, size_t);
113 static void		 handle_imsg_check_cert(struct imsg *, size_t);
114 static void		 handle_check_cert_user_choice(int, struct tab *);
115 static void		 handle_maybe_save_new_cert(int, struct tab *);
116 static void		 handle_imsg_got_code(struct imsg *, size_t);
117 static void		 handle_imsg_got_meta(struct imsg *, size_t);
118 static void		 handle_maybe_save_page(int, struct tab *);
119 static void		 handle_save_page_path(const char *, struct tab *);
120 static void		 handle_imsg_file_opened(struct imsg *, size_t);
121 static void		 handle_imsg_buf(struct imsg *, size_t);
122 static void		 handle_imsg_eof(struct imsg *, size_t);
123 static void		 handle_imsg_tofu(struct imsg *, size_t);
124 static void		 handle_imsg_bookmark_ok(struct imsg *, size_t);
125 static void		 handle_imsg_save_cert_ok(struct imsg *, size_t);
126 static void		 handle_imsg_update_cert_ok(struct imsg *, size_t);
127 static void		 handle_imsg_session(struct imsg *, size_t);
128 static void		 handle_imsg_history(struct imsg *, size_t);
129 static void		 handle_dispatch_imsg(int, short, void *);
130 static int		 load_about_url(struct tab *, const char *);
131 static int		 load_file_url(struct tab *, const char *);
132 static int		 load_finger_url(struct tab *, const char *);
133 static int		 load_gemini_url(struct tab *, const char *);
134 static int		 load_gopher_url(struct tab *, const char *);
135 static int		 load_via_proxy(struct tab *, const char *,
136 			     struct proxy *);
137 static int		 make_request(struct tab *, struct get_req *, int,
138 			     const char *);
139 static int		 make_fs_request(struct tab *, int, const char *);
140 static int		 do_load_url(struct tab *, const char *, const char *, int);
141 static pid_t		 start_child(enum telescope_process, const char *, int);
142 static void		 send_url(const char *);
143 
144 static const struct proto {
145 	const char	*schema;
146 	const char	*port;
147 	int		 (*loadfn)(struct tab *, const char *);
148 } protos[] = {
149 	{"about",	NULL,	load_about_url},
150 	{"file",	NULL,	load_file_url},
151 	{"finger",	"79",	load_finger_url},
152 	{"gemini",	"1965",	load_gemini_url},
153 	{"gopher",	"70",	load_gopher_url},
154 	{NULL, NULL, NULL},
155 };
156 
157 static imsg_handlerfn *handlers[] = {
158 	[IMSG_ERR] = handle_imsg_err,
159 	[IMSG_CHECK_CERT] = handle_imsg_check_cert,
160 	[IMSG_GOT_CODE] = handle_imsg_got_code,
161 	[IMSG_GOT_META] = handle_imsg_got_meta,
162 	[IMSG_BUF] = handle_imsg_buf,
163 	[IMSG_EOF] = handle_imsg_eof,
164 	[IMSG_TOFU] = handle_imsg_tofu,
165 	[IMSG_BOOKMARK_OK] = handle_imsg_bookmark_ok,
166 	[IMSG_SAVE_CERT_OK] = handle_imsg_save_cert_ok,
167 	[IMSG_UPDATE_CERT_OK] = handle_imsg_update_cert_ok,
168 	[IMSG_FILE_OPENED] = handle_imsg_file_opened,
169 	[IMSG_SESSION_TAB] = handle_imsg_session,
170 	[IMSG_SESSION_TAB_HIST] = handle_imsg_session,
171 	[IMSG_SESSION_END] = handle_imsg_session,
172 	[IMSG_HIST_ITEM] = handle_imsg_history,
173 	[IMSG_HIST_END] = handle_imsg_history,
174 };
175 
176 static struct ohash	certs;
177 
178 static void __attribute__((__noreturn__))
179 die(void)
180 {
181 	abort(); 		/* TODO */
182 }
183 
184 static struct tab *
185 tab_by_id(uint32_t id)
186 {
187 	struct tab *t;
188 
189 	TAILQ_FOREACH(t, &tabshead, tabs) {
190 		if (t->id == id)
191 			return t;
192 	}
193 
194 	return NULL;
195 }
196 
197 static void
198 handle_imsg_err(struct imsg *imsg, size_t datalen)
199 {
200 	struct tab	*tab;
201 	char		*page;
202 
203 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
204 		return;
205 
206 	page = imsg->data;
207 	page[datalen-1] = '\0';
208 
209 	if (asprintf(&page, "# Error loading %s\n\n> %s\n",
210 	    tab->hist_cur->h, page) == -1)
211 		die();
212 	load_page_from_str(tab, page);
213 	free(page);
214 }
215 
216 static void
217 handle_imsg_check_cert(struct imsg *imsg, size_t datalen)
218 {
219 	const char		*hash, *host, *port;
220 	int			 tofu_res;
221 	struct tofu_entry	*e;
222 	struct tab		*tab;
223 
224 	hash = imsg->data;
225 	if (hash[datalen-1] != '\0')
226 		abort();
227 
228 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
229 		return;
230 
231 	if (tab->proxy != NULL) {
232 		host = tab->proxy->host;
233 		port = tab->proxy->port;
234 	} else {
235 		host = tab->uri.host;
236 		port = tab->uri.port;
237 	}
238 
239 	if ((e = tofu_lookup(&certs, host, port)) == NULL) {
240 		/* TODO: an update in libressl/libretls changed
241 		 * significantly.  Find a better approach at storing
242 		 * the certs! */
243 		if (datalen > sizeof(e->hash))
244 			abort();
245 
246 		tofu_res = 1;	/* trust on first use */
247 		if ((e = calloc(1, sizeof(*e))) == NULL)
248 			abort();
249 		strlcpy(e->domain, host, sizeof(e->domain));
250 		if (*port != '\0' && strcmp(port, "1965")) {
251 			strlcat(e->domain, ":", sizeof(e->domain));
252 			strlcat(e->domain, port, sizeof(e->domain));
253 		}
254 		strlcpy(e->hash, hash, sizeof(e->hash));
255 		tofu_add(&certs, e);
256 		ui_send_fs(IMSG_SAVE_CERT, tab->id, e, sizeof(*e));
257 	} else
258 		tofu_res = !strcmp(hash, e->hash);
259 
260 	if (tofu_res) {
261 		if (e->verified == -1)
262 			tab->trust = TS_TEMP_TRUSTED;
263 		else if (e->verified == 1)
264 			tab->trust = TS_VERIFIED;
265 		else
266 			tab->trust = TS_TRUSTED;
267 
268 		ui_send_net(IMSG_CERT_STATUS, imsg->hdr.peerid,
269 		    &tofu_res, sizeof(tofu_res));
270 	} else {
271 		tab->trust = TS_UNTRUSTED;
272 		load_page_from_str(tab, "# Certificate mismatch\n");
273 		if ((tab->cert = strdup(hash)) == NULL)
274 			die();
275 		ui_yornp("Certificate mismatch.  Proceed?",
276 		    handle_check_cert_user_choice, tab);
277 	}
278 }
279 
280 static void
281 handle_check_cert_user_choice(int accept, struct tab *tab)
282 {
283 	ui_send_net(IMSG_CERT_STATUS, tab->id, &accept,
284 	    sizeof(accept));
285 
286 	if (accept) {
287 		/*
288 		 * trust the certificate for this session only.  If
289 		 * the page results in a redirect while we're asking
290 		 * the user to save, we'll end up with an invalid
291 		 * tabid (one request == one tab id).  It also makes
292 		 * sense to save it for the current session if the
293 		 * user accepted it.
294 		 */
295 		tofu_temp_trust(&certs, tab->uri.host, tab->uri.port,
296 		    tab->cert);
297 
298 		if (!safe_mode)
299 			ui_yornp("Save the new certificate?",
300 			    handle_maybe_save_new_cert, tab);
301 		else
302 			message("Certificate temporarly trusted");
303 	} else {
304 		free(tab->cert);
305 		tab->cert = NULL;
306 	}
307 }
308 
309 static void
310 handle_maybe_save_new_cert(int accept, struct tab *tab)
311 {
312 	struct tofu_entry *e;
313 	const char *host, *port;
314 
315 	if (tab->proxy != NULL) {
316 		host = tab->proxy->host;
317 		port = tab->proxy->port;
318 	} else {
319 		host = tab->uri.host;
320 		port = tab->uri.port;
321 	}
322 
323 	if (!accept)
324 		goto end;
325 
326 	if ((e = calloc(1, sizeof(*e))) == NULL)
327 		die();
328 
329 	strlcpy(e->domain, host, sizeof(e->domain));
330 	if (*port != '\0' && strcmp(port, "1965")) {
331 		strlcat(e->domain, ":", sizeof(e->domain));
332 		strlcat(e->domain, port, sizeof(e->domain));
333 	}
334 	strlcpy(e->hash, tab->cert, sizeof(e->hash));
335 	ui_send_fs(IMSG_UPDATE_CERT, 0, e, sizeof(*e));
336 
337 	tofu_update(&certs, e);
338 
339 	tab->trust = TS_TRUSTED;
340 
341 end:
342 	free(tab->cert);
343 	tab->cert = NULL;
344 }
345 
346 static inline int
347 normalize_code(int n)
348 {
349 	if (n < 20) {
350 		if (n == 10 || n == 11)
351 			return n;
352 		return 10;
353 	} else if (n < 30) {
354 		return 20;
355 	} else if (n < 40) {
356 		if (n == 30 || n == 31)
357 			return n;
358 		return 30;
359 	} else if (n < 50) {
360 		if (n <= 44)
361 			return n;
362 		return 40;
363 	} else if (n < 60) {
364 		if (n <= 53 || n == 59)
365 			return n;
366 		return 50;
367 	} else if (n < 70) {
368 		if (n <= 62)
369 			return n;
370 		return 60;
371 	} else
372 		return MALFORMED_RESPONSE;
373 }
374 
375 static void
376 handle_imsg_got_code(struct imsg *imsg, size_t datalen)
377 {
378 	struct tab	*tab;
379 
380 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
381 		return;
382 
383 	if (sizeof(tab->code) != datalen)
384 		die();
385 
386 	memcpy(&tab->code, imsg->data, sizeof(tab->code));
387 	tab->code = normalize_code(tab->code);
388 	if (tab->code != 30 && tab->code != 31)
389 		tab->redirect_count = 0;
390 }
391 
392 static void
393 handle_imsg_got_meta(struct imsg *imsg, size_t datalen)
394 {
395 	struct tab	*tab;
396 	char		 buf[128];
397 
398 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
399 		return;
400 
401 	if (sizeof(tab->meta) <= datalen)
402 		die();
403 
404 	memcpy(tab->meta, imsg->data, datalen);
405 
406 	if (tab->code < 10) {	/* internal errors */
407 		load_page_from_str(tab, err_pages[tab->code]);
408 	} else if (tab->code < 20) {	/* 1x */
409 		free(tab->last_input_url);
410 		tab->last_input_url = strdup(tab->hist_cur->h);
411 		if (tab->last_input_url == NULL)
412 			die();
413 
414 		load_page_from_str(tab, err_pages[tab->code]);
415 		ui_require_input(tab, tab->code == 11, ir_select_gemini);
416 	} else if (tab->code == 20) {
417 		history_add(tab->hist_cur->h);
418 		if (setup_parser_for(tab)) {
419 			ui_send_net(IMSG_PROCEED, tab->id, NULL, 0);
420 		} else if (safe_mode) {
421 			load_page_from_str(tab,
422 			    err_pages[UNKNOWN_TYPE_OR_CSET]);
423 		} else {
424 			struct hist *h;
425 
426 			if ((h = hist_pop(&tab->hist)) != NULL)
427 				tab->hist_cur = h;
428 			snprintf(buf, sizeof(buf),
429 			    "Can't display \"%s\", save it?", tab->meta);
430 			ui_yornp(buf, handle_maybe_save_page, tab);
431 		}
432 	} else if (tab->code < 40) { /* 3x */
433 		tab->redirect_count++;
434 
435 		/* TODO: make customizable? */
436 		if (tab->redirect_count > 5) {
437 			load_page_from_str(tab,
438 			    err_pages[TOO_MUCH_REDIRECTS]);
439 		} else
440 			do_load_url(tab, tab->meta, NULL, LU_MODE_NOCACHE);
441 	} else { /* 4x, 5x & 6x */
442 		load_page_from_str(tab, err_pages[tab->code]);
443 	}
444 }
445 
446 static void
447 handle_maybe_save_page(int dosave, struct tab *tab)
448 {
449 	const char	*f;
450 	char		 input[PATH_MAX];
451 
452 	/* XXX: this print a message that is confusing  */
453 	ui_on_tab_loaded(tab);
454 
455 	if (!dosave) {
456 		stop_tab(tab);
457 		return;
458 	}
459 
460 	if ((f = strrchr(tab->uri.path, '/')) == NULL)
461 		f = "";
462 	else
463 		f++;
464 
465 	strlcpy(input, download_path, sizeof(input));
466 	strlcat(input, f, sizeof(input));
467 
468 	ui_read("Save to path", handle_save_page_path, tab, input);
469 }
470 
471 static void
472 handle_save_page_path(const char *path, struct tab *tab)
473 {
474 	if (path == NULL) {
475 		stop_tab(tab);
476 		return;
477 	}
478 
479 	ui_show_downloads_pane();
480 
481 	enqueue_download(tab->id, path);
482 	ui_send_fs(IMSG_FILE_OPEN, tab->id, path, strlen(path)+1);
483 
484 	/*
485 	 * Change this tab id, the old one is associated with the
486 	 * download now.
487 	 */
488 	tab->id = tab_new_id();
489 }
490 
491 static void
492 handle_imsg_file_opened(struct imsg *imsg, size_t datalen)
493 {
494 	struct download	*d;
495 	const char	*e;
496 
497 	/*
498 	 * There are no reason we shouldn't be able to find the
499 	 * required download.
500 	 */
501 	if ((d = download_by_id(imsg->hdr.peerid)) == NULL)
502 		die();
503 
504 	if (imsg->fd == -1) {
505 		e = imsg->data;
506 		if (e[datalen-1] != '\0')
507 			die();
508 		message("Can't open file %s: %s", d->path, e);
509 	} else {
510 		d->fd = imsg->fd;
511 		ui_send_net(IMSG_PROCEED, d->id, NULL, 0);
512 	}
513 }
514 
515 static void
516 handle_imsg_session(struct imsg *imsg, size_t datalen)
517 {
518 	static struct tab	*curr;
519 	static struct tab	*tab;
520 	struct session_tab	 st;
521 	struct session_tab_hist	 sth;
522 	struct hist		*h;
523 	int			 first_time;
524 
525 	/*
526 	 * The fs process tried to send tabs after it has announced
527 	 * that he's done.  Something fishy is going on, better die.
528 	 */
529 	if (operating)
530 		die();
531 
532 	switch (imsg->hdr.type) {
533 	case IMSG_SESSION_TAB:
534 		if (datalen != sizeof(st))
535 			die();
536 
537 		memcpy(&st, imsg->data, sizeof(st));
538 		if ((tab = new_tab(st.uri, NULL, NULL)) == NULL)
539 			die();
540 		tab->hist_cur->line_off = st.top_line;
541 		tab->hist_cur->current_off = st.current_line;
542 		strlcpy(tab->buffer.page.title, st.title,
543 		    sizeof(tab->buffer.page.title));
544 		if (st.flags & TAB_CURRENT)
545 			curr = tab;
546 		if (st.flags & TAB_KILLED)
547 			kill_tab(tab, 1);
548 		break;
549 
550 	case IMSG_SESSION_TAB_HIST:
551 		if (tab == NULL || datalen != sizeof(sth))
552 			die();
553 
554 		memcpy(&sth, imsg->data, sizeof(sth));
555 		if (sth.uri[sizeof(sth.uri)-1] != '\0')
556 			die();
557 
558 		if ((h = calloc(1, sizeof(*h))) == NULL)
559 			die();
560 		strlcpy(h->h, sth.uri, sizeof(h->h));
561 
562 		if (sth.future)
563 			hist_push(&tab->hist, h);
564 		else
565 			hist_add_before(&tab->hist, tab->hist_cur, h);
566 		break;
567 
568 	case IMSG_SESSION_END:
569 		if (datalen != sizeof(first_time))
570 			die();
571 		memcpy(&first_time, imsg->data, sizeof(first_time));
572 		if (first_time) {
573 			new_tab("about:new", NULL, NULL);
574 			curr = new_tab("about:help", NULL, NULL);
575 		}
576 
577 		operating = 1;
578 		if (curr != NULL)
579 			switch_to_tab(curr);
580 		if (has_url || TAILQ_EMPTY(&tabshead))
581 			new_tab(url, NULL, NULL);
582 		ui_main_loop();
583 		break;
584 
585 	default:
586 		die();
587 	}
588 }
589 
590 static void
591 handle_imsg_history(struct imsg *imsg, size_t datalen)
592 {
593 	struct histitem hi;
594 
595 	/*
596 	 * The fs process tried to send history item after it
597 	 * has announced that it's done.  Something fishy is
598 	 * going on, better die
599 	 */
600 	if (operating)
601 		die();
602 
603 	switch (imsg->hdr.type) {
604 	case IMSG_HIST_ITEM:
605 		if (datalen != sizeof(hi))
606 			die();
607 
608 		memcpy(&hi, imsg->data, sizeof(hi));
609 		history_push(&hi);
610 		break;
611 
612 	case IMSG_HIST_END:
613 		history_sort();
614 		break;
615 
616 	default:
617 		die();
618 	}
619 }
620 
621 static void
622 handle_imsg_buf(struct imsg *imsg, size_t datalen)
623 {
624 	struct tab	*tab = NULL;
625 	struct download	*d = NULL;
626 
627 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL &&
628 	    (d = download_by_id(imsg->hdr.peerid)) == NULL)
629 		return;
630 
631 	if (tab != NULL) {
632 		if (!parser_parse(tab, imsg->data, datalen))
633 			die();
634 		ui_on_tab_refresh(tab);
635 	} else {
636 		d->bytes += datalen;
637 		write(d->fd, imsg->data, datalen);
638 		ui_on_download_refresh();
639 	}
640 }
641 
642 static void
643 handle_imsg_eof(struct imsg *imsg, size_t datalen)
644 {
645 	struct tab	*tab = NULL;
646 	struct download	*d = NULL;
647 
648 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL &&
649 	    (d = download_by_id(imsg->hdr.peerid)) == NULL)
650 		return;
651 
652 	if (tab != NULL) {
653 		if (!parser_free(tab))
654 			die();
655 		if (has_prefix(tab->hist_cur->h, "gemini://") ||
656 		    has_prefix(tab->hist_cur->h, "gopher://"))
657 			mcache_tab(tab);
658 		ui_on_tab_refresh(tab);
659 		ui_on_tab_loaded(tab);
660 	} else {
661 		close(d->fd);
662 		d->fd = -1;
663 		ui_on_download_refresh();
664 	}
665 }
666 
667 static void
668 handle_imsg_tofu(struct imsg *imsg, size_t datalen)
669 {
670 	struct tofu_entry *e;
671 
672 	if (operating)
673 		die();
674 
675 	if ((e = calloc(1, sizeof(*e))) == NULL)
676 		die();
677 
678 	if (datalen != sizeof(*e))
679 		die();
680 	memcpy(e, imsg->data, sizeof(*e));
681 	if (e->domain[sizeof(e->domain)-1] != '\0' ||
682 	    e->hash[sizeof(e->hash)-1] != '\0')
683 		die();
684 	tofu_add(&certs, e);
685 }
686 
687 static void
688 handle_imsg_bookmark_ok(struct imsg *imsg, size_t datalen)
689 {
690 	int res;
691 
692 	if (datalen != sizeof(res))
693 		die();
694 
695 	memcpy(&res, imsg->data, sizeof(res));
696 	if (res == 0)
697 		message("Added to bookmarks!");
698 	else
699 		message("Failed to add to bookmarks: %s",
700 		    strerror(res));
701 }
702 
703 static void
704 handle_imsg_save_cert_ok(struct imsg *imsg, size_t datalen)
705 {
706 	int res;
707 
708 	if (datalen != sizeof(res))
709 		die();
710 	memcpy(&res, imsg->data, datalen);
711 	if (res != 0)
712 		message("Failed to save the cert for: %s",
713 		    strerror(res));
714 }
715 
716 static void
717 handle_imsg_update_cert_ok(struct imsg *imsg, size_t datalen)
718 {
719 	int res;
720 
721 	if (datalen != sizeof(res))
722 		die();
723 	memcpy(&res, imsg->data, datalen);
724 	if (!res)
725 		message("Failed to update the certificate");
726 }
727 
728 static void
729 handle_dispatch_imsg(int fd, short ev, void *d)
730 {
731 	struct imsgev	*iev = d;
732 
733 	if (dispatch_imsg(iev, ev, handlers, sizeof(handlers)) == -1)
734 		err(1, "connection closed");
735 }
736 
737 static int
738 load_about_url(struct tab *tab, const char *url)
739 {
740 	tab->trust = TS_UNKNOWN;
741 	parser_init(tab, gemtext_initparser);
742 	return make_fs_request(tab, IMSG_GET, url);
743 }
744 
745 static int
746 load_file_url(struct tab *tab, const char *url)
747 {
748 	tab->trust = TS_UNKNOWN;
749 	return make_fs_request(tab, IMSG_GET_FILE, tab->uri.path);
750 }
751 
752 static int
753 load_finger_url(struct tab *tab, const char *url)
754 {
755 	struct get_req	 req;
756 	size_t		 len;
757 	char		*at;
758 
759 	memset(&req, 0, sizeof(req));
760 	strlcpy(req.port, tab->uri.port, sizeof(req.port));
761 
762 	/*
763 	 * Sometimes the finger url have the user as path component
764 	 * (e.g. finger://thelambdalab.xyz/plugd), sometimes as
765 	 * userinfo (e.g. finger://cobradile@finger.farm).
766 	 */
767 	if ((at = strchr(tab->uri.host, '@')) != NULL) {
768 		len = at - tab->uri.host;
769 		memcpy(req.req, tab->uri.host, len);
770 
771 		if (len >= sizeof(req.req))
772 			die();
773 		req.req[len] = '\0';
774 
775 		strlcpy(req.host, at+1, sizeof(req.host));
776 	} else {
777 		strlcpy(req.host, tab->uri.host, sizeof(req.host));
778 
779 		/* +1 to skip the initial `/' */
780 		strlcpy(req.req, tab->uri.path+1, sizeof(req.req));
781 	}
782 	strlcat(req.req, "\r\n", sizeof(req.req));
783 
784 	parser_init(tab, textplain_initparser);
785 	return make_request(tab, &req, PROTO_FINGER, NULL);
786 }
787 
788 static int
789 load_gemini_url(struct tab *tab, const char *url)
790 {
791 	struct get_req	 req;
792 
793 	memset(&req, 0, sizeof(req));
794 	strlcpy(req.host, tab->uri.host, sizeof(req.host));
795 	strlcpy(req.port, tab->uri.port, sizeof(req.port));
796 
797 	return make_request(tab, &req, PROTO_GEMINI, tab->hist_cur->h);
798 }
799 
800 static inline const char *
801 gopher_skip_selector(const char *path, int *ret_type)
802 {
803 	*ret_type = 0;
804 
805 	if (!strcmp(path, "/") || *path == '\0') {
806 		*ret_type = '1';
807 		return path;
808 	}
809 
810 	if (*path != '/')
811 		return path;
812 	path++;
813 
814 	switch (*ret_type = *path) {
815 	case '0':
816 	case '1':
817 	case '7':
818 		break;
819 
820 	default:
821 		*ret_type = 0;
822 		path -= 1;
823 		return path;
824 	}
825 
826 	return ++path;
827 }
828 
829 static int
830 load_gopher_url(struct tab *tab, const char *url)
831 {
832 	struct get_req	 req;
833 	int		 type;
834 	const char	*path;
835 
836 	memset(&req, 0, sizeof(req));
837 	strlcpy(req.host, tab->uri.host, sizeof(req.host));
838 	strlcpy(req.port, tab->uri.port, sizeof(req.port));
839 
840 	path = gopher_skip_selector(tab->uri.path, &type);
841 	switch (type) {
842 	case '0':
843 		parser_init(tab, textplain_initparser);
844 		break;
845 	case '1':
846 		parser_init(tab, gophermap_initparser);
847 		break;
848 	case '7':
849 		free(tab->last_input_url);
850 		tab->last_input_url = strdup(url);
851 		if (tab->last_input_url == NULL)
852 			die();
853 		ui_require_input(tab, 0, ir_select_gopher);
854 		return load_page_from_str(tab, err_pages[10]);
855 	default:
856 		return load_page_from_str(tab, "Unknown gopher selector");
857 	}
858 
859 	strlcpy(req.req, path, sizeof(req.req));
860 	if (*tab->uri.query != '\0') {
861 		strlcat(req.req, "?", sizeof(req.req));
862 		strlcat(req.req, tab->uri.query, sizeof(req.req));
863 	}
864 	strlcat(req.req, "\r\n", sizeof(req.req));
865 
866 	return make_request(tab, &req, PROTO_GOPHER, NULL);
867 }
868 
869 static int
870 load_via_proxy(struct tab *tab, const char *url, struct proxy *p)
871 {
872 	struct get_req req;
873 
874 	memset(&req, 0, sizeof(req));
875 	strlcpy(req.host, p->host, sizeof(req.host));
876 	strlcpy(req.port, p->port, sizeof(req.port));
877 
878 	tab->proxy = p;
879 
880 	return make_request(tab, &req, p->proto, tab->hist_cur->h);
881 }
882 
883 static int
884 make_request(struct tab *tab, struct get_req *req, int proto, const char *r)
885 {
886 	stop_tab(tab);
887 	tab->id = tab_new_id();
888 	req->proto = proto;
889 
890 	if (r != NULL) {
891 		strlcpy(req->req, r, sizeof(req->req));
892 		strlcat(req->req, "\r\n", sizeof(req->req));
893 	}
894 
895 	start_loading_anim(tab);
896 	ui_send_net(IMSG_GET_RAW, tab->id, req, sizeof(*req));
897 
898 	/*
899 	 * So the various load_*_url can `return make_request` and
900 	 * do_load_url is happy.
901 	 */
902 	return 1;
903 }
904 
905 static int
906 make_fs_request(struct tab *tab, int type, const char *r)
907 {
908 	stop_tab(tab);
909 	tab->id = tab_new_id();
910 
911 	ui_send_fs(type, tab->id, r, strlen(r)+1);
912 
913 	/*
914 	 * So load_{about,file}_url can `return make_fs_request` and
915 	 * do_load_url is happy.
916 	 */
917 	return 1;
918 }
919 
920 void
921 gopher_send_search_req(struct tab *tab, const char *text)
922 {
923 	struct get_req	req;
924 
925 	memset(&req, 0, sizeof(req));
926 	strlcpy(req.host, tab->uri.host, sizeof(req.host));
927 	strlcpy(req.port, tab->uri.port, sizeof(req.port));
928 
929 	/* +2 to skip /7 */
930 	strlcpy(req.req, tab->uri.path+2, sizeof(req.req));
931 	if (*tab->uri.query != '\0') {
932 		strlcat(req.req, "?", sizeof(req.req));
933 		strlcat(req.req, tab->uri.query, sizeof(req.req));
934 	}
935 
936 	strlcat(req.req, "\t", sizeof(req.req));
937 	strlcat(req.req, text, sizeof(req.req));
938 	strlcat(req.req, "\r\n", sizeof(req.req));
939 
940 	erase_buffer(&tab->buffer);
941 	parser_init(tab, gophermap_initparser);
942 
943 	make_request(tab, &req, PROTO_GOPHER, NULL);
944 }
945 
946 int
947 load_page_from_str(struct tab *tab, const char *page)
948 {
949 	parser_init(tab, gemtext_initparser);
950 	if (!tab->buffer.page.parse(&tab->buffer.page, page, strlen(page)))
951 		abort();
952 	if (!tab->buffer.page.free(&tab->buffer.page))
953 		abort();
954 	ui_on_tab_refresh(tab);
955 	ui_on_tab_loaded(tab);
956 	return 0;
957 }
958 
959 /*
960  * Effectively load the given url in the given tab.  Return 1 when
961  * loading the page asynchronously, and thus when an erase_buffer can
962  * be done right after this function return, or 0 when loading the
963  * page synchronously.
964  */
965 static int
966 do_load_url(struct tab *tab, const char *url, const char *base, int mode)
967 {
968 	const struct proto	*p;
969 	struct phos_uri	 uri;
970 	struct proxy	*proxy;
971 	int		 nocache = mode & LU_MODE_NOCACHE;
972 	char		*t;
973 
974 	tab->proxy = NULL;
975 	tab->trust = TS_UNKNOWN;
976 
977 	if (base == NULL)
978 		memcpy(&uri, &tab->uri, sizeof(tab->uri));
979 	else
980 		phos_parse_absolute_uri(base, &uri);
981 
982 	if (!phos_resolve_uri_from_str(&uri, url, &tab->uri)) {
983 		if (asprintf(&t, "#error loading %s\n>%s\n",
984 		    url, "Can't parse the URI") == -1)
985 			die();
986 		strlcpy(tab->hist_cur->h, url, sizeof(tab->hist_cur->h));
987 		load_page_from_str(tab, t);
988 		free(t);
989 		return 0;
990 	}
991 
992 	phos_serialize_uri(&tab->uri, tab->hist_cur->h,
993 	    sizeof(tab->hist_cur->h));
994 
995 	if (!nocache && mcache_lookup(tab->hist_cur->h, tab)) {
996 		ui_on_tab_refresh(tab);
997 		ui_on_tab_loaded(tab);
998 		return 0;
999 	}
1000 
1001 	for (p = protos; p->schema != NULL; ++p) {
1002 		if (!strcmp(tab->uri.scheme, p->schema)) {
1003 			/* patch the port */
1004 			if (*tab->uri.port == '\0' && p->port != NULL)
1005 				strlcpy(tab->uri.port, p->port,
1006 				    sizeof(tab->uri.port));
1007 
1008 			return p->loadfn(tab, tab->hist_cur->h);
1009 		}
1010 	}
1011 
1012 	TAILQ_FOREACH(proxy, &proxies, proxies) {
1013 		if (!strcmp(tab->uri.scheme, proxy->match_proto))
1014 			return load_via_proxy(tab, url, proxy);
1015 	}
1016 
1017 	return load_page_from_str(tab, err_pages[UNKNOWN_PROTOCOL]);
1018 }
1019 
1020 /*
1021  * Load url in tab and handle history.  If a tab is marked as lazy, only
1022  * prepare the url but don't load it.
1023  */
1024 void
1025 load_url(struct tab *tab, const char *url, const char *base, int mode)
1026 {
1027 	int lazy = tab->flags & TAB_LAZY;
1028 	int nohist = mode & LU_MODE_NOHIST;
1029 
1030 	if (operating && lazy) {
1031 		tab->flags ^= TAB_LAZY;
1032 		lazy = 0;
1033 	} else if (tab->hist_cur != NULL)
1034 		get_scroll_position(tab, &tab->hist_cur->line_off,
1035 		    &tab->hist_cur->current_off);
1036 
1037 	if (!nohist && (!lazy || tab->hist_cur == NULL)) {
1038 		if (tab->hist_cur != NULL)
1039 			hist_clear_forward(&tab->hist,
1040 			    TAILQ_NEXT(tab->hist_cur, entries));
1041 
1042 		if ((tab->hist_cur = calloc(1, sizeof(*tab->hist_cur)))
1043 		    == NULL) {
1044 			event_loopbreak();
1045 			return;
1046 		}
1047 
1048 		strlcpy(tab->buffer.page.title, url,
1049 		    sizeof(tab->buffer.page.title));
1050 		hist_push(&tab->hist, tab->hist_cur);
1051 
1052 		if (lazy)
1053 			strlcpy(tab->hist_cur->h, url,
1054 			    sizeof(tab->hist_cur->h));
1055 	}
1056 
1057 	if (!lazy)
1058 		do_load_url(tab, url, base, mode);
1059 }
1060 
1061 void
1062 load_url_in_tab(struct tab *tab, const char *url, const char *base, int mode)
1063 {
1064 	if (operating) {
1065 		autosave_hook();
1066 		message("Loading %s...", url);
1067 	}
1068 
1069 	load_url(tab, url, base, mode);
1070 }
1071 
1072 int
1073 load_previous_page(struct tab *tab)
1074 {
1075 	struct hist	*h;
1076 
1077 	if ((h = TAILQ_PREV(tab->hist_cur, mhisthead, entries)) == NULL)
1078 		return 0;
1079 	tab->hist_cur = h;
1080 	do_load_url(tab, h->h, NULL, LU_MODE_NONE);
1081 	return 1;
1082 }
1083 
1084 int
1085 load_next_page(struct tab *tab)
1086 {
1087 	struct hist	*h;
1088 
1089 	if ((h = TAILQ_NEXT(tab->hist_cur, entries)) == NULL)
1090 		return 0;
1091 	tab->hist_cur = h;
1092 	do_load_url(tab, h->h, NULL, LU_MODE_NONE);
1093 	return 1;
1094 }
1095 
1096 void
1097 add_to_bookmarks(const char *str)
1098 {
1099 	ui_send_fs(IMSG_BOOKMARK_PAGE, 0,
1100 	    str, strlen(str)+1);
1101 }
1102 
1103 /*
1104  * Given a user-entered URL, apply some heuristics to use it:
1105  *
1106  * - if it's a proper url use it
1107  * - if it starts with a `./' or a `/' assume its a file:// url
1108  * - assume it's a gemini:// url
1109  *
1110  * `ret' (of which len is the size) will be filled with the resulting
1111  * url.
1112  */
1113 void
1114 humanify_url(const char *raw, char *ret, size_t len)
1115 {
1116 	struct phos_uri	uri;
1117 	char		buf[PATH_MAX];
1118 
1119 	if (phos_parse_absolute_uri(raw, &uri)) {
1120 		strlcpy(ret, raw, len);
1121 		return;
1122 	}
1123 
1124 	if (has_prefix(raw, "./")) {
1125 		strlcpy(ret, "file://", len);
1126 		getcwd(buf, sizeof(buf));
1127 		strlcat(ret, buf, len);
1128 		strlcat(ret, "/", len);
1129 		strlcat(ret, raw+2, len);
1130 		return;
1131 	}
1132 
1133 	if (*raw == '/') {
1134 		strlcpy(ret, "file://", len);
1135 		strlcat(ret, raw, len);
1136 		return;
1137 	}
1138 
1139 	strlcpy(ret, "gemini://", len);
1140 	strlcat(ret, raw, len);
1141 }
1142 
1143 static pid_t
1144 start_child(enum telescope_process p, const char *argv0, int fd)
1145 {
1146 	const char	*argv[5];
1147 	int		 argc = 0;
1148 	pid_t		 pid;
1149 
1150 	switch (pid = fork()) {
1151 	case -1:
1152 		die();
1153 	case 0:
1154 		break;
1155 	default:
1156 		close(fd);
1157 		return pid;
1158 	}
1159 
1160 	if (dup2(fd, 3) == -1)
1161 		err(1, "cannot setup imsg fd");
1162 
1163 	argv[argc++] = argv0;
1164 	switch (p) {
1165 	case PROC_UI:
1166 		errx(1, "Can't start ui process");
1167 	case PROC_FS:
1168 		argv[argc++] = "-Tf";
1169 		break;
1170 	case PROC_NET:
1171 		argv[argc++] = "-Tn";
1172 		break;
1173 	}
1174 
1175 	if (safe_mode)
1176 		argv[argc++] = "-S";
1177 
1178 	argv[argc++] = NULL;
1179 	execvp(argv0, (char *const *)argv);
1180 	err(1, "execvp(%s)", argv0);
1181 }
1182 
1183 static void
1184 send_url(const char *url)
1185 {
1186 	struct sockaddr_un	 sun;
1187 	struct imsgbuf		 ibuf;
1188 	int			 ctl_sock;
1189 
1190 	if ((ctl_sock = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
1191 		err(1, "socket");
1192 
1193 	memset(&sun, 0, sizeof(sun));
1194 	sun.sun_family = AF_UNIX;
1195 	strlcpy(sun.sun_path, ctlsock_path, sizeof(sun.sun_path));
1196 
1197 	if (connect(ctl_sock, (struct sockaddr *)&sun, sizeof(sun)) == -1)
1198 		err(1, "connect: %s", ctlsock_path);
1199 
1200 	imsg_init(&ibuf, ctl_sock);
1201 	imsg_compose(&ibuf, IMSG_CTL_OPEN_URL, 0, 0, -1, url,
1202 	    strlen(url) + 1);
1203 	imsg_flush(&ibuf);
1204 	close(ctl_sock);
1205 }
1206 
1207 int
1208 ui_send_net(int type, uint32_t peerid, const void *data,
1209     uint16_t datalen)
1210 {
1211 	return imsg_compose_event(iev_net, type, peerid, 0, -1, data,
1212 	    datalen);
1213 }
1214 
1215 int
1216 ui_send_fs(int type, uint32_t peerid, const void *data, uint16_t datalen)
1217 {
1218 	return imsg_compose_event(iev_fs, type, peerid, 0, -1, data,
1219 	    datalen);
1220 }
1221 
1222 static void __attribute__((noreturn))
1223 usage(int r)
1224 {
1225 	fprintf(stderr, "USAGE: %s [-ChnSv] [-c config] [url]\n",
1226 	    getprogname());
1227 	fprintf(stderr, "version: " PACKAGE " " VERSION "\n");
1228 	exit(r);
1229 }
1230 
1231 int
1232 main(int argc, char * const *argv)
1233 {
1234 	struct imsgev	 net_ibuf, fs_ibuf;
1235 	pid_t		 pid;
1236 	int		 control_fd;
1237 	int		 pipe2net[2], pipe2fs[2];
1238 	int		 ch, configtest = 0, fail = 0;
1239 	int		 proc = -1;
1240 	int		 sessionfd = -1;
1241 	int		 status;
1242 	const char	*argv0;
1243 
1244 	argv0 = argv[0];
1245 
1246 	signal(SIGCHLD, SIG_IGN);
1247 	signal(SIGPIPE, SIG_IGN);
1248 
1249 	if (getenv("NO_COLOR") != NULL)
1250 		enable_colors = 0;
1251 
1252 	fs_init();
1253 
1254 	while ((ch = getopt_long(argc, argv, opts, longopts, NULL)) != -1) {
1255 		switch (ch) {
1256 		case 'C':
1257 			exit(ui_print_colors());
1258 		case 'c':
1259 			fail = 1;
1260 			strlcpy(config_path, optarg, sizeof(config_path));
1261 			break;
1262 		case 'n':
1263 			configtest = 1;
1264 			break;
1265 		case 'h':
1266 			usage(0);
1267 		case 'S':
1268 			safe_mode = 1;
1269 			break;
1270 		case 'T':
1271 			switch (*optarg) {
1272 			case 'f':
1273 				proc = PROC_FS;
1274 				break;
1275 			case 'n':
1276 				proc = PROC_NET;
1277 				break;
1278 			default:
1279 				errx(1, "invalid process spec %c",
1280 				    *optarg);
1281 			}
1282 			break;
1283 		case 'v':
1284 			printf("%s %s\n", PACKAGE_NAME, PACKAGE_VERSION);
1285 			exit(0);
1286 			break;
1287 		default:
1288 			usage(1);
1289 		}
1290 	}
1291 
1292 	argc -= optind;
1293 	argv += optind;
1294 
1295 	if (proc != -1) {
1296 		if (argc > 0)
1297 			usage(1);
1298 		else if (proc == PROC_FS)
1299 			return fs_main();
1300 		else if (proc == PROC_NET)
1301 			return net_main();
1302 		else
1303 			usage(1);
1304 	}
1305 
1306 	if (argc != 0) {
1307 		has_url = 1;
1308 		humanify_url(argv[0], url, sizeof(url));
1309 	}
1310 
1311 	/* setup keys before reading the config */
1312 	TAILQ_INIT(&global_map.m);
1313 	global_map.unhandled_input = global_key_unbound;
1314 	TAILQ_INIT(&minibuffer_map.m);
1315 
1316 	config_init();
1317 	parseconfig(config_path, fail);
1318 	if (configtest) {
1319 		puts("config OK");
1320 		exit(0);
1321 	}
1322 
1323 	if (download_path == NULL &&
1324 	    (download_path = strdup("/tmp/")) == NULL)
1325 		errx(1, "strdup");
1326 
1327 	if (!safe_mode && (sessionfd = lock_session()) == -1) {
1328 		if (has_url) {
1329 			send_url(url);
1330 			exit(0);
1331 		}
1332 
1333 		errx(1, "can't lock session, is another instance of "
1334 		    "telescope already running?");
1335 	}
1336 
1337 	/* Start children. */
1338 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe2fs) == -1)
1339 		err(1, "socketpair");
1340 	start_child(PROC_FS, argv0, pipe2fs[1]);
1341 	imsg_init(&fs_ibuf.ibuf, pipe2fs[0]);
1342 	iev_fs = &fs_ibuf;
1343 	iev_fs->handler = handle_dispatch_imsg;
1344 
1345 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe2net) == -1)
1346 		err(1, "socketpair");
1347 	start_child(PROC_NET, argv0, pipe2net[1]);
1348 	imsg_init(&net_ibuf.ibuf, pipe2net[0]);
1349 	iev_net = &net_ibuf;
1350 	iev_net->handler = handle_dispatch_imsg;
1351 
1352 	setproctitle("ui");
1353 
1354 	/* initialize tofu store */
1355 	tofu_init(&certs, 5, offsetof(struct tofu_entry, domain));
1356 
1357 	event_init();
1358 
1359 	if (!safe_mode) {
1360 		if ((control_fd = control_init(ctlsock_path)) == -1)
1361 			err(1, "control_init %s", ctlsock_path);
1362 		control_listen(control_fd);
1363 	}
1364 
1365 	/* initialize the in-memory cache store */
1366 	mcache_init();
1367 
1368 	/* Setup event handler for the autosave */
1369 	autosave_init();
1370 
1371 	/* Setup event handlers for pipes to fs/net */
1372 	iev_fs->events = EV_READ;
1373 	event_set(&iev_fs->ev, iev_fs->ibuf.fd, iev_fs->events,
1374 	    iev_fs->handler, iev_fs);
1375 	event_add(&iev_fs->ev, NULL);
1376 
1377 	iev_net->events = EV_READ;
1378 	event_set(&iev_net->ev, iev_net->ibuf.fd, iev_net->events,
1379 	    iev_net->handler, iev_net);
1380 	event_add(&iev_net->ev, NULL);
1381 
1382 	if (ui_init()) {
1383 		sandbox_ui_process();
1384 		ui_send_fs(IMSG_INIT, 0, NULL, 0);
1385 		event_dispatch();
1386 		ui_end();
1387 	}
1388 
1389 	ui_send_fs(IMSG_QUIT, 0, NULL, 0);
1390 	ui_send_net(IMSG_QUIT, 0, NULL, 0);
1391 	imsg_flush(&iev_fs->ibuf);
1392 	imsg_flush(&iev_net->ibuf);
1393 
1394 	/* wait for children to terminate */
1395 	do {
1396 		pid = wait(&status);
1397 		if (pid == -1) {
1398 			if (errno != EINTR && errno != ECHILD)
1399 				err(1, "wait");
1400 		} else if (WIFSIGNALED(status))
1401 			warnx("child terminated; signal %d", WTERMSIG(status));
1402 	} while (pid != -1 || (pid == -1 && errno == EINTR));
1403 
1404 	if (!safe_mode && close(sessionfd) == -1)
1405 		err(1, "close(sessionfd = %d)", sessionfd);
1406 
1407 	return 0;
1408 }