3 #include <sys/socket.h>
13 struct event netev, fsev;
14 struct tabshead tabshead;
16 /* the first is also the fallback one */
17 static struct proto protos[] = {
18 { "gemini:", load_gemini_url },
19 { "about:", load_about_url },
23 static struct imsgbuf *netibuf, *fsibuf;
25 static void die(void) __attribute__((__noreturn__));
26 static struct tab *tab_by_id(uint32_t);
27 static void handle_imsg_err(struct imsg*, size_t);
28 static void handle_imsg_check_cert(struct imsg*, size_t);
29 static void handle_imsg_got_code(struct imsg*, size_t);
30 static void handle_imsg_got_meta(struct imsg*, size_t);
31 static void handle_imsg_buf(struct imsg*, size_t);
32 static void handle_imsg_eof(struct imsg*, size_t);
33 static void handle_imsg_bookmark_ok(struct imsg*, size_t);
34 static void dispatch_imsg(int, short, void*);
35 static void load_page_from_str(struct tab*, const char*);
36 static void do_load_url(struct tab*, const char*);
38 static imsg_handlerfn *handlers[] = {
39 [IMSG_ERR] = handle_imsg_err,
40 [IMSG_CHECK_CERT] = handle_imsg_check_cert,
41 [IMSG_GOT_CODE] = handle_imsg_got_code,
42 [IMSG_GOT_META] = handle_imsg_got_meta,
43 [IMSG_BUF] = handle_imsg_buf,
44 [IMSG_EOF] = handle_imsg_eof,
45 [IMSG_BOOKMARK_OK] = handle_imsg_bookmark_ok,
48 static void __attribute__((__noreturn__))
55 tab_by_id(uint32_t id)
59 TAILQ_FOREACH(t, &tabshead, tabs) {
68 handle_imsg_err(struct imsg *imsg, size_t datalen)
73 tab = tab_by_id(imsg->hdr.peerid);
76 page[datalen-1] = '\0';
78 if (asprintf(&page, "# Error loading %s\n\n> %s\n",
79 tab->hist_cur->h, page) == -1)
81 load_page_from_str(tab, page);
86 handle_imsg_check_cert(struct imsg *imsg, size_t datalen)
90 imsg_compose(netibuf, IMSG_CERT_STATUS, imsg->hdr.peerid, 0, -1, &tofu_res, sizeof(tofu_res));
98 if (n == 10 || n == 11)
104 if (n == 30 || n == 31)
112 if (n <= 53 || n == 59)
120 return MALFORMED_RESPONSE;
124 handle_imsg_got_code(struct imsg *imsg, size_t datalen)
128 tab = tab_by_id(imsg->hdr.peerid);
130 if (sizeof(tab->code) != datalen)
133 memcpy(&tab->code, imsg->data, sizeof(tab->code));
134 tab->code = normalize_code(tab->code);
135 if (tab->code != 30 && tab->code != 31)
136 tab->redirect_count = 0;
140 handle_imsg_got_meta(struct imsg *imsg, size_t datalen)
144 tab = tab_by_id(imsg->hdr.peerid);
146 if (sizeof(tab->meta) <= datalen)
149 memcpy(tab->meta, imsg->data, datalen);
151 if (tab->code < 10) { /* internal errors */
152 load_page_from_str(tab, err_pages[tab->code]);
153 } else if (tab->code < 20) { /* 1x */
154 load_page_from_str(tab, err_pages[tab->code]);
155 ui_require_input(tab, tab->code == 11);
156 } else if (tab->code == 20) {
157 if (setup_parser_for(tab)) {
158 imsg_compose(netibuf, IMSG_PROCEED, tab->id, 0, -1, NULL, 0);
161 load_page_from_str(tab, err_pages[UNKNOWN_TYPE_OR_CSET]);
163 } else if (tab->code < 40) { /* 3x */
164 tab->redirect_count++;
166 /* TODO: make customizable? */
167 if (tab->redirect_count > 5) {
168 load_page_from_str(tab,
169 err_pages[TOO_MUCH_REDIRECTS]);
171 do_load_url(tab, tab->meta);
172 } else { /* 4x, 5x & 6x */
173 load_page_from_str(tab, err_pages[tab->code]);
178 handle_imsg_buf(struct imsg *imsg, size_t datalen)
182 tab = tab_by_id(imsg->hdr.peerid);
184 if (!tab->page.parse(&tab->page, imsg->data, datalen))
187 ui_on_tab_refresh(tab);
191 handle_imsg_eof(struct imsg *imsg, size_t datalen)
195 t = tab_by_id(imsg->hdr.peerid);
196 if (!t->page.free(&t->page))
199 ui_on_tab_refresh(t);
204 handle_imsg_bookmark_ok(struct imsg *imsg, size_t datalen)
208 if (datalen != sizeof(res))
211 memcpy(&res, imsg->data, sizeof(res));
213 ui_notify("Added to bookmarks!");
215 ui_notify("Failed to add to bookmarks: %s",
220 dispatch_imsg(int fd, short ev, void *d)
222 struct imsgbuf *ibuf = d;
227 if ((n = imsg_read(ibuf)) == -1) {
228 if (errno == EAGAIN || errno == EWOULDBLOCK)
237 if ((n = imsg_get(ibuf, &imsg)) == -1)
241 datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
242 handlers[imsg.hdr.type](&imsg, datalen);
248 load_page_from_str(struct tab *tab, const char *page)
250 gemtext_initparser(&tab->page);
251 if (!tab->page.parse(&tab->page, page, strlen(page)))
253 if (!tab->page.free(&tab->page))
255 ui_on_tab_refresh(tab);
256 ui_on_tab_loaded(tab);
260 load_about_url(struct tab *tab, const char *url)
266 memset(&tab->url, 0, sizeof(tab->url));
268 m = strchr(url, ':');
269 strlcpy(tab->url.scheme, "about", sizeof(tab->url.scheme));
270 strlcpy(tab->url.path, m+1, sizeof(tab->url.path));
272 len = sizeof(tab->hist_cur->h)-1;
273 strlcpy(tab->hist_cur->h, url, len);
275 gemtext_initparser(&tab->page);
277 imsg_compose(fsibuf, IMSG_GET, tab->id, 0, -1,
278 tab->hist_cur->h, len+1);
283 load_gemini_url(struct tab *tab, const char *url)
289 len = sizeof(tab->hist_cur->h)-1;
291 if (has_prefix(url, "gemini:")) {
292 if (!url_parse(url, &tab->url, &err))
295 if (!url_resolve_from(&tab->url, url, &err))
299 url_unparse(&tab->url, tab->hist_cur->h, len);
300 imsg_compose(netibuf, IMSG_GET, tab->id, 0, -1,
301 tab->hist_cur->h, len+1);
306 if (asprintf(&p, "#error loading %s\n>%s\n",
309 strlcpy(tab->hist_cur->h, url, len);
310 load_page_from_str(tab, p);
315 do_load_url(struct tab *tab, const char *url)
319 for (p = protos; p->schema != NULL; ++p) {
320 if (has_prefix(url, p->schema)) {
326 protos[0].loadfn(tab, url);
330 load_url(struct tab *tab, const char *url)
332 if (tab->hist_cur != NULL)
333 hist_clear_forward(&tab->hist, TAILQ_NEXT(tab->hist_cur, entries));
335 if ((tab->hist_cur = calloc(1, sizeof(*tab->hist_cur))) == NULL) {
339 hist_push(&tab->hist, tab->hist_cur);
340 do_load_url(tab, url);
344 load_previous_page(struct tab *tab)
348 if ((h = TAILQ_PREV(tab->hist_cur, mhisthead, entries)) == NULL)
351 do_load_url(tab, h->h);
356 load_next_page(struct tab *tab)
360 if ((h = TAILQ_NEXT(tab->hist_cur, entries)) == NULL)
363 do_load_url(tab, h->h);
368 stop_tab(struct tab *tab)
370 imsg_compose(netibuf, IMSG_STOP, tab->id, 0, -1, NULL, 0);
375 add_to_bookmarks(const char *str)
377 imsg_compose(fsibuf, IMSG_BOOKMARK_PAGE, 0, 0, -1, str, strlen(str)+1);
384 struct imsgbuf network_ibuf, fs_ibuf;
385 int net_fds[2], fs_fds[2];
390 signal(SIGCHLD, SIG_IGN);
392 if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, fs_fds) == -1)
393 err(1, "socketpair");
400 setproctitle("(%d) fs", pid);
402 imsg_init(&fs_ibuf, fs_fds[1]);
403 exit(fs_main(&fs_ibuf));
406 imsg_init(&fs_ibuf, fs_fds[0]);
410 if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, net_fds) == -1)
411 err(1, "socketpair");
418 setproctitle("(%d) client", pid);
421 imsg_init(&network_ibuf, net_fds[1]);
422 exit(client_main(&network_ibuf));
425 imsg_init(&network_ibuf, net_fds[0]);
426 netibuf = &network_ibuf;
429 setproctitle("(%d) ui", pid);
431 TAILQ_INIT(&tabshead);
435 event_set(&netev, netibuf->fd, EV_READ | EV_PERSIST, dispatch_imsg, netibuf);
436 event_add(&netev, NULL);
438 event_set(&fsev, fsibuf->fd, EV_READ | EV_PERSIST, dispatch_imsg, fsibuf);
439 event_add(&fsev, NULL);
443 sandbox_ui_process();
447 imsg_compose(netibuf, IMSG_QUIT, 0, 0, -1, NULL, 0);
450 imsg_compose(fsibuf, IMSG_QUIT, 0, 0, -1, NULL, 0);