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 5e11c00c 2021-03-02 op static imsg_handlerfn *handlers[] = {
 28 5e11c00c 2021-03-02 op 	[IMSG_ERR] = handle_imsg_err,
 29 5e11c00c 2021-03-02 op 	[IMSG_CHECK_CERT] = handle_imsg_check_cert,
 30 5e11c00c 2021-03-02 op 	[IMSG_GOT_CODE] = handle_imsg_got_code,
 31 5e11c00c 2021-03-02 op 	[IMSG_GOT_META] = handle_imsg_got_meta,
 32 5e11c00c 2021-03-02 op 	[IMSG_BUF] = handle_imsg_buf,
 33 5e11c00c 2021-03-02 op 	[IMSG_EOF] = handle_imsg_eof,
 34 5e11c00c 2021-03-02 op };
 35 5e11c00c 2021-03-02 op 
 36 5e11c00c 2021-03-02 op static void __attribute__((__noreturn__))
 37 5e11c00c 2021-03-02 op die(void)
 38 5e11c00c 2021-03-02 op {
 39 5e11c00c 2021-03-02 op 	abort(); 		/* TODO */
 40 5e11c00c 2021-03-02 op }
 41 5e11c00c 2021-03-02 op 
 42 5e11c00c 2021-03-02 op static struct tab *
 43 5e11c00c 2021-03-02 op tab_by_id(uint32_t id)
 44 5e11c00c 2021-03-02 op {
 45 5e11c00c 2021-03-02 op 	struct tab *t;
 46 5e11c00c 2021-03-02 op 
 47 5e11c00c 2021-03-02 op 	TAILQ_FOREACH(t, &tabshead, tabs) {
 48 5e11c00c 2021-03-02 op 		if (t->id == id)
 49 5e11c00c 2021-03-02 op 			return t;
 50 5e11c00c 2021-03-02 op 	}
 51 5e11c00c 2021-03-02 op 
 52 5e11c00c 2021-03-02 op 	die();
 53 5e11c00c 2021-03-02 op }
 54 5e11c00c 2021-03-02 op 
 55 5e11c00c 2021-03-02 op static void
 56 5e11c00c 2021-03-02 op handle_imsg_err(struct imsg *imsg, size_t datalen)
 57 5e11c00c 2021-03-02 op {
 58 5e11c00c 2021-03-02 op 	/* write(2, imsg->data, datalen); */
 59 5e11c00c 2021-03-02 op 	/* fprintf(stderr, "\nEOF\n"); */
 60 5e11c00c 2021-03-02 op 	/* event_loopbreak(); */
 61 5e11c00c 2021-03-02 op }
 62 5e11c00c 2021-03-02 op 
 63 5e11c00c 2021-03-02 op static void
 64 5e11c00c 2021-03-02 op handle_imsg_check_cert(struct imsg *imsg, size_t datalen)
 65 5e11c00c 2021-03-02 op {
 66 5e11c00c 2021-03-02 op 	int	tofu_res = 1;
 67 5e11c00c 2021-03-02 op 
 68 5e11c00c 2021-03-02 op 	imsg_compose(ibuf, IMSG_CERT_STATUS, imsg->hdr.peerid, 0, -1, &tofu_res, sizeof(tofu_res));
 69 5e11c00c 2021-03-02 op 	imsg_flush(ibuf);
 70 5e11c00c 2021-03-02 op }
 71 5e11c00c 2021-03-02 op 
 72 5e11c00c 2021-03-02 op static void
 73 5e11c00c 2021-03-02 op handle_imsg_got_code(struct imsg *imsg, size_t datalen)
 74 5e11c00c 2021-03-02 op {
 75 5e11c00c 2021-03-02 op 	int code;
 76 5e11c00c 2021-03-02 op 
 77 5e11c00c 2021-03-02 op 	if (sizeof(code) != datalen)
 78 5e11c00c 2021-03-02 op 		die();
 79 5e11c00c 2021-03-02 op 
 80 5e11c00c 2021-03-02 op 	memcpy(&code, imsg->data, sizeof(code));
 81 5e11c00c 2021-03-02 op 
 82 5e11c00c 2021-03-02 op 	/* fprintf(stderr, "got status code: %d\n", code); */
 83 5e11c00c 2021-03-02 op }
 84 5e11c00c 2021-03-02 op 
 85 5e11c00c 2021-03-02 op static void
 86 5e11c00c 2021-03-02 op handle_imsg_got_meta(struct imsg *imsg, size_t datalen)
 87 5e11c00c 2021-03-02 op {
 88 5e11c00c 2021-03-02 op 	/* fprintf(stderr, "got meta: "); */
 89 5e11c00c 2021-03-02 op 	/* fflush(stderr); */
 90 5e11c00c 2021-03-02 op 	/* write(2, imsg->data, datalen); */
 91 5e11c00c 2021-03-02 op 	/* fprintf(stderr, "\n"); */
 92 5e11c00c 2021-03-02 op }
 93 5e11c00c 2021-03-02 op 
 94 5e11c00c 2021-03-02 op static void
 95 5e11c00c 2021-03-02 op handle_imsg_buf(struct imsg *imsg, size_t datalen)
 96 5e11c00c 2021-03-02 op {
 97 5e11c00c 2021-03-02 op         struct tab	*t;
 98 5e11c00c 2021-03-02 op 	struct line	*l;
 99 5e11c00c 2021-03-02 op 
100 5e11c00c 2021-03-02 op 	t = tab_by_id(imsg->hdr.peerid);
101 5e11c00c 2021-03-02 op 
102 5e11c00c 2021-03-02 op 	if (!t->page.parse(&t->page, imsg->data, datalen))
103 5e11c00c 2021-03-02 op 		die();
104 5e11c00c 2021-03-02 op 
105 5e11c00c 2021-03-02 op 	ui_on_tab_refresh(t);
106 5e11c00c 2021-03-02 op }
107 5e11c00c 2021-03-02 op 
108 5e11c00c 2021-03-02 op static void
109 5e11c00c 2021-03-02 op handle_imsg_eof(struct imsg *imsg, size_t datalen)
110 5e11c00c 2021-03-02 op {
111 a5c3e03d 2021-03-02 op 	struct tab	*t;
112 a5c3e03d 2021-03-02 op 
113 a5c3e03d 2021-03-02 op 	t = tab_by_id(imsg->hdr.peerid);
114 a5c3e03d 2021-03-02 op 	if (!t->page.free(&t->page))
115 a5c3e03d 2021-03-02 op 		die();
116 a5c3e03d 2021-03-02 op 
117 a5c3e03d 2021-03-02 op 	ui_on_tab_refresh(t);
118 5e11c00c 2021-03-02 op }
119 5e11c00c 2021-03-02 op 
120 5e11c00c 2021-03-02 op static void
121 5e11c00c 2021-03-02 op dispatch_imsg(int fd, short ev, void *d)
122 5e11c00c 2021-03-02 op {
123 5e11c00c 2021-03-02 op 	struct imsg	imsg;
124 5e11c00c 2021-03-02 op 	size_t		datalen;
125 5e11c00c 2021-03-02 op 	ssize_t		n;
126 5e11c00c 2021-03-02 op 
127 5e11c00c 2021-03-02 op 	if ((n = imsg_read(ibuf)) == -1) {
128 5e11c00c 2021-03-02 op 		if (errno == EAGAIN || errno == EWOULDBLOCK)
129 5e11c00c 2021-03-02 op 			return;
130 5e11c00c 2021-03-02 op 		die();
131 5e11c00c 2021-03-02 op 	}
132 5e11c00c 2021-03-02 op 
133 5e11c00c 2021-03-02 op 	if (n == 0) {
134 5e11c00c 2021-03-02 op 		fprintf(stderr, "other side is dead\n");
135 5e11c00c 2021-03-02 op 		exit(0);
136 5e11c00c 2021-03-02 op 	}
137 5e11c00c 2021-03-02 op 
138 5e11c00c 2021-03-02 op 	for (;;) {
139 5e11c00c 2021-03-02 op 		if ((n = imsg_get(ibuf, &imsg)) == -1)
140 5e11c00c 2021-03-02 op 			die();
141 5e11c00c 2021-03-02 op 		if (n == 0)
142 5e11c00c 2021-03-02 op 			return;
143 5e11c00c 2021-03-02 op 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
144 5e11c00c 2021-03-02 op 		handlers[imsg.hdr.type](&imsg, datalen);
145 5e11c00c 2021-03-02 op 		imsg_free(&imsg);
146 5e11c00c 2021-03-02 op 	}
147 5e11c00c 2021-03-02 op }
148 5e11c00c 2021-03-02 op 
149 5e11c00c 2021-03-02 op void
150 5e11c00c 2021-03-02 op new_tab(void)
151 5e11c00c 2021-03-02 op {
152 5e11c00c 2021-03-02 op 	struct tab	*tab;
153 5e11c00c 2021-03-02 op 	const char	*url = "about:new";
154 5e11c00c 2021-03-02 op 	/* const char	*url = "gemini://localhost/cgi/slow-out"; */
155 5e11c00c 2021-03-02 op 
156 5e11c00c 2021-03-02 op 	if ((tab = calloc(1, sizeof(*tab))) == NULL)
157 5e11c00c 2021-03-02 op 		die();
158 5e11c00c 2021-03-02 op 
159 5e11c00c 2021-03-02 op 	TAILQ_INSERT_HEAD(&tabshead, tab, tabs);
160 5e11c00c 2021-03-02 op 
161 5e11c00c 2021-03-02 op 	tab->id = tab_counter++;
162 5e11c00c 2021-03-02 op 	TAILQ_INIT(&tab->page.head);
163 5e11c00c 2021-03-02 op 	gemtext_initparser(&tab->page);
164 5e11c00c 2021-03-02 op 
165 5e11c00c 2021-03-02 op 	imsg_compose(ibuf, IMSG_GET, tab->id, 0, -1, url, strlen(url)+1);
166 5e11c00c 2021-03-02 op 	imsg_flush(ibuf);
167 5e11c00c 2021-03-02 op 
168 5e11c00c 2021-03-02 op 	ui_on_new_tab(tab);
169 5e11c00c 2021-03-02 op }
170 5e11c00c 2021-03-02 op 
171 5e11c00c 2021-03-02 op int
172 5e11c00c 2021-03-02 op main(void)
173 5e11c00c 2021-03-02 op {
174 5e11c00c 2021-03-02 op 	struct imsgbuf	main_ibuf, network_ibuf;
175 5e11c00c 2021-03-02 op 	int		imsg_fds[2];
176 5e11c00c 2021-03-02 op 
177 5e11c00c 2021-03-02 op 	signal(SIGCHLD, SIG_IGN);
178 5e11c00c 2021-03-02 op 	signal(SIGINT, SIG_IGN);
179 5e11c00c 2021-03-02 op 
180 5e11c00c 2021-03-02 op 	if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, imsg_fds) == -1)
181 5e11c00c 2021-03-02 op 		err(1, "socketpair");
182 5e11c00c 2021-03-02 op 
183 5e11c00c 2021-03-02 op 	switch (fork()) {
184 5e11c00c 2021-03-02 op 	case -1:
185 5e11c00c 2021-03-02 op 		err(1, "fork");
186 5e11c00c 2021-03-02 op 	case 0:
187 5e11c00c 2021-03-02 op 		/* child */
188 5e11c00c 2021-03-02 op 		setproctitle("client");
189 5e11c00c 2021-03-02 op 		close(imsg_fds[0]);
190 5e11c00c 2021-03-02 op 		imsg_init(&network_ibuf, imsg_fds[1]);
191 5e11c00c 2021-03-02 op 		exit(client_main(&network_ibuf));
192 5e11c00c 2021-03-02 op 	}
193 5e11c00c 2021-03-02 op 
194 5e11c00c 2021-03-02 op 	close(imsg_fds[1]);
195 5e11c00c 2021-03-02 op 	imsg_init(&main_ibuf, imsg_fds[0]);
196 5e11c00c 2021-03-02 op 	ibuf = &main_ibuf;
197 5e11c00c 2021-03-02 op 
198 5e11c00c 2021-03-02 op 	TAILQ_INIT(&tabshead);
199 5e11c00c 2021-03-02 op 
200 5e11c00c 2021-03-02 op 	event_init();
201 5e11c00c 2021-03-02 op 
202 5e11c00c 2021-03-02 op 	event_set(&imsgev, ibuf->fd, EV_READ | EV_PERSIST, dispatch_imsg, ibuf);
203 5e11c00c 2021-03-02 op 	event_add(&imsgev, NULL);
204 5e11c00c 2021-03-02 op 
205 5e11c00c 2021-03-02 op 	ui_init();
206 5e11c00c 2021-03-02 op 
207 5e11c00c 2021-03-02 op 	new_tab();
208 5e11c00c 2021-03-02 op 
209 5e11c00c 2021-03-02 op 	event_dispatch();
210 5e11c00c 2021-03-02 op 
211 5e11c00c 2021-03-02 op 	imsg_compose(ibuf, IMSG_QUIT, 0, 0, -1, NULL, 0);
212 5e11c00c 2021-03-02 op 	imsg_flush(ibuf);
213 5e11c00c 2021-03-02 op 
214 5e11c00c 2021-03-02 op 	ui_end();
215 5e11c00c 2021-03-02 op 
216 5e11c00c 2021-03-02 op 	return 0;
217 5e11c00c 2021-03-02 op }