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 	ui_on_tab_loaded(t);
161 }
162 
163 static void
164 dispatch_imsg(int fd, short ev, void *d)
165 {
166 	struct imsg	imsg;
167 	size_t		datalen;
168 	ssize_t		n;
169 
170 	if ((n = imsg_read(ibuf)) == -1) {
171 		if (errno == EAGAIN || errno == EWOULDBLOCK)
172 			return;
173 		die();
174 	}
175 
176 	if (n == 0) {
177 		fprintf(stderr, "other side is dead\n");
178 		exit(0);
179 	}
180 
181 	for (;;) {
182 		if ((n = imsg_get(ibuf, &imsg)) == -1)
183 			die();
184 		if (n == 0)
185 			return;
186 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
187 		handlers[imsg.hdr.type](&imsg, datalen);
188 		imsg_free(&imsg);
189 	}
190 }
191 
192 static void
193 load_page_from_str(struct tab *tab, const char *page)
194 {
195 	gemtext_initparser(&tab->page);
196 	if (!tab->page.parse(&tab->page, about_new, strlen(about_new)))
197 		die();
198 	if (!tab->page.free(&tab->page))
199 		die();
200 	ui_on_tab_refresh(tab);
201 	ui_on_tab_loaded(tab);
202 }
203 
204 void
205 load_url(struct tab *tab, const char *url)
206 {
207 	strlcpy(tab->url, url, sizeof(tab->url));
208 
209 	if (!strcmp(url, "about:new")) {
210 		load_page_from_str(tab, about_new);
211 		return;
212 	}
213 
214 	imsg_compose(ibuf, IMSG_GET, tab->id, 0, -1, url, strlen(url)+1);
215 	imsg_flush(ibuf);
216 }
217 
218 int
219 main(void)
220 {
221 	struct imsgbuf	main_ibuf, network_ibuf;
222 	int		imsg_fds[2];
223 
224 	signal(SIGCHLD, SIG_IGN);
225 
226 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
227 		err(1, "socketpair");
228 
229 	switch (fork()) {
230 	case -1:
231 		err(1, "fork");
232 	case 0:
233 		/* child */
234 		setproctitle("client");
235 		close(imsg_fds[0]);
236 		imsg_init(&network_ibuf, imsg_fds[1]);
237 		exit(client_main(&network_ibuf));
238 	}
239 
240 	close(imsg_fds[1]);
241 	imsg_init(&main_ibuf, imsg_fds[0]);
242 	ibuf = &main_ibuf;
243 
244 	TAILQ_INIT(&tabshead);
245 
246 	event_init();
247 
248 	event_set(&imsgev, ibuf->fd, EV_READ | EV_PERSIST, dispatch_imsg, ibuf);
249 	event_add(&imsgev, NULL);
250 
251 	ui_init();
252 
253 	event_dispatch();
254 
255 	imsg_compose(ibuf, IMSG_QUIT, 0, 0, -1, NULL, 0);
256 	imsg_flush(ibuf);
257 
258 	ui_end();
259 
260 	return 0;
261 }