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