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