Merge branch 'jc/conflict-marker-size' into next
[git/dscho.git] / refs.c
blob5583f4baff9b77a8095280e1f30c051dd70972bc
1 #include "cache.h"
2 #include "refs.h"
3 #include "object.h"
4 #include "tag.h"
5 #include "dir.h"
7 /* ISSYMREF=01 and ISPACKED=02 are public interfaces */
8 #define REF_KNOWS_PEELED 04
10 struct ref_list {
11 struct ref_list *next;
12 unsigned char flag; /* ISSYMREF? ISPACKED? */
13 unsigned char sha1[20];
14 unsigned char peeled[20];
15 char name[FLEX_ARRAY];
18 static const char *parse_ref_line(char *line, unsigned char *sha1)
21 * 42: the answer to everything.
23 * In this case, it happens to be the answer to
24 * 40 (length of sha1 hex representation)
25 * +1 (space in between hex and name)
26 * +1 (newline at the end of the line)
28 int len = strlen(line) - 42;
30 if (len <= 0)
31 return NULL;
32 if (get_sha1_hex(line, sha1) < 0)
33 return NULL;
34 if (!isspace(line[40]))
35 return NULL;
36 line += 41;
37 if (isspace(*line))
38 return NULL;
39 if (line[len] != '\n')
40 return NULL;
41 line[len] = 0;
43 return line;
46 static struct ref_list *add_ref(const char *name, const unsigned char *sha1,
47 int flag, struct ref_list *list,
48 struct ref_list **new_entry)
50 int len;
51 struct ref_list *entry;
53 /* Allocate it and add it in.. */
54 len = strlen(name) + 1;
55 entry = xmalloc(sizeof(struct ref_list) + len);
56 hashcpy(entry->sha1, sha1);
57 hashclr(entry->peeled);
58 memcpy(entry->name, name, len);
59 entry->flag = flag;
60 entry->next = list;
61 if (new_entry)
62 *new_entry = entry;
63 return entry;
66 /* merge sort the ref list */
67 static struct ref_list *sort_ref_list(struct ref_list *list)
69 int psize, qsize, last_merge_count, cmp;
70 struct ref_list *p, *q, *l, *e;
71 struct ref_list *new_list = list;
72 int k = 1;
73 int merge_count = 0;
75 if (!list)
76 return list;
78 do {
79 last_merge_count = merge_count;
80 merge_count = 0;
82 psize = 0;
84 p = new_list;
85 q = new_list;
86 new_list = NULL;
87 l = NULL;
89 while (p) {
90 merge_count++;
92 while (psize < k && q->next) {
93 q = q->next;
94 psize++;
96 qsize = k;
98 while ((psize > 0) || (qsize > 0 && q)) {
99 if (qsize == 0 || !q) {
100 e = p;
101 p = p->next;
102 psize--;
103 } else if (psize == 0) {
104 e = q;
105 q = q->next;
106 qsize--;
107 } else {
108 cmp = strcmp(q->name, p->name);
109 if (cmp < 0) {
110 e = q;
111 q = q->next;
112 qsize--;
113 } else if (cmp > 0) {
114 e = p;
115 p = p->next;
116 psize--;
117 } else {
118 if (hashcmp(q->sha1, p->sha1))
119 die("Duplicated ref, and SHA1s don't match: %s",
120 q->name);
121 warning("Duplicated ref: %s", q->name);
122 e = q;
123 q = q->next;
124 qsize--;
125 free(e);
126 e = p;
127 p = p->next;
128 psize--;
132 e->next = NULL;
134 if (l)
135 l->next = e;
136 if (!new_list)
137 new_list = e;
138 l = e;
141 p = q;
144 k = k * 2;
145 } while ((last_merge_count != merge_count) || (last_merge_count != 1));
147 return new_list;
151 * Future: need to be in "struct repository"
152 * when doing a full libification.
154 static struct cached_refs {
155 char did_loose;
156 char did_packed;
157 struct ref_list *loose;
158 struct ref_list *packed;
159 } cached_refs;
160 static struct ref_list *current_ref;
162 static struct ref_list *extra_refs;
164 static void free_ref_list(struct ref_list *list)
166 struct ref_list *next;
167 for ( ; list; list = next) {
168 next = list->next;
169 free(list);
173 static void invalidate_cached_refs(void)
175 struct cached_refs *ca = &cached_refs;
177 if (ca->did_loose && ca->loose)
178 free_ref_list(ca->loose);
179 if (ca->did_packed && ca->packed)
180 free_ref_list(ca->packed);
181 ca->loose = ca->packed = NULL;
182 ca->did_loose = ca->did_packed = 0;
185 static void read_packed_refs(FILE *f, struct cached_refs *cached_refs)
187 struct ref_list *list = NULL;
188 struct ref_list *last = NULL;
189 char refline[PATH_MAX];
190 int flag = REF_ISPACKED;
192 while (fgets(refline, sizeof(refline), f)) {
193 unsigned char sha1[20];
194 const char *name;
195 static const char header[] = "# pack-refs with:";
197 if (!strncmp(refline, header, sizeof(header)-1)) {
198 const char *traits = refline + sizeof(header) - 1;
199 if (strstr(traits, " peeled "))
200 flag |= REF_KNOWS_PEELED;
201 /* perhaps other traits later as well */
202 continue;
205 name = parse_ref_line(refline, sha1);
206 if (name) {
207 list = add_ref(name, sha1, flag, list, &last);
208 continue;
210 if (last &&
211 refline[0] == '^' &&
212 strlen(refline) == 42 &&
213 refline[41] == '\n' &&
214 !get_sha1_hex(refline + 1, sha1))
215 hashcpy(last->peeled, sha1);
217 cached_refs->packed = sort_ref_list(list);
220 void add_extra_ref(const char *name, const unsigned char *sha1, int flag)
222 extra_refs = add_ref(name, sha1, flag, extra_refs, NULL);
225 void clear_extra_refs(void)
227 free_ref_list(extra_refs);
228 extra_refs = NULL;
231 static struct ref_list *get_packed_refs(void)
233 if (!cached_refs.did_packed) {
234 FILE *f = fopen(git_path("packed-refs"), "r");
235 cached_refs.packed = NULL;
236 if (f) {
237 read_packed_refs(f, &cached_refs);
238 fclose(f);
240 cached_refs.did_packed = 1;
242 return cached_refs.packed;
245 static struct ref_list *get_ref_dir(const char *base, struct ref_list *list)
247 DIR *dir = opendir(git_path("%s", base));
249 if (dir) {
250 struct dirent *de;
251 int baselen = strlen(base);
252 char *ref = xmalloc(baselen + 257);
254 memcpy(ref, base, baselen);
255 if (baselen && base[baselen-1] != '/')
256 ref[baselen++] = '/';
258 while ((de = readdir(dir)) != NULL) {
259 unsigned char sha1[20];
260 struct stat st;
261 int flag;
262 int namelen;
264 if (de->d_name[0] == '.')
265 continue;
266 namelen = strlen(de->d_name);
267 if (namelen > 255)
268 continue;
269 if (has_extension(de->d_name, ".lock"))
270 continue;
271 memcpy(ref + baselen, de->d_name, namelen+1);
272 if (stat(git_path("%s", ref), &st) < 0)
273 continue;
274 if (S_ISDIR(st.st_mode)) {
275 list = get_ref_dir(ref, list);
276 continue;
278 if (!resolve_ref(ref, sha1, 1, &flag))
279 hashclr(sha1);
280 list = add_ref(ref, sha1, flag, list, NULL);
282 free(ref);
283 closedir(dir);
285 return sort_ref_list(list);
288 struct warn_if_dangling_data {
289 FILE *fp;
290 const char *refname;
291 const char *msg_fmt;
294 static int warn_if_dangling_symref(const char *refname, const unsigned char *sha1,
295 int flags, void *cb_data)
297 struct warn_if_dangling_data *d = cb_data;
298 const char *resolves_to;
299 unsigned char junk[20];
301 if (!(flags & REF_ISSYMREF))
302 return 0;
304 resolves_to = resolve_ref(refname, junk, 0, NULL);
305 if (!resolves_to || strcmp(resolves_to, d->refname))
306 return 0;
308 fprintf(d->fp, d->msg_fmt, refname);
309 return 0;
312 void warn_dangling_symref(FILE *fp, const char *msg_fmt, const char *refname)
314 struct warn_if_dangling_data data = { fp, refname, msg_fmt };
315 for_each_rawref(warn_if_dangling_symref, &data);
318 static struct ref_list *get_loose_refs(void)
320 if (!cached_refs.did_loose) {
321 cached_refs.loose = get_ref_dir("refs", NULL);
322 cached_refs.did_loose = 1;
324 return cached_refs.loose;
327 /* We allow "recursive" symbolic refs. Only within reason, though */
328 #define MAXDEPTH 5
329 #define MAXREFLEN (1024)
331 static int resolve_gitlink_packed_ref(char *name, int pathlen, const char *refname, unsigned char *result)
333 FILE *f;
334 struct cached_refs refs;
335 struct ref_list *ref;
336 int retval;
338 strcpy(name + pathlen, "packed-refs");
339 f = fopen(name, "r");
340 if (!f)
341 return -1;
342 read_packed_refs(f, &refs);
343 fclose(f);
344 ref = refs.packed;
345 retval = -1;
346 while (ref) {
347 if (!strcmp(ref->name, refname)) {
348 retval = 0;
349 memcpy(result, ref->sha1, 20);
350 break;
352 ref = ref->next;
354 free_ref_list(refs.packed);
355 return retval;
358 static int resolve_gitlink_ref_recursive(char *name, int pathlen, const char *refname, unsigned char *result, int recursion)
360 int fd, len = strlen(refname);
361 char buffer[128], *p;
363 if (recursion > MAXDEPTH || len > MAXREFLEN)
364 return -1;
365 memcpy(name + pathlen, refname, len+1);
366 fd = open(name, O_RDONLY);
367 if (fd < 0)
368 return resolve_gitlink_packed_ref(name, pathlen, refname, result);
370 len = read(fd, buffer, sizeof(buffer)-1);
371 close(fd);
372 if (len < 0)
373 return -1;
374 while (len && isspace(buffer[len-1]))
375 len--;
376 buffer[len] = 0;
378 /* Was it a detached head or an old-fashioned symlink? */
379 if (!get_sha1_hex(buffer, result))
380 return 0;
382 /* Symref? */
383 if (strncmp(buffer, "ref:", 4))
384 return -1;
385 p = buffer + 4;
386 while (isspace(*p))
387 p++;
389 return resolve_gitlink_ref_recursive(name, pathlen, p, result, recursion+1);
392 int resolve_gitlink_ref(const char *path, const char *refname, unsigned char *result)
394 int len = strlen(path), retval;
395 char *gitdir;
396 const char *tmp;
398 while (len && path[len-1] == '/')
399 len--;
400 if (!len)
401 return -1;
402 gitdir = xmalloc(len + MAXREFLEN + 8);
403 memcpy(gitdir, path, len);
404 memcpy(gitdir + len, "/.git", 6);
405 len += 5;
407 tmp = read_gitfile_gently(gitdir);
408 if (tmp) {
409 free(gitdir);
410 len = strlen(tmp);
411 gitdir = xmalloc(len + MAXREFLEN + 3);
412 memcpy(gitdir, tmp, len);
414 gitdir[len] = '/';
415 gitdir[++len] = '\0';
416 retval = resolve_gitlink_ref_recursive(gitdir, len, refname, result, 0);
417 free(gitdir);
418 return retval;
422 * If the "reading" argument is set, this function finds out what _object_
423 * the ref points at by "reading" the ref. The ref, if it is not symbolic,
424 * has to exist, and if it is symbolic, it has to point at an existing ref,
425 * because the "read" goes through the symref to the ref it points at.
427 * The access that is not "reading" may often be "writing", but does not
428 * have to; it can be merely checking _where it leads to_. If it is a
429 * prelude to "writing" to the ref, a write to a symref that points at
430 * yet-to-be-born ref will create the real ref pointed by the symref.
431 * reading=0 allows the caller to check where such a symref leads to.
433 const char *resolve_ref(const char *ref, unsigned char *sha1, int reading, int *flag)
435 int depth = MAXDEPTH;
436 ssize_t len;
437 char buffer[256];
438 static char ref_buffer[256];
440 if (flag)
441 *flag = 0;
443 for (;;) {
444 char path[PATH_MAX];
445 struct stat st;
446 char *buf;
447 int fd;
449 if (--depth < 0)
450 return NULL;
452 git_snpath(path, sizeof(path), "%s", ref);
453 /* Special case: non-existing file. */
454 if (lstat(path, &st) < 0) {
455 struct ref_list *list = get_packed_refs();
456 while (list) {
457 if (!strcmp(ref, list->name)) {
458 hashcpy(sha1, list->sha1);
459 if (flag)
460 *flag |= REF_ISPACKED;
461 return ref;
463 list = list->next;
465 if (reading || errno != ENOENT)
466 return NULL;
467 hashclr(sha1);
468 return ref;
471 /* Follow "normalized" - ie "refs/.." symlinks by hand */
472 if (S_ISLNK(st.st_mode)) {
473 len = readlink(path, buffer, sizeof(buffer)-1);
474 if (len >= 5 && !memcmp("refs/", buffer, 5)) {
475 buffer[len] = 0;
476 strcpy(ref_buffer, buffer);
477 ref = ref_buffer;
478 if (flag)
479 *flag |= REF_ISSYMREF;
480 continue;
484 /* Is it a directory? */
485 if (S_ISDIR(st.st_mode)) {
486 errno = EISDIR;
487 return NULL;
491 * Anything else, just open it and try to use it as
492 * a ref
494 fd = open(path, O_RDONLY);
495 if (fd < 0)
496 return NULL;
497 len = read_in_full(fd, buffer, sizeof(buffer)-1);
498 close(fd);
501 * Is it a symbolic ref?
503 if (len < 4 || memcmp("ref:", buffer, 4))
504 break;
505 buf = buffer + 4;
506 len -= 4;
507 while (len && isspace(*buf))
508 buf++, len--;
509 while (len && isspace(buf[len-1]))
510 len--;
511 buf[len] = 0;
512 memcpy(ref_buffer, buf, len + 1);
513 ref = ref_buffer;
514 if (flag)
515 *flag |= REF_ISSYMREF;
517 if (len < 40 || get_sha1_hex(buffer, sha1))
518 return NULL;
519 return ref;
522 int read_ref(const char *ref, unsigned char *sha1)
524 if (resolve_ref(ref, sha1, 1, NULL))
525 return 0;
526 return -1;
529 #define DO_FOR_EACH_INCLUDE_BROKEN 01
530 static int do_one_ref(const char *base, each_ref_fn fn, int trim,
531 int flags, void *cb_data, struct ref_list *entry)
533 if (strncmp(base, entry->name, trim))
534 return 0;
535 /* Is this a "negative ref" that represents a deleted ref? */
536 if (is_null_sha1(entry->sha1))
537 return 0;
538 if (!(flags & DO_FOR_EACH_INCLUDE_BROKEN)) {
539 if (!has_sha1_file(entry->sha1)) {
540 error("%s does not point to a valid object!", entry->name);
541 return 0;
544 current_ref = entry;
545 return fn(entry->name + trim, entry->sha1, entry->flag, cb_data);
548 int peel_ref(const char *ref, unsigned char *sha1)
550 int flag;
551 unsigned char base[20];
552 struct object *o;
554 if (current_ref && (current_ref->name == ref
555 || !strcmp(current_ref->name, ref))) {
556 if (current_ref->flag & REF_KNOWS_PEELED) {
557 hashcpy(sha1, current_ref->peeled);
558 return 0;
560 hashcpy(base, current_ref->sha1);
561 goto fallback;
564 if (!resolve_ref(ref, base, 1, &flag))
565 return -1;
567 if ((flag & REF_ISPACKED)) {
568 struct ref_list *list = get_packed_refs();
570 while (list) {
571 if (!strcmp(list->name, ref)) {
572 if (list->flag & REF_KNOWS_PEELED) {
573 hashcpy(sha1, list->peeled);
574 return 0;
576 /* older pack-refs did not leave peeled ones */
577 break;
579 list = list->next;
583 fallback:
584 o = parse_object(base);
585 if (o && o->type == OBJ_TAG) {
586 o = deref_tag(o, ref, 0);
587 if (o) {
588 hashcpy(sha1, o->sha1);
589 return 0;
592 return -1;
595 static int do_for_each_ref(const char *base, each_ref_fn fn, int trim,
596 int flags, void *cb_data)
598 int retval = 0;
599 struct ref_list *packed = get_packed_refs();
600 struct ref_list *loose = get_loose_refs();
602 struct ref_list *extra;
604 for (extra = extra_refs; extra; extra = extra->next)
605 retval = do_one_ref(base, fn, trim, flags, cb_data, extra);
607 while (packed && loose) {
608 struct ref_list *entry;
609 int cmp = strcmp(packed->name, loose->name);
610 if (!cmp) {
611 packed = packed->next;
612 continue;
614 if (cmp > 0) {
615 entry = loose;
616 loose = loose->next;
617 } else {
618 entry = packed;
619 packed = packed->next;
621 retval = do_one_ref(base, fn, trim, flags, cb_data, entry);
622 if (retval)
623 goto end_each;
626 for (packed = packed ? packed : loose; packed; packed = packed->next) {
627 retval = do_one_ref(base, fn, trim, flags, cb_data, packed);
628 if (retval)
629 goto end_each;
632 end_each:
633 current_ref = NULL;
634 return retval;
637 int head_ref(each_ref_fn fn, void *cb_data)
639 unsigned char sha1[20];
640 int flag;
642 if (resolve_ref("HEAD", sha1, 1, &flag))
643 return fn("HEAD", sha1, flag, cb_data);
644 return 0;
647 int for_each_ref(each_ref_fn fn, void *cb_data)
649 return do_for_each_ref("refs/", fn, 0, 0, cb_data);
652 int for_each_ref_in(const char *prefix, each_ref_fn fn, void *cb_data)
654 return do_for_each_ref(prefix, fn, strlen(prefix), 0, cb_data);
657 int for_each_tag_ref(each_ref_fn fn, void *cb_data)
659 return for_each_ref_in("refs/tags/", fn, cb_data);
662 int for_each_branch_ref(each_ref_fn fn, void *cb_data)
664 return for_each_ref_in("refs/heads/", fn, cb_data);
667 int for_each_remote_ref(each_ref_fn fn, void *cb_data)
669 return for_each_ref_in("refs/remotes/", fn, cb_data);
672 int for_each_replace_ref(each_ref_fn fn, void *cb_data)
674 return do_for_each_ref("refs/replace/", fn, 13, 0, cb_data);
677 int for_each_namespace_ref(each_ref_fn fn, const char *ns_name, void *cb_data)
679 struct strbuf real_prefix = STRBUF_INIT;
680 int ret;
682 if (prefixcmp(ns_name, "refs/"))
683 strbuf_addstr(&real_prefix, "refs/");
684 strbuf_addstr(&real_prefix, ns_name);
685 if (real_prefix.buf[real_prefix.len - 1] != '/')
686 strbuf_addch(&real_prefix, '/');
688 ret = for_each_ref_in(real_prefix.buf, fn, cb_data);
689 strbuf_release(&real_prefix);
690 return ret;
693 int for_each_rawref(each_ref_fn fn, void *cb_data)
695 return do_for_each_ref("refs/", fn, 0,
696 DO_FOR_EACH_INCLUDE_BROKEN, cb_data);
700 * Make sure "ref" is something reasonable to have under ".git/refs/";
701 * We do not like it if:
703 * - any path component of it begins with ".", or
704 * - it has double dots "..", or
705 * - it has ASCII control character, "~", "^", ":" or SP, anywhere, or
706 * - it ends with a "/".
707 * - it ends with ".lock"
708 * - it contains a "\" (backslash)
711 static inline int bad_ref_char(int ch)
713 if (((unsigned) ch) <= ' ' ||
714 ch == '~' || ch == '^' || ch == ':' || ch == '\\')
715 return 1;
716 /* 2.13 Pattern Matching Notation */
717 if (ch == '?' || ch == '[') /* Unsupported */
718 return 1;
719 if (ch == '*') /* Supported at the end */
720 return 2;
721 return 0;
724 int check_ref_format(const char *ref)
726 int ch, level, bad_type, last;
727 int ret = CHECK_REF_FORMAT_OK;
728 const char *cp = ref;
730 level = 0;
731 while (1) {
732 while ((ch = *cp++) == '/')
733 ; /* tolerate duplicated slashes */
734 if (!ch)
735 /* should not end with slashes */
736 return CHECK_REF_FORMAT_ERROR;
738 /* we are at the beginning of the path component */
739 if (ch == '.')
740 return CHECK_REF_FORMAT_ERROR;
741 bad_type = bad_ref_char(ch);
742 if (bad_type) {
743 if (bad_type == 2 && (!*cp || *cp == '/') &&
744 ret == CHECK_REF_FORMAT_OK)
745 ret = CHECK_REF_FORMAT_WILDCARD;
746 else
747 return CHECK_REF_FORMAT_ERROR;
750 last = ch;
751 /* scan the rest of the path component */
752 while ((ch = *cp++) != 0) {
753 bad_type = bad_ref_char(ch);
754 if (bad_type)
755 return CHECK_REF_FORMAT_ERROR;
756 if (ch == '/')
757 break;
758 if (last == '.' && ch == '.')
759 return CHECK_REF_FORMAT_ERROR;
760 if (last == '@' && ch == '{')
761 return CHECK_REF_FORMAT_ERROR;
762 last = ch;
764 level++;
765 if (!ch) {
766 if (ref <= cp - 2 && cp[-2] == '.')
767 return CHECK_REF_FORMAT_ERROR;
768 if (level < 2)
769 return CHECK_REF_FORMAT_ONELEVEL;
770 if (has_extension(ref, ".lock"))
771 return CHECK_REF_FORMAT_ERROR;
772 return ret;
777 const char *prettify_refname(const char *name)
779 return name + (
780 !prefixcmp(name, "refs/heads/") ? 11 :
781 !prefixcmp(name, "refs/tags/") ? 10 :
782 !prefixcmp(name, "refs/remotes/") ? 13 :
786 const char *ref_rev_parse_rules[] = {
787 "%.*s",
788 "refs/%.*s",
789 "refs/tags/%.*s",
790 "refs/heads/%.*s",
791 "refs/remotes/%.*s",
792 "refs/remotes/%.*s/HEAD",
793 NULL
796 const char *ref_fetch_rules[] = {
797 "%.*s",
798 "refs/%.*s",
799 "refs/heads/%.*s",
800 NULL
803 int refname_match(const char *abbrev_name, const char *full_name, const char **rules)
805 const char **p;
806 const int abbrev_name_len = strlen(abbrev_name);
808 for (p = rules; *p; p++) {
809 if (!strcmp(full_name, mkpath(*p, abbrev_name_len, abbrev_name))) {
810 return 1;
814 return 0;
817 static struct ref_lock *verify_lock(struct ref_lock *lock,
818 const unsigned char *old_sha1, int mustexist)
820 if (!resolve_ref(lock->ref_name, lock->old_sha1, mustexist, NULL)) {
821 error("Can't verify ref %s", lock->ref_name);
822 unlock_ref(lock);
823 return NULL;
825 if (hashcmp(lock->old_sha1, old_sha1)) {
826 error("Ref %s is at %s but expected %s", lock->ref_name,
827 sha1_to_hex(lock->old_sha1), sha1_to_hex(old_sha1));
828 unlock_ref(lock);
829 return NULL;
831 return lock;
834 static int remove_empty_directories(const char *file)
836 /* we want to create a file but there is a directory there;
837 * if that is an empty directory (or a directory that contains
838 * only empty directories), remove them.
840 struct strbuf path;
841 int result;
843 strbuf_init(&path, 20);
844 strbuf_addstr(&path, file);
846 result = remove_dir_recursively(&path, REMOVE_DIR_EMPTY_ONLY);
848 strbuf_release(&path);
850 return result;
853 static int is_refname_available(const char *ref, const char *oldref,
854 struct ref_list *list, int quiet)
856 int namlen = strlen(ref); /* e.g. 'foo/bar' */
857 while (list) {
858 /* list->name could be 'foo' or 'foo/bar/baz' */
859 if (!oldref || strcmp(oldref, list->name)) {
860 int len = strlen(list->name);
861 int cmplen = (namlen < len) ? namlen : len;
862 const char *lead = (namlen < len) ? list->name : ref;
863 if (!strncmp(ref, list->name, cmplen) &&
864 lead[cmplen] == '/') {
865 if (!quiet)
866 error("'%s' exists; cannot create '%s'",
867 list->name, ref);
868 return 0;
871 list = list->next;
873 return 1;
876 static struct ref_lock *lock_ref_sha1_basic(const char *ref, const unsigned char *old_sha1, int flags, int *type_p)
878 char *ref_file;
879 const char *orig_ref = ref;
880 struct ref_lock *lock;
881 int last_errno = 0;
882 int type, lflags;
883 int mustexist = (old_sha1 && !is_null_sha1(old_sha1));
884 int missing = 0;
886 lock = xcalloc(1, sizeof(struct ref_lock));
887 lock->lock_fd = -1;
889 ref = resolve_ref(ref, lock->old_sha1, mustexist, &type);
890 if (!ref && errno == EISDIR) {
891 /* we are trying to lock foo but we used to
892 * have foo/bar which now does not exist;
893 * it is normal for the empty directory 'foo'
894 * to remain.
896 ref_file = git_path("%s", orig_ref);
897 if (remove_empty_directories(ref_file)) {
898 last_errno = errno;
899 error("there are still refs under '%s'", orig_ref);
900 goto error_return;
902 ref = resolve_ref(orig_ref, lock->old_sha1, mustexist, &type);
904 if (type_p)
905 *type_p = type;
906 if (!ref) {
907 last_errno = errno;
908 error("unable to resolve reference %s: %s",
909 orig_ref, strerror(errno));
910 goto error_return;
912 missing = is_null_sha1(lock->old_sha1);
913 /* When the ref did not exist and we are creating it,
914 * make sure there is no existing ref that is packed
915 * whose name begins with our refname, nor a ref whose
916 * name is a proper prefix of our refname.
918 if (missing &&
919 !is_refname_available(ref, NULL, get_packed_refs(), 0)) {
920 last_errno = ENOTDIR;
921 goto error_return;
924 lock->lk = xcalloc(1, sizeof(struct lock_file));
926 lflags = LOCK_DIE_ON_ERROR;
927 if (flags & REF_NODEREF) {
928 ref = orig_ref;
929 lflags |= LOCK_NODEREF;
931 lock->ref_name = xstrdup(ref);
932 lock->orig_ref_name = xstrdup(orig_ref);
933 ref_file = git_path("%s", ref);
934 if (missing)
935 lock->force_write = 1;
936 if ((flags & REF_NODEREF) && (type & REF_ISSYMREF))
937 lock->force_write = 1;
939 if (safe_create_leading_directories(ref_file)) {
940 last_errno = errno;
941 error("unable to create directory for %s", ref_file);
942 goto error_return;
945 lock->lock_fd = hold_lock_file_for_update(lock->lk, ref_file, lflags);
946 return old_sha1 ? verify_lock(lock, old_sha1, mustexist) : lock;
948 error_return:
949 unlock_ref(lock);
950 errno = last_errno;
951 return NULL;
954 struct ref_lock *lock_ref_sha1(const char *ref, const unsigned char *old_sha1)
956 char refpath[PATH_MAX];
957 if (check_ref_format(ref))
958 return NULL;
959 strcpy(refpath, mkpath("refs/%s", ref));
960 return lock_ref_sha1_basic(refpath, old_sha1, 0, NULL);
963 struct ref_lock *lock_any_ref_for_update(const char *ref, const unsigned char *old_sha1, int flags)
965 switch (check_ref_format(ref)) {
966 default:
967 return NULL;
968 case 0:
969 case CHECK_REF_FORMAT_ONELEVEL:
970 return lock_ref_sha1_basic(ref, old_sha1, flags, NULL);
974 static struct lock_file packlock;
976 static int repack_without_ref(const char *refname)
978 struct ref_list *list, *packed_ref_list;
979 int fd;
980 int found = 0;
982 packed_ref_list = get_packed_refs();
983 for (list = packed_ref_list; list; list = list->next) {
984 if (!strcmp(refname, list->name)) {
985 found = 1;
986 break;
989 if (!found)
990 return 0;
991 fd = hold_lock_file_for_update(&packlock, git_path("packed-refs"), 0);
992 if (fd < 0) {
993 unable_to_lock_error(git_path("packed-refs"), errno);
994 return error("cannot delete '%s' from packed refs", refname);
997 for (list = packed_ref_list; list; list = list->next) {
998 char line[PATH_MAX + 100];
999 int len;
1001 if (!strcmp(refname, list->name))
1002 continue;
1003 len = snprintf(line, sizeof(line), "%s %s\n",
1004 sha1_to_hex(list->sha1), list->name);
1005 /* this should not happen but just being defensive */
1006 if (len > sizeof(line))
1007 die("too long a refname '%s'", list->name);
1008 write_or_die(fd, line, len);
1010 return commit_lock_file(&packlock);
1013 int delete_ref(const char *refname, const unsigned char *sha1, int delopt)
1015 struct ref_lock *lock;
1016 int err, i = 0, ret = 0, flag = 0;
1018 lock = lock_ref_sha1_basic(refname, sha1, 0, &flag);
1019 if (!lock)
1020 return 1;
1021 if (!(flag & REF_ISPACKED) || flag & REF_ISSYMREF) {
1022 /* loose */
1023 const char *path;
1025 if (!(delopt & REF_NODEREF)) {
1026 i = strlen(lock->lk->filename) - 5; /* .lock */
1027 lock->lk->filename[i] = 0;
1028 path = lock->lk->filename;
1029 } else {
1030 path = git_path("%s", refname);
1032 err = unlink_or_warn(path);
1033 if (err && errno != ENOENT)
1034 ret = 1;
1036 if (!(delopt & REF_NODEREF))
1037 lock->lk->filename[i] = '.';
1039 /* removing the loose one could have resurrected an earlier
1040 * packed one. Also, if it was not loose we need to repack
1041 * without it.
1043 ret |= repack_without_ref(refname);
1045 unlink_or_warn(git_path("logs/%s", lock->ref_name));
1046 invalidate_cached_refs();
1047 unlock_ref(lock);
1048 return ret;
1051 int rename_ref(const char *oldref, const char *newref, const char *logmsg)
1053 static const char renamed_ref[] = "RENAMED-REF";
1054 unsigned char sha1[20], orig_sha1[20];
1055 int flag = 0, logmoved = 0;
1056 struct ref_lock *lock;
1057 struct stat loginfo;
1058 int log = !lstat(git_path("logs/%s", oldref), &loginfo);
1059 const char *symref = NULL;
1061 if (log && S_ISLNK(loginfo.st_mode))
1062 return error("reflog for %s is a symlink", oldref);
1064 symref = resolve_ref(oldref, orig_sha1, 1, &flag);
1065 if (flag & REF_ISSYMREF)
1066 return error("refname %s is a symbolic ref, renaming it is not supported",
1067 oldref);
1068 if (!symref)
1069 return error("refname %s not found", oldref);
1071 if (!is_refname_available(newref, oldref, get_packed_refs(), 0))
1072 return 1;
1074 if (!is_refname_available(newref, oldref, get_loose_refs(), 0))
1075 return 1;
1077 lock = lock_ref_sha1_basic(renamed_ref, NULL, 0, NULL);
1078 if (!lock)
1079 return error("unable to lock %s", renamed_ref);
1080 lock->force_write = 1;
1081 if (write_ref_sha1(lock, orig_sha1, logmsg))
1082 return error("unable to save current sha1 in %s", renamed_ref);
1084 if (log && rename(git_path("logs/%s", oldref), git_path("tmp-renamed-log")))
1085 return error("unable to move logfile logs/%s to tmp-renamed-log: %s",
1086 oldref, strerror(errno));
1088 if (delete_ref(oldref, orig_sha1, REF_NODEREF)) {
1089 error("unable to delete old %s", oldref);
1090 goto rollback;
1093 if (resolve_ref(newref, sha1, 1, &flag) && delete_ref(newref, sha1, REF_NODEREF)) {
1094 if (errno==EISDIR) {
1095 if (remove_empty_directories(git_path("%s", newref))) {
1096 error("Directory not empty: %s", newref);
1097 goto rollback;
1099 } else {
1100 error("unable to delete existing %s", newref);
1101 goto rollback;
1105 if (log && safe_create_leading_directories(git_path("logs/%s", newref))) {
1106 error("unable to create directory for %s", newref);
1107 goto rollback;
1110 retry:
1111 if (log && rename(git_path("tmp-renamed-log"), git_path("logs/%s", newref))) {
1112 if (errno==EISDIR || errno==ENOTDIR) {
1114 * rename(a, b) when b is an existing
1115 * directory ought to result in ISDIR, but
1116 * Solaris 5.8 gives ENOTDIR. Sheesh.
1118 if (remove_empty_directories(git_path("logs/%s", newref))) {
1119 error("Directory not empty: logs/%s", newref);
1120 goto rollback;
1122 goto retry;
1123 } else {
1124 error("unable to move logfile tmp-renamed-log to logs/%s: %s",
1125 newref, strerror(errno));
1126 goto rollback;
1129 logmoved = log;
1131 lock = lock_ref_sha1_basic(newref, NULL, 0, NULL);
1132 if (!lock) {
1133 error("unable to lock %s for update", newref);
1134 goto rollback;
1136 lock->force_write = 1;
1137 hashcpy(lock->old_sha1, orig_sha1);
1138 if (write_ref_sha1(lock, orig_sha1, logmsg)) {
1139 error("unable to write current sha1 into %s", newref);
1140 goto rollback;
1143 return 0;
1145 rollback:
1146 lock = lock_ref_sha1_basic(oldref, NULL, 0, NULL);
1147 if (!lock) {
1148 error("unable to lock %s for rollback", oldref);
1149 goto rollbacklog;
1152 lock->force_write = 1;
1153 flag = log_all_ref_updates;
1154 log_all_ref_updates = 0;
1155 if (write_ref_sha1(lock, orig_sha1, NULL))
1156 error("unable to write current sha1 into %s", oldref);
1157 log_all_ref_updates = flag;
1159 rollbacklog:
1160 if (logmoved && rename(git_path("logs/%s", newref), git_path("logs/%s", oldref)))
1161 error("unable to restore logfile %s from %s: %s",
1162 oldref, newref, strerror(errno));
1163 if (!logmoved && log &&
1164 rename(git_path("tmp-renamed-log"), git_path("logs/%s", oldref)))
1165 error("unable to restore logfile %s from tmp-renamed-log: %s",
1166 oldref, strerror(errno));
1168 return 1;
1171 int close_ref(struct ref_lock *lock)
1173 if (close_lock_file(lock->lk))
1174 return -1;
1175 lock->lock_fd = -1;
1176 return 0;
1179 int commit_ref(struct ref_lock *lock)
1181 if (commit_lock_file(lock->lk))
1182 return -1;
1183 lock->lock_fd = -1;
1184 return 0;
1187 void unlock_ref(struct ref_lock *lock)
1189 /* Do not free lock->lk -- atexit() still looks at them */
1190 if (lock->lk)
1191 rollback_lock_file(lock->lk);
1192 free(lock->ref_name);
1193 free(lock->orig_ref_name);
1194 free(lock);
1198 * copy the reflog message msg to buf, which has been allocated sufficiently
1199 * large, while cleaning up the whitespaces. Especially, convert LF to space,
1200 * because reflog file is one line per entry.
1202 static int copy_msg(char *buf, const char *msg)
1204 char *cp = buf;
1205 char c;
1206 int wasspace = 1;
1208 *cp++ = '\t';
1209 while ((c = *msg++)) {
1210 if (wasspace && isspace(c))
1211 continue;
1212 wasspace = isspace(c);
1213 if (wasspace)
1214 c = ' ';
1215 *cp++ = c;
1217 while (buf < cp && isspace(cp[-1]))
1218 cp--;
1219 *cp++ = '\n';
1220 return cp - buf;
1223 static int log_ref_write(const char *ref_name, const unsigned char *old_sha1,
1224 const unsigned char *new_sha1, const char *msg)
1226 int logfd, written, oflags = O_APPEND | O_WRONLY;
1227 unsigned maxlen, len;
1228 int msglen;
1229 char log_file[PATH_MAX];
1230 char *logrec;
1231 const char *committer;
1233 if (log_all_ref_updates < 0)
1234 log_all_ref_updates = !is_bare_repository();
1236 git_snpath(log_file, sizeof(log_file), "logs/%s", ref_name);
1238 if (log_all_ref_updates &&
1239 (!prefixcmp(ref_name, "refs/heads/") ||
1240 !prefixcmp(ref_name, "refs/remotes/") ||
1241 !strcmp(ref_name, "HEAD"))) {
1242 if (safe_create_leading_directories(log_file) < 0)
1243 return error("unable to create directory for %s",
1244 log_file);
1245 oflags |= O_CREAT;
1248 logfd = open(log_file, oflags, 0666);
1249 if (logfd < 0) {
1250 if (!(oflags & O_CREAT) && errno == ENOENT)
1251 return 0;
1253 if ((oflags & O_CREAT) && errno == EISDIR) {
1254 if (remove_empty_directories(log_file)) {
1255 return error("There are still logs under '%s'",
1256 log_file);
1258 logfd = open(log_file, oflags, 0666);
1261 if (logfd < 0)
1262 return error("Unable to append to %s: %s",
1263 log_file, strerror(errno));
1266 adjust_shared_perm(log_file);
1268 msglen = msg ? strlen(msg) : 0;
1269 committer = git_committer_info(0);
1270 maxlen = strlen(committer) + msglen + 100;
1271 logrec = xmalloc(maxlen);
1272 len = sprintf(logrec, "%s %s %s\n",
1273 sha1_to_hex(old_sha1),
1274 sha1_to_hex(new_sha1),
1275 committer);
1276 if (msglen)
1277 len += copy_msg(logrec + len - 1, msg) - 1;
1278 written = len <= maxlen ? write_in_full(logfd, logrec, len) : -1;
1279 free(logrec);
1280 if (close(logfd) != 0 || written != len)
1281 return error("Unable to append to %s", log_file);
1282 return 0;
1285 static int is_branch(const char *refname)
1287 return !strcmp(refname, "HEAD") || !prefixcmp(refname, "refs/heads/");
1290 int write_ref_sha1(struct ref_lock *lock,
1291 const unsigned char *sha1, const char *logmsg)
1293 static char term = '\n';
1294 struct object *o;
1296 if (!lock)
1297 return -1;
1298 if (!lock->force_write && !hashcmp(lock->old_sha1, sha1)) {
1299 unlock_ref(lock);
1300 return 0;
1302 o = parse_object(sha1);
1303 if (!o) {
1304 error("Trying to write ref %s with nonexistant object %s",
1305 lock->ref_name, sha1_to_hex(sha1));
1306 unlock_ref(lock);
1307 return -1;
1309 if (o->type != OBJ_COMMIT && is_branch(lock->ref_name)) {
1310 error("Trying to write non-commit object %s to branch %s",
1311 sha1_to_hex(sha1), lock->ref_name);
1312 unlock_ref(lock);
1313 return -1;
1315 if (write_in_full(lock->lock_fd, sha1_to_hex(sha1), 40) != 40 ||
1316 write_in_full(lock->lock_fd, &term, 1) != 1
1317 || close_ref(lock) < 0) {
1318 error("Couldn't write %s", lock->lk->filename);
1319 unlock_ref(lock);
1320 return -1;
1322 invalidate_cached_refs();
1323 if (log_ref_write(lock->ref_name, lock->old_sha1, sha1, logmsg) < 0 ||
1324 (strcmp(lock->ref_name, lock->orig_ref_name) &&
1325 log_ref_write(lock->orig_ref_name, lock->old_sha1, sha1, logmsg) < 0)) {
1326 unlock_ref(lock);
1327 return -1;
1329 if (strcmp(lock->orig_ref_name, "HEAD") != 0) {
1331 * Special hack: If a branch is updated directly and HEAD
1332 * points to it (may happen on the remote side of a push
1333 * for example) then logically the HEAD reflog should be
1334 * updated too.
1335 * A generic solution implies reverse symref information,
1336 * but finding all symrefs pointing to the given branch
1337 * would be rather costly for this rare event (the direct
1338 * update of a branch) to be worth it. So let's cheat and
1339 * check with HEAD only which should cover 99% of all usage
1340 * scenarios (even 100% of the default ones).
1342 unsigned char head_sha1[20];
1343 int head_flag;
1344 const char *head_ref;
1345 head_ref = resolve_ref("HEAD", head_sha1, 1, &head_flag);
1346 if (head_ref && (head_flag & REF_ISSYMREF) &&
1347 !strcmp(head_ref, lock->ref_name))
1348 log_ref_write("HEAD", lock->old_sha1, sha1, logmsg);
1350 if (commit_ref(lock)) {
1351 error("Couldn't set %s", lock->ref_name);
1352 unlock_ref(lock);
1353 return -1;
1355 unlock_ref(lock);
1356 return 0;
1359 int create_symref(const char *ref_target, const char *refs_heads_master,
1360 const char *logmsg)
1362 const char *lockpath;
1363 char ref[1000];
1364 int fd, len, written;
1365 char *git_HEAD = git_pathdup("%s", ref_target);
1366 unsigned char old_sha1[20], new_sha1[20];
1368 if (logmsg && read_ref(ref_target, old_sha1))
1369 hashclr(old_sha1);
1371 if (safe_create_leading_directories(git_HEAD) < 0)
1372 return error("unable to create directory for %s", git_HEAD);
1374 #ifndef NO_SYMLINK_HEAD
1375 if (prefer_symlink_refs) {
1376 unlink(git_HEAD);
1377 if (!symlink(refs_heads_master, git_HEAD))
1378 goto done;
1379 fprintf(stderr, "no symlink - falling back to symbolic ref\n");
1381 #endif
1383 len = snprintf(ref, sizeof(ref), "ref: %s\n", refs_heads_master);
1384 if (sizeof(ref) <= len) {
1385 error("refname too long: %s", refs_heads_master);
1386 goto error_free_return;
1388 lockpath = mkpath("%s.lock", git_HEAD);
1389 fd = open(lockpath, O_CREAT | O_EXCL | O_WRONLY, 0666);
1390 if (fd < 0) {
1391 error("Unable to open %s for writing", lockpath);
1392 goto error_free_return;
1394 written = write_in_full(fd, ref, len);
1395 if (close(fd) != 0 || written != len) {
1396 error("Unable to write to %s", lockpath);
1397 goto error_unlink_return;
1399 if (rename(lockpath, git_HEAD) < 0) {
1400 error("Unable to create %s", git_HEAD);
1401 goto error_unlink_return;
1403 if (adjust_shared_perm(git_HEAD)) {
1404 error("Unable to fix permissions on %s", lockpath);
1405 error_unlink_return:
1406 unlink_or_warn(lockpath);
1407 error_free_return:
1408 free(git_HEAD);
1409 return -1;
1412 #ifndef NO_SYMLINK_HEAD
1413 done:
1414 #endif
1415 if (logmsg && !read_ref(refs_heads_master, new_sha1))
1416 log_ref_write(ref_target, old_sha1, new_sha1, logmsg);
1418 free(git_HEAD);
1419 return 0;
1422 static char *ref_msg(const char *line, const char *endp)
1424 const char *ep;
1425 line += 82;
1426 ep = memchr(line, '\n', endp - line);
1427 if (!ep)
1428 ep = endp;
1429 return xmemdupz(line, ep - line);
1432 int read_ref_at(const char *ref, unsigned long at_time, int cnt, unsigned char *sha1, char **msg, unsigned long *cutoff_time, int *cutoff_tz, int *cutoff_cnt)
1434 const char *logfile, *logdata, *logend, *rec, *lastgt, *lastrec;
1435 char *tz_c;
1436 int logfd, tz, reccnt = 0;
1437 struct stat st;
1438 unsigned long date;
1439 unsigned char logged_sha1[20];
1440 void *log_mapped;
1441 size_t mapsz;
1443 logfile = git_path("logs/%s", ref);
1444 logfd = open(logfile, O_RDONLY, 0);
1445 if (logfd < 0)
1446 die_errno("Unable to read log '%s'", logfile);
1447 fstat(logfd, &st);
1448 if (!st.st_size)
1449 die("Log %s is empty.", logfile);
1450 mapsz = xsize_t(st.st_size);
1451 log_mapped = xmmap(NULL, mapsz, PROT_READ, MAP_PRIVATE, logfd, 0);
1452 logdata = log_mapped;
1453 close(logfd);
1455 lastrec = NULL;
1456 rec = logend = logdata + st.st_size;
1457 while (logdata < rec) {
1458 reccnt++;
1459 if (logdata < rec && *(rec-1) == '\n')
1460 rec--;
1461 lastgt = NULL;
1462 while (logdata < rec && *(rec-1) != '\n') {
1463 rec--;
1464 if (*rec == '>')
1465 lastgt = rec;
1467 if (!lastgt)
1468 die("Log %s is corrupt.", logfile);
1469 date = strtoul(lastgt + 1, &tz_c, 10);
1470 if (date <= at_time || cnt == 0) {
1471 tz = strtoul(tz_c, NULL, 10);
1472 if (msg)
1473 *msg = ref_msg(rec, logend);
1474 if (cutoff_time)
1475 *cutoff_time = date;
1476 if (cutoff_tz)
1477 *cutoff_tz = tz;
1478 if (cutoff_cnt)
1479 *cutoff_cnt = reccnt - 1;
1480 if (lastrec) {
1481 if (get_sha1_hex(lastrec, logged_sha1))
1482 die("Log %s is corrupt.", logfile);
1483 if (get_sha1_hex(rec + 41, sha1))
1484 die("Log %s is corrupt.", logfile);
1485 if (hashcmp(logged_sha1, sha1)) {
1486 warning("Log %s has gap after %s.",
1487 logfile, show_date(date, tz, DATE_RFC2822));
1490 else if (date == at_time) {
1491 if (get_sha1_hex(rec + 41, sha1))
1492 die("Log %s is corrupt.", logfile);
1494 else {
1495 if (get_sha1_hex(rec + 41, logged_sha1))
1496 die("Log %s is corrupt.", logfile);
1497 if (hashcmp(logged_sha1, sha1)) {
1498 warning("Log %s unexpectedly ended on %s.",
1499 logfile, show_date(date, tz, DATE_RFC2822));
1502 munmap(log_mapped, mapsz);
1503 return 0;
1505 lastrec = rec;
1506 if (cnt > 0)
1507 cnt--;
1510 rec = logdata;
1511 while (rec < logend && *rec != '>' && *rec != '\n')
1512 rec++;
1513 if (rec == logend || *rec == '\n')
1514 die("Log %s is corrupt.", logfile);
1515 date = strtoul(rec + 1, &tz_c, 10);
1516 tz = strtoul(tz_c, NULL, 10);
1517 if (get_sha1_hex(logdata, sha1))
1518 die("Log %s is corrupt.", logfile);
1519 if (is_null_sha1(sha1)) {
1520 if (get_sha1_hex(logdata + 41, sha1))
1521 die("Log %s is corrupt.", logfile);
1523 if (msg)
1524 *msg = ref_msg(logdata, logend);
1525 munmap(log_mapped, mapsz);
1527 if (cutoff_time)
1528 *cutoff_time = date;
1529 if (cutoff_tz)
1530 *cutoff_tz = tz;
1531 if (cutoff_cnt)
1532 *cutoff_cnt = reccnt;
1533 return 1;
1536 int for_each_recent_reflog_ent(const char *ref, each_reflog_ent_fn fn, long ofs, void *cb_data)
1538 const char *logfile;
1539 FILE *logfp;
1540 char buf[1024];
1541 int ret = 0;
1543 logfile = git_path("logs/%s", ref);
1544 logfp = fopen(logfile, "r");
1545 if (!logfp)
1546 return -1;
1548 if (ofs) {
1549 struct stat statbuf;
1550 if (fstat(fileno(logfp), &statbuf) ||
1551 statbuf.st_size < ofs ||
1552 fseek(logfp, -ofs, SEEK_END) ||
1553 fgets(buf, sizeof(buf), logfp)) {
1554 fclose(logfp);
1555 return -1;
1559 while (fgets(buf, sizeof(buf), logfp)) {
1560 unsigned char osha1[20], nsha1[20];
1561 char *email_end, *message;
1562 unsigned long timestamp;
1563 int len, tz;
1565 /* old SP new SP name <email> SP time TAB msg LF */
1566 len = strlen(buf);
1567 if (len < 83 || buf[len-1] != '\n' ||
1568 get_sha1_hex(buf, osha1) || buf[40] != ' ' ||
1569 get_sha1_hex(buf + 41, nsha1) || buf[81] != ' ' ||
1570 !(email_end = strchr(buf + 82, '>')) ||
1571 email_end[1] != ' ' ||
1572 !(timestamp = strtoul(email_end + 2, &message, 10)) ||
1573 !message || message[0] != ' ' ||
1574 (message[1] != '+' && message[1] != '-') ||
1575 !isdigit(message[2]) || !isdigit(message[3]) ||
1576 !isdigit(message[4]) || !isdigit(message[5]))
1577 continue; /* corrupt? */
1578 email_end[1] = '\0';
1579 tz = strtol(message + 1, NULL, 10);
1580 if (message[6] != '\t')
1581 message += 6;
1582 else
1583 message += 7;
1584 ret = fn(osha1, nsha1, buf+82, timestamp, tz, message, cb_data);
1585 if (ret)
1586 break;
1588 fclose(logfp);
1589 return ret;
1592 int for_each_reflog_ent(const char *ref, each_reflog_ent_fn fn, void *cb_data)
1594 return for_each_recent_reflog_ent(ref, fn, 0, cb_data);
1597 static int do_for_each_reflog(const char *base, each_ref_fn fn, void *cb_data)
1599 DIR *dir = opendir(git_path("logs/%s", base));
1600 int retval = 0;
1602 if (dir) {
1603 struct dirent *de;
1604 int baselen = strlen(base);
1605 char *log = xmalloc(baselen + 257);
1607 memcpy(log, base, baselen);
1608 if (baselen && base[baselen-1] != '/')
1609 log[baselen++] = '/';
1611 while ((de = readdir(dir)) != NULL) {
1612 struct stat st;
1613 int namelen;
1615 if (de->d_name[0] == '.')
1616 continue;
1617 namelen = strlen(de->d_name);
1618 if (namelen > 255)
1619 continue;
1620 if (has_extension(de->d_name, ".lock"))
1621 continue;
1622 memcpy(log + baselen, de->d_name, namelen+1);
1623 if (stat(git_path("logs/%s", log), &st) < 0)
1624 continue;
1625 if (S_ISDIR(st.st_mode)) {
1626 retval = do_for_each_reflog(log, fn, cb_data);
1627 } else {
1628 unsigned char sha1[20];
1629 if (!resolve_ref(log, sha1, 0, NULL))
1630 retval = error("bad ref for %s", log);
1631 else
1632 retval = fn(log, sha1, 0, cb_data);
1634 if (retval)
1635 break;
1637 free(log);
1638 closedir(dir);
1640 else if (*base)
1641 return errno;
1642 return retval;
1645 int for_each_reflog(each_ref_fn fn, void *cb_data)
1647 return do_for_each_reflog("", fn, cb_data);
1650 int update_ref(const char *action, const char *refname,
1651 const unsigned char *sha1, const unsigned char *oldval,
1652 int flags, enum action_on_err onerr)
1654 static struct ref_lock *lock;
1655 lock = lock_any_ref_for_update(refname, oldval, flags);
1656 if (!lock) {
1657 const char *str = "Cannot lock the ref '%s'.";
1658 switch (onerr) {
1659 case MSG_ON_ERR: error(str, refname); break;
1660 case DIE_ON_ERR: die(str, refname); break;
1661 case QUIET_ON_ERR: break;
1663 return 1;
1665 if (write_ref_sha1(lock, sha1, action) < 0) {
1666 const char *str = "Cannot update the ref '%s'.";
1667 switch (onerr) {
1668 case MSG_ON_ERR: error(str, refname); break;
1669 case DIE_ON_ERR: die(str, refname); break;
1670 case QUIET_ON_ERR: break;
1672 return 1;
1674 return 0;
1677 struct ref *find_ref_by_name(const struct ref *list, const char *name)
1679 for ( ; list; list = list->next)
1680 if (!strcmp(list->name, name))
1681 return (struct ref *)list;
1682 return NULL;
1686 * generate a format suitable for scanf from a ref_rev_parse_rules
1687 * rule, that is replace the "%.*s" spec with a "%s" spec
1689 static void gen_scanf_fmt(char *scanf_fmt, const char *rule)
1691 char *spec;
1693 spec = strstr(rule, "%.*s");
1694 if (!spec || strstr(spec + 4, "%.*s"))
1695 die("invalid rule in ref_rev_parse_rules: %s", rule);
1697 /* copy all until spec */
1698 strncpy(scanf_fmt, rule, spec - rule);
1699 scanf_fmt[spec - rule] = '\0';
1700 /* copy new spec */
1701 strcat(scanf_fmt, "%s");
1702 /* copy remaining rule */
1703 strcat(scanf_fmt, spec + 4);
1705 return;
1708 char *shorten_unambiguous_ref(const char *ref, int strict)
1710 int i;
1711 static char **scanf_fmts;
1712 static int nr_rules;
1713 char *short_name;
1715 /* pre generate scanf formats from ref_rev_parse_rules[] */
1716 if (!nr_rules) {
1717 size_t total_len = 0;
1719 /* the rule list is NULL terminated, count them first */
1720 for (; ref_rev_parse_rules[nr_rules]; nr_rules++)
1721 /* no +1 because strlen("%s") < strlen("%.*s") */
1722 total_len += strlen(ref_rev_parse_rules[nr_rules]);
1724 scanf_fmts = xmalloc(nr_rules * sizeof(char *) + total_len);
1726 total_len = 0;
1727 for (i = 0; i < nr_rules; i++) {
1728 scanf_fmts[i] = (char *)&scanf_fmts[nr_rules]
1729 + total_len;
1730 gen_scanf_fmt(scanf_fmts[i], ref_rev_parse_rules[i]);
1731 total_len += strlen(ref_rev_parse_rules[i]);
1735 /* bail out if there are no rules */
1736 if (!nr_rules)
1737 return xstrdup(ref);
1739 /* buffer for scanf result, at most ref must fit */
1740 short_name = xstrdup(ref);
1742 /* skip first rule, it will always match */
1743 for (i = nr_rules - 1; i > 0 ; --i) {
1744 int j;
1745 int rules_to_fail = i;
1746 int short_name_len;
1748 if (1 != sscanf(ref, scanf_fmts[i], short_name))
1749 continue;
1751 short_name_len = strlen(short_name);
1754 * in strict mode, all (except the matched one) rules
1755 * must fail to resolve to a valid non-ambiguous ref
1757 if (strict)
1758 rules_to_fail = nr_rules;
1761 * check if the short name resolves to a valid ref,
1762 * but use only rules prior to the matched one
1764 for (j = 0; j < rules_to_fail; j++) {
1765 const char *rule = ref_rev_parse_rules[j];
1766 unsigned char short_objectname[20];
1767 char refname[PATH_MAX];
1769 /* skip matched rule */
1770 if (i == j)
1771 continue;
1774 * the short name is ambiguous, if it resolves
1775 * (with this previous rule) to a valid ref
1776 * read_ref() returns 0 on success
1778 mksnpath(refname, sizeof(refname),
1779 rule, short_name_len, short_name);
1780 if (!read_ref(refname, short_objectname))
1781 break;
1785 * short name is non-ambiguous if all previous rules
1786 * haven't resolved to a valid ref
1788 if (j == rules_to_fail)
1789 return short_name;
1792 free(short_name);
1793 return xstrdup(ref);