op public repos

Blob

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