1 35e1f40a 2021-03-14 op /*
  2 35e1f40a 2021-03-14 op  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
  3 35e1f40a 2021-03-14 op  *
  4 35e1f40a 2021-03-14 op  * Permission to use, copy, modify, and distribute this software for any
  5 35e1f40a 2021-03-14 op  * purpose with or without fee is hereby granted, provided that the above
  6 35e1f40a 2021-03-14 op  * copyright notice and this permission notice appear in all copies.
  7 35e1f40a 2021-03-14 op  *
  8 35e1f40a 2021-03-14 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 35e1f40a 2021-03-14 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 35e1f40a 2021-03-14 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 35e1f40a 2021-03-14 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 35e1f40a 2021-03-14 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 35e1f40a 2021-03-14 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 35e1f40a 2021-03-14 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 35e1f40a 2021-03-14 op  */
 16 35e1f40a 2021-03-14 op 
 17 740f578b 2021-03-15 op /*
 18 740f578b 2021-03-15 op  * Handles the data in ~/.telescope
 19 3a227e9a 2021-03-18 op  *
 20 3a227e9a 2021-03-18 op  * TODO: add some form of locking on the files
 21 740f578b 2021-03-15 op  */
 22 786e6deb 2021-07-21 op 
 23 786e6deb 2021-07-21 op #include "compat.h"
 24 35e1f40a 2021-03-14 op 
 25 6cd6a9e1 2021-03-20 op #include <sys/stat.h>
 26 65124267 2021-08-13 op #include <sys/types.h>
 27 6cd6a9e1 2021-03-20 op 
 28 65124267 2021-08-13 op #include <dirent.h>
 29 35e1f40a 2021-03-14 op #include <errno.h>
 30 de2a69bb 2021-05-17 op #include <fcntl.h>
 31 35e1f40a 2021-03-14 op #include <limits.h>
 32 35e1f40a 2021-03-14 op #include <stdio.h>
 33 35e1f40a 2021-03-14 op #include <stdlib.h>
 34 35e1f40a 2021-03-14 op #include <string.h>
 35 35e1f40a 2021-03-14 op #include <unistd.h>
 36 4913b479 2021-07-12 op 
 37 4913b479 2021-07-12 op #include "telescope.h"
 38 5a824be4 2021-07-13 op #include "pages.h"
 39 35e1f40a 2021-03-14 op 
 40 35e1f40a 2021-03-14 op static void		 die(void) __attribute__((__noreturn__));
 41 fb4dc49f 2021-08-13 op static void		 send_file(uint32_t, FILE *);
 42 35e1f40a 2021-03-14 op static void		 handle_get(struct imsg*, size_t);
 43 fb4dc49f 2021-08-13 op static void		 handle_get_file(struct imsg*, size_t);
 44 35e1f40a 2021-03-14 op static void		 handle_quit(struct imsg*, size_t);
 45 740f578b 2021-03-15 op static void		 handle_bookmark_page(struct imsg*, size_t);
 46 3a227e9a 2021-03-18 op static void		 handle_save_cert(struct imsg*, size_t);
 47 288fd238 2021-04-25 op static void		 handle_update_cert(struct imsg*, size_t);
 48 de2a69bb 2021-05-17 op static void		 handle_file_open(struct imsg*, size_t);
 49 c7107cec 2021-04-01 op static void		 handle_session_start(struct imsg*, size_t);
 50 c7107cec 2021-04-01 op static void		 handle_session_tab(struct imsg*, size_t);
 51 2f524b17 2021-07-17 op static void		 handle_session_tab_title(struct imsg*, size_t);
 52 c7107cec 2021-04-01 op static void		 handle_session_end(struct imsg*, size_t);
 53 1304bbdd 2021-03-15 op static void		 handle_dispatch_imsg(int, short, void*);
 54 bc10f6a5 2021-07-12 op static int		 fs_send_ui(int, uint32_t, int, const void *, uint16_t);
 55 35e1f40a 2021-03-14 op 
 56 bc10f6a5 2021-07-12 op static struct imsgev		*iev_ui;
 57 c7107cec 2021-04-01 op static FILE			*session;
 58 c7107cec 2021-04-01 op 
 59 56e7efb4 2021-07-21 op static char	base_path[PATH_MAX];
 60 d0fd368a 2021-07-15 op static char	lockfile_path[PATH_MAX];
 61 740f578b 2021-03-15 op static char	bookmark_file[PATH_MAX];
 62 288fd238 2021-04-25 op static char	known_hosts_file[PATH_MAX], known_hosts_tmp[PATH_MAX];
 63 ea080950 2021-07-20 op static char	crashed_file[PATH_MAX];
 64 87e3e801 2021-07-17 op 
 65 87e3e801 2021-07-17 op char	session_file[PATH_MAX];
 66 740f578b 2021-03-15 op 
 67 35e1f40a 2021-03-14 op static imsg_handlerfn *handlers[] = {
 68 35e1f40a 2021-03-14 op 	[IMSG_GET] = handle_get,
 69 fb4dc49f 2021-08-13 op 	[IMSG_GET_FILE] = handle_get_file,
 70 35e1f40a 2021-03-14 op 	[IMSG_QUIT] = handle_quit,
 71 740f578b 2021-03-15 op 	[IMSG_BOOKMARK_PAGE] = handle_bookmark_page,
 72 3a227e9a 2021-03-18 op 	[IMSG_SAVE_CERT] = handle_save_cert,
 73 288fd238 2021-04-25 op 	[IMSG_UPDATE_CERT] = handle_update_cert,
 74 de2a69bb 2021-05-17 op 	[IMSG_FILE_OPEN] = handle_file_open,
 75 c7107cec 2021-04-01 op 	[IMSG_SESSION_START] = handle_session_start,
 76 c7107cec 2021-04-01 op 	[IMSG_SESSION_TAB] = handle_session_tab,
 77 2f524b17 2021-07-17 op 	[IMSG_SESSION_TAB_TITLE] = handle_session_tab_title,
 78 c7107cec 2021-04-01 op 	[IMSG_SESSION_END] = handle_session_end,
 79 35e1f40a 2021-03-14 op };
 80 35e1f40a 2021-03-14 op 
 81 35e1f40a 2021-03-14 op static void __attribute__((__noreturn__))
 82 35e1f40a 2021-03-14 op die(void)
 83 35e1f40a 2021-03-14 op {
 84 35e1f40a 2021-03-14 op 	abort(); 		/* TODO */
 85 35e1f40a 2021-03-14 op }
 86 35e1f40a 2021-03-14 op 
 87 35e1f40a 2021-03-14 op static void
 88 fb4dc49f 2021-08-13 op send_file(uint32_t peerid, FILE *f)
 89 fb4dc49f 2021-08-13 op {
 90 fb4dc49f 2021-08-13 op 	ssize_t	 r;
 91 fb4dc49f 2021-08-13 op 	char	 buf[BUFSIZ];
 92 fb4dc49f 2021-08-13 op 
 93 fb4dc49f 2021-08-13 op 	for (;;) {
 94 fb4dc49f 2021-08-13 op 		r = fread(buf, 1, sizeof(buf), f);
 95 fb4dc49f 2021-08-13 op 		if (r != 0)
 96 fb4dc49f 2021-08-13 op 			fs_send_ui(IMSG_BUF, peerid, -1, buf, r);
 97 fb4dc49f 2021-08-13 op 		if (r != sizeof(buf))
 98 fb4dc49f 2021-08-13 op 			break;
 99 fb4dc49f 2021-08-13 op 	}
100 fb4dc49f 2021-08-13 op 	fs_send_ui(IMSG_EOF, peerid, -1, NULL, 0);
101 fb4dc49f 2021-08-13 op 	fclose(f);
102 fb4dc49f 2021-08-13 op }
103 fb4dc49f 2021-08-13 op 
104 fb4dc49f 2021-08-13 op static void
105 35e1f40a 2021-03-14 op handle_get(struct imsg *imsg, size_t datalen)
106 35e1f40a 2021-03-14 op {
107 56e7efb4 2021-07-21 op 	const char	*bpath = "bookmarks.gmi";
108 fb4dc49f 2021-08-13 op 	char		 path[PATH_MAX];
109 56e7efb4 2021-07-21 op 	FILE		*f;
110 a2728733 2021-07-18 op 	const char	*data, *p;
111 a2728733 2021-07-18 op 	size_t		 i;
112 a2728733 2021-07-18 op 	struct page {
113 a2728733 2021-07-18 op 		const char	*name;
114 56e7efb4 2021-07-21 op 		const char	*path;
115 a2728733 2021-07-18 op 		const uint8_t	*data;
116 a2728733 2021-07-18 op 		size_t		 len;
117 a2728733 2021-07-18 op 	} pages[] = {
118 56e7efb4 2021-07-21 op 		{"about",	NULL,	about_about,	about_about_len},
119 56e7efb4 2021-07-21 op 		{"blank",	NULL,	about_blank,	about_blank_len},
120 56e7efb4 2021-07-21 op 		{"bookmarks",	bpath,	bookmarks,	bookmarks_len},
121 56e7efb4 2021-07-21 op 		{"crash",	NULL,	about_crash,	about_crash_len},
122 56e7efb4 2021-07-21 op 		{"help",	NULL,	about_help,	about_help_len},
123 56e7efb4 2021-07-21 op 		{"license",	NULL,	about_license,	about_license_len},
124 56e7efb4 2021-07-21 op 		{"new",		NULL,	about_new,	about_new_len},
125 56e7efb4 2021-07-21 op 	}, *page = NULL;
126 35e1f40a 2021-03-14 op 
127 35e1f40a 2021-03-14 op 	data = imsg->data;
128 56e7efb4 2021-07-21 op 	if (data[datalen-1] != '\0') /* make sure it's NUL-terminated */
129 35e1f40a 2021-03-14 op 		die();
130 56e7efb4 2021-07-21 op 	if ((data = strchr(data, ':')) == NULL)
131 56e7efb4 2021-07-21 op 		goto notfound;
132 56e7efb4 2021-07-21 op 	data++;
133 35e1f40a 2021-03-14 op 
134 56e7efb4 2021-07-21 op 	for (i = 0; i < sizeof(pages)/sizeof(pages[0]); ++i)
135 56e7efb4 2021-07-21 op 		if (!strcmp(data, pages[i].name)) {
136 56e7efb4 2021-07-21 op 			page = &pages[i];
137 56e7efb4 2021-07-21 op 			break;
138 56e7efb4 2021-07-21 op 		}
139 56e7efb4 2021-07-21 op 
140 56e7efb4 2021-07-21 op 	if (page == NULL)
141 56e7efb4 2021-07-21 op 		goto notfound;
142 56e7efb4 2021-07-21 op 
143 56e7efb4 2021-07-21 op 	strlcpy(path, base_path, sizeof(path));
144 56e7efb4 2021-07-21 op 	strlcat(path, "/", sizeof(path));
145 56e7efb4 2021-07-21 op 	if (page->path != NULL)
146 56e7efb4 2021-07-21 op 		strlcat(path, page->path, sizeof(path));
147 56e7efb4 2021-07-21 op 	else {
148 56e7efb4 2021-07-21 op 		strlcat(path, "pages/about_", sizeof(path));
149 56e7efb4 2021-07-21 op 		strlcat(path, page->name, sizeof(path));
150 56e7efb4 2021-07-21 op 		strlcat(path, ".gmi", sizeof(path));
151 56e7efb4 2021-07-21 op 	}
152 56e7efb4 2021-07-21 op 
153 56e7efb4 2021-07-21 op 	if ((f = fopen(path, "r")) == NULL) {
154 56e7efb4 2021-07-21 op 		fs_send_ui(IMSG_BUF, imsg->hdr.peerid, -1,
155 56e7efb4 2021-07-21 op 		    page->data, page->len);
156 56e7efb4 2021-07-21 op 		fs_send_ui(IMSG_EOF, imsg->hdr.peerid, -1,
157 56e7efb4 2021-07-21 op 		    NULL, 0);
158 a2728733 2021-07-18 op 		return;
159 35e1f40a 2021-03-14 op 	}
160 a2728733 2021-07-18 op 
161 fb4dc49f 2021-08-13 op 	send_file(imsg->hdr.peerid, f);
162 56e7efb4 2021-07-21 op 	return;
163 56e7efb4 2021-07-21 op 
164 56e7efb4 2021-07-21 op notfound:
165 a2728733 2021-07-18 op 	p = "# not found!\n";
166 a2728733 2021-07-18 op 	fs_send_ui(IMSG_BUF, imsg->hdr.peerid, -1, p, strlen(p));
167 a2728733 2021-07-18 op 	fs_send_ui(IMSG_EOF, imsg->hdr.peerid, -1, NULL, 0);
168 fb4dc49f 2021-08-13 op }
169 fb4dc49f 2021-08-13 op 
170 fb4dc49f 2021-08-13 op static inline void
171 fb4dc49f 2021-08-13 op send_hdr(uint32_t peerid, int code, const char *meta)
172 fb4dc49f 2021-08-13 op {
173 fb4dc49f 2021-08-13 op 	fs_send_ui(IMSG_GOT_CODE, peerid, -1, &code, sizeof(code));
174 fb4dc49f 2021-08-13 op 	fs_send_ui(IMSG_GOT_META, peerid, -1, meta, strlen(meta)+1);
175 fb4dc49f 2021-08-13 op }
176 fb4dc49f 2021-08-13 op 
177 65124267 2021-08-13 op static inline const char *
178 65124267 2021-08-13 op file_type(const char *path)
179 fb4dc49f 2021-08-13 op {
180 fb4dc49f 2021-08-13 op 	struct mapping {
181 fb4dc49f 2021-08-13 op 		const char	*ext;
182 fb4dc49f 2021-08-13 op 		const char	*mime;
183 fb4dc49f 2021-08-13 op 	} ms[] = {
184 fb4dc49f 2021-08-13 op 		{"diff",	"text/x-patch"},
185 fb4dc49f 2021-08-13 op 		{"gemini",	"text/gemini"},
186 fb4dc49f 2021-08-13 op 		{"gmi",		"text/gemini"},
187 fb4dc49f 2021-08-13 op 		{"markdown",	"text/plain"},
188 fb4dc49f 2021-08-13 op 		{"md",		"text/plain"},
189 fb4dc49f 2021-08-13 op 		{"patch",	"text/x-patch"},
190 fb4dc49f 2021-08-13 op 		{"txt",		"text/plain"},
191 fb4dc49f 2021-08-13 op 		{NULL, NULL},
192 65124267 2021-08-13 op 	}, *m;
193 65124267 2021-08-13 op 	char *dot;
194 65124267 2021-08-13 op 
195 65124267 2021-08-13 op 	if ((dot = strrchr(path, '.')) == NULL)
196 65124267 2021-08-13 op 		return NULL;
197 65124267 2021-08-13 op 
198 65124267 2021-08-13 op 	dot++;
199 65124267 2021-08-13 op 
200 65124267 2021-08-13 op 	for (m = ms; m->ext != NULL; ++m)
201 65124267 2021-08-13 op 		if (!strcmp(m->ext, dot))
202 65124267 2021-08-13 op 			return m->mime;
203 65124267 2021-08-13 op 
204 65124267 2021-08-13 op 	return NULL;
205 65124267 2021-08-13 op }
206 65124267 2021-08-13 op 
207 65124267 2021-08-13 op static inline void
208 65124267 2021-08-13 op send_dir(uint32_t peerid, const char *path)
209 65124267 2021-08-13 op {
210 65124267 2021-08-13 op 	struct dirent	**names;
211 65124267 2021-08-13 op 	struct evbuffer	 *ev;
212 65124267 2021-08-13 op 	int		  i, len;
213 65124267 2021-08-13 op 
214 65124267 2021-08-13 op 	if ((ev = evbuffer_new()) == NULL ||
215 65124267 2021-08-13 op 	    (len = scandir(path, &names, NULL, alphasort)) == -1) {
216 65124267 2021-08-13 op 		evbuffer_free(ev);
217 65124267 2021-08-13 op 		send_hdr(peerid, 40, "failure reading the directory");
218 65124267 2021-08-13 op 		return;
219 65124267 2021-08-13 op 	}
220 65124267 2021-08-13 op 
221 65124267 2021-08-13 op 	evbuffer_add_printf(ev, "# Index of %s\n\n", path);
222 65124267 2021-08-13 op 	for (i = 0; i < len; ++i) {
223 65124267 2021-08-13 op 		evbuffer_add_printf(ev, "=> %s", names[i]->d_name);
224 65124267 2021-08-13 op 		if (names[i]->d_type == DT_DIR)
225 65124267 2021-08-13 op 			evbuffer_add(ev, "/", 1);
226 65124267 2021-08-13 op 		evbuffer_add(ev, "\n", 1);
227 65124267 2021-08-13 op 	}
228 65124267 2021-08-13 op 
229 65124267 2021-08-13 op 	send_hdr(peerid, 20, "text/gemini");
230 65124267 2021-08-13 op 	fs_send_ui(IMSG_BUF, peerid, -1,
231 65124267 2021-08-13 op 	    EVBUFFER_DATA(ev), EVBUFFER_LENGTH(ev));
232 65124267 2021-08-13 op 	fs_send_ui(IMSG_EOF, peerid, -1, NULL, 0);
233 65124267 2021-08-13 op 
234 65124267 2021-08-13 op 	evbuffer_free(ev);
235 65124267 2021-08-13 op 	free(names);
236 65124267 2021-08-13 op }
237 65124267 2021-08-13 op 
238 65124267 2021-08-13 op static void
239 65124267 2021-08-13 op handle_get_file(struct imsg *imsg, size_t datalen)
240 65124267 2021-08-13 op {
241 65124267 2021-08-13 op 	struct stat	 sb;
242 fb4dc49f 2021-08-13 op 	FILE		*f;
243 65124267 2021-08-13 op 	char		*data;
244 fb4dc49f 2021-08-13 op 	const char	*meta = NULL;
245 fb4dc49f 2021-08-13 op 
246 fb4dc49f 2021-08-13 op 	data = imsg->data;
247 fb4dc49f 2021-08-13 op 	data[datalen-1] = '\0';
248 fb4dc49f 2021-08-13 op 
249 65124267 2021-08-13 op 	if ((f = fopen(data, "r")) == NULL) {
250 65124267 2021-08-13 op 		send_hdr(imsg->hdr.peerid, 51, "can't open the file");
251 fb4dc49f 2021-08-13 op 		return;
252 fb4dc49f 2021-08-13 op 	}
253 fb4dc49f 2021-08-13 op 
254 65124267 2021-08-13 op 	if (fstat(fileno(f), &sb) == -1) {
255 65124267 2021-08-13 op 		send_hdr(imsg->hdr.peerid, 40, "fstat failed");
256 65124267 2021-08-13 op 		return;
257 fb4dc49f 2021-08-13 op 	}
258 fb4dc49f 2021-08-13 op 
259 65124267 2021-08-13 op 	if (S_ISDIR(sb.st_mode)) {
260 65124267 2021-08-13 op 		fclose(f);
261 65124267 2021-08-13 op 		send_dir(imsg->hdr.peerid, data);
262 fb4dc49f 2021-08-13 op 		return;
263 fb4dc49f 2021-08-13 op 	}
264 fb4dc49f 2021-08-13 op 
265 65124267 2021-08-13 op 	if ((meta = file_type(data)) == NULL) {
266 65124267 2021-08-13 op 		fclose(f);
267 65124267 2021-08-13 op 		send_hdr(imsg->hdr.peerid, 51,
268 65124267 2021-08-13 op 		    "don't know how to visualize this file");
269 fb4dc49f 2021-08-13 op 		return;
270 fb4dc49f 2021-08-13 op 	}
271 fb4dc49f 2021-08-13 op 
272 fb4dc49f 2021-08-13 op 	send_hdr(imsg->hdr.peerid, 20, meta);
273 fb4dc49f 2021-08-13 op 	send_file(imsg->hdr.peerid, f);
274 35e1f40a 2021-03-14 op }
275 35e1f40a 2021-03-14 op 
276 35e1f40a 2021-03-14 op static void
277 35e1f40a 2021-03-14 op handle_quit(struct imsg *imsg, size_t datalen)
278 35e1f40a 2021-03-14 op {
279 35e1f40a 2021-03-14 op 	event_loopbreak();
280 35e1f40a 2021-03-14 op }
281 35e1f40a 2021-03-14 op 
282 35e1f40a 2021-03-14 op static void
283 740f578b 2021-03-15 op handle_bookmark_page(struct imsg *imsg, size_t datalen)
284 740f578b 2021-03-15 op {
285 740f578b 2021-03-15 op 	char	*data;
286 740f578b 2021-03-15 op 	int	 res;
287 740f578b 2021-03-15 op 	FILE	*f;
288 740f578b 2021-03-15 op 
289 740f578b 2021-03-15 op 	data = imsg->data;
290 740f578b 2021-03-15 op 	if (data[datalen-1] != '\0')
291 740f578b 2021-03-15 op 		die();
292 740f578b 2021-03-15 op 
293 740f578b 2021-03-15 op 	if ((f = fopen(bookmark_file, "a")) == NULL) {
294 740f578b 2021-03-15 op 		res = errno;
295 740f578b 2021-03-15 op 		goto end;
296 740f578b 2021-03-15 op 	}
297 740f578b 2021-03-15 op 	fprintf(f, "=> %s\n", data);
298 740f578b 2021-03-15 op 	fclose(f);
299 740f578b 2021-03-15 op 
300 740f578b 2021-03-15 op 	res = 0;
301 740f578b 2021-03-15 op end:
302 bc10f6a5 2021-07-12 op 	fs_send_ui(IMSG_BOOKMARK_OK, 0, -1, &res, sizeof(res));
303 740f578b 2021-03-15 op }
304 740f578b 2021-03-15 op 
305 740f578b 2021-03-15 op static void
306 3a227e9a 2021-03-18 op handle_save_cert(struct imsg *imsg, size_t datalen)
307 3a227e9a 2021-03-18 op {
308 3a227e9a 2021-03-18 op 	struct tofu_entry	 e;
309 3a227e9a 2021-03-18 op 	FILE			*f;
310 3a227e9a 2021-03-18 op 	int			 res;
311 3a227e9a 2021-03-18 op 
312 3a227e9a 2021-03-18 op 	/* TODO: traverse the file to avoid duplications? */
313 3a227e9a 2021-03-18 op 
314 3a227e9a 2021-03-18 op 	if (datalen != sizeof(e))
315 3a227e9a 2021-03-18 op 		die();
316 3a227e9a 2021-03-18 op 	memcpy(&e, imsg->data, datalen);
317 3a227e9a 2021-03-18 op 
318 3a227e9a 2021-03-18 op 	if ((f = fopen(known_hosts_file, "a")) == NULL) {
319 3a227e9a 2021-03-18 op 		res = errno;
320 3a227e9a 2021-03-18 op 		goto end;
321 3a227e9a 2021-03-18 op 	}
322 3a227e9a 2021-03-18 op 	fprintf(f, "%s %s %d\n", e.domain, e.hash, e.verified);
323 3a227e9a 2021-03-18 op 	fclose(f);
324 3a227e9a 2021-03-18 op 
325 3a227e9a 2021-03-18 op 	res = 0;
326 3a227e9a 2021-03-18 op end:
327 bc10f6a5 2021-07-12 op 	fs_send_ui(IMSG_SAVE_CERT_OK, imsg->hdr.peerid, -1,
328 288fd238 2021-04-25 op 	    &res, sizeof(res));
329 288fd238 2021-04-25 op }
330 288fd238 2021-04-25 op 
331 288fd238 2021-04-25 op static void
332 288fd238 2021-04-25 op handle_update_cert(struct imsg *imsg, size_t datalen)
333 288fd238 2021-04-25 op {
334 288fd238 2021-04-25 op 	FILE	*tmp, *f;
335 288fd238 2021-04-25 op 	struct	 tofu_entry entry;
336 288fd238 2021-04-25 op 	char	 sfn[PATH_MAX], *line = NULL, *t;
337 288fd238 2021-04-25 op 	size_t	 l, linesize = 0;
338 288fd238 2021-04-25 op 	ssize_t	 linelen;
339 288fd238 2021-04-25 op 	int	 fd, e, res = 0;
340 288fd238 2021-04-25 op 
341 288fd238 2021-04-25 op 	if (datalen != sizeof(entry))
342 288fd238 2021-04-25 op 		die();
343 288fd238 2021-04-25 op 	memcpy(&entry, imsg->data, datalen);
344 288fd238 2021-04-25 op 
345 288fd238 2021-04-25 op 	strlcpy(sfn, known_hosts_tmp, sizeof(sfn));
346 288fd238 2021-04-25 op 	if ((fd = mkstemp(sfn)) == -1 ||
347 288fd238 2021-04-25 op 	    (tmp = fdopen(fd, "w")) == NULL) {
348 288fd238 2021-04-25 op 		if (fd != -1) {
349 288fd238 2021-04-25 op 			unlink(sfn);
350 288fd238 2021-04-25 op 			close(fd);
351 288fd238 2021-04-25 op 		}
352 288fd238 2021-04-25 op 		res = 0;
353 288fd238 2021-04-25 op 		goto end;
354 288fd238 2021-04-25 op 	}
355 288fd238 2021-04-25 op 
356 288fd238 2021-04-25 op 	if ((f = fopen(known_hosts_file, "r")) == NULL) {
357 288fd238 2021-04-25 op 		unlink(sfn);
358 288fd238 2021-04-25 op 		fclose(tmp);
359 288fd238 2021-04-25 op                 res = 0;
360 288fd238 2021-04-25 op 		goto end;
361 288fd238 2021-04-25 op 	}
362 288fd238 2021-04-25 op 
363 288fd238 2021-04-25 op 	l = strlen(entry.domain);
364 288fd238 2021-04-25 op 	while ((linelen = getline(&line, &linesize, f)) != -1) {
365 288fd238 2021-04-25 op 		if ((t = strstr(line, entry.domain)) != NULL &&
366 288fd238 2021-04-25 op 		    (line[l] == ' ' || line[l] == '\t'))
367 288fd238 2021-04-25 op 			continue;
368 288fd238 2021-04-25 op 		/* line has a trailing \n */
369 288fd238 2021-04-25 op 		fprintf(tmp, "%s", line);
370 288fd238 2021-04-25 op 	}
371 288fd238 2021-04-25 op 	fprintf(tmp, "%s %s %d\n", entry.domain, entry.hash, entry.verified);
372 288fd238 2021-04-25 op 
373 288fd238 2021-04-25 op 	free(line);
374 288fd238 2021-04-25 op 	e = ferror(tmp);
375 288fd238 2021-04-25 op 
376 288fd238 2021-04-25 op 	fclose(tmp);
377 288fd238 2021-04-25 op 	fclose(f);
378 288fd238 2021-04-25 op 
379 288fd238 2021-04-25 op 	if (e) {
380 288fd238 2021-04-25 op 		unlink(sfn);
381 288fd238 2021-04-25 op 		res = 0;
382 288fd238 2021-04-25 op 		goto end;
383 288fd238 2021-04-25 op 	}
384 288fd238 2021-04-25 op 
385 288fd238 2021-04-25 op 	res = rename(sfn, known_hosts_file) != -1;
386 288fd238 2021-04-25 op 
387 288fd238 2021-04-25 op end:
388 bc10f6a5 2021-07-12 op 	fs_send_ui(IMSG_UPDATE_CERT_OK, imsg->hdr.peerid, -1,
389 3a227e9a 2021-03-18 op 	    &res, sizeof(res));
390 de2a69bb 2021-05-17 op }
391 de2a69bb 2021-05-17 op 
392 de2a69bb 2021-05-17 op static void
393 de2a69bb 2021-05-17 op handle_file_open(struct imsg *imsg, size_t datalen)
394 de2a69bb 2021-05-17 op {
395 de2a69bb 2021-05-17 op 	char	*path, *e;
396 de2a69bb 2021-05-17 op 	int	 fd;
397 de2a69bb 2021-05-17 op 
398 de2a69bb 2021-05-17 op 	path = imsg->data;
399 de2a69bb 2021-05-17 op 	if (path[datalen-1] != '\0')
400 de2a69bb 2021-05-17 op 		die();
401 de2a69bb 2021-05-17 op 
402 de2a69bb 2021-05-17 op 	if ((fd = open(path, O_WRONLY | O_TRUNC | O_CREAT, 0644)) == -1) {
403 de2a69bb 2021-05-17 op 		e = strerror(errno);
404 bc10f6a5 2021-07-12 op 		fs_send_ui(IMSG_FILE_OPENED, imsg->hdr.peerid, -1,
405 de2a69bb 2021-05-17 op 		    e, strlen(e)+1);
406 de2a69bb 2021-05-17 op 	} else
407 bc10f6a5 2021-07-12 op 		fs_send_ui(IMSG_FILE_OPENED, imsg->hdr.peerid, fd,
408 de2a69bb 2021-05-17 op 		    NULL, 0);
409 3a227e9a 2021-03-18 op }
410 3a227e9a 2021-03-18 op 
411 3a227e9a 2021-03-18 op static void
412 c7107cec 2021-04-01 op handle_session_start(struct imsg *imsg, size_t datalen)
413 c7107cec 2021-04-01 op {
414 c7107cec 2021-04-01 op 	if (datalen != 0)
415 c7107cec 2021-04-01 op 		die();
416 c7107cec 2021-04-01 op 
417 c7107cec 2021-04-01 op 	if ((session = fopen(session_file, "w")) == NULL)
418 c7107cec 2021-04-01 op 		die();
419 c7107cec 2021-04-01 op }
420 c7107cec 2021-04-01 op 
421 c7107cec 2021-04-01 op static void
422 c7107cec 2021-04-01 op handle_session_tab(struct imsg *imsg, size_t datalen)
423 c7107cec 2021-04-01 op {
424 87e3e801 2021-07-17 op 	char		*url;
425 87e3e801 2021-07-17 op 	uint32_t	 flags;
426 58df4f47 2021-07-17 op 
427 c7107cec 2021-04-01 op 	if (session == NULL)
428 c7107cec 2021-04-01 op 		die();
429 c7107cec 2021-04-01 op 
430 87e3e801 2021-07-17 op 	flags = imsg->hdr.peerid;
431 58df4f47 2021-07-17 op 	url = imsg->data;
432 58df4f47 2021-07-17 op 	if (datalen == 0 || url[datalen-1] != '\0')
433 c7107cec 2021-04-01 op 		die();
434 58df4f47 2021-07-17 op 	fprintf(session, "%s", url);
435 87e3e801 2021-07-17 op 
436 87e3e801 2021-07-17 op 	if (flags & TAB_CURRENT)
437 2f524b17 2021-07-17 op 		fprintf(session, " current ");
438 2f524b17 2021-07-17 op 	else
439 2f524b17 2021-07-17 op 		fprintf(session, " - ");
440 2f524b17 2021-07-17 op }
441 87e3e801 2021-07-17 op 
442 2f524b17 2021-07-17 op static void
443 2f524b17 2021-07-17 op handle_session_tab_title(struct imsg *imsg, size_t datalen)
444 2f524b17 2021-07-17 op {
445 2f524b17 2021-07-17 op 	const char	*title;
446 2f524b17 2021-07-17 op 
447 2f524b17 2021-07-17 op 	title = imsg->data;
448 2f524b17 2021-07-17 op 	if (title == NULL) {
449 2f524b17 2021-07-17 op 		datalen = 1;
450 2f524b17 2021-07-17 op 		title = "";
451 2f524b17 2021-07-17 op 	}
452 2f524b17 2021-07-17 op 
453 2f524b17 2021-07-17 op 	if (title[datalen-1] != '\0')
454 2f524b17 2021-07-17 op 		die();
455 2f524b17 2021-07-17 op 
456 2f524b17 2021-07-17 op 	fprintf(session, "%s\n", title);
457 c7107cec 2021-04-01 op }
458 c7107cec 2021-04-01 op 
459 c7107cec 2021-04-01 op static void
460 c7107cec 2021-04-01 op handle_session_end(struct imsg *imsg, size_t datalen)
461 c7107cec 2021-04-01 op {
462 c7107cec 2021-04-01 op 	if (session == NULL)
463 c7107cec 2021-04-01 op 		die();
464 c7107cec 2021-04-01 op 	fclose(session);
465 c7107cec 2021-04-01 op 	session = NULL;
466 c7107cec 2021-04-01 op }
467 c7107cec 2021-04-01 op 
468 c7107cec 2021-04-01 op static void
469 1304bbdd 2021-03-15 op handle_dispatch_imsg(int fd, short ev, void *d)
470 35e1f40a 2021-03-14 op {
471 bc10f6a5 2021-07-12 op 	struct imsgev	*iev = d;
472 ea080950 2021-07-20 op 	int		 e;
473 ea080950 2021-07-20 op 
474 ea080950 2021-07-20 op 	if (dispatch_imsg(iev, ev, handlers, sizeof(handlers)) == -1) {
475 ea080950 2021-07-20 op 		/*
476 ea080950 2021-07-20 op 		 * This should leave a ~/.telescope/crashed file to
477 ea080950 2021-07-20 op 		 * trigger about:crash on next run.  Unfortunately, if
478 ea080950 2021-07-20 op 		 * the main process dies the fs sticks around and
479 ea080950 2021-07-20 op 		 * doesn't notice that the fd was closed.  Why EV_READ
480 ea080950 2021-07-20 op 		 * is not triggered when a fd is closed on the other end?
481 ea080950 2021-07-20 op 		 */
482 ea080950 2021-07-20 op 		e = errno;
483 ea080950 2021-07-20 op 		if ((fd = open(crashed_file, O_CREAT|O_TRUNC|O_WRONLY, 0600))
484 ea080950 2021-07-20 op 		    == -1)
485 ea080950 2021-07-20 op 			err(1, "open");
486 ea080950 2021-07-20 op 		close(fd);
487 ea080950 2021-07-20 op 		errx(1, "connection closed: %s", strerror(e));
488 ea080950 2021-07-20 op 	}
489 35e1f40a 2021-03-14 op }
490 35e1f40a 2021-03-14 op 
491 bc10f6a5 2021-07-12 op static int
492 bc10f6a5 2021-07-12 op fs_send_ui(int type, uint32_t peerid, int fd, const void *data,
493 bc10f6a5 2021-07-12 op     uint16_t datalen)
494 bc10f6a5 2021-07-12 op {
495 bc10f6a5 2021-07-12 op 	return imsg_compose_event(iev_ui, type, peerid, 0, fd,
496 bc10f6a5 2021-07-12 op 	    data, datalen);
497 bc10f6a5 2021-07-12 op }
498 bc10f6a5 2021-07-12 op 
499 35e1f40a 2021-03-14 op int
500 3a227e9a 2021-03-18 op fs_init(void)
501 35e1f40a 2021-03-14 op {
502 56e7efb4 2021-07-21 op 	strlcpy(base_path, getenv("HOME"), sizeof(base_path));
503 56e7efb4 2021-07-21 op 	strlcat(base_path, "/.telescope", sizeof(base_path));
504 56e7efb4 2021-07-21 op 	mkdir(base_path, 0700);
505 d0fd368a 2021-07-15 op 
506 56e7efb4 2021-07-21 op 	strlcpy(lockfile_path, base_path, sizeof(lockfile_path));
507 56e7efb4 2021-07-21 op 	strlcat(lockfile_path, "/lock", sizeof(lockfile_path));
508 35e1f40a 2021-03-14 op 
509 56e7efb4 2021-07-21 op 	strlcpy(bookmark_file, base_path, sizeof(bookmark_file));
510 56e7efb4 2021-07-21 op 	strlcat(bookmark_file, "/bookmarks.gmi", sizeof(bookmark_file));
511 740f578b 2021-03-15 op 
512 56e7efb4 2021-07-21 op 	strlcpy(known_hosts_file, base_path, sizeof(known_hosts_file));
513 56e7efb4 2021-07-21 op 	strlcat(known_hosts_file, "/known_hosts", sizeof(known_hosts_file));
514 3a227e9a 2021-03-18 op 
515 56e7efb4 2021-07-21 op 	strlcpy(known_hosts_tmp, base_path, sizeof(known_hosts_tmp));
516 56e7efb4 2021-07-21 op 	strlcat(known_hosts_tmp, "/known_hosts.tmp.XXXXXXXXXX",
517 288fd238 2021-04-25 op 	    sizeof(known_hosts_file));
518 288fd238 2021-04-25 op 
519 56e7efb4 2021-07-21 op 	strlcpy(session_file, base_path, sizeof(session_file));
520 56e7efb4 2021-07-21 op 	strlcat(session_file, "/session", sizeof(session_file));
521 c7107cec 2021-04-01 op 
522 56e7efb4 2021-07-21 op 	strlcpy(crashed_file, base_path, sizeof(crashed_file));
523 56e7efb4 2021-07-21 op 	strlcat(crashed_file, "/crashed", sizeof(crashed_file));
524 ea080950 2021-07-20 op 
525 3a227e9a 2021-03-18 op 	return 1;
526 3a227e9a 2021-03-18 op }
527 3a227e9a 2021-03-18 op 
528 3a227e9a 2021-03-18 op int
529 6cc5fcfe 2021-07-08 op fs_main(void)
530 3a227e9a 2021-03-18 op {
531 6cc5fcfe 2021-07-08 op 	setproctitle("fs");
532 3a227e9a 2021-03-18 op 
533 6cc5fcfe 2021-07-08 op 	fs_init();
534 6cc5fcfe 2021-07-08 op 
535 35e1f40a 2021-03-14 op 	event_init();
536 35e1f40a 2021-03-14 op 
537 bc10f6a5 2021-07-12 op 	/* Setup pipe and event handler to the main process */
538 bc10f6a5 2021-07-12 op 	if ((iev_ui = malloc(sizeof(*iev_ui))) == NULL)
539 6cc5fcfe 2021-07-08 op 		die();
540 bc10f6a5 2021-07-12 op 	imsg_init(&iev_ui->ibuf, 3);
541 bc10f6a5 2021-07-12 op 	iev_ui->handler = handle_dispatch_imsg;
542 bc10f6a5 2021-07-12 op 	iev_ui->events = EV_READ;
543 bc10f6a5 2021-07-12 op 	event_set(&iev_ui->ev, iev_ui->ibuf.fd, iev_ui->events,
544 bc10f6a5 2021-07-12 op 	    iev_ui->handler, iev_ui);
545 bc10f6a5 2021-07-12 op 	event_add(&iev_ui->ev, NULL);
546 35e1f40a 2021-03-14 op 
547 35e1f40a 2021-03-14 op 	sandbox_fs_process();
548 35e1f40a 2021-03-14 op 
549 35e1f40a 2021-03-14 op 	event_dispatch();
550 35e1f40a 2021-03-14 op 	return 0;
551 35e1f40a 2021-03-14 op }
552 3a227e9a 2021-03-18 op 
553 3a227e9a 2021-03-18 op 
554 d0fd368a 2021-07-15 op 
555 d0fd368a 2021-07-15 op int
556 b6171794 2021-07-20 op last_time_crashed(void)
557 b6171794 2021-07-20 op {
558 b6171794 2021-07-20 op 	int fd;
559 b6171794 2021-07-20 op 
560 b6171794 2021-07-20 op 	if ((fd = open(crashed_file, O_RDONLY)) == -1)
561 b6171794 2021-07-20 op 		return 0;
562 b6171794 2021-07-20 op 
563 b6171794 2021-07-20 op 	close(fd);
564 b6171794 2021-07-20 op 	unlink(crashed_file);
565 b6171794 2021-07-20 op 	return 1;
566 b6171794 2021-07-20 op }
567 b6171794 2021-07-20 op 
568 b6171794 2021-07-20 op int
569 d0fd368a 2021-07-15 op lock_session(void)
570 d0fd368a 2021-07-15 op {
571 d0fd368a 2021-07-15 op 	struct flock	lock;
572 d0fd368a 2021-07-15 op 	int		fd;
573 d0fd368a 2021-07-15 op 
574 d0fd368a 2021-07-15 op 	if ((fd = open(lockfile_path, O_WRONLY|O_CREAT, 0600)) == -1)
575 d0fd368a 2021-07-15 op 		return -1;
576 d0fd368a 2021-07-15 op 
577 d0fd368a 2021-07-15 op 	lock.l_start = 0;
578 d0fd368a 2021-07-15 op 	lock.l_len = 0;
579 d0fd368a 2021-07-15 op 	lock.l_type = F_WRLCK;
580 d0fd368a 2021-07-15 op 	lock.l_whence = SEEK_SET;
581 c6d03cf5 2021-04-25 op 
582 d0fd368a 2021-07-15 op 	if (fcntl(fd, F_SETLK, &lock) == -1) {
583 d0fd368a 2021-07-15 op 		close(fd);
584 d0fd368a 2021-07-15 op 		return -1;
585 d0fd368a 2021-07-15 op 	}
586 d0fd368a 2021-07-15 op 
587 d0fd368a 2021-07-15 op 	return fd;
588 d0fd368a 2021-07-15 op }
589 d0fd368a 2021-07-15 op 
590 c6d03cf5 2021-04-25 op static int
591 c6d03cf5 2021-04-25 op parse_khost_line(char *line, char *tmp[3])
592 c6d03cf5 2021-04-25 op {
593 c6d03cf5 2021-04-25 op 	char **ap;
594 c6d03cf5 2021-04-25 op 
595 c6d03cf5 2021-04-25 op 	for (ap = tmp; ap < &tmp[3] &&
596 c6d03cf5 2021-04-25 op 	    (*ap = strsep(&line, " \t\n")) != NULL;) {
597 c6d03cf5 2021-04-25 op 		if (**ap != '\0')
598 c6d03cf5 2021-04-25 op 			ap++;
599 c6d03cf5 2021-04-25 op 	}
600 3a227e9a 2021-03-18 op 
601 c6d03cf5 2021-04-25 op 	return ap == &tmp[3] && *line == '\0';
602 c6d03cf5 2021-04-25 op }
603 c6d03cf5 2021-04-25 op 
604 3a227e9a 2021-03-18 op int
605 3a227e9a 2021-03-18 op load_certs(struct ohash *h)
606 3a227e9a 2021-03-18 op {
607 c6d03cf5 2021-04-25 op 	char		*tmp[3], *line = NULL;
608 6cd6a9e1 2021-03-20 op 	const char	*errstr;
609 ec1fa0b0 2021-04-25 op 	size_t		 lineno = 0, linesize = 0;
610 3a227e9a 2021-03-18 op 	ssize_t		 linelen;
611 3a227e9a 2021-03-18 op 	FILE		*f;
612 3a227e9a 2021-03-18 op 	struct tofu_entry *e;
613 3a227e9a 2021-03-18 op 
614 3a227e9a 2021-03-18 op 	if ((f = fopen(known_hosts_file, "r")) == NULL)
615 3a227e9a 2021-03-18 op 		return 0;
616 3a227e9a 2021-03-18 op 
617 3a227e9a 2021-03-18 op 	while ((linelen = getline(&line, &linesize, f)) != -1) {
618 3a227e9a 2021-03-18 op 		if ((e = calloc(1, sizeof(*e))) == NULL)
619 3a227e9a 2021-03-18 op 			abort();
620 3a227e9a 2021-03-18 op 
621 ec1fa0b0 2021-04-25 op 		lineno++;
622 3a227e9a 2021-03-18 op 
623 c6d03cf5 2021-04-25 op 		if (parse_khost_line(line, tmp)) {
624 c6d03cf5 2021-04-25 op 			strlcpy(e->domain, tmp[0], sizeof(e->domain));
625 c6d03cf5 2021-04-25 op 			strlcpy(e->hash, tmp[1], sizeof(e->hash));
626 3a227e9a 2021-03-18 op 
627 c6d03cf5 2021-04-25 op 			e->verified = strtonum(tmp[2], 0, 1, &errstr);
628 c6d03cf5 2021-04-25 op 			if (errstr != NULL)
629 c6d03cf5 2021-04-25 op 				errx(1, "%s:%zu verification for %s is %s: %s",
630 c6d03cf5 2021-04-25 op 				    known_hosts_file, lineno,
631 c6d03cf5 2021-04-25 op 				    e->domain, errstr, tmp[2]);
632 c6d03cf5 2021-04-25 op 			tofu_add(h, e);
633 c6d03cf5 2021-04-25 op 		} else {
634 ec1fa0b0 2021-04-25 op 			warnx("%s:%zu invalid entry",
635 ec1fa0b0 2021-04-25 op 			    known_hosts_file, lineno);
636 ec1fa0b0 2021-04-25 op 			free(e);
637 c6d03cf5 2021-04-25 op 		}
638 3a227e9a 2021-03-18 op 	}
639 3a227e9a 2021-03-18 op 
640 3a227e9a 2021-03-18 op 	free(line);
641 3a227e9a 2021-03-18 op 	return ferror(f);
642 3a227e9a 2021-03-18 op }
643 c7107cec 2021-04-01 op