1 /*
2  * Copyright (c) 2019 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <sys/types.h>
18 #include <sys/queue.h>
19 #include <sys/uio.h>
20 #include <sys/time.h>
21 
22 #include <stdint.h>
23 #include <imsg.h>
24 #include <limits.h>
25 #include <signal.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <sha1.h>
30 #include <unistd.h>
31 #include <zlib.h>
32 
33 #include "got_error.h"
34 #include "got_object.h"
35 #include "got_repository.h"
36 
37 #include "got_lib_delta.h"
38 #include "got_lib_object.h"
39 #include "got_lib_privsep.h"
40 #include "got_lib_gitconfig.h"
41 
42 static volatile sig_atomic_t sigint_received;
43 
44 static void
45 catch_sigint(int signo)
46 {
47 	sigint_received = 1;
48 }
49 
50 static const struct got_error *
51 send_gitconfig_int(struct imsgbuf *ibuf, int value)
52 {
53 	if (imsg_compose(ibuf, GOT_IMSG_GITCONFIG_INT_VAL, 0, 0, -1,
54 	    &value, sizeof(value)) == -1)
55 		return got_error_from_errno("imsg_compose GITCONFIG_INT_VAL");
56 
57 	return got_privsep_flush_imsg(ibuf);
58 }
59 
60 static const struct got_error *
61 gitconfig_num_request(struct imsgbuf *ibuf, struct got_gitconfig *gitconfig,
62     const char *section, const char *tag, int def)
63 {
64 	int value;
65 
66 	if (gitconfig == NULL)
67 		return got_error(GOT_ERR_PRIVSEP_MSG);
68 
69 	value = got_gitconfig_get_num(gitconfig, section, tag, def);
70 	return send_gitconfig_int(ibuf, value);
71 }
72 
73 static const struct got_error *
74 send_gitconfig_str(struct imsgbuf *ibuf, const char *value)
75 {
76 	size_t len = value ? strlen(value) : 0;
77 
78 	if (imsg_compose(ibuf, GOT_IMSG_GITCONFIG_STR_VAL, 0, 0, -1,
79 	    value, len) == -1)
80 		return got_error_from_errno("imsg_compose GITCONFIG_STR_VAL");
81 
82 	return got_privsep_flush_imsg(ibuf);
83 }
84 
85 static const struct got_error *
86 send_gitconfig_pair(struct imsgbuf *ibuf, const char *key, const char *val)
87 {
88 	struct ibuf *wbuf;
89 	size_t klen = key ? strlen(key) : 0;
90 	size_t vlen = val ? strlen(val) : 0;
91 	size_t tot = sizeof(klen) + sizeof(vlen) + klen + vlen;
92 
93 	if (tot > MAX_IMSGSIZE - IMSG_HEADER_SIZE)
94 		return got_error(GOT_ERR_NO_SPACE);
95 
96 	wbuf = imsg_create(ibuf, GOT_IMSG_GITCONFIG_PAIR, 0, 0, tot);
97 	if (wbuf == NULL)
98 		return got_error_from_errno("imsg_create GITCONFIG_PAIR");
99 
100 	/* Keep in sync with got_imsg_gitconfig_pair */
101 	if (imsg_add(wbuf, &klen, sizeof(klen)) == -1)
102 		return got_error_from_errno("imsg_add GITCONFIG_PAIR");
103 	if (imsg_add(wbuf, &vlen, sizeof(vlen)) == -1)
104 		return got_error_from_errno("imsg_add GITCONFIG_PAIR");
105 	if (imsg_add(wbuf, key, klen) == -1)
106 		return got_error_from_errno("imsg_add GITCONFIG_PAIR");
107 	if (imsg_add(wbuf, val, vlen) == -1)
108 		return got_error_from_errno("imsg_add GITCONFIG_PAIR");
109 
110 	wbuf->fd = -1;
111 	imsg_close(ibuf, wbuf);
112 	return got_privsep_flush_imsg(ibuf);
113 }
114 
115 static const struct got_error *
116 gitconfig_str_request(struct imsgbuf *ibuf, struct got_gitconfig *gitconfig,
117     const char *section, const char *tag)
118 {
119 	char *value;
120 
121 	if (gitconfig == NULL)
122 		return got_error(GOT_ERR_PRIVSEP_MSG);
123 
124 	value = got_gitconfig_get_str(gitconfig, section, tag);
125 	return send_gitconfig_str(ibuf, value);
126 }
127 
128 static const struct got_error *
129 send_gitconfig_remotes(struct imsgbuf *ibuf, struct got_remote_repo *remotes,
130     int nremotes)
131 {
132 	const struct got_error *err = NULL;
133 	struct got_imsg_remotes iremotes;
134 	int i;
135 
136 	iremotes.nremotes = nremotes;
137 	if (imsg_compose(ibuf, GOT_IMSG_GITCONFIG_REMOTES, 0, 0, -1,
138 	    &iremotes, sizeof(iremotes)) == -1)
139 		return got_error_from_errno("imsg_compose GITCONFIG_REMOTES");
140 
141 	err = got_privsep_flush_imsg(ibuf);
142 	imsg_clear(ibuf);
143 	if (err)
144 		return err;
145 
146 	for (i = 0; i < nremotes; i++) {
147 		struct got_imsg_remote iremote;
148 		size_t len = sizeof(iremote);
149 		struct ibuf *wbuf;
150 
151 		iremote.mirror_references = remotes[i].mirror_references;
152 		iremote.name_len = strlen(remotes[i].name);
153 		len += iremote.name_len;
154 		iremote.fetch_url_len = strlen(remotes[i].fetch_url);
155 		len += iremote.fetch_url_len;
156 		iremote.send_url_len = strlen(remotes[i].send_url);
157 		len += iremote.send_url_len;
158 
159 		wbuf = imsg_create(ibuf, GOT_IMSG_GITCONFIG_REMOTE, 0, 0, len);
160 		if (wbuf == NULL)
161 			return got_error_from_errno(
162 			    "imsg_create GITCONFIG_REMOTE");
163 
164 		if (imsg_add(wbuf, &iremote, sizeof(iremote)) == -1)
165 			return got_error_from_errno(
166 			    "imsg_add GITCONFIG_REMOTE");
167 
168 		if (imsg_add(wbuf, remotes[i].name, iremote.name_len) == -1)
169 			return got_error_from_errno(
170 			    "imsg_add GITCONFIG_REMOTE");
171 		if (imsg_add(wbuf, remotes[i].fetch_url, iremote.fetch_url_len) == -1)
172 			return got_error_from_errno(
173 			    "imsg_add GITCONFIG_REMOTE");
174 		if (imsg_add(wbuf, remotes[i].send_url, iremote.send_url_len) == -1)
175 			return got_error_from_errno(
176 			    "imsg_add GITCONFIG_REMOTE");
177 
178 		wbuf->fd = -1;
179 		imsg_close(ibuf, wbuf);
180 		err = got_privsep_flush_imsg(ibuf);
181 		if (err)
182 			return err;
183 	}
184 
185 	return NULL;
186 }
187 
188 static int
189 get_boolean_val(char *val)
190 {
191     return (strcasecmp(val, "true") == 0 ||
192 	strcasecmp(val, "on") == 0 ||
193 	strcasecmp(val, "yes") == 0 ||
194 	strcmp(val, "1") == 0);
195 }
196 
197 static const struct got_error *
198 gitconfig_remotes_request(struct imsgbuf *ibuf, struct got_gitconfig *gitconfig)
199 {
200 	const struct got_error *err = NULL;
201 	struct got_gitconfig_list *sections;
202 	struct got_gitconfig_list_node *node;
203 	struct got_remote_repo *remotes = NULL;
204 	int nremotes = 0, i;
205 
206 	if (gitconfig == NULL)
207 		return got_error(GOT_ERR_PRIVSEP_MSG);
208 
209 	err = got_gitconfig_get_section_list(&sections, gitconfig);
210 	if (err)
211 		return err;
212 
213 	TAILQ_FOREACH(node, &sections->fields, link) {
214 		if (strncasecmp("remote \"", node->field, 8) != 0)
215 			continue;
216 		nremotes++;
217 	}
218 
219 	if (nremotes == 0) {
220 		err = send_gitconfig_remotes(ibuf, NULL, 0);
221 		goto done;
222 	}
223 
224 	remotes = recallocarray(NULL, 0, nremotes, sizeof(*remotes));
225 	if (remotes == NULL) {
226 		err = got_error_from_errno("recallocarray");
227 		goto done;
228 	}
229 
230 	i = 0;
231 	TAILQ_FOREACH(node, &sections->fields, link) {
232 		char *name, *end, *mirror;
233 
234 		if (strncasecmp("remote \"", node->field, 8) != 0)
235 			continue;
236 
237 		name = strdup(node->field + 8);
238 		if (name == NULL) {
239 			err = got_error_from_errno("strdup");
240 			goto done;
241 		}
242 		end = strrchr(name, '"');
243 		if (end)
244 			*end = '\0';
245 		remotes[i].name = name;
246 
247 		remotes[i].fetch_url = got_gitconfig_get_str(gitconfig,
248 		    node->field, "url");
249 		if (remotes[i].fetch_url == NULL) {
250 			err = got_error(GOT_ERR_GITCONFIG_SYNTAX);
251 			goto done;
252 		}
253 
254 		remotes[i].send_url = got_gitconfig_get_str(gitconfig,
255 		    node->field, "pushurl");
256 		if (remotes[i].send_url == NULL)
257 			remotes[i].send_url = got_gitconfig_get_str(gitconfig,
258 			    node->field, "url");
259 		if (remotes[i].send_url == NULL) {
260 			err = got_error(GOT_ERR_GITCONFIG_SYNTAX);
261 			goto done;
262 		}
263 
264 		remotes[i].mirror_references = 0;
265 		mirror = got_gitconfig_get_str(gitconfig, node->field,
266 		    "mirror");
267 		if (mirror != NULL && get_boolean_val(mirror))
268 			remotes[i].mirror_references = 1;
269 
270 		i++;
271 	}
272 
273 	err = send_gitconfig_remotes(ibuf, remotes, nremotes);
274 done:
275 	for (i = 0; i < nremotes; i++)
276 		free(remotes[i].name);
277 	free(remotes);
278 	got_gitconfig_free_list(sections);
279 	return err;
280 }
281 
282 static const struct got_error *
283 gitconfig_owner_request(struct imsgbuf *ibuf, struct got_gitconfig *gitconfig)
284 {
285 	char *value;
286 
287 	if (gitconfig == NULL)
288 		return got_error(GOT_ERR_PRIVSEP_MSG);
289 
290 	value = got_gitconfig_get_str(gitconfig, "gotweb", "owner");
291 	if (value)
292 		return send_gitconfig_str(ibuf, value);
293 	value = got_gitconfig_get_str(gitconfig, "gitweb", "owner");
294 	return send_gitconfig_str(ibuf, value);
295 }
296 
297 static const struct got_error *
298 gitconfig_extensions_request(struct imsgbuf *ibuf,
299     struct got_gitconfig *gitconfig)
300 {
301 	const struct got_error *err = NULL;
302 	struct got_gitconfig_list *tags;
303 	struct got_gitconfig_list_node *node;
304 	int nextensions = 0;
305 	char *val;
306 
307 	if (gitconfig == NULL)
308 		return got_error(GOT_ERR_PRIVSEP_MSG);
309 
310 	tags = got_gitconfig_get_tag_list(gitconfig, "extensions");
311 	if (tags == NULL)
312 		return send_gitconfig_int(ibuf, 0);
313 
314 	TAILQ_FOREACH(node, &tags->fields, link)
315 		nextensions++;
316 
317 	err = send_gitconfig_int(ibuf, nextensions);
318 	if (err)
319 		goto done;
320 
321 	TAILQ_FOREACH(node, &tags->fields, link) {
322 		val = got_gitconfig_get_str(gitconfig, "extensions",
323 		    node->field);
324 		err = send_gitconfig_pair(ibuf, node->field, val);
325 		if (err)
326 			goto done;
327 	}
328 done:
329 	got_gitconfig_free_list(tags);
330 	return err;
331 }
332 
333 int
334 main(int argc, char *argv[])
335 {
336 	const struct got_error *err = NULL;
337 	struct imsgbuf ibuf;
338 	size_t datalen;
339 	struct got_gitconfig *gitconfig = NULL;
340 #if 0
341 	static int attached;
342 
343 	while (!attached)
344 		sleep(1);
345 #endif
346 	signal(SIGINT, catch_sigint);
347 
348 	imsg_init(&ibuf, GOT_IMSG_FD_CHILD);
349 
350 #ifndef PROFILE
351 	/* revoke access to most system calls */
352 	if (pledge("stdio recvfd", NULL) == -1) {
353 		err = got_error_from_errno("pledge");
354 		got_privsep_send_error(&ibuf, err);
355 		return 1;
356 	}
357 #endif
358 
359 	for (;;) {
360 		struct imsg imsg;
361 
362 		memset(&imsg, 0, sizeof(imsg));
363 		imsg.fd = -1;
364 
365 		if (sigint_received) {
366 			err = got_error(GOT_ERR_CANCELLED);
367 			break;
368 		}
369 
370 		err = got_privsep_recv_imsg(&imsg, &ibuf, 0);
371 		if (err) {
372 			if (err->code == GOT_ERR_PRIVSEP_PIPE)
373 				err = NULL;
374 			break;
375 		}
376 
377 		if (imsg.hdr.type == GOT_IMSG_STOP)
378 			break;
379 
380 		switch (imsg.hdr.type) {
381 		case GOT_IMSG_GITCONFIG_PARSE_REQUEST:
382 			datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
383 			if (datalen != 0) {
384 				err = got_error(GOT_ERR_PRIVSEP_LEN);
385 				break;
386 			}
387 			if (imsg.fd == -1){
388 				err = got_error(GOT_ERR_PRIVSEP_NO_FD);
389 				break;
390 			}
391 
392 			if (gitconfig)
393 				got_gitconfig_close(gitconfig);
394 			err = got_gitconfig_open(&gitconfig, imsg.fd);
395 			break;
396 		case GOT_IMSG_GITCONFIG_REPOSITORY_FORMAT_VERSION_REQUEST:
397 			err = gitconfig_num_request(&ibuf, gitconfig, "core",
398 			    "repositoryformatversion", 0);
399 			break;
400 		case GOT_IMSG_GITCONFIG_REPOSITORY_EXTENSIONS_REQUEST:
401 			err = gitconfig_extensions_request(&ibuf, gitconfig);
402 			break;
403 		case GOT_IMSG_GITCONFIG_AUTHOR_NAME_REQUEST:
404 			err = gitconfig_str_request(&ibuf, gitconfig, "user",
405 			    "name");
406 			break;
407 		case GOT_IMSG_GITCONFIG_AUTHOR_EMAIL_REQUEST:
408 			err = gitconfig_str_request(&ibuf, gitconfig, "user",
409 			    "email");
410 			break;
411 		case GOT_IMSG_GITCONFIG_REMOTES_REQUEST:
412 			err = gitconfig_remotes_request(&ibuf, gitconfig);
413 			break;
414 		case GOT_IMSG_GITCONFIG_OWNER_REQUEST:
415 			err = gitconfig_owner_request(&ibuf, gitconfig);
416 			break;
417 		default:
418 			err = got_error(GOT_ERR_PRIVSEP_MSG);
419 			break;
420 		}
421 
422 		if (imsg.fd != -1) {
423 			if (close(imsg.fd) == -1 && err == NULL)
424 				err = got_error_from_errno("close");
425 		}
426 
427 		imsg_free(&imsg);
428 		if (err)
429 			break;
430 	}
431 
432 	imsg_clear(&ibuf);
433 	if (err) {
434 		if (!sigint_received && err->code != GOT_ERR_PRIVSEP_PIPE) {
435 			fprintf(stderr, "%s: %s\n", getprogname(), err->msg);
436 			got_privsep_send_error(&ibuf, err);
437 		}
438 	}
439 	if (close(GOT_IMSG_FD_CHILD) == -1 && err == NULL)
440 		err = got_error_from_errno("close");
441 	return err ? 1 : 0;
442 }