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