refs.con commit do_for_each_ref(): only iterate over the subtree that was requested (933ac03)
   1#include "cache.h"
   2#include "refs.h"
   3#include "object.h"
   4#include "tag.h"
   5#include "dir.h"
   6
   7/*
   8 * Make sure "ref" is something reasonable to have under ".git/refs/";
   9 * We do not like it if:
  10 *
  11 * - any path component of it begins with ".", or
  12 * - it has double dots "..", or
  13 * - it has ASCII control character, "~", "^", ":" or SP, anywhere, or
  14 * - it ends with a "/".
  15 * - it ends with ".lock"
  16 * - it contains a "\" (backslash)
  17 */
  18
  19/* Return true iff ch is not allowed in reference names. */
  20static inline int bad_ref_char(int ch)
  21{
  22        if (((unsigned) ch) <= ' ' || ch == 0x7f ||
  23            ch == '~' || ch == '^' || ch == ':' || ch == '\\')
  24                return 1;
  25        /* 2.13 Pattern Matching Notation */
  26        if (ch == '*' || ch == '?' || ch == '[') /* Unsupported */
  27                return 1;
  28        return 0;
  29}
  30
  31/*
  32 * Try to read one refname component from the front of refname.  Return
  33 * the length of the component found, or -1 if the component is not
  34 * legal.
  35 */
  36static int check_refname_component(const char *refname, int flags)
  37{
  38        const char *cp;
  39        char last = '\0';
  40
  41        for (cp = refname; ; cp++) {
  42                char ch = *cp;
  43                if (ch == '\0' || ch == '/')
  44                        break;
  45                if (bad_ref_char(ch))
  46                        return -1; /* Illegal character in refname. */
  47                if (last == '.' && ch == '.')
  48                        return -1; /* Refname contains "..". */
  49                if (last == '@' && ch == '{')
  50                        return -1; /* Refname contains "@{". */
  51                last = ch;
  52        }
  53        if (cp == refname)
  54                return 0; /* Component has zero length. */
  55        if (refname[0] == '.') {
  56                if (!(flags & REFNAME_DOT_COMPONENT))
  57                        return -1; /* Component starts with '.'. */
  58                /*
  59                 * Even if leading dots are allowed, don't allow "."
  60                 * as a component (".." is prevented by a rule above).
  61                 */
  62                if (refname[1] == '\0')
  63                        return -1; /* Component equals ".". */
  64        }
  65        if (cp - refname >= 5 && !memcmp(cp - 5, ".lock", 5))
  66                return -1; /* Refname ends with ".lock". */
  67        return cp - refname;
  68}
  69
  70int check_refname_format(const char *refname, int flags)
  71{
  72        int component_len, component_count = 0;
  73
  74        while (1) {
  75                /* We are at the start of a path component. */
  76                component_len = check_refname_component(refname, flags);
  77                if (component_len <= 0) {
  78                        if ((flags & REFNAME_REFSPEC_PATTERN) &&
  79                                        refname[0] == '*' &&
  80                                        (refname[1] == '\0' || refname[1] == '/')) {
  81                                /* Accept one wildcard as a full refname component. */
  82                                flags &= ~REFNAME_REFSPEC_PATTERN;
  83                                component_len = 1;
  84                        } else {
  85                                return -1;
  86                        }
  87                }
  88                component_count++;
  89                if (refname[component_len] == '\0')
  90                        break;
  91                /* Skip to next component. */
  92                refname += component_len + 1;
  93        }
  94
  95        if (refname[component_len - 1] == '.')
  96                return -1; /* Refname ends with '.'. */
  97        if (!(flags & REFNAME_ALLOW_ONELEVEL) && component_count < 2)
  98                return -1; /* Refname has only one component. */
  99        return 0;
 100}
 101
 102struct ref_entry;
 103
 104struct ref_value {
 105        unsigned char sha1[20];
 106        unsigned char peeled[20];
 107};
 108
 109struct ref_dir {
 110        int nr, alloc;
 111
 112        /*
 113         * Entries with index 0 <= i < sorted are sorted by name.  New
 114         * entries are appended to the list unsorted, and are sorted
 115         * only when required; thus we avoid the need to sort the list
 116         * after the addition of every reference.
 117         */
 118        int sorted;
 119
 120        struct ref_entry **entries;
 121};
 122
 123/* ISSYMREF=0x01, ISPACKED=0x02, and ISBROKEN=0x04 are public interfaces */
 124#define REF_KNOWS_PEELED 0x08
 125#define REF_DIR 0x10
 126
 127/*
 128 * A ref_entry represents either a reference or a "subdirectory" of
 129 * references.  Each directory in the reference namespace is
 130 * represented by a ref_entry with (flags & REF_DIR) set and
 131 * containing a subdir member that holds the entries in that
 132 * directory.  References are represented by a ref_entry with (flags &
 133 * REF_DIR) unset and a value member that describes the reference's
 134 * value.  The flag member is at the ref_entry level, but it is also
 135 * needed to interpret the contents of the value field (in other
 136 * words, a ref_value object is not very much use without the
 137 * enclosing ref_entry).
 138 *
 139 * Reference names cannot end with slash and directories' names are
 140 * always stored with a trailing slash (except for the top-level
 141 * directory, which is always denoted by "").  This has two nice
 142 * consequences: (1) when the entries in each subdir are sorted
 143 * lexicographically by name (as they usually are), the references in
 144 * a whole tree can be generated in lexicographic order by traversing
 145 * the tree in left-to-right, depth-first order; (2) the names of
 146 * references and subdirectories cannot conflict, and therefore the
 147 * presence of an empty subdirectory does not block the creation of a
 148 * similarly-named reference.  (The fact that reference names with the
 149 * same leading components can conflict *with each other* is a
 150 * separate issue that is regulated by is_refname_available().)
 151 *
 152 * Please note that the name field contains the fully-qualified
 153 * reference (or subdirectory) name.  Space could be saved by only
 154 * storing the relative names.  But that would require the full names
 155 * to be generated on the fly when iterating in do_for_each_ref(), and
 156 * would break callback functions, who have always been able to assume
 157 * that the name strings that they are passed will not be freed during
 158 * the iteration.
 159 */
 160struct ref_entry {
 161        unsigned char flag; /* ISSYMREF? ISPACKED? */
 162        union {
 163                struct ref_value value; /* if not (flags&REF_DIR) */
 164                struct ref_dir subdir; /* if (flags&REF_DIR) */
 165        } u;
 166        /*
 167         * The full name of the reference (e.g., "refs/heads/master")
 168         * or the full name of the directory with a trailing slash
 169         * (e.g., "refs/heads/"):
 170         */
 171        char name[FLEX_ARRAY];
 172};
 173
 174static struct ref_entry *create_ref_entry(const char *refname,
 175                                          const unsigned char *sha1, int flag,
 176                                          int check_name)
 177{
 178        int len;
 179        struct ref_entry *ref;
 180
 181        if (check_name &&
 182            check_refname_format(refname, REFNAME_ALLOW_ONELEVEL|REFNAME_DOT_COMPONENT))
 183                die("Reference has invalid format: '%s'", refname);
 184        len = strlen(refname) + 1;
 185        ref = xmalloc(sizeof(struct ref_entry) + len);
 186        hashcpy(ref->u.value.sha1, sha1);
 187        hashclr(ref->u.value.peeled);
 188        memcpy(ref->name, refname, len);
 189        ref->flag = flag;
 190        return ref;
 191}
 192
 193static void clear_ref_dir(struct ref_dir *dir);
 194
 195static void free_ref_entry(struct ref_entry *entry)
 196{
 197        if (entry->flag & REF_DIR)
 198                clear_ref_dir(&entry->u.subdir);
 199        free(entry);
 200}
 201
 202/*
 203 * Add a ref_entry to the end of dir (unsorted).  Entry is always
 204 * stored directly in dir; no recursion into subdirectories is
 205 * done.
 206 */
 207static void add_entry_to_dir(struct ref_dir *dir, struct ref_entry *entry)
 208{
 209        ALLOC_GROW(dir->entries, dir->nr + 1, dir->alloc);
 210        dir->entries[dir->nr++] = entry;
 211}
 212
 213/*
 214 * Clear and free all entries in dir, recursively.
 215 */
 216static void clear_ref_dir(struct ref_dir *dir)
 217{
 218        int i;
 219        for (i = 0; i < dir->nr; i++)
 220                free_ref_entry(dir->entries[i]);
 221        free(dir->entries);
 222        dir->sorted = dir->nr = dir->alloc = 0;
 223        dir->entries = NULL;
 224}
 225
 226/*
 227 * Create a struct ref_entry object for the specified dirname.
 228 * dirname is the name of the directory with a trailing slash (e.g.,
 229 * "refs/heads/") or "" for the top-level directory.
 230 */
 231static struct ref_entry *create_dir_entry(const char *dirname)
 232{
 233        struct ref_entry *direntry;
 234        int len = strlen(dirname);
 235        direntry = xcalloc(1, sizeof(struct ref_entry) + len + 1);
 236        memcpy(direntry->name, dirname, len + 1);
 237        direntry->flag = REF_DIR;
 238        return direntry;
 239}
 240
 241static int ref_entry_cmp(const void *a, const void *b)
 242{
 243        struct ref_entry *one = *(struct ref_entry **)a;
 244        struct ref_entry *two = *(struct ref_entry **)b;
 245        return strcmp(one->name, two->name);
 246}
 247
 248static void sort_ref_dir(struct ref_dir *dir);
 249
 250/*
 251 * Return the entry with the given refname from the ref_dir
 252 * (non-recursively), sorting dir if necessary.  Return NULL if no
 253 * such entry is found.
 254 */
 255static struct ref_entry *search_ref_dir(struct ref_dir *dir, const char *refname)
 256{
 257        struct ref_entry *e, **r;
 258        int len;
 259
 260        if (refname == NULL || !dir->nr)
 261                return NULL;
 262
 263        sort_ref_dir(dir);
 264
 265        len = strlen(refname) + 1;
 266        e = xmalloc(sizeof(struct ref_entry) + len);
 267        memcpy(e->name, refname, len);
 268
 269        r = bsearch(&e, dir->entries, dir->nr, sizeof(*dir->entries), ref_entry_cmp);
 270
 271        free(e);
 272
 273        if (r == NULL)
 274                return NULL;
 275
 276        return *r;
 277}
 278
 279/*
 280 * If refname is a reference name, find the ref_dir within the dir
 281 * tree that should hold refname.  If refname is a directory name
 282 * (i.e., ends in '/'), then return that ref_dir itself.  dir must
 283 * represent the top-level directory.  Sort ref_dirs and recurse into
 284 * subdirectories as necessary.  If mkdir is set, then create any
 285 * missing directories; otherwise, return NULL if the desired
 286 * directory cannot be found.
 287 */
 288static struct ref_dir *find_containing_dir(struct ref_dir *dir,
 289                                           const char *refname, int mkdir)
 290{
 291        char *refname_copy = xstrdup(refname);
 292        char *slash;
 293        struct ref_entry *entry;
 294        for (slash = strchr(refname_copy, '/'); slash; slash = strchr(slash + 1, '/')) {
 295                char tmp = slash[1];
 296                slash[1] = '\0';
 297                entry = search_ref_dir(dir, refname_copy);
 298                if (!entry) {
 299                        if (!mkdir) {
 300                                dir = NULL;
 301                                break;
 302                        }
 303                        entry = create_dir_entry(refname_copy);
 304                        add_entry_to_dir(dir, entry);
 305                }
 306                slash[1] = tmp;
 307                assert(entry->flag & REF_DIR);
 308                dir = &entry->u.subdir;
 309        }
 310
 311        free(refname_copy);
 312        return dir;
 313}
 314
 315/*
 316 * Find the value entry with the given name in dir, sorting ref_dirs
 317 * and recursing into subdirectories as necessary.  If the name is not
 318 * found or it corresponds to a directory entry, return NULL.
 319 */
 320static struct ref_entry *find_ref(struct ref_dir *dir, const char *refname)
 321{
 322        struct ref_entry *entry;
 323        dir = find_containing_dir(dir, refname, 0);
 324        if (!dir)
 325                return NULL;
 326        entry = search_ref_dir(dir, refname);
 327        return (entry && !(entry->flag & REF_DIR)) ? entry : NULL;
 328}
 329
 330/*
 331 * Add a ref_entry to the ref_dir (unsorted), recursing into
 332 * subdirectories as necessary.  dir must represent the top-level
 333 * directory.  Return 0 on success.
 334 */
 335static int add_ref(struct ref_dir *dir, struct ref_entry *ref)
 336{
 337        dir = find_containing_dir(dir, ref->name, 1);
 338        if (!dir)
 339                return -1;
 340        add_entry_to_dir(dir, ref);
 341        return 0;
 342}
 343
 344/*
 345 * Emit a warning and return true iff ref1 and ref2 have the same name
 346 * and the same sha1.  Die if they have the same name but different
 347 * sha1s.
 348 */
 349static int is_dup_ref(const struct ref_entry *ref1, const struct ref_entry *ref2)
 350{
 351        if (strcmp(ref1->name, ref2->name))
 352                return 0;
 353
 354        /* Duplicate name; make sure that they don't conflict: */
 355
 356        if ((ref1->flag & REF_DIR) || (ref2->flag & REF_DIR))
 357                /* This is impossible by construction */
 358                die("Reference directory conflict: %s", ref1->name);
 359
 360        if (hashcmp(ref1->u.value.sha1, ref2->u.value.sha1))
 361                die("Duplicated ref, and SHA1s don't match: %s", ref1->name);
 362
 363        warning("Duplicated ref: %s", ref1->name);
 364        return 1;
 365}
 366
 367/*
 368 * Sort the entries in dir non-recursively (if they are not already
 369 * sorted) and remove any duplicate entries.
 370 */
 371static void sort_ref_dir(struct ref_dir *dir)
 372{
 373        int i, j;
 374        struct ref_entry *last = NULL;
 375
 376        /*
 377         * This check also prevents passing a zero-length array to qsort(),
 378         * which is a problem on some platforms.
 379         */
 380        if (dir->sorted == dir->nr)
 381                return;
 382
 383        qsort(dir->entries, dir->nr, sizeof(*dir->entries), ref_entry_cmp);
 384
 385        /* Remove any duplicates: */
 386        for (i = 0, j = 0; j < dir->nr; j++) {
 387                struct ref_entry *entry = dir->entries[j];
 388                if (last && is_dup_ref(last, entry))
 389                        free_ref_entry(entry);
 390                else
 391                        last = dir->entries[i++] = entry;
 392        }
 393        dir->sorted = dir->nr = i;
 394}
 395
 396#define DO_FOR_EACH_INCLUDE_BROKEN 01
 397
 398static struct ref_entry *current_ref;
 399
 400static int do_one_ref(const char *base, each_ref_fn fn, int trim,
 401                      int flags, void *cb_data, struct ref_entry *entry)
 402{
 403        int retval;
 404        if (prefixcmp(entry->name, base))
 405                return 0;
 406
 407        if (!(flags & DO_FOR_EACH_INCLUDE_BROKEN)) {
 408                if (entry->flag & REF_ISBROKEN)
 409                        return 0; /* ignore broken refs e.g. dangling symref */
 410                if (!has_sha1_file(entry->u.value.sha1)) {
 411                        error("%s does not point to a valid object!", entry->name);
 412                        return 0;
 413                }
 414        }
 415        current_ref = entry;
 416        retval = fn(entry->name + trim, entry->u.value.sha1, entry->flag, cb_data);
 417        current_ref = NULL;
 418        return retval;
 419}
 420
 421/*
 422 * Call fn for each reference in dir that has index in the range
 423 * offset <= index < dir->nr.  Recurse into subdirectories that are in
 424 * that index range, sorting them before iterating.  This function
 425 * does not sort dir itself; it should be sorted beforehand.
 426 */
 427static int do_for_each_ref_in_dir(struct ref_dir *dir, int offset,
 428                                  const char *base,
 429                                  each_ref_fn fn, int trim, int flags, void *cb_data)
 430{
 431        int i;
 432        assert(dir->sorted == dir->nr);
 433        for (i = offset; i < dir->nr; i++) {
 434                struct ref_entry *entry = dir->entries[i];
 435                int retval;
 436                if (entry->flag & REF_DIR) {
 437                        sort_ref_dir(&entry->u.subdir);
 438                        retval = do_for_each_ref_in_dir(&entry->u.subdir, 0,
 439                                                        base, fn, trim, flags, cb_data);
 440                } else {
 441                        retval = do_one_ref(base, fn, trim, flags, cb_data, entry);
 442                }
 443                if (retval)
 444                        return retval;
 445        }
 446        return 0;
 447}
 448
 449/*
 450 * Call fn for each reference in the union of dir1 and dir2, in order
 451 * by refname.  Recurse into subdirectories.  If a value entry appears
 452 * in both dir1 and dir2, then only process the version that is in
 453 * dir2.  The input dirs must already be sorted, but subdirs will be
 454 * sorted as needed.
 455 */
 456static int do_for_each_ref_in_dirs(struct ref_dir *dir1,
 457                                   struct ref_dir *dir2,
 458                                   const char *base, each_ref_fn fn, int trim,
 459                                   int flags, void *cb_data)
 460{
 461        int retval;
 462        int i1 = 0, i2 = 0;
 463
 464        assert(dir1->sorted == dir1->nr);
 465        assert(dir2->sorted == dir2->nr);
 466        while (1) {
 467                struct ref_entry *e1, *e2;
 468                int cmp;
 469                if (i1 == dir1->nr) {
 470                        return do_for_each_ref_in_dir(dir2, i2,
 471                                                      base, fn, trim, flags, cb_data);
 472                }
 473                if (i2 == dir2->nr) {
 474                        return do_for_each_ref_in_dir(dir1, i1,
 475                                                      base, fn, trim, flags, cb_data);
 476                }
 477                e1 = dir1->entries[i1];
 478                e2 = dir2->entries[i2];
 479                cmp = strcmp(e1->name, e2->name);
 480                if (cmp == 0) {
 481                        if ((e1->flag & REF_DIR) && (e2->flag & REF_DIR)) {
 482                                /* Both are directories; descend them in parallel. */
 483                                sort_ref_dir(&e1->u.subdir);
 484                                sort_ref_dir(&e2->u.subdir);
 485                                retval = do_for_each_ref_in_dirs(
 486                                                &e1->u.subdir, &e2->u.subdir,
 487                                                base, fn, trim, flags, cb_data);
 488                                i1++;
 489                                i2++;
 490                        } else if (!(e1->flag & REF_DIR) && !(e2->flag & REF_DIR)) {
 491                                /* Both are references; ignore the one from dir1. */
 492                                retval = do_one_ref(base, fn, trim, flags, cb_data, e2);
 493                                i1++;
 494                                i2++;
 495                        } else {
 496                                die("conflict between reference and directory: %s",
 497                                    e1->name);
 498                        }
 499                } else {
 500                        struct ref_entry *e;
 501                        if (cmp < 0) {
 502                                e = e1;
 503                                i1++;
 504                        } else {
 505                                e = e2;
 506                                i2++;
 507                        }
 508                        if (e->flag & REF_DIR) {
 509                                sort_ref_dir(&e->u.subdir);
 510                                retval = do_for_each_ref_in_dir(
 511                                                &e->u.subdir, 0,
 512                                                base, fn, trim, flags, cb_data);
 513                        } else {
 514                                retval = do_one_ref(base, fn, trim, flags, cb_data, e);
 515                        }
 516                }
 517                if (retval)
 518                        return retval;
 519        }
 520        if (i1 < dir1->nr)
 521                return do_for_each_ref_in_dir(dir1, i1,
 522                                              base, fn, trim, flags, cb_data);
 523        if (i2 < dir2->nr)
 524                return do_for_each_ref_in_dir(dir2, i2,
 525                                              base, fn, trim, flags, cb_data);
 526        return 0;
 527}
 528
 529/*
 530 * Return true iff refname1 and refname2 conflict with each other.
 531 * Two reference names conflict if one of them exactly matches the
 532 * leading components of the other; e.g., "foo/bar" conflicts with
 533 * both "foo" and with "foo/bar/baz" but not with "foo/bar" or
 534 * "foo/barbados".
 535 */
 536static int names_conflict(const char *refname1, const char *refname2)
 537{
 538        for (; *refname1 && *refname1 == *refname2; refname1++, refname2++)
 539                ;
 540        return (*refname1 == '\0' && *refname2 == '/')
 541                || (*refname1 == '/' && *refname2 == '\0');
 542}
 543
 544struct name_conflict_cb {
 545        const char *refname;
 546        const char *oldrefname;
 547        const char *conflicting_refname;
 548};
 549
 550static int name_conflict_fn(const char *existingrefname, const unsigned char *sha1,
 551                            int flags, void *cb_data)
 552{
 553        struct name_conflict_cb *data = (struct name_conflict_cb *)cb_data;
 554        if (data->oldrefname && !strcmp(data->oldrefname, existingrefname))
 555                return 0;
 556        if (names_conflict(data->refname, existingrefname)) {
 557                data->conflicting_refname = existingrefname;
 558                return 1;
 559        }
 560        return 0;
 561}
 562
 563/*
 564 * Return true iff a reference named refname could be created without
 565 * conflicting with the name of an existing reference in array.  If
 566 * oldrefname is non-NULL, ignore potential conflicts with oldrefname
 567 * (e.g., because oldrefname is scheduled for deletion in the same
 568 * operation).
 569 */
 570static int is_refname_available(const char *refname, const char *oldrefname,
 571                                struct ref_dir *dir)
 572{
 573        struct name_conflict_cb data;
 574        data.refname = refname;
 575        data.oldrefname = oldrefname;
 576        data.conflicting_refname = NULL;
 577
 578        sort_ref_dir(dir);
 579        if (do_for_each_ref_in_dir(dir, 0, "", name_conflict_fn,
 580                                   0, DO_FOR_EACH_INCLUDE_BROKEN,
 581                                   &data)) {
 582                error("'%s' exists; cannot create '%s'",
 583                      data.conflicting_refname, refname);
 584                return 0;
 585        }
 586        return 1;
 587}
 588
 589/*
 590 * Future: need to be in "struct repository"
 591 * when doing a full libification.
 592 */
 593static struct ref_cache {
 594        struct ref_cache *next;
 595        char did_loose;
 596        char did_packed;
 597        struct ref_dir loose;
 598        struct ref_dir packed;
 599        /* The submodule name, or "" for the main repo. */
 600        char name[FLEX_ARRAY];
 601} *ref_cache;
 602
 603static void clear_packed_ref_cache(struct ref_cache *refs)
 604{
 605        if (refs->did_packed)
 606                clear_ref_dir(&refs->packed);
 607        refs->did_packed = 0;
 608}
 609
 610static void clear_loose_ref_cache(struct ref_cache *refs)
 611{
 612        if (refs->did_loose)
 613                clear_ref_dir(&refs->loose);
 614        refs->did_loose = 0;
 615}
 616
 617static struct ref_cache *create_ref_cache(const char *submodule)
 618{
 619        int len;
 620        struct ref_cache *refs;
 621        if (!submodule)
 622                submodule = "";
 623        len = strlen(submodule) + 1;
 624        refs = xcalloc(1, sizeof(struct ref_cache) + len);
 625        memcpy(refs->name, submodule, len);
 626        return refs;
 627}
 628
 629/*
 630 * Return a pointer to a ref_cache for the specified submodule. For
 631 * the main repository, use submodule==NULL. The returned structure
 632 * will be allocated and initialized but not necessarily populated; it
 633 * should not be freed.
 634 */
 635static struct ref_cache *get_ref_cache(const char *submodule)
 636{
 637        struct ref_cache *refs = ref_cache;
 638        if (!submodule)
 639                submodule = "";
 640        while (refs) {
 641                if (!strcmp(submodule, refs->name))
 642                        return refs;
 643                refs = refs->next;
 644        }
 645
 646        refs = create_ref_cache(submodule);
 647        refs->next = ref_cache;
 648        ref_cache = refs;
 649        return refs;
 650}
 651
 652void invalidate_ref_cache(const char *submodule)
 653{
 654        struct ref_cache *refs = get_ref_cache(submodule);
 655        clear_packed_ref_cache(refs);
 656        clear_loose_ref_cache(refs);
 657}
 658
 659/*
 660 * Parse one line from a packed-refs file.  Write the SHA1 to sha1.
 661 * Return a pointer to the refname within the line (null-terminated),
 662 * or NULL if there was a problem.
 663 */
 664static const char *parse_ref_line(char *line, unsigned char *sha1)
 665{
 666        /*
 667         * 42: the answer to everything.
 668         *
 669         * In this case, it happens to be the answer to
 670         *  40 (length of sha1 hex representation)
 671         *  +1 (space in between hex and name)
 672         *  +1 (newline at the end of the line)
 673         */
 674        int len = strlen(line) - 42;
 675
 676        if (len <= 0)
 677                return NULL;
 678        if (get_sha1_hex(line, sha1) < 0)
 679                return NULL;
 680        if (!isspace(line[40]))
 681                return NULL;
 682        line += 41;
 683        if (isspace(*line))
 684                return NULL;
 685        if (line[len] != '\n')
 686                return NULL;
 687        line[len] = 0;
 688
 689        return line;
 690}
 691
 692static void read_packed_refs(FILE *f, struct ref_dir *dir)
 693{
 694        struct ref_entry *last = NULL;
 695        char refline[PATH_MAX];
 696        int flag = REF_ISPACKED;
 697
 698        while (fgets(refline, sizeof(refline), f)) {
 699                unsigned char sha1[20];
 700                const char *refname;
 701                static const char header[] = "# pack-refs with:";
 702
 703                if (!strncmp(refline, header, sizeof(header)-1)) {
 704                        const char *traits = refline + sizeof(header) - 1;
 705                        if (strstr(traits, " peeled "))
 706                                flag |= REF_KNOWS_PEELED;
 707                        /* perhaps other traits later as well */
 708                        continue;
 709                }
 710
 711                refname = parse_ref_line(refline, sha1);
 712                if (refname) {
 713                        last = create_ref_entry(refname, sha1, flag, 1);
 714                        add_ref(dir, last);
 715                        continue;
 716                }
 717                if (last &&
 718                    refline[0] == '^' &&
 719                    strlen(refline) == 42 &&
 720                    refline[41] == '\n' &&
 721                    !get_sha1_hex(refline + 1, sha1))
 722                        hashcpy(last->u.value.peeled, sha1);
 723        }
 724}
 725
 726static struct ref_dir *get_packed_refs(struct ref_cache *refs)
 727{
 728        if (!refs->did_packed) {
 729                const char *packed_refs_file;
 730                FILE *f;
 731
 732                if (*refs->name)
 733                        packed_refs_file = git_path_submodule(refs->name, "packed-refs");
 734                else
 735                        packed_refs_file = git_path("packed-refs");
 736                f = fopen(packed_refs_file, "r");
 737                if (f) {
 738                        read_packed_refs(f, &refs->packed);
 739                        fclose(f);
 740                }
 741                refs->did_packed = 1;
 742        }
 743        return &refs->packed;
 744}
 745
 746void add_packed_ref(const char *refname, const unsigned char *sha1)
 747{
 748        add_ref(get_packed_refs(get_ref_cache(NULL)),
 749                        create_ref_entry(refname, sha1, REF_ISPACKED, 1));
 750}
 751
 752static void get_ref_dir(struct ref_cache *refs, const char *base,
 753                        struct ref_dir *dir)
 754{
 755        DIR *d;
 756        const char *path;
 757
 758        if (*refs->name)
 759                path = git_path_submodule(refs->name, "%s", base);
 760        else
 761                path = git_path("%s", base);
 762
 763        d = opendir(path);
 764        if (d) {
 765                struct dirent *de;
 766                int baselen = strlen(base);
 767                char *refname = xmalloc(baselen + 257);
 768
 769                memcpy(refname, base, baselen);
 770                if (baselen && base[baselen-1] != '/')
 771                        refname[baselen++] = '/';
 772
 773                while ((de = readdir(d)) != NULL) {
 774                        unsigned char sha1[20];
 775                        struct stat st;
 776                        int flag;
 777                        int namelen;
 778                        const char *refdir;
 779
 780                        if (de->d_name[0] == '.')
 781                                continue;
 782                        namelen = strlen(de->d_name);
 783                        if (namelen > 255)
 784                                continue;
 785                        if (has_extension(de->d_name, ".lock"))
 786                                continue;
 787                        memcpy(refname + baselen, de->d_name, namelen+1);
 788                        refdir = *refs->name
 789                                ? git_path_submodule(refs->name, "%s", refname)
 790                                : git_path("%s", refname);
 791                        if (stat(refdir, &st) < 0)
 792                                continue;
 793                        if (S_ISDIR(st.st_mode)) {
 794                                get_ref_dir(refs, refname, dir);
 795                                continue;
 796                        }
 797                        if (*refs->name) {
 798                                hashclr(sha1);
 799                                flag = 0;
 800                                if (resolve_gitlink_ref(refs->name, refname, sha1) < 0) {
 801                                        hashclr(sha1);
 802                                        flag |= REF_ISBROKEN;
 803                                }
 804                        } else if (read_ref_full(refname, sha1, 1, &flag)) {
 805                                hashclr(sha1);
 806                                flag |= REF_ISBROKEN;
 807                        }
 808                        add_ref(dir, create_ref_entry(refname, sha1, flag, 1));
 809                }
 810                free(refname);
 811                closedir(d);
 812        }
 813}
 814
 815static struct ref_dir *get_loose_refs(struct ref_cache *refs)
 816{
 817        if (!refs->did_loose) {
 818                get_ref_dir(refs, "refs", &refs->loose);
 819                refs->did_loose = 1;
 820        }
 821        return &refs->loose;
 822}
 823
 824/* We allow "recursive" symbolic refs. Only within reason, though */
 825#define MAXDEPTH 5
 826#define MAXREFLEN (1024)
 827
 828/*
 829 * Called by resolve_gitlink_ref_recursive() after it failed to read
 830 * from the loose refs in ref_cache refs. Find <refname> in the
 831 * packed-refs file for the submodule.
 832 */
 833static int resolve_gitlink_packed_ref(struct ref_cache *refs,
 834                                      const char *refname, unsigned char *sha1)
 835{
 836        struct ref_entry *ref;
 837        struct ref_dir *dir = get_packed_refs(refs);
 838
 839        ref = find_ref(dir, refname);
 840        if (ref == NULL)
 841                return -1;
 842
 843        memcpy(sha1, ref->u.value.sha1, 20);
 844        return 0;
 845}
 846
 847static int resolve_gitlink_ref_recursive(struct ref_cache *refs,
 848                                         const char *refname, unsigned char *sha1,
 849                                         int recursion)
 850{
 851        int fd, len;
 852        char buffer[128], *p;
 853        char *path;
 854
 855        if (recursion > MAXDEPTH || strlen(refname) > MAXREFLEN)
 856                return -1;
 857        path = *refs->name
 858                ? git_path_submodule(refs->name, "%s", refname)
 859                : git_path("%s", refname);
 860        fd = open(path, O_RDONLY);
 861        if (fd < 0)
 862                return resolve_gitlink_packed_ref(refs, refname, sha1);
 863
 864        len = read(fd, buffer, sizeof(buffer)-1);
 865        close(fd);
 866        if (len < 0)
 867                return -1;
 868        while (len && isspace(buffer[len-1]))
 869                len--;
 870        buffer[len] = 0;
 871
 872        /* Was it a detached head or an old-fashioned symlink? */
 873        if (!get_sha1_hex(buffer, sha1))
 874                return 0;
 875
 876        /* Symref? */
 877        if (strncmp(buffer, "ref:", 4))
 878                return -1;
 879        p = buffer + 4;
 880        while (isspace(*p))
 881                p++;
 882
 883        return resolve_gitlink_ref_recursive(refs, p, sha1, recursion+1);
 884}
 885
 886int resolve_gitlink_ref(const char *path, const char *refname, unsigned char *sha1)
 887{
 888        int len = strlen(path), retval;
 889        char *submodule;
 890        struct ref_cache *refs;
 891
 892        while (len && path[len-1] == '/')
 893                len--;
 894        if (!len)
 895                return -1;
 896        submodule = xstrndup(path, len);
 897        refs = get_ref_cache(submodule);
 898        free(submodule);
 899
 900        retval = resolve_gitlink_ref_recursive(refs, refname, sha1, 0);
 901        return retval;
 902}
 903
 904/*
 905 * Try to read ref from the packed references.  On success, set sha1
 906 * and return 0; otherwise, return -1.
 907 */
 908static int get_packed_ref(const char *refname, unsigned char *sha1)
 909{
 910        struct ref_dir *packed = get_packed_refs(get_ref_cache(NULL));
 911        struct ref_entry *entry = find_ref(packed, refname);
 912        if (entry) {
 913                hashcpy(sha1, entry->u.value.sha1);
 914                return 0;
 915        }
 916        return -1;
 917}
 918
 919const char *resolve_ref_unsafe(const char *refname, unsigned char *sha1, int reading, int *flag)
 920{
 921        int depth = MAXDEPTH;
 922        ssize_t len;
 923        char buffer[256];
 924        static char refname_buffer[256];
 925
 926        if (flag)
 927                *flag = 0;
 928
 929        if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL))
 930                return NULL;
 931
 932        for (;;) {
 933                char path[PATH_MAX];
 934                struct stat st;
 935                char *buf;
 936                int fd;
 937
 938                if (--depth < 0)
 939                        return NULL;
 940
 941                git_snpath(path, sizeof(path), "%s", refname);
 942
 943                if (lstat(path, &st) < 0) {
 944                        if (errno != ENOENT)
 945                                return NULL;
 946                        /*
 947                         * The loose reference file does not exist;
 948                         * check for a packed reference.
 949                         */
 950                        if (!get_packed_ref(refname, sha1)) {
 951                                if (flag)
 952                                        *flag |= REF_ISPACKED;
 953                                return refname;
 954                        }
 955                        /* The reference is not a packed reference, either. */
 956                        if (reading) {
 957                                return NULL;
 958                        } else {
 959                                hashclr(sha1);
 960                                return refname;
 961                        }
 962                }
 963
 964                /* Follow "normalized" - ie "refs/.." symlinks by hand */
 965                if (S_ISLNK(st.st_mode)) {
 966                        len = readlink(path, buffer, sizeof(buffer)-1);
 967                        if (len < 0)
 968                                return NULL;
 969                        buffer[len] = 0;
 970                        if (!prefixcmp(buffer, "refs/") &&
 971                                        !check_refname_format(buffer, 0)) {
 972                                strcpy(refname_buffer, buffer);
 973                                refname = refname_buffer;
 974                                if (flag)
 975                                        *flag |= REF_ISSYMREF;
 976                                continue;
 977                        }
 978                }
 979
 980                /* Is it a directory? */
 981                if (S_ISDIR(st.st_mode)) {
 982                        errno = EISDIR;
 983                        return NULL;
 984                }
 985
 986                /*
 987                 * Anything else, just open it and try to use it as
 988                 * a ref
 989                 */
 990                fd = open(path, O_RDONLY);
 991                if (fd < 0)
 992                        return NULL;
 993                len = read_in_full(fd, buffer, sizeof(buffer)-1);
 994                close(fd);
 995                if (len < 0)
 996                        return NULL;
 997                while (len && isspace(buffer[len-1]))
 998                        len--;
 999                buffer[len] = '\0';
1000
1001                /*
1002                 * Is it a symbolic ref?
1003                 */
1004                if (prefixcmp(buffer, "ref:"))
1005                        break;
1006                if (flag)
1007                        *flag |= REF_ISSYMREF;
1008                buf = buffer + 4;
1009                while (isspace(*buf))
1010                        buf++;
1011                if (check_refname_format(buf, REFNAME_ALLOW_ONELEVEL)) {
1012                        if (flag)
1013                                *flag |= REF_ISBROKEN;
1014                        return NULL;
1015                }
1016                refname = strcpy(refname_buffer, buf);
1017        }
1018        /* Please note that FETCH_HEAD has a second line containing other data. */
1019        if (get_sha1_hex(buffer, sha1) || (buffer[40] != '\0' && !isspace(buffer[40]))) {
1020                if (flag)
1021                        *flag |= REF_ISBROKEN;
1022                return NULL;
1023        }
1024        return refname;
1025}
1026
1027char *resolve_refdup(const char *ref, unsigned char *sha1, int reading, int *flag)
1028{
1029        const char *ret = resolve_ref_unsafe(ref, sha1, reading, flag);
1030        return ret ? xstrdup(ret) : NULL;
1031}
1032
1033/* The argument to filter_refs */
1034struct ref_filter {
1035        const char *pattern;
1036        each_ref_fn *fn;
1037        void *cb_data;
1038};
1039
1040int read_ref_full(const char *refname, unsigned char *sha1, int reading, int *flags)
1041{
1042        if (resolve_ref_unsafe(refname, sha1, reading, flags))
1043                return 0;
1044        return -1;
1045}
1046
1047int read_ref(const char *refname, unsigned char *sha1)
1048{
1049        return read_ref_full(refname, sha1, 1, NULL);
1050}
1051
1052int ref_exists(const char *refname)
1053{
1054        unsigned char sha1[20];
1055        return !!resolve_ref_unsafe(refname, sha1, 1, NULL);
1056}
1057
1058static int filter_refs(const char *refname, const unsigned char *sha1, int flags,
1059                       void *data)
1060{
1061        struct ref_filter *filter = (struct ref_filter *)data;
1062        if (fnmatch(filter->pattern, refname, 0))
1063                return 0;
1064        return filter->fn(refname, sha1, flags, filter->cb_data);
1065}
1066
1067int peel_ref(const char *refname, unsigned char *sha1)
1068{
1069        int flag;
1070        unsigned char base[20];
1071        struct object *o;
1072
1073        if (current_ref && (current_ref->name == refname
1074                || !strcmp(current_ref->name, refname))) {
1075                if (current_ref->flag & REF_KNOWS_PEELED) {
1076                        hashcpy(sha1, current_ref->u.value.peeled);
1077                        return 0;
1078                }
1079                hashcpy(base, current_ref->u.value.sha1);
1080                goto fallback;
1081        }
1082
1083        if (read_ref_full(refname, base, 1, &flag))
1084                return -1;
1085
1086        if ((flag & REF_ISPACKED)) {
1087                struct ref_dir *dir = get_packed_refs(get_ref_cache(NULL));
1088                struct ref_entry *r = find_ref(dir, refname);
1089
1090                if (r != NULL && r->flag & REF_KNOWS_PEELED) {
1091                        hashcpy(sha1, r->u.value.peeled);
1092                        return 0;
1093                }
1094        }
1095
1096fallback:
1097        o = parse_object(base);
1098        if (o && o->type == OBJ_TAG) {
1099                o = deref_tag(o, refname, 0);
1100                if (o) {
1101                        hashcpy(sha1, o->sha1);
1102                        return 0;
1103                }
1104        }
1105        return -1;
1106}
1107
1108struct warn_if_dangling_data {
1109        FILE *fp;
1110        const char *refname;
1111        const char *msg_fmt;
1112};
1113
1114static int warn_if_dangling_symref(const char *refname, const unsigned char *sha1,
1115                                   int flags, void *cb_data)
1116{
1117        struct warn_if_dangling_data *d = cb_data;
1118        const char *resolves_to;
1119        unsigned char junk[20];
1120
1121        if (!(flags & REF_ISSYMREF))
1122                return 0;
1123
1124        resolves_to = resolve_ref_unsafe(refname, junk, 0, NULL);
1125        if (!resolves_to || strcmp(resolves_to, d->refname))
1126                return 0;
1127
1128        fprintf(d->fp, d->msg_fmt, refname);
1129        return 0;
1130}
1131
1132void warn_dangling_symref(FILE *fp, const char *msg_fmt, const char *refname)
1133{
1134        struct warn_if_dangling_data data;
1135
1136        data.fp = fp;
1137        data.refname = refname;
1138        data.msg_fmt = msg_fmt;
1139        for_each_rawref(warn_if_dangling_symref, &data);
1140}
1141
1142static int do_for_each_ref(const char *submodule, const char *base, each_ref_fn fn,
1143                           int trim, int flags, void *cb_data)
1144{
1145        struct ref_cache *refs = get_ref_cache(submodule);
1146        struct ref_dir *packed_dir = get_packed_refs(refs);
1147        struct ref_dir *loose_dir = get_loose_refs(refs);
1148        int retval = 0;
1149
1150        if (base && *base) {
1151                packed_dir = find_containing_dir(packed_dir, base, 0);
1152                loose_dir = find_containing_dir(loose_dir, base, 0);
1153        }
1154
1155        if (packed_dir && loose_dir) {
1156                sort_ref_dir(packed_dir);
1157                sort_ref_dir(loose_dir);
1158                retval = do_for_each_ref_in_dirs(
1159                                packed_dir, loose_dir,
1160                                base, fn, trim, flags, cb_data);
1161        } else if (packed_dir) {
1162                sort_ref_dir(packed_dir);
1163                retval = do_for_each_ref_in_dir(
1164                                packed_dir, 0,
1165                                base, fn, trim, flags, cb_data);
1166        } else if (loose_dir) {
1167                sort_ref_dir(loose_dir);
1168                retval = do_for_each_ref_in_dir(
1169                                loose_dir, 0,
1170                                base, fn, trim, flags, cb_data);
1171        }
1172
1173        return retval;
1174}
1175
1176static int do_head_ref(const char *submodule, each_ref_fn fn, void *cb_data)
1177{
1178        unsigned char sha1[20];
1179        int flag;
1180
1181        if (submodule) {
1182                if (resolve_gitlink_ref(submodule, "HEAD", sha1) == 0)
1183                        return fn("HEAD", sha1, 0, cb_data);
1184
1185                return 0;
1186        }
1187
1188        if (!read_ref_full("HEAD", sha1, 1, &flag))
1189                return fn("HEAD", sha1, flag, cb_data);
1190
1191        return 0;
1192}
1193
1194int head_ref(each_ref_fn fn, void *cb_data)
1195{
1196        return do_head_ref(NULL, fn, cb_data);
1197}
1198
1199int head_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
1200{
1201        return do_head_ref(submodule, fn, cb_data);
1202}
1203
1204int for_each_ref(each_ref_fn fn, void *cb_data)
1205{
1206        return do_for_each_ref(NULL, "", fn, 0, 0, cb_data);
1207}
1208
1209int for_each_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
1210{
1211        return do_for_each_ref(submodule, "", fn, 0, 0, cb_data);
1212}
1213
1214int for_each_ref_in(const char *prefix, each_ref_fn fn, void *cb_data)
1215{
1216        return do_for_each_ref(NULL, prefix, fn, strlen(prefix), 0, cb_data);
1217}
1218
1219int for_each_ref_in_submodule(const char *submodule, const char *prefix,
1220                each_ref_fn fn, void *cb_data)
1221{
1222        return do_for_each_ref(submodule, prefix, fn, strlen(prefix), 0, cb_data);
1223}
1224
1225int for_each_tag_ref(each_ref_fn fn, void *cb_data)
1226{
1227        return for_each_ref_in("refs/tags/", fn, cb_data);
1228}
1229
1230int for_each_tag_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
1231{
1232        return for_each_ref_in_submodule(submodule, "refs/tags/", fn, cb_data);
1233}
1234
1235int for_each_branch_ref(each_ref_fn fn, void *cb_data)
1236{
1237        return for_each_ref_in("refs/heads/", fn, cb_data);
1238}
1239
1240int for_each_branch_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
1241{
1242        return for_each_ref_in_submodule(submodule, "refs/heads/", fn, cb_data);
1243}
1244
1245int for_each_remote_ref(each_ref_fn fn, void *cb_data)
1246{
1247        return for_each_ref_in("refs/remotes/", fn, cb_data);
1248}
1249
1250int for_each_remote_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
1251{
1252        return for_each_ref_in_submodule(submodule, "refs/remotes/", fn, cb_data);
1253}
1254
1255int for_each_replace_ref(each_ref_fn fn, void *cb_data)
1256{
1257        return do_for_each_ref(NULL, "refs/replace/", fn, 13, 0, cb_data);
1258}
1259
1260int head_ref_namespaced(each_ref_fn fn, void *cb_data)
1261{
1262        struct strbuf buf = STRBUF_INIT;
1263        int ret = 0;
1264        unsigned char sha1[20];
1265        int flag;
1266
1267        strbuf_addf(&buf, "%sHEAD", get_git_namespace());
1268        if (!read_ref_full(buf.buf, sha1, 1, &flag))
1269                ret = fn(buf.buf, sha1, flag, cb_data);
1270        strbuf_release(&buf);
1271
1272        return ret;
1273}
1274
1275int for_each_namespaced_ref(each_ref_fn fn, void *cb_data)
1276{
1277        struct strbuf buf = STRBUF_INIT;
1278        int ret;
1279        strbuf_addf(&buf, "%srefs/", get_git_namespace());
1280        ret = do_for_each_ref(NULL, buf.buf, fn, 0, 0, cb_data);
1281        strbuf_release(&buf);
1282        return ret;
1283}
1284
1285int for_each_glob_ref_in(each_ref_fn fn, const char *pattern,
1286        const char *prefix, void *cb_data)
1287{
1288        struct strbuf real_pattern = STRBUF_INIT;
1289        struct ref_filter filter;
1290        int ret;
1291
1292        if (!prefix && prefixcmp(pattern, "refs/"))
1293                strbuf_addstr(&real_pattern, "refs/");
1294        else if (prefix)
1295                strbuf_addstr(&real_pattern, prefix);
1296        strbuf_addstr(&real_pattern, pattern);
1297
1298        if (!has_glob_specials(pattern)) {
1299                /* Append implied '/' '*' if not present. */
1300                if (real_pattern.buf[real_pattern.len - 1] != '/')
1301                        strbuf_addch(&real_pattern, '/');
1302                /* No need to check for '*', there is none. */
1303                strbuf_addch(&real_pattern, '*');
1304        }
1305
1306        filter.pattern = real_pattern.buf;
1307        filter.fn = fn;
1308        filter.cb_data = cb_data;
1309        ret = for_each_ref(filter_refs, &filter);
1310
1311        strbuf_release(&real_pattern);
1312        return ret;
1313}
1314
1315int for_each_glob_ref(each_ref_fn fn, const char *pattern, void *cb_data)
1316{
1317        return for_each_glob_ref_in(fn, pattern, NULL, cb_data);
1318}
1319
1320int for_each_rawref(each_ref_fn fn, void *cb_data)
1321{
1322        return do_for_each_ref(NULL, "", fn, 0,
1323                               DO_FOR_EACH_INCLUDE_BROKEN, cb_data);
1324}
1325
1326const char *prettify_refname(const char *name)
1327{
1328        return name + (
1329                !prefixcmp(name, "refs/heads/") ? 11 :
1330                !prefixcmp(name, "refs/tags/") ? 10 :
1331                !prefixcmp(name, "refs/remotes/") ? 13 :
1332                0);
1333}
1334
1335const char *ref_rev_parse_rules[] = {
1336        "%.*s",
1337        "refs/%.*s",
1338        "refs/tags/%.*s",
1339        "refs/heads/%.*s",
1340        "refs/remotes/%.*s",
1341        "refs/remotes/%.*s/HEAD",
1342        NULL
1343};
1344
1345int refname_match(const char *abbrev_name, const char *full_name, const char **rules)
1346{
1347        const char **p;
1348        const int abbrev_name_len = strlen(abbrev_name);
1349
1350        for (p = rules; *p; p++) {
1351                if (!strcmp(full_name, mkpath(*p, abbrev_name_len, abbrev_name))) {
1352                        return 1;
1353                }
1354        }
1355
1356        return 0;
1357}
1358
1359static struct ref_lock *verify_lock(struct ref_lock *lock,
1360        const unsigned char *old_sha1, int mustexist)
1361{
1362        if (read_ref_full(lock->ref_name, lock->old_sha1, mustexist, NULL)) {
1363                error("Can't verify ref %s", lock->ref_name);
1364                unlock_ref(lock);
1365                return NULL;
1366        }
1367        if (hashcmp(lock->old_sha1, old_sha1)) {
1368                error("Ref %s is at %s but expected %s", lock->ref_name,
1369                        sha1_to_hex(lock->old_sha1), sha1_to_hex(old_sha1));
1370                unlock_ref(lock);
1371                return NULL;
1372        }
1373        return lock;
1374}
1375
1376static int remove_empty_directories(const char *file)
1377{
1378        /* we want to create a file but there is a directory there;
1379         * if that is an empty directory (or a directory that contains
1380         * only empty directories), remove them.
1381         */
1382        struct strbuf path;
1383        int result;
1384
1385        strbuf_init(&path, 20);
1386        strbuf_addstr(&path, file);
1387
1388        result = remove_dir_recursively(&path, REMOVE_DIR_EMPTY_ONLY);
1389
1390        strbuf_release(&path);
1391
1392        return result;
1393}
1394
1395/*
1396 * *string and *len will only be substituted, and *string returned (for
1397 * later free()ing) if the string passed in is a magic short-hand form
1398 * to name a branch.
1399 */
1400static char *substitute_branch_name(const char **string, int *len)
1401{
1402        struct strbuf buf = STRBUF_INIT;
1403        int ret = interpret_branch_name(*string, &buf);
1404
1405        if (ret == *len) {
1406                size_t size;
1407                *string = strbuf_detach(&buf, &size);
1408                *len = size;
1409                return (char *)*string;
1410        }
1411
1412        return NULL;
1413}
1414
1415int dwim_ref(const char *str, int len, unsigned char *sha1, char **ref)
1416{
1417        char *last_branch = substitute_branch_name(&str, &len);
1418        const char **p, *r;
1419        int refs_found = 0;
1420
1421        *ref = NULL;
1422        for (p = ref_rev_parse_rules; *p; p++) {
1423                char fullref[PATH_MAX];
1424                unsigned char sha1_from_ref[20];
1425                unsigned char *this_result;
1426                int flag;
1427
1428                this_result = refs_found ? sha1_from_ref : sha1;
1429                mksnpath(fullref, sizeof(fullref), *p, len, str);
1430                r = resolve_ref_unsafe(fullref, this_result, 1, &flag);
1431                if (r) {
1432                        if (!refs_found++)
1433                                *ref = xstrdup(r);
1434                        if (!warn_ambiguous_refs)
1435                                break;
1436                } else if ((flag & REF_ISSYMREF) && strcmp(fullref, "HEAD")) {
1437                        warning("ignoring dangling symref %s.", fullref);
1438                } else if ((flag & REF_ISBROKEN) && strchr(fullref, '/')) {
1439                        warning("ignoring broken ref %s.", fullref);
1440                }
1441        }
1442        free(last_branch);
1443        return refs_found;
1444}
1445
1446int dwim_log(const char *str, int len, unsigned char *sha1, char **log)
1447{
1448        char *last_branch = substitute_branch_name(&str, &len);
1449        const char **p;
1450        int logs_found = 0;
1451
1452        *log = NULL;
1453        for (p = ref_rev_parse_rules; *p; p++) {
1454                struct stat st;
1455                unsigned char hash[20];
1456                char path[PATH_MAX];
1457                const char *ref, *it;
1458
1459                mksnpath(path, sizeof(path), *p, len, str);
1460                ref = resolve_ref_unsafe(path, hash, 1, NULL);
1461                if (!ref)
1462                        continue;
1463                if (!stat(git_path("logs/%s", path), &st) &&
1464                    S_ISREG(st.st_mode))
1465                        it = path;
1466                else if (strcmp(ref, path) &&
1467                         !stat(git_path("logs/%s", ref), &st) &&
1468                         S_ISREG(st.st_mode))
1469                        it = ref;
1470                else
1471                        continue;
1472                if (!logs_found++) {
1473                        *log = xstrdup(it);
1474                        hashcpy(sha1, hash);
1475                }
1476                if (!warn_ambiguous_refs)
1477                        break;
1478        }
1479        free(last_branch);
1480        return logs_found;
1481}
1482
1483static struct ref_lock *lock_ref_sha1_basic(const char *refname,
1484                                            const unsigned char *old_sha1,
1485                                            int flags, int *type_p)
1486{
1487        char *ref_file;
1488        const char *orig_refname = refname;
1489        struct ref_lock *lock;
1490        int last_errno = 0;
1491        int type, lflags;
1492        int mustexist = (old_sha1 && !is_null_sha1(old_sha1));
1493        int missing = 0;
1494
1495        lock = xcalloc(1, sizeof(struct ref_lock));
1496        lock->lock_fd = -1;
1497
1498        refname = resolve_ref_unsafe(refname, lock->old_sha1, mustexist, &type);
1499        if (!refname && errno == EISDIR) {
1500                /* we are trying to lock foo but we used to
1501                 * have foo/bar which now does not exist;
1502                 * it is normal for the empty directory 'foo'
1503                 * to remain.
1504                 */
1505                ref_file = git_path("%s", orig_refname);
1506                if (remove_empty_directories(ref_file)) {
1507                        last_errno = errno;
1508                        error("there are still refs under '%s'", orig_refname);
1509                        goto error_return;
1510                }
1511                refname = resolve_ref_unsafe(orig_refname, lock->old_sha1, mustexist, &type);
1512        }
1513        if (type_p)
1514            *type_p = type;
1515        if (!refname) {
1516                last_errno = errno;
1517                error("unable to resolve reference %s: %s",
1518                        orig_refname, strerror(errno));
1519                goto error_return;
1520        }
1521        missing = is_null_sha1(lock->old_sha1);
1522        /* When the ref did not exist and we are creating it,
1523         * make sure there is no existing ref that is packed
1524         * whose name begins with our refname, nor a ref whose
1525         * name is a proper prefix of our refname.
1526         */
1527        if (missing &&
1528             !is_refname_available(refname, NULL, get_packed_refs(get_ref_cache(NULL)))) {
1529                last_errno = ENOTDIR;
1530                goto error_return;
1531        }
1532
1533        lock->lk = xcalloc(1, sizeof(struct lock_file));
1534
1535        lflags = LOCK_DIE_ON_ERROR;
1536        if (flags & REF_NODEREF) {
1537                refname = orig_refname;
1538                lflags |= LOCK_NODEREF;
1539        }
1540        lock->ref_name = xstrdup(refname);
1541        lock->orig_ref_name = xstrdup(orig_refname);
1542        ref_file = git_path("%s", refname);
1543        if (missing)
1544                lock->force_write = 1;
1545        if ((flags & REF_NODEREF) && (type & REF_ISSYMREF))
1546                lock->force_write = 1;
1547
1548        if (safe_create_leading_directories(ref_file)) {
1549                last_errno = errno;
1550                error("unable to create directory for %s", ref_file);
1551                goto error_return;
1552        }
1553
1554        lock->lock_fd = hold_lock_file_for_update(lock->lk, ref_file, lflags);
1555        return old_sha1 ? verify_lock(lock, old_sha1, mustexist) : lock;
1556
1557 error_return:
1558        unlock_ref(lock);
1559        errno = last_errno;
1560        return NULL;
1561}
1562
1563struct ref_lock *lock_ref_sha1(const char *refname, const unsigned char *old_sha1)
1564{
1565        char refpath[PATH_MAX];
1566        if (check_refname_format(refname, 0))
1567                return NULL;
1568        strcpy(refpath, mkpath("refs/%s", refname));
1569        return lock_ref_sha1_basic(refpath, old_sha1, 0, NULL);
1570}
1571
1572struct ref_lock *lock_any_ref_for_update(const char *refname,
1573                                         const unsigned char *old_sha1, int flags)
1574{
1575        if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL))
1576                return NULL;
1577        return lock_ref_sha1_basic(refname, old_sha1, flags, NULL);
1578}
1579
1580struct repack_without_ref_sb {
1581        const char *refname;
1582        int fd;
1583};
1584
1585static int repack_without_ref_fn(const char *refname, const unsigned char *sha1,
1586                                 int flags, void *cb_data)
1587{
1588        struct repack_without_ref_sb *data = cb_data;
1589        char line[PATH_MAX + 100];
1590        int len;
1591
1592        if (!strcmp(data->refname, refname))
1593                return 0;
1594        len = snprintf(line, sizeof(line), "%s %s\n",
1595                       sha1_to_hex(sha1), refname);
1596        /* this should not happen but just being defensive */
1597        if (len > sizeof(line))
1598                die("too long a refname '%s'", refname);
1599        write_or_die(data->fd, line, len);
1600        return 0;
1601}
1602
1603static struct lock_file packlock;
1604
1605static int repack_without_ref(const char *refname)
1606{
1607        struct repack_without_ref_sb data;
1608        struct ref_dir *packed = get_packed_refs(get_ref_cache(NULL));
1609        if (find_ref(packed, refname) == NULL)
1610                return 0;
1611        data.refname = refname;
1612        data.fd = hold_lock_file_for_update(&packlock, git_path("packed-refs"), 0);
1613        if (data.fd < 0) {
1614                unable_to_lock_error(git_path("packed-refs"), errno);
1615                return error("cannot delete '%s' from packed refs", refname);
1616        }
1617        do_for_each_ref_in_dir(packed, 0, "", repack_without_ref_fn, 0, 0, &data);
1618        return commit_lock_file(&packlock);
1619}
1620
1621int delete_ref(const char *refname, const unsigned char *sha1, int delopt)
1622{
1623        struct ref_lock *lock;
1624        int err, i = 0, ret = 0, flag = 0;
1625
1626        lock = lock_ref_sha1_basic(refname, sha1, 0, &flag);
1627        if (!lock)
1628                return 1;
1629        if (!(flag & REF_ISPACKED) || flag & REF_ISSYMREF) {
1630                /* loose */
1631                const char *path;
1632
1633                if (!(delopt & REF_NODEREF)) {
1634                        i = strlen(lock->lk->filename) - 5; /* .lock */
1635                        lock->lk->filename[i] = 0;
1636                        path = lock->lk->filename;
1637                } else {
1638                        path = git_path("%s", refname);
1639                }
1640                err = unlink_or_warn(path);
1641                if (err && errno != ENOENT)
1642                        ret = 1;
1643
1644                if (!(delopt & REF_NODEREF))
1645                        lock->lk->filename[i] = '.';
1646        }
1647        /* removing the loose one could have resurrected an earlier
1648         * packed one.  Also, if it was not loose we need to repack
1649         * without it.
1650         */
1651        ret |= repack_without_ref(refname);
1652
1653        unlink_or_warn(git_path("logs/%s", lock->ref_name));
1654        invalidate_ref_cache(NULL);
1655        unlock_ref(lock);
1656        return ret;
1657}
1658
1659/*
1660 * People using contrib's git-new-workdir have .git/logs/refs ->
1661 * /some/other/path/.git/logs/refs, and that may live on another device.
1662 *
1663 * IOW, to avoid cross device rename errors, the temporary renamed log must
1664 * live into logs/refs.
1665 */
1666#define TMP_RENAMED_LOG  "logs/refs/.tmp-renamed-log"
1667
1668int rename_ref(const char *oldrefname, const char *newrefname, const char *logmsg)
1669{
1670        unsigned char sha1[20], orig_sha1[20];
1671        int flag = 0, logmoved = 0;
1672        struct ref_lock *lock;
1673        struct stat loginfo;
1674        int log = !lstat(git_path("logs/%s", oldrefname), &loginfo);
1675        const char *symref = NULL;
1676        struct ref_cache *refs = get_ref_cache(NULL);
1677
1678        if (log && S_ISLNK(loginfo.st_mode))
1679                return error("reflog for %s is a symlink", oldrefname);
1680
1681        symref = resolve_ref_unsafe(oldrefname, orig_sha1, 1, &flag);
1682        if (flag & REF_ISSYMREF)
1683                return error("refname %s is a symbolic ref, renaming it is not supported",
1684                        oldrefname);
1685        if (!symref)
1686                return error("refname %s not found", oldrefname);
1687
1688        if (!is_refname_available(newrefname, oldrefname, get_packed_refs(refs)))
1689                return 1;
1690
1691        if (!is_refname_available(newrefname, oldrefname, get_loose_refs(refs)))
1692                return 1;
1693
1694        if (log && rename(git_path("logs/%s", oldrefname), git_path(TMP_RENAMED_LOG)))
1695                return error("unable to move logfile logs/%s to "TMP_RENAMED_LOG": %s",
1696                        oldrefname, strerror(errno));
1697
1698        if (delete_ref(oldrefname, orig_sha1, REF_NODEREF)) {
1699                error("unable to delete old %s", oldrefname);
1700                goto rollback;
1701        }
1702
1703        if (!read_ref_full(newrefname, sha1, 1, &flag) &&
1704            delete_ref(newrefname, sha1, REF_NODEREF)) {
1705                if (errno==EISDIR) {
1706                        if (remove_empty_directories(git_path("%s", newrefname))) {
1707                                error("Directory not empty: %s", newrefname);
1708                                goto rollback;
1709                        }
1710                } else {
1711                        error("unable to delete existing %s", newrefname);
1712                        goto rollback;
1713                }
1714        }
1715
1716        if (log && safe_create_leading_directories(git_path("logs/%s", newrefname))) {
1717                error("unable to create directory for %s", newrefname);
1718                goto rollback;
1719        }
1720
1721 retry:
1722        if (log && rename(git_path(TMP_RENAMED_LOG), git_path("logs/%s", newrefname))) {
1723                if (errno==EISDIR || errno==ENOTDIR) {
1724                        /*
1725                         * rename(a, b) when b is an existing
1726                         * directory ought to result in ISDIR, but
1727                         * Solaris 5.8 gives ENOTDIR.  Sheesh.
1728                         */
1729                        if (remove_empty_directories(git_path("logs/%s", newrefname))) {
1730                                error("Directory not empty: logs/%s", newrefname);
1731                                goto rollback;
1732                        }
1733                        goto retry;
1734                } else {
1735                        error("unable to move logfile "TMP_RENAMED_LOG" to logs/%s: %s",
1736                                newrefname, strerror(errno));
1737                        goto rollback;
1738                }
1739        }
1740        logmoved = log;
1741
1742        lock = lock_ref_sha1_basic(newrefname, NULL, 0, NULL);
1743        if (!lock) {
1744                error("unable to lock %s for update", newrefname);
1745                goto rollback;
1746        }
1747        lock->force_write = 1;
1748        hashcpy(lock->old_sha1, orig_sha1);
1749        if (write_ref_sha1(lock, orig_sha1, logmsg)) {
1750                error("unable to write current sha1 into %s", newrefname);
1751                goto rollback;
1752        }
1753
1754        return 0;
1755
1756 rollback:
1757        lock = lock_ref_sha1_basic(oldrefname, NULL, 0, NULL);
1758        if (!lock) {
1759                error("unable to lock %s for rollback", oldrefname);
1760                goto rollbacklog;
1761        }
1762
1763        lock->force_write = 1;
1764        flag = log_all_ref_updates;
1765        log_all_ref_updates = 0;
1766        if (write_ref_sha1(lock, orig_sha1, NULL))
1767                error("unable to write current sha1 into %s", oldrefname);
1768        log_all_ref_updates = flag;
1769
1770 rollbacklog:
1771        if (logmoved && rename(git_path("logs/%s", newrefname), git_path("logs/%s", oldrefname)))
1772                error("unable to restore logfile %s from %s: %s",
1773                        oldrefname, newrefname, strerror(errno));
1774        if (!logmoved && log &&
1775            rename(git_path(TMP_RENAMED_LOG), git_path("logs/%s", oldrefname)))
1776                error("unable to restore logfile %s from "TMP_RENAMED_LOG": %s",
1777                        oldrefname, strerror(errno));
1778
1779        return 1;
1780}
1781
1782int close_ref(struct ref_lock *lock)
1783{
1784        if (close_lock_file(lock->lk))
1785                return -1;
1786        lock->lock_fd = -1;
1787        return 0;
1788}
1789
1790int commit_ref(struct ref_lock *lock)
1791{
1792        if (commit_lock_file(lock->lk))
1793                return -1;
1794        lock->lock_fd = -1;
1795        return 0;
1796}
1797
1798void unlock_ref(struct ref_lock *lock)
1799{
1800        /* Do not free lock->lk -- atexit() still looks at them */
1801        if (lock->lk)
1802                rollback_lock_file(lock->lk);
1803        free(lock->ref_name);
1804        free(lock->orig_ref_name);
1805        free(lock);
1806}
1807
1808/*
1809 * copy the reflog message msg to buf, which has been allocated sufficiently
1810 * large, while cleaning up the whitespaces.  Especially, convert LF to space,
1811 * because reflog file is one line per entry.
1812 */
1813static int copy_msg(char *buf, const char *msg)
1814{
1815        char *cp = buf;
1816        char c;
1817        int wasspace = 1;
1818
1819        *cp++ = '\t';
1820        while ((c = *msg++)) {
1821                if (wasspace && isspace(c))
1822                        continue;
1823                wasspace = isspace(c);
1824                if (wasspace)
1825                        c = ' ';
1826                *cp++ = c;
1827        }
1828        while (buf < cp && isspace(cp[-1]))
1829                cp--;
1830        *cp++ = '\n';
1831        return cp - buf;
1832}
1833
1834int log_ref_setup(const char *refname, char *logfile, int bufsize)
1835{
1836        int logfd, oflags = O_APPEND | O_WRONLY;
1837
1838        git_snpath(logfile, bufsize, "logs/%s", refname);
1839        if (log_all_ref_updates &&
1840            (!prefixcmp(refname, "refs/heads/") ||
1841             !prefixcmp(refname, "refs/remotes/") ||
1842             !prefixcmp(refname, "refs/notes/") ||
1843             !strcmp(refname, "HEAD"))) {
1844                if (safe_create_leading_directories(logfile) < 0)
1845                        return error("unable to create directory for %s",
1846                                     logfile);
1847                oflags |= O_CREAT;
1848        }
1849
1850        logfd = open(logfile, oflags, 0666);
1851        if (logfd < 0) {
1852                if (!(oflags & O_CREAT) && errno == ENOENT)
1853                        return 0;
1854
1855                if ((oflags & O_CREAT) && errno == EISDIR) {
1856                        if (remove_empty_directories(logfile)) {
1857                                return error("There are still logs under '%s'",
1858                                             logfile);
1859                        }
1860                        logfd = open(logfile, oflags, 0666);
1861                }
1862
1863                if (logfd < 0)
1864                        return error("Unable to append to %s: %s",
1865                                     logfile, strerror(errno));
1866        }
1867
1868        adjust_shared_perm(logfile);
1869        close(logfd);
1870        return 0;
1871}
1872
1873static int log_ref_write(const char *refname, const unsigned char *old_sha1,
1874                         const unsigned char *new_sha1, const char *msg)
1875{
1876        int logfd, result, written, oflags = O_APPEND | O_WRONLY;
1877        unsigned maxlen, len;
1878        int msglen;
1879        char log_file[PATH_MAX];
1880        char *logrec;
1881        const char *committer;
1882
1883        if (log_all_ref_updates < 0)
1884                log_all_ref_updates = !is_bare_repository();
1885
1886        result = log_ref_setup(refname, log_file, sizeof(log_file));
1887        if (result)
1888                return result;
1889
1890        logfd = open(log_file, oflags);
1891        if (logfd < 0)
1892                return 0;
1893        msglen = msg ? strlen(msg) : 0;
1894        committer = git_committer_info(0);
1895        maxlen = strlen(committer) + msglen + 100;
1896        logrec = xmalloc(maxlen);
1897        len = sprintf(logrec, "%s %s %s\n",
1898                      sha1_to_hex(old_sha1),
1899                      sha1_to_hex(new_sha1),
1900                      committer);
1901        if (msglen)
1902                len += copy_msg(logrec + len - 1, msg) - 1;
1903        written = len <= maxlen ? write_in_full(logfd, logrec, len) : -1;
1904        free(logrec);
1905        if (close(logfd) != 0 || written != len)
1906                return error("Unable to append to %s", log_file);
1907        return 0;
1908}
1909
1910static int is_branch(const char *refname)
1911{
1912        return !strcmp(refname, "HEAD") || !prefixcmp(refname, "refs/heads/");
1913}
1914
1915int write_ref_sha1(struct ref_lock *lock,
1916        const unsigned char *sha1, const char *logmsg)
1917{
1918        static char term = '\n';
1919        struct object *o;
1920
1921        if (!lock)
1922                return -1;
1923        if (!lock->force_write && !hashcmp(lock->old_sha1, sha1)) {
1924                unlock_ref(lock);
1925                return 0;
1926        }
1927        o = parse_object(sha1);
1928        if (!o) {
1929                error("Trying to write ref %s with nonexistent object %s",
1930                        lock->ref_name, sha1_to_hex(sha1));
1931                unlock_ref(lock);
1932                return -1;
1933        }
1934        if (o->type != OBJ_COMMIT && is_branch(lock->ref_name)) {
1935                error("Trying to write non-commit object %s to branch %s",
1936                        sha1_to_hex(sha1), lock->ref_name);
1937                unlock_ref(lock);
1938                return -1;
1939        }
1940        if (write_in_full(lock->lock_fd, sha1_to_hex(sha1), 40) != 40 ||
1941            write_in_full(lock->lock_fd, &term, 1) != 1
1942                || close_ref(lock) < 0) {
1943                error("Couldn't write %s", lock->lk->filename);
1944                unlock_ref(lock);
1945                return -1;
1946        }
1947        clear_loose_ref_cache(get_ref_cache(NULL));
1948        if (log_ref_write(lock->ref_name, lock->old_sha1, sha1, logmsg) < 0 ||
1949            (strcmp(lock->ref_name, lock->orig_ref_name) &&
1950             log_ref_write(lock->orig_ref_name, lock->old_sha1, sha1, logmsg) < 0)) {
1951                unlock_ref(lock);
1952                return -1;
1953        }
1954        if (strcmp(lock->orig_ref_name, "HEAD") != 0) {
1955                /*
1956                 * Special hack: If a branch is updated directly and HEAD
1957                 * points to it (may happen on the remote side of a push
1958                 * for example) then logically the HEAD reflog should be
1959                 * updated too.
1960                 * A generic solution implies reverse symref information,
1961                 * but finding all symrefs pointing to the given branch
1962                 * would be rather costly for this rare event (the direct
1963                 * update of a branch) to be worth it.  So let's cheat and
1964                 * check with HEAD only which should cover 99% of all usage
1965                 * scenarios (even 100% of the default ones).
1966                 */
1967                unsigned char head_sha1[20];
1968                int head_flag;
1969                const char *head_ref;
1970                head_ref = resolve_ref_unsafe("HEAD", head_sha1, 1, &head_flag);
1971                if (head_ref && (head_flag & REF_ISSYMREF) &&
1972                    !strcmp(head_ref, lock->ref_name))
1973                        log_ref_write("HEAD", lock->old_sha1, sha1, logmsg);
1974        }
1975        if (commit_ref(lock)) {
1976                error("Couldn't set %s", lock->ref_name);
1977                unlock_ref(lock);
1978                return -1;
1979        }
1980        unlock_ref(lock);
1981        return 0;
1982}
1983
1984int create_symref(const char *ref_target, const char *refs_heads_master,
1985                  const char *logmsg)
1986{
1987        const char *lockpath;
1988        char ref[1000];
1989        int fd, len, written;
1990        char *git_HEAD = git_pathdup("%s", ref_target);
1991        unsigned char old_sha1[20], new_sha1[20];
1992
1993        if (logmsg && read_ref(ref_target, old_sha1))
1994                hashclr(old_sha1);
1995
1996        if (safe_create_leading_directories(git_HEAD) < 0)
1997                return error("unable to create directory for %s", git_HEAD);
1998
1999#ifndef NO_SYMLINK_HEAD
2000        if (prefer_symlink_refs) {
2001                unlink(git_HEAD);
2002                if (!symlink(refs_heads_master, git_HEAD))
2003                        goto done;
2004                fprintf(stderr, "no symlink - falling back to symbolic ref\n");
2005        }
2006#endif
2007
2008        len = snprintf(ref, sizeof(ref), "ref: %s\n", refs_heads_master);
2009        if (sizeof(ref) <= len) {
2010                error("refname too long: %s", refs_heads_master);
2011                goto error_free_return;
2012        }
2013        lockpath = mkpath("%s.lock", git_HEAD);
2014        fd = open(lockpath, O_CREAT | O_EXCL | O_WRONLY, 0666);
2015        if (fd < 0) {
2016                error("Unable to open %s for writing", lockpath);
2017                goto error_free_return;
2018        }
2019        written = write_in_full(fd, ref, len);
2020        if (close(fd) != 0 || written != len) {
2021                error("Unable to write to %s", lockpath);
2022                goto error_unlink_return;
2023        }
2024        if (rename(lockpath, git_HEAD) < 0) {
2025                error("Unable to create %s", git_HEAD);
2026                goto error_unlink_return;
2027        }
2028        if (adjust_shared_perm(git_HEAD)) {
2029                error("Unable to fix permissions on %s", lockpath);
2030        error_unlink_return:
2031                unlink_or_warn(lockpath);
2032        error_free_return:
2033                free(git_HEAD);
2034                return -1;
2035        }
2036
2037#ifndef NO_SYMLINK_HEAD
2038        done:
2039#endif
2040        if (logmsg && !read_ref(refs_heads_master, new_sha1))
2041                log_ref_write(ref_target, old_sha1, new_sha1, logmsg);
2042
2043        free(git_HEAD);
2044        return 0;
2045}
2046
2047static char *ref_msg(const char *line, const char *endp)
2048{
2049        const char *ep;
2050        line += 82;
2051        ep = memchr(line, '\n', endp - line);
2052        if (!ep)
2053                ep = endp;
2054        return xmemdupz(line, ep - line);
2055}
2056
2057int read_ref_at(const char *refname, unsigned long at_time, int cnt,
2058                unsigned char *sha1, char **msg,
2059                unsigned long *cutoff_time, int *cutoff_tz, int *cutoff_cnt)
2060{
2061        const char *logfile, *logdata, *logend, *rec, *lastgt, *lastrec;
2062        char *tz_c;
2063        int logfd, tz, reccnt = 0;
2064        struct stat st;
2065        unsigned long date;
2066        unsigned char logged_sha1[20];
2067        void *log_mapped;
2068        size_t mapsz;
2069
2070        logfile = git_path("logs/%s", refname);
2071        logfd = open(logfile, O_RDONLY, 0);
2072        if (logfd < 0)
2073                die_errno("Unable to read log '%s'", logfile);
2074        fstat(logfd, &st);
2075        if (!st.st_size)
2076                die("Log %s is empty.", logfile);
2077        mapsz = xsize_t(st.st_size);
2078        log_mapped = xmmap(NULL, mapsz, PROT_READ, MAP_PRIVATE, logfd, 0);
2079        logdata = log_mapped;
2080        close(logfd);
2081
2082        lastrec = NULL;
2083        rec = logend = logdata + st.st_size;
2084        while (logdata < rec) {
2085                reccnt++;
2086                if (logdata < rec && *(rec-1) == '\n')
2087                        rec--;
2088                lastgt = NULL;
2089                while (logdata < rec && *(rec-1) != '\n') {
2090                        rec--;
2091                        if (*rec == '>')
2092                                lastgt = rec;
2093                }
2094                if (!lastgt)
2095                        die("Log %s is corrupt.", logfile);
2096                date = strtoul(lastgt + 1, &tz_c, 10);
2097                if (date <= at_time || cnt == 0) {
2098                        tz = strtoul(tz_c, NULL, 10);
2099                        if (msg)
2100                                *msg = ref_msg(rec, logend);
2101                        if (cutoff_time)
2102                                *cutoff_time = date;
2103                        if (cutoff_tz)
2104                                *cutoff_tz = tz;
2105                        if (cutoff_cnt)
2106                                *cutoff_cnt = reccnt - 1;
2107                        if (lastrec) {
2108                                if (get_sha1_hex(lastrec, logged_sha1))
2109                                        die("Log %s is corrupt.", logfile);
2110                                if (get_sha1_hex(rec + 41, sha1))
2111                                        die("Log %s is corrupt.", logfile);
2112                                if (hashcmp(logged_sha1, sha1)) {
2113                                        warning("Log %s has gap after %s.",
2114                                                logfile, show_date(date, tz, DATE_RFC2822));
2115                                }
2116                        }
2117                        else if (date == at_time) {
2118                                if (get_sha1_hex(rec + 41, sha1))
2119                                        die("Log %s is corrupt.", logfile);
2120                        }
2121                        else {
2122                                if (get_sha1_hex(rec + 41, logged_sha1))
2123                                        die("Log %s is corrupt.", logfile);
2124                                if (hashcmp(logged_sha1, sha1)) {
2125                                        warning("Log %s unexpectedly ended on %s.",
2126                                                logfile, show_date(date, tz, DATE_RFC2822));
2127                                }
2128                        }
2129                        munmap(log_mapped, mapsz);
2130                        return 0;
2131                }
2132                lastrec = rec;
2133                if (cnt > 0)
2134                        cnt--;
2135        }
2136
2137        rec = logdata;
2138        while (rec < logend && *rec != '>' && *rec != '\n')
2139                rec++;
2140        if (rec == logend || *rec == '\n')
2141                die("Log %s is corrupt.", logfile);
2142        date = strtoul(rec + 1, &tz_c, 10);
2143        tz = strtoul(tz_c, NULL, 10);
2144        if (get_sha1_hex(logdata, sha1))
2145                die("Log %s is corrupt.", logfile);
2146        if (is_null_sha1(sha1)) {
2147                if (get_sha1_hex(logdata + 41, sha1))
2148                        die("Log %s is corrupt.", logfile);
2149        }
2150        if (msg)
2151                *msg = ref_msg(logdata, logend);
2152        munmap(log_mapped, mapsz);
2153
2154        if (cutoff_time)
2155                *cutoff_time = date;
2156        if (cutoff_tz)
2157                *cutoff_tz = tz;
2158        if (cutoff_cnt)
2159                *cutoff_cnt = reccnt;
2160        return 1;
2161}
2162
2163int for_each_recent_reflog_ent(const char *refname, each_reflog_ent_fn fn, long ofs, void *cb_data)
2164{
2165        const char *logfile;
2166        FILE *logfp;
2167        struct strbuf sb = STRBUF_INIT;
2168        int ret = 0;
2169
2170        logfile = git_path("logs/%s", refname);
2171        logfp = fopen(logfile, "r");
2172        if (!logfp)
2173                return -1;
2174
2175        if (ofs) {
2176                struct stat statbuf;
2177                if (fstat(fileno(logfp), &statbuf) ||
2178                    statbuf.st_size < ofs ||
2179                    fseek(logfp, -ofs, SEEK_END) ||
2180                    strbuf_getwholeline(&sb, logfp, '\n')) {
2181                        fclose(logfp);
2182                        strbuf_release(&sb);
2183                        return -1;
2184                }
2185        }
2186
2187        while (!strbuf_getwholeline(&sb, logfp, '\n')) {
2188                unsigned char osha1[20], nsha1[20];
2189                char *email_end, *message;
2190                unsigned long timestamp;
2191                int tz;
2192
2193                /* old SP new SP name <email> SP time TAB msg LF */
2194                if (sb.len < 83 || sb.buf[sb.len - 1] != '\n' ||
2195                    get_sha1_hex(sb.buf, osha1) || sb.buf[40] != ' ' ||
2196                    get_sha1_hex(sb.buf + 41, nsha1) || sb.buf[81] != ' ' ||
2197                    !(email_end = strchr(sb.buf + 82, '>')) ||
2198                    email_end[1] != ' ' ||
2199                    !(timestamp = strtoul(email_end + 2, &message, 10)) ||
2200                    !message || message[0] != ' ' ||
2201                    (message[1] != '+' && message[1] != '-') ||
2202                    !isdigit(message[2]) || !isdigit(message[3]) ||
2203                    !isdigit(message[4]) || !isdigit(message[5]))
2204                        continue; /* corrupt? */
2205                email_end[1] = '\0';
2206                tz = strtol(message + 1, NULL, 10);
2207                if (message[6] != '\t')
2208                        message += 6;
2209                else
2210                        message += 7;
2211                ret = fn(osha1, nsha1, sb.buf + 82, timestamp, tz, message,
2212                         cb_data);
2213                if (ret)
2214                        break;
2215        }
2216        fclose(logfp);
2217        strbuf_release(&sb);
2218        return ret;
2219}
2220
2221int for_each_reflog_ent(const char *refname, each_reflog_ent_fn fn, void *cb_data)
2222{
2223        return for_each_recent_reflog_ent(refname, fn, 0, cb_data);
2224}
2225
2226static int do_for_each_reflog(const char *base, each_ref_fn fn, void *cb_data)
2227{
2228        DIR *d = opendir(git_path("logs/%s", base));
2229        int retval = 0;
2230
2231        if (d) {
2232                struct dirent *de;
2233                int baselen = strlen(base);
2234                char *log = xmalloc(baselen + 257);
2235
2236                memcpy(log, base, baselen);
2237                if (baselen && base[baselen-1] != '/')
2238                        log[baselen++] = '/';
2239
2240                while ((de = readdir(d)) != NULL) {
2241                        struct stat st;
2242                        int namelen;
2243
2244                        if (de->d_name[0] == '.')
2245                                continue;
2246                        namelen = strlen(de->d_name);
2247                        if (namelen > 255)
2248                                continue;
2249                        if (has_extension(de->d_name, ".lock"))
2250                                continue;
2251                        memcpy(log + baselen, de->d_name, namelen+1);
2252                        if (stat(git_path("logs/%s", log), &st) < 0)
2253                                continue;
2254                        if (S_ISDIR(st.st_mode)) {
2255                                retval = do_for_each_reflog(log, fn, cb_data);
2256                        } else {
2257                                unsigned char sha1[20];
2258                                if (read_ref_full(log, sha1, 0, NULL))
2259                                        retval = error("bad ref for %s", log);
2260                                else
2261                                        retval = fn(log, sha1, 0, cb_data);
2262                        }
2263                        if (retval)
2264                                break;
2265                }
2266                free(log);
2267                closedir(d);
2268        }
2269        else if (*base)
2270                return errno;
2271        return retval;
2272}
2273
2274int for_each_reflog(each_ref_fn fn, void *cb_data)
2275{
2276        return do_for_each_reflog("", fn, cb_data);
2277}
2278
2279int update_ref(const char *action, const char *refname,
2280                const unsigned char *sha1, const unsigned char *oldval,
2281                int flags, enum action_on_err onerr)
2282{
2283        static struct ref_lock *lock;
2284        lock = lock_any_ref_for_update(refname, oldval, flags);
2285        if (!lock) {
2286                const char *str = "Cannot lock the ref '%s'.";
2287                switch (onerr) {
2288                case MSG_ON_ERR: error(str, refname); break;
2289                case DIE_ON_ERR: die(str, refname); break;
2290                case QUIET_ON_ERR: break;
2291                }
2292                return 1;
2293        }
2294        if (write_ref_sha1(lock, sha1, action) < 0) {
2295                const char *str = "Cannot update the ref '%s'.";
2296                switch (onerr) {
2297                case MSG_ON_ERR: error(str, refname); break;
2298                case DIE_ON_ERR: die(str, refname); break;
2299                case QUIET_ON_ERR: break;
2300                }
2301                return 1;
2302        }
2303        return 0;
2304}
2305
2306struct ref *find_ref_by_name(const struct ref *list, const char *name)
2307{
2308        for ( ; list; list = list->next)
2309                if (!strcmp(list->name, name))
2310                        return (struct ref *)list;
2311        return NULL;
2312}
2313
2314/*
2315 * generate a format suitable for scanf from a ref_rev_parse_rules
2316 * rule, that is replace the "%.*s" spec with a "%s" spec
2317 */
2318static void gen_scanf_fmt(char *scanf_fmt, const char *rule)
2319{
2320        char *spec;
2321
2322        spec = strstr(rule, "%.*s");
2323        if (!spec || strstr(spec + 4, "%.*s"))
2324                die("invalid rule in ref_rev_parse_rules: %s", rule);
2325
2326        /* copy all until spec */
2327        strncpy(scanf_fmt, rule, spec - rule);
2328        scanf_fmt[spec - rule] = '\0';
2329        /* copy new spec */
2330        strcat(scanf_fmt, "%s");
2331        /* copy remaining rule */
2332        strcat(scanf_fmt, spec + 4);
2333
2334        return;
2335}
2336
2337char *shorten_unambiguous_ref(const char *refname, int strict)
2338{
2339        int i;
2340        static char **scanf_fmts;
2341        static int nr_rules;
2342        char *short_name;
2343
2344        /* pre generate scanf formats from ref_rev_parse_rules[] */
2345        if (!nr_rules) {
2346                size_t total_len = 0;
2347
2348                /* the rule list is NULL terminated, count them first */
2349                for (; ref_rev_parse_rules[nr_rules]; nr_rules++)
2350                        /* no +1 because strlen("%s") < strlen("%.*s") */
2351                        total_len += strlen(ref_rev_parse_rules[nr_rules]);
2352
2353                scanf_fmts = xmalloc(nr_rules * sizeof(char *) + total_len);
2354
2355                total_len = 0;
2356                for (i = 0; i < nr_rules; i++) {
2357                        scanf_fmts[i] = (char *)&scanf_fmts[nr_rules]
2358                                        + total_len;
2359                        gen_scanf_fmt(scanf_fmts[i], ref_rev_parse_rules[i]);
2360                        total_len += strlen(ref_rev_parse_rules[i]);
2361                }
2362        }
2363
2364        /* bail out if there are no rules */
2365        if (!nr_rules)
2366                return xstrdup(refname);
2367
2368        /* buffer for scanf result, at most refname must fit */
2369        short_name = xstrdup(refname);
2370
2371        /* skip first rule, it will always match */
2372        for (i = nr_rules - 1; i > 0 ; --i) {
2373                int j;
2374                int rules_to_fail = i;
2375                int short_name_len;
2376
2377                if (1 != sscanf(refname, scanf_fmts[i], short_name))
2378                        continue;
2379
2380                short_name_len = strlen(short_name);
2381
2382                /*
2383                 * in strict mode, all (except the matched one) rules
2384                 * must fail to resolve to a valid non-ambiguous ref
2385                 */
2386                if (strict)
2387                        rules_to_fail = nr_rules;
2388
2389                /*
2390                 * check if the short name resolves to a valid ref,
2391                 * but use only rules prior to the matched one
2392                 */
2393                for (j = 0; j < rules_to_fail; j++) {
2394                        const char *rule = ref_rev_parse_rules[j];
2395                        char refname[PATH_MAX];
2396
2397                        /* skip matched rule */
2398                        if (i == j)
2399                                continue;
2400
2401                        /*
2402                         * the short name is ambiguous, if it resolves
2403                         * (with this previous rule) to a valid ref
2404                         * read_ref() returns 0 on success
2405                         */
2406                        mksnpath(refname, sizeof(refname),
2407                                 rule, short_name_len, short_name);
2408                        if (ref_exists(refname))
2409                                break;
2410                }
2411
2412                /*
2413                 * short name is non-ambiguous if all previous rules
2414                 * haven't resolved to a valid ref
2415                 */
2416                if (j == rules_to_fail)
2417                        return short_name;
2418        }
2419
2420        free(short_name);
2421        return xstrdup(refname);
2422}