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