2 * Copyright (c) 2018, 2019 Stefan Sperling <stsp@openbsd.org>
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.
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.
17 #include <sys/types.h>
18 #include <sys/queue.h>
29 #include "got_error.h"
30 #include "got_object.h"
31 #include "got_repository.h"
32 #include "got_reference.h"
34 #include "got_lib_sha1.h"
35 #include "got_lib_path.h"
36 #include "got_lib_delta.h"
37 #include "got_lib_inflate.h"
38 #include "got_lib_object.h"
41 #define nitems(_a) (sizeof(_a) / sizeof((_a)[0]))
44 #define GOT_REF_HEADS "heads"
45 #define GOT_REF_TAGS "tags"
46 #define GOT_REF_REMOTES "remotes"
48 /* A symbolic reference. */
54 /* A non-symbolic reference (there is no better designation). */
57 u_int8_t sha1[SHA1_DIGEST_LENGTH];
60 /* A reference which points to an arbitrary object. */
61 struct got_reference {
63 #define GOT_REF_IS_SYMBOLIC 0x01
67 struct got_symref symref;
71 static const struct got_error *
72 parse_symref(struct got_reference **ref, const char *name, const char *line)
74 struct got_symref *symref;
79 return got_error(GOT_ERR_BAD_REF_DATA);
81 symref_name = strdup(name);
82 if (symref_name == NULL)
83 return got_error_from_errno();
84 symref_ref = strdup(line);
85 if (symref_ref == NULL) {
86 const struct got_error *err = got_error_from_errno();
91 *ref = calloc(1, sizeof(**ref));
93 return got_error_from_errno();
94 (*ref)->flags |= GOT_REF_IS_SYMBOLIC;
95 symref = &((*ref)->ref.symref);
96 symref->name = symref_name;
97 symref->ref = symref_ref;
101 static const struct got_error *
102 parse_ref_line(struct got_reference **ref, const char *name, const char *line)
104 struct got_object_id id;
106 if (strncmp(line, "ref: ", 5) == 0) {
108 return parse_symref(ref, name, line);
111 if (!got_parse_sha1_digest(id.sha1, line))
112 return got_error(GOT_ERR_BAD_REF_DATA);
114 return got_ref_alloc(ref, name, &id);
117 static const struct got_error *
118 parse_ref_file(struct got_reference **ref, const char *name,
121 const struct got_error *err = NULL;
122 FILE *f = fopen(abspath, "rb");
125 const char delim[3] = {'\0', '\0', '\0'};
130 line = fparseln(f, &len, NULL, delim, 0);
132 err = got_error(GOT_ERR_BAD_REF_DATA);
136 err = parse_ref_line(ref, name, line);
139 if (fclose(f) != 0 && err == NULL)
140 err = got_error_from_errno();
145 is_well_known_ref(const char *refname)
147 return (strcmp(refname, GOT_REF_HEAD) == 0 ||
148 strcmp(refname, GOT_REF_ORIG_HEAD) == 0 ||
149 strcmp(refname, GOT_REF_MERGE_HEAD) == 0 ||
150 strcmp(refname, GOT_REF_FETCH_HEAD) == 0);
154 get_refs_dir_path(struct got_repository *repo, const char *refname)
156 if (is_well_known_ref(refname) || strncmp(refname, "refs/", 5) == 0)
157 return strdup(got_repo_get_path_git_dir(repo));
159 return got_repo_get_path_refs(repo);
162 const struct got_error *
163 got_ref_alloc(struct got_reference **ref, const char *name,
164 struct got_object_id *id)
166 const struct got_error *err = NULL;
168 *ref = calloc(1, sizeof(**ref));
170 return got_error_from_errno();
172 memcpy(&(*ref)->ref.ref.sha1, id->sha1, SHA1_DIGEST_LENGTH);
173 (*ref)->ref.ref.name = strdup(name);
174 if ((*ref)->ref.ref.name == NULL) {
175 err = got_error_from_errno();
182 static const struct got_error *
183 parse_packed_ref_line(struct got_reference **ref, const char *abs_refname,
186 struct got_object_id id;
191 if (line[0] == '#' || line[0] == '^')
194 if (!got_parse_sha1_digest(id.sha1, line))
195 return got_error(GOT_ERR_BAD_REF_DATA);
198 if (strcmp(line + SHA1_DIGEST_STRING_LENGTH, abs_refname) != 0)
202 name = line + SHA1_DIGEST_STRING_LENGTH;
204 return got_ref_alloc(ref, name, &id);
207 static const struct got_error *
208 open_packed_ref(struct got_reference **ref, FILE *f, const char **subdirs,
209 int nsubdirs, const char *refname)
211 const struct got_error *err = NULL;
215 const char delim[3] = {'\0', '\0', '\0'};
216 int i, ref_is_absolute = (strncmp(refname, "refs/", 5) == 0);
221 abs_refname = (char *)refname;
223 line = fparseln(f, &len, NULL, delim, 0);
226 for (i = 0; i < nsubdirs; i++) {
227 if (!ref_is_absolute &&
228 asprintf(&abs_refname, "refs/%s/%s", subdirs[i],
230 return got_error_from_errno();
231 err = parse_packed_ref_line(ref, abs_refname, line);
232 if (!ref_is_absolute)
234 if (err || *ref != NULL)
240 } while (*ref == NULL);
245 static const struct got_error *
246 open_ref(struct got_reference **ref, const char *path_refs, const char *subdir,
249 const struct got_error *err = NULL;
251 char *normpath = NULL;
252 char *absname = NULL;
253 int ref_is_absolute = (strncmp(name, "refs/", 5) == 0);
254 int ref_is_well_known = is_well_known_ref(name);
258 if (ref_is_absolute || ref_is_well_known) {
259 if (asprintf(&path, "%s/%s", path_refs, name) == -1)
260 return got_error_from_errno();
261 absname = (char *)name;
263 if (asprintf(&path, "%s/%s/%s", path_refs, subdir, name) == -1)
264 return got_error_from_errno();
266 if (asprintf(&absname, "refs/%s/%s", subdir, name) == -1) {
267 err = got_error_from_errno();
272 normpath = got_path_normalize(path);
273 if (normpath == NULL) {
274 err = got_error_from_errno();
278 err = parse_ref_file(ref, absname, normpath);
280 if (!ref_is_absolute && !ref_is_well_known)
287 const struct got_error *
288 got_ref_open(struct got_reference **ref, struct got_repository *repo,
291 const struct got_error *err = NULL;
292 char *path_refs = NULL;
293 const char *subdirs[] = {
294 GOT_REF_HEADS, GOT_REF_TAGS, GOT_REF_REMOTES
296 int i, well_known = is_well_known_ref(refname);
300 path_refs = get_refs_dir_path(repo, refname);
301 if (path_refs == NULL) {
302 err = got_error_from_errno();
307 char *packed_refs_path;
310 /* Search on-disk refs before packed refs! */
311 for (i = 0; i < nitems(subdirs); i++) {
312 err = open_ref(ref, path_refs, subdirs[i], refname);
317 packed_refs_path = got_repo_get_path_packed_refs(repo);
318 if (packed_refs_path == NULL) {
319 err = got_error_from_errno();
323 f = fopen(packed_refs_path, "rb");
324 free(packed_refs_path);
326 err = open_packed_ref(ref, f, subdirs, nitems(subdirs),
328 if (fclose(f) != 0 && err == NULL)
329 err = got_error_from_errno();
335 err = open_ref(ref, path_refs, "", refname);
340 err = got_error_not_ref(refname);
346 got_ref_close(struct got_reference *ref)
348 if (ref->flags & GOT_REF_IS_SYMBOLIC)
349 free(ref->ref.symref.name);
351 free(ref->ref.ref.name);
355 struct got_reference *
356 got_ref_dup(struct got_reference *ref)
358 struct got_reference *ret;
360 ret = calloc(1, sizeof(*ret));
364 ret->flags = ref->flags;
365 if (ref->flags & GOT_REF_IS_SYMBOLIC) {
366 ret->ref.symref.name = strdup(ref->ref.symref.name);
367 if (ret->ref.symref.name == NULL) {
371 ret->ref.symref.ref = strdup(ref->ref.symref.ref);
372 if (ret->ref.symref.ref == NULL) {
373 free(ret->ref.symref.name);
378 ref->ref.ref.name = strdup(ref->ref.ref.name);
379 if (ref->ref.ref.name == NULL) {
383 memcpy(ret->ref.ref.sha1, ref->ref.ref.sha1,
390 static const struct got_error *
391 resolve_symbolic_ref(struct got_reference **resolved,
392 struct got_repository *repo, struct got_reference *ref)
394 struct got_reference *nextref;
395 const struct got_error *err;
397 err = got_ref_open(&nextref, repo, ref->ref.symref.ref);
401 if (nextref->flags & GOT_REF_IS_SYMBOLIC)
402 err = resolve_symbolic_ref(resolved, repo, nextref);
404 *resolved = got_ref_dup(nextref);
406 got_ref_close(nextref);
410 const struct got_error *
411 got_ref_resolve(struct got_object_id **id, struct got_repository *repo,
412 struct got_reference *ref)
414 const struct got_error *err;
416 if (ref->flags & GOT_REF_IS_SYMBOLIC) {
417 struct got_reference *resolved = NULL;
418 err = resolve_symbolic_ref(&resolved, repo, ref);
420 err = got_ref_resolve(id, repo, resolved);
425 *id = calloc(1, sizeof(**id));
427 return got_error_from_errno();
428 memcpy((*id)->sha1, ref->ref.ref.sha1, SHA1_DIGEST_LENGTH);
433 got_ref_to_str(struct got_reference *ref)
437 if (ref->flags & GOT_REF_IS_SYMBOLIC)
438 return strdup(ref->ref.symref.ref);
440 str = malloc(SHA1_DIGEST_STRING_LENGTH);
444 if (got_sha1_digest_to_str(ref->ref.ref.sha1, str,
445 SHA1_DIGEST_STRING_LENGTH) == NULL) {
454 got_ref_get_name(struct got_reference *ref)
456 if (ref->flags & GOT_REF_IS_SYMBOLIC)
457 return ref->ref.symref.name;
459 return ref->ref.ref.name;
462 static const struct got_error *
463 insert_ref(struct got_reflist_head *refs, struct got_reference *ref,
464 struct got_repository *repo)
466 const struct got_error *err;
467 struct got_object_id *id;
468 struct got_reflist_entry *new, *re, *prev;
471 err = got_ref_resolve(&id, repo, ref);
475 new = malloc(sizeof(*re));
478 return got_error_from_errno();
484 * We must de-duplicate entries on insert because packed-refs may
485 * contain redundant entries. On-disk refs take precedence.
486 * This code assumes that on-disk revs are read before packed-refs.
487 * We're iterating the list anyway, so insert elements sorted by name.
489 re = SIMPLEQ_FIRST(refs);
491 cmp = got_path_cmp(got_ref_get_name(re->ref),
492 got_ref_get_name(ref));
494 free(ref); /* duplicate */
496 } else if (cmp > 0) {
498 SIMPLEQ_INSERT_AFTER(refs, prev, new, entry);
500 SIMPLEQ_INSERT_HEAD(refs, new, entry);
504 re = SIMPLEQ_NEXT(re, entry);
508 SIMPLEQ_INSERT_TAIL(refs, new, entry);
512 static const struct got_error *
513 gather_on_disk_refs(struct got_reflist_head *refs, const char *path_refs,
514 const char *subdir, struct got_repository *repo)
516 const struct got_error *err = NULL;
520 if (asprintf(&path_subdir, "%s/%s", path_refs, subdir) == -1)
521 return got_error_from_errno();
523 d = opendir(path_subdir);
529 struct got_reference *ref;
536 if (strcmp(dent->d_name, ".") == 0 ||
537 strcmp(dent->d_name, "..") == 0)
540 switch (dent->d_type) {
542 err = open_ref(&ref, path_refs, subdir, dent->d_name);
546 err = insert_ref(refs, ref, repo);
552 if (asprintf(&child, "%s%s%s", subdir,
553 subdir[0] == '\0' ? "" : "/", dent->d_name) == -1) {
554 err = got_error_from_errno();
557 err = gather_on_disk_refs(refs, path_refs, child, repo);
571 const struct got_error *
572 got_ref_list(struct got_reflist_head *refs, struct got_repository *repo)
574 const struct got_error *err;
575 char *packed_refs_path, *path_refs = NULL;
577 struct got_reference *ref;
579 /* HEAD ref should always exist. */
580 path_refs = get_refs_dir_path(repo, GOT_REF_HEAD);
581 if (path_refs == NULL) {
582 err = got_error_from_errno();
585 err = open_ref(&ref, path_refs, "", GOT_REF_HEAD);
588 err = insert_ref(refs, ref, repo);
592 /* Gather on-disk refs before parsing packed-refs. */
594 path_refs = get_refs_dir_path(repo, "");
595 if (path_refs == NULL) {
596 err = got_error_from_errno();
599 err = gather_on_disk_refs(refs, path_refs, "", repo);
604 * The packed-refs file may contain redundant entries, in which
605 * case on-disk refs take precedence.
607 packed_refs_path = got_repo_get_path_packed_refs(repo);
608 if (packed_refs_path == NULL) {
609 err = got_error_from_errno();
613 f = fopen(packed_refs_path, "r");
614 free(packed_refs_path);
618 const char delim[3] = {'\0', '\0', '\0'};
620 line = fparseln(f, &len, NULL, delim, 0);
623 err = parse_packed_ref_line(&ref, NULL, line);
627 err = insert_ref(refs, ref, repo);
635 if (f && fclose(f) != 0 && err == NULL)
636 err = got_error_from_errno();