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