Blob


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 */
17 #include "compat.h"
19 #include <sys/socket.h>
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>
30 #include "defaults.h"
31 #include "minibuffer.h"
32 #include "parser.h"
33 #include "telescope.h"
34 #include "ui.h"
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 };
43 static const char *opts = "Cc:hnT:v";
45 static struct imsgev *iev_fs, *iev_net;
47 struct tabshead tabshead = TAILQ_HEAD_INITIALIZER(tabshead);
48 struct proxylist proxies = TAILQ_HEAD_INITIALIZER(proxies);
50 enum telescope_process {
51 PROC_UI,
52 PROC_FS,
53 PROC_NET,
54 };
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
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",
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 };
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);
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 };
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 };
145 static struct ohash certs;
147 static void __attribute__((__noreturn__))
148 die(void)
150 abort(); /* TODO */
153 static struct tab *
154 tab_by_id(uint32_t id)
156 struct tab *t;
158 TAILQ_FOREACH(t, &tabshead, tabs) {
159 if (t->id == id)
160 return t;
163 return NULL;
166 static void
167 handle_imsg_err(struct imsg *imsg, size_t datalen)
169 struct tab *tab;
170 char *page;
172 if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
173 return;
175 page = imsg->data;
176 page[datalen-1] = '\0';
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);
185 static void
186 handle_imsg_check_cert(struct imsg *imsg, size_t datalen)
188 const char *hash, *host, *port;
189 int tofu_res;
190 struct tofu_entry *e;
191 struct tab *tab;
193 hash = imsg->data;
194 if (hash[datalen-1] != '\0')
195 abort();
197 if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
198 return;
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;
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();
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));
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);
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;
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);
249 static void
250 handle_check_cert_user_choice(int accept, struct tab *tab)
252 ui_send_net(IMSG_CERT_STATUS, tab->id, &accept,
253 sizeof(accept));
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);
266 ui_yornp("Save the new certificate?",
267 handle_maybe_save_new_cert, tab);
268 } else {
269 free(tab->cert);
270 tab->cert = NULL;
274 static void
275 handle_maybe_save_new_cert(int accept, struct tab *tab)
277 struct tofu_entry *e;
278 const char *host, *port;
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;
288 if (!accept)
289 goto end;
291 if ((e = calloc(1, sizeof(*e))) == NULL)
292 die();
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));
299 strlcpy(e->hash, tab->cert, sizeof(e->hash));
300 ui_send_fs(IMSG_UPDATE_CERT, 0, e, sizeof(*e));
302 tofu_update(&certs, e);
304 tab->trust = TS_TRUSTED;
306 end:
307 free(tab->cert);
308 tab->cert = NULL;
311 static inline int
312 normalize_code(int n)
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;
340 static void
341 handle_imsg_got_code(struct imsg *imsg, size_t datalen)
343 struct tab *tab;
345 if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
346 return;
348 if (sizeof(tab->code) != datalen)
349 die();
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;
357 static void
358 handle_imsg_got_meta(struct imsg *imsg, size_t datalen)
360 struct tab *tab;
362 if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
363 return;
365 if (sizeof(tab->meta) <= datalen)
366 die();
368 memcpy(tab->meta, imsg->data, datalen);
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);
383 } else if (tab->code < 40) { /* 3x */
384 tab->redirect_count++;
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]);
397 static void
398 handle_maybe_save_page(int dosave, struct tab *tab)
400 if (dosave)
401 ui_read("Save to path", handle_save_page_path, tab);
402 else
403 stop_tab(tab);
406 static void
407 handle_save_page_path(const char *path, struct tab *tab)
409 if (path == NULL) {
410 stop_tab(tab);
411 return;
414 tab->path = strdup(path);
416 ui_send_fs(IMSG_FILE_OPEN, tab->id, path, strlen(path)+1);
419 static void
420 handle_imsg_file_opened(struct imsg *imsg, size_t datalen)
422 struct tab *tab;
423 char *page;
424 const char *e;
425 int l;
427 if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL) {
428 if (imsg->fd != -1)
429 close(imsg->fd);
430 return;
433 if (imsg->fd == -1) {
434 stop_tab(tab);
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);
451 static void
452 handle_imsg_buf(struct imsg *imsg, size_t datalen)
454 struct tab *tab;
455 int l;
456 char *page, buf[FMT_SCALED_STRSIZE] = {0};
458 if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
459 return;
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);
477 ui_on_tab_refresh(tab);
480 static void
481 handle_imsg_eof(struct imsg *imsg, size_t datalen)
483 struct tab *tab;
484 int l;
485 char *page, buf[FMT_SCALED_STRSIZE] = {0};
487 if ((tab = tab_by_id(imsg->hdr.peerid)) == NULL)
488 return;
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);
503 close(tab->fd);
504 tab->fd = -1;
505 free(tab->path);
506 tab->path = NULL;
509 ui_on_tab_refresh(tab);
510 ui_on_tab_loaded(tab);
513 static void
514 handle_imsg_bookmark_ok(struct imsg *imsg, size_t datalen)
516 int res;
518 if (datalen != sizeof(res))
519 die();
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));
529 static void
530 handle_imsg_save_cert_ok(struct imsg *imsg, size_t datalen)
532 int res;
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));
542 static void
543 handle_imsg_update_cert_ok(struct imsg *imsg, size_t datalen)
545 int res;
547 if (datalen != sizeof(res))
548 die();
549 memcpy(&res, imsg->data, datalen);
550 if (!res)
551 message("Failed to update the certificate");
554 static void
555 handle_dispatch_imsg(int fd, short ev, void *d)
557 struct imsgev *iev = d;
559 if (dispatch_imsg(iev, ev, handlers, sizeof(handlers)) == -1)
560 err(1, "connection closed");
563 static int
564 load_page_from_str(struct tab *tab, const char *page)
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;
577 static int
578 load_about_url(struct tab *tab, const char *url)
580 tab->trust = TS_VERIFIED;
582 parser_init(tab, gemtext_initparser);
584 ui_send_fs(IMSG_GET, tab->id,
585 tab->hist_cur->h, strlen(tab->hist_cur->h)+1);
587 return 1;
590 static int
591 load_finger_url(struct tab *tab, const char *url)
593 struct get_req req;
594 size_t len;
595 char *at;
597 memset(&req, 0, sizeof(req));
598 strlcpy(req.port, tab->uri.port, sizeof(req.port));
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);
609 if (len >= sizeof(req.req))
610 die();
611 req.req[len] = '\0';
613 strlcpy(req.host, at+1, sizeof(req.host));
614 } else {
615 strlcpy(req.host, tab->uri.host, sizeof(req.host));
617 /* +1 to skip the initial `/' */
618 strlcpy(req.req, tab->uri.path+1, sizeof(req.req));
620 strlcat(req.req, "\r\n", sizeof(req.req));
622 parser_init(tab, textplain_initparser);
623 return make_request(tab, &req, PROTO_FINGER, NULL);
626 static int
627 load_gemini_url(struct tab *tab, const char *url)
629 struct get_req req;
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));
635 return make_request(tab, &req, PROTO_GEMINI, tab->hist_cur->h);
638 static int
639 load_gopher_url(struct tab *tab, const char *url)
641 struct get_req req;
642 const char *path;
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));
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;
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));
672 strlcpy(req.req, "\r\n", sizeof(req.req));
674 return make_request(tab, &req, PROTO_GOPHER, path);
677 static int
678 load_via_proxy(struct tab *tab, const char *url, struct proxy *p)
680 struct get_req req;
682 memset(&req, 0, sizeof(req));
683 strlcpy(req.host, p->host, sizeof(req.host));
684 strlcpy(req.port, p->port, sizeof(req.host));
686 tab->proxy = p;
688 return make_request(tab, &req, p->proto, tab->hist_cur->h);
691 static int
692 make_request(struct tab *tab, struct get_req *req, int proto, const char *r)
694 stop_tab(tab);
695 tab->id = tab_new_id();
696 req->proto = proto;
698 if (r != NULL) {
699 strlcpy(req->req, r, sizeof(req->req));
700 strlcat(req->req, "\r\n", sizeof(req->req));
703 ui_send_net(IMSG_GET_RAW, tab->id, req, sizeof(*req));
705 /*
706 * So the various load_*_url can `return make_request` and
707 * do_load_url is happy.
708 */
709 return 1;
712 void
713 gopher_send_search_req(struct tab *tab, const char *text)
715 struct get_req req;
717 start_loading_anim(tab);
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));
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));
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));
734 erase_buffer(&tab->buffer);
735 parser_init(tab, gophermap_initparser);
737 make_request(tab, &req, PROTO_GOPHER, NULL);
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)
750 struct phos_uri uri;
751 struct proto *p;
752 struct proxy *proxy;
753 char *t;
755 tab->proxy = NULL;
757 if (tab->fd != -1) {
758 close(tab->fd);
759 tab->fd = -1;
760 free(tab->path);
761 tab->path = NULL;
764 tab->trust = TS_UNKNOWN;
766 if (base == NULL)
767 memcpy(&uri, &tab->uri, sizeof(tab->uri));
768 else
769 phos_parse_absolute_uri(base, &uri);
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;
781 phos_serialize_uri(&tab->uri, tab->hist_cur->h,
782 sizeof(tab->hist_cur->h));
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));
791 return p->loadfn(tab, url);
795 TAILQ_FOREACH(proxy, &proxies, proxies) {
796 if (!strcmp(tab->uri.scheme, proxy->match_proto))
797 return load_via_proxy(tab, url, proxy);
800 return load_page_from_str(tab, err_pages[UNKNOWN_PROTOCOL]);
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)
811 int lazy;
813 lazy = tab->flags & TAB_LAZY;
815 if (lazy && tab->hist_cur != NULL) {
816 lazy = 0;
817 tab->flags &= ~TAB_LAZY;
820 if (!lazy || tab->hist_cur == NULL) {
821 if (tab->hist_cur != NULL)
822 hist_clear_forward(&tab->hist,
823 TAILQ_NEXT(tab->hist_cur, entries));
825 if ((tab->hist_cur = calloc(1, sizeof(*tab->hist_cur))) == NULL) {
826 event_loopbreak();
827 return;
830 strlcpy(tab->buffer.page.title, url,
831 sizeof(tab->buffer.page.title));
832 hist_push(&tab->hist, tab->hist_cur);
834 if (lazy)
835 strlcpy(tab->hist_cur->h, url,
836 sizeof(tab->hist_cur->h));
839 if (!lazy && do_load_url(tab, url, base))
840 erase_buffer(&tab->buffer);
843 int
844 load_previous_page(struct tab *tab)
846 struct hist *h;
848 if ((h = TAILQ_PREV(tab->hist_cur, mhisthead, entries)) == NULL)
849 return 0;
850 tab->hist_cur = h;
851 do_load_url(tab, h->h, NULL);
852 return 1;
855 int
856 load_next_page(struct tab *tab)
858 struct hist *h;
860 if ((h = TAILQ_NEXT(tab->hist_cur, entries)) == NULL)
861 return 0;
862 tab->hist_cur = h;
863 do_load_url(tab, h->h, NULL);
864 return 1;
867 /*
868 * Free every resource linked to the tab, including the tab itself.
869 * Removes the tab from the tablist, but doesn't update the
870 * current_tab though.
871 */
872 void
873 free_tab(struct tab *tab)
875 stop_tab(tab);
877 if (evtimer_pending(&tab->loadingev, NULL))
878 evtimer_del(&tab->loadingev);
880 TAILQ_REMOVE(&tabshead, tab, tabs);
881 free(tab);
884 void
885 stop_tab(struct tab *tab)
887 ui_send_net(IMSG_STOP, tab->id, NULL, 0);
889 if (tab->fd != -1) {
890 close(tab->fd);
891 tab->fd = -1;
892 free(tab->path);
893 tab->path = NULL;
894 load_page_from_str(tab, "Stopped.\n");
898 void
899 add_to_bookmarks(const char *str)
901 ui_send_fs(IMSG_BOOKMARK_PAGE, 0,
902 str, strlen(str)+1);
905 void
906 save_session(void)
908 struct tab *tab;
909 char *t;
910 int flags;
912 ui_send_fs(IMSG_SESSION_START, 0, NULL, 0);
914 TAILQ_FOREACH(tab, &tabshead, tabs) {
915 flags = tab->flags;
916 if (tab == current_tab)
917 flags |= TAB_CURRENT;
919 t = tab->hist_cur->h;
920 ui_send_fs(IMSG_SESSION_TAB, flags, t, strlen(t)+1);
922 t = tab->buffer.page.title;
923 ui_send_fs(IMSG_SESSION_TAB_TITLE, 0, t, strlen(t)+1);
926 ui_send_fs(IMSG_SESSION_END, 0, NULL, 0);
929 /*
930 * Parse a line of the session file. The format is:
932 * URL [flags,...] [title]\n
933 */
934 static void
935 parse_session_line(char *line, const char **title, uint32_t *flags)
937 char *s, *t, *ap;
939 *title = "";
940 *flags = 0;
941 if ((s = strchr(line, ' ')) == NULL)
942 return;
944 *s++ = '\0';
946 if ((t = strchr(s, ' ')) != NULL) {
947 *t++ = '\0';
948 *title = t;
951 while ((ap = strsep(&s, ",")) != NULL) {
952 if (*ap == '\0')
954 else if (!strcmp(ap, "current"))
955 *flags |= TAB_CURRENT;
956 else
957 message("unknown tab flag: %s", ap);
961 static void
962 load_last_session(void)
964 const char *title;
965 char *nl, *line = NULL;
966 uint32_t flags;
967 size_t linesize = 0;
968 ssize_t linelen;
969 FILE *session;
970 struct tab *tab, *curr = NULL;
972 if ((session = fopen(session_file, "r")) == NULL) {
973 /* first time? */
974 new_tab("about:new", NULL);
975 switch_to_tab(new_tab("about:help", NULL));
976 return;
979 while ((linelen = getline(&line, &linesize, session)) != -1) {
980 if ((nl = strchr(line, '\n')) != NULL)
981 *nl = '\0';
982 parse_session_line(line, &title, &flags);
983 if ((tab = new_tab(line, NULL)) == NULL)
984 err(1, "new_tab");
985 strlcpy(tab->buffer.page.title, title,
986 sizeof(tab->buffer.page.title));
987 if (flags & TAB_CURRENT)
988 curr = tab;
991 if (ferror(session))
992 message("error reading %s: %s",
993 session_file, strerror(errno));
994 fclose(session);
995 free(line);
997 if (curr != NULL)
998 switch_to_tab(curr);
1000 if (last_time_crashed())
1001 switch_to_tab(new_tab("about:crash", NULL));
1003 return;
1006 static pid_t
1007 start_child(enum telescope_process p, const char *argv0, int fd)
1009 const char *argv[4];
1010 int argc = 0;
1011 pid_t pid;
1013 switch (pid = fork()) {
1014 case -1:
1015 die();
1016 case 0:
1017 break;
1018 default:
1019 close(fd);
1020 return pid;
1023 if (dup2(fd, 3) == -1)
1024 err(1, "cannot setup imsg fd");
1026 argv[argc++] = argv0;
1027 switch (p) {
1028 case PROC_UI:
1029 errx(1, "Can't start ui process");
1030 case PROC_FS:
1031 argv[argc++] = "-Tf";
1032 break;
1033 case PROC_NET:
1034 argv[argc++] = "-Tn";
1035 break;
1038 argv[argc++] = NULL;
1039 execvp(argv0, (char *const *)argv);
1040 err(1, "execvp(%s)", argv0);
1043 static int
1044 ui_send_net(int type, uint32_t peerid, const void *data,
1045 uint16_t datalen)
1047 return imsg_compose_event(iev_net, type, peerid, 0, -1, data,
1048 datalen);
1051 static int
1052 ui_send_fs(int type, uint32_t peerid, const void *data, uint16_t datalen)
1054 return imsg_compose_event(iev_fs, type, peerid, 0, -1, data,
1055 datalen);
1058 static void __attribute__((noreturn))
1059 usage(int r)
1061 fprintf(stderr, "USAGE: %s [-hnv] [-c config] [url]\n",
1062 getprogname());
1063 fprintf(stderr, "version: " PACKAGE " " VERSION "\n");
1064 exit(r);
1067 int
1068 main(int argc, char * const *argv)
1070 struct imsgev net_ibuf, fs_ibuf;
1071 int pipe2net[2], pipe2fs[2];
1072 int ch, configtest = 0, fail = 0;
1073 int has_url = 0;
1074 int proc = -1;
1075 int sessionfd;
1076 char path[PATH_MAX];
1077 const char *url = NEW_TAB_URL;
1078 const char *argv0;
1080 argv0 = argv[0];
1082 signal(SIGCHLD, SIG_IGN);
1083 signal(SIGPIPE, SIG_IGN);
1085 if (getenv("NO_COLOR") != NULL)
1086 enable_colors = 0;
1088 strlcpy(path, getenv("HOME"), sizeof(path));
1089 strlcat(path, "/.telescope/config", sizeof(path));
1091 while ((ch = getopt_long(argc, argv, opts, longopts, NULL)) != -1) {
1092 switch (ch) {
1093 case 'C':
1094 exit(ui_print_colors());
1095 case 'c':
1096 fail = 1;
1097 strlcpy(path, optarg, sizeof(path));
1098 break;
1099 case 'n':
1100 configtest = 1;
1101 break;
1102 case 'h':
1103 usage(0);
1104 case 'T':
1105 switch (*optarg) {
1106 case 'f':
1107 proc = PROC_FS;
1108 break;
1109 case 'n':
1110 proc = PROC_NET;
1111 break;
1112 default:
1113 errx(1, "invalid process spec %c",
1114 *optarg);
1116 break;
1117 case 'v':
1118 printf("%s %s\n", PACKAGE_NAME, PACKAGE_VERSION);
1119 exit(0);
1120 break;
1121 default:
1122 usage(1);
1126 argc -= optind;
1127 argv += optind;
1129 if (proc != -1) {
1130 if (argc > 0)
1131 usage(1);
1132 else if (proc == PROC_FS)
1133 return fs_main();
1134 else if (proc == PROC_NET)
1135 return net_main();
1136 else
1137 usage(1);
1140 if (argc != 0) {
1141 has_url = 1;
1142 url = argv[0];
1145 /* setup keys before reading the config */
1146 TAILQ_INIT(&global_map.m);
1147 global_map.unhandled_input = global_key_unbound;
1148 TAILQ_INIT(&minibuffer_map.m);
1150 config_init();
1151 parseconfig(path, fail);
1152 if (configtest){
1153 puts("config OK");
1154 exit(0);
1157 fs_init();
1158 if ((sessionfd = lock_session()) == -1)
1159 errx(1, "can't lock session, is another instance of "
1160 "telescope already running?");
1162 /* Start children. */
1163 if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe2fs) == -1)
1164 err(1, "socketpair");
1165 start_child(PROC_FS, argv0, pipe2fs[1]);
1166 imsg_init(&fs_ibuf.ibuf, pipe2fs[0]);
1167 iev_fs = &fs_ibuf;
1168 iev_fs->handler = handle_dispatch_imsg;
1170 if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe2net) == -1)
1171 err(1, "socketpair");
1172 start_child(PROC_NET, argv0, pipe2net[1]);
1173 imsg_init(&net_ibuf.ibuf, pipe2net[0]);
1174 iev_net = &net_ibuf;
1175 iev_net->handler = handle_dispatch_imsg;
1177 setproctitle("ui");
1179 /* initialize tofu & load certificates */
1180 tofu_init(&certs, 5, offsetof(struct tofu_entry, domain));
1181 load_certs(&certs);
1183 event_init();
1185 /* Setup event handlers for pipes to fs/net */
1186 iev_fs->events = EV_READ;
1187 event_set(&iev_fs->ev, iev_fs->ibuf.fd, iev_fs->events,
1188 iev_fs->handler, iev_fs);
1189 event_add(&iev_fs->ev, NULL);
1191 iev_net->events = EV_READ;
1192 event_set(&iev_net->ev, iev_net->ibuf.fd, iev_net->events,
1193 iev_net->handler, iev_net);
1194 event_add(&iev_net->ev, NULL);
1196 if (ui_init()) {
1197 load_last_session();
1198 if (has_url || TAILQ_EMPTY(&tabshead))
1199 new_tab(url, NULL);
1201 sandbox_ui_process();
1202 ui_main_loop();
1203 ui_end();
1206 ui_send_fs(IMSG_QUIT, 0, NULL, 0);
1207 ui_send_net(IMSG_QUIT, 0, NULL, 0);
1208 imsg_flush(&iev_fs->ibuf);
1209 imsg_flush(&iev_net->ibuf);
1211 close(sessionfd);
1213 return 0;