op public repos

Blob

Date:: Tue Nov 1 07:41:39 2022 UTC
Message:: use mkstemps(3) instead of mkstemp(3) for opening named temporary files Allows 'got commit' to use a ".diff" suffix for temporary diff files. ok op@
Actions:: History | Blame | Raw File
1 /*
2  * Copyright (c) 2018, 2019 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <sys/types.h>
18 #include <sys/queue.h>
19 #include <sys/stat.h>
20 
21 #include <errno.h>
22 #include <dirent.h>
23 #include <limits.h>
24 #include <sha1.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <unistd.h>
29 #include <util.h>
30 #include <zlib.h>
31 #include <time.h>
32 #include <libgen.h>
33 
34 #include "got_error.h"
35 #include "got_object.h"
36 #include "got_repository.h"
37 #include "got_reference.h"
38 #include "got_opentemp.h"
39 #include "got_path.h"
40 
41 #include "got_lib_sha1.h"
42 #include "got_lib_delta.h"
43 #include "got_lib_inflate.h"
44 #include "got_lib_object.h"
45 #include "got_lib_object_idset.h"
46 #include "got_lib_lockfile.h"
47 
48 #ifndef nitems
49 #define nitems(_a) (sizeof(_a) / sizeof((_a)[0]))
50 #endif
51 
52 #define GOT_REF_HEADS	"heads"
53 #define GOT_REF_TAGS	"tags"
54 #define GOT_REF_REMOTES	"remotes"
55 
56 /*
57  * We do not resolve tags yet, and don't yet care about sorting refs either,
58  * so packed-refs files we write contain a minimal header which disables all
59  * packed-refs "traits" supported by Git.
60  */
61 #define GOT_PACKED_REFS_HEADER	"# pack-refs with:"
62 
63 /* A symbolic reference. */
64 struct got_symref {
65 	char *name;
66 	char *ref;
67 };
68 
69 #define GOT_REF_RECURSE_MAX	20
70 
71 /* A non-symbolic reference (there is no better designation). */
72 struct got_ref {
73 	char *name;
74 	u_int8_t sha1[SHA1_DIGEST_LENGTH];
75 };
76 
77 /* A reference which points to an arbitrary object. */
78 struct got_reference {
79 	unsigned int flags;
80 #define GOT_REF_IS_SYMBOLIC	0x01
81 #define GOT_REF_IS_PACKED	0x02
82 
83 	union {
84 		struct got_ref ref;
85 		struct got_symref symref;
86 	} ref;
87 
88 	struct got_lockfile *lf;
89 	time_t mtime;
90 
91 	/* Cached timestamp for got_ref_cmp_by_commit_timestamp_descending() */
92 	time_t committer_time;
93 };
94 
95 static const struct got_error *
96 alloc_ref(struct got_reference **ref, const char *name,
97     struct got_object_id *id, int flags, time_t mtime)
98 {
99 	const struct got_error *err = NULL;
100 
101 	*ref = calloc(1, sizeof(**ref));
102 	if (*ref == NULL)
103 		return got_error_from_errno("calloc");
104 
105 	memcpy((*ref)->ref.ref.sha1, id->sha1, sizeof((*ref)->ref.ref.sha1));
106 	(*ref)->flags = flags;
107 	(*ref)->ref.ref.name = strdup(name);
108 	(*ref)->mtime = mtime;
109 	if ((*ref)->ref.ref.name == NULL) {
110 		err = got_error_from_errno("strdup");
111 		got_ref_close(*ref);
112 		*ref = NULL;
113 	}
114 	return err;
115 }
116 
117 static const struct got_error *
118 alloc_symref(struct got_reference **ref, const char *name,
119     const char *target_ref, int flags)
120 {
121 	const struct got_error *err = NULL;
122 
123 	*ref = calloc(1, sizeof(**ref));
124 	if (*ref == NULL)
125 		return got_error_from_errno("calloc");
126 
127 	(*ref)->flags = GOT_REF_IS_SYMBOLIC | flags;
128 	(*ref)->ref.symref.name = strdup(name);
129 	if ((*ref)->ref.symref.name == NULL) {
130 		err = got_error_from_errno("strdup");
131 		got_ref_close(*ref);
132 		*ref = NULL;
133 		return err;
134 	}
135 	(*ref)->ref.symref.ref = strdup(target_ref);
136 	if ((*ref)->ref.symref.ref == NULL) {
137 		err = got_error_from_errno("strdup");
138 		got_ref_close(*ref);
139 		*ref = NULL;
140 	}
141 	return err;
142 }
143 
144 static const struct got_error *
145 parse_symref(struct got_reference **ref, const char *name, const char *line)
146 {
147 	if (line[0] == '\0')
148 		return got_error(GOT_ERR_BAD_REF_DATA);
149 
150 	return alloc_symref(ref, name, line, 0);
151 }
152 
153 static const struct got_error *
154 parse_ref_line(struct got_reference **ref, const char *name, const char *line,
155     time_t mtime)
156 {
157 	struct got_object_id id;
158 
159 	if (strncmp(line, "ref: ", 5) == 0) {
160 		line += 5;
161 		return parse_symref(ref, name, line);
162 	}
163 
164 	if (!got_parse_sha1_digest(id.sha1, line))
165 		return got_error(GOT_ERR_BAD_REF_DATA);
166 
167 	return alloc_ref(ref, name, &id, 0, mtime);
168 }
169 
170 static const struct got_error *
171 parse_ref_file(struct got_reference **ref, const char *name,
172     const char *absname, const char *abspath, int lock)
173 {
174 	const struct got_error *err = NULL;
175 	FILE *f;
176 	char *line = NULL;
177 	size_t linesize = 0;
178 	ssize_t linelen;
179 	struct got_lockfile *lf = NULL;
180 	struct stat sb;
181 
182 	if (lock) {
183 		err = got_lockfile_lock(&lf, abspath, -1);
184 		if (err) {
185 			if (err->code == GOT_ERR_ERRNO && errno == ENOENT)
186 				err = got_error_not_ref(name);
187 			return err;
188 		}
189 	}
190 
191 	f = fopen(abspath, "rbe");
192 	if (f == NULL) {
193 		if (errno != ENOTDIR && errno != ENOENT)
194 			err = got_error_from_errno2("fopen", abspath);
195 		else
196 			err = got_error_not_ref(name);
197 		if (lock)
198 			got_lockfile_unlock(lf, -1);
199 		return err;
200 	}
201 	if (fstat(fileno(f), &sb) == -1) {
202 		err = got_error_from_errno2("fstat", abspath);
203 		goto done;
204 	}
205 
206 	linelen = getline(&line, &linesize, f);
207 	if (linelen == -1) {
208 		if (feof(f))
209 			err = NULL; /* ignore empty files (could be locks) */
210 		else {
211 			if (errno == EISDIR)
212 				err = got_error(GOT_ERR_NOT_REF);
213 			else if (ferror(f))
214 				err = got_ferror(f, GOT_ERR_IO);
215 			else
216 				err = got_error_from_errno2("getline", abspath);
217 		}
218 		if (lock)
219 			got_lockfile_unlock(lf, -1);
220 		goto done;
221 	}
222 	while (linelen > 0 && line[linelen - 1] == '\n') {
223 		line[linelen - 1] = '\0';
224 		linelen--;
225 	}
226 
227 	err = parse_ref_line(ref, absname, line, sb.st_mtime);
228 	if (lock) {
229 		if (err)
230 			got_lockfile_unlock(lf, -1);
231 		else {
232 			if (*ref)
233 				(*ref)->lf = lf;
234 			else
235 				got_lockfile_unlock(lf, -1);
236 		}
237 	}
238 done:
239 	free(line);
240 	if (fclose(f) == EOF && err == NULL) {
241 		err = got_error_from_errno("fclose");
242 		if (*ref) {
243 			if (lock)
244 				got_ref_unlock(*ref);
245 			got_ref_close(*ref);
246 			*ref = NULL;
247 		}
248 	}
249 	return err;
250 }
251 
252 static int
253 is_well_known_ref(const char *refname)
254 {
255 	return (strcmp(refname, GOT_REF_HEAD) == 0 ||
256 	    strcmp(refname, GOT_REF_ORIG_HEAD) == 0 ||
257 	    strcmp(refname, GOT_REF_MERGE_HEAD) == 0 ||
258 	    strcmp(refname, GOT_REF_FETCH_HEAD) == 0);
259 }
260 
261 static char *
262 get_refs_dir_path(struct got_repository *repo, const char *refname)
263 {
264 	if (is_well_known_ref(refname) || strncmp(refname, "refs/", 5) == 0)
265 		return strdup(got_repo_get_path_git_dir(repo));
266 
267 	return got_repo_get_path_refs(repo);
268 }
269 
270 const struct got_error *
271 got_ref_alloc(struct got_reference **ref, const char *name,
272     struct got_object_id *id)
273 {
274 	if (!got_ref_name_is_valid(name))
275 		return got_error_path(name, GOT_ERR_BAD_REF_NAME);
276 
277 	return alloc_ref(ref, name, id, 0, 0);
278 }
279 
280 const struct got_error *
281 got_ref_alloc_symref(struct got_reference **ref, const char *name,
282 	struct got_reference *target_ref)
283 {
284 	if (!got_ref_name_is_valid(name))
285 		return got_error_path(name, GOT_ERR_BAD_REF_NAME);
286 
287 	return alloc_symref(ref, name, got_ref_get_name(target_ref), 0);
288 }
289 
290 static const struct got_error *
291 parse_packed_ref_line(struct got_reference **ref, const char *abs_refname,
292     const char *line, time_t mtime)
293 {
294 	struct got_object_id id;
295 	const char *name;
296 
297 	*ref = NULL;
298 
299 	if (line[0] == '#' || line[0] == '^')
300 		return NULL;
301 
302 	if (!got_parse_sha1_digest(id.sha1, line))
303 		return got_error(GOT_ERR_BAD_REF_DATA);
304 
305 	if (abs_refname) {
306 		if (strcmp(line + SHA1_DIGEST_STRING_LENGTH, abs_refname) != 0)
307 			return NULL;
308 		name = abs_refname;
309 	} else
310 		name = line + SHA1_DIGEST_STRING_LENGTH;
311 
312 	return alloc_ref(ref, name, &id, GOT_REF_IS_PACKED, mtime);
313 }
314 
315 static const struct got_error *
316 open_packed_ref(struct got_reference **ref, FILE *f, const char **subdirs,
317     int nsubdirs, const char *refname, time_t mtime)
318 {
319 	const struct got_error *err = NULL;
320 	char *abs_refname;
321 	char *line = NULL;
322 	size_t linesize = 0;
323 	ssize_t linelen;
324 	int i, ref_is_absolute = (strncmp(refname, "refs/", 5) == 0);
325 
326 	*ref = NULL;
327 
328 	if (ref_is_absolute)
329 		abs_refname = (char *)refname;
330 	do {
331 		linelen = getline(&line, &linesize, f);
332 		if (linelen == -1) {
333 			if (feof(f))
334 				break;
335 			err = got_ferror(f, GOT_ERR_BAD_REF_DATA);
336 			break;
337 		}
338 		if (linelen > 0 && line[linelen - 1] == '\n')
339 			line[linelen - 1] = '\0';
340 		for (i = 0; i < nsubdirs; i++) {
341 			if (!ref_is_absolute &&
342 			    asprintf(&abs_refname, "refs/%s/%s", subdirs[i],
343 			    refname) == -1)
344 				return got_error_from_errno("asprintf");
345 			err = parse_packed_ref_line(ref, abs_refname, line,
346 			    mtime);
347 			if (!ref_is_absolute)
348 				free(abs_refname);
349 			if (err || *ref != NULL)
350 				break;
351 		}
352 		if (err)
353 			break;
354 	} while (*ref == NULL);
355 	free(line);
356 
357 	return err;
358 }
359 
360 static const struct got_error *
361 open_ref(struct got_reference **ref, const char *path_refs, const char *subdir,
362     const char *name, int lock)
363 {
364 	const struct got_error *err = NULL;
365 	char *path = NULL;
366 	char *absname = NULL;
367 	int ref_is_absolute = (strncmp(name, "refs/", 5) == 0);
368 	int ref_is_well_known = (subdir[0] == '\0' && is_well_known_ref(name));
369 
370 	*ref = NULL;
371 
372 	if (!got_ref_name_is_valid(name))
373 		return got_error_path(name, GOT_ERR_BAD_REF_NAME);
374 
375 	if (ref_is_absolute || ref_is_well_known) {
376 		if (asprintf(&path, "%s/%s", path_refs, name) == -1)
377 			return got_error_from_errno("asprintf");
378 		absname = (char *)name;
379 	} else {
380 		if (asprintf(&path, "%s/%s%s%s", path_refs, subdir,
381 		    subdir[0] ? "/" : "", name) == -1)
382 			return got_error_from_errno("asprintf");
383 
384 		if (asprintf(&absname, "refs/%s%s%s",
385 		    subdir, subdir[0] ? "/" : "", name) == -1) {
386 			err = got_error_from_errno("asprintf");
387 			goto done;
388 		}
389 	}
390 
391 	err = parse_ref_file(ref, name, absname, path, lock);
392 done:
393 	if (!ref_is_absolute && !ref_is_well_known)
394 		free(absname);
395 	free(path);
396 	return err;
397 }
398 
399 const struct got_error *
400 got_ref_open(struct got_reference **ref, struct got_repository *repo,
401     const char *refname, int lock)
402 {
403 	const struct got_error *err = NULL;
404 	char *packed_refs_path = NULL, *path_refs = NULL;
405 	const char *subdirs[] = {
406 	    GOT_REF_HEADS, GOT_REF_TAGS, GOT_REF_REMOTES
407 	};
408 	size_t i;
409 	int well_known = is_well_known_ref(refname);
410 	struct got_lockfile *lf = NULL;
411 
412 	*ref = NULL;
413 
414 	path_refs = get_refs_dir_path(repo, refname);
415 	if (path_refs == NULL) {
416 		err = got_error_from_errno2("get_refs_dir_path", refname);
417 		goto done;
418 	}
419 
420 	if (well_known) {
421 		err = open_ref(ref, path_refs, "", refname, lock);
422 	} else {
423 		FILE *f;
424 
425 		/* Search on-disk refs before packed refs! */
426 		for (i = 0; i < nitems(subdirs); i++) {
427 			err = open_ref(ref, path_refs, subdirs[i], refname,
428 			    lock);
429 			if ((err && err->code != GOT_ERR_NOT_REF) || *ref)
430 				goto done;
431 		}
432 
433 		packed_refs_path = got_repo_get_path_packed_refs(repo);
434 		if (packed_refs_path == NULL) {
435 			err = got_error_from_errno(
436 			    "got_repo_get_path_packed_refs");
437 			goto done;
438 		}
439 
440 		if (lock) {
441 			err = got_lockfile_lock(&lf, packed_refs_path, -1);
442 			if (err)
443 				goto done;
444 		}
445 		f = fopen(packed_refs_path, "rbe");
446 		if (f != NULL) {
447 			struct stat sb;
448 			if (fstat(fileno(f), &sb) == -1) {
449 				err = got_error_from_errno2("fstat",
450 				    packed_refs_path);
451 				goto done;
452 			}
453 			err = open_packed_ref(ref, f, subdirs, nitems(subdirs),
454 			    refname, sb.st_mtime);
455 			if (!err) {
456 				if (fclose(f) == EOF) {
457 					err = got_error_from_errno("fclose");
458 					got_ref_close(*ref);
459 					*ref = NULL;
460 				} else if (*ref)
461 					(*ref)->lf = lf;
462 			}
463 		}
464 	}
465 done:
466 	if (!err && *ref == NULL)
467 		err = got_error_not_ref(refname);
468 	if (err && lf)
469 		got_lockfile_unlock(lf, -1);
470 	free(packed_refs_path);
471 	free(path_refs);
472 	return err;
473 }
474 
475 void
476 got_ref_close(struct got_reference *ref)
477 {
478 	if (ref->flags & GOT_REF_IS_SYMBOLIC) {
479 		free(ref->ref.symref.name);
480 		free(ref->ref.symref.ref);
481 	} else
482 		free(ref->ref.ref.name);
483 	free(ref);
484 }
485 
486 struct got_reference *
487 got_ref_dup(struct got_reference *ref)
488 {
489 	struct got_reference *ret;
490 
491 	ret = calloc(1, sizeof(*ret));
492 	if (ret == NULL)
493 		return NULL;
494 
495 	ret->flags = ref->flags;
496 	if (ref->flags & GOT_REF_IS_SYMBOLIC) {
497 		ret->ref.symref.name = strdup(ref->ref.symref.name);
498 		if (ret->ref.symref.name == NULL) {
499 			free(ret);
500 			return NULL;
501 		}
502 		ret->ref.symref.ref = strdup(ref->ref.symref.ref);
503 		if (ret->ref.symref.ref == NULL) {
504 			free(ret->ref.symref.name);
505 			free(ret);
506 			return NULL;
507 		}
508 	} else {
509 		ret->ref.ref.name = strdup(ref->ref.ref.name);
510 		if (ret->ref.ref.name == NULL) {
511 			free(ret);
512 			return NULL;
513 		}
514 		memcpy(ret->ref.ref.sha1, ref->ref.ref.sha1,
515 		    sizeof(ret->ref.ref.sha1));
516 	}
517 
518 	return ret;
519 }
520 
521 const struct got_error *
522 got_reflist_entry_dup(struct got_reflist_entry **newp,
523     struct got_reflist_entry *re)
524 {
525 	const struct got_error *err = NULL;
526 	struct got_reflist_entry *new;
527 
528 	*newp = NULL;
529 
530 	new = malloc(sizeof(*new));
531 	if (new == NULL)
532 		return got_error_from_errno("malloc");
533 
534 	new->ref = got_ref_dup(re->ref);
535 	if (new->ref == NULL) {
536 		err = got_error_from_errno("got_ref_dup");
537 		free(new);
538 		return err;
539 	}
540 
541 	*newp = new;
542 	return NULL;
543 }
544 
545 const struct got_error *
546 got_ref_resolve_symbolic(struct got_reference **resolved,
547     struct got_repository *repo, struct got_reference *ref)
548 {
549 	struct got_reference *nextref;
550 	const struct got_error *err;
551 
552 	err = got_ref_open(&nextref, repo, ref->ref.symref.ref, 0);
553 	if (err)
554 		return err;
555 
556 	if (nextref->flags & GOT_REF_IS_SYMBOLIC)
557 		err = got_ref_resolve_symbolic(resolved, repo, nextref);
558 	else
559 		*resolved = got_ref_dup(nextref);
560 
561 	got_ref_close(nextref);
562 	return err;
563 }
564 
565 static const struct got_error *
566 ref_resolve(struct got_object_id **id, struct got_repository *repo,
567     struct got_reference *ref, int recursion)
568 {
569 	const struct got_error *err;
570 
571 	if (recursion <= 0)
572 		return got_error_msg(GOT_ERR_RECURSION,
573 		    "reference recursion limit reached");
574 
575 	if (ref->flags & GOT_REF_IS_SYMBOLIC) {
576 		struct got_reference *resolved = NULL;
577 		err = got_ref_resolve_symbolic(&resolved, repo, ref);
578 		if (err == NULL)
579 			err = ref_resolve(id, repo, resolved, --recursion);
580 		if (resolved)
581 			got_ref_close(resolved);
582 		return err;
583 	}
584 
585 	*id = calloc(1, sizeof(**id));
586 	if (*id == NULL)
587 		return got_error_from_errno("calloc");
588 	memcpy((*id)->sha1, ref->ref.ref.sha1, sizeof((*id)->sha1));
589 	return NULL;
590 }
591 
592 const struct got_error *
593 got_ref_resolve(struct got_object_id **id, struct got_repository *repo,
594     struct got_reference *ref)
595 {
596 	return ref_resolve(id, repo, ref, GOT_REF_RECURSE_MAX);
597 }
598 
599 char *
600 got_ref_to_str(struct got_reference *ref)
601 {
602 	char *str;
603 
604 	if (ref->flags & GOT_REF_IS_SYMBOLIC)
605 		return strdup(ref->ref.symref.ref);
606 
607 	str = malloc(SHA1_DIGEST_STRING_LENGTH);
608 	if (str == NULL)
609 		return NULL;
610 
611 	if (got_sha1_digest_to_str(ref->ref.ref.sha1, str,
612 	    SHA1_DIGEST_STRING_LENGTH) == NULL) {
613 		free(str);
614 		return NULL;
615 	}
616 
617 	return str;
618 }
619 
620 const char *
621 got_ref_get_name(struct got_reference *ref)
622 {
623 	if (ref->flags & GOT_REF_IS_SYMBOLIC)
624 		return ref->ref.symref.name;
625 
626 	return ref->ref.ref.name;
627 }
628 
629 const char *
630 got_ref_get_symref_target(struct got_reference *ref)
631 {
632 	if (ref->flags & GOT_REF_IS_SYMBOLIC)
633 		return ref->ref.symref.ref;
634 
635 	return NULL;
636 }
637 
638 time_t
639 got_ref_get_mtime(struct got_reference *ref)
640 {
641 	return ref->mtime;
642 }
643 
644 const struct got_error *
645 got_ref_cmp_by_name(void *arg, int *cmp, struct got_reference *re1,
646     struct got_reference* re2)
647 {
648 	const char *name1 = got_ref_get_name(re1);
649 	const char *name2 = got_ref_get_name(re2);
650 
651 	*cmp = got_path_cmp(name1, name2, strlen(name1), strlen(name2));
652 	return NULL;
653 }
654 
655 const struct got_error *
656 got_ref_cmp_tags(void *arg, int *cmp, struct got_reference *ref1,
657     struct got_reference *ref2)
658 {
659 	const struct got_error *err = NULL;
660 	struct got_repository *repo = arg;
661 	struct got_object_id *id1, *id2 = NULL;
662 	struct got_tag_object *tag1 = NULL, *tag2 = NULL;
663 	struct got_commit_object *commit1 = NULL, *commit2 = NULL;
664 	time_t time1, time2;
665 
666 	*cmp = 0;
667 
668 	err = got_ref_resolve(&id1, repo, ref1);
669 	if (err)
670 		return err;
671 	err = got_object_open_as_tag(&tag1, repo, id1);
672 	if (err) {
673 		if (err->code != GOT_ERR_OBJ_TYPE)
674 			goto done;
675 		/* "lightweight" tag */
676 		err = got_object_open_as_commit(&commit1, repo, id1);
677 		if (err)
678 			goto done;
679 		time1 = got_object_commit_get_committer_time(commit1);
680 	} else
681 		time1 = got_object_tag_get_tagger_time(tag1);
682 
683 	err = got_ref_resolve(&id2, repo, ref2);
684 	if (err)
685 		goto done;
686 	err = got_object_open_as_tag(&tag2, repo, id2);
687 	if (err) {
688 		if (err->code != GOT_ERR_OBJ_TYPE)
689 			goto done;
690 		/* "lightweight" tag */
691 		err = got_object_open_as_commit(&commit2, repo, id2);
692 		if (err)
693 			goto done;
694 		time2 = got_object_commit_get_committer_time(commit2);
695 	} else
696 		time2 = got_object_tag_get_tagger_time(tag2);
697 
698 	/* Put latest tags first. */
699 	if (time1 < time2)
700 		*cmp = 1;
701 	else if (time1 > time2)
702 		*cmp = -1;
703 	else
704 		err = got_ref_cmp_by_name(NULL, cmp, ref2, ref1);
705 done:
706 	free(id1);
707 	free(id2);
708 	if (tag1)
709 		got_object_tag_close(tag1);
710 	if (tag2)
711 		got_object_tag_close(tag2);
712 	if (commit1)
713 		got_object_commit_close(commit1);
714 	if (commit2)
715 		got_object_commit_close(commit2);
716 	return err;
717 }
718 
719 static const struct got_error *
720 get_committer_time(struct got_reference *ref, struct got_repository *repo)
721 {
722 	const struct got_error *err = NULL;
723 	int obj_type;
724 	struct got_commit_object *commit = NULL;
725 	struct got_tag_object *tag = NULL;
726 	struct got_object_id *id = NULL;
727 
728 	err = got_ref_resolve(&id, repo, ref);
729 	if (err)
730 		return err;
731 
732 	err = got_object_get_type(&obj_type, repo, id);
733 	if (err)
734 		goto done;
735 
736 	switch (obj_type) {
737 	case GOT_OBJ_TYPE_COMMIT:
738 		err = got_object_open_as_commit(&commit, repo, id);
739 		if (err)
740 			goto done;
741 		ref->committer_time =
742 		    got_object_commit_get_committer_time(commit);
743 		break;
744 	case GOT_OBJ_TYPE_TAG:
745 		err = got_object_open_as_tag(&tag, repo, id);
746 		if (err)
747 			goto done;
748 		ref->committer_time = got_object_tag_get_tagger_time(tag);
749 		break;
750 	default:
751 		/* best effort for other object types */
752 		ref->committer_time = got_ref_get_mtime(ref);
753 		break;
754 	}
755 done:
756 	free(id);
757 	if (commit)
758 		got_object_commit_close(commit);
759 	if (tag)
760 		got_object_tag_close(tag);
761 	return err;
762 }
763 
764 const struct got_error *
765 got_ref_cmp_by_commit_timestamp_descending(void *arg, int *cmp,
766     struct got_reference *ref1, struct got_reference *ref2)
767 {
768 	const struct got_error *err = NULL;
769 	struct got_repository *repo = arg;
770 
771 	*cmp = 0;
772 
773 	if (ref1->committer_time == 0) {
774 		err = get_committer_time(ref1, repo);
775 		if (err)
776 			return err;
777 	}
778 	if (ref2->committer_time == 0) {
779 		err = get_committer_time(ref2, repo);
780 		if (err)
781 			return err;
782 	}
783 
784 	if (ref1->committer_time < ref2->committer_time)
785 		*cmp = 1;
786 	else if (ref2->committer_time < ref1->committer_time)
787 		*cmp = -1;
788 	else
789 		return got_ref_cmp_by_name(arg, cmp, ref1, ref2);
790 
791 	return err;
792 }
793 
794 const struct got_error *
795 got_reflist_insert(struct got_reflist_entry **newp,
796     struct got_reflist_head *refs, struct got_reference *ref,
797     got_ref_cmp_cb cmp_cb, void *cmp_arg)
798 {
799 	const struct got_error *err;
800 	struct got_reflist_entry *new, *re;
801 	int cmp;
802 
803 	*newp = NULL;
804 
805 	if (cmp_cb != got_ref_cmp_by_name && (ref->flags & GOT_REF_IS_PACKED)) {
806 		/*
807 		 * If we are not sorting elements by name then we must still
808 		 * detect collisions between a packed ref and an on-disk ref
809 		 * using the same name. On-disk refs take precedence and are
810 		 * already present on the list before packed refs get added.
811 		 */
812 		TAILQ_FOREACH(re, refs, entry) {
813 			err = got_ref_cmp_by_name(NULL, &cmp, re->ref, ref);
814 			if (err)
815 				return err;
816 			if (cmp == 0)
817 				return NULL;
818 		}
819 	}
820 
821 	new = malloc(sizeof(*new));
822 	if (new == NULL)
823 		return got_error_from_errno("malloc");
824 	new->ref = ref;
825 	*newp = new;
826 
827 	/*
828 	 * We must de-duplicate entries on insert because packed-refs may
829 	 * contain redundant entries. On-disk refs take precedence.
830 	 * This code assumes that on-disk revs are read before packed-refs.
831 	 * We're iterating the list anyway, so insert elements sorted by name.
832 	 *
833 	 * Many callers will provide paths in a somewhat sorted order.
834 	 * Iterating backwards from the tail of the list should be more
835 	 * efficient than traversing through the entire list each time
836 	 * an element is inserted.
837 	 */
838 	re = TAILQ_LAST(refs, got_reflist_head);
839 	while (re) {
840 		err = (*cmp_cb)(cmp_arg, &cmp, re->ref, new->ref);
841 		if (err)
842 			return err;
843 		if (cmp == 0) {
844 			/* duplicate */
845 			free(new);
846 			*newp = NULL;
847 			return NULL;
848 		} else if (cmp < 0) {
849 			TAILQ_INSERT_AFTER(refs, re, new, entry);
850 			return NULL;
851 		}
852 		re = TAILQ_PREV(re, got_reflist_head, entry);
853 	}
854 
855 	TAILQ_INSERT_HEAD(refs, new, entry);
856 	return NULL;
857 }
858 
859 const struct got_error *
860 got_reflist_sort(struct got_reflist_head *refs,
861     got_ref_cmp_cb cmp_cb, void *cmp_arg)
862 {
863 	const struct got_error *err = NULL;
864 	struct got_reflist_entry *re, *tmp, *new;
865 	struct got_reflist_head sorted;
866 
867 	TAILQ_INIT(&sorted);
868 
869 	TAILQ_FOREACH_SAFE(re, refs, entry, tmp) {
870 		struct got_reference *ref = re->ref;
871 		TAILQ_REMOVE(refs, re, entry);
872 		free(re);
873 		err = got_reflist_insert(&new, &sorted, ref, cmp_cb, cmp_arg);
874 		if (err || new == NULL /* duplicate */)
875 			got_ref_close(ref);
876 		if (err)
877 			return err;
878 	}
879 
880 	TAILQ_CONCAT(refs, &sorted, entry);
881 	return NULL;
882 }
883 
884 static const struct got_error *
885 gather_on_disk_refs(struct got_reflist_head *refs, const char *path_refs,
886     const char *subdir, struct got_repository *repo,
887     got_ref_cmp_cb cmp_cb, void *cmp_arg)
888 {
889 	const struct got_error *err = NULL;
890 	DIR *d = NULL;
891 	char *path_subdir;
892 
893 	while (subdir[0] == '/')
894 		subdir++;
895 
896 	if (asprintf(&path_subdir, "%s/%s", path_refs, subdir) == -1)
897 		return got_error_from_errno("asprintf");
898 
899 	d = opendir(path_subdir);
900 	if (d == NULL)
901 		goto done;
902 
903 	for (;;) {
904 		struct dirent *dent;
905 		struct got_reference *ref;
906 		char *child;
907 		int type;
908 
909 		dent = readdir(d);
910 		if (dent == NULL)
911 			break;
912 
913 		if (strcmp(dent->d_name, ".") == 0 ||
914 		    strcmp(dent->d_name, "..") == 0)
915 			continue;
916 
917 		err = got_path_dirent_type(&type, path_subdir, dent);
918 		if (err)
919 			break;
920 
921 		switch (type) {
922 		case DT_REG:
923 			err = open_ref(&ref, path_refs, subdir, dent->d_name,
924 			    0);
925 			if (err)
926 				goto done;
927 			if (ref) {
928 				struct got_reflist_entry *new;
929 				err = got_reflist_insert(&new, refs, ref,
930 				    cmp_cb, cmp_arg);
931 				if (err || new == NULL /* duplicate */)
932 					got_ref_close(ref);
933 				if (err)
934 					goto done;
935 			}
936 			break;
937 		case DT_DIR:
938 			if (asprintf(&child, "%s%s%s", subdir,
939 			    subdir[0] == '\0' ? "" : "/", dent->d_name) == -1) {
940 				err = got_error_from_errno("asprintf");
941 				break;
942 			}
943 			err = gather_on_disk_refs(refs, path_refs, child, repo,
944 			    cmp_cb, cmp_arg);
945 			free(child);
946 			break;
947 		default:
948 			break;
949 		}
950 	}
951 done:
952 	if (d)
953 		closedir(d);
954 	free(path_subdir);
955 	return err;
956 }
957 
958 const struct got_error *
959 got_ref_list(struct got_reflist_head *refs, struct got_repository *repo,
960     const char *ref_namespace, got_ref_cmp_cb cmp_cb, void *cmp_arg)
961 {
962 	const struct got_error *err;
963 	char *packed_refs_path = NULL, *path_refs = NULL;
964 	char *abs_namespace = NULL, *buf = NULL;
965 	const char *ondisk_ref_namespace = NULL;
966 	char *line = NULL;
967 	FILE *f = NULL;
968 	struct got_reference *ref;
969 	struct got_reflist_entry *new;
970 
971 	if (ref_namespace == NULL || ref_namespace[0] == '\0') {
972 		path_refs = get_refs_dir_path(repo, GOT_REF_HEAD);
973 		if (path_refs == NULL) {
974 			err = got_error_from_errno("get_refs_dir_path");
975 			goto done;
976 		}
977 		err = open_ref(&ref, path_refs, "", GOT_REF_HEAD, 0);
978 		if (err)
979 			goto done;
980 		err = got_reflist_insert(&new, refs, ref, cmp_cb, cmp_arg);
981 		if (err || new == NULL /* duplicate */)
982 			got_ref_close(ref);
983 		if (err && err->code != GOT_ERR_NOT_REF)
984 			goto done;
985 	} else {
986 		/* Try listing a single reference. */
987 		const char *refname = ref_namespace;
988 		path_refs = get_refs_dir_path(repo, refname);
989 		if (path_refs == NULL) {
990 			err = got_error_from_errno("get_refs_dir_path");
991 			goto done;
992 		}
993 		err = open_ref(&ref, path_refs, "", refname, 0);
994 		if (err) {
995 			if (err->code != GOT_ERR_NOT_REF)
996 				goto done;
997 			/* Try to look up references in a given namespace. */
998 		} else {
999 			err = got_reflist_insert(&new, refs, ref,
1000 			    cmp_cb, cmp_arg);
1001 			if (err || new == NULL /* duplicate */)
1002 				got_ref_close(ref);
1003 			return err;
1004 		}
1005 	}
1006 
1007 	if (ref_namespace) {
1008 		size_t len;
1009 		/* Canonicalize the path to eliminate double-slashes if any. */
1010 		if (asprintf(&abs_namespace, "/%s", ref_namespace) == -1) {
1011 			err = got_error_from_errno("asprintf");
1012 			goto done;
1013 		}
1014 		len = strlen(abs_namespace) + 1;
1015 		buf = malloc(len);
1016 		if (buf == NULL) {
1017 			err = got_error_from_errno("malloc");
1018 			goto done;
1019 		}
1020 		err = got_canonpath(abs_namespace, buf, len);
1021 		if (err)
1022 			goto done;
1023 		ondisk_ref_namespace = buf;
1024 		while (ondisk_ref_namespace[0] == '/')
1025 			ondisk_ref_namespace++;
1026 		if (strncmp(ondisk_ref_namespace, "refs/", 5) == 0)
1027 			ondisk_ref_namespace += 5;
1028 		else if (strcmp(ondisk_ref_namespace, "refs") == 0)
1029 			ondisk_ref_namespace = "";
1030 	}
1031 
1032 	/* Gather on-disk refs before parsing packed-refs. */
1033 	free(path_refs);
1034 	path_refs = get_refs_dir_path(repo, "");
1035 	if (path_refs == NULL) {
1036 		err = got_error_from_errno("get_refs_dir_path");
1037 		goto done;
1038 	}
1039 	err = gather_on_disk_refs(refs, path_refs,
1040 	    ondisk_ref_namespace ? ondisk_ref_namespace : "", repo,
1041 	    cmp_cb, cmp_arg);
1042 	if (err)
1043 		goto done;
1044 
1045 	/*
1046 	 * The packed-refs file may contain redundant entries, in which
1047 	 * case on-disk refs take precedence.
1048 	 */
1049 	packed_refs_path = got_repo_get_path_packed_refs(repo);
1050 	if (packed_refs_path == NULL) {
1051 		err = got_error_from_errno("got_repo_get_path_packed_refs");
1052 		goto done;
1053 	}
1054 
1055 	f = fopen(packed_refs_path, "re");
1056 	if (f) {
1057 		size_t linesize = 0;
1058 		ssize_t linelen;
1059 		struct stat sb;
1060 
1061 		if (fstat(fileno(f), &sb) == -1) {
1062 			err = got_error_from_errno2("fstat", packed_refs_path);
1063 			goto done;
1064 		}
1065 		for (;;) {
1066 			linelen = getline(&line, &linesize, f);
1067 			if (linelen == -1) {
1068 				if (feof(f))
1069 					break;
1070 				err = got_ferror(f, GOT_ERR_BAD_REF_DATA);
1071 				goto done;
1072 			}
1073 			if (linelen > 0 && line[linelen - 1] == '\n')
1074 				line[linelen - 1] = '\0';
1075 			err = parse_packed_ref_line(&ref, NULL, line,
1076 			    sb.st_mtime);
1077 			if (err)
1078 				goto done;
1079 			if (ref) {
1080 				if (ref_namespace) {
1081 					const char *name;
1082 					name = got_ref_get_name(ref);
1083 					if (!got_path_is_child(name,
1084 					    ref_namespace,
1085 					    strlen(ref_namespace))) {
1086 						got_ref_close(ref);
1087 						continue;
1088 					}
1089 				}
1090 				err = got_reflist_insert(&new, refs, ref,
1091 				    cmp_cb, cmp_arg);
1092 				if (err || new == NULL /* duplicate */)
1093 					got_ref_close(ref);
1094 				if (err)
1095 					goto done;
1096 			}
1097 		}
1098 	}
1099 done:
1100 	free(packed_refs_path);
1101 	free(abs_namespace);
1102 	free(buf);
1103 	free(line);
1104 	free(path_refs);
1105 	if (f && fclose(f) == EOF && err == NULL)
1106 		err = got_error_from_errno("fclose");
1107 	return err;
1108 }
1109 
1110 void
1111 got_ref_list_free(struct got_reflist_head *refs)
1112 {
1113 	struct got_reflist_entry *re;
1114 
1115 	while ((re = TAILQ_FIRST(refs))) {
1116 		TAILQ_REMOVE(refs, re, entry);
1117 		got_ref_close(re->ref);
1118 		free(re);
1119 	}
1120 
1121 }
1122 
1123 int
1124 got_ref_is_symbolic(struct got_reference *ref)
1125 {
1126 	return (ref->flags & GOT_REF_IS_SYMBOLIC);
1127 }
1128 
1129 const struct got_error *
1130 got_ref_change_ref(struct got_reference *ref, struct got_object_id *id)
1131 {
1132 	if (ref->flags & GOT_REF_IS_SYMBOLIC)
1133 		return got_error(GOT_ERR_BAD_REF_TYPE);
1134 
1135 	memcpy(ref->ref.ref.sha1, id->sha1, sizeof(ref->ref.ref.sha1));
1136 	return NULL;
1137 }
1138 
1139 const struct got_error *
1140 got_ref_change_symref(struct got_reference *ref, const char *refname)
1141 {
1142 	char *new_name;
1143 
1144 	if ((ref->flags & GOT_REF_IS_SYMBOLIC) == 0)
1145 		return got_error(GOT_ERR_BAD_REF_TYPE);
1146 
1147 	new_name = strdup(refname);
1148 	if (new_name == NULL)
1149 		return got_error_from_errno("strdup");
1150 
1151 	free(ref->ref.symref.ref);
1152 	ref->ref.symref.ref = new_name;
1153 	return NULL;
1154 }
1155 
1156 const struct got_error *
1157 got_ref_change_symref_to_ref(struct got_reference *symref,
1158     struct got_object_id *id)
1159 {
1160 	if ((symref->flags & GOT_REF_IS_SYMBOLIC) == 0)
1161 		return got_error(GOT_ERR_BAD_REF_TYPE);
1162 
1163 	symref->ref.ref.name = symref->ref.symref.name;
1164 	memcpy(symref->ref.ref.sha1, id->sha1, SHA1_DIGEST_LENGTH);
1165 	symref->flags &= ~GOT_REF_IS_SYMBOLIC;
1166 	return NULL;
1167 }
1168 
1169 const struct got_error *
1170 got_ref_write(struct got_reference *ref, struct got_repository *repo)
1171 {
1172 	const struct got_error *err = NULL, *unlock_err = NULL;
1173 	const char *name = got_ref_get_name(ref);
1174 	char *path_refs = NULL, *path = NULL, *tmppath = NULL;
1175 	struct got_lockfile *lf = NULL;
1176 	FILE *f = NULL;
1177 	size_t n;
1178 	struct stat sb;
1179 
1180 	path_refs = get_refs_dir_path(repo, name);
1181 	if (path_refs == NULL) {
1182 		err = got_error_from_errno2("get_refs_dir_path", name);
1183 		goto done;
1184 	}
1185 
1186 	if (asprintf(&path, "%s/%s", path_refs, name) == -1) {
1187 		err = got_error_from_errno("asprintf");
1188 		goto done;
1189 	}
1190 
1191 	err = got_opentemp_named(&tmppath, &f, path, "");
1192 	if (err) {
1193 		char *parent;
1194 		if (!(err->code == GOT_ERR_ERRNO && errno == ENOENT))
1195 			goto done;
1196 		err = got_path_dirname(&parent, path);
1197 		if (err)
1198 			goto done;
1199 		err = got_path_mkdir(parent);
1200 		free(parent);
1201 		if (err)
1202 			goto done;
1203 		err = got_opentemp_named(&tmppath, &f, path, "");
1204 		if (err)
1205 			goto done;
1206 	}
1207 
1208 	if (ref->flags & GOT_REF_IS_SYMBOLIC) {
1209 		n = fprintf(f, "ref: %s\n", ref->ref.symref.ref);
1210 		if (n != strlen(ref->ref.symref.ref) + 6) {
1211 			err = got_ferror(f, GOT_ERR_IO);
1212 			goto done;
1213 		}
1214 	} else {
1215 		char hex[SHA1_DIGEST_STRING_LENGTH];
1216 		if (got_sha1_digest_to_str(ref->ref.ref.sha1, hex,
1217 		    sizeof(hex)) == NULL) {
1218 			err = got_error(GOT_ERR_BAD_REF_DATA);
1219 			goto done;
1220 		}
1221 		n = fprintf(f, "%s\n", hex);
1222 		if (n != sizeof(hex)) {
1223 			err = got_ferror(f, GOT_ERR_IO);
1224 			goto done;
1225 		}
1226 	}
1227 
1228 	if (ref->lf == NULL) {
1229 		err = got_lockfile_lock(&lf, path, -1);
1230 		if (err)
1231 			goto done;
1232 	}
1233 
1234 	/* XXX: check if old content matches our expectations? */
1235 
1236 	if (stat(path, &sb) != 0) {
1237 		if (errno != ENOENT) {
1238 			err = got_error_from_errno2("stat", path);
1239 			goto done;
1240 		}
1241 		sb.st_mode = GOT_DEFAULT_FILE_MODE;
1242 	}
1243 
1244 	if (fchmod(fileno(f), sb.st_mode) != 0) {
1245 		err = got_error_from_errno2("fchmod", tmppath);
1246 		goto done;
1247 	}
1248 
1249 	if (rename(tmppath, path) != 0) {
1250 		err = got_error_from_errno3("rename", tmppath, path);
1251 		goto done;
1252 	}
1253 	free(tmppath);
1254 	tmppath = NULL;
1255 
1256 	if (stat(path, &sb) == -1) {
1257 		err = got_error_from_errno2("stat", path);
1258 		goto done;
1259 	}
1260 	ref->mtime = sb.st_mtime;
1261 done:
1262 	if (ref->lf == NULL && lf)
1263 		unlock_err = got_lockfile_unlock(lf, -1);
1264 	if (f) {
1265 		if (fclose(f) == EOF && err == NULL)
1266 			err = got_error_from_errno("fclose");
1267 	}
1268 	free(path_refs);
1269 	free(path);
1270 	if (tmppath) {
1271 		if (unlink(tmppath) == -1 && err == NULL)
1272 			err = got_error_from_errno2("unlink", tmppath);
1273 		free(tmppath);
1274 	}
1275 	return err ? err : unlock_err;
1276 }
1277 
1278 static const struct got_error *
1279 delete_packed_ref(struct got_reference *delref, struct got_repository *repo)
1280 {
1281 	const struct got_error *err = NULL, *unlock_err = NULL;
1282 	struct got_lockfile *lf = NULL;
1283 	FILE *f = NULL, *tmpf = NULL;
1284 	char *line = NULL, *packed_refs_path, *tmppath = NULL;
1285 	size_t linesize = 0;
1286 	struct got_reflist_head refs;
1287 	int found_delref = 0;
1288 
1289 	/* The packed-refs file does not cotain symbolic references. */
1290 	if (delref->flags & GOT_REF_IS_SYMBOLIC)
1291 		return got_error(GOT_ERR_BAD_REF_DATA);
1292 
1293 	TAILQ_INIT(&refs);
1294 
1295 	packed_refs_path = got_repo_get_path_packed_refs(repo);
1296 	if (packed_refs_path == NULL)
1297 		return got_error_from_errno("got_repo_get_path_packed_refs");
1298 
1299 	err = got_opentemp_named(&tmppath, &tmpf, packed_refs_path, "");
1300 	if (err)
1301 		goto done;
1302 
1303 	if (delref->lf == NULL) {
1304 		err = got_lockfile_lock(&lf, packed_refs_path, -1);
1305 		if (err)
1306 			goto done;
1307 	}
1308 
1309 	f = fopen(packed_refs_path, "re");
1310 	if (f == NULL) {
1311 		err = got_error_from_errno2("fopen", packed_refs_path);
1312 		goto done;
1313 	}
1314 	for (;;) {
1315 		ssize_t linelen;
1316 		struct got_reference *ref;
1317 		struct got_reflist_entry *new;
1318 
1319 		linelen = getline(&line, &linesize, f);
1320 		if (linelen == -1) {
1321 			if (feof(f))
1322 				break;
1323 			err = got_ferror(f, GOT_ERR_BAD_REF_DATA);
1324 			goto done;
1325 		}
1326 		if (linelen > 0 && line[linelen - 1] == '\n')
1327 			line[linelen - 1] = '\0';
1328 		err = parse_packed_ref_line(&ref, NULL, line, 0);
1329 		if (err)
1330 			goto done;
1331 		if (ref == NULL)
1332 			continue;
1333 
1334 		if (strcmp(ref->ref.ref.name, delref->ref.ref.name) == 0 &&
1335 		    memcmp(ref->ref.ref.sha1, delref->ref.ref.sha1,
1336 		    sizeof(delref->ref.ref.sha1)) == 0) {
1337 			found_delref = 1;
1338 			got_ref_close(ref);
1339 			continue;
1340 		}
1341 
1342 		err = got_reflist_insert(&new, &refs, ref,
1343 		    got_ref_cmp_by_name, NULL);
1344 		if (err || new == NULL /* duplicate */)
1345 			got_ref_close(ref);
1346 		if (err)
1347 			goto done;
1348 	}
1349 
1350 	if (found_delref) {
1351 		struct got_reflist_entry *re;
1352 		size_t n;
1353 		struct stat sb;
1354 
1355 		n = fprintf(tmpf, "%s\n", GOT_PACKED_REFS_HEADER);
1356 		if (n != sizeof(GOT_PACKED_REFS_HEADER)) {
1357 			err = got_ferror(f, GOT_ERR_IO);
1358 			goto done;
1359 		}
1360 
1361 		TAILQ_FOREACH(re, &refs, entry) {
1362 			char hex[SHA1_DIGEST_STRING_LENGTH];
1363 
1364 			if (got_sha1_digest_to_str(re->ref->ref.ref.sha1, hex,
1365 			    sizeof(hex)) == NULL) {
1366 				err = got_error(GOT_ERR_BAD_REF_DATA);
1367 				goto done;
1368 			}
1369 			n = fprintf(tmpf, "%s ", hex);
1370 			if (n != sizeof(hex)) {
1371 				err = got_ferror(f, GOT_ERR_IO);
1372 				goto done;
1373 			}
1374 			n = fprintf(tmpf, "%s\n", re->ref->ref.ref.name);
1375 			if (n != strlen(re->ref->ref.ref.name) + 1) {
1376 				err = got_ferror(f, GOT_ERR_IO);
1377 				goto done;
1378 			}
1379 		}
1380 
1381 		if (fflush(tmpf) != 0) {
1382 			err = got_error_from_errno("fflush");
1383 			goto done;
1384 		}
1385 
1386 		if (fstat(fileno(f), &sb) != 0) {
1387 			if (errno != ENOENT) {
1388 				err = got_error_from_errno2("fstat",
1389 				    packed_refs_path);
1390 				goto done;
1391 			}
1392 			sb.st_mode = GOT_DEFAULT_FILE_MODE;
1393 		}
1394 
1395 		if (fchmod(fileno(tmpf), sb.st_mode) != 0) {
1396 			err = got_error_from_errno2("fchmod", tmppath);
1397 			goto done;
1398 		}
1399 
1400 		if (rename(tmppath, packed_refs_path) != 0) {
1401 			err = got_error_from_errno3("rename", tmppath,
1402 			    packed_refs_path);
1403 			goto done;
1404 		}
1405 		free(tmppath);
1406 		tmppath = NULL;
1407 	}
1408 done:
1409 	if (delref->lf == NULL && lf)
1410 		unlock_err = got_lockfile_unlock(lf, -1);
1411 	if (f) {
1412 		if (fclose(f) == EOF && err == NULL)
1413 			err = got_error_from_errno("fclose");
1414 	}
1415 	if (tmppath && unlink(tmppath) == -1 && err == NULL)
1416 		err = got_error_from_errno2("unlink", tmppath);
1417 	if (tmpf && fclose(tmpf) == EOF && err == NULL)
1418 		err = got_error_from_errno("fclose");
1419 	free(tmppath);
1420 	free(packed_refs_path);
1421 	free(line);
1422 	got_ref_list_free(&refs);
1423 	return err ? err : unlock_err;
1424 }
1425 
1426 static const struct got_error *
1427 delete_loose_ref(struct got_reference *ref, struct got_repository *repo)
1428 {
1429 	const struct got_error *err = NULL, *unlock_err = NULL;
1430 	const char *name = got_ref_get_name(ref);
1431 	char *path_refs = NULL, *path = NULL;
1432 	struct got_lockfile *lf = NULL;
1433 
1434 	path_refs = get_refs_dir_path(repo, name);
1435 	if (path_refs == NULL) {
1436 		err = got_error_from_errno2("get_refs_dir_path", name);
1437 		goto done;
1438 	}
1439 
1440 	if (asprintf(&path, "%s/%s", path_refs, name) == -1) {
1441 		err = got_error_from_errno("asprintf");
1442 		goto done;
1443 	}
1444 
1445 	if (ref->lf == NULL) {
1446 		err = got_lockfile_lock(&lf, path, -1);
1447 		if (err)
1448 			goto done;
1449 	}
1450 
1451 	/* XXX: check if old content matches our expectations? */
1452 
1453 	if (unlink(path) == -1)
1454 		err = got_error_from_errno2("unlink", path);
1455 done:
1456 	if (ref->lf == NULL && lf)
1457 		unlock_err = got_lockfile_unlock(lf, -1);
1458 
1459 	free(path_refs);
1460 	free(path);
1461 	return err ? err : unlock_err;
1462 }
1463 
1464 const struct got_error *
1465 got_ref_delete(struct got_reference *ref, struct got_repository *repo)
1466 {
1467 	const struct got_error *err = NULL;
1468 	struct got_reference *ref2;
1469 
1470 	if (ref->flags & GOT_REF_IS_PACKED) {
1471 		err = delete_packed_ref(ref, repo);
1472 		if (err)
1473 			return err;
1474 
1475 		err = got_ref_open(&ref2, repo, got_ref_get_name(ref), 0);
1476 		if (err) {
1477 			if (err->code == GOT_ERR_NOT_REF)
1478 				return NULL;
1479 			return err;
1480 		}
1481 
1482 		err = delete_loose_ref(ref2, repo);
1483 		got_ref_close(ref2);
1484 		return err;
1485 	} else {
1486 		err = delete_loose_ref(ref, repo);
1487 		if (err)
1488 			return err;
1489 
1490 		err = got_ref_open(&ref2, repo, got_ref_get_name(ref), 0);
1491 		if (err) {
1492 			if (err->code == GOT_ERR_NOT_REF)
1493 				return NULL;
1494 			return err;
1495 		}
1496 
1497 		err = delete_packed_ref(ref2, repo);
1498 		got_ref_close(ref2);
1499 		return err;
1500 	}
1501 }
1502 
1503 const struct got_error *
1504 got_ref_unlock(struct got_reference *ref)
1505 {
1506 	const struct got_error *err;
1507 	err = got_lockfile_unlock(ref->lf, -1);
1508 	ref->lf = NULL;
1509 	return err;
1510 }
1511 
1512 struct got_reflist_object_id_map {
1513 	struct got_object_idset *idset;
1514 };
1515 
1516 struct got_reflist_object_id_map_entry {
1517 	struct got_reflist_head refs;
1518 };
1519 
1520 static const struct got_error *
1521 add_object_id_map_entry(struct got_object_idset *idset,
1522     struct got_object_id *id, struct got_reflist_entry *re)
1523 {
1524 	const struct got_error *err = NULL;
1525 	struct got_reflist_object_id_map_entry *ent;
1526 	struct got_reflist_entry *new;
1527 
1528 	ent = got_object_idset_get(idset, id);
1529 	if (ent == NULL) {
1530 		ent = malloc(sizeof(*ent));
1531 		if (ent == NULL)
1532 			return got_error_from_errno("malloc");
1533 
1534 		TAILQ_INIT(&ent->refs);
1535 		err = got_object_idset_add(idset, id, ent);
1536 		if (err) {
1537 			free(ent);
1538 			return err;
1539 		}
1540 	}
1541 
1542 	err = got_reflist_entry_dup(&new, re);
1543 	if (err)
1544 		return err;
1545 
1546 	TAILQ_INSERT_TAIL(&ent->refs, new, entry);
1547 	return NULL;
1548 }
1549 
1550 const struct got_error *
1551 got_reflist_object_id_map_create(struct got_reflist_object_id_map **map,
1552     struct got_reflist_head *refs, struct got_repository *repo)
1553 {
1554 	const struct got_error *err = NULL;
1555 	struct got_object_idset *idset;
1556 	struct got_object_id *id = NULL;
1557 	struct got_reflist_entry *re;
1558 
1559 	idset = got_object_idset_alloc();
1560 	if (idset == NULL)
1561 		return got_error_from_errno("got_object_idset_alloc");
1562 
1563 	*map = malloc(sizeof(**map));
1564 	if (*map == NULL) {
1565 		got_object_idset_free(idset);
1566 		return got_error_from_errno("malloc");
1567 	}
1568 	(*map)->idset = idset;
1569 
1570 	TAILQ_FOREACH(re, refs, entry) {
1571 		struct got_tag_object *tag = NULL;
1572 
1573 		err = got_ref_resolve(&id, repo, re->ref);
1574 		if (err)
1575 			goto done;
1576 
1577 		err = add_object_id_map_entry(idset, id, re);
1578 		if (err)
1579 			goto done;
1580 
1581 		if (strstr(got_ref_get_name(re->ref), "/tags/") == NULL) {
1582 			free(id);
1583 			id = NULL;
1584 			continue;
1585 		}
1586 
1587 		err = got_object_open_as_tag(&tag, repo, id);
1588 		if (err) {
1589 			if (err->code != GOT_ERR_OBJ_TYPE)
1590 				goto done;
1591 			/* Ref points at something other than a tag. */
1592 			err = NULL;
1593 			tag = NULL;
1594 			free(id);
1595 			id = NULL;
1596 			continue;
1597 		}
1598 
1599 		err = add_object_id_map_entry(idset,
1600 		    got_object_tag_get_object_id(tag), re);
1601 		got_object_tag_close(tag);
1602 		if (err)
1603 			goto done;
1604 
1605 		free(id);
1606 		id = NULL;
1607 	}
1608 done:
1609 	free(id);
1610 	if (err) {
1611 		got_reflist_object_id_map_free(*map);
1612 		*map = NULL;
1613 	}
1614 	return err;
1615 }
1616 
1617 struct got_reflist_head *
1618 got_reflist_object_id_map_lookup(struct got_reflist_object_id_map *map,
1619     struct got_object_id *id)
1620 {
1621 	struct got_reflist_object_id_map_entry *ent;
1622 	ent = got_object_idset_get(map->idset, id);
1623 	if (ent)
1624 		return &ent->refs;
1625 	return NULL;
1626 }
1627 
1628 static const struct got_error *
1629 free_id_map_entry(struct got_object_id *id, void *data, void *arg)
1630 {
1631 	struct got_reflist_object_id_map_entry *ent = data;
1632 
1633 	got_ref_list_free(&ent->refs);
1634 	free(ent);
1635 	return NULL;
1636 }
1637 
1638 void
1639 got_reflist_object_id_map_free(struct got_reflist_object_id_map *map)
1640 {
1641 	got_object_idset_for_each(map->idset, free_id_map_entry, NULL);
1642 	got_object_idset_free(map->idset);
1643 	free(map);
1644 }