op public repos

Blame

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