Code

create_ref_entry(): extract function from add_ref()
[git.git] / refs.c
1 #include "cache.h"
2 #include "refs.h"
3 #include "object.h"
4 #include "tag.h"
5 #include "dir.h"
7 /* ISSYMREF=0x01, ISPACKED=0x02 and ISBROKEN=0x04 are public interfaces */
8 #define REF_KNOWS_PEELED 0x10
10 struct ref_entry {
11         unsigned char flag; /* ISSYMREF? ISPACKED? */
12         unsigned char sha1[20];
13         unsigned char peeled[20];
14         /* The full name of the reference (e.g., "refs/heads/master"): */
15         char name[FLEX_ARRAY];
16 };
18 struct ref_array {
19         int nr, alloc;
20         struct ref_entry **refs;
21 };
23 /*
24  * Parse one line from a packed-refs file.  Write the SHA1 to sha1.
25  * Return a pointer to the refname within the line (null-terminated),
26  * or NULL if there was a problem.
27  */
28 static const char *parse_ref_line(char *line, unsigned char *sha1)
29 {
30         /*
31          * 42: the answer to everything.
32          *
33          * In this case, it happens to be the answer to
34          *  40 (length of sha1 hex representation)
35          *  +1 (space in between hex and name)
36          *  +1 (newline at the end of the line)
37          */
38         int len = strlen(line) - 42;
40         if (len <= 0)
41                 return NULL;
42         if (get_sha1_hex(line, sha1) < 0)
43                 return NULL;
44         if (!isspace(line[40]))
45                 return NULL;
46         line += 41;
47         if (isspace(*line))
48                 return NULL;
49         if (line[len] != '\n')
50                 return NULL;
51         line[len] = 0;
53         return line;
54 }
56 static struct ref_entry *create_ref_entry(const char *refname,
57                                           const unsigned char *sha1, int flag,
58                                           int check_name)
59 {
60         int len;
61         struct ref_entry *ref;
63         if (check_name &&
64             check_refname_format(refname, REFNAME_ALLOW_ONELEVEL|REFNAME_DOT_COMPONENT))
65                 die("Reference has invalid format: '%s'", refname);
66         len = strlen(refname) + 1;
67         ref = xmalloc(sizeof(struct ref_entry) + len);
68         hashcpy(ref->sha1, sha1);
69         hashclr(ref->peeled);
70         memcpy(ref->name, refname, len);
71         ref->flag = flag;
72         return ref;
73 }
75 /* Add a ref_entry to the end of the ref_array (unsorted). */
76 static void add_ref(const char *refname, const unsigned char *sha1,
77                     int flag, int check_name, struct ref_array *refs,
78                     struct ref_entry **new_ref)
79 {
80         struct ref_entry *ref = create_ref_entry(refname, sha1, flag, check_name);
81         if (new_ref)
82                 *new_ref = ref;
83         ALLOC_GROW(refs->refs, refs->nr + 1, refs->alloc);
84         refs->refs[refs->nr++] = ref;
85 }
87 static int ref_entry_cmp(const void *a, const void *b)
88 {
89         struct ref_entry *one = *(struct ref_entry **)a;
90         struct ref_entry *two = *(struct ref_entry **)b;
91         return strcmp(one->name, two->name);
92 }
94 /*
95  * Emit a warning and return true iff ref1 and ref2 have the same name
96  * and the same sha1.  Die if they have the same name but different
97  * sha1s.
98  */
99 static int is_dup_ref(const struct ref_entry *ref1, const struct ref_entry *ref2)
101         if (!strcmp(ref1->name, ref2->name)) {
102                 /* Duplicate name; make sure that the SHA1s match: */
103                 if (hashcmp(ref1->sha1, ref2->sha1))
104                         die("Duplicated ref, and SHA1s don't match: %s",
105                             ref1->name);
106                 warning("Duplicated ref: %s", ref1->name);
107                 return 1;
108         } else {
109                 return 0;
110         }
113 static void sort_ref_array(struct ref_array *array)
115         int i, j;
117         /* Nothing to sort unless there are at least two entries */
118         if (array->nr < 2)
119                 return;
121         qsort(array->refs, array->nr, sizeof(*array->refs), ref_entry_cmp);
123         /* Remove any duplicates from the ref_array */
124         i = 0;
125         for (j = 1; j < array->nr; j++) {
126                 if (is_dup_ref(array->refs[i], array->refs[j])) {
127                         free(array->refs[j]);
128                         continue;
129                 }
130                 array->refs[++i] = array->refs[j];
131         }
132         array->nr = i + 1;
135 static struct ref_entry *search_ref_array(struct ref_array *array, const char *refname)
137         struct ref_entry *e, **r;
138         int len;
140         if (refname == NULL)
141                 return NULL;
143         if (!array->nr)
144                 return NULL;
146         len = strlen(refname) + 1;
147         e = xmalloc(sizeof(struct ref_entry) + len);
148         memcpy(e->name, refname, len);
150         r = bsearch(&e, array->refs, array->nr, sizeof(*array->refs), ref_entry_cmp);
152         free(e);
154         if (r == NULL)
155                 return NULL;
157         return *r;
160 /*
161  * Future: need to be in "struct repository"
162  * when doing a full libification.
163  */
164 static struct ref_cache {
165         struct ref_cache *next;
166         char did_loose;
167         char did_packed;
168         struct ref_array loose;
169         struct ref_array packed;
170         /* The submodule name, or "" for the main repo. */
171         char name[FLEX_ARRAY];
172 } *ref_cache;
174 static struct ref_entry *current_ref;
176 static struct ref_array extra_refs;
178 static void clear_ref_array(struct ref_array *array)
180         int i;
181         for (i = 0; i < array->nr; i++)
182                 free(array->refs[i]);
183         free(array->refs);
184         array->nr = array->alloc = 0;
185         array->refs = NULL;
188 static void clear_packed_ref_cache(struct ref_cache *refs)
190         if (refs->did_packed)
191                 clear_ref_array(&refs->packed);
192         refs->did_packed = 0;
195 static void clear_loose_ref_cache(struct ref_cache *refs)
197         if (refs->did_loose)
198                 clear_ref_array(&refs->loose);
199         refs->did_loose = 0;
202 static struct ref_cache *create_ref_cache(const char *submodule)
204         int len;
205         struct ref_cache *refs;
206         if (!submodule)
207                 submodule = "";
208         len = strlen(submodule) + 1;
209         refs = xcalloc(1, sizeof(struct ref_cache) + len);
210         memcpy(refs->name, submodule, len);
211         return refs;
214 /*
215  * Return a pointer to a ref_cache for the specified submodule. For
216  * the main repository, use submodule==NULL. The returned structure
217  * will be allocated and initialized but not necessarily populated; it
218  * should not be freed.
219  */
220 static struct ref_cache *get_ref_cache(const char *submodule)
222         struct ref_cache *refs = ref_cache;
223         if (!submodule)
224                 submodule = "";
225         while (refs) {
226                 if (!strcmp(submodule, refs->name))
227                         return refs;
228                 refs = refs->next;
229         }
231         refs = create_ref_cache(submodule);
232         refs->next = ref_cache;
233         ref_cache = refs;
234         return refs;
237 void invalidate_ref_cache(const char *submodule)
239         struct ref_cache *refs = get_ref_cache(submodule);
240         clear_packed_ref_cache(refs);
241         clear_loose_ref_cache(refs);
244 static void read_packed_refs(FILE *f, struct ref_array *array)
246         struct ref_entry *last = NULL;
247         char refline[PATH_MAX];
248         int flag = REF_ISPACKED;
250         while (fgets(refline, sizeof(refline), f)) {
251                 unsigned char sha1[20];
252                 const char *refname;
253                 static const char header[] = "# pack-refs with:";
255                 if (!strncmp(refline, header, sizeof(header)-1)) {
256                         const char *traits = refline + sizeof(header) - 1;
257                         if (strstr(traits, " peeled "))
258                                 flag |= REF_KNOWS_PEELED;
259                         /* perhaps other traits later as well */
260                         continue;
261                 }
263                 refname = parse_ref_line(refline, sha1);
264                 if (refname) {
265                         add_ref(refname, sha1, flag, 1, array, &last);
266                         continue;
267                 }
268                 if (last &&
269                     refline[0] == '^' &&
270                     strlen(refline) == 42 &&
271                     refline[41] == '\n' &&
272                     !get_sha1_hex(refline + 1, sha1))
273                         hashcpy(last->peeled, sha1);
274         }
275         sort_ref_array(array);
278 void add_extra_ref(const char *refname, const unsigned char *sha1, int flag)
280         add_ref(refname, sha1, flag, 0, &extra_refs, NULL);
283 void clear_extra_refs(void)
285         clear_ref_array(&extra_refs);
288 static struct ref_array *get_packed_refs(struct ref_cache *refs)
290         if (!refs->did_packed) {
291                 const char *packed_refs_file;
292                 FILE *f;
294                 if (*refs->name)
295                         packed_refs_file = git_path_submodule(refs->name, "packed-refs");
296                 else
297                         packed_refs_file = git_path("packed-refs");
298                 f = fopen(packed_refs_file, "r");
299                 if (f) {
300                         read_packed_refs(f, &refs->packed);
301                         fclose(f);
302                 }
303                 refs->did_packed = 1;
304         }
305         return &refs->packed;
308 static void get_ref_dir(struct ref_cache *refs, const char *base,
309                         struct ref_array *array)
311         DIR *dir;
312         const char *path;
314         if (*refs->name)
315                 path = git_path_submodule(refs->name, "%s", base);
316         else
317                 path = git_path("%s", base);
320         dir = opendir(path);
322         if (dir) {
323                 struct dirent *de;
324                 int baselen = strlen(base);
325                 char *refname = xmalloc(baselen + 257);
327                 memcpy(refname, base, baselen);
328                 if (baselen && base[baselen-1] != '/')
329                         refname[baselen++] = '/';
331                 while ((de = readdir(dir)) != NULL) {
332                         unsigned char sha1[20];
333                         struct stat st;
334                         int flag;
335                         int namelen;
336                         const char *refdir;
338                         if (de->d_name[0] == '.')
339                                 continue;
340                         namelen = strlen(de->d_name);
341                         if (namelen > 255)
342                                 continue;
343                         if (has_extension(de->d_name, ".lock"))
344                                 continue;
345                         memcpy(refname + baselen, de->d_name, namelen+1);
346                         refdir = *refs->name
347                                 ? git_path_submodule(refs->name, "%s", refname)
348                                 : git_path("%s", refname);
349                         if (stat(refdir, &st) < 0)
350                                 continue;
351                         if (S_ISDIR(st.st_mode)) {
352                                 get_ref_dir(refs, refname, array);
353                                 continue;
354                         }
355                         if (*refs->name) {
356                                 hashclr(sha1);
357                                 flag = 0;
358                                 if (resolve_gitlink_ref(refs->name, refname, sha1) < 0) {
359                                         hashclr(sha1);
360                                         flag |= REF_ISBROKEN;
361                                 }
362                         } else if (read_ref_full(refname, sha1, 1, &flag)) {
363                                 hashclr(sha1);
364                                 flag |= REF_ISBROKEN;
365                         }
366                         add_ref(refname, sha1, flag, 1, array, NULL);
367                 }
368                 free(refname);
369                 closedir(dir);
370         }
373 struct warn_if_dangling_data {
374         FILE *fp;
375         const char *refname;
376         const char *msg_fmt;
377 };
379 static int warn_if_dangling_symref(const char *refname, const unsigned char *sha1,
380                                    int flags, void *cb_data)
382         struct warn_if_dangling_data *d = cb_data;
383         const char *resolves_to;
384         unsigned char junk[20];
386         if (!(flags & REF_ISSYMREF))
387                 return 0;
389         resolves_to = resolve_ref(refname, junk, 0, NULL);
390         if (!resolves_to || strcmp(resolves_to, d->refname))
391                 return 0;
393         fprintf(d->fp, d->msg_fmt, refname);
394         return 0;
397 void warn_dangling_symref(FILE *fp, const char *msg_fmt, const char *refname)
399         struct warn_if_dangling_data data;
401         data.fp = fp;
402         data.refname = refname;
403         data.msg_fmt = msg_fmt;
404         for_each_rawref(warn_if_dangling_symref, &data);
407 static struct ref_array *get_loose_refs(struct ref_cache *refs)
409         if (!refs->did_loose) {
410                 get_ref_dir(refs, "refs", &refs->loose);
411                 sort_ref_array(&refs->loose);
412                 refs->did_loose = 1;
413         }
414         return &refs->loose;
417 /* We allow "recursive" symbolic refs. Only within reason, though */
418 #define MAXDEPTH 5
419 #define MAXREFLEN (1024)
421 /*
422  * Called by resolve_gitlink_ref_recursive() after it failed to read
423  * from the loose refs in ref_cache refs. Find <refname> in the
424  * packed-refs file for the submodule.
425  */
426 static int resolve_gitlink_packed_ref(struct ref_cache *refs,
427                                       const char *refname, unsigned char *sha1)
429         struct ref_entry *ref;
430         struct ref_array *array = get_packed_refs(refs);
432         ref = search_ref_array(array, refname);
433         if (ref == NULL)
434                 return -1;
436         memcpy(sha1, ref->sha1, 20);
437         return 0;
440 static int resolve_gitlink_ref_recursive(struct ref_cache *refs,
441                                          const char *refname, unsigned char *sha1,
442                                          int recursion)
444         int fd, len;
445         char buffer[128], *p;
446         char *path;
448         if (recursion > MAXDEPTH || strlen(refname) > MAXREFLEN)
449                 return -1;
450         path = *refs->name
451                 ? git_path_submodule(refs->name, "%s", refname)
452                 : git_path("%s", refname);
453         fd = open(path, O_RDONLY);
454         if (fd < 0)
455                 return resolve_gitlink_packed_ref(refs, refname, sha1);
457         len = read(fd, buffer, sizeof(buffer)-1);
458         close(fd);
459         if (len < 0)
460                 return -1;
461         while (len && isspace(buffer[len-1]))
462                 len--;
463         buffer[len] = 0;
465         /* Was it a detached head or an old-fashioned symlink? */
466         if (!get_sha1_hex(buffer, sha1))
467                 return 0;
469         /* Symref? */
470         if (strncmp(buffer, "ref:", 4))
471                 return -1;
472         p = buffer + 4;
473         while (isspace(*p))
474                 p++;
476         return resolve_gitlink_ref_recursive(refs, p, sha1, recursion+1);
479 int resolve_gitlink_ref(const char *path, const char *refname, unsigned char *sha1)
481         int len = strlen(path), retval;
482         char *submodule;
483         struct ref_cache *refs;
485         while (len && path[len-1] == '/')
486                 len--;
487         if (!len)
488                 return -1;
489         submodule = xstrndup(path, len);
490         refs = get_ref_cache(submodule);
491         free(submodule);
493         retval = resolve_gitlink_ref_recursive(refs, refname, sha1, 0);
494         return retval;
497 /*
498  * Try to read ref from the packed references.  On success, set sha1
499  * and return 0; otherwise, return -1.
500  */
501 static int get_packed_ref(const char *refname, unsigned char *sha1)
503         struct ref_array *packed = get_packed_refs(get_ref_cache(NULL));
504         struct ref_entry *entry = search_ref_array(packed, refname);
505         if (entry) {
506                 hashcpy(sha1, entry->sha1);
507                 return 0;
508         }
509         return -1;
512 const char *resolve_ref(const char *refname, unsigned char *sha1, int reading, int *flag)
514         int depth = MAXDEPTH;
515         ssize_t len;
516         char buffer[256];
517         static char refname_buffer[256];
519         if (flag)
520                 *flag = 0;
522         if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL))
523                 return NULL;
525         for (;;) {
526                 char path[PATH_MAX];
527                 struct stat st;
528                 char *buf;
529                 int fd;
531                 if (--depth < 0)
532                         return NULL;
534                 git_snpath(path, sizeof(path), "%s", refname);
536                 if (lstat(path, &st) < 0) {
537                         if (errno != ENOENT)
538                                 return NULL;
539                         /*
540                          * The loose reference file does not exist;
541                          * check for a packed reference.
542                          */
543                         if (!get_packed_ref(refname, sha1)) {
544                                 if (flag)
545                                         *flag |= REF_ISPACKED;
546                                 return refname;
547                         }
548                         /* The reference is not a packed reference, either. */
549                         if (reading) {
550                                 return NULL;
551                         } else {
552                                 hashclr(sha1);
553                                 return refname;
554                         }
555                 }
557                 /* Follow "normalized" - ie "refs/.." symlinks by hand */
558                 if (S_ISLNK(st.st_mode)) {
559                         len = readlink(path, buffer, sizeof(buffer)-1);
560                         if (len < 0)
561                                 return NULL;
562                         buffer[len] = 0;
563                         if (!prefixcmp(buffer, "refs/") &&
564                                         !check_refname_format(buffer, 0)) {
565                                 strcpy(refname_buffer, buffer);
566                                 refname = refname_buffer;
567                                 if (flag)
568                                         *flag |= REF_ISSYMREF;
569                                 continue;
570                         }
571                 }
573                 /* Is it a directory? */
574                 if (S_ISDIR(st.st_mode)) {
575                         errno = EISDIR;
576                         return NULL;
577                 }
579                 /*
580                  * Anything else, just open it and try to use it as
581                  * a ref
582                  */
583                 fd = open(path, O_RDONLY);
584                 if (fd < 0)
585                         return NULL;
586                 len = read_in_full(fd, buffer, sizeof(buffer)-1);
587                 close(fd);
588                 if (len < 0)
589                         return NULL;
590                 while (len && isspace(buffer[len-1]))
591                         len--;
592                 buffer[len] = '\0';
594                 /*
595                  * Is it a symbolic ref?
596                  */
597                 if (prefixcmp(buffer, "ref:"))
598                         break;
599                 if (flag)
600                         *flag |= REF_ISSYMREF;
601                 buf = buffer + 4;
602                 while (isspace(*buf))
603                         buf++;
604                 if (check_refname_format(buf, REFNAME_ALLOW_ONELEVEL)) {
605                         if (flag)
606                                 *flag |= REF_ISBROKEN;
607                         return NULL;
608                 }
609                 refname = strcpy(refname_buffer, buf);
610         }
611         /* Please note that FETCH_HEAD has a second line containing other data. */
612         if (get_sha1_hex(buffer, sha1) || (buffer[40] != '\0' && !isspace(buffer[40]))) {
613                 if (flag)
614                         *flag |= REF_ISBROKEN;
615                 return NULL;
616         }
617         return refname;
620 /* The argument to filter_refs */
621 struct ref_filter {
622         const char *pattern;
623         each_ref_fn *fn;
624         void *cb_data;
625 };
627 int read_ref_full(const char *refname, unsigned char *sha1, int reading, int *flags)
629         if (resolve_ref(refname, sha1, reading, flags))
630                 return 0;
631         return -1;
634 int read_ref(const char *refname, unsigned char *sha1)
636         return read_ref_full(refname, sha1, 1, NULL);
639 #define DO_FOR_EACH_INCLUDE_BROKEN 01
640 static int do_one_ref(const char *base, each_ref_fn fn, int trim,
641                       int flags, void *cb_data, struct ref_entry *entry)
643         if (prefixcmp(entry->name, base))
644                 return 0;
646         if (!(flags & DO_FOR_EACH_INCLUDE_BROKEN)) {
647                 if (entry->flag & REF_ISBROKEN)
648                         return 0; /* ignore broken refs e.g. dangling symref */
649                 if (!has_sha1_file(entry->sha1)) {
650                         error("%s does not point to a valid object!", entry->name);
651                         return 0;
652                 }
653         }
654         current_ref = entry;
655         return fn(entry->name + trim, entry->sha1, entry->flag, cb_data);
658 static int filter_refs(const char *refname, const unsigned char *sha1, int flags,
659                        void *data)
661         struct ref_filter *filter = (struct ref_filter *)data;
662         if (fnmatch(filter->pattern, refname, 0))
663                 return 0;
664         return filter->fn(refname, sha1, flags, filter->cb_data);
667 int peel_ref(const char *refname, unsigned char *sha1)
669         int flag;
670         unsigned char base[20];
671         struct object *o;
673         if (current_ref && (current_ref->name == refname
674                 || !strcmp(current_ref->name, refname))) {
675                 if (current_ref->flag & REF_KNOWS_PEELED) {
676                         hashcpy(sha1, current_ref->peeled);
677                         return 0;
678                 }
679                 hashcpy(base, current_ref->sha1);
680                 goto fallback;
681         }
683         if (read_ref_full(refname, base, 1, &flag))
684                 return -1;
686         if ((flag & REF_ISPACKED)) {
687                 struct ref_array *array = get_packed_refs(get_ref_cache(NULL));
688                 struct ref_entry *r = search_ref_array(array, refname);
690                 if (r != NULL && r->flag & REF_KNOWS_PEELED) {
691                         hashcpy(sha1, r->peeled);
692                         return 0;
693                 }
694         }
696 fallback:
697         o = parse_object(base);
698         if (o && o->type == OBJ_TAG) {
699                 o = deref_tag(o, refname, 0);
700                 if (o) {
701                         hashcpy(sha1, o->sha1);
702                         return 0;
703                 }
704         }
705         return -1;
708 static int do_for_each_ref(const char *submodule, const char *base, each_ref_fn fn,
709                            int trim, int flags, void *cb_data)
711         int retval = 0, i, p = 0, l = 0;
712         struct ref_cache *refs = get_ref_cache(submodule);
713         struct ref_array *packed = get_packed_refs(refs);
714         struct ref_array *loose = get_loose_refs(refs);
716         struct ref_array *extra = &extra_refs;
718         for (i = 0; i < extra->nr; i++)
719                 retval = do_one_ref(base, fn, trim, flags, cb_data, extra->refs[i]);
721         while (p < packed->nr && l < loose->nr) {
722                 struct ref_entry *entry;
723                 int cmp = strcmp(packed->refs[p]->name, loose->refs[l]->name);
724                 if (!cmp) {
725                         p++;
726                         continue;
727                 }
728                 if (cmp > 0) {
729                         entry = loose->refs[l++];
730                 } else {
731                         entry = packed->refs[p++];
732                 }
733                 retval = do_one_ref(base, fn, trim, flags, cb_data, entry);
734                 if (retval)
735                         goto end_each;
736         }
738         if (l < loose->nr) {
739                 p = l;
740                 packed = loose;
741         }
743         for (; p < packed->nr; p++) {
744                 retval = do_one_ref(base, fn, trim, flags, cb_data, packed->refs[p]);
745                 if (retval)
746                         goto end_each;
747         }
749 end_each:
750         current_ref = NULL;
751         return retval;
755 static int do_head_ref(const char *submodule, each_ref_fn fn, void *cb_data)
757         unsigned char sha1[20];
758         int flag;
760         if (submodule) {
761                 if (resolve_gitlink_ref(submodule, "HEAD", sha1) == 0)
762                         return fn("HEAD", sha1, 0, cb_data);
764                 return 0;
765         }
767         if (!read_ref_full("HEAD", sha1, 1, &flag))
768                 return fn("HEAD", sha1, flag, cb_data);
770         return 0;
773 int head_ref(each_ref_fn fn, void *cb_data)
775         return do_head_ref(NULL, fn, cb_data);
778 int head_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
780         return do_head_ref(submodule, fn, cb_data);
783 int for_each_ref(each_ref_fn fn, void *cb_data)
785         return do_for_each_ref(NULL, "", fn, 0, 0, cb_data);
788 int for_each_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
790         return do_for_each_ref(submodule, "", fn, 0, 0, cb_data);
793 int for_each_ref_in(const char *prefix, each_ref_fn fn, void *cb_data)
795         return do_for_each_ref(NULL, prefix, fn, strlen(prefix), 0, cb_data);
798 int for_each_ref_in_submodule(const char *submodule, const char *prefix,
799                 each_ref_fn fn, void *cb_data)
801         return do_for_each_ref(submodule, prefix, fn, strlen(prefix), 0, cb_data);
804 int for_each_tag_ref(each_ref_fn fn, void *cb_data)
806         return for_each_ref_in("refs/tags/", fn, cb_data);
809 int for_each_tag_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
811         return for_each_ref_in_submodule(submodule, "refs/tags/", fn, cb_data);
814 int for_each_branch_ref(each_ref_fn fn, void *cb_data)
816         return for_each_ref_in("refs/heads/", fn, cb_data);
819 int for_each_branch_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
821         return for_each_ref_in_submodule(submodule, "refs/heads/", fn, cb_data);
824 int for_each_remote_ref(each_ref_fn fn, void *cb_data)
826         return for_each_ref_in("refs/remotes/", fn, cb_data);
829 int for_each_remote_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
831         return for_each_ref_in_submodule(submodule, "refs/remotes/", fn, cb_data);
834 int for_each_replace_ref(each_ref_fn fn, void *cb_data)
836         return do_for_each_ref(NULL, "refs/replace/", fn, 13, 0, cb_data);
839 int head_ref_namespaced(each_ref_fn fn, void *cb_data)
841         struct strbuf buf = STRBUF_INIT;
842         int ret = 0;
843         unsigned char sha1[20];
844         int flag;
846         strbuf_addf(&buf, "%sHEAD", get_git_namespace());
847         if (!read_ref_full(buf.buf, sha1, 1, &flag))
848                 ret = fn(buf.buf, sha1, flag, cb_data);
849         strbuf_release(&buf);
851         return ret;
854 int for_each_namespaced_ref(each_ref_fn fn, void *cb_data)
856         struct strbuf buf = STRBUF_INIT;
857         int ret;
858         strbuf_addf(&buf, "%srefs/", get_git_namespace());
859         ret = do_for_each_ref(NULL, buf.buf, fn, 0, 0, cb_data);
860         strbuf_release(&buf);
861         return ret;
864 int for_each_glob_ref_in(each_ref_fn fn, const char *pattern,
865         const char *prefix, void *cb_data)
867         struct strbuf real_pattern = STRBUF_INIT;
868         struct ref_filter filter;
869         int ret;
871         if (!prefix && prefixcmp(pattern, "refs/"))
872                 strbuf_addstr(&real_pattern, "refs/");
873         else if (prefix)
874                 strbuf_addstr(&real_pattern, prefix);
875         strbuf_addstr(&real_pattern, pattern);
877         if (!has_glob_specials(pattern)) {
878                 /* Append implied '/' '*' if not present. */
879                 if (real_pattern.buf[real_pattern.len - 1] != '/')
880                         strbuf_addch(&real_pattern, '/');
881                 /* No need to check for '*', there is none. */
882                 strbuf_addch(&real_pattern, '*');
883         }
885         filter.pattern = real_pattern.buf;
886         filter.fn = fn;
887         filter.cb_data = cb_data;
888         ret = for_each_ref(filter_refs, &filter);
890         strbuf_release(&real_pattern);
891         return ret;
894 int for_each_glob_ref(each_ref_fn fn, const char *pattern, void *cb_data)
896         return for_each_glob_ref_in(fn, pattern, NULL, cb_data);
899 int for_each_rawref(each_ref_fn fn, void *cb_data)
901         return do_for_each_ref(NULL, "", fn, 0,
902                                DO_FOR_EACH_INCLUDE_BROKEN, cb_data);
905 /*
906  * Make sure "ref" is something reasonable to have under ".git/refs/";
907  * We do not like it if:
908  *
909  * - any path component of it begins with ".", or
910  * - it has double dots "..", or
911  * - it has ASCII control character, "~", "^", ":" or SP, anywhere, or
912  * - it ends with a "/".
913  * - it ends with ".lock"
914  * - it contains a "\" (backslash)
915  */
917 /* Return true iff ch is not allowed in reference names. */
918 static inline int bad_ref_char(int ch)
920         if (((unsigned) ch) <= ' ' || ch == 0x7f ||
921             ch == '~' || ch == '^' || ch == ':' || ch == '\\')
922                 return 1;
923         /* 2.13 Pattern Matching Notation */
924         if (ch == '*' || ch == '?' || ch == '[') /* Unsupported */
925                 return 1;
926         return 0;
929 /*
930  * Try to read one refname component from the front of refname.  Return
931  * the length of the component found, or -1 if the component is not
932  * legal.
933  */
934 static int check_refname_component(const char *refname, int flags)
936         const char *cp;
937         char last = '\0';
939         for (cp = refname; ; cp++) {
940                 char ch = *cp;
941                 if (ch == '\0' || ch == '/')
942                         break;
943                 if (bad_ref_char(ch))
944                         return -1; /* Illegal character in refname. */
945                 if (last == '.' && ch == '.')
946                         return -1; /* Refname contains "..". */
947                 if (last == '@' && ch == '{')
948                         return -1; /* Refname contains "@{". */
949                 last = ch;
950         }
951         if (cp == refname)
952                 return -1; /* Component has zero length. */
953         if (refname[0] == '.') {
954                 if (!(flags & REFNAME_DOT_COMPONENT))
955                         return -1; /* Component starts with '.'. */
956                 /*
957                  * Even if leading dots are allowed, don't allow "."
958                  * as a component (".." is prevented by a rule above).
959                  */
960                 if (refname[1] == '\0')
961                         return -1; /* Component equals ".". */
962         }
963         if (cp - refname >= 5 && !memcmp(cp - 5, ".lock", 5))
964                 return -1; /* Refname ends with ".lock". */
965         return cp - refname;
968 int check_refname_format(const char *refname, int flags)
970         int component_len, component_count = 0;
972         while (1) {
973                 /* We are at the start of a path component. */
974                 component_len = check_refname_component(refname, flags);
975                 if (component_len < 0) {
976                         if ((flags & REFNAME_REFSPEC_PATTERN) &&
977                                         refname[0] == '*' &&
978                                         (refname[1] == '\0' || refname[1] == '/')) {
979                                 /* Accept one wildcard as a full refname component. */
980                                 flags &= ~REFNAME_REFSPEC_PATTERN;
981                                 component_len = 1;
982                         } else {
983                                 return -1;
984                         }
985                 }
986                 component_count++;
987                 if (refname[component_len] == '\0')
988                         break;
989                 /* Skip to next component. */
990                 refname += component_len + 1;
991         }
993         if (refname[component_len - 1] == '.')
994                 return -1; /* Refname ends with '.'. */
995         if (!(flags & REFNAME_ALLOW_ONELEVEL) && component_count < 2)
996                 return -1; /* Refname has only one component. */
997         return 0;
1000 const char *prettify_refname(const char *name)
1002         return name + (
1003                 !prefixcmp(name, "refs/heads/") ? 11 :
1004                 !prefixcmp(name, "refs/tags/") ? 10 :
1005                 !prefixcmp(name, "refs/remotes/") ? 13 :
1006                 0);
1009 const char *ref_rev_parse_rules[] = {
1010         "%.*s",
1011         "refs/%.*s",
1012         "refs/tags/%.*s",
1013         "refs/heads/%.*s",
1014         "refs/remotes/%.*s",
1015         "refs/remotes/%.*s/HEAD",
1016         NULL
1017 };
1019 int refname_match(const char *abbrev_name, const char *full_name, const char **rules)
1021         const char **p;
1022         const int abbrev_name_len = strlen(abbrev_name);
1024         for (p = rules; *p; p++) {
1025                 if (!strcmp(full_name, mkpath(*p, abbrev_name_len, abbrev_name))) {
1026                         return 1;
1027                 }
1028         }
1030         return 0;
1033 static struct ref_lock *verify_lock(struct ref_lock *lock,
1034         const unsigned char *old_sha1, int mustexist)
1036         if (read_ref_full(lock->ref_name, lock->old_sha1, mustexist, NULL)) {
1037                 error("Can't verify ref %s", lock->ref_name);
1038                 unlock_ref(lock);
1039                 return NULL;
1040         }
1041         if (hashcmp(lock->old_sha1, old_sha1)) {
1042                 error("Ref %s is at %s but expected %s", lock->ref_name,
1043                         sha1_to_hex(lock->old_sha1), sha1_to_hex(old_sha1));
1044                 unlock_ref(lock);
1045                 return NULL;
1046         }
1047         return lock;
1050 static int remove_empty_directories(const char *file)
1052         /* we want to create a file but there is a directory there;
1053          * if that is an empty directory (or a directory that contains
1054          * only empty directories), remove them.
1055          */
1056         struct strbuf path;
1057         int result;
1059         strbuf_init(&path, 20);
1060         strbuf_addstr(&path, file);
1062         result = remove_dir_recursively(&path, REMOVE_DIR_EMPTY_ONLY);
1064         strbuf_release(&path);
1066         return result;
1069 /*
1070  * Return true iff a reference named refname could be created without
1071  * conflicting with the name of an existing reference.  If oldrefname
1072  * is non-NULL, ignore potential conflicts with oldrefname (e.g.,
1073  * because oldrefname is scheduled for deletion in the same
1074  * operation).
1075  */
1076 static int is_refname_available(const char *refname, const char *oldrefname,
1077                                 struct ref_array *array)
1079         int i, namlen = strlen(refname); /* e.g. 'foo/bar' */
1080         for (i = 0; i < array->nr; i++ ) {
1081                 struct ref_entry *entry = array->refs[i];
1082                 /* entry->name could be 'foo' or 'foo/bar/baz' */
1083                 if (!oldrefname || strcmp(oldrefname, entry->name)) {
1084                         int len = strlen(entry->name);
1085                         int cmplen = (namlen < len) ? namlen : len;
1086                         const char *lead = (namlen < len) ? entry->name : refname;
1087                         if (!strncmp(refname, entry->name, cmplen) &&
1088                             lead[cmplen] == '/') {
1089                                 error("'%s' exists; cannot create '%s'",
1090                                       entry->name, refname);
1091                                 return 0;
1092                         }
1093                 }
1094         }
1095         return 1;
1098 /*
1099  * *string and *len will only be substituted, and *string returned (for
1100  * later free()ing) if the string passed in is a magic short-hand form
1101  * to name a branch.
1102  */
1103 static char *substitute_branch_name(const char **string, int *len)
1105         struct strbuf buf = STRBUF_INIT;
1106         int ret = interpret_branch_name(*string, &buf);
1108         if (ret == *len) {
1109                 size_t size;
1110                 *string = strbuf_detach(&buf, &size);
1111                 *len = size;
1112                 return (char *)*string;
1113         }
1115         return NULL;
1118 int dwim_ref(const char *str, int len, unsigned char *sha1, char **ref)
1120         char *last_branch = substitute_branch_name(&str, &len);
1121         const char **p, *r;
1122         int refs_found = 0;
1124         *ref = NULL;
1125         for (p = ref_rev_parse_rules; *p; p++) {
1126                 char fullref[PATH_MAX];
1127                 unsigned char sha1_from_ref[20];
1128                 unsigned char *this_result;
1129                 int flag;
1131                 this_result = refs_found ? sha1_from_ref : sha1;
1132                 mksnpath(fullref, sizeof(fullref), *p, len, str);
1133                 r = resolve_ref(fullref, this_result, 1, &flag);
1134                 if (r) {
1135                         if (!refs_found++)
1136                                 *ref = xstrdup(r);
1137                         if (!warn_ambiguous_refs)
1138                                 break;
1139                 } else if ((flag & REF_ISSYMREF) && strcmp(fullref, "HEAD")) {
1140                         warning("ignoring dangling symref %s.", fullref);
1141                 } else if ((flag & REF_ISBROKEN) && strchr(fullref, '/')) {
1142                         warning("ignoring broken ref %s.", fullref);
1143                 }
1144         }
1145         free(last_branch);
1146         return refs_found;
1149 int dwim_log(const char *str, int len, unsigned char *sha1, char **log)
1151         char *last_branch = substitute_branch_name(&str, &len);
1152         const char **p;
1153         int logs_found = 0;
1155         *log = NULL;
1156         for (p = ref_rev_parse_rules; *p; p++) {
1157                 struct stat st;
1158                 unsigned char hash[20];
1159                 char path[PATH_MAX];
1160                 const char *ref, *it;
1162                 mksnpath(path, sizeof(path), *p, len, str);
1163                 ref = resolve_ref(path, hash, 1, NULL);
1164                 if (!ref)
1165                         continue;
1166                 if (!stat(git_path("logs/%s", path), &st) &&
1167                     S_ISREG(st.st_mode))
1168                         it = path;
1169                 else if (strcmp(ref, path) &&
1170                          !stat(git_path("logs/%s", ref), &st) &&
1171                          S_ISREG(st.st_mode))
1172                         it = ref;
1173                 else
1174                         continue;
1175                 if (!logs_found++) {
1176                         *log = xstrdup(it);
1177                         hashcpy(sha1, hash);
1178                 }
1179                 if (!warn_ambiguous_refs)
1180                         break;
1181         }
1182         free(last_branch);
1183         return logs_found;
1186 static struct ref_lock *lock_ref_sha1_basic(const char *refname,
1187                                             const unsigned char *old_sha1,
1188                                             int flags, int *type_p)
1190         char *ref_file;
1191         const char *orig_refname = refname;
1192         struct ref_lock *lock;
1193         int last_errno = 0;
1194         int type, lflags;
1195         int mustexist = (old_sha1 && !is_null_sha1(old_sha1));
1196         int missing = 0;
1198         lock = xcalloc(1, sizeof(struct ref_lock));
1199         lock->lock_fd = -1;
1201         refname = resolve_ref(refname, lock->old_sha1, mustexist, &type);
1202         if (!refname && errno == EISDIR) {
1203                 /* we are trying to lock foo but we used to
1204                  * have foo/bar which now does not exist;
1205                  * it is normal for the empty directory 'foo'
1206                  * to remain.
1207                  */
1208                 ref_file = git_path("%s", orig_refname);
1209                 if (remove_empty_directories(ref_file)) {
1210                         last_errno = errno;
1211                         error("there are still refs under '%s'", orig_refname);
1212                         goto error_return;
1213                 }
1214                 refname = resolve_ref(orig_refname, lock->old_sha1, mustexist, &type);
1215         }
1216         if (type_p)
1217             *type_p = type;
1218         if (!refname) {
1219                 last_errno = errno;
1220                 error("unable to resolve reference %s: %s",
1221                         orig_refname, strerror(errno));
1222                 goto error_return;
1223         }
1224         missing = is_null_sha1(lock->old_sha1);
1225         /* When the ref did not exist and we are creating it,
1226          * make sure there is no existing ref that is packed
1227          * whose name begins with our refname, nor a ref whose
1228          * name is a proper prefix of our refname.
1229          */
1230         if (missing &&
1231              !is_refname_available(refname, NULL, get_packed_refs(get_ref_cache(NULL)))) {
1232                 last_errno = ENOTDIR;
1233                 goto error_return;
1234         }
1236         lock->lk = xcalloc(1, sizeof(struct lock_file));
1238         lflags = LOCK_DIE_ON_ERROR;
1239         if (flags & REF_NODEREF) {
1240                 refname = orig_refname;
1241                 lflags |= LOCK_NODEREF;
1242         }
1243         lock->ref_name = xstrdup(refname);
1244         lock->orig_ref_name = xstrdup(orig_refname);
1245         ref_file = git_path("%s", refname);
1246         if (missing)
1247                 lock->force_write = 1;
1248         if ((flags & REF_NODEREF) && (type & REF_ISSYMREF))
1249                 lock->force_write = 1;
1251         if (safe_create_leading_directories(ref_file)) {
1252                 last_errno = errno;
1253                 error("unable to create directory for %s", ref_file);
1254                 goto error_return;
1255         }
1257         lock->lock_fd = hold_lock_file_for_update(lock->lk, ref_file, lflags);
1258         return old_sha1 ? verify_lock(lock, old_sha1, mustexist) : lock;
1260  error_return:
1261         unlock_ref(lock);
1262         errno = last_errno;
1263         return NULL;
1266 struct ref_lock *lock_ref_sha1(const char *refname, const unsigned char *old_sha1)
1268         char refpath[PATH_MAX];
1269         if (check_refname_format(refname, 0))
1270                 return NULL;
1271         strcpy(refpath, mkpath("refs/%s", refname));
1272         return lock_ref_sha1_basic(refpath, old_sha1, 0, NULL);
1275 struct ref_lock *lock_any_ref_for_update(const char *refname,
1276                                          const unsigned char *old_sha1, int flags)
1278         if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL))
1279                 return NULL;
1280         return lock_ref_sha1_basic(refname, old_sha1, flags, NULL);
1283 static struct lock_file packlock;
1285 static int repack_without_ref(const char *refname)
1287         struct ref_array *packed;
1288         int fd, i;
1290         packed = get_packed_refs(get_ref_cache(NULL));
1291         if (search_ref_array(packed, refname) == NULL)
1292                 return 0;
1293         fd = hold_lock_file_for_update(&packlock, git_path("packed-refs"), 0);
1294         if (fd < 0) {
1295                 unable_to_lock_error(git_path("packed-refs"), errno);
1296                 return error("cannot delete '%s' from packed refs", refname);
1297         }
1299         for (i = 0; i < packed->nr; i++) {
1300                 char line[PATH_MAX + 100];
1301                 int len;
1302                 struct ref_entry *ref = packed->refs[i];
1304                 if (!strcmp(refname, ref->name))
1305                         continue;
1306                 len = snprintf(line, sizeof(line), "%s %s\n",
1307                                sha1_to_hex(ref->sha1), ref->name);
1308                 /* this should not happen but just being defensive */
1309                 if (len > sizeof(line))
1310                         die("too long a refname '%s'", ref->name);
1311                 write_or_die(fd, line, len);
1312         }
1313         return commit_lock_file(&packlock);
1316 int delete_ref(const char *refname, const unsigned char *sha1, int delopt)
1318         struct ref_lock *lock;
1319         int err, i = 0, ret = 0, flag = 0;
1321         lock = lock_ref_sha1_basic(refname, sha1, 0, &flag);
1322         if (!lock)
1323                 return 1;
1324         if (!(flag & REF_ISPACKED) || flag & REF_ISSYMREF) {
1325                 /* loose */
1326                 const char *path;
1328                 if (!(delopt & REF_NODEREF)) {
1329                         i = strlen(lock->lk->filename) - 5; /* .lock */
1330                         lock->lk->filename[i] = 0;
1331                         path = lock->lk->filename;
1332                 } else {
1333                         path = git_path("%s", refname);
1334                 }
1335                 err = unlink_or_warn(path);
1336                 if (err && errno != ENOENT)
1337                         ret = 1;
1339                 if (!(delopt & REF_NODEREF))
1340                         lock->lk->filename[i] = '.';
1341         }
1342         /* removing the loose one could have resurrected an earlier
1343          * packed one.  Also, if it was not loose we need to repack
1344          * without it.
1345          */
1346         ret |= repack_without_ref(refname);
1348         unlink_or_warn(git_path("logs/%s", lock->ref_name));
1349         invalidate_ref_cache(NULL);
1350         unlock_ref(lock);
1351         return ret;
1354 /*
1355  * People using contrib's git-new-workdir have .git/logs/refs ->
1356  * /some/other/path/.git/logs/refs, and that may live on another device.
1357  *
1358  * IOW, to avoid cross device rename errors, the temporary renamed log must
1359  * live into logs/refs.
1360  */
1361 #define TMP_RENAMED_LOG  "logs/refs/.tmp-renamed-log"
1363 int rename_ref(const char *oldrefname, const char *newrefname, const char *logmsg)
1365         unsigned char sha1[20], orig_sha1[20];
1366         int flag = 0, logmoved = 0;
1367         struct ref_lock *lock;
1368         struct stat loginfo;
1369         int log = !lstat(git_path("logs/%s", oldrefname), &loginfo);
1370         const char *symref = NULL;
1371         struct ref_cache *refs = get_ref_cache(NULL);
1373         if (log && S_ISLNK(loginfo.st_mode))
1374                 return error("reflog for %s is a symlink", oldrefname);
1376         symref = resolve_ref(oldrefname, orig_sha1, 1, &flag);
1377         if (flag & REF_ISSYMREF)
1378                 return error("refname %s is a symbolic ref, renaming it is not supported",
1379                         oldrefname);
1380         if (!symref)
1381                 return error("refname %s not found", oldrefname);
1383         if (!is_refname_available(newrefname, oldrefname, get_packed_refs(refs)))
1384                 return 1;
1386         if (!is_refname_available(newrefname, oldrefname, get_loose_refs(refs)))
1387                 return 1;
1389         if (log && rename(git_path("logs/%s", oldrefname), git_path(TMP_RENAMED_LOG)))
1390                 return error("unable to move logfile logs/%s to "TMP_RENAMED_LOG": %s",
1391                         oldrefname, strerror(errno));
1393         if (delete_ref(oldrefname, orig_sha1, REF_NODEREF)) {
1394                 error("unable to delete old %s", oldrefname);
1395                 goto rollback;
1396         }
1398         if (!read_ref_full(newrefname, sha1, 1, &flag) &&
1399             delete_ref(newrefname, sha1, REF_NODEREF)) {
1400                 if (errno==EISDIR) {
1401                         if (remove_empty_directories(git_path("%s", newrefname))) {
1402                                 error("Directory not empty: %s", newrefname);
1403                                 goto rollback;
1404                         }
1405                 } else {
1406                         error("unable to delete existing %s", newrefname);
1407                         goto rollback;
1408                 }
1409         }
1411         if (log && safe_create_leading_directories(git_path("logs/%s", newrefname))) {
1412                 error("unable to create directory for %s", newrefname);
1413                 goto rollback;
1414         }
1416  retry:
1417         if (log && rename(git_path(TMP_RENAMED_LOG), git_path("logs/%s", newrefname))) {
1418                 if (errno==EISDIR || errno==ENOTDIR) {
1419                         /*
1420                          * rename(a, b) when b is an existing
1421                          * directory ought to result in ISDIR, but
1422                          * Solaris 5.8 gives ENOTDIR.  Sheesh.
1423                          */
1424                         if (remove_empty_directories(git_path("logs/%s", newrefname))) {
1425                                 error("Directory not empty: logs/%s", newrefname);
1426                                 goto rollback;
1427                         }
1428                         goto retry;
1429                 } else {
1430                         error("unable to move logfile "TMP_RENAMED_LOG" to logs/%s: %s",
1431                                 newrefname, strerror(errno));
1432                         goto rollback;
1433                 }
1434         }
1435         logmoved = log;
1437         lock = lock_ref_sha1_basic(newrefname, NULL, 0, NULL);
1438         if (!lock) {
1439                 error("unable to lock %s for update", newrefname);
1440                 goto rollback;
1441         }
1442         lock->force_write = 1;
1443         hashcpy(lock->old_sha1, orig_sha1);
1444         if (write_ref_sha1(lock, orig_sha1, logmsg)) {
1445                 error("unable to write current sha1 into %s", newrefname);
1446                 goto rollback;
1447         }
1449         return 0;
1451  rollback:
1452         lock = lock_ref_sha1_basic(oldrefname, NULL, 0, NULL);
1453         if (!lock) {
1454                 error("unable to lock %s for rollback", oldrefname);
1455                 goto rollbacklog;
1456         }
1458         lock->force_write = 1;
1459         flag = log_all_ref_updates;
1460         log_all_ref_updates = 0;
1461         if (write_ref_sha1(lock, orig_sha1, NULL))
1462                 error("unable to write current sha1 into %s", oldrefname);
1463         log_all_ref_updates = flag;
1465  rollbacklog:
1466         if (logmoved && rename(git_path("logs/%s", newrefname), git_path("logs/%s", oldrefname)))
1467                 error("unable to restore logfile %s from %s: %s",
1468                         oldrefname, newrefname, strerror(errno));
1469         if (!logmoved && log &&
1470             rename(git_path(TMP_RENAMED_LOG), git_path("logs/%s", oldrefname)))
1471                 error("unable to restore logfile %s from "TMP_RENAMED_LOG": %s",
1472                         oldrefname, strerror(errno));
1474         return 1;
1477 int close_ref(struct ref_lock *lock)
1479         if (close_lock_file(lock->lk))
1480                 return -1;
1481         lock->lock_fd = -1;
1482         return 0;
1485 int commit_ref(struct ref_lock *lock)
1487         if (commit_lock_file(lock->lk))
1488                 return -1;
1489         lock->lock_fd = -1;
1490         return 0;
1493 void unlock_ref(struct ref_lock *lock)
1495         /* Do not free lock->lk -- atexit() still looks at them */
1496         if (lock->lk)
1497                 rollback_lock_file(lock->lk);
1498         free(lock->ref_name);
1499         free(lock->orig_ref_name);
1500         free(lock);
1503 /*
1504  * copy the reflog message msg to buf, which has been allocated sufficiently
1505  * large, while cleaning up the whitespaces.  Especially, convert LF to space,
1506  * because reflog file is one line per entry.
1507  */
1508 static int copy_msg(char *buf, const char *msg)
1510         char *cp = buf;
1511         char c;
1512         int wasspace = 1;
1514         *cp++ = '\t';
1515         while ((c = *msg++)) {
1516                 if (wasspace && isspace(c))
1517                         continue;
1518                 wasspace = isspace(c);
1519                 if (wasspace)
1520                         c = ' ';
1521                 *cp++ = c;
1522         }
1523         while (buf < cp && isspace(cp[-1]))
1524                 cp--;
1525         *cp++ = '\n';
1526         return cp - buf;
1529 int log_ref_setup(const char *refname, char *logfile, int bufsize)
1531         int logfd, oflags = O_APPEND | O_WRONLY;
1533         git_snpath(logfile, bufsize, "logs/%s", refname);
1534         if (log_all_ref_updates &&
1535             (!prefixcmp(refname, "refs/heads/") ||
1536              !prefixcmp(refname, "refs/remotes/") ||
1537              !prefixcmp(refname, "refs/notes/") ||
1538              !strcmp(refname, "HEAD"))) {
1539                 if (safe_create_leading_directories(logfile) < 0)
1540                         return error("unable to create directory for %s",
1541                                      logfile);
1542                 oflags |= O_CREAT;
1543         }
1545         logfd = open(logfile, oflags, 0666);
1546         if (logfd < 0) {
1547                 if (!(oflags & O_CREAT) && errno == ENOENT)
1548                         return 0;
1550                 if ((oflags & O_CREAT) && errno == EISDIR) {
1551                         if (remove_empty_directories(logfile)) {
1552                                 return error("There are still logs under '%s'",
1553                                              logfile);
1554                         }
1555                         logfd = open(logfile, oflags, 0666);
1556                 }
1558                 if (logfd < 0)
1559                         return error("Unable to append to %s: %s",
1560                                      logfile, strerror(errno));
1561         }
1563         adjust_shared_perm(logfile);
1564         close(logfd);
1565         return 0;
1568 static int log_ref_write(const char *refname, const unsigned char *old_sha1,
1569                          const unsigned char *new_sha1, const char *msg)
1571         int logfd, result, written, oflags = O_APPEND | O_WRONLY;
1572         unsigned maxlen, len;
1573         int msglen;
1574         char log_file[PATH_MAX];
1575         char *logrec;
1576         const char *committer;
1578         if (log_all_ref_updates < 0)
1579                 log_all_ref_updates = !is_bare_repository();
1581         result = log_ref_setup(refname, log_file, sizeof(log_file));
1582         if (result)
1583                 return result;
1585         logfd = open(log_file, oflags);
1586         if (logfd < 0)
1587                 return 0;
1588         msglen = msg ? strlen(msg) : 0;
1589         committer = git_committer_info(0);
1590         maxlen = strlen(committer) + msglen + 100;
1591         logrec = xmalloc(maxlen);
1592         len = sprintf(logrec, "%s %s %s\n",
1593                       sha1_to_hex(old_sha1),
1594                       sha1_to_hex(new_sha1),
1595                       committer);
1596         if (msglen)
1597                 len += copy_msg(logrec + len - 1, msg) - 1;
1598         written = len <= maxlen ? write_in_full(logfd, logrec, len) : -1;
1599         free(logrec);
1600         if (close(logfd) != 0 || written != len)
1601                 return error("Unable to append to %s", log_file);
1602         return 0;
1605 static int is_branch(const char *refname)
1607         return !strcmp(refname, "HEAD") || !prefixcmp(refname, "refs/heads/");
1610 int write_ref_sha1(struct ref_lock *lock,
1611         const unsigned char *sha1, const char *logmsg)
1613         static char term = '\n';
1614         struct object *o;
1616         if (!lock)
1617                 return -1;
1618         if (!lock->force_write && !hashcmp(lock->old_sha1, sha1)) {
1619                 unlock_ref(lock);
1620                 return 0;
1621         }
1622         o = parse_object(sha1);
1623         if (!o) {
1624                 error("Trying to write ref %s with nonexistent object %s",
1625                         lock->ref_name, sha1_to_hex(sha1));
1626                 unlock_ref(lock);
1627                 return -1;
1628         }
1629         if (o->type != OBJ_COMMIT && is_branch(lock->ref_name)) {
1630                 error("Trying to write non-commit object %s to branch %s",
1631                         sha1_to_hex(sha1), lock->ref_name);
1632                 unlock_ref(lock);
1633                 return -1;
1634         }
1635         if (write_in_full(lock->lock_fd, sha1_to_hex(sha1), 40) != 40 ||
1636             write_in_full(lock->lock_fd, &term, 1) != 1
1637                 || close_ref(lock) < 0) {
1638                 error("Couldn't write %s", lock->lk->filename);
1639                 unlock_ref(lock);
1640                 return -1;
1641         }
1642         clear_loose_ref_cache(get_ref_cache(NULL));
1643         if (log_ref_write(lock->ref_name, lock->old_sha1, sha1, logmsg) < 0 ||
1644             (strcmp(lock->ref_name, lock->orig_ref_name) &&
1645              log_ref_write(lock->orig_ref_name, lock->old_sha1, sha1, logmsg) < 0)) {
1646                 unlock_ref(lock);
1647                 return -1;
1648         }
1649         if (strcmp(lock->orig_ref_name, "HEAD") != 0) {
1650                 /*
1651                  * Special hack: If a branch is updated directly and HEAD
1652                  * points to it (may happen on the remote side of a push
1653                  * for example) then logically the HEAD reflog should be
1654                  * updated too.
1655                  * A generic solution implies reverse symref information,
1656                  * but finding all symrefs pointing to the given branch
1657                  * would be rather costly for this rare event (the direct
1658                  * update of a branch) to be worth it.  So let's cheat and
1659                  * check with HEAD only which should cover 99% of all usage
1660                  * scenarios (even 100% of the default ones).
1661                  */
1662                 unsigned char head_sha1[20];
1663                 int head_flag;
1664                 const char *head_ref;
1665                 head_ref = resolve_ref("HEAD", head_sha1, 1, &head_flag);
1666                 if (head_ref && (head_flag & REF_ISSYMREF) &&
1667                     !strcmp(head_ref, lock->ref_name))
1668                         log_ref_write("HEAD", lock->old_sha1, sha1, logmsg);
1669         }
1670         if (commit_ref(lock)) {
1671                 error("Couldn't set %s", lock->ref_name);
1672                 unlock_ref(lock);
1673                 return -1;
1674         }
1675         unlock_ref(lock);
1676         return 0;
1679 int create_symref(const char *ref_target, const char *refs_heads_master,
1680                   const char *logmsg)
1682         const char *lockpath;
1683         char ref[1000];
1684         int fd, len, written;
1685         char *git_HEAD = git_pathdup("%s", ref_target);
1686         unsigned char old_sha1[20], new_sha1[20];
1688         if (logmsg && read_ref(ref_target, old_sha1))
1689                 hashclr(old_sha1);
1691         if (safe_create_leading_directories(git_HEAD) < 0)
1692                 return error("unable to create directory for %s", git_HEAD);
1694 #ifndef NO_SYMLINK_HEAD
1695         if (prefer_symlink_refs) {
1696                 unlink(git_HEAD);
1697                 if (!symlink(refs_heads_master, git_HEAD))
1698                         goto done;
1699                 fprintf(stderr, "no symlink - falling back to symbolic ref\n");
1700         }
1701 #endif
1703         len = snprintf(ref, sizeof(ref), "ref: %s\n", refs_heads_master);
1704         if (sizeof(ref) <= len) {
1705                 error("refname too long: %s", refs_heads_master);
1706                 goto error_free_return;
1707         }
1708         lockpath = mkpath("%s.lock", git_HEAD);
1709         fd = open(lockpath, O_CREAT | O_EXCL | O_WRONLY, 0666);
1710         if (fd < 0) {
1711                 error("Unable to open %s for writing", lockpath);
1712                 goto error_free_return;
1713         }
1714         written = write_in_full(fd, ref, len);
1715         if (close(fd) != 0 || written != len) {
1716                 error("Unable to write to %s", lockpath);
1717                 goto error_unlink_return;
1718         }
1719         if (rename(lockpath, git_HEAD) < 0) {
1720                 error("Unable to create %s", git_HEAD);
1721                 goto error_unlink_return;
1722         }
1723         if (adjust_shared_perm(git_HEAD)) {
1724                 error("Unable to fix permissions on %s", lockpath);
1725         error_unlink_return:
1726                 unlink_or_warn(lockpath);
1727         error_free_return:
1728                 free(git_HEAD);
1729                 return -1;
1730         }
1732 #ifndef NO_SYMLINK_HEAD
1733         done:
1734 #endif
1735         if (logmsg && !read_ref(refs_heads_master, new_sha1))
1736                 log_ref_write(ref_target, old_sha1, new_sha1, logmsg);
1738         free(git_HEAD);
1739         return 0;
1742 static char *ref_msg(const char *line, const char *endp)
1744         const char *ep;
1745         line += 82;
1746         ep = memchr(line, '\n', endp - line);
1747         if (!ep)
1748                 ep = endp;
1749         return xmemdupz(line, ep - line);
1752 int read_ref_at(const char *refname, unsigned long at_time, int cnt,
1753                 unsigned char *sha1, char **msg,
1754                 unsigned long *cutoff_time, int *cutoff_tz, int *cutoff_cnt)
1756         const char *logfile, *logdata, *logend, *rec, *lastgt, *lastrec;
1757         char *tz_c;
1758         int logfd, tz, reccnt = 0;
1759         struct stat st;
1760         unsigned long date;
1761         unsigned char logged_sha1[20];
1762         void *log_mapped;
1763         size_t mapsz;
1765         logfile = git_path("logs/%s", refname);
1766         logfd = open(logfile, O_RDONLY, 0);
1767         if (logfd < 0)
1768                 die_errno("Unable to read log '%s'", logfile);
1769         fstat(logfd, &st);
1770         if (!st.st_size)
1771                 die("Log %s is empty.", logfile);
1772         mapsz = xsize_t(st.st_size);
1773         log_mapped = xmmap(NULL, mapsz, PROT_READ, MAP_PRIVATE, logfd, 0);
1774         logdata = log_mapped;
1775         close(logfd);
1777         lastrec = NULL;
1778         rec = logend = logdata + st.st_size;
1779         while (logdata < rec) {
1780                 reccnt++;
1781                 if (logdata < rec && *(rec-1) == '\n')
1782                         rec--;
1783                 lastgt = NULL;
1784                 while (logdata < rec && *(rec-1) != '\n') {
1785                         rec--;
1786                         if (*rec == '>')
1787                                 lastgt = rec;
1788                 }
1789                 if (!lastgt)
1790                         die("Log %s is corrupt.", logfile);
1791                 date = strtoul(lastgt + 1, &tz_c, 10);
1792                 if (date <= at_time || cnt == 0) {
1793                         tz = strtoul(tz_c, NULL, 10);
1794                         if (msg)
1795                                 *msg = ref_msg(rec, logend);
1796                         if (cutoff_time)
1797                                 *cutoff_time = date;
1798                         if (cutoff_tz)
1799                                 *cutoff_tz = tz;
1800                         if (cutoff_cnt)
1801                                 *cutoff_cnt = reccnt - 1;
1802                         if (lastrec) {
1803                                 if (get_sha1_hex(lastrec, logged_sha1))
1804                                         die("Log %s is corrupt.", logfile);
1805                                 if (get_sha1_hex(rec + 41, sha1))
1806                                         die("Log %s is corrupt.", logfile);
1807                                 if (hashcmp(logged_sha1, sha1)) {
1808                                         warning("Log %s has gap after %s.",
1809                                                 logfile, show_date(date, tz, DATE_RFC2822));
1810                                 }
1811                         }
1812                         else if (date == at_time) {
1813                                 if (get_sha1_hex(rec + 41, sha1))
1814                                         die("Log %s is corrupt.", logfile);
1815                         }
1816                         else {
1817                                 if (get_sha1_hex(rec + 41, logged_sha1))
1818                                         die("Log %s is corrupt.", logfile);
1819                                 if (hashcmp(logged_sha1, sha1)) {
1820                                         warning("Log %s unexpectedly ended on %s.",
1821                                                 logfile, show_date(date, tz, DATE_RFC2822));
1822                                 }
1823                         }
1824                         munmap(log_mapped, mapsz);
1825                         return 0;
1826                 }
1827                 lastrec = rec;
1828                 if (cnt > 0)
1829                         cnt--;
1830         }
1832         rec = logdata;
1833         while (rec < logend && *rec != '>' && *rec != '\n')
1834                 rec++;
1835         if (rec == logend || *rec == '\n')
1836                 die("Log %s is corrupt.", logfile);
1837         date = strtoul(rec + 1, &tz_c, 10);
1838         tz = strtoul(tz_c, NULL, 10);
1839         if (get_sha1_hex(logdata, sha1))
1840                 die("Log %s is corrupt.", logfile);
1841         if (is_null_sha1(sha1)) {
1842                 if (get_sha1_hex(logdata + 41, sha1))
1843                         die("Log %s is corrupt.", logfile);
1844         }
1845         if (msg)
1846                 *msg = ref_msg(logdata, logend);
1847         munmap(log_mapped, mapsz);
1849         if (cutoff_time)
1850                 *cutoff_time = date;
1851         if (cutoff_tz)
1852                 *cutoff_tz = tz;
1853         if (cutoff_cnt)
1854                 *cutoff_cnt = reccnt;
1855         return 1;
1858 int for_each_recent_reflog_ent(const char *refname, each_reflog_ent_fn fn, long ofs, void *cb_data)
1860         const char *logfile;
1861         FILE *logfp;
1862         struct strbuf sb = STRBUF_INIT;
1863         int ret = 0;
1865         logfile = git_path("logs/%s", refname);
1866         logfp = fopen(logfile, "r");
1867         if (!logfp)
1868                 return -1;
1870         if (ofs) {
1871                 struct stat statbuf;
1872                 if (fstat(fileno(logfp), &statbuf) ||
1873                     statbuf.st_size < ofs ||
1874                     fseek(logfp, -ofs, SEEK_END) ||
1875                     strbuf_getwholeline(&sb, logfp, '\n')) {
1876                         fclose(logfp);
1877                         strbuf_release(&sb);
1878                         return -1;
1879                 }
1880         }
1882         while (!strbuf_getwholeline(&sb, logfp, '\n')) {
1883                 unsigned char osha1[20], nsha1[20];
1884                 char *email_end, *message;
1885                 unsigned long timestamp;
1886                 int tz;
1888                 /* old SP new SP name <email> SP time TAB msg LF */
1889                 if (sb.len < 83 || sb.buf[sb.len - 1] != '\n' ||
1890                     get_sha1_hex(sb.buf, osha1) || sb.buf[40] != ' ' ||
1891                     get_sha1_hex(sb.buf + 41, nsha1) || sb.buf[81] != ' ' ||
1892                     !(email_end = strchr(sb.buf + 82, '>')) ||
1893                     email_end[1] != ' ' ||
1894                     !(timestamp = strtoul(email_end + 2, &message, 10)) ||
1895                     !message || message[0] != ' ' ||
1896                     (message[1] != '+' && message[1] != '-') ||
1897                     !isdigit(message[2]) || !isdigit(message[3]) ||
1898                     !isdigit(message[4]) || !isdigit(message[5]))
1899                         continue; /* corrupt? */
1900                 email_end[1] = '\0';
1901                 tz = strtol(message + 1, NULL, 10);
1902                 if (message[6] != '\t')
1903                         message += 6;
1904                 else
1905                         message += 7;
1906                 ret = fn(osha1, nsha1, sb.buf + 82, timestamp, tz, message,
1907                          cb_data);
1908                 if (ret)
1909                         break;
1910         }
1911         fclose(logfp);
1912         strbuf_release(&sb);
1913         return ret;
1916 int for_each_reflog_ent(const char *refname, each_reflog_ent_fn fn, void *cb_data)
1918         return for_each_recent_reflog_ent(refname, fn, 0, cb_data);
1921 static int do_for_each_reflog(const char *base, each_ref_fn fn, void *cb_data)
1923         DIR *dir = opendir(git_path("logs/%s", base));
1924         int retval = 0;
1926         if (dir) {
1927                 struct dirent *de;
1928                 int baselen = strlen(base);
1929                 char *log = xmalloc(baselen + 257);
1931                 memcpy(log, base, baselen);
1932                 if (baselen && base[baselen-1] != '/')
1933                         log[baselen++] = '/';
1935                 while ((de = readdir(dir)) != NULL) {
1936                         struct stat st;
1937                         int namelen;
1939                         if (de->d_name[0] == '.')
1940                                 continue;
1941                         namelen = strlen(de->d_name);
1942                         if (namelen > 255)
1943                                 continue;
1944                         if (has_extension(de->d_name, ".lock"))
1945                                 continue;
1946                         memcpy(log + baselen, de->d_name, namelen+1);
1947                         if (stat(git_path("logs/%s", log), &st) < 0)
1948                                 continue;
1949                         if (S_ISDIR(st.st_mode)) {
1950                                 retval = do_for_each_reflog(log, fn, cb_data);
1951                         } else {
1952                                 unsigned char sha1[20];
1953                                 if (read_ref_full(log, sha1, 0, NULL))
1954                                         retval = error("bad ref for %s", log);
1955                                 else
1956                                         retval = fn(log, sha1, 0, cb_data);
1957                         }
1958                         if (retval)
1959                                 break;
1960                 }
1961                 free(log);
1962                 closedir(dir);
1963         }
1964         else if (*base)
1965                 return errno;
1966         return retval;
1969 int for_each_reflog(each_ref_fn fn, void *cb_data)
1971         return do_for_each_reflog("", fn, cb_data);
1974 int update_ref(const char *action, const char *refname,
1975                 const unsigned char *sha1, const unsigned char *oldval,
1976                 int flags, enum action_on_err onerr)
1978         static struct ref_lock *lock;
1979         lock = lock_any_ref_for_update(refname, oldval, flags);
1980         if (!lock) {
1981                 const char *str = "Cannot lock the ref '%s'.";
1982                 switch (onerr) {
1983                 case MSG_ON_ERR: error(str, refname); break;
1984                 case DIE_ON_ERR: die(str, refname); break;
1985                 case QUIET_ON_ERR: break;
1986                 }
1987                 return 1;
1988         }
1989         if (write_ref_sha1(lock, sha1, action) < 0) {
1990                 const char *str = "Cannot update the ref '%s'.";
1991                 switch (onerr) {
1992                 case MSG_ON_ERR: error(str, refname); break;
1993                 case DIE_ON_ERR: die(str, refname); break;
1994                 case QUIET_ON_ERR: break;
1995                 }
1996                 return 1;
1997         }
1998         return 0;
2001 int ref_exists(const char *refname)
2003         unsigned char sha1[20];
2004         return !!resolve_ref(refname, sha1, 1, NULL);
2007 struct ref *find_ref_by_name(const struct ref *list, const char *name)
2009         for ( ; list; list = list->next)
2010                 if (!strcmp(list->name, name))
2011                         return (struct ref *)list;
2012         return NULL;
2015 /*
2016  * generate a format suitable for scanf from a ref_rev_parse_rules
2017  * rule, that is replace the "%.*s" spec with a "%s" spec
2018  */
2019 static void gen_scanf_fmt(char *scanf_fmt, const char *rule)
2021         char *spec;
2023         spec = strstr(rule, "%.*s");
2024         if (!spec || strstr(spec + 4, "%.*s"))
2025                 die("invalid rule in ref_rev_parse_rules: %s", rule);
2027         /* copy all until spec */
2028         strncpy(scanf_fmt, rule, spec - rule);
2029         scanf_fmt[spec - rule] = '\0';
2030         /* copy new spec */
2031         strcat(scanf_fmt, "%s");
2032         /* copy remaining rule */
2033         strcat(scanf_fmt, spec + 4);
2035         return;
2038 char *shorten_unambiguous_ref(const char *refname, int strict)
2040         int i;
2041         static char **scanf_fmts;
2042         static int nr_rules;
2043         char *short_name;
2045         /* pre generate scanf formats from ref_rev_parse_rules[] */
2046         if (!nr_rules) {
2047                 size_t total_len = 0;
2049                 /* the rule list is NULL terminated, count them first */
2050                 for (; ref_rev_parse_rules[nr_rules]; nr_rules++)
2051                         /* no +1 because strlen("%s") < strlen("%.*s") */
2052                         total_len += strlen(ref_rev_parse_rules[nr_rules]);
2054                 scanf_fmts = xmalloc(nr_rules * sizeof(char *) + total_len);
2056                 total_len = 0;
2057                 for (i = 0; i < nr_rules; i++) {
2058                         scanf_fmts[i] = (char *)&scanf_fmts[nr_rules]
2059                                         + total_len;
2060                         gen_scanf_fmt(scanf_fmts[i], ref_rev_parse_rules[i]);
2061                         total_len += strlen(ref_rev_parse_rules[i]);
2062                 }
2063         }
2065         /* bail out if there are no rules */
2066         if (!nr_rules)
2067                 return xstrdup(refname);
2069         /* buffer for scanf result, at most refname must fit */
2070         short_name = xstrdup(refname);
2072         /* skip first rule, it will always match */
2073         for (i = nr_rules - 1; i > 0 ; --i) {
2074                 int j;
2075                 int rules_to_fail = i;
2076                 int short_name_len;
2078                 if (1 != sscanf(refname, scanf_fmts[i], short_name))
2079                         continue;
2081                 short_name_len = strlen(short_name);
2083                 /*
2084                  * in strict mode, all (except the matched one) rules
2085                  * must fail to resolve to a valid non-ambiguous ref
2086                  */
2087                 if (strict)
2088                         rules_to_fail = nr_rules;
2090                 /*
2091                  * check if the short name resolves to a valid ref,
2092                  * but use only rules prior to the matched one
2093                  */
2094                 for (j = 0; j < rules_to_fail; j++) {
2095                         const char *rule = ref_rev_parse_rules[j];
2096                         char refname[PATH_MAX];
2098                         /* skip matched rule */
2099                         if (i == j)
2100                                 continue;
2102                         /*
2103                          * the short name is ambiguous, if it resolves
2104                          * (with this previous rule) to a valid ref
2105                          * read_ref() returns 0 on success
2106                          */
2107                         mksnpath(refname, sizeof(refname),
2108                                  rule, short_name_len, short_name);
2109                         if (ref_exists(refname))
2110                                 break;
2111                 }
2113                 /*
2114                  * short name is non-ambiguous if all previous rules
2115                  * haven't resolved to a valid ref
2116                  */
2117                 if (j == rules_to_fail)
2118                         return short_name;
2119         }
2121         free(short_name);
2122         return xstrdup(refname);