Gotweb

Blob

Date:: Fri Apr 22 11:37:09 2022 UTC
Message:: inline struct got_object_id in struct got_object_qid Saves us from doing a malloc/free call for every item on the list. ok op@
Actions:: History | Blame | Raw File
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, struct got_pathlist_head *entries,
1493     int nentries)
1494 {
1495 	const struct got_error *err = NULL;
1496 	struct got_imsg_tree_object itree;
1497 	struct got_pathlist_entry *pe;
1498 	size_t totlen;
1499 	int nimsg; /* number of imsg queued in ibuf */
1500 
1501 	itree.nentries = nentries;
1502 	if (imsg_compose(ibuf, GOT_IMSG_TREE, 0, 0, -1, &itree, sizeof(itree))
1503 	    == -1)
1504 		return got_error_from_errno("imsg_compose TREE");
1505 
1506 	totlen = sizeof(itree);
1507 	nimsg = 1;
1508 	TAILQ_FOREACH(pe, entries, entry) {
1509 		const char *name = pe->path;
1510 		struct got_parsed_tree_entry *pte = pe->data;
1511 		struct ibuf *wbuf;
1512 		size_t namelen = strlen(name);
1513 		size_t len = sizeof(struct got_imsg_tree_entry) + namelen;
1514 
1515 		if (len > MAX_IMSGSIZE)
1516 			return got_error(GOT_ERR_NO_SPACE);
1517 
1518 		nimsg++;
1519 		if (totlen + len >= MAX_IMSGSIZE - (IMSG_HEADER_SIZE * nimsg)) {
1520 			err = flush_imsg(ibuf);
1521 			if (err)
1522 				return err;
1523 			nimsg = 0;
1524 		}
1525 
1526 		wbuf = imsg_create(ibuf, GOT_IMSG_TREE_ENTRY, 0, 0, len);
1527 		if (wbuf == NULL)
1528 			return got_error_from_errno("imsg_create TREE_ENTRY");
1529 
1530 		/* Keep in sync with struct got_imsg_tree_object definition! */
1531 		if (imsg_add(wbuf, pte->id, SHA1_DIGEST_LENGTH) == -1) {
1532 			err = got_error_from_errno("imsg_add TREE_ENTRY");
1533 			ibuf_free(wbuf);
1534 			return err;
1535 		}
1536 		if (imsg_add(wbuf, &pte->mode, sizeof(pte->mode)) == -1) {
1537 			err = got_error_from_errno("imsg_add TREE_ENTRY");
1538 			ibuf_free(wbuf);
1539 			return err;
1540 		}
1541 
1542 		if (imsg_add(wbuf, name, namelen) == -1) {
1543 			err = got_error_from_errno("imsg_add TREE_ENTRY");
1544 			ibuf_free(wbuf);
1545 			return err;
1546 		}
1547 
1548 		wbuf->fd = -1;
1549 		imsg_close(ibuf, wbuf);
1550 
1551 		totlen += len;
1552 	}
1553 
1554 	return flush_imsg(ibuf);
1555 }
1556 
1557 const struct got_error *
1558 got_privsep_recv_tree(struct got_tree_object **tree, struct imsgbuf *ibuf)
1559 {
1560 	const struct got_error *err = NULL;
1561 	const size_t min_datalen =
1562 	    MIN(sizeof(struct got_imsg_error),
1563 	    sizeof(struct got_imsg_tree_object));
1564 	struct got_imsg_tree_object *itree;
1565 	int nentries = 0;
1566 
1567 	*tree = NULL;
1568 get_more:
1569 	err = read_imsg(ibuf);
1570 	if (err)
1571 		goto done;
1572 
1573 	for (;;) {
1574 		struct imsg imsg;
1575 		size_t n;
1576 		size_t datalen;
1577 		struct got_imsg_tree_entry *ite;
1578 		struct got_tree_entry *te = NULL;
1579 
1580 		n = imsg_get(ibuf, &imsg);
1581 		if (n == 0) {
1582 			if (*tree && (*tree)->nentries != nentries)
1583 				goto get_more;
1584 			break;
1585 		}
1586 
1587 		if (imsg.hdr.len < IMSG_HEADER_SIZE + min_datalen) {
1588 			imsg_free(&imsg);
1589 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1590 			break;
1591 		}
1592 
1593 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1594 
1595 		switch (imsg.hdr.type) {
1596 		case GOT_IMSG_ERROR:
1597 			err = recv_imsg_error(&imsg, datalen);
1598 			break;
1599 		case GOT_IMSG_TREE:
1600 			/* This message should only appear once. */
1601 			if (*tree != NULL) {
1602 				err = got_error(GOT_ERR_PRIVSEP_MSG);
1603 				break;
1604 			}
1605 			if (datalen != sizeof(*itree)) {
1606 				err = got_error(GOT_ERR_PRIVSEP_LEN);
1607 				break;
1608 			}
1609 			itree = imsg.data;
1610 			*tree = malloc(sizeof(**tree));
1611 			if (*tree == NULL) {
1612 				err = got_error_from_errno("malloc");
1613 				break;
1614 			}
1615 			(*tree)->entries = calloc(itree->nentries,
1616 			    sizeof(struct got_tree_entry));
1617 			if ((*tree)->entries == NULL) {
1618 				err = got_error_from_errno("malloc");
1619 				break;
1620 			}
1621 			(*tree)->nentries = itree->nentries;
1622 			(*tree)->refcnt = 0;
1623 			break;
1624 		case GOT_IMSG_TREE_ENTRY:
1625 			/* This message should be preceeded by GOT_IMSG_TREE. */
1626 			if (*tree == NULL) {
1627 				err = got_error(GOT_ERR_PRIVSEP_MSG);
1628 				break;
1629 			}
1630 			if (datalen < sizeof(*ite) || datalen > MAX_IMSGSIZE) {
1631 				err = got_error(GOT_ERR_PRIVSEP_LEN);
1632 				break;
1633 			}
1634 
1635 			/* Remaining data contains the entry's name. */
1636 			datalen -= sizeof(*ite);
1637 			if (datalen == 0 || datalen > MAX_IMSGSIZE) {
1638 				err = got_error(GOT_ERR_PRIVSEP_LEN);
1639 				break;
1640 			}
1641 			ite = imsg.data;
1642 
1643 			if (datalen + 1 > sizeof(te->name)) {
1644 				err = got_error(GOT_ERR_NO_SPACE);
1645 				break;
1646 			}
1647 			te = &(*tree)->entries[nentries];
1648 			memcpy(te->name, imsg.data + sizeof(*ite), datalen);
1649 			te->name[datalen] = '\0';
1650 
1651 			memcpy(te->id.sha1, ite->id, SHA1_DIGEST_LENGTH);
1652 			te->mode = ite->mode;
1653 			te->idx = nentries;
1654 			nentries++;
1655 			break;
1656 		default:
1657 			err = got_error(GOT_ERR_PRIVSEP_MSG);
1658 			break;
1659 		}
1660 
1661 		imsg_free(&imsg);
1662 		if (err)
1663 			break;
1664 	}
1665 done:
1666 	if (*tree && (*tree)->nentries != nentries) {
1667 		if (err == NULL)
1668 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1669 		got_object_tree_close(*tree);
1670 		*tree = NULL;
1671 	}
1672 
1673 	return err;
1674 }
1675 
1676 const struct got_error *
1677 got_privsep_send_blob(struct imsgbuf *ibuf, size_t size, size_t hdrlen,
1678     const uint8_t *data)
1679 {
1680 	struct got_imsg_blob iblob;
1681 
1682 	iblob.size = size;
1683 	iblob.hdrlen = hdrlen;
1684 
1685 	if (data) {
1686 		uint8_t *buf;
1687 
1688 		if (size > GOT_PRIVSEP_INLINE_BLOB_DATA_MAX)
1689 			return got_error(GOT_ERR_NO_SPACE);
1690 
1691 		buf = malloc(sizeof(iblob) + size);
1692 		if (buf == NULL)
1693 			return got_error_from_errno("malloc");
1694 
1695 		memcpy(buf, &iblob, sizeof(iblob));
1696 		memcpy(buf + sizeof(iblob), data, size);
1697 		if (imsg_compose(ibuf, GOT_IMSG_BLOB, 0, 0, -1, buf,
1698 		    sizeof(iblob) + size) == -1) {
1699 			free(buf);
1700 			return got_error_from_errno("imsg_compose BLOB");
1701 		}
1702 		free(buf);
1703 	} else {
1704 		/* Data has already been written to file descriptor. */
1705 		if (imsg_compose(ibuf, GOT_IMSG_BLOB, 0, 0, -1, &iblob,
1706 		    sizeof(iblob)) == -1)
1707 			return got_error_from_errno("imsg_compose BLOB");
1708 	}
1709 
1710 
1711 	return flush_imsg(ibuf);
1712 }
1713 
1714 const struct got_error *
1715 got_privsep_recv_blob(uint8_t **outbuf, size_t *size, size_t *hdrlen,
1716     struct imsgbuf *ibuf)
1717 {
1718 	const struct got_error *err = NULL;
1719 	struct imsg imsg;
1720 	struct got_imsg_blob *iblob;
1721 	size_t datalen;
1722 
1723 	*outbuf = NULL;
1724 
1725 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
1726 	if (err)
1727 		return err;
1728 
1729 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1730 
1731 	switch (imsg.hdr.type) {
1732 	case GOT_IMSG_BLOB:
1733 		if (datalen < sizeof(*iblob)) {
1734 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1735 			break;
1736 		}
1737 		iblob = imsg.data;
1738 		*size = iblob->size;
1739 		*hdrlen = iblob->hdrlen;
1740 
1741 		if (datalen == sizeof(*iblob)) {
1742 			/* Data has been written to file descriptor. */
1743 			break;
1744 		}
1745 
1746 		if (*size > GOT_PRIVSEP_INLINE_BLOB_DATA_MAX) {
1747 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1748 			break;
1749 		}
1750 
1751 		*outbuf = malloc(*size);
1752 		if (*outbuf == NULL) {
1753 			err = got_error_from_errno("malloc");
1754 			break;
1755 		}
1756 		memcpy(*outbuf, imsg.data + sizeof(*iblob), *size);
1757 		break;
1758 	default:
1759 		err = got_error(GOT_ERR_PRIVSEP_MSG);
1760 		break;
1761 	}
1762 
1763 	imsg_free(&imsg);
1764 
1765 	return err;
1766 }
1767 
1768 static const struct got_error *
1769 send_tagmsg(struct imsgbuf *ibuf, struct got_tag_object *tag, size_t tagmsg_len)
1770 {
1771 	const struct got_error *err = NULL;
1772 	size_t offset, remain;
1773 
1774 	offset = 0;
1775 	remain = tagmsg_len;
1776 	while (remain > 0) {
1777 		size_t n = MIN(MAX_IMSGSIZE - IMSG_HEADER_SIZE, remain);
1778 
1779 		if (imsg_compose(ibuf, GOT_IMSG_TAG_TAGMSG, 0, 0, -1,
1780 		    tag->tagmsg + offset, n) == -1) {
1781 			err = got_error_from_errno("imsg_compose TAG_TAGMSG");
1782 			break;
1783 		}
1784 
1785 		err = flush_imsg(ibuf);
1786 		if (err)
1787 			break;
1788 
1789 		offset += n;
1790 		remain -= n;
1791 	}
1792 
1793 	return err;
1794 }
1795 
1796 const struct got_error *
1797 got_privsep_send_tag(struct imsgbuf *ibuf, struct got_tag_object *tag)
1798 {
1799 	const struct got_error *err = NULL;
1800 	struct got_imsg_tag_object *itag;
1801 	uint8_t *buf;
1802 	size_t len, total;
1803 	size_t tag_len = strlen(tag->tag);
1804 	size_t tagger_len = strlen(tag->tagger);
1805 	size_t tagmsg_len = strlen(tag->tagmsg);
1806 
1807 	total = sizeof(*itag) + tag_len + tagger_len + tagmsg_len;
1808 
1809 	buf = malloc(total);
1810 	if (buf == NULL)
1811 		return got_error_from_errno("malloc");
1812 
1813 	itag = (struct got_imsg_tag_object *)buf;
1814 	memcpy(itag->id, tag->id.sha1, sizeof(itag->id));
1815 	itag->obj_type = tag->obj_type;
1816 	itag->tag_len = tag_len;
1817 	itag->tagger_len = tagger_len;
1818 	itag->tagger_time = tag->tagger_time;
1819 	itag->tagger_gmtoff = tag->tagger_gmtoff;
1820 	itag->tagmsg_len = tagmsg_len;
1821 
1822 	len = sizeof(*itag);
1823 	memcpy(buf + len, tag->tag, tag_len);
1824 	len += tag_len;
1825 	memcpy(buf + len, tag->tagger, tagger_len);
1826 	len += tagger_len;
1827 
1828 	if (imsg_compose(ibuf, GOT_IMSG_TAG, 0, 0, -1, buf, len) == -1) {
1829 		err = got_error_from_errno("imsg_compose TAG");
1830 		goto done;
1831 	}
1832 
1833 	if (tagmsg_len == 0 ||
1834 	    tagmsg_len + len > MAX_IMSGSIZE - IMSG_HEADER_SIZE) {
1835 		err = flush_imsg(ibuf);
1836 		if (err)
1837 			goto done;
1838 	}
1839 	err = send_tagmsg(ibuf, tag, tagmsg_len);
1840 done:
1841 	free(buf);
1842 	return err;
1843 }
1844 
1845 const struct got_error *
1846 got_privsep_recv_tag(struct got_tag_object **tag, struct imsgbuf *ibuf)
1847 {
1848 	const struct got_error *err = NULL;
1849 	struct imsg imsg;
1850 	struct got_imsg_tag_object *itag;
1851 	size_t len, datalen;
1852 	const size_t min_datalen =
1853 	    MIN(sizeof(struct got_imsg_error),
1854 	    sizeof(struct got_imsg_tag_object));
1855 
1856 	*tag = NULL;
1857 
1858 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
1859 	if (err)
1860 		return err;
1861 
1862 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1863 	len = 0;
1864 
1865 	switch (imsg.hdr.type) {
1866 	case GOT_IMSG_TAG:
1867 		if (datalen < sizeof(*itag)) {
1868 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1869 			break;
1870 		}
1871 		itag = imsg.data;
1872 		if (datalen != sizeof(*itag) + itag->tag_len +
1873 		    itag->tagger_len) {
1874 			err = got_error(GOT_ERR_PRIVSEP_LEN);
1875 			break;
1876 		}
1877 		len += sizeof(*itag);
1878 
1879 		*tag = calloc(1, sizeof(**tag));
1880 		if (*tag == NULL) {
1881 			err = got_error_from_errno("calloc");
1882 			break;
1883 		}
1884 
1885 		memcpy((*tag)->id.sha1, itag->id, SHA1_DIGEST_LENGTH);
1886 
1887 		if (itag->tag_len == 0) {
1888 			(*tag)->tag = strdup("");
1889 			if ((*tag)->tag == NULL) {
1890 				err = got_error_from_errno("strdup");
1891 				break;
1892 			}
1893 		} else {
1894 			(*tag)->tag = malloc(itag->tag_len + 1);
1895 			if ((*tag)->tag == NULL) {
1896 				err = got_error_from_errno("malloc");
1897 				break;
1898 			}
1899 			memcpy((*tag)->tag, imsg.data + len,
1900 			    itag->tag_len);
1901 			(*tag)->tag[itag->tag_len] = '\0';
1902 		}
1903 		len += itag->tag_len;
1904 
1905 		(*tag)->obj_type = itag->obj_type;
1906 		(*tag)->tagger_time = itag->tagger_time;
1907 		(*tag)->tagger_gmtoff = itag->tagger_gmtoff;
1908 
1909 		if (itag->tagger_len == 0) {
1910 			(*tag)->tagger = strdup("");
1911 			if ((*tag)->tagger == NULL) {
1912 				err = got_error_from_errno("strdup");
1913 				break;
1914 			}
1915 		} else {
1916 			(*tag)->tagger = malloc(itag->tagger_len + 1);
1917 			if ((*tag)->tagger == NULL) {
1918 				err = got_error_from_errno("malloc");
1919 				break;
1920 			}
1921 			memcpy((*tag)->tagger, imsg.data + len,
1922 			    itag->tagger_len);
1923 			(*tag)->tagger[itag->tagger_len] = '\0';
1924 		}
1925 		len += itag->tagger_len;
1926 
1927 		if (itag->tagmsg_len == 0) {
1928 			(*tag)->tagmsg = strdup("");
1929 			if ((*tag)->tagmsg == NULL) {
1930 				err = got_error_from_errno("strdup");
1931 				break;
1932 			}
1933 		} else {
1934 			size_t offset = 0, remain = itag->tagmsg_len;
1935 
1936 			(*tag)->tagmsg = malloc(itag->tagmsg_len + 1);
1937 			if ((*tag)->tagmsg == NULL) {
1938 				err = got_error_from_errno("malloc");
1939 				break;
1940 			}
1941 			while (remain > 0) {
1942 				struct imsg imsg_log;
1943 				size_t n = MIN(MAX_IMSGSIZE - IMSG_HEADER_SIZE,
1944 				    remain);
1945 
1946 				err = got_privsep_recv_imsg(&imsg_log, ibuf, n);
1947 				if (err)
1948 					return err;
1949 
1950 				if (imsg_log.hdr.type != GOT_IMSG_TAG_TAGMSG)
1951 					return got_error(GOT_ERR_PRIVSEP_MSG);
1952 
1953 				memcpy((*tag)->tagmsg + offset, imsg_log.data,
1954 				    n);
1955 				imsg_free(&imsg_log);
1956 				offset += n;
1957 				remain -= n;
1958 			}
1959 			(*tag)->tagmsg[itag->tagmsg_len] = '\0';
1960 		}
1961 
1962 		break;
1963 	default:
1964 		err = got_error(GOT_ERR_PRIVSEP_MSG);
1965 		break;
1966 	}
1967 
1968 	imsg_free(&imsg);
1969 
1970 	return err;
1971 }
1972 
1973 const struct got_error *
1974 got_privsep_init_pack_child(struct imsgbuf *ibuf, struct got_pack *pack,
1975     struct got_packidx *packidx)
1976 {
1977 	const struct got_error *err = NULL;
1978 	struct got_imsg_packidx ipackidx;
1979 	struct got_imsg_pack ipack;
1980 	int fd;
1981 
1982 	ipackidx.len = packidx->len;
1983 	ipackidx.packfile_size = pack->filesize;
1984 	fd = dup(packidx->fd);
1985 	if (fd == -1)
1986 		return got_error_from_errno("dup");
1987 
1988 	if (imsg_compose(ibuf, GOT_IMSG_PACKIDX, 0, 0, fd, &ipackidx,
1989 	    sizeof(ipackidx)) == -1) {
1990 		err = got_error_from_errno("imsg_compose PACKIDX");
1991 		close(fd);
1992 		return err;
1993 	}
1994 
1995 	if (strlcpy(ipack.path_packfile, pack->path_packfile,
1996 	    sizeof(ipack.path_packfile)) >= sizeof(ipack.path_packfile))
1997 		return got_error(GOT_ERR_NO_SPACE);
1998 	ipack.filesize = pack->filesize;
1999 
2000 	fd = dup(pack->fd);
2001 	if (fd == -1)
2002 		return got_error_from_errno("dup");
2003 
2004 	if (imsg_compose(ibuf, GOT_IMSG_PACK, 0, 0, fd, &ipack, sizeof(ipack))
2005 	    == -1) {
2006 		err = got_error_from_errno("imsg_compose PACK");
2007 		close(fd);
2008 		return err;
2009 	}
2010 
2011 	return flush_imsg(ibuf);
2012 }
2013 
2014 const struct got_error *
2015 got_privsep_send_packed_obj_req(struct imsgbuf *ibuf, int idx,
2016     struct got_object_id *id)
2017 {
2018 	struct got_imsg_packed_object iobj;
2019 
2020 	iobj.idx = idx;
2021 	memcpy(iobj.id, id->sha1, sizeof(iobj.id));
2022 
2023 	if (imsg_compose(ibuf, GOT_IMSG_PACKED_OBJECT_REQUEST, 0, 0, -1,
2024 	    &iobj, sizeof(iobj)) == -1)
2025 		return got_error_from_errno("imsg_compose "
2026 		    "PACKED_OBJECT_REQUEST");
2027 
2028 	return flush_imsg(ibuf);
2029 }
2030 
2031 const struct got_error *
2032 got_privsep_send_packed_raw_obj_req(struct imsgbuf *ibuf, int idx,
2033     struct got_object_id *id)
2034 {
2035 	struct got_imsg_packed_object iobj;
2036 
2037 	iobj.idx = idx;
2038 	memcpy(iobj.id, id->sha1, sizeof(iobj.id));
2039 
2040 	if (imsg_compose(ibuf, GOT_IMSG_PACKED_RAW_OBJECT_REQUEST, 0, 0, -1,
2041 	    &iobj, sizeof(iobj)) == -1)
2042 		return got_error_from_errno("imsg_compose "
2043 		    "PACKED_OBJECT_REQUEST");
2044 
2045 	return flush_imsg(ibuf);
2046 }
2047 
2048 const struct got_error *
2049 got_privsep_send_gitconfig_parse_req(struct imsgbuf *ibuf, int fd)
2050 {
2051 	const struct got_error *err = NULL;
2052 
2053 	if (imsg_compose(ibuf, GOT_IMSG_GITCONFIG_PARSE_REQUEST, 0, 0, fd,
2054 	    NULL, 0) == -1) {
2055 		err = got_error_from_errno("imsg_compose "
2056 		    "GITCONFIG_PARSE_REQUEST");
2057 		close(fd);
2058 		return err;
2059 	}
2060 
2061 	return flush_imsg(ibuf);
2062 }
2063 
2064 const struct got_error *
2065 got_privsep_send_gitconfig_repository_format_version_req(struct imsgbuf *ibuf)
2066 {
2067 	if (imsg_compose(ibuf,
2068 	    GOT_IMSG_GITCONFIG_REPOSITORY_FORMAT_VERSION_REQUEST, 0, 0, -1,
2069 	    NULL, 0) == -1)
2070 		return got_error_from_errno("imsg_compose "
2071 		    "GITCONFIG_REPOSITORY_FORMAT_VERSION_REQUEST");
2072 
2073 	return flush_imsg(ibuf);
2074 }
2075 
2076 const struct got_error *
2077 got_privsep_send_gitconfig_repository_extensions_req(struct imsgbuf *ibuf)
2078 {
2079 	if (imsg_compose(ibuf,
2080 	    GOT_IMSG_GITCONFIG_REPOSITORY_EXTENSIONS_REQUEST, 0, 0, -1,
2081 	    NULL, 0) == -1)
2082 		return got_error_from_errno("imsg_compose "
2083 		    "GITCONFIG_REPOSITORY_EXTENSIONS_REQUEST");
2084 
2085 	return flush_imsg(ibuf);
2086 }
2087 
2088 
2089 const struct got_error *
2090 got_privsep_send_gitconfig_author_name_req(struct imsgbuf *ibuf)
2091 {
2092 	if (imsg_compose(ibuf,
2093 	    GOT_IMSG_GITCONFIG_AUTHOR_NAME_REQUEST, 0, 0, -1, NULL, 0) == -1)
2094 		return got_error_from_errno("imsg_compose "
2095 		    "GITCONFIG_AUTHOR_NAME_REQUEST");
2096 
2097 	return flush_imsg(ibuf);
2098 }
2099 
2100 const struct got_error *
2101 got_privsep_send_gitconfig_author_email_req(struct imsgbuf *ibuf)
2102 {
2103 	if (imsg_compose(ibuf,
2104 	    GOT_IMSG_GITCONFIG_AUTHOR_EMAIL_REQUEST, 0, 0, -1, NULL, 0) == -1)
2105 		return got_error_from_errno("imsg_compose "
2106 		    "GITCONFIG_AUTHOR_EMAIL_REQUEST");
2107 
2108 	return flush_imsg(ibuf);
2109 }
2110 
2111 const struct got_error *
2112 got_privsep_send_gitconfig_remotes_req(struct imsgbuf *ibuf)
2113 {
2114 	if (imsg_compose(ibuf,
2115 	    GOT_IMSG_GITCONFIG_REMOTES_REQUEST, 0, 0, -1, NULL, 0) == -1)
2116 		return got_error_from_errno("imsg_compose "
2117 		    "GITCONFIG_REMOTE_REQUEST");
2118 
2119 	return flush_imsg(ibuf);
2120 }
2121 
2122 const struct got_error *
2123 got_privsep_send_gitconfig_owner_req(struct imsgbuf *ibuf)
2124 {
2125 	if (imsg_compose(ibuf,
2126 	    GOT_IMSG_GITCONFIG_OWNER_REQUEST, 0, 0, -1, NULL, 0) == -1)
2127 		return got_error_from_errno("imsg_compose "
2128 		    "GITCONFIG_OWNER_REQUEST");
2129 
2130 	return flush_imsg(ibuf);
2131 }
2132 
2133 const struct got_error *
2134 got_privsep_recv_gitconfig_str(char **str, struct imsgbuf *ibuf)
2135 {
2136 	const struct got_error *err = NULL;
2137 	struct imsg imsg;
2138 	size_t datalen;
2139 	const size_t min_datalen = 0;
2140 
2141 	*str = NULL;
2142 
2143 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2144 	if (err)
2145 		return err;
2146 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2147 
2148 	switch (imsg.hdr.type) {
2149 	case GOT_IMSG_GITCONFIG_STR_VAL:
2150 		if (datalen == 0)
2151 			break;
2152 		/* datalen does not include terminating \0 */
2153 		*str = malloc(datalen + 1);
2154 		if (*str == NULL) {
2155 			err = got_error_from_errno("malloc");
2156 			break;
2157 		}
2158 		memcpy(*str, imsg.data, datalen);
2159 		(*str)[datalen] = '\0';
2160 		break;
2161 	default:
2162 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2163 		break;
2164 	}
2165 
2166 	imsg_free(&imsg);
2167 	return err;
2168 }
2169 
2170 const struct got_error *
2171 got_privsep_recv_gitconfig_int(int *val, struct imsgbuf *ibuf)
2172 {
2173 	const struct got_error *err = NULL;
2174 	struct imsg imsg;
2175 	size_t datalen;
2176 	const size_t min_datalen =
2177 	    MIN(sizeof(struct got_imsg_error), sizeof(int));
2178 
2179 	*val = 0;
2180 
2181 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2182 	if (err)
2183 		return err;
2184 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2185 
2186 	switch (imsg.hdr.type) {
2187 	case GOT_IMSG_GITCONFIG_INT_VAL:
2188 		if (datalen != sizeof(*val)) {
2189 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2190 			break;
2191 		}
2192 		memcpy(val, imsg.data, sizeof(*val));
2193 		break;
2194 	default:
2195 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2196 		break;
2197 	}
2198 
2199 	imsg_free(&imsg);
2200 	return err;
2201 }
2202 
2203 static void
2204 free_remote_data(struct got_remote_repo *remote)
2205 {
2206 	int i;
2207 
2208 	free(remote->name);
2209 	free(remote->fetch_url);
2210 	free(remote->send_url);
2211 	for (i = 0; i < remote->nfetch_branches; i++)
2212 		free(remote->fetch_branches[i]);
2213 	free(remote->fetch_branches);
2214 	for (i = 0; i < remote->nsend_branches; i++)
2215 		free(remote->send_branches[i]);
2216 	free(remote->send_branches);
2217 	for (i = 0; i < remote->nfetch_refs; i++)
2218 		free(remote->fetch_refs[i]);
2219 	free(remote->fetch_refs);
2220 }
2221 
2222 const struct got_error *
2223 got_privsep_recv_gitconfig_remotes(struct got_remote_repo **remotes,
2224     int *nremotes, struct imsgbuf *ibuf)
2225 {
2226 	const struct got_error *err = NULL;
2227 	struct imsg imsg;
2228 	size_t datalen;
2229 	struct got_imsg_remotes iremotes;
2230 	struct got_imsg_remote iremote;
2231 
2232 	*remotes = NULL;
2233 	*nremotes = 0;
2234 	iremotes.nremotes = 0;
2235 
2236 	err = got_privsep_recv_imsg(&imsg, ibuf, sizeof(iremotes));
2237 	if (err)
2238 		return err;
2239 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2240 
2241 	switch (imsg.hdr.type) {
2242 	case GOT_IMSG_GITCONFIG_REMOTES:
2243 		if (datalen != sizeof(iremotes)) {
2244 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2245 			break;
2246 		}
2247 		memcpy(&iremotes, imsg.data, sizeof(iremotes));
2248 		if (iremotes.nremotes == 0) {
2249 			imsg_free(&imsg);
2250 			return NULL;
2251 		}
2252 		break;
2253 	default:
2254 		imsg_free(&imsg);
2255 		return got_error(GOT_ERR_PRIVSEP_MSG);
2256 	}
2257 
2258 	imsg_free(&imsg);
2259 
2260 	*remotes = recallocarray(NULL, 0, iremotes.nremotes, sizeof(**remotes));
2261 	if (*remotes == NULL)
2262 		return got_error_from_errno("recallocarray");
2263 
2264 	while (*nremotes < iremotes.nremotes) {
2265 		struct got_remote_repo *remote;
2266 
2267 		err = got_privsep_recv_imsg(&imsg, ibuf, sizeof(iremote));
2268 		if (err)
2269 			break;
2270 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2271 
2272 		switch (imsg.hdr.type) {
2273 		case GOT_IMSG_GITCONFIG_REMOTE:
2274 			remote = &(*remotes)[*nremotes];
2275 			memset(remote, 0, sizeof(*remote));
2276 			if (datalen < sizeof(iremote)) {
2277 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2278 				break;
2279 			}
2280 			memcpy(&iremote, imsg.data, sizeof(iremote));
2281 			if (iremote.name_len == 0 ||
2282 			    iremote.fetch_url_len == 0 ||
2283 			    iremote.send_url_len == 0 ||
2284 			    (sizeof(iremote) + iremote.name_len +
2285 			    iremote.fetch_url_len + iremote.send_url_len) > datalen) {
2286 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2287 				break;
2288 			}
2289 			remote->name = strndup(imsg.data + sizeof(iremote),
2290 			    iremote.name_len);
2291 			if (remote->name == NULL) {
2292 				err = got_error_from_errno("strndup");
2293 				break;
2294 			}
2295 			remote->fetch_url = strndup(imsg.data + sizeof(iremote) +
2296 			    iremote.name_len, iremote.fetch_url_len);
2297 			if (remote->fetch_url == NULL) {
2298 				err = got_error_from_errno("strndup");
2299 				free_remote_data(remote);
2300 				break;
2301 			}
2302 			remote->send_url = strndup(imsg.data + sizeof(iremote) +
2303 			    iremote.name_len + iremote.fetch_url_len,
2304 			    iremote.send_url_len);
2305 			if (remote->send_url == NULL) {
2306 				err = got_error_from_errno("strndup");
2307 				free_remote_data(remote);
2308 				break;
2309 			}
2310 			remote->mirror_references = iremote.mirror_references;
2311 			remote->fetch_all_branches = iremote.fetch_all_branches;
2312 			remote->nfetch_branches = 0;
2313 			remote->fetch_branches = NULL;
2314 			remote->nsend_branches = 0;
2315 			remote->send_branches = NULL;
2316 			remote->nfetch_refs = 0;
2317 			remote->fetch_refs = NULL;
2318 			(*nremotes)++;
2319 			break;
2320 		default:
2321 			err = got_error(GOT_ERR_PRIVSEP_MSG);
2322 			break;
2323 		}
2324 
2325 		imsg_free(&imsg);
2326 		if (err)
2327 			break;
2328 	}
2329 
2330 	if (err) {
2331 		int i;
2332 		for (i = 0; i < *nremotes; i++)
2333 			free_remote_data(&(*remotes)[i]);
2334 		free(*remotes);
2335 		*remotes = NULL;
2336 		*nremotes = 0;
2337 	}
2338 	return err;
2339 }
2340 
2341 const struct got_error *
2342 got_privsep_send_gotconfig_parse_req(struct imsgbuf *ibuf, int fd)
2343 {
2344 	const struct got_error *err = NULL;
2345 
2346 	if (imsg_compose(ibuf, GOT_IMSG_GOTCONFIG_PARSE_REQUEST, 0, 0, fd,
2347 	    NULL, 0) == -1) {
2348 		err = got_error_from_errno("imsg_compose "
2349 		    "GOTCONFIG_PARSE_REQUEST");
2350 		close(fd);
2351 		return err;
2352 	}
2353 
2354 	return flush_imsg(ibuf);
2355 }
2356 
2357 const struct got_error *
2358 got_privsep_send_gotconfig_author_req(struct imsgbuf *ibuf)
2359 {
2360 	if (imsg_compose(ibuf,
2361 	    GOT_IMSG_GOTCONFIG_AUTHOR_REQUEST, 0, 0, -1, NULL, 0) == -1)
2362 		return got_error_from_errno("imsg_compose "
2363 		    "GOTCONFIG_AUTHOR_REQUEST");
2364 
2365 	return flush_imsg(ibuf);
2366 }
2367 
2368 const struct got_error *
2369 got_privsep_send_gotconfig_remotes_req(struct imsgbuf *ibuf)
2370 {
2371 	if (imsg_compose(ibuf,
2372 	    GOT_IMSG_GOTCONFIG_REMOTES_REQUEST, 0, 0, -1, NULL, 0) == -1)
2373 		return got_error_from_errno("imsg_compose "
2374 		    "GOTCONFIG_REMOTE_REQUEST");
2375 
2376 	return flush_imsg(ibuf);
2377 }
2378 
2379 const struct got_error *
2380 got_privsep_recv_gotconfig_str(char **str, struct imsgbuf *ibuf)
2381 {
2382 	const struct got_error *err = NULL;
2383 	struct imsg imsg;
2384 	size_t datalen;
2385 	const size_t min_datalen = 0;
2386 
2387 	*str = NULL;
2388 
2389 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2390 	if (err)
2391 		return err;
2392 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2393 
2394 	switch (imsg.hdr.type) {
2395 	case GOT_IMSG_ERROR:
2396 		if (datalen < sizeof(struct got_imsg_error)) {
2397 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2398 			break;
2399 		}
2400 		err = recv_imsg_error(&imsg, datalen);
2401 		break;
2402 	case GOT_IMSG_GOTCONFIG_STR_VAL:
2403 		if (datalen == 0)
2404 			break;
2405 		/* datalen does not include terminating \0 */
2406 		*str = malloc(datalen + 1);
2407 		if (*str == NULL) {
2408 			err = got_error_from_errno("malloc");
2409 			break;
2410 		}
2411 		memcpy(*str, imsg.data, datalen);
2412 		(*str)[datalen] = '\0';
2413 		break;
2414 	default:
2415 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2416 		break;
2417 	}
2418 
2419 	imsg_free(&imsg);
2420 	return err;
2421 }
2422 
2423 const struct got_error *
2424 got_privsep_recv_gotconfig_remotes(struct got_remote_repo **remotes,
2425     int *nremotes, struct imsgbuf *ibuf)
2426 {
2427 	const struct got_error *err = NULL;
2428 	struct imsg imsg;
2429 	size_t datalen;
2430 	struct got_imsg_remotes iremotes;
2431 	struct got_imsg_remote iremote;
2432 	const size_t min_datalen =
2433 	    MIN(sizeof(struct got_imsg_error), sizeof(iremotes));
2434 
2435 	*remotes = NULL;
2436 	*nremotes = 0;
2437 	iremotes.nremotes = 0;
2438 
2439 	err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2440 	if (err)
2441 		return err;
2442 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2443 
2444 	switch (imsg.hdr.type) {
2445 	case GOT_IMSG_ERROR:
2446 		if (datalen < sizeof(struct got_imsg_error)) {
2447 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2448 			break;
2449 		}
2450 		err = recv_imsg_error(&imsg, datalen);
2451 		break;
2452 	case GOT_IMSG_GOTCONFIG_REMOTES:
2453 		if (datalen != sizeof(iremotes)) {
2454 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2455 			break;
2456 		}
2457 		memcpy(&iremotes, imsg.data, sizeof(iremotes));
2458 		if (iremotes.nremotes == 0) {
2459 			imsg_free(&imsg);
2460 			return NULL;
2461 		}
2462 		break;
2463 	default:
2464 		imsg_free(&imsg);
2465 		return got_error(GOT_ERR_PRIVSEP_MSG);
2466 	}
2467 
2468 	imsg_free(&imsg);
2469 
2470 	*remotes = recallocarray(NULL, 0, iremotes.nremotes, sizeof(**remotes));
2471 	if (*remotes == NULL)
2472 		return got_error_from_errno("recallocarray");
2473 
2474 	while (*nremotes < iremotes.nremotes) {
2475 		struct got_remote_repo *remote;
2476 		const size_t min_datalen =
2477 		    MIN(sizeof(struct got_imsg_error), sizeof(iremote));
2478 		int i;
2479 
2480 		err = got_privsep_recv_imsg(&imsg, ibuf, min_datalen);
2481 		if (err)
2482 			break;
2483 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2484 
2485 		switch (imsg.hdr.type) {
2486 		case GOT_IMSG_ERROR:
2487 			if (datalen < sizeof(struct got_imsg_error)) {
2488 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2489 				break;
2490 			}
2491 			err = recv_imsg_error(&imsg, datalen);
2492 			break;
2493 		case GOT_IMSG_GOTCONFIG_REMOTE:
2494 			remote = &(*remotes)[*nremotes];
2495 			memset(remote, 0, sizeof(*remote));
2496 			if (datalen < sizeof(iremote)) {
2497 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2498 				break;
2499 			}
2500 			memcpy(&iremote, imsg.data, sizeof(iremote));
2501 			if (iremote.name_len == 0 ||
2502 			    (iremote.fetch_url_len == 0 &&
2503 			    iremote.send_url_len == 0) ||
2504 			    (sizeof(iremote) + iremote.name_len +
2505 			    iremote.fetch_url_len + iremote.send_url_len) >
2506 			    datalen) {
2507 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2508 				break;
2509 			}
2510 			remote->name = strndup(imsg.data + sizeof(iremote),
2511 			    iremote.name_len);
2512 			if (remote->name == NULL) {
2513 				err = got_error_from_errno("strndup");
2514 				break;
2515 			}
2516 			remote->fetch_url = strndup(imsg.data +
2517 			    sizeof(iremote) + iremote.name_len,
2518 			    iremote.fetch_url_len);
2519 			if (remote->fetch_url == NULL) {
2520 				err = got_error_from_errno("strndup");
2521 				free_remote_data(remote);
2522 				break;
2523 			}
2524 			remote->send_url = strndup(imsg.data +
2525 			    sizeof(iremote) + iremote.name_len +
2526 			    iremote.fetch_url_len, iremote.send_url_len);
2527 			if (remote->send_url == NULL) {
2528 				err = got_error_from_errno("strndup");
2529 				free_remote_data(remote);
2530 				break;
2531 			}
2532 			remote->mirror_references = iremote.mirror_references;
2533 			remote->fetch_all_branches = iremote.fetch_all_branches;
2534 			if (iremote.nfetch_branches > 0) {
2535 				remote->fetch_branches = recallocarray(NULL, 0,
2536 				    iremote.nfetch_branches, sizeof(char *));
2537 				if (remote->fetch_branches == NULL) {
2538 					err = got_error_from_errno("calloc");
2539 					free_remote_data(remote);
2540 					break;
2541 				}
2542 			}
2543 			remote->nfetch_branches = 0;
2544 			for (i = 0; i < iremote.nfetch_branches; i++) {
2545 				char *branch;
2546 				err = got_privsep_recv_gotconfig_str(&branch,
2547 				    ibuf);
2548 				if (err) {
2549 					free_remote_data(remote);
2550 					goto done;
2551 				}
2552 				remote->fetch_branches[i] = branch;
2553 				remote->nfetch_branches++;
2554 			}
2555 			if (iremote.nsend_branches > 0) {
2556 				remote->send_branches = recallocarray(NULL, 0,
2557 				    iremote.nsend_branches, sizeof(char *));
2558 				if (remote->send_branches == NULL) {
2559 					err = got_error_from_errno("calloc");
2560 					free_remote_data(remote);
2561 					break;
2562 				}
2563 			}
2564 			remote->nsend_branches = 0;
2565 			for (i = 0; i < iremote.nsend_branches; i++) {
2566 				char *branch;
2567 				err = got_privsep_recv_gotconfig_str(&branch,
2568 				    ibuf);
2569 				if (err) {
2570 					free_remote_data(remote);
2571 					goto done;
2572 				}
2573 				remote->send_branches[i] = branch;
2574 				remote->nsend_branches++;
2575 			}
2576 			if (iremote.nfetch_refs > 0) {
2577 				remote->fetch_refs = recallocarray(NULL, 0,
2578 				    iremote.nfetch_refs, sizeof(char *));
2579 				if (remote->fetch_refs == NULL) {
2580 					err = got_error_from_errno("calloc");
2581 					free_remote_data(remote);
2582 					break;
2583 				}
2584 			}
2585 			remote->nfetch_refs = 0;
2586 			for (i = 0; i < iremote.nfetch_refs; i++) {
2587 				char *ref;
2588 				err = got_privsep_recv_gotconfig_str(&ref,
2589 				    ibuf);
2590 				if (err) {
2591 					free_remote_data(remote);
2592 					goto done;
2593 				}
2594 				remote->fetch_refs[i] = ref;
2595 				remote->nfetch_refs++;
2596 			}
2597 			(*nremotes)++;
2598 			break;
2599 		default:
2600 			err = got_error(GOT_ERR_PRIVSEP_MSG);
2601 			break;
2602 		}
2603 
2604 		imsg_free(&imsg);
2605 		if (err)
2606 			break;
2607 	}
2608 done:
2609 	if (err) {
2610 		int i;
2611 		for (i = 0; i < *nremotes; i++)
2612 			free_remote_data(&(*remotes)[i]);
2613 		free(*remotes);
2614 		*remotes = NULL;
2615 		*nremotes = 0;
2616 	}
2617 	return err;
2618 }
2619 
2620 const struct got_error *
2621 got_privsep_send_commit_traversal_request(struct imsgbuf *ibuf,
2622     struct got_object_id *id, int idx, const char *path)
2623 {
2624 	const struct got_error *err = NULL;
2625 	struct ibuf *wbuf;
2626 	size_t path_len = strlen(path) + 1;
2627 
2628 	wbuf = imsg_create(ibuf, GOT_IMSG_COMMIT_TRAVERSAL_REQUEST, 0, 0,
2629 	    sizeof(struct got_imsg_commit_traversal_request) + path_len);
2630 	if (wbuf == NULL)
2631 		return got_error_from_errno(
2632 		    "imsg_create COMMIT_TRAVERSAL_REQUEST");
2633 	if (imsg_add(wbuf, id->sha1, SHA1_DIGEST_LENGTH) == -1) {
2634 		err = got_error_from_errno("imsg_add COMMIT_TRAVERSAL_REQUEST");
2635 		ibuf_free(wbuf);
2636 		return err;
2637 	}
2638 	if (imsg_add(wbuf, &idx, sizeof(idx)) == -1) {
2639 		err = got_error_from_errno("imsg_add COMMIT_TRAVERSAL_REQUEST");
2640 		ibuf_free(wbuf);
2641 		return err;
2642 	}
2643 	if (imsg_add(wbuf, path, path_len) == -1) {
2644 		err = got_error_from_errno("imsg_add COMMIT_TRAVERSAL_REQUEST");
2645 		ibuf_free(wbuf);
2646 		return err;
2647 	}
2648 
2649 	wbuf->fd = -1;
2650 	imsg_close(ibuf, wbuf);
2651 
2652 	return flush_imsg(ibuf);
2653 }
2654 
2655 const struct got_error *
2656 got_privsep_recv_traversed_commits(struct got_commit_object **changed_commit,
2657     struct got_object_id **changed_commit_id,
2658     struct got_object_id_queue *commit_ids, struct imsgbuf *ibuf)
2659 {
2660 	const struct got_error *err = NULL;
2661 	struct imsg imsg;
2662 	struct got_imsg_traversed_commits *icommits;
2663 	size_t datalen;
2664 	int i, done = 0;
2665 
2666 	*changed_commit = NULL;
2667 	*changed_commit_id = NULL;
2668 
2669 	while (!done) {
2670 		err = got_privsep_recv_imsg(&imsg, ibuf, 0);
2671 		if (err)
2672 			return err;
2673 
2674 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2675 		switch (imsg.hdr.type) {
2676 		case GOT_IMSG_TRAVERSED_COMMITS:
2677 			icommits = imsg.data;
2678 			if (datalen != sizeof(*icommits) +
2679 			    icommits->ncommits * SHA1_DIGEST_LENGTH) {
2680 				err = got_error(GOT_ERR_PRIVSEP_LEN);
2681 				break;
2682 			}
2683 			for (i = 0; i < icommits->ncommits; i++) {
2684 				struct got_object_qid *qid;
2685 				uint8_t *sha1 = (uint8_t *)imsg.data +
2686 				    sizeof(*icommits) + i * SHA1_DIGEST_LENGTH;
2687 				err = got_object_qid_alloc_partial(&qid);
2688 				if (err)
2689 					break;
2690 				memcpy(qid->id.sha1, sha1, SHA1_DIGEST_LENGTH);
2691 				STAILQ_INSERT_TAIL(commit_ids, qid, entry);
2692 
2693 				/* The last commit may contain a change. */
2694 				if (i == icommits->ncommits - 1) {
2695 					*changed_commit_id =
2696 					    got_object_id_dup(&qid->id);
2697 					if (*changed_commit_id == NULL) {
2698 						err = got_error_from_errno(
2699 						    "got_object_id_dup");
2700 						break;
2701 					}
2702 				}
2703 			}
2704 			break;
2705 		case GOT_IMSG_COMMIT:
2706 			if (*changed_commit_id == NULL) {
2707 				err = got_error(GOT_ERR_PRIVSEP_MSG);
2708 				break;
2709 			}
2710 			err = get_commit_from_imsg(changed_commit, &imsg,
2711 			    datalen, ibuf);
2712 			break;
2713 		case GOT_IMSG_COMMIT_TRAVERSAL_DONE:
2714 			done = 1;
2715 			break;
2716 		default:
2717 			err = got_error(GOT_ERR_PRIVSEP_MSG);
2718 			break;
2719 		}
2720 
2721 		imsg_free(&imsg);
2722 		if (err)
2723 			break;
2724 	}
2725 
2726 	if (err)
2727 		got_object_id_queue_free(commit_ids);
2728 	return err;
2729 }
2730 
2731 const struct got_error *
2732 got_privsep_send_raw_delta_req(struct imsgbuf *ibuf, int idx,
2733     struct got_object_id *id)
2734 {
2735 	struct got_imsg_raw_delta_request dreq;
2736 
2737 	dreq.idx = idx;
2738 	memcpy(dreq.id, id->sha1, SHA1_DIGEST_LENGTH);
2739 
2740 	if (imsg_compose(ibuf, GOT_IMSG_RAW_DELTA_REQUEST, 0, 0, -1,
2741 	    &dreq, sizeof(dreq)) == -1)
2742 		return got_error_from_errno("imsg_compose RAW_DELTA_REQUEST");
2743 
2744 	return flush_imsg(ibuf);
2745 }
2746 
2747 const struct got_error *
2748 got_privsep_send_raw_delta_outfd(struct imsgbuf *ibuf, int fd)
2749 {
2750 	return send_fd(ibuf, GOT_IMSG_RAW_DELTA_OUTFD, fd);
2751 }
2752 
2753 const struct got_error *
2754 got_privsep_send_raw_delta(struct imsgbuf *ibuf, uint64_t base_size,
2755     uint64_t result_size,  off_t delta_size, off_t delta_offset,
2756     off_t delta_out_offset, struct got_object_id *base_id)
2757 {
2758 	struct got_imsg_raw_delta idelta;
2759 	int ret;
2760 
2761 	idelta.base_size = base_size;
2762 	idelta.result_size = result_size;
2763 	idelta.delta_size = delta_size;
2764 	idelta.delta_offset = delta_offset;
2765 	idelta.delta_out_offset = delta_out_offset;
2766 	memcpy(idelta.base_id, base_id->sha1, SHA1_DIGEST_LENGTH);
2767 
2768 	ret = imsg_compose(ibuf, GOT_IMSG_RAW_DELTA, 0, 0, -1,
2769 	    &idelta, sizeof(idelta));
2770 	if (ret == -1)
2771 		return got_error_from_errno("imsg_compose RAW_DELTA");
2772 
2773 	return flush_imsg(ibuf);
2774 }
2775 
2776 const struct got_error *
2777 got_privsep_recv_raw_delta(uint64_t *base_size, uint64_t *result_size,
2778     off_t *delta_size, off_t *delta_offset, off_t *delta_out_offset,
2779     struct got_object_id **base_id, struct imsgbuf *ibuf)
2780 {
2781 	const struct got_error *err = NULL;
2782 	struct imsg imsg;
2783 	struct got_imsg_raw_delta *delta;
2784 	size_t datalen;
2785 
2786 	*base_size = 0;
2787 	*result_size = 0;
2788 	*delta_size = 0;
2789 	*delta_offset = 0;
2790 	*delta_out_offset = 0;
2791 	*base_id = NULL;
2792 
2793 	err = got_privsep_recv_imsg(&imsg, ibuf, 0);
2794 	if (err)
2795 		return err;
2796 
2797 	datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
2798 
2799 	switch (imsg.hdr.type) {
2800 	case GOT_IMSG_RAW_DELTA:
2801 		if (datalen != sizeof(*delta)) {
2802 			err = got_error(GOT_ERR_PRIVSEP_LEN);
2803 			break;
2804 		}
2805 		delta = imsg.data;
2806 		*base_size = delta->base_size;
2807 		*result_size = delta->result_size;
2808 		*delta_size = delta->delta_size;
2809 		*delta_offset = delta->delta_offset;
2810 		*delta_out_offset = delta->delta_out_offset;
2811 		*base_id = calloc(1, sizeof(**base_id));
2812 		if (*base_id == NULL) {
2813 			err = got_error_from_errno("malloc");
2814 			break;
2815 		}
2816 		memcpy((*base_id)->sha1, delta->base_id, SHA1_DIGEST_LENGTH);
2817 		break;
2818 	default:
2819 		err = got_error(GOT_ERR_PRIVSEP_MSG);
2820 		break;
2821 	}
2822 
2823 	imsg_free(&imsg);
2824 
2825 	if (err) {
2826 		free(*base_id);
2827 		*base_id = NULL;
2828 	}
2829 	return err;
2830 }
2831 
2832 const struct got_error *
2833 got_privsep_unveil_exec_helpers(void)
2834 {
2835 	const char *helpers[] = {
2836 	    GOT_PATH_PROG_READ_PACK,
2837 	    GOT_PATH_PROG_READ_OBJECT,
2838 	    GOT_PATH_PROG_READ_COMMIT,
2839 	    GOT_PATH_PROG_READ_TREE,
2840 	    GOT_PATH_PROG_READ_BLOB,
2841 	    GOT_PATH_PROG_READ_TAG,
2842 	    GOT_PATH_PROG_READ_GITCONFIG,
2843 	    GOT_PATH_PROG_READ_GOTCONFIG,
2844 	    GOT_PATH_PROG_READ_PATCH,
2845 	    GOT_PATH_PROG_FETCH_PACK,
2846 	    GOT_PATH_PROG_INDEX_PACK,
2847 	    GOT_PATH_PROG_SEND_PACK,
2848 	};
2849 	size_t i;
2850 
2851 	for (i = 0; i < nitems(helpers); i++) {
2852 		if (unveil(helpers[i], "x") == 0)
2853 			continue;
2854 		return got_error_from_errno2("unveil", helpers[i]);
2855 	}
2856 
2857 	return NULL;
2858 }
2859 
2860 void
2861 got_privsep_exec_child(int imsg_fds[2], const char *path, const char *repo_path)
2862 {
2863 	if (close(imsg_fds[0]) == -1) {
2864 		fprintf(stderr, "%s: %s\n", getprogname(), strerror(errno));
2865 		_exit(1);
2866 	}
2867 
2868 	if (dup2(imsg_fds[1], GOT_IMSG_FD_CHILD) == -1) {
2869 		fprintf(stderr, "%s: %s\n", getprogname(), strerror(errno));
2870 		_exit(1);
2871 	}
2872 
2873 	closefrom(GOT_IMSG_FD_CHILD + 1);
2874 
2875 	if (execl(path, path, repo_path, (char *)NULL) == -1) {
2876 		fprintf(stderr, "%s: %s: %s\n", getprogname(), path,
2877 		    strerror(errno));
2878 		_exit(1);
2879 	}
2880 }