1 /*
2  * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <sys/types.h>
18 #include <sys/stat.h>
19 #include <sys/queue.h>
20 #include <sys/mman.h>
21 
22 #include <dirent.h>
23 #include <fcntl.h>
24 #include <errno.h>
25 #include <stdio.h>
26 #include <stdint.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <limits.h>
30 #include <sha1.h>
31 #include <endian.h>
32 #include <zlib.h>
33 
34 #include "got_error.h"
35 #include "got_object.h"
36 #include "got_repository.h"
37 #include "got_opentemp.h"
38 
39 #include "got_lib_sha1.h"
40 #include "got_lib_pack.h"
41 #include "got_lib_path.h"
42 #include "got_lib_delta.h"
43 #include "got_lib_inflate.h"
44 #include "got_lib_object.h"
45 #include "got_lib_repository.h"
46 
47 #ifndef nitems
48 #define nitems(_a) (sizeof(_a) / sizeof((_a)[0]))
49 #endif
50 
51 #define GOT_PACK_PREFIX		"pack-"
52 #define GOT_PACKFILE_SUFFIX	".pack"
53 #define GOT_PACKIDX_SUFFIX		".idx"
54 #define GOT_PACKFILE_NAMELEN	(strlen(GOT_PACK_PREFIX) + \
55 				SHA1_DIGEST_STRING_LENGTH - 1 + \
56 				strlen(GOT_PACKFILE_SUFFIX))
57 #define GOT_PACKIDX_NAMELEN	(strlen(GOT_PACK_PREFIX) + \
58 				SHA1_DIGEST_STRING_LENGTH - 1 + \
59 				strlen(GOT_PACKIDX_SUFFIX))
60 
61 #ifndef MIN
62 #define	MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
63 #endif
64 
65 static const struct got_error *
66 verify_fanout_table(uint32_t *fanout_table)
67 {
68 	int i;
69 
70 	for (i = 0; i < 0xff - 1; i++) {
71 		if (be32toh(fanout_table[i]) > be32toh(fanout_table[i + 1]))
72 			return got_error(GOT_ERR_BAD_PACKIDX);
73 	}
74 
75 	return NULL;
76 }
77 
78 static const struct got_error *
79 get_packfile_size(size_t *size, const char *path)
80 {
81 	struct stat sb;
82 	char *dot;
83 
84 	*size = 0;
85 
86 	dot = strrchr(path, '.');
87 	if (dot == NULL)
88 		return got_error(GOT_ERR_BAD_PATH);
89 
90 	/* Path must point to a pack index or to a pack file. */
91 	if (strcmp(dot, GOT_PACKIDX_SUFFIX) == 0) {
92 		const struct got_error *err = NULL;
93 		char *path_pack;
94 		char base_path[PATH_MAX];
95 
96 		/* Convert pack index path to pack file path. */
97 		if (strlcpy(base_path, path, PATH_MAX) > PATH_MAX)
98 			return got_error(GOT_ERR_NO_SPACE);
99 		dot = strrchr(base_path, '.');
100 		if (dot == NULL)
101 			return got_error(GOT_ERR_BAD_PATH);
102 		*dot = '\0';
103 		if (asprintf(&path_pack, "%s.pack", base_path) == -1)
104 			return got_error_from_errno();
105 
106 		if (stat(path_pack, &sb) != 0)
107 			err = got_error_from_errno();
108 		free(path_pack);
109 		if (err)
110 			return err;
111 	} else if (strcmp(dot, GOT_PACKFILE_SUFFIX) == 0) {
112 		if (stat(path, &sb) != 0)
113 			return got_error_from_errno();
114 	} else
115 		return got_error(GOT_ERR_BAD_PATH);
116 
117 	*size = sb.st_size;
118 	return 0;
119 }
120 
121 const struct got_error *
122 got_packidx_open(struct got_packidx **packidx, const char *path, int verify)
123 {
124 	struct got_packidx *p;
125 	struct got_packidx_v2_hdr *h;
126 	const struct got_error *err = NULL;
127 	size_t nobj, len_fanout, len_ids, offset, remain;
128 	ssize_t n;
129 	SHA1_CTX ctx;
130 	uint8_t sha1[SHA1_DIGEST_LENGTH];
131 
132 	*packidx = NULL;
133 
134 	SHA1Init(&ctx);
135 
136 	p = calloc(1, sizeof(*p));
137 	if (p == NULL)
138 		return got_error_from_errno();
139 
140 	p->fd = open(path, O_RDONLY | O_NOFOLLOW, GOT_DEFAULT_FILE_MODE);
141 	if (p->fd == -1)
142 		return got_error_from_errno();
143 
144 	err = get_packfile_size(&p->len, path);
145 	if (err) {
146 		close(p->fd);
147 		free(p);
148 		return err;
149 	}
150 	if (p->len < sizeof(p->hdr)) {
151 		err = got_error(GOT_ERR_BAD_PACKIDX);
152 		close(p->fd);
153 		free(p);
154 		return err;
155 	}
156 
157 	p->path_packidx = strdup(path);
158 	if (p->path_packidx == NULL) {
159 		err = got_error_from_errno();
160 		goto done;
161 	}
162 
163 #ifndef GOT_PACK_NO_MMAP
164 	p->map = mmap(NULL, p->len, PROT_READ, MAP_PRIVATE, p->fd, 0);
165 	if (p->map == MAP_FAILED)
166 		p->map = NULL; /* fall back to read(2) */
167 #endif
168 
169 	h = &p->hdr;
170 	offset = 0;
171 	remain = p->len;
172 
173 	if (remain < sizeof(*h->magic)) {
174 		err = got_error(GOT_ERR_BAD_PACKIDX);
175 		goto done;
176 	}
177 	if (p->map)
178 		h->magic = (uint32_t *)(p->map + offset);
179 	else {
180 		h->magic = malloc(sizeof(*h->magic));
181 		if (h->magic == NULL) {
182 			err = got_error_from_errno();
183 			goto done;
184 		}
185 		n = read(p->fd, h->magic, sizeof(*h->magic));
186 		if (n != sizeof(*h->magic)) {
187 			err = got_error(GOT_ERR_BAD_PACKIDX);
188 			goto done;
189 		}
190 	}
191 	if (betoh32(*h->magic) != GOT_PACKIDX_V2_MAGIC) {
192 		err = got_error(GOT_ERR_BAD_PACKIDX);
193 		goto done;
194 	}
195 	offset += sizeof(*h->magic);
196 	remain -= sizeof(*h->magic);
197 
198 	if (verify)
199 		SHA1Update(&ctx, (uint8_t *)h->magic, sizeof(*h->magic));
200 
201 	if (remain < sizeof(*h->version)) {
202 		err = got_error(GOT_ERR_BAD_PACKIDX);
203 		goto done;
204 	}
205 	if (p->map)
206 		h->version = (uint32_t *)(p->map + offset);
207 	else {
208 		h->version = malloc(sizeof(*h->version));
209 		if (h->version == NULL) {
210 			err = got_error_from_errno();
211 			goto done;
212 		}
213 		n = read(p->fd, h->version, sizeof(*h->version));
214 		if (n != sizeof(*h->version)) {
215 			err = got_error(GOT_ERR_BAD_PACKIDX);
216 			goto done;
217 		}
218 	}
219 	if (betoh32(*h->version) != GOT_PACKIDX_VERSION) {
220 		err = got_error(GOT_ERR_BAD_PACKIDX);
221 		goto done;
222 	}
223 	offset += sizeof(*h->version);
224 	remain -= sizeof(*h->version);
225 
226 	if (verify)
227 		SHA1Update(&ctx, (uint8_t *)h->version, sizeof(*h->version));
228 
229 	len_fanout =
230 	    sizeof(*h->fanout_table) * GOT_PACKIDX_V2_FANOUT_TABLE_ITEMS;
231 	if (remain < len_fanout) {
232 		err = got_error(GOT_ERR_BAD_PACKIDX);
233 		goto done;
234 	}
235 	if (p->map)
236 		h->fanout_table = (uint32_t *)(p->map + offset);
237 	else {
238 		h->fanout_table = malloc(len_fanout);
239 		if (h->fanout_table == NULL) {
240 			err = got_error_from_errno();
241 			goto done;
242 		}
243 		n = read(p->fd, h->fanout_table, len_fanout);
244 		if (n != len_fanout) {
245 			err = got_error(GOT_ERR_BAD_PACKIDX);
246 			goto done;
247 		}
248 	}
249 	err = verify_fanout_table(h->fanout_table);
250 	if (err)
251 		goto done;
252 	if (verify)
253 		SHA1Update(&ctx, (uint8_t *)h->fanout_table, len_fanout);
254 	offset += len_fanout;
255 	remain -= len_fanout;
256 
257 	nobj = betoh32(h->fanout_table[0xff]);
258 	len_ids = nobj * sizeof(*h->sorted_ids);
259 	if (len_ids <= nobj || len_ids > remain) {
260 		err = got_error(GOT_ERR_BAD_PACKIDX);
261 		goto done;
262 	}
263 	if (p->map)
264 		h->sorted_ids =
265 		    (struct got_packidx_object_id *)((uint8_t*)(p->map + offset));
266 	else {
267 		h->sorted_ids = malloc(len_ids);
268 		if (h->sorted_ids == NULL) {
269 			err = got_error(GOT_ERR_BAD_PACKIDX);
270 			goto done;
271 		}
272 		n = read(p->fd, h->sorted_ids, len_ids);
273 		if (n != len_ids) {
274 			err = got_error(GOT_ERR_BAD_PACKIDX);
275 			goto done;
276 		}
277 	}
278 	if (verify)
279 		SHA1Update(&ctx, (uint8_t *)h->sorted_ids, len_ids);
280 	offset += len_ids;
281 	remain -= len_ids;
282 
283 	if (remain < nobj * sizeof(*h->crc32)) {
284 		err = got_error(GOT_ERR_BAD_PACKIDX);
285 		goto done;
286 	}
287 	if (p->map)
288 		h->crc32 = (uint32_t *)((uint8_t*)(p->map + offset));
289 	else {
290 		h->crc32 = malloc(nobj * sizeof(*h->crc32));
291 		if (h->crc32 == NULL) {
292 			err = got_error_from_errno();
293 			goto done;
294 		}
295 		n = read(p->fd, h->crc32, nobj * sizeof(*h->crc32));
296 		if (n != nobj * sizeof(*h->crc32)) {
297 			err = got_error(GOT_ERR_BAD_PACKIDX);
298 			goto done;
299 		}
300 	}
301 	if (verify)
302 		SHA1Update(&ctx, (uint8_t *)h->crc32, nobj * sizeof(*h->crc32));
303 	remain -= nobj * sizeof(*h->crc32);
304 	offset += nobj * sizeof(*h->crc32);
305 
306 	if (remain < nobj * sizeof(*h->offsets)) {
307 		err = got_error(GOT_ERR_BAD_PACKIDX);
308 		goto done;
309 	}
310 	if (p->map)
311 		h->offsets = (uint32_t *)((uint8_t*)(p->map + offset));
312 	else {
313 		h->offsets = malloc(nobj * sizeof(*h->offsets));
314 		if (h->offsets == NULL) {
315 			err = got_error_from_errno();
316 			goto done;
317 		}
318 		n = read(p->fd, h->offsets, nobj * sizeof(*h->offsets));
319 		if (n != nobj * sizeof(*h->offsets)) {
320 			err = got_error(GOT_ERR_BAD_PACKIDX);
321 			goto done;
322 		}
323 	}
324 	if (verify)
325 		SHA1Update(&ctx, (uint8_t *)h->offsets,
326 		    nobj * sizeof(*h->offsets));
327 	remain -= nobj * sizeof(*h->offsets);
328 	offset += nobj * sizeof(*h->offsets);
329 
330 	/* Large file offsets are contained only in files > 2GB. */
331 	if (p->len <= 0x80000000)
332 		goto checksum;
333 
334 	if (remain < nobj * sizeof(*h->large_offsets)) {
335 		err = got_error(GOT_ERR_BAD_PACKIDX);
336 		goto done;
337 	}
338 	if (p->map)
339 		h->large_offsets = (uint64_t *)((uint8_t*)(p->map + offset));
340 	else {
341 		h->offsets = malloc(nobj * sizeof(*h->large_offsets));
342 		if (h->offsets == NULL) {
343 			err = got_error_from_errno();
344 			goto done;
345 		}
346 		n = read(p->fd, h->large_offsets,
347 		    nobj * sizeof(*h->large_offsets));
348 		if (n != nobj * sizeof(*h->large_offsets)) {
349 			err = got_error(GOT_ERR_BAD_PACKIDX);
350 			goto done;
351 		}
352 	}
353 	if (verify)
354 		SHA1Update(&ctx, (uint8_t*)h->large_offsets,
355 		    nobj * sizeof(*h->large_offsets));
356 	remain -= nobj * sizeof(*h->large_offsets);
357 	offset += nobj * sizeof(*h->large_offsets);
358 
359 checksum:
360 	if (remain < sizeof(*h->trailer)) {
361 		err = got_error(GOT_ERR_BAD_PACKIDX);
362 		goto done;
363 	}
364 	if (p->map)
365 		h->trailer =
366 		    (struct got_packidx_trailer *)((uint8_t*)(p->map + offset));
367 	else {
368 		h->trailer = malloc(sizeof(*h->trailer));
369 		if (h->trailer == NULL) {
370 			err = got_error_from_errno();
371 			goto done;
372 		}
373 		n = read(p->fd, h->trailer, sizeof(*h->trailer));
374 		if (n != sizeof(*h->trailer)) {
375 			err = got_error(GOT_ERR_BAD_PACKIDX);
376 			goto done;
377 		}
378 	}
379 	if (verify) {
380 		SHA1Update(&ctx, h->trailer->packfile_sha1, SHA1_DIGEST_LENGTH);
381 		SHA1Final(sha1, &ctx);
382 		if (memcmp(h->trailer->packidx_sha1, sha1,
383 		    SHA1_DIGEST_LENGTH) != 0)
384 			err = got_error(GOT_ERR_PACKIDX_CSUM);
385 	}
386 done:
387 	if (err)
388 		got_packidx_close(p);
389 	else
390 		*packidx = p;
391 	return err;
392 }
393 
394 const struct got_error *
395 got_packidx_close(struct got_packidx *packidx)
396 {
397 	const struct got_error *err = NULL;
398 
399 	free(packidx->path_packidx);
400 	if (packidx->map) {
401 		if (munmap(packidx->map, packidx->len) == -1)
402 			err = got_error_from_errno();
403 	} else {
404 		free(packidx->hdr.magic);
405 		free(packidx->hdr.version);
406 		free(packidx->hdr.fanout_table);
407 		free(packidx->hdr.sorted_ids);
408 		free(packidx->hdr.crc32);
409 		free(packidx->hdr.offsets);
410 		free(packidx->hdr.large_offsets);
411 		free(packidx->hdr.trailer);
412 	}
413 	close(packidx->fd);
414 	free(packidx);
415 
416 	return err;
417 }
418 
419 static int
420 is_packidx_filename(const char *name, size_t len)
421 {
422 	if (len != GOT_PACKIDX_NAMELEN)
423 		return 0;
424 
425 	if (strncmp(name, GOT_PACK_PREFIX, strlen(GOT_PACK_PREFIX)) != 0)
426 		return 0;
427 
428 	if (strcmp(name + strlen(GOT_PACK_PREFIX) +
429 	    SHA1_DIGEST_STRING_LENGTH - 1, GOT_PACKIDX_SUFFIX) != 0)
430 		return 0;
431 
432 	return 1;
433 }
434 
435 static off_t
436 get_object_offset(struct got_packidx *packidx, int idx)
437 {
438 	uint32_t totobj = betoh32(packidx->hdr.fanout_table[0xff]);
439 	uint32_t offset = betoh32(packidx->hdr.offsets[idx]);
440 	if (offset & GOT_PACKIDX_OFFSET_VAL_IS_LARGE_IDX) {
441 		uint64_t loffset;
442 		idx = offset & GOT_PACKIDX_OFFSET_VAL_MASK;
443 		if (idx < 0 || idx > totobj ||
444 		    packidx->hdr.large_offsets == NULL)
445 			return -1;
446 		loffset = betoh64(packidx->hdr.large_offsets[idx]);
447 		return (loffset > INT64_MAX ? -1 : (off_t)loffset);
448 	}
449 	return (off_t)(offset & GOT_PACKIDX_OFFSET_VAL_MASK);
450 }
451 
452 static int
453 get_object_idx(struct got_packidx *packidx, struct got_object_id *id,
454     struct got_repository *repo)
455 {
456 	u_int8_t id0 = id->sha1[0];
457 	uint32_t totobj = betoh32(packidx->hdr.fanout_table[0xff]);
458 	int left = 0, right = totobj - 1;
459 
460 	if (id0 > 0)
461 		left = betoh32(packidx->hdr.fanout_table[id0 - 1]);
462 
463 	while (left <= right) {
464 		struct got_packidx_object_id *oid;
465 		int i, cmp;
466 
467 		i = ((left + right) / 2);
468 		oid = &packidx->hdr.sorted_ids[i];
469 		cmp = memcmp(id->sha1, oid->sha1, SHA1_DIGEST_LENGTH);
470 		if (cmp == 0)
471 			return i;
472 		else if (cmp > 0)
473 			left = i + 1;
474 		else if (cmp < 0)
475 			right = i - 1;
476 	}
477 
478 	return -1;
479 }
480 
481 static const struct got_error *
482 cache_packidx(struct got_packidx *packidx, struct got_repository *repo)
483 {
484 	const struct got_error *err = NULL;
485 	int i;
486 
487 	for (i = 0; i < nitems(repo->packidx_cache); i++) {
488 		if (repo->packidx_cache[i] == NULL)
489 			break;
490 	}
491 
492 	if (i == nitems(repo->packidx_cache)) {
493 		err = got_packidx_close(repo->packidx_cache[i - 1]);
494 		if (err)
495 			return err;
496 		memmove(&repo->packidx_cache[1], &repo->packidx_cache[0],
497 		    sizeof(repo->packidx_cache) -
498 		    sizeof(repo->packidx_cache[0]));
499 		i = 0;
500 	}
501 
502 	repo->packidx_cache[i] = packidx;
503 	return NULL;
504 }
505 
506 static const struct got_error *
507 search_packidx(struct got_packidx **packidx, int *idx,
508     struct got_repository *repo, struct got_object_id *id)
509 {
510 	const struct got_error *err;
511 	char *path_packdir;
512 	DIR *packdir;
513 	struct dirent *dent;
514 	char *path_packidx;
515 	int i;
516 
517 	/* Search pack index cache. */
518 	for (i = 0; i < nitems(repo->packidx_cache); i++) {
519 		if (repo->packidx_cache[i] == NULL)
520 			break;
521 		*idx = get_object_idx(repo->packidx_cache[i], id, repo);
522 		if (*idx != -1) {
523 			*packidx = repo->packidx_cache[i];
524 			return NULL;
525 		}
526 	}
527 	/* No luck. Search the filesystem. */
528 
529 	path_packdir = got_repo_get_path_objects_pack(repo);
530 	if (path_packdir == NULL)
531 		return got_error_from_errno();
532 
533 	packdir = opendir(path_packdir);
534 	if (packdir == NULL) {
535 		err = got_error_from_errno();
536 		goto done;
537 	}
538 
539 	while ((dent = readdir(packdir)) != NULL) {
540 		if (!is_packidx_filename(dent->d_name, dent->d_namlen))
541 			continue;
542 
543 		if (asprintf(&path_packidx, "%s/%s", path_packdir,
544 		    dent->d_name) == -1) {
545 			err = got_error_from_errno();
546 			goto done;
547 		}
548 
549 		err = got_packidx_open(packidx, path_packidx, 0);
550 		free(path_packidx);
551 		if (err)
552 			goto done;
553 
554 		*idx = get_object_idx(*packidx, id, repo);
555 		if (*idx != -1) {
556 			err = NULL; /* found the object */
557 			err = cache_packidx(*packidx, repo);
558 			goto done;
559 		}
560 
561 		err = got_packidx_close(*packidx);
562 		*packidx = NULL;
563 		if (err)
564 			goto done;
565 	}
566 
567 	err = got_error(GOT_ERR_NO_OBJ);
568 done:
569 	free(path_packdir);
570 	if (packdir && closedir(packdir) != 0 && err == 0)
571 		err = got_error_from_errno();
572 	return err;
573 }
574 
575 static const struct got_error *
576 get_packfile_path(char **path_packfile, struct got_repository *repo,
577     struct got_packidx *packidx)
578 {
579 	size_t size;
580 
581 	/* Packfile path contains ".pack" instead of ".idx", so add one byte. */
582 	size = strlen(packidx->path_packidx) + 2;
583 	if (size < GOT_PACKFILE_NAMELEN + 1)
584 		return got_error(GOT_ERR_BAD_PATH);
585 
586 	*path_packfile = calloc(size, sizeof(**path_packfile));
587 	if (*path_packfile == NULL)
588 		return got_error_from_errno();
589 
590 	/* Copy up to and excluding ".idx". */
591 	if (strlcpy(*path_packfile, packidx->path_packidx,
592 	    size - strlen(GOT_PACKIDX_SUFFIX) - 1) >= size)
593 		return got_error(GOT_ERR_NO_SPACE);
594 
595 	if (strlcat(*path_packfile, GOT_PACKFILE_SUFFIX, size) >= size)
596 		return got_error(GOT_ERR_NO_SPACE);
597 
598 	return NULL;
599 }
600 
601 static const struct got_error *
602 read_packfile_hdr(int fd, struct got_packidx *packidx)
603 {
604 	const struct got_error *err = NULL;
605 	uint32_t totobj = betoh32(packidx->hdr.fanout_table[0xff]);
606 	struct got_packfile_hdr hdr;
607 	ssize_t n;
608 
609 	n = read(fd, &hdr, sizeof(hdr));
610 	if (n < 0)
611 		return got_error_from_errno();
612 	if (n != sizeof(hdr))
613 		return got_error(GOT_ERR_BAD_PACKFILE);
614 
615 	if (betoh32(hdr.signature) != GOT_PACKFILE_SIGNATURE ||
616 	    betoh32(hdr.version) != GOT_PACKFILE_VERSION ||
617 	    betoh32(hdr.nobjects) != totobj)
618 		err = got_error(GOT_ERR_BAD_PACKFILE);
619 
620 	return err;
621 }
622 
623 static const struct got_error *
624 open_packfile(int *fd, const char *path_packfile,
625     struct got_repository *repo, struct got_packidx *packidx)
626 {
627 	const struct got_error *err = NULL;
628 
629 	*fd = open(path_packfile, O_RDONLY | O_NOFOLLOW, GOT_DEFAULT_FILE_MODE);
630 	if (*fd == -1)
631 		return got_error_from_errno();
632 
633 	if (packidx) {
634 		err = read_packfile_hdr(*fd, packidx);
635 		if (err) {
636 			close(*fd);
637 			*fd = -1;
638 		}
639 	}
640 	return err;
641 }
642 
643 const struct got_error *
644 got_pack_close(struct got_pack *pack)
645 {
646 	const struct got_error *err = NULL;
647 
648 	if (pack->map && munmap(pack->map, pack->filesize) == -1)
649 		err = got_error_from_errno();
650 	close(pack->fd);
651 	pack->fd = -1;
652 	free(pack->path_packfile);
653 	pack->path_packfile = NULL;
654 	pack->filesize = 0;
655 
656 	return err;
657 }
658 
659 static const struct got_error *
660 cache_pack(struct got_pack **packp, const char *path_packfile,
661     struct got_packidx *packidx, struct got_repository *repo)
662 {
663 	const struct got_error *err = NULL;
664 	struct got_pack *pack = NULL;
665 	int i;
666 
667 	if (packp)
668 		*packp = NULL;
669 
670 	for (i = 0; i < nitems(repo->packs); i++) {
671 		pack = &repo->packs[i];
672 		if (pack->path_packfile == NULL)
673 			break;
674 		if (strcmp(pack->path_packfile, path_packfile) == 0)
675 			return NULL;
676 	}
677 
678 	if (i == nitems(repo->packs) - 1) {
679 		err = got_pack_close(&repo->packs[i - 1]);
680 		if (err)
681 			return err;
682 		memmove(&repo->packs[1], &repo->packs[0],
683 		    sizeof(repo->packs) - sizeof(repo->packs[0]));
684 		i = 0;
685 	}
686 
687 	pack = &repo->packs[i];
688 
689 	pack->path_packfile = strdup(path_packfile);
690 	if (pack->path_packfile == NULL) {
691 		err = got_error_from_errno();
692 		goto done;
693 	}
694 
695 	err = open_packfile(&pack->fd, path_packfile, repo, packidx);
696 	if (err)
697 		goto done;
698 
699 	err = get_packfile_size(&pack->filesize, path_packfile);
700 	if (err)
701 		goto done;
702 
703 #ifndef GOT_PACK_NO_MMAP
704 	pack->map = mmap(NULL, pack->filesize, PROT_READ, MAP_PRIVATE,
705 	    pack->fd, 0);
706 	if (pack->map == MAP_FAILED)
707 		pack->map = NULL; /* fall back to read(2) */
708 #endif
709 done:
710 	if (err) {
711 		if (pack) {
712 			free(pack->path_packfile);
713 			memset(pack, 0, sizeof(*pack));
714 		}
715 	} else if (packp)
716 		*packp = pack;
717 	return err;
718 }
719 
720 struct got_pack *
721 get_cached_pack(const char *path_packfile, struct got_repository *repo)
722 {
723 	struct got_pack *pack = NULL;
724 	int i;
725 
726 	for (i = 0; i < nitems(repo->packs); i++) {
727 		pack = &repo->packs[i];
728 		if (pack->path_packfile == NULL)
729 			break;
730 		if (strcmp(pack->path_packfile, path_packfile) == 0)
731 			return pack;
732 	}
733 
734 	return NULL;
735 }
736 
737 static const struct got_error *
738 parse_object_type_and_size(uint8_t *type, uint64_t *size, size_t *len,
739     struct got_pack *pack, off_t offset)
740 {
741 	uint8_t t = 0;
742 	uint64_t s = 0;
743 	uint8_t sizeN;
744 	size_t mapoff = 0;
745 	int i = 0;
746 
747 	*len = 0;
748 
749 	if (offset >= pack->filesize)
750 		return got_error(GOT_ERR_PACK_OFFSET);
751 
752 	if (pack->map) {
753 		mapoff = (size_t)offset;
754 	} else {
755 		if (lseek(pack->fd, offset, SEEK_SET) == -1)
756 			return got_error_from_errno();
757 	}
758 
759 	do {
760 		/* We do not support size values which don't fit in 64 bit. */
761 		if (i > 9)
762 			return got_error(GOT_ERR_NO_SPACE);
763 
764 		if (pack->map) {
765 			sizeN = *(pack->map + mapoff);
766 			mapoff += sizeof(sizeN);
767 		} else {
768 			ssize_t n = read(pack->fd, &sizeN, sizeof(sizeN));
769 			if (n < 0)
770 				return got_error_from_errno();
771 			if (n != sizeof(sizeN))
772 				return got_error(GOT_ERR_BAD_PACKFILE);
773 		}
774 		*len += sizeof(sizeN);
775 
776 		if (i == 0) {
777 			t = (sizeN & GOT_PACK_OBJ_SIZE0_TYPE_MASK) >>
778 			    GOT_PACK_OBJ_SIZE0_TYPE_MASK_SHIFT;
779 			s = (sizeN & GOT_PACK_OBJ_SIZE0_VAL_MASK);
780 		} else {
781 			size_t shift = 4 + 7 * (i - 1);
782 			s |= ((sizeN & GOT_PACK_OBJ_SIZE_VAL_MASK) << shift);
783 		}
784 		i++;
785 	} while (sizeN & GOT_PACK_OBJ_SIZE_MORE);
786 
787 	*type = t;
788 	*size = s;
789 	return NULL;
790 }
791 
792 static const struct got_error *
793 open_plain_object(struct got_object **obj, const char *path_packfile,
794     struct got_object_id *id, uint8_t type, off_t offset, size_t size)
795 {
796 	*obj = calloc(1, sizeof(**obj));
797 	if (*obj == NULL)
798 		return got_error_from_errno();
799 
800 	(*obj)->path_packfile = strdup(path_packfile);
801 	if ((*obj)->path_packfile == NULL) {
802 		const struct got_error *err = got_error_from_errno();
803 		free(*obj);
804 		*obj = NULL;
805 		return err;
806 	}
807 
808 	(*obj)->type = type;
809 	(*obj)->flags = GOT_OBJ_FLAG_PACKED;
810 	(*obj)->hdrlen = 0;
811 	(*obj)->size = size;
812 	memcpy(&(*obj)->id, id, sizeof((*obj)->id));
813 	(*obj)->pack_offset = offset;
814 
815 	return NULL;
816 }
817 
818 static const struct got_error *
819 parse_negative_offset(int64_t *offset, size_t *len, struct got_pack *pack,
820     off_t delta_offset)
821 {
822 	int64_t o = 0;
823 	uint8_t offN;
824 	int i = 0;
825 
826 	*len = 0;
827 
828 	do {
829 		/* We do not support offset values which don't fit in 64 bit. */
830 		if (i > 8)
831 			return got_error(GOT_ERR_NO_SPACE);
832 
833 		if (pack->map) {
834 			size_t mapoff;
835 			if (delta_offset >= pack->filesize)
836 				return got_error(GOT_ERR_PACK_OFFSET);
837 			mapoff = (size_t)delta_offset + *len;
838 			offN = *(pack->map + mapoff);
839 		} else {
840 			ssize_t n;
841 			n = read(pack->fd, &offN, sizeof(offN));
842 			if (n < 0)
843 				return got_error_from_errno();
844 			if (n != sizeof(offN))
845 				return got_error(GOT_ERR_BAD_PACKFILE);
846 		}
847 		*len += sizeof(offN);
848 
849 		if (i == 0)
850 			o = (offN & GOT_PACK_OBJ_DELTA_OFF_VAL_MASK);
851 		else {
852 			o++;
853 			o <<= 7;
854 			o += (offN & GOT_PACK_OBJ_DELTA_OFF_VAL_MASK);
855 		}
856 		i++;
857 	} while (offN & GOT_PACK_OBJ_DELTA_OFF_MORE);
858 
859 	*offset = o;
860 	return NULL;
861 }
862 
863 static const struct got_error *
864 parse_offset_delta(off_t *base_offset, size_t *len, struct got_pack *pack,
865     off_t offset, int tslen)
866 {
867 	const struct got_error *err;
868 	int64_t negoffset;
869 	size_t negofflen;
870 
871 	*len = 0;
872 
873 	err = parse_negative_offset(&negoffset, &negofflen, pack,
874 	    offset + tslen);
875 	if (err)
876 		return err;
877 
878 	/* Compute the base object's offset (must be in the same pack file). */
879 	*base_offset = (offset - negoffset);
880 	if (*base_offset <= 0)
881 		return got_error(GOT_ERR_BAD_PACKFILE);
882 
883 	*len = negofflen;
884 	return NULL;
885 }
886 
887 static const struct got_error *
888 resolve_delta_chain(struct got_delta_chain *, struct got_repository *,
889     struct got_packidx *, struct got_pack *, off_t, size_t, int,
890     size_t, unsigned int);
891 
892 static const struct got_error *
893 add_delta(struct got_delta_chain *deltas, off_t delta_offset, size_t tslen,
894     int delta_type, size_t delta_size, size_t delta_data_offset,
895     uint8_t *delta_buf, size_t delta_len)
896 {
897 	struct got_delta *delta;
898 
899 	delta = got_delta_open(delta_offset, tslen, delta_type, delta_size,
900 	    delta_data_offset, delta_buf,
901 	    delta_len);
902 	if (delta == NULL)
903 		return got_error_from_errno();
904 	/* delta is freed in got_object_close() */
905 	deltas->nentries++;
906 	SIMPLEQ_INSERT_HEAD(&deltas->entries, delta, entry);
907 	return NULL;
908 }
909 
910 static const struct got_error *
911 resolve_offset_delta(struct got_delta_chain *deltas,
912     struct got_repository *repo, struct got_packidx *packidx,
913     struct got_pack *pack, off_t delta_offset, size_t tslen,
914     int delta_type, size_t delta_size, unsigned int recursion)
915 
916 {
917 	const struct got_error *err;
918 	off_t base_offset;
919 	uint8_t base_type;
920 	uint64_t base_size;
921 	size_t base_tslen;
922 	off_t delta_data_offset;
923 	uint8_t *delta_buf;
924 	size_t delta_len, consumed;
925 
926 	err = parse_offset_delta(&base_offset, &consumed, pack,
927 	    delta_offset, tslen);
928 	if (err)
929 		return err;
930 
931 	delta_data_offset = delta_offset + tslen + consumed;
932 	if (delta_data_offset >= pack->filesize)
933 		return got_error(GOT_ERR_PACK_OFFSET);
934 
935 	if (pack->map == NULL) {
936 		delta_data_offset = lseek(pack->fd, 0, SEEK_CUR);
937 		if (delta_data_offset == -1)
938 			return got_error_from_errno();
939 	}
940 
941 	if (pack->map) {
942 		size_t mapoff = (size_t)delta_data_offset;
943 		err = got_inflate_to_mem_mmap(&delta_buf, &delta_len, pack->map,
944 		    mapoff, pack->filesize - mapoff);
945 		if (err)
946 			return err;
947 	} else {
948 
949 		err = got_inflate_to_mem_fd(&delta_buf, &delta_len, pack->fd);
950 		if (err)
951 			return err;
952 	}
953 
954 	err = add_delta(deltas, delta_offset, tslen, delta_type, delta_size,
955 	    delta_data_offset, delta_buf, delta_len);
956 	if (err)
957 		goto done;
958 
959 	/* An offset delta must be in the same packfile. */
960 	if (base_offset >= pack->filesize) {
961 		err = got_error(GOT_ERR_PACK_OFFSET);
962 		goto done;
963 	}
964 
965 	err = parse_object_type_and_size(&base_type, &base_size, &base_tslen,
966 	    pack, base_offset);
967 	if (err)
968 		goto done;
969 
970 	err = resolve_delta_chain(deltas, repo, packidx, pack, base_offset,
971 	    base_tslen, base_type, base_size, recursion - 1);
972 done:
973 	if (err)
974 		free(delta_buf);
975 	return err;
976 }
977 
978 static const struct got_error *
979 resolve_ref_delta(struct got_delta_chain *deltas, struct got_repository *repo,
980     struct got_packidx *packidx, struct got_pack *pack,
981     off_t delta_offset, size_t tslen, int delta_type, size_t delta_size,
982     unsigned int recursion)
983 {
984 	const struct got_error *err;
985 	struct got_object_id id;
986 	int idx;
987 	off_t base_offset;
988 	uint8_t base_type;
989 	uint64_t base_size;
990 	size_t base_tslen;
991 	off_t delta_data_offset;
992 	uint8_t *delta_buf = NULL;
993 	size_t delta_len;
994 
995 	if (delta_offset >= pack->filesize)
996 		return got_error(GOT_ERR_PACK_OFFSET);
997 	delta_data_offset = delta_offset + tslen;
998 	if (delta_data_offset >= pack->filesize)
999 		return got_error(GOT_ERR_PACK_OFFSET);
1000 
1001 	if (pack->map == NULL) {
1002 		delta_data_offset = lseek(pack->fd, 0, SEEK_CUR);
1003 		if (delta_data_offset == -1)
1004 			return got_error_from_errno();
1005 	}
1006 
1007 
1008 	if (pack->map) {
1009 		size_t mapoff = (size_t)delta_data_offset;
1010 		memcpy(&id, pack->map + mapoff, sizeof(id));
1011 		mapoff += sizeof(id);
1012 		err = got_inflate_to_mem_mmap(&delta_buf, &delta_len, pack->map,
1013 		    mapoff, pack->filesize - mapoff);
1014 		if (err)
1015 			goto done;
1016 	} else {
1017 		ssize_t n = read(pack->fd, &id, sizeof(id));
1018 		if (n < 0)
1019 			return got_error_from_errno();
1020 		if (n != sizeof(id))
1021 			return got_error(GOT_ERR_BAD_PACKFILE);
1022 		err = got_inflate_to_mem_fd(&delta_buf, &delta_len, pack->fd);
1023 		if (err)
1024 			goto done;
1025 	}
1026 
1027 	err = add_delta(deltas, delta_offset, tslen, delta_type, delta_size,
1028 	    delta_data_offset, delta_buf, delta_len);
1029 	if (err)
1030 		goto done;
1031 
1032 	/* Delta base must be in the same pack file. */
1033 	idx = get_object_idx(packidx, &id, repo);
1034 	if (idx == -1) {
1035 		err = got_error(GOT_ERR_BAD_PACKFILE);
1036 		goto done;
1037 	}
1038 
1039 	base_offset = get_object_offset(packidx, idx);
1040 	if (base_offset == (uint64_t)-1) {
1041 		err = got_error(GOT_ERR_BAD_PACKIDX);
1042 		goto done;
1043 	}
1044 
1045 	if (base_offset >= pack->filesize) {
1046 		err = got_error(GOT_ERR_PACK_OFFSET);
1047 		goto done;
1048 	}
1049 
1050 	err = parse_object_type_and_size(&base_type, &base_size, &base_tslen,
1051 	    pack, base_offset);
1052 	if (err)
1053 		goto done;
1054 
1055 	err = resolve_delta_chain(deltas, repo, packidx, pack, base_offset,
1056 	    base_tslen, base_type, base_size, recursion - 1);
1057 done:
1058 	if (err)
1059 		free(delta_buf);
1060 	return err;
1061 }
1062 
1063 static const struct got_error *
1064 resolve_delta_chain(struct got_delta_chain *deltas, struct got_repository *repo,
1065     struct got_packidx *packidx, struct got_pack *pack, off_t delta_offset,
1066     size_t tslen, int delta_type, size_t delta_size, unsigned int recursion)
1067 {
1068 	const struct got_error *err = NULL;
1069 
1070 	if (--recursion == 0)
1071 		return got_error(GOT_ERR_RECURSION);
1072 
1073 	switch (delta_type) {
1074 	case GOT_OBJ_TYPE_COMMIT:
1075 	case GOT_OBJ_TYPE_TREE:
1076 	case GOT_OBJ_TYPE_BLOB:
1077 	case GOT_OBJ_TYPE_TAG:
1078 		/* Plain types are the final delta base. Recursion ends. */
1079 		err = add_delta(deltas, delta_offset, tslen, delta_type,
1080 		    delta_size, 0, NULL, 0);
1081 		break;
1082 	case GOT_OBJ_TYPE_OFFSET_DELTA:
1083 		err = resolve_offset_delta(deltas, repo, packidx, pack,
1084 		    delta_offset, tslen, delta_type, delta_size, recursion - 1);
1085 		break;
1086 	case GOT_OBJ_TYPE_REF_DELTA:
1087 		err = resolve_ref_delta(deltas, repo, packidx, pack,
1088 		    delta_offset, tslen, delta_type, delta_size, recursion - 1);
1089 		break;
1090 	default:
1091 		return got_error(GOT_ERR_OBJ_TYPE);
1092 	}
1093 
1094 	return err;
1095 }
1096 
1097 static const struct got_error *
1098 open_delta_object(struct got_object **obj, struct got_repository *repo,
1099     struct got_packidx *packidx, struct got_pack *pack,
1100     struct got_object_id *id, off_t offset, size_t tslen,
1101     int delta_type, size_t delta_size)
1102 {
1103 	const struct got_error *err = NULL;
1104 	int resolved_type;
1105 
1106 	*obj = calloc(1, sizeof(**obj));
1107 	if (*obj == NULL)
1108 		return got_error_from_errno();
1109 
1110 	(*obj)->flags = 0;
1111 	(*obj)->hdrlen = 0;
1112 	(*obj)->size = 0; /* Not known because deltas aren't applied yet. */
1113 	memcpy(&(*obj)->id, id, sizeof((*obj)->id));
1114 	(*obj)->pack_offset = offset + tslen;
1115 
1116 	(*obj)->path_packfile = strdup(pack->path_packfile);
1117 	if ((*obj)->path_packfile == NULL) {
1118 		err = got_error_from_errno();
1119 		goto done;
1120 	}
1121 	(*obj)->flags |= GOT_OBJ_FLAG_PACKED;
1122 
1123 	SIMPLEQ_INIT(&(*obj)->deltas.entries);
1124 	(*obj)->flags |= GOT_OBJ_FLAG_DELTIFIED;
1125 
1126 	err = resolve_delta_chain(&(*obj)->deltas, repo, packidx, pack, offset,
1127 	    tslen, delta_type, delta_size, GOT_DELTA_CHAIN_RECURSION_MAX);
1128 	if (err)
1129 		goto done;
1130 
1131 	err = got_delta_chain_get_base_type(&resolved_type, &(*obj)->deltas);
1132 	if (err)
1133 		goto done;
1134 	(*obj)->type = resolved_type;
1135 
1136 done:
1137 	if (err) {
1138 		got_object_close(*obj);
1139 		*obj = NULL;
1140 	}
1141 	return err;
1142 }
1143 
1144 static const struct got_error *
1145 open_packed_object(struct got_object **obj, struct got_repository *repo,
1146     struct got_packidx *packidx, int idx, struct got_object_id *id)
1147 {
1148 	const struct got_error *err = NULL;
1149 	off_t offset;
1150 	char *path_packfile;
1151 	struct got_pack *pack;
1152 	uint8_t type;
1153 	uint64_t size;
1154 	size_t tslen;
1155 
1156 	*obj = NULL;
1157 
1158 	offset = get_object_offset(packidx, idx);
1159 	if (offset == (uint64_t)-1)
1160 		return got_error(GOT_ERR_BAD_PACKIDX);
1161 
1162 	err = get_packfile_path(&path_packfile, repo, packidx);
1163 	if (err)
1164 		return err;
1165 
1166 	pack = get_cached_pack(path_packfile, repo);
1167 	if (pack == NULL) {
1168 		err = cache_pack(&pack, path_packfile, packidx, repo);
1169 		if (err)
1170 			goto done;
1171 	}
1172 
1173 	err = parse_object_type_and_size(&type, &size, &tslen, pack, offset);
1174 	if (err)
1175 		goto done;
1176 
1177 	switch (type) {
1178 	case GOT_OBJ_TYPE_COMMIT:
1179 	case GOT_OBJ_TYPE_TREE:
1180 	case GOT_OBJ_TYPE_BLOB:
1181 	case GOT_OBJ_TYPE_TAG:
1182 		err = open_plain_object(obj, path_packfile, id, type,
1183 		    offset + tslen, size);
1184 		break;
1185 
1186 	case GOT_OBJ_TYPE_OFFSET_DELTA:
1187 	case GOT_OBJ_TYPE_REF_DELTA:
1188 		err = open_delta_object(obj, repo, packidx, pack, id, offset,
1189 		    tslen, type, size);
1190 		break;
1191 
1192 	default:
1193 		err = got_error(GOT_ERR_OBJ_TYPE);
1194 		goto done;
1195 	}
1196 done:
1197 	free(path_packfile);
1198 	return err;
1199 }
1200 
1201 const struct got_error *
1202 got_packfile_open_object(struct got_object **obj, struct got_object_id *id,
1203     struct got_repository *repo)
1204 {
1205 	const struct got_error *err = NULL;
1206 	struct got_packidx *packidx = NULL;
1207 	int idx;
1208 
1209 	err = search_packidx(&packidx, &idx, repo, id);
1210 	if (err)
1211 		return err;
1212 
1213 	err = open_packed_object(obj, repo, packidx, idx, id);
1214 	if (err)
1215 		return err;
1216 
1217 	err = cache_pack(NULL, (*obj)->path_packfile, packidx, repo);
1218 	return err;
1219 }
1220 
1221 static const struct got_error *
1222 get_delta_chain_max_size(uint64_t *max_size, struct got_delta_chain *deltas)
1223 {
1224 	struct got_delta *delta;
1225 	uint64_t base_size = 0, result_size = 0;
1226 
1227 	*max_size = 0;
1228 	SIMPLEQ_FOREACH(delta, &deltas->entries, entry) {
1229 		/* Plain object types are the delta base. */
1230 		if (delta->type != GOT_OBJ_TYPE_COMMIT &&
1231 		    delta->type != GOT_OBJ_TYPE_TREE &&
1232 		    delta->type != GOT_OBJ_TYPE_BLOB &&
1233 		    delta->type != GOT_OBJ_TYPE_TAG) {
1234 			const struct got_error *err;
1235 			err = got_delta_get_sizes(&base_size, &result_size,
1236 			    delta->delta_buf, delta->delta_len);
1237 			if (err)
1238 				return err;
1239 		} else
1240 			base_size = delta->size;
1241 		if (base_size > *max_size)
1242 			*max_size = base_size;
1243 		if (result_size > *max_size)
1244 			*max_size = result_size;
1245 	}
1246 
1247 	return NULL;
1248 }
1249 
1250 static const struct got_error *
1251 dump_delta_chain_to_file(size_t *result_size, struct got_delta_chain *deltas,
1252     const char *path_packfile, FILE *outfile, struct got_repository *repo)
1253 {
1254 	const struct got_error *err = NULL;
1255 	struct got_delta *delta;
1256 	FILE *base_file = NULL, *accum_file = NULL;
1257 	uint8_t *base_buf = NULL, *accum_buf = NULL;
1258 	size_t accum_size = 0;
1259 	uint64_t max_size;
1260 	int n = 0;
1261 
1262 	*result_size = 0;
1263 
1264 	if (SIMPLEQ_EMPTY(&deltas->entries))
1265 		return got_error(GOT_ERR_BAD_DELTA_CHAIN);
1266 
1267 	/* We process small enough files entirely in memory for speed. */
1268 	err = get_delta_chain_max_size(&max_size, deltas);
1269 	if (err)
1270 		return err;
1271 	if (max_size < GOT_DELTA_RESULT_SIZE_CACHED_MAX) {
1272 		accum_buf = malloc(max_size);
1273 		if (accum_buf == NULL)
1274 			return got_error_from_errno();
1275 	} else {
1276 		base_file = got_opentemp();
1277 		if (base_file == NULL)
1278 			return got_error_from_errno();
1279 
1280 		accum_file = got_opentemp();
1281 		if (accum_file == NULL) {
1282 			err = got_error_from_errno();
1283 			fclose(base_file);
1284 			return err;
1285 		}
1286 	}
1287 
1288 	/* Deltas are ordered in ascending order. */
1289 	SIMPLEQ_FOREACH(delta, &deltas->entries, entry) {
1290 		if (n == 0) {
1291 			struct got_pack *pack;
1292 			size_t base_len, mapoff;
1293 			off_t delta_data_offset;
1294 
1295 			/* Plain object types are the delta base. */
1296 			if (delta->type != GOT_OBJ_TYPE_COMMIT &&
1297 			    delta->type != GOT_OBJ_TYPE_TREE &&
1298 			    delta->type != GOT_OBJ_TYPE_BLOB &&
1299 			    delta->type != GOT_OBJ_TYPE_TAG) {
1300 				err = got_error(GOT_ERR_BAD_DELTA_CHAIN);
1301 				goto done;
1302 			}
1303 
1304 			pack = get_cached_pack(path_packfile, repo);
1305 			if (pack == NULL) {
1306 				err = got_error(GOT_ERR_BAD_DELTA_CHAIN);
1307 				goto done;
1308 			}
1309 
1310 			delta_data_offset = delta->offset + delta->tslen;
1311 			if (delta_data_offset >= pack->filesize) {
1312 				err = got_error(GOT_ERR_PACK_OFFSET);
1313 				goto done;
1314 			}
1315 			if (pack->map == NULL) {
1316 				if (lseek(pack->fd, delta_data_offset, SEEK_SET)
1317 				    == -1) {
1318 					err = got_error_from_errno();
1319 					goto done;
1320 				}
1321 			}
1322 			if (base_file) {
1323 				if (pack->map) {
1324 					mapoff = (size_t)delta_data_offset;
1325 					err = got_inflate_to_file_mmap(
1326 					    &base_len, pack->map, mapoff,
1327 					    pack->filesize - mapoff, base_file);
1328 				} else
1329 					err = got_inflate_to_file_fd(&base_len,
1330 					    pack->fd, base_file);
1331 			} else {
1332 				if (pack->map) {
1333 					mapoff = (size_t)delta_data_offset;
1334 					err = got_inflate_to_mem_mmap(&base_buf,
1335 					    &base_len, pack->map, mapoff,
1336 					    pack->filesize - mapoff);
1337 				} else
1338 					err = got_inflate_to_mem_fd(&base_buf,
1339 					    &base_len, pack->fd);
1340 				if (base_len < max_size) {
1341 					uint8_t *p;
1342 					p = reallocarray(base_buf, 1, max_size);
1343 					if (p == NULL) {
1344 						err = got_error_from_errno();
1345 						goto done;
1346 					}
1347 					base_buf = p;
1348 				}
1349 			}
1350 			if (err)
1351 				goto done;
1352 			n++;
1353 			if (base_file)
1354 				rewind(base_file);
1355 			continue;
1356 		}
1357 
1358 		if (base_buf) {
1359 			err = got_delta_apply_in_mem(base_buf, delta->delta_buf,
1360 			    delta->delta_len, accum_buf, &accum_size);
1361 			n++;
1362 		} else {
1363 			err = got_delta_apply(base_file, delta->delta_buf,
1364 			    delta->delta_len,
1365 			    /* Final delta application writes to output file. */
1366 			    ++n < deltas->nentries ? accum_file : outfile,
1367 			    &accum_size);
1368 		}
1369 		if (err)
1370 			goto done;
1371 
1372 		if (n < deltas->nentries) {
1373 			/* Accumulated delta becomes the new base. */
1374 			if (base_buf) {
1375 				uint8_t *tmp = accum_buf;
1376 				accum_buf = base_buf;
1377 				base_buf = tmp;
1378 			} else {
1379 				FILE *tmp = accum_file;
1380 				accum_file = base_file;
1381 				base_file = tmp;
1382 				rewind(base_file);
1383 				rewind(accum_file);
1384 			}
1385 		}
1386 	}
1387 
1388 done:
1389 	free(base_buf);
1390 	if (accum_buf) {
1391 		size_t len = fwrite(accum_buf, 1, accum_size, outfile);
1392 		free(accum_buf);
1393 		if (len != accum_size)
1394 			err = got_ferror(outfile, GOT_ERR_IO);
1395 	}
1396 	if (base_file)
1397 		fclose(base_file);
1398 	if (accum_file)
1399 		fclose(accum_file);
1400 	rewind(outfile);
1401 	if (err == NULL)
1402 		*result_size = accum_size;
1403 	return err;
1404 }
1405 
1406 static const struct got_error *
1407 dump_delta_chain_to_mem(uint8_t **outbuf, size_t *outlen,
1408     struct got_delta_chain *deltas, const char *path_packfile,
1409     struct got_repository *repo)
1410 {
1411 	const struct got_error *err = NULL;
1412 	struct got_delta *delta;
1413 	uint8_t *base_buf = NULL, *accum_buf = NULL;
1414 	size_t accum_size;
1415 	uint64_t max_size;
1416 	int n = 0;
1417 
1418 	*outbuf = NULL;
1419 	*outlen = 0;
1420 
1421 	if (SIMPLEQ_EMPTY(&deltas->entries))
1422 		return got_error(GOT_ERR_BAD_DELTA_CHAIN);
1423 
1424 	err = get_delta_chain_max_size(&max_size, deltas);
1425 	if (err)
1426 		return err;
1427 	accum_buf = malloc(max_size);
1428 	if (accum_buf == NULL)
1429 		return got_error_from_errno();
1430 
1431 	/* Deltas are ordered in ascending order. */
1432 	SIMPLEQ_FOREACH(delta, &deltas->entries, entry) {
1433 		if (n == 0) {
1434 			struct got_pack *pack;
1435 			size_t base_len;
1436 			size_t delta_data_offset;
1437 
1438 			/* Plain object types are the delta base. */
1439 			if (delta->type != GOT_OBJ_TYPE_COMMIT &&
1440 			    delta->type != GOT_OBJ_TYPE_TREE &&
1441 			    delta->type != GOT_OBJ_TYPE_BLOB &&
1442 			    delta->type != GOT_OBJ_TYPE_TAG) {
1443 				err = got_error(GOT_ERR_BAD_DELTA_CHAIN);
1444 				goto done;
1445 			}
1446 
1447 			pack = get_cached_pack(path_packfile, repo);
1448 			if (pack == NULL) {
1449 				err = got_error(GOT_ERR_BAD_DELTA_CHAIN);
1450 				goto done;
1451 			}
1452 
1453 			delta_data_offset = delta->offset + delta->tslen;
1454 			if (delta_data_offset >= pack->filesize) {
1455 				err = got_error(GOT_ERR_PACK_OFFSET);
1456 				goto done;
1457 			}
1458 			if (pack->map) {
1459 				size_t mapoff = (size_t)delta_data_offset;
1460 				err = got_inflate_to_mem_mmap(&base_buf,
1461 				    &base_len, pack->map, mapoff,
1462 				    pack->filesize - mapoff);
1463 			} else {
1464 				if (lseek(pack->fd, delta_data_offset, SEEK_SET)
1465 				    == -1) {
1466 					err = got_error_from_errno();
1467 					goto done;
1468 				}
1469 				err = got_inflate_to_mem_fd(&base_buf,
1470 				    &base_len, pack->fd);
1471 			}
1472 			if (err)
1473 				goto done;
1474 			if (base_len < max_size) {
1475 				uint8_t *p;
1476 				p = reallocarray(base_buf, 1, max_size);
1477 				if (p == NULL) {
1478 					err = got_error_from_errno();
1479 					goto done;
1480 				}
1481 				base_buf = p;
1482 			}
1483 			n++;
1484 			continue;
1485 		}
1486 
1487 		err = got_delta_apply_in_mem(base_buf, delta->delta_buf,
1488 		    delta->delta_len, accum_buf, &accum_size);
1489 		n++;
1490 		if (err)
1491 			goto done;
1492 
1493 		if (n < deltas->nentries) {
1494 			/* Accumulated delta becomes the new base. */
1495 			uint8_t *tmp = accum_buf;
1496 			accum_buf = base_buf;
1497 			base_buf = tmp;
1498 		}
1499 	}
1500 
1501 done:
1502 	free(base_buf);
1503 	if (err) {
1504 		free(accum_buf);
1505 		*outbuf = NULL;
1506 		*outlen = 0;
1507 	} else {
1508 		*outbuf = accum_buf;
1509 		*outlen = accum_size;
1510 	}
1511 	return err;
1512 }
1513 
1514 const struct got_error *
1515 got_packfile_extract_object(FILE **f, struct got_object *obj,
1516     struct got_repository *repo)
1517 {
1518 	const struct got_error *err = NULL;
1519 
1520 	*f = NULL;
1521 
1522 	if ((obj->flags & GOT_OBJ_FLAG_PACKED) == 0)
1523 		return got_error(GOT_ERR_OBJ_NOT_PACKED);
1524 
1525 	*f = got_opentemp();
1526 	if (*f == NULL) {
1527 		err = got_error_from_errno();
1528 		goto done;
1529 	}
1530 
1531 	if ((obj->flags & GOT_OBJ_FLAG_DELTIFIED) == 0) {
1532 		struct got_pack *pack;
1533 
1534 		pack = get_cached_pack(obj->path_packfile, repo);
1535 		if (pack == NULL) {
1536 			err = cache_pack(&pack, obj->path_packfile, NULL, repo);
1537 			if (err)
1538 				goto done;
1539 		}
1540 
1541 		if (obj->pack_offset >= pack->filesize) {
1542 			err = got_error(GOT_ERR_PACK_OFFSET);
1543 			goto done;
1544 		}
1545 
1546 		if (pack->map) {
1547 			size_t mapoff = (size_t)obj->pack_offset;
1548 			err = got_inflate_to_file_mmap(&obj->size, pack->map,
1549 			    mapoff, pack->filesize - mapoff, *f);
1550 		} else {
1551 			if (lseek(pack->fd, obj->pack_offset, SEEK_SET) == -1) {
1552 				err = got_error_from_errno();
1553 				goto done;
1554 			}
1555 			err = got_inflate_to_file_fd(&obj->size, pack->fd, *f);
1556 		}
1557 	} else
1558 		err = dump_delta_chain_to_file(&obj->size,
1559 		    &obj->deltas, obj->path_packfile, *f, repo);
1560 done:
1561 	if (err && *f) {
1562 		fclose(*f);
1563 		*f = NULL;
1564 	}
1565 	return err;
1566 }
1567 
1568 const struct got_error *
1569 got_packfile_extract_object_to_mem(uint8_t **buf, size_t *len,
1570     struct got_object *obj, struct got_repository *repo)
1571 {
1572 	const struct got_error *err = NULL;
1573 
1574 	if ((obj->flags & GOT_OBJ_FLAG_PACKED) == 0)
1575 		return got_error(GOT_ERR_OBJ_NOT_PACKED);
1576 
1577 	if ((obj->flags & GOT_OBJ_FLAG_DELTIFIED) == 0) {
1578 		struct got_pack *pack;
1579 
1580 		pack = get_cached_pack(obj->path_packfile, repo);
1581 		if (pack == NULL) {
1582 			err = cache_pack(&pack, obj->path_packfile, NULL, repo);
1583 			if (err)
1584 				goto done;
1585 		}
1586 
1587 		if (obj->pack_offset >= pack->filesize) {
1588 			err = got_error(GOT_ERR_PACK_OFFSET);
1589 			goto done;
1590 		}
1591 		if (pack->map) {
1592 			size_t mapoff = (size_t)obj->pack_offset;
1593 			err = got_inflate_to_mem_mmap(buf, len, pack->map,
1594 			    mapoff, pack->filesize - mapoff);
1595 		} else {
1596 			if (lseek(pack->fd, obj->pack_offset, SEEK_SET) == -1) {
1597 				err = got_error_from_errno();
1598 				goto done;
1599 			}
1600 			err = got_inflate_to_mem_fd(buf, len, pack->fd);
1601 		}
1602 	} else
1603 		err = dump_delta_chain_to_mem(buf, len, &obj->deltas,
1604 		    obj->path_packfile, repo);
1605 done:
1606 	return err;
1607 }