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 
21 #include <errno.h>
22 #include <getopt.h>
23 #include <limits.h>
24 #include <signal.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <unistd.h>
29 
30 #include "defaults.h"
31 #include "minibuffer.h"
32 #include "parser.h"
33 #include "telescope.h"
34 #include "ui.h"
35 
36 static struct option longopts[] = {
37 	{"colors",	no_argument,	NULL,	'c'},
38 	{"help",	no_argument,	NULL,	'h'},
39 	{"version",	no_argument,	NULL,	'v'},
40 	{NULL,		0,		NULL,	0},
41 };
42 
43 static const char *opts = "Cc:hnT:v";
44 
45 static struct imsgev	*iev_fs, *iev_net;
46 
47 struct tabshead		 tabshead = TAILQ_HEAD_INITIALIZER(tabshead);
48 struct proxylist	 proxies = TAILQ_HEAD_INITIALIZER(proxies);
49 
50 enum telescope_process {
51 	PROC_UI,
52 	PROC_FS,
53 	PROC_NET,
54 };
55 
56 #define CANNOT_FETCH		0
57 #define TOO_MUCH_REDIRECTS	1
58 #define MALFORMED_RESPONSE	2
59 #define UNKNOWN_TYPE_OR_CSET	3
60 #define UNKNOWN_PROTOCOL	4
61 
62 /* XXX: keep in sync with telescope.c:/^normalize_code/ */
63 const char *err_pages[70] = {
64 	[CANNOT_FETCH]		= "# Couldn't load the page\n",
65 	[TOO_MUCH_REDIRECTS]	= "# Too much redirects\n",
66 	[MALFORMED_RESPONSE]	= "# Got a malformed response\n",
67 	[UNKNOWN_TYPE_OR_CSET]	= "# Unsupported type or charset\n",
68 	[UNKNOWN_PROTOCOL]	= "# Unknown protocol\n",
69 
70 	[10] = "# Input required\n",
71 	[11] = "# Input required\n",
72 	[40] = "# Temporary failure\n",
73 	[41] = "# Server unavailable\n",
74 	[42] = "# CGI error\n",
75 	[43] = "# Proxy error\n",
76 	[44] = "# Slow down\n",
77 	[50] = "# Permanent failure\n",
78 	[51] = "# Not found\n",
79 	[52] = "# Gone\n",
80 	[53] = "# Proxy request refused\n",
81 	[59] = "# Bad request\n",
82 	[60] = "# Client certificate required\n",
83 	[61] = "# Certificate not authorised\n",
84 	[62] = "# Certificate not valid\n"
85 };
86 
87 static void		 die(void) __attribute__((__noreturn__));
88 static struct tab	*tab_by_id(uint32_t);
89 static void		 handle_imsg_err(struct imsg*, size_t);
90 static void		 handle_imsg_check_cert(struct imsg*, size_t);
91 static void		 handle_check_cert_user_choice(int, struct tab *);
92 static void		 handle_maybe_save_new_cert(int, struct tab *);
93 static void		 handle_imsg_got_code(struct imsg*, size_t);
94 static void		 handle_imsg_got_meta(struct imsg*, size_t);
95 static void		 handle_maybe_save_page(int, struct tab *);
96 static void		 handle_save_page_path(const char *, struct tab *);
97 static void		 handle_imsg_file_opened(struct imsg*, size_t);
98 static void		 handle_imsg_buf(struct imsg*, size_t);
99 static void		 handle_imsg_eof(struct imsg*, size_t);
100 static void		 handle_imsg_bookmark_ok(struct imsg*, size_t);
101 static void		 handle_imsg_save_cert_ok(struct imsg*, size_t);
102 static void		 handle_imsg_update_cert_ok(struct imsg *, size_t);
103 static void		 handle_dispatch_imsg(int, short, void*);
104 static int		 load_page_from_str(struct tab*, const char*);
105 static int		 load_about_url(struct tab*, const char*);
106 static int		 load_finger_url(struct tab *, const char *);
107 static int		 load_gemini_url(struct tab*, const char*);
108 static int		 load_gopher_url(struct tab *, const char *);
109 static int		 load_via_proxy(struct tab *, const char *,
110 			     struct proxy *);
111 static int		 make_request(struct tab *, struct get_req *, int,
112 			     const char *);
113 static int		 do_load_url(struct tab*, const char *, const char *);
114 static void		 parse_session_line(char *, const char **, uint32_t *);
115 static void		 load_last_session(void);
116 static pid_t		 start_child(enum telescope_process, const char *, int);
117 static int		 ui_send_net(int, uint32_t, const void *, uint16_t);
118 static int		 ui_send_fs(int, uint32_t, const void *, uint16_t);
119 
120 static struct proto {
121 	const char	*schema;
122 	const char	*port;
123 	int		 (*loadfn)(struct tab*, const char*);
124 } protos[] = {
125 	{"about",	NULL,	load_about_url},
126 	{"finger",	"79",	load_finger_url},
127 	{"gemini",	"1965",	load_gemini_url},
128 	{"gopher",	"70",	load_gopher_url},
129 	{NULL, NULL, NULL},
130 };
131 
132 static imsg_handlerfn *handlers[] = {
133 	[IMSG_ERR] = handle_imsg_err,
134 	[IMSG_CHECK_CERT] = handle_imsg_check_cert,
135 	[IMSG_GOT_CODE] = handle_imsg_got_code,
136 	[IMSG_GOT_META] = handle_imsg_got_meta,
137 	[IMSG_BUF] = handle_imsg_buf,
138 	[IMSG_EOF] = handle_imsg_eof,
139 	[IMSG_BOOKMARK_OK] = handle_imsg_bookmark_ok,
140 	[IMSG_SAVE_CERT_OK] = handle_imsg_save_cert_ok,
141 	[IMSG_UPDATE_CERT_OK] = handle_imsg_update_cert_ok,
142 	[IMSG_FILE_OPENED] = handle_imsg_file_opened,
143 };
144 
145 static struct ohash	certs;
146 
147 static void __attribute__((__noreturn__))
148 die(void)
149 {
150 	abort(); 		/* TODO */
151 }
152 
153 static struct tab *
154 tab_by_id(uint32_t id)
155 {
156 	struct tab *t;
157 
158 	TAILQ_FOREACH(t, &tabshead, tabs) {
159 		if (t->id == id)
160 			return t;
161 	}
162 
163 	return NULL;
164 }
165 
166 static void
167 handle_imsg_err(struct imsg *imsg, size_t datalen)
168 {
169 	struct tab	*tab;
170 	char		*page;
171 
172 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
173 		return;
174 
175 	page = imsg->data;
176 	page[datalen-1] = '\0';
177 
178 	if (asprintf(&page, "# Error loading %s\n\n> %s\n",
179 	    tab->hist_cur->h, page) == -1)
180 		die();
181 	load_page_from_str(tab, page);
182 	free(page);
183 }
184 
185 static void
186 handle_imsg_check_cert(struct imsg *imsg, size_t datalen)
187 {
188 	const char		*hash, *host, *port;
189 	int			 tofu_res;
190 	struct tofu_entry	*e;
191 	struct tab		*tab;
192 
193 	hash = imsg->data;
194 	if (hash[datalen-1] != '\0')
195 		abort();
196 
197 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
198 		return;
199 
200 	if (tab->proxy != NULL) {
201 		host = tab->proxy->host;
202 		port = tab->proxy->port;
203 	} else {
204 		host = tab->uri.host;
205 		port = tab->uri.port;
206 	}
207 
208 	if ((e = tofu_lookup(&certs, host, port)) == NULL) {
209 		/* TODO: an update in libressl/libretls changed
210 		 * significantly.  Find a better approach at storing
211 		 * the certs! */
212 		if (datalen > sizeof(e->hash))
213 			abort();
214 
215 		tofu_res = 1;	/* trust on first use */
216 		if ((e = calloc(1, sizeof(*e))) == NULL)
217 			abort();
218 		strlcpy(e->domain, host, sizeof(e->domain));
219 		if (*port != '\0' && strcmp(port, "1965")) {
220 			strlcat(e->domain, ":", sizeof(e->domain));
221 			strlcat(e->domain, port, sizeof(e->domain));
222 		}
223 		strlcpy(e->hash, hash, sizeof(e->hash));
224 		tofu_add(&certs, e);
225 		ui_send_fs(IMSG_SAVE_CERT, tab->id, e, sizeof(*e));
226 	} else
227 		tofu_res = !strcmp(hash, e->hash);
228 
229 	if (tofu_res) {
230 		if (e->verified == -1)
231 			tab->trust = TS_TEMP_TRUSTED;
232 		else if (e->verified == 1)
233 			tab->trust = TS_VERIFIED;
234 		else
235 			tab->trust = TS_TRUSTED;
236 
237 		ui_send_net(IMSG_CERT_STATUS, imsg->hdr.peerid,
238 		    &tofu_res, sizeof(tofu_res));
239 	} else {
240 		tab->trust = TS_UNTRUSTED;
241 		load_page_from_str(tab, "# Certificate mismatch\n");
242 		if ((tab->cert = strdup(hash)) == NULL)
243 			die();
244 		ui_yornp("Certificate mismatch.  Proceed?",
245 		    handle_check_cert_user_choice, tab);
246 	}
247 }
248 
249 static void
250 handle_check_cert_user_choice(int accept, struct tab *tab)
251 {
252 	ui_send_net(IMSG_CERT_STATUS, tab->id, &accept,
253 	    sizeof(accept));
254 
255 	if (accept) {
256 		/*
257 		 * trust the certificate for this session only.  If
258 		 * the page results in a redirect while we're asking
259 		 * the user to save, we'll end up with an invalid
260 		 * tabid (one request == one tab id) and crash.  It
261 		 * also makes sense to save it for the current session
262 		 * if the user accepted it.
263 		 */
264 		tofu_temp_trust(&certs, tab->uri.host, tab->uri.port, tab->cert);
265 
266 		ui_yornp("Save the new certificate?",
267 		    handle_maybe_save_new_cert, tab);
268 	} else {
269 		free(tab->cert);
270 		tab->cert = NULL;
271 	}
272 }
273 
274 static void
275 handle_maybe_save_new_cert(int accept, struct tab *tab)
276 {
277 	struct tofu_entry *e;
278 	const char *host, *port;
279 
280 	if (tab->proxy != NULL) {
281 		host = tab->proxy->host;
282 		port = tab->proxy->port;
283 	} else {
284 		host = tab->uri.host;
285 		port = tab->uri.port;
286 	}
287 
288 	if (!accept)
289 		goto end;
290 
291 	if ((e = calloc(1, sizeof(*e))) == NULL)
292 		die();
293 
294 	strlcpy(e->domain, host, sizeof(e->domain));
295 	if (*port != '\0' && strcmp(port, "1965")) {
296 		strlcat(e->domain, ":", sizeof(e->domain));
297 		strlcat(e->domain, port, sizeof(e->domain));
298 	}
299 	strlcpy(e->hash, tab->cert, sizeof(e->hash));
300 	ui_send_fs(IMSG_UPDATE_CERT, 0, e, sizeof(*e));
301 
302 	tofu_update(&certs, e);
303 
304 	tab->trust = TS_TRUSTED;
305 
306 end:
307 	free(tab->cert);
308 	tab->cert = NULL;
309 }
310 
311 static inline int
312 normalize_code(int n)
313 {
314 	if (n < 20) {
315 		if (n == 10 || n == 11)
316 			return n;
317 		return 10;
318 	} else if (n < 30) {
319 		return 20;
320 	} else if (n < 40) {
321 		if (n == 30 || n == 31)
322 			return n;
323 		return 30;
324 	} else if (n < 50) {
325 		if (n <= 44)
326 			return n;
327 		return 40;
328 	} else if (n < 60) {
329 		if (n <= 53 || n == 59)
330 			return n;
331 		return 50;
332 	} else if (n < 70) {
333 		if (n <= 62)
334 			return n;
335 		return 60;
336 	} else
337 		return MALFORMED_RESPONSE;
338 }
339 
340 static void
341 handle_imsg_got_code(struct imsg *imsg, size_t datalen)
342 {
343 	struct tab	*tab;
344 
345 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
346 		return;
347 
348 	if (sizeof(tab->code) != datalen)
349 		die();
350 
351 	memcpy(&tab->code, imsg->data, sizeof(tab->code));
352 	tab->code = normalize_code(tab->code);
353 	if (tab->code != 30 && tab->code != 31)
354 		tab->redirect_count = 0;
355 }
356 
357 static void
358 handle_imsg_got_meta(struct imsg *imsg, size_t datalen)
359 {
360 	struct tab	*tab;
361 
362 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
363 		return;
364 
365 	if (sizeof(tab->meta) <= datalen)
366 		die();
367 
368 	memcpy(tab->meta, imsg->data, datalen);
369 
370 	if (tab->code < 10) {	/* internal errors */
371 		load_page_from_str(tab, err_pages[tab->code]);
372 	} else if (tab->code < 20) {	/* 1x */
373 		load_page_from_str(tab, err_pages[tab->code]);
374 		ui_require_input(tab, tab->code == 11, PROTO_GEMINI);
375 	} else if (tab->code == 20) {
376 		if (setup_parser_for(tab)) {
377 			ui_send_net(IMSG_PROCEED, tab->id, NULL, 0);
378 		} else {
379 			load_page_from_str(tab, err_pages[UNKNOWN_TYPE_OR_CSET]);
380 			ui_yornp("Can't display page, wanna save?",
381 			    handle_maybe_save_page, tab);
382 		}
383 	} else if (tab->code < 40) { /* 3x */
384 		tab->redirect_count++;
385 
386 		/* TODO: make customizable? */
387 		if (tab->redirect_count > 5) {
388 			load_page_from_str(tab,
389 			    err_pages[TOO_MUCH_REDIRECTS]);
390 		} else
391 			do_load_url(tab, tab->meta, NULL);
392 	} else { /* 4x, 5x & 6x */
393 		load_page_from_str(tab, err_pages[tab->code]);
394 	}
395 }
396 
397 static void
398 handle_maybe_save_page(int dosave, struct tab *tab)
399 {
400 	if (dosave)
401 		ui_read("Save to path", handle_save_page_path, tab);
402 	else
403 		stop_tab(tab);
404 }
405 
406 static void
407 handle_save_page_path(const char *path, struct tab *tab)
408 {
409 	if (path == NULL) {
410 		stop_tab(tab);
411 		return;
412 	}
413 
414 	tab->path = strdup(path);
415 
416 	ui_send_fs(IMSG_FILE_OPEN, tab->id, path, strlen(path)+1);
417 }
418 
419 static void
420 handle_imsg_file_opened(struct imsg *imsg, size_t datalen)
421 {
422 	struct tab	*tab;
423 	char		*page;
424 	const char	*e;
425 	int		 l;
426 
427 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL) {
428 		if (imsg->fd != -1)
429 			close(imsg->fd);
430 		return;
431 	}
432 
433 	if (imsg->fd == -1) {
434 		stop_tab(tab);
435 
436 		e = imsg->data;
437 		if (e[datalen-1] != '\0')
438 			die();
439 		l = asprintf(&page, "# Can't open file\n\n> %s: %s\n",
440 		    tab->path, e);
441 		if (l == -1)
442 			die();
443 		load_page_from_str(tab, page);
444 		free(page);
445 	} else {
446 		tab->fd = imsg->fd;
447 		ui_send_net(IMSG_PROCEED, tab->id, NULL, 0);
448 	}
449 }
450 
451 static void
452 handle_imsg_buf(struct imsg *imsg, size_t datalen)
453 {
454 	struct tab	*tab;
455 	int		 l;
456 	char		*page, buf[FMT_SCALED_STRSIZE] = {0};
457 
458 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
459 		return;
460 
461 	tab->bytes += datalen;
462 	if (tab->fd == -1) {
463 		if (!parser_parse(tab, imsg->data, datalen))
464 			die();
465 	} else {
466 		write(tab->fd, imsg->data, datalen);
467 		fmt_scaled(tab->bytes, buf);
468 		l = asprintf(&page, "Saving to \"%s\"... (%s)\n",
469 		    tab->path,
470 		    buf);
471 		if (l == -1)
472 			die();
473 		load_page_from_str(tab, page);
474 		free(page);
475 	}
476 
477 	ui_on_tab_refresh(tab);
478 }
479 
480 static void
481 handle_imsg_eof(struct imsg *imsg, size_t datalen)
482 {
483 	struct tab	*tab;
484 	int		 l;
485 	char		*page, buf[FMT_SCALED_STRSIZE] = {0};
486 
487 	if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
488 		return;
489 
490 	if (tab->fd == -1) {
491 		if (!parser_free(tab))
492 			die();
493 	} else {
494 		fmt_scaled(tab->bytes, buf);
495 		l = asprintf(&page, "Saved to \"%s\" (%s)\n",
496 		    tab->path,
497 		    buf);
498 		if (l == -1)
499 			die();
500 		load_page_from_str(tab, page);
501 		free(page);
502 
503 		close(tab->fd);
504 		tab->fd = -1;
505 		free(tab->path);
506 		tab->path = NULL;
507 	}
508 
509 	ui_on_tab_refresh(tab);
510 	ui_on_tab_loaded(tab);
511 }
512 
513 static void
514 handle_imsg_bookmark_ok(struct imsg *imsg, size_t datalen)
515 {
516 	int res;
517 
518 	if (datalen != sizeof(res))
519 		die();
520 
521 	memcpy(&res, imsg->data, sizeof(res));
522 	if (res == 0)
523 		message("Added to bookmarks!");
524 	else
525 		message("Failed to add to bookmarks: %s",
526 		    strerror(res));
527 }
528 
529 static void
530 handle_imsg_save_cert_ok(struct imsg *imsg, size_t datalen)
531 {
532 	int res;
533 
534 	if (datalen != sizeof(res))
535 		die();
536 	memcpy(&res, imsg->data, datalen);
537 	if (res != 0)
538 		message("Failed to save the cert for: %s",
539 		    strerror(res));
540 }
541 
542 static void
543 handle_imsg_update_cert_ok(struct imsg *imsg, size_t datalen)
544 {
545 	int res;
546 
547 	if (datalen != sizeof(res))
548 		die();
549 	memcpy(&res, imsg->data, datalen);
550 	if (!res)
551 		message("Failed to update the certificate");
552 }
553 
554 static void
555 handle_dispatch_imsg(int fd, short ev, void *d)
556 {
557 	struct imsgev	*iev = d;
558 
559 	if (dispatch_imsg(iev, ev, handlers, sizeof(handlers)) == -1)
560 		err(1, "connection closed");
561 }
562 
563 static int
564 load_page_from_str(struct tab *tab, const char *page)
565 {
566 	erase_buffer(&tab->buffer);
567 	parser_init(tab, gemtext_initparser);
568 	if (!tab->buffer.page.parse(&tab->buffer.page, page, strlen(page)))
569 		die();
570 	if (!tab->buffer.page.free(&tab->buffer.page))
571 		die();
572 	ui_on_tab_refresh(tab);
573 	ui_on_tab_loaded(tab);
574 	return 0;
575 }
576 
577 static int
578 load_about_url(struct tab *tab, const char *url)
579 {
580 	tab->trust = TS_VERIFIED;
581 
582 	parser_init(tab, gemtext_initparser);
583 
584 	ui_send_fs(IMSG_GET, tab->id,
585 	    tab->hist_cur->h, strlen(tab->hist_cur->h)+1);
586 
587 	return 1;
588 }
589 
590 static int
591 load_finger_url(struct tab *tab, const char *url)
592 {
593 	struct get_req	 req;
594 	size_t		 len;
595 	char		*at;
596 
597 	memset(&req, 0, sizeof(req));
598 	strlcpy(req.port, tab->uri.port, sizeof(req.port));
599 
600 	/*
601 	 * Sometimes the finger url have the user as path component
602 	 * (e.g. finger://thelambdalab.xyz/plugd), sometimes as
603 	 * userinfo (e.g. finger://cobradile@finger.farm).
604 	 */
605 	if ((at = strchr(tab->uri.host, '@')) != NULL) {
606 		len = at - tab->uri.host;
607 		memcpy(req.req, tab->uri.host, len);
608 
609 		if (len >= sizeof(req.req))
610 			die();
611 		req.req[len] = '\0';
612 
613 		strlcpy(req.host, at+1, sizeof(req.host));
614 	} else {
615 		strlcpy(req.host, tab->uri.host, sizeof(req.host));
616 
617 		/* +1 to skip the initial `/' */
618 		strlcpy(req.req, tab->uri.path+1, sizeof(req.req));
619 	}
620 	strlcat(req.req, "\r\n", sizeof(req.req));
621 
622 	parser_init(tab, textplain_initparser);
623 	return make_request(tab, &req, PROTO_FINGER, NULL);
624 }
625 
626 static int
627 load_gemini_url(struct tab *tab, const char *url)
628 {
629 	struct get_req	 req;
630 
631 	memset(&req, 0, sizeof(req));
632 	strlcpy(req.host, tab->uri.host, sizeof(req.host));
633 	strlcpy(req.port, tab->uri.port, sizeof(req.host));
634 
635 	return make_request(tab, &req, PROTO_GEMINI, tab->hist_cur->h);
636 }
637 
638 static int
639 load_gopher_url(struct tab *tab, const char *url)
640 {
641 	struct get_req	 req;
642 	const char	*path;
643 
644 	memset(&req, 0, sizeof(req));
645 	strlcpy(req.host, tab->uri.host, sizeof(req.host));
646 	strlcpy(req.port, tab->uri.port, sizeof(req.host));
647 
648 	path = tab->uri.path;
649 	if (!strcmp(path, "/") || *path == '\0') {
650 		/* expect the top directory to be a gophermap */
651 		parser_init(tab, gophermap_initparser);
652 	} else if (has_prefix(path, "/1/")) {
653 		/* gophermap menu/submenu */
654 		parser_init(tab, gophermap_initparser);
655 		path += 2;
656 	} else if (has_prefix(path, "/0/")) {
657 		parser_init(tab, textplain_initparser);
658 		path += 2;
659 	} else if (has_prefix(path, "/7/")) {
660 		/* show input request if there is no query */
661 		ui_require_input(tab, 0, PROTO_GOPHER);
662 		return load_page_from_str(tab, err_pages[10]);
663 	} else {
664 		return 0;
665 	}
666 
667 	strlcpy(req.req, path, sizeof(req.req));
668 	if (*tab->uri.query != '\0') {
669 		strlcat(req.req, "?", sizeof(req.req));
670 		strlcat(req.req, tab->uri.query, sizeof(req.req));
671 	}
672 	strlcpy(req.req, "\r\n", sizeof(req.req));
673 
674 	return make_request(tab, &req, PROTO_GOPHER, path);
675 }
676 
677 static int
678 load_via_proxy(struct tab *tab, const char *url, struct proxy *p)
679 {
680 	struct get_req req;
681 
682 	memset(&req, 0, sizeof(req));
683 	strlcpy(req.host, p->host, sizeof(req.host));
684 	strlcpy(req.port, p->port, sizeof(req.host));
685 
686 	tab->proxy = p;
687 
688 	return make_request(tab, &req, p->proto, tab->hist_cur->h);
689 }
690 
691 static int
692 make_request(struct tab *tab, struct get_req *req, int proto, const char *r)
693 {
694 	stop_tab(tab);
695 	tab->id = tab_new_id();
696 	req->proto = proto;
697 
698 	if (r != NULL) {
699 		strlcpy(req->req, r, sizeof(req->req));
700 		strlcat(req->req, "\r\n", sizeof(req->req));
701 	}
702 
703 	ui_send_net(IMSG_GET_RAW, tab->id, req, sizeof(*req));
704 
705 	/*
706 	 * So the various load_*_url can `return make_request` and
707 	 * do_load_url is happy.
708 	 */
709 	return 1;
710 }
711 
712 void
713 gopher_send_search_req(struct tab *tab, const char *text)
714 {
715 	struct get_req	req;
716 
717 	start_loading_anim(tab);
718 
719 	memset(&req, 0, sizeof(req));
720 	strlcpy(req.host, tab->uri.host, sizeof(req.host));
721 	strlcpy(req.port, tab->uri.port, sizeof(req.host));
722 
723 	/* +2 to skip /7 */
724 	strlcpy(req.req, tab->uri.path+2, sizeof(req.req));
725 	if (*tab->uri.query != '\0') {
726 		strlcat(req.req, "?", sizeof(req.req));
727 		strlcat(req.req, tab->uri.query, sizeof(req.req));
728 	}
729 
730 	strlcat(req.req, "\t", sizeof(req.req));
731 	strlcat(req.req, text, sizeof(req.req));
732 	strlcat(req.req, "\r\n", sizeof(req.req));
733 
734 	erase_buffer(&tab->buffer);
735 	parser_init(tab, gophermap_initparser);
736 
737 	make_request(tab, &req, PROTO_GOPHER, NULL);
738 }
739 
740 /*
741  * Effectively load the given url in the given tab.  Return 1 when
742  * loading the page asynchronously, and thus when an erase_buffer can
743  * be done right after this function return, or 0 when loading the
744  * page synchronously.  In this last case, erase_buffer *MUST* be
745  * called by the handling function (such as load_page_from_str).
746  */
747 static int
748 do_load_url(struct tab *tab, const char *url, const char *base)
749 {
750 	struct phos_uri	 uri;
751 	struct proto	*p;
752 	struct proxy	*proxy;
753 	char		*t;
754 
755 	tab->proxy = NULL;
756 
757 	if (tab->fd != -1) {
758 		close(tab->fd);
759 		tab->fd = -1;
760 		free(tab->path);
761 		tab->path = NULL;
762 	}
763 
764 	tab->trust = TS_UNKNOWN;
765 
766 	if (base == NULL)
767 		memcpy(&uri, &tab->uri, sizeof(tab->uri));
768 	else
769 		phos_parse_absolute_uri(base, &uri);
770 
771 	if (!phos_resolve_uri_from_str(&uri, url, &tab->uri)) {
772 		if (asprintf(&t, "#error loading %s\n>%s\n",
773 		    url, "Can't parse the URI") == -1)
774 			die();
775 		strlcpy(tab->hist_cur->h, url, sizeof(tab->hist_cur->h));
776 		load_page_from_str(tab, t);
777 		free(t);
778 		return 0;
779 	}
780 
781 	phos_serialize_uri(&tab->uri, tab->hist_cur->h,
782 	    sizeof(tab->hist_cur->h));
783 
784 	for (p = protos; p->schema != NULL; ++p) {
785 		if (!strcmp(tab->uri.scheme, p->schema)) {
786 			/* patch the port */
787 			if (*tab->uri.port == '\0' && p->port != NULL)
788 				strlcpy(tab->uri.port, p->port,
789 				    sizeof(tab->uri.port));
790 
791 			return p->loadfn(tab, url);
792 		}
793 	}
794 
795 	TAILQ_FOREACH(proxy, &proxies, proxies) {
796 		if (!strcmp(tab->uri.scheme, proxy->match_proto))
797 			return load_via_proxy(tab, url, proxy);
798 	}
799 
800 	return load_page_from_str(tab, err_pages[UNKNOWN_PROTOCOL]);
801 }
802 
803 /*
804  * Load url in tab.  If tab is marked as lazy, only prepare the url
805  * but don't load it.  If tab is lazy and a url was already prepared,
806  * do load it!
807  */
808 void
809 load_url(struct tab *tab, const char *url, const char *base, int nohist)
810 {
811 	int lazy;
812 
813 	lazy = tab->flags & TAB_LAZY;
814 
815 	if (lazy && tab->hist_cur != NULL) {
816 		lazy = 0;
817 		tab->flags &= ~TAB_LAZY;
818 	}
819 
820 	/* can't have both nohist and lazy at the same time. */
821 	if (nohist && lazy)
822 		nohist = 0;
823 
824 	if (!nohist && (!lazy || tab->hist_cur == NULL)) {
825 		if (tab->hist_cur != NULL)
826 			hist_clear_forward(&tab->hist,
827 			    TAILQ_NEXT(tab->hist_cur, entries));
828 
829 		if ((tab->hist_cur = calloc(1, sizeof(*tab->hist_cur))) == NULL) {
830 			event_loopbreak();
831 			return;
832 		}
833 
834 		strlcpy(tab->buffer.page.title, url,
835 		    sizeof(tab->buffer.page.title));
836 		hist_push(&tab->hist, tab->hist_cur);
837 
838 		if (lazy)
839 			strlcpy(tab->hist_cur->h, url,
840 			    sizeof(tab->hist_cur->h));
841 	}
842 
843 	if (!lazy && do_load_url(tab, url, base))
844 		erase_buffer(&tab->buffer);
845 }
846 
847 int
848 load_previous_page(struct tab *tab)
849 {
850 	struct hist	*h;
851 
852 	if ((h = TAILQ_PREV(tab->hist_cur, mhisthead, entries)) == NULL)
853 		return 0;
854 	tab->hist_cur = h;
855 	do_load_url(tab, h->h, NULL);
856 	return 1;
857 }
858 
859 int
860 load_next_page(struct tab *tab)
861 {
862 	struct hist	*h;
863 
864 	if ((h = TAILQ_NEXT(tab->hist_cur, entries)) == NULL)
865 		return 0;
866 	tab->hist_cur = h;
867 	do_load_url(tab, h->h, NULL);
868 	return 1;
869 }
870 
871 /*
872  * Free every resource linked to the tab, including the tab itself.
873  * Removes the tab from the tablist, but doesn't update the
874  * current_tab though.
875  */
876 void
877 free_tab(struct tab *tab)
878 {
879 	stop_tab(tab);
880 
881 	if (evtimer_pending(&tab->loadingev, NULL))
882 		evtimer_del(&tab->loadingev);
883 
884 	TAILQ_REMOVE(&tabshead, tab, tabs);
885 	free(tab);
886 }
887 
888 void
889 stop_tab(struct tab *tab)
890 {
891 	ui_send_net(IMSG_STOP, tab->id, NULL, 0);
892 
893 	if (tab->fd != -1) {
894 		close(tab->fd);
895 		tab->fd = -1;
896 		free(tab->path);
897 		tab->path = NULL;
898 		load_page_from_str(tab, "Stopped.\n");
899 	}
900 }
901 
902 void
903 add_to_bookmarks(const char *str)
904 {
905 	ui_send_fs(IMSG_BOOKMARK_PAGE, 0,
906 	    str, strlen(str)+1);
907 }
908 
909 void
910 save_session(void)
911 {
912 	struct tab	*tab;
913 	char		*t;
914 	int		 flags;
915 
916 	ui_send_fs(IMSG_SESSION_START, 0, NULL, 0);
917 
918 	TAILQ_FOREACH(tab, &tabshead, tabs) {
919 		flags = tab->flags;
920 		if (tab == current_tab)
921 			flags |= TAB_CURRENT;
922 
923 		t = tab->hist_cur->h;
924 		ui_send_fs(IMSG_SESSION_TAB, flags, t, strlen(t)+1);
925 
926 		t = tab->buffer.page.title;
927 		ui_send_fs(IMSG_SESSION_TAB_TITLE, 0, t, strlen(t)+1);
928 	}
929 
930 	ui_send_fs(IMSG_SESSION_END, 0, NULL, 0);
931 }
932 
933 /*
934  * Parse a line of the session file.  The format is:
935  *
936  *	URL [flags,...] [title]\n
937  */
938 static void
939 parse_session_line(char *line, const char **title, uint32_t *flags)
940 {
941 	char *s, *t, *ap;
942 
943 	*title = "";
944 	*flags = 0;
945 	if ((s = strchr(line, ' ')) == NULL)
946 		return;
947 
948 	*s++ = '\0';
949 
950         if ((t = strchr(s, ' ')) != NULL) {
951 		*t++ = '\0';
952 		*title = t;
953 	}
954 
955 	while ((ap = strsep(&s, ",")) != NULL) {
956 		if (*ap == '\0')
957 			;
958 		else if (!strcmp(ap, "current"))
959 			*flags |= TAB_CURRENT;
960 		else
961 			message("unknown tab flag: %s", ap);
962 	}
963 }
964 
965 static void
966 load_last_session(void)
967 {
968 	const char	*title;
969 	char		*nl, *line = NULL;
970 	uint32_t	 flags;
971 	size_t		 linesize = 0;
972 	ssize_t		 linelen;
973 	FILE		*session;
974 	struct tab	*tab, *curr = NULL;
975 
976 	if ((session = fopen(session_file, "r")) == NULL) {
977 		/* first time? */
978 		new_tab("about:new", NULL, NULL);
979 		switch_to_tab(new_tab("about:help", NULL, NULL));
980 		return;
981 	}
982 
983 	while ((linelen = getline(&line, &linesize, session)) != -1) {
984                 if ((nl = strchr(line, '\n')) != NULL)
985 			*nl = '\0';
986 		parse_session_line(line, &title, &flags);
987 		if ((tab = new_tab(line, NULL, NULL)) == NULL)
988 			err(1, "new_tab");
989                 strlcpy(tab->buffer.page.title, title,
990 		    sizeof(tab->buffer.page.title));
991 		if (flags & TAB_CURRENT)
992 			curr = tab;
993 	}
994 
995 	if (ferror(session))
996 		message("error reading %s: %s",
997 		    session_file, strerror(errno));
998 	fclose(session);
999 	free(line);
1000 
1001 	if (curr != NULL)
1002 		switch_to_tab(curr);
1003 
1004 	if (last_time_crashed())
1005 		switch_to_tab(new_tab("about:crash", NULL, NULL));
1006 
1007 	return;
1008 }
1009 
1010 static pid_t
1011 start_child(enum telescope_process p, const char *argv0, int fd)
1012 {
1013 	const char	*argv[4];
1014 	int		 argc = 0;
1015 	pid_t		 pid;
1016 
1017 	switch (pid = fork()) {
1018 	case -1:
1019 		die();
1020 	case 0:
1021 		break;
1022 	default:
1023 		close(fd);
1024 		return pid;
1025 	}
1026 
1027 	if (dup2(fd, 3) == -1)
1028 		err(1, "cannot setup imsg fd");
1029 
1030 	argv[argc++] = argv0;
1031 	switch (p) {
1032 	case PROC_UI:
1033 		errx(1, "Can't start ui process");
1034 	case PROC_FS:
1035 		argv[argc++] = "-Tf";
1036 		break;
1037 	case PROC_NET:
1038 		argv[argc++] = "-Tn";
1039 		break;
1040 	}
1041 
1042 	argv[argc++] = NULL;
1043 	execvp(argv0, (char *const *)argv);
1044 	err(1, "execvp(%s)", argv0);
1045 }
1046 
1047 static int
1048 ui_send_net(int type, uint32_t peerid, const void *data,
1049     uint16_t datalen)
1050 {
1051 	return imsg_compose_event(iev_net, type, peerid, 0, -1, data,
1052 	    datalen);
1053 }
1054 
1055 static int
1056 ui_send_fs(int type, uint32_t peerid, const void *data, uint16_t datalen)
1057 {
1058 	return imsg_compose_event(iev_fs, type, peerid, 0, -1, data,
1059 	    datalen);
1060 }
1061 
1062 static void __attribute__((noreturn))
1063 usage(int r)
1064 {
1065 	fprintf(stderr, "USAGE: %s [-hnv] [-c config] [url]\n",
1066 	    getprogname());
1067 	fprintf(stderr, "version: " PACKAGE " " VERSION "\n");
1068 	exit(r);
1069 }
1070 
1071 int
1072 main(int argc, char * const *argv)
1073 {
1074 	struct imsgev	 net_ibuf, fs_ibuf;
1075 	int		 pipe2net[2], pipe2fs[2];
1076 	int		 ch, configtest = 0, fail = 0;
1077 	int		 has_url = 0;
1078 	int		 proc = -1;
1079 	int		 sessionfd;
1080 	char		 path[PATH_MAX];
1081 	const char	*url = NEW_TAB_URL;
1082 	const char	*argv0;
1083 
1084 	argv0 = argv[0];
1085 
1086 	signal(SIGCHLD, SIG_IGN);
1087 	signal(SIGPIPE, SIG_IGN);
1088 
1089 	if (getenv("NO_COLOR") != NULL)
1090 		enable_colors = 0;
1091 
1092 	strlcpy(path, getenv("HOME"), sizeof(path));
1093 	strlcat(path, "/.telescope/config", sizeof(path));
1094 
1095 	while ((ch = getopt_long(argc, argv, opts, longopts, NULL)) != -1) {
1096 		switch (ch) {
1097 		case 'C':
1098 			exit(ui_print_colors());
1099 		case 'c':
1100 			fail = 1;
1101 			strlcpy(path, optarg, sizeof(path));
1102 			break;
1103 		case 'n':
1104 			configtest = 1;
1105 			break;
1106 		case 'h':
1107 			usage(0);
1108 		case 'T':
1109 			switch (*optarg) {
1110 			case 'f':
1111 				proc = PROC_FS;
1112 				break;
1113 			case 'n':
1114 				proc = PROC_NET;
1115 				break;
1116 			default:
1117 				errx(1, "invalid process spec %c",
1118 				    *optarg);
1119 			}
1120 			break;
1121 		case 'v':
1122 			printf("%s %s\n", PACKAGE_NAME, PACKAGE_VERSION);
1123 			exit(0);
1124 			break;
1125 		default:
1126 			usage(1);
1127 		}
1128 	}
1129 
1130 	argc -= optind;
1131 	argv += optind;
1132 
1133 	if (proc != -1) {
1134 		if (argc > 0)
1135 			usage(1);
1136 		else if (proc == PROC_FS)
1137 			return fs_main();
1138 		else if (proc == PROC_NET)
1139 			return net_main();
1140 		else
1141 			usage(1);
1142 	}
1143 
1144 	if (argc != 0) {
1145 		has_url = 1;
1146 		url = argv[0];
1147 	}
1148 
1149 	/* setup keys before reading the config */
1150 	TAILQ_INIT(&global_map.m);
1151 	global_map.unhandled_input = global_key_unbound;
1152 	TAILQ_INIT(&minibuffer_map.m);
1153 
1154 	config_init();
1155 	parseconfig(path, fail);
1156 	if (configtest){
1157 		puts("config OK");
1158 		exit(0);
1159 	}
1160 
1161 	fs_init();
1162 	if ((sessionfd = lock_session()) == -1)
1163 		errx(1, "can't lock session, is another instance of "
1164 		    "telescope already running?");
1165 
1166 	/* Start children. */
1167 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe2fs) == -1)
1168 		err(1, "socketpair");
1169 	start_child(PROC_FS, argv0, pipe2fs[1]);
1170 	imsg_init(&fs_ibuf.ibuf, pipe2fs[0]);
1171 	iev_fs = &fs_ibuf;
1172 	iev_fs->handler = handle_dispatch_imsg;
1173 
1174 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe2net) == -1)
1175 		err(1, "socketpair");
1176 	start_child(PROC_NET, argv0, pipe2net[1]);
1177 	imsg_init(&net_ibuf.ibuf, pipe2net[0]);
1178 	iev_net = &net_ibuf;
1179 	iev_net->handler = handle_dispatch_imsg;
1180 
1181 	setproctitle("ui");
1182 
1183 	/* initialize tofu & load certificates */
1184 	tofu_init(&certs, 5, offsetof(struct tofu_entry, domain));
1185 	load_certs(&certs);
1186 
1187 	event_init();
1188 
1189 	/* Setup event handlers for pipes to fs/net */
1190 	iev_fs->events = EV_READ;
1191 	event_set(&iev_fs->ev, iev_fs->ibuf.fd, iev_fs->events,
1192 	    iev_fs->handler, iev_fs);
1193 	event_add(&iev_fs->ev, NULL);
1194 
1195 	iev_net->events = EV_READ;
1196 	event_set(&iev_net->ev, iev_net->ibuf.fd, iev_net->events,
1197 	    iev_net->handler, iev_net);
1198 	event_add(&iev_net->ev, NULL);
1199 
1200 	if (ui_init()) {
1201 		load_last_session();
1202 		if (has_url || TAILQ_EMPTY(&tabshead))
1203 			new_tab(url, NULL, NULL);
1204 
1205 		sandbox_ui_process();
1206 		ui_main_loop();
1207 		ui_end();
1208 	}
1209 
1210 	ui_send_fs(IMSG_QUIT, 0, NULL, 0);
1211 	ui_send_net(IMSG_QUIT, 0, NULL, 0);
1212 	imsg_flush(&iev_fs->ibuf);
1213 	imsg_flush(&iev_net->ibuf);
1214 
1215 	close(sessionfd);
1216 
1217 	return 0;
1218 }