1 /*
2  * Copyright (c) 2018, 2019, 2020 Stefan Sperling <stsp@openbsd.org>
3  * Copyright (c) 2020 Ori Bernstein <ori@openbsd.org>
4  *
5  * Permission to use, copy, modify, and distribute this software for any
6  * purpose with or without fee is hereby granted, provided that the above
7  * copyright notice and this permission notice appear in all copies.
8  *
9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16  */
17 
18 #include <sys/types.h>
19 #include <sys/uio.h>
20 #include <sys/wait.h>
21 
22 #include <ctype.h>
23 #include <limits.h>
24 #include <signal.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <errno.h>
29 #include <stdint.h>
30 #include <poll.h>
31 #include <unistd.h>
32 #include <zlib.h>
33 #include <time.h>
34 
35 #include "got_compat.h"
36 
37 #include "got_object.h"
38 #include "got_error.h"
39 #include "got_path.h"
40 #include "got_repository.h"
41 
42 #include "got_lib_sha1.h"
43 #include "got_lib_delta.h"
44 #include "got_lib_inflate.h"
45 #include "got_lib_object.h"
46 #include "got_lib_object_parse.h"
47 #include "got_lib_privsep.h"
48 #include "got_lib_pack.h"
49 
50 #ifndef MIN
51 #define	MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
52 #endif
53 
54 #ifndef nitems
55 #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
56 #endif
57 
58 static const struct got_error *
59 poll_fd(int fd, int events, int timeout)
60 {
61 	struct pollfd pfd[1];
62 	struct timespec ts;
63 	sigset_t sigset;
64 	int n;
65 
66 	pfd[0].fd = fd;
67 	pfd[0].events = events;
68 
69 	ts.tv_sec = timeout;
70 	ts.tv_nsec = 0;
71 
72 	if (sigemptyset(&sigset) == -1)
73 		return got_error_from_errno("sigemptyset");
74 	if (sigaddset(&sigset, SIGWINCH) == -1)
75 		return got_error_from_errno("sigaddset");
76 
77 	n = ppoll(pfd, 1, timeout == INFTIM ? NULL : &ts, &sigset);
78 	if (n == -1)
79 		return got_error_from_errno("ppoll");
80 	if (n == 0)
81 		return got_error(GOT_ERR_TIMEOUT);
82 	if (pfd[0].revents & (POLLERR | POLLNVAL))
83 		return got_error_from_errno("poll error");
84 	if (pfd[0].revents & (events | POLLHUP))
85 		return NULL;
86 
87 	return got_error(GOT_ERR_INTERRUPT);
88 }
89 
90 static const struct got_error *
91 read_imsg(struct imsgbuf *ibuf)
92 {
93 	const struct got_error *err;
94 	size_t n;
95 
96 	err = poll_fd(ibuf->fd, POLLIN, INFTIM);
97 	if (err)
98 		return err;
99 
100 	n = imsg_read(ibuf);
101 	if (n == -1) {
102 		if (errno == EAGAIN) /* Could be a file-descriptor leak. */
103 			return got_error(GOT_ERR_PRIVSEP_NO_FD);
104 		return got_error(GOT_ERR_PRIVSEP_READ);
105 	}
106 	if (n == 0)
107 		return got_error(GOT_ERR_PRIVSEP_PIPE);
108 
109 	return NULL;
110 }
111 
112 const struct got_error *
113 got_privsep_wait_for_child(pid_t pid)
114 {
115 	int child_status;
116 
117 	if (waitpid(pid, &child_status, 0) == -1)
118 		return got_error_from_errno("waitpid");
119 
120 	if (!WIFEXITED(child_status))
121 		return got_error(GOT_ERR_PRIVSEP_DIED);
122 
123 	if (WEXITSTATUS(child_status) != 0)
124 		return got_error(GOT_ERR_PRIVSEP_EXIT);
125 
126 	return NULL;
127 }
128 
129 static const struct got_error *
130 recv_imsg_error(struct imsg *imsg, size_t datalen)
131 {
132 	struct got_imsg_error *ierr;
133 
134 	if (datalen != sizeof(*ierr))
135 		return got_error(GOT_ERR_PRIVSEP_LEN);
136 
137 	ierr = imsg->data;
138 	if (ierr->code == GOT_ERR_ERRNO) {
139 		static struct got_error serr;
140 		serr.code = GOT_ERR_ERRNO;
141 		serr.msg = strerror(ierr->errno_code);
142 		return &serr;
143 	}
144 
145 	return got_error(ierr->code);
146 }
147 
148 const struct got_error *
149 got_privsep_recv_imsg(struct imsg *imsg, struct imsgbuf *ibuf,
150     size_t min_datalen)
151 {
152 	const struct got_error *err;
153 	ssize_t n;
154 
155 	n = imsg_get(ibuf, imsg);
156 	if (n == -1)
157 		return got_error_from_errno("imsg_get");
158 
159 	while (n == 0) {
160 		err = read_imsg(ibuf);
161 		if (err)
162 			return err;
163 		n = imsg_get(ibuf, imsg);
164 		if (n == -1)
165 			return got_error_from_errno("imsg_get");
166 	}
167 
168 	if (imsg->hdr.len < IMSG_HEADER_SIZE + min_datalen)
169 		return got_error(GOT_ERR_PRIVSEP_LEN);
170 
171 	if (imsg->hdr.type == GOT_IMSG_ERROR) {
172 		size_t datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
173 		return recv_imsg_error(imsg, datalen);
174 	}
175 
176 	return NULL;
177 }
178 
179 /* Attempt to send an error in an imsg. Complain on stderr as a last resort. */
180 void
181 got_privsep_send_error(struct imsgbuf *ibuf, const struct got_error *err)
182 {
183 	const struct got_error *poll_err;
184 	struct got_imsg_error ierr;
185 	int ret;
186 
187 	ierr.code = err->code;
188 	if (err->code == GOT_ERR_ERRNO)
189 		ierr.errno_code = errno;
190 	else
191 		ierr.errno_code = 0;
192 	ret = imsg_compose(ibuf, GOT_IMSG_ERROR, 0, 0, -1, &ierr, sizeof(ierr));
193 	if (ret == -1) {
194 		fprintf(stderr, "%s: error %d \"%s\": imsg_compose: %s\n",
195 		    getprogname(), err->code, err->msg, strerror(errno));
196 		return;
197 	}
198 
199 	poll_err = poll_fd(ibuf->fd, POLLOUT, INFTIM);
200 	if (poll_err) {
201 		fprintf(stderr, "%s: error %d \"%s\": poll: %s\n",
202 		    getprogname(), err->code, err->msg, poll_err->msg);
203 		return;
204 	}
205 
206 	ret = imsg_flush(ibuf);
207 	if (ret == -1) {
208 		fprintf(stderr, "%s: error %d \"%s\": imsg_flush: %s\n",
209 		    getprogname(), err->code, err->msg, strerror(errno));
210 		imsg_clear(ibuf);
211 		return;
212 	}
213 }
214 
215 static const struct got_error *
216 flush_imsg(struct imsgbuf *ibuf)
217 {
218 	const struct got_error *err;
219 
220 	err = poll_fd(ibuf->fd, POLLOUT, INFTIM);
221 	if (err)
222 		return err;
223 
224 	if (imsg_flush(ibuf) == -1) {
225 		imsg_clear(ibuf);
226 		return got_error_from_errno("imsg_flush");
227 	}
228 
229 	return NULL;
230 }
231 
232 const struct got_error *
233 got_privsep_flush_imsg(struct imsgbuf *ibuf)
234 {
235 	return flush_imsg(ibuf);
236 }
237 
238 const struct got_error *
239 got_privsep_send_stop(int fd)
240 {
241 	const struct got_error *err = NULL;
242 	struct imsgbuf ibuf;
243 
244 	imsg_init(&ibuf, fd);
245 
246 	if (imsg_compose(&ibuf, GOT_IMSG_STOP, 0, 0, -1, NULL, 0) == -1)
247 		return got_error_from_errno("imsg_compose STOP");
248 
249 	err = flush_imsg(&ibuf);
250 	return err;
251 }
252 
253 const struct got_error *
254 got_privsep_send_obj_req(struct imsgbuf *ibuf, int fd,
255     struct got_object_id *id)
256 {
257 	if (imsg_compose(ibuf, GOT_IMSG_OBJECT_REQUEST, 0, 0, fd,
258 	    id, sizeof(*id)) == -1)
259 		return got_error_from_errno("imsg_compose OBJECT_REQUEST");
260 
261 	return flush_imsg(ibuf);
262 }
263 
264 const struct got_error *
265 got_privsep_send_raw_obj_req(struct imsgbuf *ibuf, int fd,
266     struct got_object_id *id)
267 {
268 	if (imsg_compose(ibuf, GOT_IMSG_RAW_OBJECT_REQUEST, 0, 0, fd,
269 	    id, sizeof(*id)) == -1)
270 		return got_error_from_errno("imsg_compose RAW_OBJECT_REQUEST");
271 
272 	return flush_imsg(ibuf);
273 }
274 
275 const struct got_error *
276 got_privsep_send_raw_obj_outfd(struct imsgbuf *ibuf, int outfd)
277 {
278 	const struct got_error *err = NULL;
279 
280 	if (imsg_compose(ibuf, GOT_IMSG_RAW_OBJECT_OUTFD, 0, 0, outfd, NULL, 0)
281 	    == -1) {
282 		err = got_error_from_errno("imsg_compose RAW_OBJECT_OUTFD");
283 		close(outfd);
284 		return err;
285 	}
286 
287 	return flush_imsg(ibuf);
288 }
289 
290 const struct got_error *
291 got_privsep_send_raw_obj(struct imsgbuf *ibuf, off_t size, size_t hdrlen,
292     uint8_t *data)
293 {
294 	const struct got_error *err = NULL;
295 	struct got_imsg_raw_obj iobj;
296 	size_t len = sizeof(iobj);
297 	struct ibuf *wbuf;
298 
299 	iobj.hdrlen = hdrlen;
300 	iobj.size = size;
301 
302 	if (data && size + hdrlen <= GOT_PRIVSEP_INLINE_OBJECT_DATA_MAX)
303 		len += (size_t)size + hdrlen;
304 
305 	wbuf = imsg_create(ibuf, GOT_IMSG_RAW_OBJECT, 0, 0, len);
306 	if (wbuf == NULL) {
307 		err = got_error_from_errno("imsg_create RAW_OBJECT");
308 		return err;
309 	}
310 
311 	if (imsg_add(wbuf, &iobj, sizeof(iobj)) == -1) {
312 		err = got_error_from_errno("imsg_add RAW_OBJECT");
313 		ibuf_free(wbuf);
314 		return err;
315 	}
316 
317 	if (data && size + hdrlen <= GOT_PRIVSEP_INLINE_OBJECT_DATA_MAX) {
318 		if (imsg_add(wbuf, data, size + hdrlen) == -1) {
319 			err = got_error_from_errno("imsg_add RAW_OBJECT");
320 			ibuf_free(wbuf);
321 			return err;
322 		}
323 	}
324 
325 	wbuf->fd = -1;
326 	imsg_close(ibuf, wbuf);
327 
328 	return flush_imsg(ibuf);
329 }
330 
331 const struct got_error *
332 got_privsep_recv_raw_obj(uint8_t **outbuf, off_t *size, size_t *hdrlen,
333     struct imsgbuf *ibuf)
334 {
335 	const struct got_error *err = NULL;
336 	struct imsg imsg;
337 	struct got_imsg_raw_obj *iobj;
338 	size_t datalen;
339 
340 	*outbuf = NULL;
341 
342 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
343 	if (err)
344 		return err;
345 
346 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
347 
348 	switch (imsg.hdr.type) {
349 	case GOT_IMSG_RAW_OBJECT:
350 		if (datalen < sizeof(*iobj)) {
351 			err = got_error(GOT_ERR_PRIVSEP_LEN);
352 			break;
353 		}
354 		iobj = imsg.data;
355 		*size = iobj->size;
356 		*hdrlen = iobj->hdrlen;
357 
358 		if (datalen == sizeof(*iobj)) {
359 			/* Data has been written to file descriptor. */
360 			break;
361 		}
362 
363 		if (*size + *hdrlen > GOT_PRIVSEP_INLINE_OBJECT_DATA_MAX) {
364 			err = got_error(GOT_ERR_PRIVSEP_LEN);
365 			break;
366 		}
367 
368 		*outbuf = malloc(*size + *hdrlen);
369 		if (*outbuf == NULL) {
370 			err = got_error_from_errno("malloc");
371 			break;
372 		}
373 		memcpy(*outbuf, imsg.data + sizeof(*iobj), *size + *hdrlen);
374 		break;
375 	default:
376 		err = got_error(GOT_ERR_PRIVSEP_MSG);
377 		break;
378 	}
379 
380 	imsg_free(&imsg);
381 
382 	return err;
383 }
384 
385 const struct got_error *
386 got_privsep_send_commit_req(struct imsgbuf *ibuf, int fd,
387     struct got_object_id *id, int pack_idx)
388 {
389 	const struct got_error *err = NULL;
390 	struct got_imsg_packed_object iobj;
391 	void *data;
392 	size_t len;
393 
394 	if (pack_idx != -1) { /* commit is packed */
395 		iobj.idx = pack_idx;
396 		memcpy(iobj.id, id->sha1, sizeof(iobj.id));
397 		data = &iobj;
398 		len = sizeof(iobj);
399 	} else {
400 		data = id;
401 		len = sizeof(*id);
402 	}
403 
404 	if (imsg_compose(ibuf, GOT_IMSG_COMMIT_REQUEST, 0, 0, fd, data, len)
405 	    == -1) {
406 		err = got_error_from_errno("imsg_compose COMMIT_REQUEST");
407 		close(fd);
408 		return err;
409 	}
410 
411 	return flush_imsg(ibuf);
412 }
413 
414 const struct got_error *
415 got_privsep_send_tree_req(struct imsgbuf *ibuf, int fd,
416     struct got_object_id *id, int pack_idx)
417 {
418 	const struct got_error *err = NULL;
419 	struct ibuf *wbuf;
420 	size_t len;
421 
422 	if (pack_idx != -1)
423 		len = sizeof(struct got_imsg_packed_object);
424 	else
425 		len = sizeof(*id);
426 
427 	wbuf = imsg_create(ibuf, GOT_IMSG_TREE_REQUEST, 0, 0, len);
428 	if (wbuf == NULL)
429 		return got_error_from_errno("imsg_create TREE_REQUEST");
430 
431 	if (imsg_add(wbuf, id->sha1, SHA1_DIGEST_LENGTH) == -1) {
432 		err = got_error_from_errno("imsg_add TREE_REQUEST");
433 		ibuf_free(wbuf);
434 		return err;
435 	}
436 
437 	if (pack_idx != -1) { /* tree is packed */
438 		if (imsg_add(wbuf, &pack_idx, sizeof(pack_idx)) == -1) {
439 			err = got_error_from_errno("imsg_add TREE_REQUEST");
440 			ibuf_free(wbuf);
441 			return err;
442 		}
443 	}
444 
445 	wbuf->fd = fd;
446 	imsg_close(ibuf, wbuf);
447 
448 	return flush_imsg(ibuf);
449 }
450 
451 const struct got_error *
452 got_privsep_send_tag_req(struct imsgbuf *ibuf, int fd,
453     struct got_object_id *id, int pack_idx)
454 {
455 	struct got_imsg_packed_object iobj;
456 	void *data;
457 	size_t len;
458 
459 	if (pack_idx != -1) { /* tag is packed */
460 		iobj.idx = pack_idx;
461 		memcpy(iobj.id, id->sha1, sizeof(iobj.id));
462 		data = &iobj;
463 		len = sizeof(iobj);
464 	} else {
465 		data = id;
466 		len = sizeof(*id);
467 	}
468 
469 	if (imsg_compose(ibuf, GOT_IMSG_TAG_REQUEST, 0, 0, fd, data, len)
470 	    == -1)
471 		return got_error_from_errno("imsg_compose TAG_REQUEST");
472 
473 	return flush_imsg(ibuf);
474 }
475 
476 const struct got_error *
477 got_privsep_send_blob_req(struct imsgbuf *ibuf, int infd,
478     struct got_object_id *id, int pack_idx)
479 {
480 	const struct got_error *err = NULL;
481 	struct got_imsg_packed_object iobj;
482 	void *data;
483 	size_t len;
484 
485 	if (pack_idx != -1) { /* blob is packed */
486 		iobj.idx = pack_idx;
487 		memcpy(iobj.id, id->sha1, sizeof(iobj.id));
488 		data = &iobj;
489 		len = sizeof(iobj);
490 	} else {
491 		data = id;
492 		len = sizeof(*id);
493 	}
494 
495 	if (imsg_compose(ibuf, GOT_IMSG_BLOB_REQUEST, 0, 0, infd, data, len)
496 	    == -1) {
497 		err = got_error_from_errno("imsg_compose BLOB_REQUEST");
498 		close(infd);
499 		return err;
500 	}
501 
502 	return flush_imsg(ibuf);
503 }
504 
505 const struct got_error *
506 got_privsep_send_blob_outfd(struct imsgbuf *ibuf, int outfd)
507 {
508 	const struct got_error *err = NULL;
509 
510 	if (imsg_compose(ibuf, GOT_IMSG_BLOB_OUTFD, 0, 0, outfd, NULL, 0)
511 	    == -1) {
512 		err = got_error_from_errno("imsg_compose BLOB_OUTFD");
513 		close(outfd);
514 		return err;
515 	}
516 
517 	return flush_imsg(ibuf);
518 }
519 
520 static const struct got_error *
521 send_fd(struct imsgbuf *ibuf, int imsg_code, int fd)
522 {
523 	const struct got_error *err = NULL;
524 
525 	if (imsg_compose(ibuf, imsg_code, 0, 0, fd, NULL, 0) == -1) {
526 		err = got_error_from_errno("imsg_compose TMPFD");
527 		close(fd);
528 		return err;
529 	}
530 
531 	return flush_imsg(ibuf);
532 }
533 
534 const struct got_error *
535 got_privsep_send_tmpfd(struct imsgbuf *ibuf, int fd)
536 {
537 	return send_fd(ibuf, GOT_IMSG_TMPFD, fd);
538 }
539 
540 const struct got_error *
541 got_privsep_send_obj(struct imsgbuf *ibuf, struct got_object *obj)
542 {
543 	struct got_imsg_object iobj;
544 
545 	memcpy(iobj.id, obj->id.sha1, sizeof(iobj.id));
546 	iobj.type = obj->type;
547 	iobj.flags = obj->flags;
548 	iobj.hdrlen = obj->hdrlen;
549 	iobj.size = obj->size;
550 	if (iobj.flags & GOT_OBJ_FLAG_PACKED) {
551 		iobj.pack_offset = obj->pack_offset;
552 		iobj.pack_idx = obj->pack_idx;
553 	}
554 
555 	if (imsg_compose(ibuf, GOT_IMSG_OBJECT, 0, 0, -1, &iobj, sizeof(iobj))
556 	    == -1)
557 		return got_error_from_errno("imsg_compose OBJECT");
558 
559 	return flush_imsg(ibuf);
560 }
561 
562 const struct got_error *
563 got_privsep_send_fetch_req(struct imsgbuf *ibuf, int fd,
564     struct got_pathlist_head *have_refs, int fetch_all_branches,
565     struct got_pathlist_head *wanted_branches,
566     struct got_pathlist_head *wanted_refs, int list_refs_only, int verbosity)
567 {
568 	const struct got_error *err = NULL;
569 	struct ibuf *wbuf;
570 	size_t len;
571 	struct got_pathlist_entry *pe;
572 	struct got_imsg_fetch_request fetchreq;
573 
574 	memset(&fetchreq, 0, sizeof(fetchreq));
575 	fetchreq.fetch_all_branches = fetch_all_branches;
576 	fetchreq.list_refs_only = list_refs_only;
577 	fetchreq.verbosity = verbosity;
578 	TAILQ_FOREACH(pe, have_refs, entry)
579 		fetchreq.n_have_refs++;
580 	TAILQ_FOREACH(pe, wanted_branches, entry)
581 		fetchreq.n_wanted_branches++;
582 	TAILQ_FOREACH(pe, wanted_refs, entry)
583 		fetchreq.n_wanted_refs++;
584 	len = sizeof(struct got_imsg_fetch_request);
585 	if (len >= MAX_IMSGSIZE - IMSG_HEADER_SIZE) {
586 		close(fd);
587 		return got_error(GOT_ERR_NO_SPACE);
588 	}
589 
590 	if (imsg_compose(ibuf, GOT_IMSG_FETCH_REQUEST, 0, 0, fd,
591 	    &fetchreq, sizeof(fetchreq)) == -1)
592 		return got_error_from_errno(
593 		    "imsg_compose FETCH_SERVER_PROGRESS");
594 
595 	err = flush_imsg(ibuf);
596 	if (err) {
597 		close(fd);
598 		return err;
599 	}
600 	fd = -1;
601 
602 	TAILQ_FOREACH(pe, have_refs, entry) {
603 		const char *name = pe->path;
604 		size_t name_len = pe->path_len;
605 		struct got_object_id *id = pe->data;
606 
607 		len = sizeof(struct got_imsg_fetch_have_ref) + name_len;
608 		wbuf = imsg_create(ibuf, GOT_IMSG_FETCH_HAVE_REF, 0, 0, len);
609 		if (wbuf == NULL)
610 			return got_error_from_errno("imsg_create FETCH_HAVE_REF");
611 
612 		/* Keep in sync with struct got_imsg_fetch_have_ref! */
613 		if (imsg_add(wbuf, id->sha1, sizeof(id->sha1)) == -1) {
614 			err = got_error_from_errno("imsg_add FETCH_HAVE_REF");
615 			ibuf_free(wbuf);
616 			return err;
617 		}
618 		if (imsg_add(wbuf, &name_len, sizeof(name_len)) == -1) {
619 			err = got_error_from_errno("imsg_add FETCH_HAVE_REF");
620 			ibuf_free(wbuf);
621 			return err;
622 		}
623 		if (imsg_add(wbuf, name, name_len) == -1) {
624 			err = got_error_from_errno("imsg_add FETCH_HAVE_REF");
625 			ibuf_free(wbuf);
626 			return err;
627 		}
628 
629 		wbuf->fd = -1;
630 		imsg_close(ibuf, wbuf);
631 		err = flush_imsg(ibuf);
632 		if (err)
633 			return err;
634 	}
635 
636 	TAILQ_FOREACH(pe, wanted_branches, entry) {
637 		const char *name = pe->path;
638 		size_t name_len = pe->path_len;
639 
640 		len = sizeof(struct got_imsg_fetch_wanted_branch) + name_len;
641 		wbuf = imsg_create(ibuf, GOT_IMSG_FETCH_WANTED_BRANCH, 0, 0,
642 		    len);
643 		if (wbuf == NULL)
644 			return got_error_from_errno(
645 			    "imsg_create FETCH_WANTED_BRANCH");
646 
647 		/* Keep in sync with struct got_imsg_fetch_wanted_branch! */
648 		if (imsg_add(wbuf, &name_len, sizeof(name_len)) == -1) {
649 			err = got_error_from_errno(
650 			    "imsg_add FETCH_WANTED_BRANCH");
651 			ibuf_free(wbuf);
652 			return err;
653 		}
654 		if (imsg_add(wbuf, name, name_len) == -1) {
655 			err = got_error_from_errno(
656 			    "imsg_add FETCH_WANTED_BRANCH");
657 			ibuf_free(wbuf);
658 			return err;
659 		}
660 
661 		wbuf->fd = -1;
662 		imsg_close(ibuf, wbuf);
663 		err = flush_imsg(ibuf);
664 		if (err)
665 			return err;
666 	}
667 
668 	TAILQ_FOREACH(pe, wanted_refs, entry) {
669 		const char *name = pe->path;
670 		size_t name_len = pe->path_len;
671 
672 		len = sizeof(struct got_imsg_fetch_wanted_ref) + name_len;
673 		wbuf = imsg_create(ibuf, GOT_IMSG_FETCH_WANTED_REF, 0, 0,
674 		    len);
675 		if (wbuf == NULL)
676 			return got_error_from_errno(
677 			    "imsg_create FETCH_WANTED_REF");
678 
679 		/* Keep in sync with struct got_imsg_fetch_wanted_ref! */
680 		if (imsg_add(wbuf, &name_len, sizeof(name_len)) == -1) {
681 			err = got_error_from_errno(
682 			    "imsg_add FETCH_WANTED_REF");
683 			ibuf_free(wbuf);
684 			return err;
685 		}
686 		if (imsg_add(wbuf, name, name_len) == -1) {
687 			err = got_error_from_errno(
688 			    "imsg_add FETCH_WANTED_REF");
689 			ibuf_free(wbuf);
690 			return err;
691 		}
692 
693 		wbuf->fd = -1;
694 		imsg_close(ibuf, wbuf);
695 		err = flush_imsg(ibuf);
696 		if (err)
697 			return err;
698 	}
699 
700 
701 	return NULL;
702 
703 }
704 
705 const struct got_error *
706 got_privsep_send_fetch_outfd(struct imsgbuf *ibuf, int fd)
707 {
708 	return send_fd(ibuf, GOT_IMSG_FETCH_OUTFD, fd);
709 }
710 
711 const struct got_error *
712 got_privsep_recv_fetch_progress(int *done, struct got_object_id **id,
713     char **refname, struct got_pathlist_head *symrefs, char **server_progress,
714     off_t *packfile_size, uint8_t *pack_sha1, struct imsgbuf *ibuf)
715 {
716 	const struct got_error *err = NULL;
717 	struct imsg imsg;
718 	size_t datalen;
719 	struct got_imsg_fetch_symrefs *isymrefs = NULL;
720 	size_t n, remain;
721 	off_t off;
722 	int i;
723 
724 	*done = 0;
725 	*id = NULL;
726 	*refname = NULL;
727 	*server_progress = NULL;
728 	*packfile_size = 0;
729 	memset(pack_sha1, 0, SHA1_DIGEST_LENGTH);
730 
731 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
732 	if (err)
733 		return err;
734 
735 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
736 	switch (imsg.hdr.type) {
737 	case GOT_IMSG_ERROR:
738 		if (datalen < sizeof(struct got_imsg_error)) {
739 			err = got_error(GOT_ERR_PRIVSEP_LEN);
740 			break;
741 		}
742 		err = recv_imsg_error(&imsg, datalen);
743 		break;
744 	case GOT_IMSG_FETCH_SYMREFS:
745 		if (datalen < sizeof(*isymrefs)) {
746 			err = got_error(GOT_ERR_PRIVSEP_LEN);
747 			break;
748 		}
749 		if (isymrefs != NULL) {
750 			err = got_error(GOT_ERR_PRIVSEP_MSG);
751 			break;
752 		}
753 		isymrefs = (struct got_imsg_fetch_symrefs *)imsg.data;
754 		off = sizeof(*isymrefs);
755 		remain = datalen - off;
756 		for (n = 0; n < isymrefs->nsymrefs; n++) {
757 			struct got_imsg_fetch_symref *s;
758 			char *name, *target;
759 			if (remain < sizeof(struct got_imsg_fetch_symref)) {
760 				err = got_error(GOT_ERR_PRIVSEP_LEN);
761 				goto done;
762 			}
763 			s = (struct got_imsg_fetch_symref *)(imsg.data + off);
764 			off += sizeof(*s);
765 			remain -= sizeof(*s);
766 			if (remain < s->name_len) {
767 				err = got_error(GOT_ERR_PRIVSEP_LEN);
768 				goto done;
769 			}
770 			name = strndup(imsg.data + off, s->name_len);
771 			if (name == NULL) {
772 				err = got_error_from_errno("strndup");
773 				goto done;
774 			}
775 			off += s->name_len;
776 			remain -= s->name_len;
777 			if (remain < s->target_len) {
778 				err = got_error(GOT_ERR_PRIVSEP_LEN);
779 				free(name);
780 				goto done;
781 			}
782 			target = strndup(imsg.data + off, s->target_len);
783 			if (target == NULL) {
784 				err = got_error_from_errno("strndup");
785 				free(name);
786 				goto done;
787 			}
788 			off += s->target_len;
789 			remain -= s->target_len;
790 			err = got_pathlist_append(symrefs, name, target);
791 			if (err) {
792 				free(name);
793 				free(target);
794 				goto done;
795 			}
796 		}
797 		break;
798 	case GOT_IMSG_FETCH_REF:
799 		if (datalen <= SHA1_DIGEST_LENGTH) {
800 			err = got_error(GOT_ERR_PRIVSEP_MSG);
801 			break;
802 		}
803 		*id = malloc(sizeof(**id));
804 		if (*id == NULL) {
805 			err = got_error_from_errno("malloc");
806 			break;
807 		}
808 		memcpy((*id)->sha1, imsg.data, SHA1_DIGEST_LENGTH);
809 		*refname = strndup(imsg.data + SHA1_DIGEST_LENGTH,
810 		    datalen - SHA1_DIGEST_LENGTH);
811 		if (*refname == NULL) {
812 			err = got_error_from_errno("strndup");
813 			break;
814 		}
815 		break;
816 	case GOT_IMSG_FETCH_SERVER_PROGRESS:
817 		if (datalen == 0) {
818 			err = got_error(GOT_ERR_PRIVSEP_LEN);
819 			break;
820 		}
821 		*server_progress = strndup(imsg.data, datalen);
822 		if (*server_progress == NULL) {
823 			err = got_error_from_errno("strndup");
824 			break;
825 		}
826 		for (i = 0; i < datalen; i++) {
827 			if (!isprint((unsigned char)(*server_progress)[i]) &&
828 			    !isspace((unsigned char)(*server_progress)[i])) {
829 				err = got_error(GOT_ERR_PRIVSEP_MSG);
830 				free(*server_progress);
831 				*server_progress = NULL;
832 				goto done;
833 			}
834 		}
835 		break;
836 	case GOT_IMSG_FETCH_DOWNLOAD_PROGRESS:
837 		if (datalen < sizeof(*packfile_size)) {
838 			err = got_error(GOT_ERR_PRIVSEP_MSG);
839 			break;
840 		}
841 		memcpy(packfile_size, imsg.data, sizeof(*packfile_size));
842 		break;
843 	case GOT_IMSG_FETCH_DONE:
844 		if (datalen != SHA1_DIGEST_LENGTH) {
845 			err = got_error(GOT_ERR_PRIVSEP_MSG);
846 			break;
847 		}
848 		memcpy(pack_sha1, imsg.data, SHA1_DIGEST_LENGTH);
849 		*done = 1;
850 		break;
851 	default:
852 		err = got_error(GOT_ERR_PRIVSEP_MSG);
853 		break;
854 	}
855 done:
856 	if (err) {
857 		free(*id);
858 		*id = NULL;
859 		free(*refname);
860 		*refname = NULL;
861 	}
862 	imsg_free(&imsg);
863 	return err;
864 }
865 
866 static const struct got_error *
867 send_send_ref(const char *name, size_t name_len, struct got_object_id *id,
868     int delete, struct imsgbuf *ibuf)
869 {
870 	const struct got_error *err = NULL;
871 	size_t len;
872 	struct ibuf *wbuf;
873 
874 	len = sizeof(struct got_imsg_send_ref) + name_len;
875 	wbuf = imsg_create(ibuf, GOT_IMSG_SEND_REF, 0, 0, len);
876 	if (wbuf == NULL)
877 		return got_error_from_errno("imsg_create SEND_REF");
878 
879 	/* Keep in sync with struct got_imsg_send_ref! */
880 	if (imsg_add(wbuf, id->sha1, sizeof(id->sha1)) == -1) {
881 		err = got_error_from_errno("imsg_add SEND_REF");
882 		ibuf_free(wbuf);
883 		return err;
884 	}
885 	if (imsg_add(wbuf, &delete, sizeof(delete)) == -1) {
886 		err = got_error_from_errno("imsg_add SEND_REF");
887 		ibuf_free(wbuf);
888 		return err;
889 	}
890 	if (imsg_add(wbuf, &name_len, sizeof(name_len)) == -1) {
891 		err = got_error_from_errno("imsg_add SEND_REF");
892 		ibuf_free(wbuf);
893 		return err;
894 	}
895 	if (imsg_add(wbuf, name, name_len) == -1) {
896 		err = got_error_from_errno("imsg_add SEND_REF");
897 		ibuf_free(wbuf);
898 		return err;
899 	}
900 
901 	wbuf->fd = -1;
902 	imsg_close(ibuf, wbuf);
903 	return flush_imsg(ibuf);
904 }
905 
906 const struct got_error *
907 got_privsep_send_send_req(struct imsgbuf *ibuf, int fd,
908     struct got_pathlist_head *have_refs,
909     struct got_pathlist_head *delete_refs,
910     int verbosity)
911 {
912 	const struct got_error *err = NULL;
913 	struct got_pathlist_entry *pe;
914 	struct got_imsg_send_request sendreq;
915 	struct got_object_id zero_id;
916 
917 	memset(&zero_id, 0, sizeof(zero_id));
918 	memset(&sendreq, 0, sizeof(sendreq));
919 	sendreq.verbosity = verbosity;
920 	TAILQ_FOREACH(pe, have_refs, entry)
921 		sendreq.nrefs++;
922 	TAILQ_FOREACH(pe, delete_refs, entry)
923 		sendreq.nrefs++;
924 	if (imsg_compose(ibuf, GOT_IMSG_SEND_REQUEST, 0, 0, fd,
925 	    &sendreq, sizeof(sendreq)) == -1) {
926 		err = got_error_from_errno(
927 		    "imsg_compose FETCH_SERVER_PROGRESS");
928 		goto done;
929 	}
930 
931 	err = flush_imsg(ibuf);
932 	if (err)
933 		goto done;
934 	fd = -1;
935 
936 	TAILQ_FOREACH(pe, have_refs, entry) {
937 		const char *name = pe->path;
938 		size_t name_len = pe->path_len;
939 		struct got_object_id *id = pe->data;
940 		err = send_send_ref(name, name_len, id, 0, ibuf);
941 		if (err)
942 			goto done;
943 	}
944 
945 	TAILQ_FOREACH(pe, delete_refs, entry) {
946 		const char *name = pe->path;
947 		size_t name_len = pe->path_len;
948 		err = send_send_ref(name, name_len, &zero_id, 1, ibuf);
949 		if (err)
950 			goto done;
951 	}
952 done:
953 	if (fd != -1 && close(fd) == -1 && err == NULL)
954 		err = got_error_from_errno("close");
955 	return err;
956 
957 }
958 
959 const struct got_error *
960 got_privsep_recv_send_remote_refs(struct got_pathlist_head *remote_refs,
961     struct imsgbuf *ibuf)
962 {
963 	const struct got_error *err = NULL;
964 	struct imsg imsg;
965 	size_t datalen;
966 	int done = 0;
967 	struct got_imsg_send_remote_ref iremote_ref;
968 	struct got_object_id *id = NULL;
969 	char *refname = NULL;
970 	struct got_pathlist_entry *new;
971 
972 	while (!done) {
973 		err = got_privsep_recv_imsg(&imsg, ibuf, 0);
974 		if (err)
975 			return err;
976 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
977 		switch (imsg.hdr.type) {
978 		case GOT_IMSG_ERROR:
979 			if (datalen < sizeof(struct got_imsg_error)) {
980 				err = got_error(GOT_ERR_PRIVSEP_LEN);
981 				goto done;
982 			}
983 			err = recv_imsg_error(&imsg, datalen);
984 			goto done;
985 		case GOT_IMSG_SEND_REMOTE_REF:
986 			if (datalen < sizeof(iremote_ref)) {
987 				err = got_error(GOT_ERR_PRIVSEP_MSG);
988 				goto done;
989 			}
990 			memcpy(&iremote_ref, imsg.data, sizeof(iremote_ref));
991 			if (datalen != sizeof(iremote_ref) +
992 			    iremote_ref.name_len) {
993 				err = got_error(GOT_ERR_PRIVSEP_MSG);
994 				goto done;
995 			}
996 			id = malloc(sizeof(*id));
997 			if (id == NULL) {
998 				err = got_error_from_errno("malloc");
999 				goto done;
1000 			}
1001 			memcpy(id->sha1, iremote_ref.id, SHA1_DIGEST_LENGTH);
1002 			refname = strndup(imsg.data + sizeof(iremote_ref),
1003 			    datalen - sizeof(iremote_ref));
1004 			if (refname == NULL) {
1005 				err = got_error_from_errno("strndup");
1006 				goto done;
1007 			}
1008 			err = got_pathlist_insert(&new, remote_refs,
1009 			    refname, id);
1010 			if (err)
1011 				goto done;
1012 			if (new == NULL) { /* duplicate which wasn't inserted */
1013 				free(id);
1014 				free(refname);
1015 			}
1016 			id = NULL;
1017 			refname = NULL;
1018 			break;
1019 		case GOT_IMSG_SEND_PACK_REQUEST:
1020 			if (datalen != 0) {
1021 				err = got_error(GOT_ERR_PRIVSEP_MSG);
1022 				goto done;
1023 			}
1024 			/* got-send-pack is now waiting for a pack file. */
1025 			done = 1;
1026 			break;
1027 		default:
1028 			err = got_error(GOT_ERR_PRIVSEP_MSG);
1029 			break;
1030 		}
1031 	}
1032 done:
1033 	free(id);
1034 	free(refname);
1035 	imsg_free(&imsg);
1036 	return err;
1037 }
1038 
1039 const struct got_error *
1040 got_privsep_send_packfd(struct imsgbuf *ibuf, int fd)
1041 {
1042 	return send_fd(ibuf, GOT_IMSG_SEND_PACKFD, fd);
1043 }
1044 
1045 const struct got_error *
1046 got_privsep_recv_send_progress(int *done, off_t *bytes_sent,
1047     int *success, char **refname, struct imsgbuf *ibuf)
1048 {
1049 	const struct got_error *err = NULL;
1050 	struct imsg imsg;
1051 	size_t datalen;
1052 	struct got_imsg_send_ref_status iref_status;
1053 
1054 	/* Do not reset the current value of 'bytes_sent', it accumulates. */
1055 	*done = 0;
1056 	*success = 0;
1057 	*refname = NULL;
1058 
1059 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
1060 	if (err)
1061 		return err;
1062 
1063 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1064 	switch (imsg.hdr.type) {
1065 	case GOT_IMSG_ERROR:
1066 		if (datalen < sizeof(struct got_imsg_error)) {
1067 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1068 			break;
1069 		}
1070 		err = recv_imsg_error(&imsg, datalen);
1071 		break;
1072 	case GOT_IMSG_SEND_UPLOAD_PROGRESS:
1073 		if (datalen < sizeof(*bytes_sent)) {
1074 			err = got_error(GOT_ERR_PRIVSEP_MSG);
1075 			break;
1076 		}
1077 		memcpy(bytes_sent, imsg.data, sizeof(*bytes_sent));
1078 		break;
1079 	case GOT_IMSG_SEND_REF_STATUS:
1080 		if (datalen < sizeof(iref_status)) {
1081 			err = got_error(GOT_ERR_PRIVSEP_MSG);
1082 			break;
1083 		}
1084 		memcpy(&iref_status, imsg.data, sizeof(iref_status));
1085 		if (datalen != sizeof(iref_status) + iref_status.name_len) {
1086 			err = got_error(GOT_ERR_PRIVSEP_MSG);
1087 			break;
1088 		}
1089 		*success = iref_status.success;
1090 		*refname = strndup(imsg.data + sizeof(iref_status),
1091 		    iref_status.name_len);
1092 		break;
1093 	case GOT_IMSG_SEND_DONE:
1094 		if (datalen != 0) {
1095 			err = got_error(GOT_ERR_PRIVSEP_MSG);
1096 			break;
1097 		}
1098 		*done = 1;
1099 		break;
1100 	default:
1101 		err = got_error(GOT_ERR_PRIVSEP_MSG);
1102 		break;
1103 	}
1104 
1105 	imsg_free(&imsg);
1106 	return err;
1107 }
1108 
1109 const struct got_error *
1110 got_privsep_send_index_pack_req(struct imsgbuf *ibuf, uint8_t *pack_sha1,
1111     int fd)
1112 {
1113 	const struct got_error *err = NULL;
1114 
1115 	/* Keep in sync with struct got_imsg_index_pack_request */
1116 	if (imsg_compose(ibuf, GOT_IMSG_IDXPACK_REQUEST, 0, 0, fd,
1117 	    pack_sha1, SHA1_DIGEST_LENGTH) == -1) {
1118 		err = got_error_from_errno("imsg_compose INDEX_REQUEST");
1119 		close(fd);
1120 		return err;
1121 	}
1122 	return flush_imsg(ibuf);
1123 }
1124 
1125 const struct got_error *
1126 got_privsep_send_index_pack_outfd(struct imsgbuf *ibuf, int fd)
1127 {
1128 	return send_fd(ibuf, GOT_IMSG_IDXPACK_OUTFD, fd);
1129 }
1130 
1131 const struct got_error *
1132 got_privsep_recv_index_progress(int *done, int *nobj_total,
1133     int *nobj_indexed, int *nobj_loose, int *nobj_resolved,
1134     struct imsgbuf *ibuf)
1135 {
1136 	const struct got_error *err = NULL;
1137 	struct imsg imsg;
1138 	struct got_imsg_index_pack_progress *iprogress;
1139 	size_t datalen;
1140 
1141 	*done = 0;
1142 	*nobj_total = 0;
1143 	*nobj_indexed = 0;
1144 	*nobj_resolved = 0;
1145 
1146 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
1147 	if (err)
1148 		return err;
1149 
1150 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1151 	switch (imsg.hdr.type) {
1152 	case GOT_IMSG_ERROR:
1153 		if (datalen < sizeof(struct got_imsg_error)) {
1154 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1155 			break;
1156 		}
1157 		err = recv_imsg_error(&imsg, datalen);
1158 		break;
1159 	case GOT_IMSG_IDXPACK_PROGRESS:
1160 		if (datalen < sizeof(*iprogress)) {
1161 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1162 			break;
1163 		}
1164 		iprogress = (struct got_imsg_index_pack_progress *)imsg.data;
1165 		*nobj_total = iprogress->nobj_total;
1166 		*nobj_indexed = iprogress->nobj_indexed;
1167 		*nobj_loose = iprogress->nobj_loose;
1168 		*nobj_resolved = iprogress->nobj_resolved;
1169 		break;
1170 	case GOT_IMSG_IDXPACK_DONE:
1171 		if (datalen != 0) {
1172 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1173 			break;
1174 		}
1175 		*done = 1;
1176 		break;
1177 	default:
1178 		err = got_error(GOT_ERR_PRIVSEP_MSG);
1179 		break;
1180 	}
1181 
1182 	imsg_free(&imsg);
1183 	return err;
1184 }
1185 
1186 const struct got_error *
1187 got_privsep_get_imsg_obj(struct got_object **obj, struct imsg *imsg,
1188     struct imsgbuf *ibuf)
1189 {
1190 	struct got_imsg_object *iobj;
1191 	size_t datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
1192 
1193 	if (datalen != sizeof(*iobj))
1194 		return got_error(GOT_ERR_PRIVSEP_LEN);
1195 	iobj = imsg->data;
1196 
1197 	*obj = calloc(1, sizeof(**obj));
1198 	if (*obj == NULL)
1199 		return got_error_from_errno("calloc");
1200 
1201 	memcpy((*obj)->id.sha1, iobj->id, SHA1_DIGEST_LENGTH);
1202 	(*obj)->type = iobj->type;
1203 	(*obj)->flags = iobj->flags;
1204 	(*obj)->hdrlen = iobj->hdrlen;
1205 	(*obj)->size = iobj->size;
1206 	/* path_packfile is handled by caller */
1207 	if (iobj->flags & GOT_OBJ_FLAG_PACKED) {
1208 		(*obj)->pack_offset = iobj->pack_offset;
1209 		(*obj)->pack_idx = iobj->pack_idx;
1210 	}
1211 	STAILQ_INIT(&(*obj)->deltas.entries);
1212 	return NULL;
1213 }
1214 
1215 const struct got_error *
1216 got_privsep_recv_obj(struct got_object **obj, struct imsgbuf *ibuf)
1217 {
1218 	const struct got_error *err = NULL;
1219 	struct imsg imsg;
1220 	const size_t min_datalen =
1221 	    MIN(sizeof(struct got_imsg_error), sizeof(struct got_imsg_object));
1222 
1223 	*obj = NULL;
1224 
1225 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
1226 	if (err)
1227 		return err;
1228 
1229 	switch (imsg.hdr.type) {
1230 	case GOT_IMSG_OBJECT:
1231 		err = got_privsep_get_imsg_obj(obj, &imsg, ibuf);
1232 		break;
1233 	default:
1234 		err = got_error(GOT_ERR_PRIVSEP_MSG);
1235 		break;
1236 	}
1237 
1238 	imsg_free(&imsg);
1239 
1240 	return err;
1241 }
1242 
1243 static const struct got_error *
1244 send_commit_logmsg(struct imsgbuf *ibuf, struct got_commit_object *commit,
1245     size_t logmsg_len)
1246 {
1247 	const struct got_error *err = NULL;
1248 	size_t offset, remain;
1249 
1250 	offset = 0;
1251 	remain = logmsg_len;
1252 	while (remain > 0) {
1253 		size_t n = MIN(MAX_IMSGSIZE - IMSG_HEADER_SIZE, remain);
1254 
1255 		if (imsg_compose(ibuf, GOT_IMSG_COMMIT_LOGMSG, 0, 0, -1,
1256 		    commit->logmsg + offset, n) == -1) {
1257 			err = got_error_from_errno("imsg_compose "
1258 			    "COMMIT_LOGMSG");
1259 			break;
1260 		}
1261 
1262 		err = flush_imsg(ibuf);
1263 		if (err)
1264 			break;
1265 
1266 		offset += n;
1267 		remain -= n;
1268 	}
1269 
1270 	return err;
1271 }
1272 
1273 const struct got_error *
1274 got_privsep_send_commit(struct imsgbuf *ibuf, struct got_commit_object *commit)
1275 {
1276 	const struct got_error *err = NULL;
1277 	struct got_imsg_commit_object *icommit;
1278 	uint8_t *buf;
1279 	size_t len, total;
1280 	struct got_object_qid *qid;
1281 	size_t author_len = strlen(commit->author);
1282 	size_t committer_len = strlen(commit->committer);
1283 	size_t logmsg_len = strlen(commit->logmsg);
1284 
1285 	total = sizeof(*icommit) + author_len + committer_len +
1286 	    commit->nparents * SHA1_DIGEST_LENGTH;
1287 
1288 	buf = malloc(total);
1289 	if (buf == NULL)
1290 		return got_error_from_errno("malloc");
1291 
1292 	icommit = (struct got_imsg_commit_object *)buf;
1293 	memcpy(icommit->tree_id, commit->tree_id->sha1,
1294 	    sizeof(icommit->tree_id));
1295 	icommit->author_len = author_len;
1296 	icommit->author_time = commit->author_time;
1297 	icommit->author_gmtoff = commit->author_gmtoff;
1298 	icommit->committer_len = committer_len;
1299 	icommit->committer_time = commit->committer_time;
1300 	icommit->committer_gmtoff = commit->committer_gmtoff;
1301 	icommit->logmsg_len = logmsg_len;
1302 	icommit->nparents = commit->nparents;
1303 
1304 	len = sizeof(*icommit);
1305 	memcpy(buf + len, commit->author, author_len);
1306 	len += author_len;
1307 	memcpy(buf + len, commit->committer, committer_len);
1308 	len += committer_len;
1309 	STAILQ_FOREACH(qid, &commit->parent_ids, entry) {
1310 		memcpy(buf + len, &qid->id, SHA1_DIGEST_LENGTH);
1311 		len += SHA1_DIGEST_LENGTH;
1312 	}
1313 
1314 	if (imsg_compose(ibuf, GOT_IMSG_COMMIT, 0, 0, -1, buf, len) == -1) {
1315 		err = got_error_from_errno("imsg_compose COMMIT");
1316 		goto done;
1317 	}
1318 
1319 	if (logmsg_len == 0 ||
1320 	    logmsg_len + len > MAX_IMSGSIZE - IMSG_HEADER_SIZE) {
1321 		err = flush_imsg(ibuf);
1322 		if (err)
1323 			goto done;
1324 	}
1325 	err = send_commit_logmsg(ibuf, commit, logmsg_len);
1326 done:
1327 	free(buf);
1328 	return err;
1329 }
1330 
1331 static const struct got_error *
1332 get_commit_from_imsg(struct got_commit_object **commit,
1333     struct imsg *imsg, size_t datalen, struct imsgbuf *ibuf)
1334 {
1335 	const struct got_error *err = NULL;
1336 	struct got_imsg_commit_object *icommit;
1337 	size_t len = 0;
1338 	int i;
1339 
1340 	if (datalen < sizeof(*icommit))
1341 		return got_error(GOT_ERR_PRIVSEP_LEN);
1342 
1343 	icommit = imsg->data;
1344 	if (datalen != sizeof(*icommit) + icommit->author_len +
1345 	    icommit->committer_len +
1346 	    icommit->nparents * SHA1_DIGEST_LENGTH)
1347 		return got_error(GOT_ERR_PRIVSEP_LEN);
1348 
1349 	if (icommit->nparents < 0)
1350 		return got_error(GOT_ERR_PRIVSEP_LEN);
1351 
1352 	len += sizeof(*icommit);
1353 
1354 	*commit = got_object_commit_alloc_partial();
1355 	if (*commit == NULL)
1356 		return got_error_from_errno(
1357 		    "got_object_commit_alloc_partial");
1358 
1359 	memcpy((*commit)->tree_id->sha1, icommit->tree_id,
1360 	    SHA1_DIGEST_LENGTH);
1361 	(*commit)->author_time = icommit->author_time;
1362 	(*commit)->author_gmtoff = icommit->author_gmtoff;
1363 	(*commit)->committer_time = icommit->committer_time;
1364 	(*commit)->committer_gmtoff = icommit->committer_gmtoff;
1365 
1366 	if (icommit->author_len == 0) {
1367 		(*commit)->author = strdup("");
1368 		if ((*commit)->author == NULL) {
1369 			err = got_error_from_errno("strdup");
1370 			goto done;
1371 		}
1372 	} else {
1373 		(*commit)->author = malloc(icommit->author_len + 1);
1374 		if ((*commit)->author == NULL) {
1375 			err = got_error_from_errno("malloc");
1376 			goto done;
1377 		}
1378 		memcpy((*commit)->author, imsg->data + len,
1379 		    icommit->author_len);
1380 		(*commit)->author[icommit->author_len] = '\0';
1381 	}
1382 	len += icommit->author_len;
1383 
1384 	if (icommit->committer_len == 0) {
1385 		(*commit)->committer = strdup("");
1386 		if ((*commit)->committer == NULL) {
1387 			err = got_error_from_errno("strdup");
1388 			goto done;
1389 		}
1390 	} else {
1391 		(*commit)->committer =
1392 		    malloc(icommit->committer_len + 1);
1393 		if ((*commit)->committer == NULL) {
1394 			err = got_error_from_errno("malloc");
1395 			goto done;
1396 		}
1397 		memcpy((*commit)->committer, imsg->data + len,
1398 		    icommit->committer_len);
1399 		(*commit)->committer[icommit->committer_len] = '\0';
1400 	}
1401 	len += icommit->committer_len;
1402 
1403 	if (icommit->logmsg_len == 0) {
1404 		(*commit)->logmsg = strdup("");
1405 		if ((*commit)->logmsg == NULL) {
1406 			err = got_error_from_errno("strdup");
1407 			goto done;
1408 		}
1409 	} else {
1410 		size_t offset = 0, remain = icommit->logmsg_len;
1411 
1412 		(*commit)->logmsg = malloc(icommit->logmsg_len + 1);
1413 		if ((*commit)->logmsg == NULL) {
1414 			err = got_error_from_errno("malloc");
1415 			goto done;
1416 		}
1417 		while (remain > 0) {
1418 			struct imsg imsg_log;
1419 			size_t n = MIN(MAX_IMSGSIZE - IMSG_HEADER_SIZE,
1420 			    remain);
1421 
1422 			err = got_privsep_recv_imsg(&imsg_log, ibuf, n);
1423 			if (err)
1424 				goto done;
1425 
1426 			if (imsg_log.hdr.type != GOT_IMSG_COMMIT_LOGMSG) {
1427 				err = got_error(GOT_ERR_PRIVSEP_MSG);
1428 				goto done;
1429 			}
1430 
1431 			memcpy((*commit)->logmsg + offset,
1432 			    imsg_log.data, n);
1433 			imsg_free(&imsg_log);
1434 			offset += n;
1435 			remain -= n;
1436 		}
1437 		(*commit)->logmsg[icommit->logmsg_len] = '\0';
1438 	}
1439 
1440 	for (i = 0; i < icommit->nparents; i++) {
1441 		struct got_object_qid *qid;
1442 
1443 		err = got_object_qid_alloc_partial(&qid);
1444 		if (err)
1445 			break;
1446 		memcpy(&qid->id, imsg->data + len +
1447 		    i * SHA1_DIGEST_LENGTH, sizeof(qid->id));
1448 		STAILQ_INSERT_TAIL(&(*commit)->parent_ids, qid, entry);
1449 		(*commit)->nparents++;
1450 	}
1451 done:
1452 	if (err) {
1453 		got_object_commit_close(*commit);
1454 		*commit = NULL;
1455 	}
1456 	return err;
1457 }
1458 
1459 const struct got_error *
1460 got_privsep_recv_commit(struct got_commit_object **commit, struct imsgbuf *ibuf)
1461 {
1462 	const struct got_error *err = NULL;
1463 	struct imsg imsg;
1464 	size_t datalen;
1465 	const size_t min_datalen =
1466 	    MIN(sizeof(struct got_imsg_error),
1467 	    sizeof(struct got_imsg_commit_object));
1468 
1469 	*commit = NULL;
1470 
1471 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
1472 	if (err)
1473 		return err;
1474 
1475 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1476 
1477 	switch (imsg.hdr.type) {
1478 	case GOT_IMSG_COMMIT:
1479 		err = get_commit_from_imsg(commit, &imsg, datalen, ibuf);
1480 		break;
1481 	default:
1482 		err = got_error(GOT_ERR_PRIVSEP_MSG);
1483 		break;
1484 	}
1485 
1486 	imsg_free(&imsg);
1487 
1488 	return err;
1489 }
1490 
1491 const struct got_error *
1492 got_privsep_send_tree(struct imsgbuf *ibuf,
1493     struct got_parsed_tree_entry *entries, int nentries)
1494 {
1495 	const struct got_error *err = NULL;
1496 	struct got_imsg_tree_object itree;
1497 	size_t totlen;
1498 	int i, nimsg; /* number of imsg queued in ibuf */
1499 
1500 	itree.nentries = nentries;
1501 	if (imsg_compose(ibuf, GOT_IMSG_TREE, 0, 0, -1, &itree, sizeof(itree))
1502 	    == -1)
1503 		return got_error_from_errno("imsg_compose TREE");
1504 
1505 	totlen = sizeof(itree);
1506 	nimsg = 1;
1507 	for (i = 0; i < nentries; i++) {
1508 		struct got_parsed_tree_entry *pte = &entries[i];
1509 		struct ibuf *wbuf;
1510 		size_t len = sizeof(struct got_imsg_tree_entry) + pte->namelen;
1511 
1512 		if (len > MAX_IMSGSIZE)
1513 			return got_error(GOT_ERR_NO_SPACE);
1514 
1515 		nimsg++;
1516 		if (totlen + len >= MAX_IMSGSIZE - (IMSG_HEADER_SIZE * nimsg)) {
1517 			err = flush_imsg(ibuf);
1518 			if (err)
1519 				return err;
1520 			nimsg = 0;
1521 		}
1522 
1523 		wbuf = imsg_create(ibuf, GOT_IMSG_TREE_ENTRY, 0, 0, len);
1524 		if (wbuf == NULL)
1525 			return got_error_from_errno("imsg_create TREE_ENTRY");
1526 
1527 		/* Keep in sync with struct got_imsg_tree_object definition! */
1528 		if (imsg_add(wbuf, pte->id, SHA1_DIGEST_LENGTH) == -1) {
1529 			err = got_error_from_errno("imsg_add TREE_ENTRY");
1530 			ibuf_free(wbuf);
1531 			return err;
1532 		}
1533 		if (imsg_add(wbuf, &pte->mode, sizeof(pte->mode)) == -1) {
1534 			err = got_error_from_errno("imsg_add TREE_ENTRY");
1535 			ibuf_free(wbuf);
1536 			return err;
1537 		}
1538 
1539 		if (imsg_add(wbuf, pte->name, pte->namelen) == -1) {
1540 			err = got_error_from_errno("imsg_add TREE_ENTRY");
1541 			ibuf_free(wbuf);
1542 			return err;
1543 		}
1544 
1545 		wbuf->fd = -1;
1546 		imsg_close(ibuf, wbuf);
1547 
1548 		totlen += len;
1549 	}
1550 
1551 	return flush_imsg(ibuf);
1552 }
1553 
1554 const struct got_error *
1555 got_privsep_recv_tree(struct got_tree_object **tree, struct imsgbuf *ibuf)
1556 {
1557 	const struct got_error *err = NULL;
1558 	const size_t min_datalen =
1559 	    MIN(sizeof(struct got_imsg_error),
1560 	    sizeof(struct got_imsg_tree_object));
1561 	struct got_imsg_tree_object *itree;
1562 	int nentries = 0;
1563 
1564 	*tree = NULL;
1565 
1566 	err = read_imsg(ibuf);
1567 	if (err)
1568 		goto done;
1569 
1570 	for (;;) {
1571 		struct imsg imsg;
1572 		size_t n;
1573 		size_t datalen;
1574 		struct got_imsg_tree_entry *ite;
1575 		struct got_tree_entry *te = NULL;
1576 
1577 		n = imsg_get(ibuf, &imsg);
1578 		if (n == 0) {
1579 			if ((*tree)) {
1580 				if (nentries < (*tree)->nentries) {
1581 					err = read_imsg(ibuf);
1582 					if (err)
1583 						break;
1584 					continue;
1585 				} else
1586 					break;
1587 			} else {
1588 				err = got_error(GOT_ERR_PRIVSEP_MSG);
1589 				break;
1590 			}
1591 		}
1592 
1593 		if (imsg.hdr.len < IMSG_HEADER_SIZE + min_datalen) {
1594 			imsg_free(&imsg);
1595 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1596 			break;
1597 		}
1598 
1599 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1600 
1601 		switch (imsg.hdr.type) {
1602 		case GOT_IMSG_ERROR:
1603 			err = recv_imsg_error(&imsg, datalen);
1604 			break;
1605 		case GOT_IMSG_TREE:
1606 			/* This message should only appear once. */
1607 			if (*tree != NULL) {
1608 				err = got_error(GOT_ERR_PRIVSEP_MSG);
1609 				break;
1610 			}
1611 			if (datalen != sizeof(*itree)) {
1612 				err = got_error(GOT_ERR_PRIVSEP_LEN);
1613 				break;
1614 			}
1615 			itree = imsg.data;
1616 			if (itree->nentries < 0) {
1617 				err = got_error(GOT_ERR_PRIVSEP_LEN);
1618 				break;
1619 			}
1620 			*tree = malloc(sizeof(**tree));
1621 			if (*tree == NULL) {
1622 				err = got_error_from_errno("malloc");
1623 				break;
1624 			}
1625 			(*tree)->entries = calloc(itree->nentries,
1626 			    sizeof(struct got_tree_entry));
1627 			if ((*tree)->entries == NULL) {
1628 				err = got_error_from_errno("malloc");
1629 				free(*tree);
1630 				*tree = NULL;
1631 				break;
1632 			}
1633 			(*tree)->nentries = itree->nentries;
1634 			(*tree)->refcnt = 0;
1635 			break;
1636 		case GOT_IMSG_TREE_ENTRY:
1637 			/* This message should be preceeded by GOT_IMSG_TREE. */
1638 			if (*tree == NULL) {
1639 				err = got_error(GOT_ERR_PRIVSEP_MSG);
1640 				break;
1641 			}
1642 			if (datalen < sizeof(*ite) || datalen > MAX_IMSGSIZE) {
1643 				err = got_error(GOT_ERR_PRIVSEP_LEN);
1644 				break;
1645 			}
1646 
1647 			/* Remaining data contains the entry's name. */
1648 			datalen -= sizeof(*ite);
1649 			if (datalen == 0 || datalen > MAX_IMSGSIZE) {
1650 				err = got_error(GOT_ERR_PRIVSEP_LEN);
1651 				break;
1652 			}
1653 			ite = imsg.data;
1654 
1655 			if (datalen + 1 > sizeof(te->name)) {
1656 				err = got_error(GOT_ERR_NO_SPACE);
1657 				break;
1658 			}
1659 			if (nentries >= (*tree)->nentries) {
1660 				err = got_error(GOT_ERR_PRIVSEP_LEN);
1661 				break;
1662 			}
1663 			te = &(*tree)->entries[nentries];
1664 			memcpy(te->name, imsg.data + sizeof(*ite), datalen);
1665 			te->name[datalen] = '\0';
1666 
1667 			memcpy(te->id.sha1, ite->id, SHA1_DIGEST_LENGTH);
1668 			te->mode = ite->mode;
1669 			te->idx = nentries;
1670 			nentries++;
1671 			break;
1672 		default:
1673 			err = got_error(GOT_ERR_PRIVSEP_MSG);
1674 			break;
1675 		}
1676 
1677 		imsg_free(&imsg);
1678 		if (err)
1679 			break;
1680 	}
1681 done:
1682 	if (*tree && (*tree)->nentries != nentries) {
1683 		if (err == NULL)
1684 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1685 		got_object_tree_close(*tree);
1686 		*tree = NULL;
1687 	}
1688 
1689 	return err;
1690 }
1691 
1692 const struct got_error *
1693 got_privsep_send_blob(struct imsgbuf *ibuf, size_t size, size_t hdrlen,
1694     const uint8_t *data)
1695 {
1696 	struct got_imsg_blob iblob;
1697 
1698 	iblob.size = size;
1699 	iblob.hdrlen = hdrlen;
1700 
1701 	if (data) {
1702 		uint8_t *buf;
1703 
1704 		if (size > GOT_PRIVSEP_INLINE_BLOB_DATA_MAX)
1705 			return got_error(GOT_ERR_NO_SPACE);
1706 
1707 		buf = malloc(sizeof(iblob) + size);
1708 		if (buf == NULL)
1709 			return got_error_from_errno("malloc");
1710 
1711 		memcpy(buf, &iblob, sizeof(iblob));
1712 		memcpy(buf + sizeof(iblob), data, size);
1713 		if (imsg_compose(ibuf, GOT_IMSG_BLOB, 0, 0, -1, buf,
1714 		    sizeof(iblob) + size) == -1) {
1715 			free(buf);
1716 			return got_error_from_errno("imsg_compose BLOB");
1717 		}
1718 		free(buf);
1719 	} else {
1720 		/* Data has already been written to file descriptor. */
1721 		if (imsg_compose(ibuf, GOT_IMSG_BLOB, 0, 0, -1, &iblob,
1722 		    sizeof(iblob)) == -1)
1723 			return got_error_from_errno("imsg_compose BLOB");
1724 	}
1725 
1726 
1727 	return flush_imsg(ibuf);
1728 }
1729 
1730 const struct got_error *
1731 got_privsep_recv_blob(uint8_t **outbuf, size_t *size, size_t *hdrlen,
1732     struct imsgbuf *ibuf)
1733 {
1734 	const struct got_error *err = NULL;
1735 	struct imsg imsg;
1736 	struct got_imsg_blob *iblob;
1737 	size_t datalen;
1738 
1739 	*outbuf = NULL;
1740 
1741 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
1742 	if (err)
1743 		return err;
1744 
1745 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1746 
1747 	switch (imsg.hdr.type) {
1748 	case GOT_IMSG_BLOB:
1749 		if (datalen < sizeof(*iblob)) {
1750 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1751 			break;
1752 		}
1753 		iblob = imsg.data;
1754 		*size = iblob->size;
1755 		*hdrlen = iblob->hdrlen;
1756 
1757 		if (datalen == sizeof(*iblob)) {
1758 			/* Data has been written to file descriptor. */
1759 			break;
1760 		}
1761 
1762 		if (*size > GOT_PRIVSEP_INLINE_BLOB_DATA_MAX) {
1763 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1764 			break;
1765 		}
1766 
1767 		*outbuf = malloc(*size);
1768 		if (*outbuf == NULL) {
1769 			err = got_error_from_errno("malloc");
1770 			break;
1771 		}
1772 		memcpy(*outbuf, imsg.data + sizeof(*iblob), *size);
1773 		break;
1774 	default:
1775 		err = got_error(GOT_ERR_PRIVSEP_MSG);
1776 		break;
1777 	}
1778 
1779 	imsg_free(&imsg);
1780 
1781 	return err;
1782 }
1783 
1784 static const struct got_error *
1785 send_tagmsg(struct imsgbuf *ibuf, struct got_tag_object *tag, size_t tagmsg_len)
1786 {
1787 	const struct got_error *err = NULL;
1788 	size_t offset, remain;
1789 
1790 	offset = 0;
1791 	remain = tagmsg_len;
1792 	while (remain > 0) {
1793 		size_t n = MIN(MAX_IMSGSIZE - IMSG_HEADER_SIZE, remain);
1794 
1795 		if (imsg_compose(ibuf, GOT_IMSG_TAG_TAGMSG, 0, 0, -1,
1796 		    tag->tagmsg + offset, n) == -1) {
1797 			err = got_error_from_errno("imsg_compose TAG_TAGMSG");
1798 			break;
1799 		}
1800 
1801 		err = flush_imsg(ibuf);
1802 		if (err)
1803 			break;
1804 
1805 		offset += n;
1806 		remain -= n;
1807 	}
1808 
1809 	return err;
1810 }
1811 
1812 const struct got_error *
1813 got_privsep_send_tag(struct imsgbuf *ibuf, struct got_tag_object *tag)
1814 {
1815 	const struct got_error *err = NULL;
1816 	struct got_imsg_tag_object *itag;
1817 	uint8_t *buf;
1818 	size_t len, total;
1819 	size_t tag_len = strlen(tag->tag);
1820 	size_t tagger_len = strlen(tag->tagger);
1821 	size_t tagmsg_len = strlen(tag->tagmsg);
1822 
1823 	total = sizeof(*itag) + tag_len + tagger_len + tagmsg_len;
1824 
1825 	buf = malloc(total);
1826 	if (buf == NULL)
1827 		return got_error_from_errno("malloc");
1828 
1829 	itag = (struct got_imsg_tag_object *)buf;
1830 	memcpy(itag->id, tag->id.sha1, sizeof(itag->id));
1831 	itag->obj_type = tag->obj_type;
1832 	itag->tag_len = tag_len;
1833 	itag->tagger_len = tagger_len;
1834 	itag->tagger_time = tag->tagger_time;
1835 	itag->tagger_gmtoff = tag->tagger_gmtoff;
1836 	itag->tagmsg_len = tagmsg_len;
1837 
1838 	len = sizeof(*itag);
1839 	memcpy(buf + len, tag->tag, tag_len);
1840 	len += tag_len;
1841 	memcpy(buf + len, tag->tagger, tagger_len);
1842 	len += tagger_len;
1843 
1844 	if (imsg_compose(ibuf, GOT_IMSG_TAG, 0, 0, -1, buf, len) == -1) {
1845 		err = got_error_from_errno("imsg_compose TAG");
1846 		goto done;
1847 	}
1848 
1849 	if (tagmsg_len == 0 ||
1850 	    tagmsg_len + len > MAX_IMSGSIZE - IMSG_HEADER_SIZE) {
1851 		err = flush_imsg(ibuf);
1852 		if (err)
1853 			goto done;
1854 	}
1855 	err = send_tagmsg(ibuf, tag, tagmsg_len);
1856 done:
1857 	free(buf);
1858 	return err;
1859 }
1860 
1861 const struct got_error *
1862 got_privsep_recv_tag(struct got_tag_object **tag, struct imsgbuf *ibuf)
1863 {
1864 	const struct got_error *err = NULL;
1865 	struct imsg imsg;
1866 	struct got_imsg_tag_object *itag;
1867 	size_t len, datalen;
1868 	const size_t min_datalen =
1869 	    MIN(sizeof(struct got_imsg_error),
1870 	    sizeof(struct got_imsg_tag_object));
1871 
1872 	*tag = NULL;
1873 
1874 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
1875 	if (err)
1876 		return err;
1877 
1878 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1879 	len = 0;
1880 
1881 	switch (imsg.hdr.type) {
1882 	case GOT_IMSG_TAG:
1883 		if (datalen < sizeof(*itag)) {
1884 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1885 			break;
1886 		}
1887 		itag = imsg.data;
1888 		if (datalen != sizeof(*itag) + itag->tag_len +
1889 		    itag->tagger_len) {
1890 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1891 			break;
1892 		}
1893 		len += sizeof(*itag);
1894 
1895 		*tag = calloc(1, sizeof(**tag));
1896 		if (*tag == NULL) {
1897 			err = got_error_from_errno("calloc");
1898 			break;
1899 		}
1900 
1901 		memcpy((*tag)->id.sha1, itag->id, SHA1_DIGEST_LENGTH);
1902 
1903 		if (itag->tag_len == 0) {
1904 			(*tag)->tag = strdup("");
1905 			if ((*tag)->tag == NULL) {
1906 				err = got_error_from_errno("strdup");
1907 				break;
1908 			}
1909 		} else {
1910 			(*tag)->tag = malloc(itag->tag_len + 1);
1911 			if ((*tag)->tag == NULL) {
1912 				err = got_error_from_errno("malloc");
1913 				break;
1914 			}
1915 			memcpy((*tag)->tag, imsg.data + len,
1916 			    itag->tag_len);
1917 			(*tag)->tag[itag->tag_len] = '\0';
1918 		}
1919 		len += itag->tag_len;
1920 
1921 		(*tag)->obj_type = itag->obj_type;
1922 		(*tag)->tagger_time = itag->tagger_time;
1923 		(*tag)->tagger_gmtoff = itag->tagger_gmtoff;
1924 
1925 		if (itag->tagger_len == 0) {
1926 			(*tag)->tagger = strdup("");
1927 			if ((*tag)->tagger == NULL) {
1928 				err = got_error_from_errno("strdup");
1929 				break;
1930 			}
1931 		} else {
1932 			(*tag)->tagger = malloc(itag->tagger_len + 1);
1933 			if ((*tag)->tagger == NULL) {
1934 				err = got_error_from_errno("malloc");
1935 				break;
1936 			}
1937 			memcpy((*tag)->tagger, imsg.data + len,
1938 			    itag->tagger_len);
1939 			(*tag)->tagger[itag->tagger_len] = '\0';
1940 		}
1941 		len += itag->tagger_len;
1942 
1943 		if (itag->tagmsg_len == 0) {
1944 			(*tag)->tagmsg = strdup("");
1945 			if ((*tag)->tagmsg == NULL) {
1946 				err = got_error_from_errno("strdup");
1947 				break;
1948 			}
1949 		} else {
1950 			size_t offset = 0, remain = itag->tagmsg_len;
1951 
1952 			(*tag)->tagmsg = malloc(itag->tagmsg_len + 1);
1953 			if ((*tag)->tagmsg == NULL) {
1954 				err = got_error_from_errno("malloc");
1955 				break;
1956 			}
1957 			while (remain > 0) {
1958 				struct imsg imsg_log;
1959 				size_t n = MIN(MAX_IMSGSIZE - IMSG_HEADER_SIZE,
1960 				    remain);
1961 
1962 				err = got_privsep_recv_imsg(&imsg_log, ibuf, n);
1963 				if (err)
1964 					return err;
1965 
1966 				if (imsg_log.hdr.type != GOT_IMSG_TAG_TAGMSG)
1967 					return got_error(GOT_ERR_PRIVSEP_MSG);
1968 
1969 				memcpy((*tag)->tagmsg + offset, imsg_log.data,
1970 				    n);
1971 				imsg_free(&imsg_log);
1972 				offset += n;
1973 				remain -= n;
1974 			}
1975 			(*tag)->tagmsg[itag->tagmsg_len] = '\0';
1976 		}
1977 
1978 		break;
1979 	default:
1980 		err = got_error(GOT_ERR_PRIVSEP_MSG);
1981 		break;
1982 	}
1983 
1984 	imsg_free(&imsg);
1985 
1986 	return err;
1987 }
1988 
1989 const struct got_error *
1990 got_privsep_init_pack_child(struct imsgbuf *ibuf, struct got_pack *pack,
1991     struct got_packidx *packidx)
1992 {
1993 	const struct got_error *err = NULL;
1994 	struct got_imsg_packidx ipackidx;
1995 	struct got_imsg_pack ipack;
1996 	int fd;
1997 
1998 	ipackidx.len = packidx->len;
1999 	ipackidx.packfile_size = pack->filesize;
2000 	fd = dup(packidx->fd);
2001 	if (fd == -1)
2002 		return got_error_from_errno("dup");
2003 
2004 	if (imsg_compose(ibuf, GOT_IMSG_PACKIDX, 0, 0, fd, &ipackidx,
2005 	    sizeof(ipackidx)) == -1) {
2006 		err = got_error_from_errno("imsg_compose PACKIDX");
2007 		close(fd);
2008 		return err;
2009 	}
2010 
2011 	if (strlcpy(ipack.path_packfile, pack->path_packfile,
2012 	    sizeof(ipack.path_packfile)) >= sizeof(ipack.path_packfile))
2013 		return got_error(GOT_ERR_NO_SPACE);
2014 	ipack.filesize = pack->filesize;
2015 
2016 	fd = dup(pack->fd);
2017 	if (fd == -1)
2018 		return got_error_from_errno("dup");
2019 
2020 	if (imsg_compose(ibuf, GOT_IMSG_PACK, 0, 0, fd, &ipack, sizeof(ipack))
2021 	    == -1) {
2022 		err = got_error_from_errno("imsg_compose PACK");
2023 		close(fd);
2024 		return err;
2025 	}
2026 
2027 	return flush_imsg(ibuf);
2028 }
2029 
2030 const struct got_error *
2031 got_privsep_send_packed_obj_req(struct imsgbuf *ibuf, int idx,
2032     struct got_object_id *id)
2033 {
2034 	struct got_imsg_packed_object iobj;
2035 
2036 	iobj.idx = idx;
2037 	memcpy(iobj.id, id->sha1, sizeof(iobj.id));
2038 
2039 	if (imsg_compose(ibuf, GOT_IMSG_PACKED_OBJECT_REQUEST, 0, 0, -1,
2040 	    &iobj, sizeof(iobj)) == -1)
2041 		return got_error_from_errno("imsg_compose "
2042 		    "PACKED_OBJECT_REQUEST");
2043 
2044 	return flush_imsg(ibuf);
2045 }
2046 
2047 const struct got_error *
2048 got_privsep_send_packed_raw_obj_req(struct imsgbuf *ibuf, int idx,
2049     struct got_object_id *id)
2050 {
2051 	struct got_imsg_packed_object iobj;
2052 
2053 	iobj.idx = idx;
2054 	memcpy(iobj.id, id->sha1, sizeof(iobj.id));
2055 
2056 	if (imsg_compose(ibuf, GOT_IMSG_PACKED_RAW_OBJECT_REQUEST, 0, 0, -1,
2057 	    &iobj, sizeof(iobj)) == -1)
2058 		return got_error_from_errno("imsg_compose "
2059 		    "PACKED_OBJECT_REQUEST");
2060 
2061 	return flush_imsg(ibuf);
2062 }
2063 
2064 const struct got_error *
2065 got_privsep_send_gitconfig_parse_req(struct imsgbuf *ibuf, int fd)
2066 {
2067 	const struct got_error *err = NULL;
2068 
2069 	if (imsg_compose(ibuf, GOT_IMSG_GITCONFIG_PARSE_REQUEST, 0, 0, fd,
2070 	    NULL, 0) == -1) {
2071 		err = got_error_from_errno("imsg_compose "
2072 		    "GITCONFIG_PARSE_REQUEST");
2073 		close(fd);
2074 		return err;
2075 	}
2076 
2077 	return flush_imsg(ibuf);
2078 }
2079 
2080 const struct got_error *
2081 got_privsep_send_gitconfig_repository_format_version_req(struct imsgbuf *ibuf)
2082 {
2083 	if (imsg_compose(ibuf,
2084 	    GOT_IMSG_GITCONFIG_REPOSITORY_FORMAT_VERSION_REQUEST, 0, 0, -1,
2085 	    NULL, 0) == -1)
2086 		return got_error_from_errno("imsg_compose "
2087 		    "GITCONFIG_REPOSITORY_FORMAT_VERSION_REQUEST");
2088 
2089 	return flush_imsg(ibuf);
2090 }
2091 
2092 const struct got_error *
2093 got_privsep_send_gitconfig_repository_extensions_req(struct imsgbuf *ibuf)
2094 {
2095 	if (imsg_compose(ibuf,
2096 	    GOT_IMSG_GITCONFIG_REPOSITORY_EXTENSIONS_REQUEST, 0, 0, -1,
2097 	    NULL, 0) == -1)
2098 		return got_error_from_errno("imsg_compose "
2099 		    "GITCONFIG_REPOSITORY_EXTENSIONS_REQUEST");
2100 
2101 	return flush_imsg(ibuf);
2102 }
2103 
2104 
2105 const struct got_error *
2106 got_privsep_send_gitconfig_author_name_req(struct imsgbuf *ibuf)
2107 {
2108 	if (imsg_compose(ibuf,
2109 	    GOT_IMSG_GITCONFIG_AUTHOR_NAME_REQUEST, 0, 0, -1, NULL, 0) == -1)
2110 		return got_error_from_errno("imsg_compose "
2111 		    "GITCONFIG_AUTHOR_NAME_REQUEST");
2112 
2113 	return flush_imsg(ibuf);
2114 }
2115 
2116 const struct got_error *
2117 got_privsep_send_gitconfig_author_email_req(struct imsgbuf *ibuf)
2118 {
2119 	if (imsg_compose(ibuf,
2120 	    GOT_IMSG_GITCONFIG_AUTHOR_EMAIL_REQUEST, 0, 0, -1, NULL, 0) == -1)
2121 		return got_error_from_errno("imsg_compose "
2122 		    "GITCONFIG_AUTHOR_EMAIL_REQUEST");
2123 
2124 	return flush_imsg(ibuf);
2125 }
2126 
2127 const struct got_error *
2128 got_privsep_send_gitconfig_remotes_req(struct imsgbuf *ibuf)
2129 {
2130 	if (imsg_compose(ibuf,
2131 	    GOT_IMSG_GITCONFIG_REMOTES_REQUEST, 0, 0, -1, NULL, 0) == -1)
2132 		return got_error_from_errno("imsg_compose "
2133 		    "GITCONFIG_REMOTE_REQUEST");
2134 
2135 	return flush_imsg(ibuf);
2136 }
2137 
2138 const struct got_error *
2139 got_privsep_send_gitconfig_owner_req(struct imsgbuf *ibuf)
2140 {
2141 	if (imsg_compose(ibuf,
2142 	    GOT_IMSG_GITCONFIG_OWNER_REQUEST, 0, 0, -1, NULL, 0) == -1)
2143 		return got_error_from_errno("imsg_compose "
2144 		    "GITCONFIG_OWNER_REQUEST");
2145 
2146 	return flush_imsg(ibuf);
2147 }
2148 
2149 const struct got_error *
2150 got_privsep_recv_gitconfig_str(char **str, struct imsgbuf *ibuf)
2151 {
2152 	const struct got_error *err = NULL;
2153 	struct imsg imsg;
2154 	size_t datalen;
2155 	const size_t min_datalen = 0;
2156 
2157 	*str = NULL;
2158 
2159 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2160 	if (err)
2161 		return err;
2162 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2163 
2164 	switch (imsg.hdr.type) {
2165 	case GOT_IMSG_GITCONFIG_STR_VAL:
2166 		if (datalen == 0)
2167 			break;
2168 		/* datalen does not include terminating \0 */
2169 		*str = malloc(datalen + 1);
2170 		if (*str == NULL) {
2171 			err = got_error_from_errno("malloc");
2172 			break;
2173 		}
2174 		memcpy(*str, imsg.data, datalen);
2175 		(*str)[datalen] = '\0';
2176 		break;
2177 	default:
2178 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2179 		break;
2180 	}
2181 
2182 	imsg_free(&imsg);
2183 	return err;
2184 }
2185 
2186 const struct got_error *
2187 got_privsep_recv_gitconfig_int(int *val, struct imsgbuf *ibuf)
2188 {
2189 	const struct got_error *err = NULL;
2190 	struct imsg imsg;
2191 	size_t datalen;
2192 	const size_t min_datalen =
2193 	    MIN(sizeof(struct got_imsg_error), sizeof(int));
2194 
2195 	*val = 0;
2196 
2197 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2198 	if (err)
2199 		return err;
2200 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2201 
2202 	switch (imsg.hdr.type) {
2203 	case GOT_IMSG_GITCONFIG_INT_VAL:
2204 		if (datalen != sizeof(*val)) {
2205 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2206 			break;
2207 		}
2208 		memcpy(val, imsg.data, sizeof(*val));
2209 		break;
2210 	default:
2211 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2212 		break;
2213 	}
2214 
2215 	imsg_free(&imsg);
2216 	return err;
2217 }
2218 
2219 static void
2220 free_remote_data(struct got_remote_repo *remote)
2221 {
2222 	int i;
2223 
2224 	free(remote->name);
2225 	free(remote->fetch_url);
2226 	free(remote->send_url);
2227 	for (i = 0; i < remote->nfetch_branches; i++)
2228 		free(remote->fetch_branches[i]);
2229 	free(remote->fetch_branches);
2230 	for (i = 0; i < remote->nsend_branches; i++)
2231 		free(remote->send_branches[i]);
2232 	free(remote->send_branches);
2233 	for (i = 0; i < remote->nfetch_refs; i++)
2234 		free(remote->fetch_refs[i]);
2235 	free(remote->fetch_refs);
2236 }
2237 
2238 const struct got_error *
2239 got_privsep_recv_gitconfig_remotes(struct got_remote_repo **remotes,
2240     int *nremotes, struct imsgbuf *ibuf)
2241 {
2242 	const struct got_error *err = NULL;
2243 	struct imsg imsg;
2244 	size_t datalen;
2245 	struct got_imsg_remotes iremotes;
2246 	struct got_imsg_remote iremote;
2247 
2248 	*remotes = NULL;
2249 	*nremotes = 0;
2250 	iremotes.nremotes = 0;
2251 
2252 	err = got_privsep_recv_imsg(&imsg, ibuf, sizeof(iremotes));
2253 	if (err)
2254 		return err;
2255 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2256 
2257 	switch (imsg.hdr.type) {
2258 	case GOT_IMSG_GITCONFIG_REMOTES:
2259 		if (datalen != sizeof(iremotes)) {
2260 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2261 			break;
2262 		}
2263 		memcpy(&iremotes, imsg.data, sizeof(iremotes));
2264 		if (iremotes.nremotes == 0) {
2265 			imsg_free(&imsg);
2266 			return NULL;
2267 		}
2268 		break;
2269 	default:
2270 		imsg_free(&imsg);
2271 		return got_error(GOT_ERR_PRIVSEP_MSG);
2272 	}
2273 
2274 	imsg_free(&imsg);
2275 
2276 	*remotes = recallocarray(NULL, 0, iremotes.nremotes, sizeof(**remotes));
2277 	if (*remotes == NULL)
2278 		return got_error_from_errno("recallocarray");
2279 
2280 	while (*nremotes < iremotes.nremotes) {
2281 		struct got_remote_repo *remote;
2282 
2283 		err = got_privsep_recv_imsg(&imsg, ibuf, sizeof(iremote));
2284 		if (err)
2285 			break;
2286 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2287 
2288 		switch (imsg.hdr.type) {
2289 		case GOT_IMSG_GITCONFIG_REMOTE:
2290 			remote = &(*remotes)[*nremotes];
2291 			memset(remote, 0, sizeof(*remote));
2292 			if (datalen < sizeof(iremote)) {
2293 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2294 				break;
2295 			}
2296 			memcpy(&iremote, imsg.data, sizeof(iremote));
2297 			if (iremote.name_len == 0 ||
2298 			    iremote.fetch_url_len == 0 ||
2299 			    iremote.send_url_len == 0 ||
2300 			    (sizeof(iremote) + iremote.name_len +
2301 			    iremote.fetch_url_len + iremote.send_url_len) > datalen) {
2302 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2303 				break;
2304 			}
2305 			remote->name = strndup(imsg.data + sizeof(iremote),
2306 			    iremote.name_len);
2307 			if (remote->name == NULL) {
2308 				err = got_error_from_errno("strndup");
2309 				break;
2310 			}
2311 			remote->fetch_url = strndup(imsg.data + sizeof(iremote) +
2312 			    iremote.name_len, iremote.fetch_url_len);
2313 			if (remote->fetch_url == NULL) {
2314 				err = got_error_from_errno("strndup");
2315 				free_remote_data(remote);
2316 				break;
2317 			}
2318 			remote->send_url = strndup(imsg.data + sizeof(iremote) +
2319 			    iremote.name_len + iremote.fetch_url_len,
2320 			    iremote.send_url_len);
2321 			if (remote->send_url == NULL) {
2322 				err = got_error_from_errno("strndup");
2323 				free_remote_data(remote);
2324 				break;
2325 			}
2326 			remote->mirror_references = iremote.mirror_references;
2327 			remote->fetch_all_branches = iremote.fetch_all_branches;
2328 			remote->nfetch_branches = 0;
2329 			remote->fetch_branches = NULL;
2330 			remote->nsend_branches = 0;
2331 			remote->send_branches = NULL;
2332 			remote->nfetch_refs = 0;
2333 			remote->fetch_refs = NULL;
2334 			(*nremotes)++;
2335 			break;
2336 		default:
2337 			err = got_error(GOT_ERR_PRIVSEP_MSG);
2338 			break;
2339 		}
2340 
2341 		imsg_free(&imsg);
2342 		if (err)
2343 			break;
2344 	}
2345 
2346 	if (err) {
2347 		int i;
2348 		for (i = 0; i < *nremotes; i++)
2349 			free_remote_data(&(*remotes)[i]);
2350 		free(*remotes);
2351 		*remotes = NULL;
2352 		*nremotes = 0;
2353 	}
2354 	return err;
2355 }
2356 
2357 const struct got_error *
2358 got_privsep_send_gotconfig_parse_req(struct imsgbuf *ibuf, int fd)
2359 {
2360 	const struct got_error *err = NULL;
2361 
2362 	if (imsg_compose(ibuf, GOT_IMSG_GOTCONFIG_PARSE_REQUEST, 0, 0, fd,
2363 	    NULL, 0) == -1) {
2364 		err = got_error_from_errno("imsg_compose "
2365 		    "GOTCONFIG_PARSE_REQUEST");
2366 		close(fd);
2367 		return err;
2368 	}
2369 
2370 	return flush_imsg(ibuf);
2371 }
2372 
2373 const struct got_error *
2374 got_privsep_send_gotconfig_author_req(struct imsgbuf *ibuf)
2375 {
2376 	if (imsg_compose(ibuf,
2377 	    GOT_IMSG_GOTCONFIG_AUTHOR_REQUEST, 0, 0, -1, NULL, 0) == -1)
2378 		return got_error_from_errno("imsg_compose "
2379 		    "GOTCONFIG_AUTHOR_REQUEST");
2380 
2381 	return flush_imsg(ibuf);
2382 }
2383 
2384 const struct got_error *
2385 got_privsep_send_gotconfig_remotes_req(struct imsgbuf *ibuf)
2386 {
2387 	if (imsg_compose(ibuf,
2388 	    GOT_IMSG_GOTCONFIG_REMOTES_REQUEST, 0, 0, -1, NULL, 0) == -1)
2389 		return got_error_from_errno("imsg_compose "
2390 		    "GOTCONFIG_REMOTE_REQUEST");
2391 
2392 	return flush_imsg(ibuf);
2393 }
2394 
2395 const struct got_error *
2396 got_privsep_recv_gotconfig_str(char **str, struct imsgbuf *ibuf)
2397 {
2398 	const struct got_error *err = NULL;
2399 	struct imsg imsg;
2400 	size_t datalen;
2401 	const size_t min_datalen = 0;
2402 
2403 	*str = NULL;
2404 
2405 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2406 	if (err)
2407 		return err;
2408 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2409 
2410 	switch (imsg.hdr.type) {
2411 	case GOT_IMSG_ERROR:
2412 		if (datalen < sizeof(struct got_imsg_error)) {
2413 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2414 			break;
2415 		}
2416 		err = recv_imsg_error(&imsg, datalen);
2417 		break;
2418 	case GOT_IMSG_GOTCONFIG_STR_VAL:
2419 		if (datalen == 0)
2420 			break;
2421 		/* datalen does not include terminating \0 */
2422 		*str = malloc(datalen + 1);
2423 		if (*str == NULL) {
2424 			err = got_error_from_errno("malloc");
2425 			break;
2426 		}
2427 		memcpy(*str, imsg.data, datalen);
2428 		(*str)[datalen] = '\0';
2429 		break;
2430 	default:
2431 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2432 		break;
2433 	}
2434 
2435 	imsg_free(&imsg);
2436 	return err;
2437 }
2438 
2439 const struct got_error *
2440 got_privsep_recv_gotconfig_remotes(struct got_remote_repo **remotes,
2441     int *nremotes, struct imsgbuf *ibuf)
2442 {
2443 	const struct got_error *err = NULL;
2444 	struct imsg imsg;
2445 	size_t datalen;
2446 	struct got_imsg_remotes iremotes;
2447 	struct got_imsg_remote iremote;
2448 	const size_t min_datalen =
2449 	    MIN(sizeof(struct got_imsg_error), sizeof(iremotes));
2450 
2451 	*remotes = NULL;
2452 	*nremotes = 0;
2453 	iremotes.nremotes = 0;
2454 
2455 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2456 	if (err)
2457 		return err;
2458 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2459 
2460 	switch (imsg.hdr.type) {
2461 	case GOT_IMSG_ERROR:
2462 		if (datalen < sizeof(struct got_imsg_error)) {
2463 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2464 			break;
2465 		}
2466 		err = recv_imsg_error(&imsg, datalen);
2467 		break;
2468 	case GOT_IMSG_GOTCONFIG_REMOTES:
2469 		if (datalen != sizeof(iremotes)) {
2470 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2471 			break;
2472 		}
2473 		memcpy(&iremotes, imsg.data, sizeof(iremotes));
2474 		if (iremotes.nremotes == 0) {
2475 			imsg_free(&imsg);
2476 			return NULL;
2477 		}
2478 		break;
2479 	default:
2480 		imsg_free(&imsg);
2481 		return got_error(GOT_ERR_PRIVSEP_MSG);
2482 	}
2483 
2484 	imsg_free(&imsg);
2485 
2486 	*remotes = recallocarray(NULL, 0, iremotes.nremotes, sizeof(**remotes));
2487 	if (*remotes == NULL)
2488 		return got_error_from_errno("recallocarray");
2489 
2490 	while (*nremotes < iremotes.nremotes) {
2491 		struct got_remote_repo *remote;
2492 		const size_t min_datalen =
2493 		    MIN(sizeof(struct got_imsg_error), sizeof(iremote));
2494 		int i;
2495 
2496 		err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2497 		if (err)
2498 			break;
2499 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2500 
2501 		switch (imsg.hdr.type) {
2502 		case GOT_IMSG_ERROR:
2503 			if (datalen < sizeof(struct got_imsg_error)) {
2504 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2505 				break;
2506 			}
2507 			err = recv_imsg_error(&imsg, datalen);
2508 			break;
2509 		case GOT_IMSG_GOTCONFIG_REMOTE:
2510 			remote = &(*remotes)[*nremotes];
2511 			memset(remote, 0, sizeof(*remote));
2512 			if (datalen < sizeof(iremote)) {
2513 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2514 				break;
2515 			}
2516 			memcpy(&iremote, imsg.data, sizeof(iremote));
2517 			if (iremote.name_len == 0 ||
2518 			    (iremote.fetch_url_len == 0 &&
2519 			    iremote.send_url_len == 0) ||
2520 			    (sizeof(iremote) + iremote.name_len +
2521 			    iremote.fetch_url_len + iremote.send_url_len) >
2522 			    datalen) {
2523 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2524 				break;
2525 			}
2526 			remote->name = strndup(imsg.data + sizeof(iremote),
2527 			    iremote.name_len);
2528 			if (remote->name == NULL) {
2529 				err = got_error_from_errno("strndup");
2530 				break;
2531 			}
2532 			remote->fetch_url = strndup(imsg.data +
2533 			    sizeof(iremote) + iremote.name_len,
2534 			    iremote.fetch_url_len);
2535 			if (remote->fetch_url == NULL) {
2536 				err = got_error_from_errno("strndup");
2537 				free_remote_data(remote);
2538 				break;
2539 			}
2540 			remote->send_url = strndup(imsg.data +
2541 			    sizeof(iremote) + iremote.name_len +
2542 			    iremote.fetch_url_len, iremote.send_url_len);
2543 			if (remote->send_url == NULL) {
2544 				err = got_error_from_errno("strndup");
2545 				free_remote_data(remote);
2546 				break;
2547 			}
2548 			remote->mirror_references = iremote.mirror_references;
2549 			remote->fetch_all_branches = iremote.fetch_all_branches;
2550 			if (iremote.nfetch_branches > 0) {
2551 				remote->fetch_branches = recallocarray(NULL, 0,
2552 				    iremote.nfetch_branches, sizeof(char *));
2553 				if (remote->fetch_branches == NULL) {
2554 					err = got_error_from_errno("calloc");
2555 					free_remote_data(remote);
2556 					break;
2557 				}
2558 			}
2559 			remote->nfetch_branches = 0;
2560 			for (i = 0; i < iremote.nfetch_branches; i++) {
2561 				char *branch;
2562 				err = got_privsep_recv_gotconfig_str(&branch,
2563 				    ibuf);
2564 				if (err) {
2565 					free_remote_data(remote);
2566 					goto done;
2567 				}
2568 				remote->fetch_branches[i] = branch;
2569 				remote->nfetch_branches++;
2570 			}
2571 			if (iremote.nsend_branches > 0) {
2572 				remote->send_branches = recallocarray(NULL, 0,
2573 				    iremote.nsend_branches, sizeof(char *));
2574 				if (remote->send_branches == NULL) {
2575 					err = got_error_from_errno("calloc");
2576 					free_remote_data(remote);
2577 					break;
2578 				}
2579 			}
2580 			remote->nsend_branches = 0;
2581 			for (i = 0; i < iremote.nsend_branches; i++) {
2582 				char *branch;
2583 				err = got_privsep_recv_gotconfig_str(&branch,
2584 				    ibuf);
2585 				if (err) {
2586 					free_remote_data(remote);
2587 					goto done;
2588 				}
2589 				remote->send_branches[i] = branch;
2590 				remote->nsend_branches++;
2591 			}
2592 			if (iremote.nfetch_refs > 0) {
2593 				remote->fetch_refs = recallocarray(NULL, 0,
2594 				    iremote.nfetch_refs, sizeof(char *));
2595 				if (remote->fetch_refs == NULL) {
2596 					err = got_error_from_errno("calloc");
2597 					free_remote_data(remote);
2598 					break;
2599 				}
2600 			}
2601 			remote->nfetch_refs = 0;
2602 			for (i = 0; i < iremote.nfetch_refs; i++) {
2603 				char *ref;
2604 				err = got_privsep_recv_gotconfig_str(&ref,
2605 				    ibuf);
2606 				if (err) {
2607 					free_remote_data(remote);
2608 					goto done;
2609 				}
2610 				remote->fetch_refs[i] = ref;
2611 				remote->nfetch_refs++;
2612 			}
2613 			(*nremotes)++;
2614 			break;
2615 		default:
2616 			err = got_error(GOT_ERR_PRIVSEP_MSG);
2617 			break;
2618 		}
2619 
2620 		imsg_free(&imsg);
2621 		if (err)
2622 			break;
2623 	}
2624 done:
2625 	if (err) {
2626 		int i;
2627 		for (i = 0; i < *nremotes; i++)
2628 			free_remote_data(&(*remotes)[i]);
2629 		free(*remotes);
2630 		*remotes = NULL;
2631 		*nremotes = 0;
2632 	}
2633 	return err;
2634 }
2635 
2636 const struct got_error *
2637 got_privsep_send_commit_traversal_request(struct imsgbuf *ibuf,
2638     struct got_object_id *id, int idx, const char *path)
2639 {
2640 	const struct got_error *err = NULL;
2641 	struct ibuf *wbuf;
2642 	size_t path_len = strlen(path) + 1;
2643 
2644 	wbuf = imsg_create(ibuf, GOT_IMSG_COMMIT_TRAVERSAL_REQUEST, 0, 0,
2645 	    sizeof(struct got_imsg_commit_traversal_request) + path_len);
2646 	if (wbuf == NULL)
2647 		return got_error_from_errno(
2648 		    "imsg_create COMMIT_TRAVERSAL_REQUEST");
2649 	if (imsg_add(wbuf, id->sha1, SHA1_DIGEST_LENGTH) == -1) {
2650 		err = got_error_from_errno("imsg_add COMMIT_TRAVERSAL_REQUEST");
2651 		ibuf_free(wbuf);
2652 		return err;
2653 	}
2654 	if (imsg_add(wbuf, &idx, sizeof(idx)) == -1) {
2655 		err = got_error_from_errno("imsg_add COMMIT_TRAVERSAL_REQUEST");
2656 		ibuf_free(wbuf);
2657 		return err;
2658 	}
2659 	if (imsg_add(wbuf, path, path_len) == -1) {
2660 		err = got_error_from_errno("imsg_add COMMIT_TRAVERSAL_REQUEST");
2661 		ibuf_free(wbuf);
2662 		return err;
2663 	}
2664 
2665 	wbuf->fd = -1;
2666 	imsg_close(ibuf, wbuf);
2667 
2668 	return flush_imsg(ibuf);
2669 }
2670 
2671 const struct got_error *
2672 got_privsep_recv_traversed_commits(struct got_commit_object **changed_commit,
2673     struct got_object_id **changed_commit_id,
2674     struct got_object_id_queue *commit_ids, struct imsgbuf *ibuf)
2675 {
2676 	const struct got_error *err = NULL;
2677 	struct imsg imsg;
2678 	struct got_imsg_traversed_commits *icommits;
2679 	size_t datalen;
2680 	int i, done = 0;
2681 
2682 	*changed_commit = NULL;
2683 	*changed_commit_id = NULL;
2684 
2685 	while (!done) {
2686 		err = got_privsep_recv_imsg(&imsg, ibuf, 0);
2687 		if (err)
2688 			return err;
2689 
2690 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2691 		switch (imsg.hdr.type) {
2692 		case GOT_IMSG_TRAVERSED_COMMITS:
2693 			icommits = imsg.data;
2694 			if (datalen != sizeof(*icommits) +
2695 			    icommits->ncommits * SHA1_DIGEST_LENGTH) {
2696 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2697 				break;
2698 			}
2699 			for (i = 0; i < icommits->ncommits; i++) {
2700 				struct got_object_qid *qid;
2701 				uint8_t *sha1 = (uint8_t *)imsg.data +
2702 				    sizeof(*icommits) + i * SHA1_DIGEST_LENGTH;
2703 				err = got_object_qid_alloc_partial(&qid);
2704 				if (err)
2705 					break;
2706 				memcpy(qid->id.sha1, sha1, SHA1_DIGEST_LENGTH);
2707 				STAILQ_INSERT_TAIL(commit_ids, qid, entry);
2708 
2709 				/* The last commit may contain a change. */
2710 				if (i == icommits->ncommits - 1) {
2711 					*changed_commit_id =
2712 					    got_object_id_dup(&qid->id);
2713 					if (*changed_commit_id == NULL) {
2714 						err = got_error_from_errno(
2715 						    "got_object_id_dup");
2716 						break;
2717 					}
2718 				}
2719 			}
2720 			break;
2721 		case GOT_IMSG_COMMIT:
2722 			if (*changed_commit_id == NULL) {
2723 				err = got_error(GOT_ERR_PRIVSEP_MSG);
2724 				break;
2725 			}
2726 			err = get_commit_from_imsg(changed_commit, &imsg,
2727 			    datalen, ibuf);
2728 			break;
2729 		case GOT_IMSG_COMMIT_TRAVERSAL_DONE:
2730 			done = 1;
2731 			break;
2732 		default:
2733 			err = got_error(GOT_ERR_PRIVSEP_MSG);
2734 			break;
2735 		}
2736 
2737 		imsg_free(&imsg);
2738 		if (err)
2739 			break;
2740 	}
2741 
2742 	if (err)
2743 		got_object_id_queue_free(commit_ids);
2744 	return err;
2745 }
2746 
2747 const struct got_error *
2748 got_privsep_send_raw_delta_req(struct imsgbuf *ibuf, int idx,
2749     struct got_object_id *id)
2750 {
2751 	struct got_imsg_raw_delta_request dreq;
2752 
2753 	dreq.idx = idx;
2754 	memcpy(dreq.id, id->sha1, SHA1_DIGEST_LENGTH);
2755 
2756 	if (imsg_compose(ibuf, GOT_IMSG_RAW_DELTA_REQUEST, 0, 0, -1,
2757 	    &dreq, sizeof(dreq)) == -1)
2758 		return got_error_from_errno("imsg_compose RAW_DELTA_REQUEST");
2759 
2760 	return flush_imsg(ibuf);
2761 }
2762 
2763 const struct got_error *
2764 got_privsep_send_raw_delta_outfd(struct imsgbuf *ibuf, int fd)
2765 {
2766 	return send_fd(ibuf, GOT_IMSG_RAW_DELTA_OUTFD, fd);
2767 }
2768 
2769 const struct got_error *
2770 got_privsep_send_raw_delta(struct imsgbuf *ibuf, uint64_t base_size,
2771     uint64_t result_size,  off_t delta_size, off_t delta_compressed_size,
2772     off_t delta_offset, off_t delta_out_offset, struct got_object_id *base_id)
2773 {
2774 	struct got_imsg_raw_delta idelta;
2775 	int ret;
2776 
2777 	idelta.base_size = base_size;
2778 	idelta.result_size = result_size;
2779 	idelta.delta_size = delta_size;
2780 	idelta.delta_compressed_size = delta_compressed_size;
2781 	idelta.delta_offset = delta_offset;
2782 	idelta.delta_out_offset = delta_out_offset;
2783 	memcpy(idelta.base_id, base_id->sha1, SHA1_DIGEST_LENGTH);
2784 
2785 	ret = imsg_compose(ibuf, GOT_IMSG_RAW_DELTA, 0, 0, -1,
2786 	    &idelta, sizeof(idelta));
2787 	if (ret == -1)
2788 		return got_error_from_errno("imsg_compose RAW_DELTA");
2789 
2790 	return flush_imsg(ibuf);
2791 }
2792 
2793 const struct got_error *
2794 got_privsep_recv_raw_delta(uint64_t *base_size, uint64_t *result_size,
2795     off_t *delta_size, off_t *delta_compressed_size, off_t *delta_offset,
2796     off_t *delta_out_offset, struct got_object_id **base_id, struct imsgbuf *ibuf)
2797 {
2798 	const struct got_error *err = NULL;
2799 	struct imsg imsg;
2800 	struct got_imsg_raw_delta *delta;
2801 	size_t datalen;
2802 
2803 	*base_size = 0;
2804 	*result_size = 0;
2805 	*delta_size = 0;
2806 	*delta_compressed_size = 0;
2807 	*delta_offset = 0;
2808 	*delta_out_offset = 0;
2809 	*base_id = NULL;
2810 
2811 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
2812 	if (err)
2813 		return err;
2814 
2815 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2816 
2817 	switch (imsg.hdr.type) {
2818 	case GOT_IMSG_RAW_DELTA:
2819 		if (datalen != sizeof(*delta)) {
2820 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2821 			break;
2822 		}
2823 		delta = imsg.data;
2824 		*base_size = delta->base_size;
2825 		*result_size = delta->result_size;
2826 		*delta_size = delta->delta_size;
2827 		*delta_compressed_size = delta->delta_compressed_size;
2828 		*delta_offset = delta->delta_offset;
2829 		*delta_out_offset = delta->delta_out_offset;
2830 		*base_id = calloc(1, sizeof(**base_id));
2831 		if (*base_id == NULL) {
2832 			err = got_error_from_errno("malloc");
2833 			break;
2834 		}
2835 		memcpy((*base_id)->sha1, delta->base_id, SHA1_DIGEST_LENGTH);
2836 		break;
2837 	default:
2838 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2839 		break;
2840 	}
2841 
2842 	imsg_free(&imsg);
2843 
2844 	if (err) {
2845 		free(*base_id);
2846 		*base_id = NULL;
2847 	}
2848 	return err;
2849 }
2850 
2851 const struct got_error *
2852 got_privsep_send_object_idlist(struct imsgbuf *ibuf,
2853     struct got_object_id **ids, size_t nids)
2854 {
2855 	const struct got_error *err = NULL;
2856 	struct got_imsg_object_idlist idlist;
2857 	struct ibuf *wbuf;
2858 	size_t i;
2859 
2860 	if (nids > GOT_IMSG_OBJ_ID_LIST_MAX_NIDS)
2861 		return got_error(GOT_ERR_NO_SPACE);
2862 
2863 	wbuf = imsg_create(ibuf, GOT_IMSG_OBJ_ID_LIST, 0, 0,
2864 	    sizeof(idlist) + nids * sizeof(**ids));
2865 	if (wbuf == NULL) {
2866 		err = got_error_from_errno("imsg_create OBJ_ID_LIST");
2867 		return err;
2868 	}
2869 
2870 	idlist.nids = nids;
2871 	if (imsg_add(wbuf, &idlist, sizeof(idlist)) == -1) {
2872 		err = got_error_from_errno("imsg_add OBJ_ID_LIST");
2873 		ibuf_free(wbuf);
2874 		return err;
2875 	}
2876 
2877 	for (i = 0; i < nids; i++) {
2878 		struct got_object_id *id = ids[i];
2879 		if (imsg_add(wbuf, id, sizeof(*id)) == -1) {
2880 			err = got_error_from_errno("imsg_add OBJ_ID_LIST");
2881 			ibuf_free(wbuf);
2882 			return err;
2883 		}
2884 	}
2885 	
2886 	wbuf->fd = -1;
2887 	imsg_close(ibuf, wbuf);
2888 
2889 	return flush_imsg(ibuf);
2890 }
2891 
2892 const struct got_error *
2893 got_privsep_send_object_idlist_done(struct imsgbuf *ibuf)
2894 {
2895 	if (imsg_compose(ibuf, GOT_IMSG_OBJ_ID_LIST_DONE, 0, 0, -1, NULL, 0)
2896 	    == -1)
2897 		return got_error_from_errno("imsg_compose OBJ_ID_LIST_DONE");
2898 
2899 	return flush_imsg(ibuf);
2900 }
2901 
2902 const struct got_error *
2903 got_privsep_recv_object_idlist(int *done, struct got_object_id **ids,
2904     size_t *nids, struct imsgbuf *ibuf)
2905 {
2906 	const struct got_error *err = NULL;
2907 	struct imsg imsg;
2908 	struct got_imsg_object_idlist *idlist;
2909 	size_t datalen;
2910 
2911 	*ids = NULL;
2912 	*done = 0;
2913 	*nids = 0;
2914 
2915 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
2916 	if (err)
2917 		return err;
2918 
2919 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2920 	switch (imsg.hdr.type) {
2921 	case GOT_IMSG_OBJ_ID_LIST:
2922 		if (datalen < sizeof(*idlist)) {
2923 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2924 			break;
2925 		}
2926 		idlist = imsg.data;
2927 		if (idlist->nids > GOT_IMSG_OBJ_ID_LIST_MAX_NIDS) {
2928 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2929 			break;
2930 		}
2931 		*nids = idlist->nids;
2932 		*ids = calloc(*nids, sizeof(**ids));
2933 		if (*ids == NULL) {
2934 			err = got_error_from_errno("calloc");
2935 			break;
2936 		}
2937 		memcpy(*ids, (uint8_t *)imsg.data + sizeof(idlist),
2938 		    *nids * sizeof(**ids));
2939 		break;
2940 	case GOT_IMSG_OBJ_ID_LIST_DONE:
2941 		*done = 1;
2942 		break;
2943 	default:
2944 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2945 		break;
2946 	}
2947 
2948 	imsg_free(&imsg);
2949 
2950 	return err;
2951 }
2952 
2953 const struct got_error *
2954 got_privsep_send_delta_reuse_req(struct imsgbuf *ibuf)
2955 {
2956 	if (imsg_compose(ibuf, GOT_IMSG_DELTA_REUSE_REQUEST, 0, 0, -1, NULL, 0)
2957 	    == -1)
2958 		return got_error_from_errno("imsg_compose DELTA_REUSE_REQUEST");
2959 
2960 	return flush_imsg(ibuf);
2961 }
2962 
2963 const struct got_error *
2964 got_privsep_send_reused_deltas(struct imsgbuf *ibuf,
2965     struct got_imsg_reused_delta *deltas, size_t ndeltas)
2966 {
2967 	const struct got_error *err = NULL;
2968 	struct ibuf *wbuf;
2969 	struct got_imsg_reused_deltas ideltas;
2970 	size_t i;
2971 
2972 	if (ndeltas > GOT_IMSG_REUSED_DELTAS_MAX_NDELTAS)
2973 		return got_error(GOT_ERR_NO_SPACE);
2974 
2975 	wbuf = imsg_create(ibuf, GOT_IMSG_REUSED_DELTAS, 0, 0,
2976 	    sizeof(ideltas) + ndeltas * sizeof(*deltas));
2977 	if (wbuf == NULL) {
2978 		err = got_error_from_errno("imsg_create REUSED_DELTAS");
2979 		return err;
2980 	}
2981 
2982 	ideltas.ndeltas = ndeltas;
2983 	if (imsg_add(wbuf, &ideltas, sizeof(ideltas)) == -1) {
2984 		err = got_error_from_errno("imsg_add REUSED_DELTAS");
2985 		ibuf_free(wbuf);
2986 		return err;
2987 	}
2988 
2989 	for (i = 0; i < ndeltas; i++) {
2990 		struct got_imsg_reused_delta *delta = &deltas[i];
2991 		if (imsg_add(wbuf, delta, sizeof(*delta)) == -1) {
2992 			err = got_error_from_errno("imsg_add REUSED_DELTAS");
2993 			ibuf_free(wbuf);
2994 			return err;
2995 		}
2996 	}
2997 	
2998 	wbuf->fd = -1;
2999 	imsg_close(ibuf, wbuf);
3000 
3001 	return flush_imsg(ibuf);
3002 }
3003 
3004 const struct got_error *
3005 got_privsep_send_reused_deltas_done(struct imsgbuf *ibuf)
3006 {
3007 	if (imsg_compose(ibuf, GOT_IMSG_DELTA_REUSE_DONE, 0, 0, -1, NULL, 0)
3008 	    == -1)
3009 		return got_error_from_errno("imsg_compose DELTA_REUSE_DONE");
3010 
3011 	return flush_imsg(ibuf);
3012 }
3013 
3014 const struct got_error *
3015 got_privsep_recv_reused_deltas(int *done, struct got_imsg_reused_delta *deltas,
3016     size_t *ndeltas, struct imsgbuf *ibuf)
3017 {
3018 	const struct got_error *err = NULL;
3019 	struct imsg imsg;
3020 	struct got_imsg_reused_deltas *ideltas;
3021 	size_t datalen;
3022 
3023 	*done = 0;
3024 	*ndeltas = 0;
3025 
3026 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
3027 	if (err)
3028 		return err;
3029 
3030 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
3031 	switch (imsg.hdr.type) {
3032 	case GOT_IMSG_REUSED_DELTAS:
3033 		if (datalen < sizeof(*ideltas)) {
3034 			err = got_error(GOT_ERR_PRIVSEP_LEN);
3035 			break;
3036 		}
3037 		ideltas = imsg.data;
3038 		if (ideltas->ndeltas > GOT_IMSG_OBJ_ID_LIST_MAX_NIDS) {
3039 			err = got_error(GOT_ERR_PRIVSEP_LEN);
3040 			break;
3041 		}
3042 		*ndeltas = ideltas->ndeltas;
3043 		memcpy(deltas, (uint8_t *)imsg.data + sizeof(ideltas),
3044 		    *ndeltas * sizeof(*deltas));
3045 		break;
3046 	case GOT_IMSG_DELTA_REUSE_DONE:
3047 		*done = 1;
3048 		break;
3049 	default:
3050 		err = got_error(GOT_ERR_PRIVSEP_MSG);
3051 		break;
3052 	}
3053 
3054 	imsg_free(&imsg);
3055 
3056 	return err;
3057 }
3058 
3059 const struct got_error *
3060 got_privsep_unveil_exec_helpers(void)
3061 {
3062 	const char *helpers[] = {
3063 	    GOT_PATH_PROG_READ_PACK,
3064 	    GOT_PATH_PROG_READ_OBJECT,
3065 	    GOT_PATH_PROG_READ_COMMIT,
3066 	    GOT_PATH_PROG_READ_TREE,
3067 	    GOT_PATH_PROG_READ_BLOB,
3068 	    GOT_PATH_PROG_READ_TAG,
3069 	    GOT_PATH_PROG_READ_GITCONFIG,
3070 	    GOT_PATH_PROG_READ_GOTCONFIG,
3071 	    GOT_PATH_PROG_READ_PATCH,
3072 	    GOT_PATH_PROG_FETCH_PACK,
3073 	    GOT_PATH_PROG_INDEX_PACK,
3074 	    GOT_PATH_PROG_SEND_PACK,
3075 	};
3076 	size_t i;
3077 
3078 	for (i = 0; i < nitems(helpers); i++) {
3079 		if (unveil(helpers[i], "x") == 0)
3080 			continue;
3081 		return got_error_from_errno2("unveil", helpers[i]);
3082 	}
3083 
3084 	return NULL;
3085 }
3086 
3087 void
3088 got_privsep_exec_child(int imsg_fds[2], const char *path, const char *repo_path)
3089 {
3090 	if (close(imsg_fds[0]) == -1) {
3091 		fprintf(stderr, "%s: %s\n", getprogname(), strerror(errno));
3092 		_exit(1);
3093 	}
3094 
3095 	if (dup2(imsg_fds[1], GOT_IMSG_FD_CHILD) == -1) {
3096 		fprintf(stderr, "%s: %s\n", getprogname(), strerror(errno));
3097 		_exit(1);
3098 	}
3099 
3100 	closefrom(GOT_IMSG_FD_CHILD + 1);
3101 
3102 	if (execl(path, path, repo_path, (char *)NULL) == -1) {
3103 		fprintf(stderr, "%s: %s: %s\n", getprogname(), path,
3104 		    strerror(errno));
3105 		_exit(1);
3106 	}
3107 }