Blame


1 5e11c00c 2021-03-02 op #include "telescope.h"
2 5e11c00c 2021-03-02 op
3 5e11c00c 2021-03-02 op #include <sys/socket.h>
4 5e11c00c 2021-03-02 op
5 5e11c00c 2021-03-02 op #include <err.h>
6 5e11c00c 2021-03-02 op #include <errno.h>
7 5e11c00c 2021-03-02 op #include <event.h>
8 5e11c00c 2021-03-02 op #include <signal.h>
9 5e11c00c 2021-03-02 op #include <stdio.h>
10 5e11c00c 2021-03-02 op #include <stdlib.h>
11 5e11c00c 2021-03-02 op #include <string.h>
12 5e11c00c 2021-03-02 op #include <unistd.h>
13 5e11c00c 2021-03-02 op
14 5e11c00c 2021-03-02 op struct event imsgev;
15 5e11c00c 2021-03-02 op struct tabshead tabshead;
16 5e11c00c 2021-03-02 op
17 c07ac996 2021-03-12 op /* the first is also the fallback one */
18 c07ac996 2021-03-12 op static struct proto protos[] = {
19 4d3785b1 2021-03-09 op { "gemini:", load_gemini_url },
20 4d3785b1 2021-03-09 op { "about:", load_about_url },
21 4d3785b1 2021-03-09 op { NULL, NULL },
22 4d3785b1 2021-03-09 op };
23 4d3785b1 2021-03-09 op
24 5e11c00c 2021-03-02 op static struct imsgbuf *ibuf;
25 5e11c00c 2021-03-02 op
26 5e11c00c 2021-03-02 op static void handle_imsg_err(struct imsg*, size_t);
27 5e11c00c 2021-03-02 op static void handle_imsg_check_cert(struct imsg*, size_t);
28 5e11c00c 2021-03-02 op static void handle_imsg_got_code(struct imsg*, size_t);
29 5e11c00c 2021-03-02 op static void handle_imsg_got_meta(struct imsg*, size_t);
30 5e11c00c 2021-03-02 op static void handle_imsg_buf(struct imsg*, size_t);
31 5e11c00c 2021-03-02 op static void handle_imsg_eof(struct imsg*, size_t);
32 5e11c00c 2021-03-02 op
33 0972d8b2 2021-03-02 op static void load_page_from_str(struct tab*, const char*);
34 0972d8b2 2021-03-02 op
35 5e11c00c 2021-03-02 op static imsg_handlerfn *handlers[] = {
36 5e11c00c 2021-03-02 op [IMSG_ERR] = handle_imsg_err,
37 5e11c00c 2021-03-02 op [IMSG_CHECK_CERT] = handle_imsg_check_cert,
38 5e11c00c 2021-03-02 op [IMSG_GOT_CODE] = handle_imsg_got_code,
39 5e11c00c 2021-03-02 op [IMSG_GOT_META] = handle_imsg_got_meta,
40 5e11c00c 2021-03-02 op [IMSG_BUF] = handle_imsg_buf,
41 5e11c00c 2021-03-02 op [IMSG_EOF] = handle_imsg_eof,
42 5e11c00c 2021-03-02 op };
43 5e11c00c 2021-03-02 op
44 5e11c00c 2021-03-02 op static void __attribute__((__noreturn__))
45 5e11c00c 2021-03-02 op die(void)
46 5e11c00c 2021-03-02 op {
47 5e11c00c 2021-03-02 op abort(); /* TODO */
48 5e11c00c 2021-03-02 op }
49 5e11c00c 2021-03-02 op
50 5e11c00c 2021-03-02 op static struct tab *
51 5e11c00c 2021-03-02 op tab_by_id(uint32_t id)
52 5e11c00c 2021-03-02 op {
53 5e11c00c 2021-03-02 op struct tab *t;
54 5e11c00c 2021-03-02 op
55 5e11c00c 2021-03-02 op TAILQ_FOREACH(t, &tabshead, tabs) {
56 5e11c00c 2021-03-02 op if (t->id == id)
57 5e11c00c 2021-03-02 op return t;
58 5e11c00c 2021-03-02 op }
59 5e11c00c 2021-03-02 op
60 5e11c00c 2021-03-02 op die();
61 5e11c00c 2021-03-02 op }
62 5e11c00c 2021-03-02 op
63 5e11c00c 2021-03-02 op static void
64 5e11c00c 2021-03-02 op handle_imsg_err(struct imsg *imsg, size_t datalen)
65 5e11c00c 2021-03-02 op {
66 3a9b9365 2021-03-09 op struct tab *tab;
67 3a9b9365 2021-03-09 op char *page;
68 3a9b9365 2021-03-09 op
69 3a9b9365 2021-03-09 op tab = tab_by_id(imsg->hdr.peerid);
70 3a9b9365 2021-03-09 op
71 3a9b9365 2021-03-09 op page = imsg->data;
72 3a9b9365 2021-03-09 op page[datalen-1] = '\0';
73 3a9b9365 2021-03-09 op
74 3a9b9365 2021-03-09 op if (asprintf(&page, "# Error loading %s\n\n> %s\n",
75 4d3785b1 2021-03-09 op tab->urlstr, page) == -1)
76 3a9b9365 2021-03-09 op die();
77 3a9b9365 2021-03-09 op load_page_from_str(tab, page);
78 3a9b9365 2021-03-09 op free(page);
79 5e11c00c 2021-03-02 op }
80 5e11c00c 2021-03-02 op
81 5e11c00c 2021-03-02 op static void
82 5e11c00c 2021-03-02 op handle_imsg_check_cert(struct imsg *imsg, size_t datalen)
83 5e11c00c 2021-03-02 op {
84 5e11c00c 2021-03-02 op int tofu_res = 1;
85 5e11c00c 2021-03-02 op
86 5e11c00c 2021-03-02 op imsg_compose(ibuf, IMSG_CERT_STATUS, imsg->hdr.peerid, 0, -1, &tofu_res, sizeof(tofu_res));
87 5e11c00c 2021-03-02 op imsg_flush(ibuf);
88 5e11c00c 2021-03-02 op }
89 5e11c00c 2021-03-02 op
90 5cd2ebb1 2021-03-11 op static inline int
91 5cd2ebb1 2021-03-11 op normalize_code(int n)
92 5cd2ebb1 2021-03-11 op {
93 5cd2ebb1 2021-03-11 op if (n < 20) {
94 5cd2ebb1 2021-03-11 op if (n == 10 || n == 11)
95 5cd2ebb1 2021-03-11 op return n;
96 5cd2ebb1 2021-03-11 op return 10;
97 5cd2ebb1 2021-03-11 op } else if (n < 30) {
98 5cd2ebb1 2021-03-11 op return 20;
99 5cd2ebb1 2021-03-11 op } else if (n < 40) {
100 5cd2ebb1 2021-03-11 op if (n == 30 || n == 31)
101 5cd2ebb1 2021-03-11 op return n;
102 5cd2ebb1 2021-03-11 op return 30;
103 5cd2ebb1 2021-03-11 op } else if (n < 50) {
104 5cd2ebb1 2021-03-11 op if (n <= 44)
105 5cd2ebb1 2021-03-11 op return n;
106 5cd2ebb1 2021-03-11 op return 40;
107 5cd2ebb1 2021-03-11 op } else if (n < 60) {
108 5cd2ebb1 2021-03-11 op if (n <= 53 || n == 59)
109 5cd2ebb1 2021-03-11 op return n;
110 5cd2ebb1 2021-03-11 op return 50;
111 5cd2ebb1 2021-03-11 op } else if (n < 70) {
112 5cd2ebb1 2021-03-11 op if (n <= 62)
113 5cd2ebb1 2021-03-11 op return n;
114 5cd2ebb1 2021-03-11 op return 60;
115 5cd2ebb1 2021-03-11 op } else
116 5cd2ebb1 2021-03-11 op return MALFORMED_RESPONSE;
117 5cd2ebb1 2021-03-11 op }
118 5cd2ebb1 2021-03-11 op
119 5e11c00c 2021-03-02 op static void
120 5e11c00c 2021-03-02 op handle_imsg_got_code(struct imsg *imsg, size_t datalen)
121 5e11c00c 2021-03-02 op {
122 0972d8b2 2021-03-02 op struct tab *tab;
123 5e11c00c 2021-03-02 op
124 0972d8b2 2021-03-02 op tab = tab_by_id(imsg->hdr.peerid);
125 0972d8b2 2021-03-02 op
126 0972d8b2 2021-03-02 op if (sizeof(tab->code) != datalen)
127 5e11c00c 2021-03-02 op die();
128 5e11c00c 2021-03-02 op
129 5cd2ebb1 2021-03-11 op memcpy(&tab->code, imsg->data, sizeof(tab->code));
130 5cd2ebb1 2021-03-11 op tab->code = normalize_code(tab->code);
131 5cd2ebb1 2021-03-11 op if (tab->code != 30 && tab->code != 31)
132 0972d8b2 2021-03-02 op tab->redirect_count = 0;
133 5e11c00c 2021-03-02 op }
134 5e11c00c 2021-03-02 op
135 5e11c00c 2021-03-02 op static void
136 5e11c00c 2021-03-02 op handle_imsg_got_meta(struct imsg *imsg, size_t datalen)
137 5e11c00c 2021-03-02 op {
138 0972d8b2 2021-03-02 op struct tab *tab;
139 0972d8b2 2021-03-02 op
140 0972d8b2 2021-03-02 op tab = tab_by_id(imsg->hdr.peerid);
141 0972d8b2 2021-03-02 op
142 0972d8b2 2021-03-02 op if (sizeof(tab->meta) <= datalen)
143 0972d8b2 2021-03-02 op die();
144 5cd2ebb1 2021-03-11 op
145 0972d8b2 2021-03-02 op memcpy(tab->meta, imsg->data, datalen);
146 0972d8b2 2021-03-02 op
147 5cd2ebb1 2021-03-11 op if (tab->code < 10) { /* internal errors */
148 5cd2ebb1 2021-03-11 op load_page_from_str(tab, err_pages[tab->code]);
149 5cd2ebb1 2021-03-11 op } else if (tab->code < 20) { /* 1x */
150 5cd2ebb1 2021-03-11 op load_page_from_str(tab, err_pages[tab->code]);
151 5cd2ebb1 2021-03-11 op ui_require_input(tab, tab->code == 11);
152 5cd2ebb1 2021-03-11 op } else if (tab->code == 20) {
153 c07ac996 2021-03-12 op if (setup_parser_for(tab)) {
154 c07ac996 2021-03-12 op imsg_compose(ibuf, IMSG_PROCEED, tab->id, 0, -1, NULL, 0);
155 c07ac996 2021-03-12 op imsg_flush(ibuf);
156 c07ac996 2021-03-12 op } else {
157 c07ac996 2021-03-12 op load_page_from_str(tab, err_pages[UNKNOWN_TYPE_OR_CSET]);
158 c07ac996 2021-03-12 op }
159 5cd2ebb1 2021-03-11 op } else if (tab->code < 40) { /* 3x */
160 3a9b9365 2021-03-09 op tab->redirect_count++;
161 0972d8b2 2021-03-02 op
162 3a9b9365 2021-03-09 op /* TODO: make customizable? */
163 3a9b9365 2021-03-09 op if (tab->redirect_count > 5) {
164 3a9b9365 2021-03-09 op load_page_from_str(tab,
165 3a9b9365 2021-03-09 op err_pages[TOO_MUCH_REDIRECTS]);
166 5cd2ebb1 2021-03-11 op } else
167 5cd2ebb1 2021-03-11 op load_url(tab, tab->meta);
168 5cd2ebb1 2021-03-11 op } else { /* 4x, 5x & 6x */
169 5cd2ebb1 2021-03-11 op load_page_from_str(tab, err_pages[tab->code]);
170 3a9b9365 2021-03-09 op }
171 5e11c00c 2021-03-02 op }
172 5e11c00c 2021-03-02 op
173 5e11c00c 2021-03-02 op static void
174 5e11c00c 2021-03-02 op handle_imsg_buf(struct imsg *imsg, size_t datalen)
175 5e11c00c 2021-03-02 op {
176 0972d8b2 2021-03-02 op struct tab *tab;
177 5e11c00c 2021-03-02 op
178 0972d8b2 2021-03-02 op tab = tab_by_id(imsg->hdr.peerid);
179 5e11c00c 2021-03-02 op
180 0972d8b2 2021-03-02 op if (!tab->page.parse(&tab->page, imsg->data, datalen))
181 5e11c00c 2021-03-02 op die();
182 5e11c00c 2021-03-02 op
183 0972d8b2 2021-03-02 op ui_on_tab_refresh(tab);
184 5e11c00c 2021-03-02 op }
185 5e11c00c 2021-03-02 op
186 5e11c00c 2021-03-02 op static void
187 5e11c00c 2021-03-02 op handle_imsg_eof(struct imsg *imsg, size_t datalen)
188 5e11c00c 2021-03-02 op {
189 a5c3e03d 2021-03-02 op struct tab *t;
190 a5c3e03d 2021-03-02 op
191 a5c3e03d 2021-03-02 op t = tab_by_id(imsg->hdr.peerid);
192 a5c3e03d 2021-03-02 op if (!t->page.free(&t->page))
193 a5c3e03d 2021-03-02 op die();
194 a5c3e03d 2021-03-02 op
195 a5c3e03d 2021-03-02 op ui_on_tab_refresh(t);
196 8af5e5ed 2021-03-08 op ui_on_tab_loaded(t);
197 5e11c00c 2021-03-02 op }
198 5e11c00c 2021-03-02 op
199 5e11c00c 2021-03-02 op static void
200 5e11c00c 2021-03-02 op dispatch_imsg(int fd, short ev, void *d)
201 5e11c00c 2021-03-02 op {
202 5e11c00c 2021-03-02 op struct imsg imsg;
203 5e11c00c 2021-03-02 op size_t datalen;
204 5e11c00c 2021-03-02 op ssize_t n;
205 5e11c00c 2021-03-02 op
206 5e11c00c 2021-03-02 op if ((n = imsg_read(ibuf)) == -1) {
207 5e11c00c 2021-03-02 op if (errno == EAGAIN || errno == EWOULDBLOCK)
208 5e11c00c 2021-03-02 op return;
209 5e11c00c 2021-03-02 op die();
210 5e11c00c 2021-03-02 op }
211 5e11c00c 2021-03-02 op
212 5e11c00c 2021-03-02 op if (n == 0) {
213 5e11c00c 2021-03-02 op fprintf(stderr, "other side is dead\n");
214 5e11c00c 2021-03-02 op exit(0);
215 5e11c00c 2021-03-02 op }
216 5e11c00c 2021-03-02 op
217 5e11c00c 2021-03-02 op for (;;) {
218 5e11c00c 2021-03-02 op if ((n = imsg_get(ibuf, &imsg)) == -1)
219 5e11c00c 2021-03-02 op die();
220 5e11c00c 2021-03-02 op if (n == 0)
221 5e11c00c 2021-03-02 op return;
222 5e11c00c 2021-03-02 op datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
223 5e11c00c 2021-03-02 op handlers[imsg.hdr.type](&imsg, datalen);
224 5e11c00c 2021-03-02 op imsg_free(&imsg);
225 5e11c00c 2021-03-02 op }
226 5e11c00c 2021-03-02 op }
227 5e11c00c 2021-03-02 op
228 0972d8b2 2021-03-02 op static void
229 0972d8b2 2021-03-02 op load_page_from_str(struct tab *tab, const char *page)
230 0972d8b2 2021-03-02 op {
231 0972d8b2 2021-03-02 op gemtext_initparser(&tab->page);
232 3a9b9365 2021-03-09 op if (!tab->page.parse(&tab->page, page, strlen(page)))
233 0972d8b2 2021-03-02 op die();
234 0972d8b2 2021-03-02 op if (!tab->page.free(&tab->page))
235 0972d8b2 2021-03-02 op die();
236 0972d8b2 2021-03-02 op ui_on_tab_refresh(tab);
237 8af5e5ed 2021-03-08 op ui_on_tab_loaded(tab);
238 0972d8b2 2021-03-02 op }
239 0972d8b2 2021-03-02 op
240 bcb0b073 2021-03-07 op void
241 4d3785b1 2021-03-09 op load_about_url(struct tab *tab, const char *url)
242 0972d8b2 2021-03-02 op {
243 4d3785b1 2021-03-09 op char *m;
244 8af5e5ed 2021-03-08 op
245 4d3785b1 2021-03-09 op memset(tab->urlstr, 0, sizeof(tab->urlstr));
246 4d3785b1 2021-03-09 op memset(&tab->url, 0, sizeof(tab->url));
247 4d3785b1 2021-03-09 op
248 4d3785b1 2021-03-09 op m = strchr(url, ':');
249 4d3785b1 2021-03-09 op strlcpy(tab->url.scheme, "about", sizeof(tab->url.scheme));
250 4d3785b1 2021-03-09 op strlcpy(tab->url.path, m+1, sizeof(tab->url.path));
251 097ff997 2021-03-09 op
252 097ff997 2021-03-09 op strlcpy(tab->urlstr, url, sizeof(tab->urlstr));
253 4d3785b1 2021-03-09 op
254 4d3785b1 2021-03-09 op if (!strcmp(url, "about:new"))
255 0972d8b2 2021-03-02 op load_page_from_str(tab, about_new);
256 4d3785b1 2021-03-09 op else
257 4d3785b1 2021-03-09 op load_page_from_str(tab, "# not found\n");
258 4d3785b1 2021-03-09 op }
259 0972d8b2 2021-03-02 op
260 4d3785b1 2021-03-09 op void
261 4d3785b1 2021-03-09 op load_gemini_url(struct tab *tab, const char *url)
262 4d3785b1 2021-03-09 op {
263 4d3785b1 2021-03-09 op const char *err;
264 4d3785b1 2021-03-09 op char *p;
265 3a9b9365 2021-03-09 op
266 4d3785b1 2021-03-09 op if (has_prefix(url, "gemini:")) {
267 4d3785b1 2021-03-09 op if (!url_parse(url, &tab->url, &err))
268 3a9b9365 2021-03-09 op goto err;
269 4d3785b1 2021-03-09 op } else {
270 4d3785b1 2021-03-09 op if (!url_resolve_from(&tab->url, url, &err))
271 4d3785b1 2021-03-09 op goto err;
272 3a9b9365 2021-03-09 op }
273 3a9b9365 2021-03-09 op
274 4d3785b1 2021-03-09 op url_unparse(&tab->url, tab->urlstr, sizeof(tab->urlstr));
275 4d3785b1 2021-03-09 op imsg_compose(ibuf, IMSG_GET, tab->id, 0, -1,
276 4d3785b1 2021-03-09 op tab->urlstr, strlen(tab->urlstr)+1);
277 0972d8b2 2021-03-02 op imsg_flush(ibuf);
278 3a9b9365 2021-03-09 op return;
279 3a9b9365 2021-03-09 op
280 3a9b9365 2021-03-09 op err:
281 4d3785b1 2021-03-09 op if (asprintf(&p, "#error loading %s\n>%s\n",
282 4d3785b1 2021-03-09 op url, err) == -1)
283 3a9b9365 2021-03-09 op die();
284 4d3785b1 2021-03-09 op strlcpy(tab->urlstr, url, sizeof(tab->urlstr));
285 4d3785b1 2021-03-09 op load_page_from_str(tab, p);
286 4d3785b1 2021-03-09 op free(p);
287 0972d8b2 2021-03-02 op }
288 0972d8b2 2021-03-02 op
289 4d3785b1 2021-03-09 op void
290 4d3785b1 2021-03-09 op load_url(struct tab *tab, const char *url)
291 4d3785b1 2021-03-09 op {
292 4d3785b1 2021-03-09 op struct proto *p;
293 4d3785b1 2021-03-09 op
294 4d3785b1 2021-03-09 op for (p = protos; p->schema != NULL; ++p) {
295 4d3785b1 2021-03-09 op if (has_prefix(url, p->schema)) {
296 4d3785b1 2021-03-09 op p->loadfn(tab, url);
297 4d3785b1 2021-03-09 op return;
298 4d3785b1 2021-03-09 op }
299 4d3785b1 2021-03-09 op }
300 4d3785b1 2021-03-09 op
301 4d3785b1 2021-03-09 op protos[0].loadfn(tab, url);
302 4d3785b1 2021-03-09 op }
303 4d3785b1 2021-03-09 op
304 9ad4627d 2021-03-10 op void
305 9ad4627d 2021-03-10 op stop_tab(struct tab *tab)
306 9ad4627d 2021-03-10 op {
307 9ad4627d 2021-03-10 op imsg_compose(ibuf, IMSG_STOP, tab->id, 0, -1, NULL, 0);
308 9ad4627d 2021-03-10 op imsg_flush(ibuf);
309 9ad4627d 2021-03-10 op }
310 9ad4627d 2021-03-10 op
311 5e11c00c 2021-03-02 op int
312 5e11c00c 2021-03-02 op main(void)
313 5e11c00c 2021-03-02 op {
314 5e11c00c 2021-03-02 op struct imsgbuf main_ibuf, network_ibuf;
315 5e11c00c 2021-03-02 op int imsg_fds[2];
316 5e11c00c 2021-03-02 op
317 5e11c00c 2021-03-02 op signal(SIGCHLD, SIG_IGN);
318 5e11c00c 2021-03-02 op
319 5e11c00c 2021-03-02 op if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
320 5e11c00c 2021-03-02 op err(1, "socketpair");
321 5e11c00c 2021-03-02 op
322 5e11c00c 2021-03-02 op switch (fork()) {
323 5e11c00c 2021-03-02 op case -1:
324 5e11c00c 2021-03-02 op err(1, "fork");
325 5e11c00c 2021-03-02 op case 0:
326 5e11c00c 2021-03-02 op /* child */
327 5e11c00c 2021-03-02 op setproctitle("client");
328 5e11c00c 2021-03-02 op close(imsg_fds[0]);
329 5e11c00c 2021-03-02 op imsg_init(&network_ibuf, imsg_fds[1]);
330 5e11c00c 2021-03-02 op exit(client_main(&network_ibuf));
331 5e11c00c 2021-03-02 op }
332 5e11c00c 2021-03-02 op
333 5e11c00c 2021-03-02 op close(imsg_fds[1]);
334 5e11c00c 2021-03-02 op imsg_init(&main_ibuf, imsg_fds[0]);
335 5e11c00c 2021-03-02 op ibuf = &main_ibuf;
336 5e11c00c 2021-03-02 op
337 5e11c00c 2021-03-02 op TAILQ_INIT(&tabshead);
338 5e11c00c 2021-03-02 op
339 5e11c00c 2021-03-02 op event_init();
340 5e11c00c 2021-03-02 op
341 5e11c00c 2021-03-02 op event_set(&imsgev, ibuf->fd, EV_READ | EV_PERSIST, dispatch_imsg, ibuf);
342 5e11c00c 2021-03-02 op event_add(&imsgev, NULL);
343 5e11c00c 2021-03-02 op
344 5e11c00c 2021-03-02 op ui_init();
345 5e11c00c 2021-03-02 op
346 5e11c00c 2021-03-02 op event_dispatch();
347 5e11c00c 2021-03-02 op
348 5e11c00c 2021-03-02 op imsg_compose(ibuf, IMSG_QUIT, 0, 0, -1, NULL, 0);
349 5e11c00c 2021-03-02 op imsg_flush(ibuf);
350 5e11c00c 2021-03-02 op
351 5e11c00c 2021-03-02 op ui_end();
352 5e11c00c 2021-03-02 op
353 5e11c00c 2021-03-02 op return 0;
354 5e11c00c 2021-03-02 op }