1#include "../cache.h" 2#include "../refs.h" 3#include "refs-internal.h" 4#include "../iterator.h" 5#include "../dir-iterator.h" 6#include "../lockfile.h" 7#include "../object.h" 8#include "../dir.h" 9 10struct ref_lock { 11 char *ref_name; 12 struct lock_file *lk; 13 struct object_id old_oid; 14}; 15 16struct ref_entry; 17 18/* 19 * Information used (along with the information in ref_entry) to 20 * describe a single cached reference. This data structure only 21 * occurs embedded in a union in struct ref_entry, and only when 22 * (ref_entry->flag & REF_DIR) is zero. 23 */ 24struct ref_value { 25 /* 26 * The name of the object to which this reference resolves 27 * (which may be a tag object). If REF_ISBROKEN, this is 28 * null. If REF_ISSYMREF, then this is the name of the object 29 * referred to by the last reference in the symlink chain. 30 */ 31 struct object_id oid; 32 33 /* 34 * If REF_KNOWS_PEELED, then this field holds the peeled value 35 * of this reference, or null if the reference is known not to 36 * be peelable. See the documentation for peel_ref() for an 37 * exact definition of "peelable". 38 */ 39 struct object_id peeled; 40}; 41 42struct files_ref_store; 43 44/* 45 * Information used (along with the information in ref_entry) to 46 * describe a level in the hierarchy of references. This data 47 * structure only occurs embedded in a union in struct ref_entry, and 48 * only when (ref_entry.flag & REF_DIR) is set. In that case, 49 * (ref_entry.flag & REF_INCOMPLETE) determines whether the references 50 * in the directory have already been read: 51 * 52 * (ref_entry.flag & REF_INCOMPLETE) unset -- a directory of loose 53 * or packed references, already read. 54 * 55 * (ref_entry.flag & REF_INCOMPLETE) set -- a directory of loose 56 * references that hasn't been read yet (nor has any of its 57 * subdirectories). 58 * 59 * Entries within a directory are stored within a growable array of 60 * pointers to ref_entries (entries, nr, alloc). Entries 0 <= i < 61 * sorted are sorted by their component name in strcmp() order and the 62 * remaining entries are unsorted. 63 * 64 * Loose references are read lazily, one directory at a time. When a 65 * directory of loose references is read, then all of the references 66 * in that directory are stored, and REF_INCOMPLETE stubs are created 67 * for any subdirectories, but the subdirectories themselves are not 68 * read. The reading is triggered by get_ref_dir(). 69 */ 70struct ref_dir { 71 int nr, alloc; 72 73 /* 74 * Entries with index 0 <= i < sorted are sorted by name. New 75 * entries are appended to the list unsorted, and are sorted 76 * only when required; thus we avoid the need to sort the list 77 * after the addition of every reference. 78 */ 79 int sorted; 80 81 /* A pointer to the files_ref_store that contains this ref_dir. */ 82 struct files_ref_store *ref_store; 83 84 struct ref_entry **entries; 85}; 86 87/* 88 * Bit values for ref_entry::flag. REF_ISSYMREF=0x01, 89 * REF_ISPACKED=0x02, REF_ISBROKEN=0x04 and REF_BAD_NAME=0x08 are 90 * public values; see refs.h. 91 */ 92 93/* 94 * The field ref_entry->u.value.peeled of this value entry contains 95 * the correct peeled value for the reference, which might be 96 * null_sha1 if the reference is not a tag or if it is broken. 97 */ 98#define REF_KNOWS_PEELED 0x10 99 100/* ref_entry represents a directory of references */ 101#define REF_DIR 0x20 102 103/* 104 * Entry has not yet been read from disk (used only for REF_DIR 105 * entries representing loose references) 106 */ 107#define REF_INCOMPLETE 0x40 108 109/* 110 * A ref_entry represents either a reference or a "subdirectory" of 111 * references. 112 * 113 * Each directory in the reference namespace is represented by a 114 * ref_entry with (flags & REF_DIR) set and containing a subdir member 115 * that holds the entries in that directory that have been read so 116 * far. If (flags & REF_INCOMPLETE) is set, then the directory and 117 * its subdirectories haven't been read yet. REF_INCOMPLETE is only 118 * used for loose reference directories. 119 * 120 * References are represented by a ref_entry with (flags & REF_DIR) 121 * unset and a value member that describes the reference's value. The 122 * flag member is at the ref_entry level, but it is also needed to 123 * interpret the contents of the value field (in other words, a 124 * ref_value object is not very much use without the enclosing 125 * ref_entry). 126 * 127 * Reference names cannot end with slash and directories' names are 128 * always stored with a trailing slash (except for the top-level 129 * directory, which is always denoted by ""). This has two nice 130 * consequences: (1) when the entries in each subdir are sorted 131 * lexicographically by name (as they usually are), the references in 132 * a whole tree can be generated in lexicographic order by traversing 133 * the tree in left-to-right, depth-first order; (2) the names of 134 * references and subdirectories cannot conflict, and therefore the 135 * presence of an empty subdirectory does not block the creation of a 136 * similarly-named reference. (The fact that reference names with the 137 * same leading components can conflict *with each other* is a 138 * separate issue that is regulated by verify_refname_available().) 139 * 140 * Please note that the name field contains the fully-qualified 141 * reference (or subdirectory) name. Space could be saved by only 142 * storing the relative names. But that would require the full names 143 * to be generated on the fly when iterating in do_for_each_ref(), and 144 * would break callback functions, who have always been able to assume 145 * that the name strings that they are passed will not be freed during 146 * the iteration. 147 */ 148struct ref_entry { 149 unsigned char flag; /* ISSYMREF? ISPACKED? */ 150 union { 151 struct ref_value value; /* if not (flags&REF_DIR) */ 152 struct ref_dir subdir; /* if (flags&REF_DIR) */ 153 } u; 154 /* 155 * The full name of the reference (e.g., "refs/heads/master") 156 * or the full name of the directory with a trailing slash 157 * (e.g., "refs/heads/"): 158 */ 159 char name[FLEX_ARRAY]; 160}; 161 162static void read_loose_refs(const char *dirname, struct ref_dir *dir); 163static int search_ref_dir(struct ref_dir *dir, const char *refname, size_t len); 164static struct ref_entry *create_dir_entry(struct files_ref_store *ref_store, 165 const char *dirname, size_t len, 166 int incomplete); 167static void add_entry_to_dir(struct ref_dir *dir, struct ref_entry *entry); 168static int files_log_ref_write(const char *refname, const unsigned char *old_sha1, 169 const unsigned char *new_sha1, const char *msg, 170 int flags, struct strbuf *err); 171 172static struct ref_dir *get_ref_dir(struct ref_entry *entry) 173{ 174 struct ref_dir *dir; 175 assert(entry->flag & REF_DIR); 176 dir = &entry->u.subdir; 177 if (entry->flag & REF_INCOMPLETE) { 178 read_loose_refs(entry->name, dir); 179 180 /* 181 * Manually add refs/bisect, which, being 182 * per-worktree, might not appear in the directory 183 * listing for refs/ in the main repo. 184 */ 185 if (!strcmp(entry->name, "refs/")) { 186 int pos = search_ref_dir(dir, "refs/bisect/", 12); 187 if (pos < 0) { 188 struct ref_entry *child_entry; 189 child_entry = create_dir_entry(dir->ref_store, 190 "refs/bisect/", 191 12, 1); 192 add_entry_to_dir(dir, child_entry); 193 read_loose_refs("refs/bisect", 194 &child_entry->u.subdir); 195 } 196 } 197 entry->flag &= ~REF_INCOMPLETE; 198 } 199 return dir; 200} 201 202static struct ref_entry *create_ref_entry(const char *refname, 203 const unsigned char *sha1, int flag, 204 int check_name) 205{ 206 struct ref_entry *ref; 207 208 if (check_name && 209 check_refname_format(refname, REFNAME_ALLOW_ONELEVEL)) 210 die("Reference has invalid format: '%s'", refname); 211 FLEX_ALLOC_STR(ref, name, refname); 212 hashcpy(ref->u.value.oid.hash, sha1); 213 oidclr(&ref->u.value.peeled); 214 ref->flag = flag; 215 return ref; 216} 217 218static void clear_ref_dir(struct ref_dir *dir); 219 220static void free_ref_entry(struct ref_entry *entry) 221{ 222 if (entry->flag & REF_DIR) { 223 /* 224 * Do not use get_ref_dir() here, as that might 225 * trigger the reading of loose refs. 226 */ 227 clear_ref_dir(&entry->u.subdir); 228 } 229 free(entry); 230} 231 232/* 233 * Add a ref_entry to the end of dir (unsorted). Entry is always 234 * stored directly in dir; no recursion into subdirectories is 235 * done. 236 */ 237static void add_entry_to_dir(struct ref_dir *dir, struct ref_entry *entry) 238{ 239 ALLOC_GROW(dir->entries, dir->nr + 1, dir->alloc); 240 dir->entries[dir->nr++] = entry; 241 /* optimize for the case that entries are added in order */ 242 if (dir->nr == 1 || 243 (dir->nr == dir->sorted + 1 && 244 strcmp(dir->entries[dir->nr - 2]->name, 245 dir->entries[dir->nr - 1]->name) < 0)) 246 dir->sorted = dir->nr; 247} 248 249/* 250 * Clear and free all entries in dir, recursively. 251 */ 252static void clear_ref_dir(struct ref_dir *dir) 253{ 254 int i; 255 for (i = 0; i < dir->nr; i++) 256 free_ref_entry(dir->entries[i]); 257 free(dir->entries); 258 dir->sorted = dir->nr = dir->alloc = 0; 259 dir->entries = NULL; 260} 261 262/* 263 * Create a struct ref_entry object for the specified dirname. 264 * dirname is the name of the directory with a trailing slash (e.g., 265 * "refs/heads/") or "" for the top-level directory. 266 */ 267static struct ref_entry *create_dir_entry(struct files_ref_store *ref_store, 268 const char *dirname, size_t len, 269 int incomplete) 270{ 271 struct ref_entry *direntry; 272 FLEX_ALLOC_MEM(direntry, name, dirname, len); 273 direntry->u.subdir.ref_store = ref_store; 274 direntry->flag = REF_DIR | (incomplete ? REF_INCOMPLETE : 0); 275 return direntry; 276} 277 278static int ref_entry_cmp(const void *a, const void *b) 279{ 280 struct ref_entry *one = *(struct ref_entry **)a; 281 struct ref_entry *two = *(struct ref_entry **)b; 282 return strcmp(one->name, two->name); 283} 284 285static void sort_ref_dir(struct ref_dir *dir); 286 287struct string_slice { 288 size_t len; 289 const char *str; 290}; 291 292static int ref_entry_cmp_sslice(const void *key_, const void *ent_) 293{ 294 const struct string_slice *key = key_; 295 const struct ref_entry *ent = *(const struct ref_entry * const *)ent_; 296 int cmp = strncmp(key->str, ent->name, key->len); 297 if (cmp) 298 return cmp; 299 return '\0' - (unsigned char)ent->name[key->len]; 300} 301 302/* 303 * Return the index of the entry with the given refname from the 304 * ref_dir (non-recursively), sorting dir if necessary. Return -1 if 305 * no such entry is found. dir must already be complete. 306 */ 307static int search_ref_dir(struct ref_dir *dir, const char *refname, size_t len) 308{ 309 struct ref_entry **r; 310 struct string_slice key; 311 312 if (refname == NULL || !dir->nr) 313 return -1; 314 315 sort_ref_dir(dir); 316 key.len = len; 317 key.str = refname; 318 r = bsearch(&key, dir->entries, dir->nr, sizeof(*dir->entries), 319 ref_entry_cmp_sslice); 320 321 if (r == NULL) 322 return -1; 323 324 return r - dir->entries; 325} 326 327/* 328 * Search for a directory entry directly within dir (without 329 * recursing). Sort dir if necessary. subdirname must be a directory 330 * name (i.e., end in '/'). If mkdir is set, then create the 331 * directory if it is missing; otherwise, return NULL if the desired 332 * directory cannot be found. dir must already be complete. 333 */ 334static struct ref_dir *search_for_subdir(struct ref_dir *dir, 335 const char *subdirname, size_t len, 336 int mkdir) 337{ 338 int entry_index = search_ref_dir(dir, subdirname, len); 339 struct ref_entry *entry; 340 if (entry_index == -1) { 341 if (!mkdir) 342 return NULL; 343 /* 344 * Since dir is complete, the absence of a subdir 345 * means that the subdir really doesn't exist; 346 * therefore, create an empty record for it but mark 347 * the record complete. 348 */ 349 entry = create_dir_entry(dir->ref_store, subdirname, len, 0); 350 add_entry_to_dir(dir, entry); 351 } else { 352 entry = dir->entries[entry_index]; 353 } 354 return get_ref_dir(entry); 355} 356 357/* 358 * If refname is a reference name, find the ref_dir within the dir 359 * tree that should hold refname. If refname is a directory name 360 * (i.e., ends in '/'), then return that ref_dir itself. dir must 361 * represent the top-level directory and must already be complete. 362 * Sort ref_dirs and recurse into subdirectories as necessary. If 363 * mkdir is set, then create any missing directories; otherwise, 364 * return NULL if the desired directory cannot be found. 365 */ 366static struct ref_dir *find_containing_dir(struct ref_dir *dir, 367 const char *refname, int mkdir) 368{ 369 const char *slash; 370 for (slash = strchr(refname, '/'); slash; slash = strchr(slash + 1, '/')) { 371 size_t dirnamelen = slash - refname + 1; 372 struct ref_dir *subdir; 373 subdir = search_for_subdir(dir, refname, dirnamelen, mkdir); 374 if (!subdir) { 375 dir = NULL; 376 break; 377 } 378 dir = subdir; 379 } 380 381 return dir; 382} 383 384/* 385 * Find the value entry with the given name in dir, sorting ref_dirs 386 * and recursing into subdirectories as necessary. If the name is not 387 * found or it corresponds to a directory entry, return NULL. 388 */ 389static struct ref_entry *find_ref(struct ref_dir *dir, const char *refname) 390{ 391 int entry_index; 392 struct ref_entry *entry; 393 dir = find_containing_dir(dir, refname, 0); 394 if (!dir) 395 return NULL; 396 entry_index = search_ref_dir(dir, refname, strlen(refname)); 397 if (entry_index == -1) 398 return NULL; 399 entry = dir->entries[entry_index]; 400 return (entry->flag & REF_DIR) ? NULL : entry; 401} 402 403/* 404 * Remove the entry with the given name from dir, recursing into 405 * subdirectories as necessary. If refname is the name of a directory 406 * (i.e., ends with '/'), then remove the directory and its contents. 407 * If the removal was successful, return the number of entries 408 * remaining in the directory entry that contained the deleted entry. 409 * If the name was not found, return -1. Please note that this 410 * function only deletes the entry from the cache; it does not delete 411 * it from the filesystem or ensure that other cache entries (which 412 * might be symbolic references to the removed entry) are updated. 413 * Nor does it remove any containing dir entries that might be made 414 * empty by the removal. dir must represent the top-level directory 415 * and must already be complete. 416 */ 417static int remove_entry(struct ref_dir *dir, const char *refname) 418{ 419 int refname_len = strlen(refname); 420 int entry_index; 421 struct ref_entry *entry; 422 int is_dir = refname[refname_len - 1] == '/'; 423 if (is_dir) { 424 /* 425 * refname represents a reference directory. Remove 426 * the trailing slash; otherwise we will get the 427 * directory *representing* refname rather than the 428 * one *containing* it. 429 */ 430 char *dirname = xmemdupz(refname, refname_len - 1); 431 dir = find_containing_dir(dir, dirname, 0); 432 free(dirname); 433 } else { 434 dir = find_containing_dir(dir, refname, 0); 435 } 436 if (!dir) 437 return -1; 438 entry_index = search_ref_dir(dir, refname, refname_len); 439 if (entry_index == -1) 440 return -1; 441 entry = dir->entries[entry_index]; 442 443 memmove(&dir->entries[entry_index], 444 &dir->entries[entry_index + 1], 445 (dir->nr - entry_index - 1) * sizeof(*dir->entries) 446 ); 447 dir->nr--; 448 if (dir->sorted > entry_index) 449 dir->sorted--; 450 free_ref_entry(entry); 451 return dir->nr; 452} 453 454/* 455 * Add a ref_entry to the ref_dir (unsorted), recursing into 456 * subdirectories as necessary. dir must represent the top-level 457 * directory. Return 0 on success. 458 */ 459static int add_ref(struct ref_dir *dir, struct ref_entry *ref) 460{ 461 dir = find_containing_dir(dir, ref->name, 1); 462 if (!dir) 463 return -1; 464 add_entry_to_dir(dir, ref); 465 return 0; 466} 467 468/* 469 * Emit a warning and return true iff ref1 and ref2 have the same name 470 * and the same sha1. Die if they have the same name but different 471 * sha1s. 472 */ 473static int is_dup_ref(const struct ref_entry *ref1, const struct ref_entry *ref2) 474{ 475 if (strcmp(ref1->name, ref2->name)) 476 return 0; 477 478 /* Duplicate name; make sure that they don't conflict: */ 479 480 if ((ref1->flag & REF_DIR) || (ref2->flag & REF_DIR)) 481 /* This is impossible by construction */ 482 die("Reference directory conflict: %s", ref1->name); 483 484 if (oidcmp(&ref1->u.value.oid, &ref2->u.value.oid)) 485 die("Duplicated ref, and SHA1s don't match: %s", ref1->name); 486 487 warning("Duplicated ref: %s", ref1->name); 488 return 1; 489} 490 491/* 492 * Sort the entries in dir non-recursively (if they are not already 493 * sorted) and remove any duplicate entries. 494 */ 495static void sort_ref_dir(struct ref_dir *dir) 496{ 497 int i, j; 498 struct ref_entry *last = NULL; 499 500 /* 501 * This check also prevents passing a zero-length array to qsort(), 502 * which is a problem on some platforms. 503 */ 504 if (dir->sorted == dir->nr) 505 return; 506 507 QSORT(dir->entries, dir->nr, ref_entry_cmp); 508 509 /* Remove any duplicates: */ 510 for (i = 0, j = 0; j < dir->nr; j++) { 511 struct ref_entry *entry = dir->entries[j]; 512 if (last && is_dup_ref(last, entry)) 513 free_ref_entry(entry); 514 else 515 last = dir->entries[i++] = entry; 516 } 517 dir->sorted = dir->nr = i; 518} 519 520/* 521 * Return true if refname, which has the specified oid and flags, can 522 * be resolved to an object in the database. If the referred-to object 523 * does not exist, emit a warning and return false. 524 */ 525static int ref_resolves_to_object(const char *refname, 526 const struct object_id *oid, 527 unsigned int flags) 528{ 529 if (flags & REF_ISBROKEN) 530 return 0; 531 if (!has_sha1_file(oid->hash)) { 532 error("%s does not point to a valid object!", refname); 533 return 0; 534 } 535 return 1; 536} 537 538/* 539 * Return true if the reference described by entry can be resolved to 540 * an object in the database; otherwise, emit a warning and return 541 * false. 542 */ 543static int entry_resolves_to_object(struct ref_entry *entry) 544{ 545 return ref_resolves_to_object(entry->name, 546 &entry->u.value.oid, entry->flag); 547} 548 549typedef int each_ref_entry_fn(struct ref_entry *entry, void *cb_data); 550 551/* 552 * Call fn for each reference in dir that has index in the range 553 * offset <= index < dir->nr. Recurse into subdirectories that are in 554 * that index range, sorting them before iterating. This function 555 * does not sort dir itself; it should be sorted beforehand. fn is 556 * called for all references, including broken ones. 557 */ 558static int do_for_each_entry_in_dir(struct ref_dir *dir, int offset, 559 each_ref_entry_fn fn, void *cb_data) 560{ 561 int i; 562 assert(dir->sorted == dir->nr); 563 for (i = offset; i < dir->nr; i++) { 564 struct ref_entry *entry = dir->entries[i]; 565 int retval; 566 if (entry->flag & REF_DIR) { 567 struct ref_dir *subdir = get_ref_dir(entry); 568 sort_ref_dir(subdir); 569 retval = do_for_each_entry_in_dir(subdir, 0, fn, cb_data); 570 } else { 571 retval = fn(entry, cb_data); 572 } 573 if (retval) 574 return retval; 575 } 576 return 0; 577} 578 579/* 580 * Load all of the refs from the dir into our in-memory cache. The hard work 581 * of loading loose refs is done by get_ref_dir(), so we just need to recurse 582 * through all of the sub-directories. We do not even need to care about 583 * sorting, as traversal order does not matter to us. 584 */ 585static void prime_ref_dir(struct ref_dir *dir) 586{ 587 int i; 588 for (i = 0; i < dir->nr; i++) { 589 struct ref_entry *entry = dir->entries[i]; 590 if (entry->flag & REF_DIR) 591 prime_ref_dir(get_ref_dir(entry)); 592 } 593} 594 595/* 596 * A level in the reference hierarchy that is currently being iterated 597 * through. 598 */ 599struct cache_ref_iterator_level { 600 /* 601 * The ref_dir being iterated over at this level. The ref_dir 602 * is sorted before being stored here. 603 */ 604 struct ref_dir *dir; 605 606 /* 607 * The index of the current entry within dir (which might 608 * itself be a directory). If index == -1, then the iteration 609 * hasn't yet begun. If index == dir->nr, then the iteration 610 * through this level is over. 611 */ 612 int index; 613}; 614 615/* 616 * Represent an iteration through a ref_dir in the memory cache. The 617 * iteration recurses through subdirectories. 618 */ 619struct cache_ref_iterator { 620 struct ref_iterator base; 621 622 /* 623 * The number of levels currently on the stack. This is always 624 * at least 1, because when it becomes zero the iteration is 625 * ended and this struct is freed. 626 */ 627 size_t levels_nr; 628 629 /* The number of levels that have been allocated on the stack */ 630 size_t levels_alloc; 631 632 /* 633 * A stack of levels. levels[0] is the uppermost level that is 634 * being iterated over in this iteration. (This is not 635 * necessary the top level in the references hierarchy. If we 636 * are iterating through a subtree, then levels[0] will hold 637 * the ref_dir for that subtree, and subsequent levels will go 638 * on from there.) 639 */ 640 struct cache_ref_iterator_level *levels; 641}; 642 643static int cache_ref_iterator_advance(struct ref_iterator *ref_iterator) 644{ 645 struct cache_ref_iterator *iter = 646 (struct cache_ref_iterator *)ref_iterator; 647 648 while (1) { 649 struct cache_ref_iterator_level *level = 650 &iter->levels[iter->levels_nr - 1]; 651 struct ref_dir *dir = level->dir; 652 struct ref_entry *entry; 653 654 if (level->index == -1) 655 sort_ref_dir(dir); 656 657 if (++level->index == level->dir->nr) { 658 /* This level is exhausted; pop up a level */ 659 if (--iter->levels_nr == 0) 660 return ref_iterator_abort(ref_iterator); 661 662 continue; 663 } 664 665 entry = dir->entries[level->index]; 666 667 if (entry->flag & REF_DIR) { 668 /* push down a level */ 669 ALLOC_GROW(iter->levels, iter->levels_nr + 1, 670 iter->levels_alloc); 671 672 level = &iter->levels[iter->levels_nr++]; 673 level->dir = get_ref_dir(entry); 674 level->index = -1; 675 } else { 676 iter->base.refname = entry->name; 677 iter->base.oid = &entry->u.value.oid; 678 iter->base.flags = entry->flag; 679 return ITER_OK; 680 } 681 } 682} 683 684static enum peel_status peel_entry(struct ref_entry *entry, int repeel); 685 686static int cache_ref_iterator_peel(struct ref_iterator *ref_iterator, 687 struct object_id *peeled) 688{ 689 struct cache_ref_iterator *iter = 690 (struct cache_ref_iterator *)ref_iterator; 691 struct cache_ref_iterator_level *level; 692 struct ref_entry *entry; 693 694 level = &iter->levels[iter->levels_nr - 1]; 695 696 if (level->index == -1) 697 die("BUG: peel called before advance for cache iterator"); 698 699 entry = level->dir->entries[level->index]; 700 701 if (peel_entry(entry, 0)) 702 return -1; 703 oidcpy(peeled, &entry->u.value.peeled); 704 return 0; 705} 706 707static int cache_ref_iterator_abort(struct ref_iterator *ref_iterator) 708{ 709 struct cache_ref_iterator *iter = 710 (struct cache_ref_iterator *)ref_iterator; 711 712 free(iter->levels); 713 base_ref_iterator_free(ref_iterator); 714 return ITER_DONE; 715} 716 717static struct ref_iterator_vtable cache_ref_iterator_vtable = { 718 cache_ref_iterator_advance, 719 cache_ref_iterator_peel, 720 cache_ref_iterator_abort 721}; 722 723static struct ref_iterator *cache_ref_iterator_begin(struct ref_dir *dir) 724{ 725 struct cache_ref_iterator *iter; 726 struct ref_iterator *ref_iterator; 727 struct cache_ref_iterator_level *level; 728 729 iter = xcalloc(1, sizeof(*iter)); 730 ref_iterator = &iter->base; 731 base_ref_iterator_init(ref_iterator, &cache_ref_iterator_vtable); 732 ALLOC_GROW(iter->levels, 10, iter->levels_alloc); 733 734 iter->levels_nr = 1; 735 level = &iter->levels[0]; 736 level->index = -1; 737 level->dir = dir; 738 739 return ref_iterator; 740} 741 742struct nonmatching_ref_data { 743 const struct string_list *skip; 744 const char *conflicting_refname; 745}; 746 747static int nonmatching_ref_fn(struct ref_entry *entry, void *vdata) 748{ 749 struct nonmatching_ref_data *data = vdata; 750 751 if (data->skip && string_list_has_string(data->skip, entry->name)) 752 return 0; 753 754 data->conflicting_refname = entry->name; 755 return 1; 756} 757 758/* 759 * Return 0 if a reference named refname could be created without 760 * conflicting with the name of an existing reference in dir. 761 * See verify_refname_available for more information. 762 */ 763static int verify_refname_available_dir(const char *refname, 764 const struct string_list *extras, 765 const struct string_list *skip, 766 struct ref_dir *dir, 767 struct strbuf *err) 768{ 769 const char *slash; 770 const char *extra_refname; 771 int pos; 772 struct strbuf dirname = STRBUF_INIT; 773 int ret = -1; 774 775 /* 776 * For the sake of comments in this function, suppose that 777 * refname is "refs/foo/bar". 778 */ 779 780 assert(err); 781 782 strbuf_grow(&dirname, strlen(refname) + 1); 783 for (slash = strchr(refname, '/'); slash; slash = strchr(slash + 1, '/')) { 784 /* Expand dirname to the new prefix, not including the trailing slash: */ 785 strbuf_add(&dirname, refname + dirname.len, slash - refname - dirname.len); 786 787 /* 788 * We are still at a leading dir of the refname (e.g., 789 * "refs/foo"; if there is a reference with that name, 790 * it is a conflict, *unless* it is in skip. 791 */ 792 if (dir) { 793 pos = search_ref_dir(dir, dirname.buf, dirname.len); 794 if (pos >= 0 && 795 (!skip || !string_list_has_string(skip, dirname.buf))) { 796 /* 797 * We found a reference whose name is 798 * a proper prefix of refname; e.g., 799 * "refs/foo", and is not in skip. 800 */ 801 strbuf_addf(err, "'%s' exists; cannot create '%s'", 802 dirname.buf, refname); 803 goto cleanup; 804 } 805 } 806 807 if (extras && string_list_has_string(extras, dirname.buf) && 808 (!skip || !string_list_has_string(skip, dirname.buf))) { 809 strbuf_addf(err, "cannot process '%s' and '%s' at the same time", 810 refname, dirname.buf); 811 goto cleanup; 812 } 813 814 /* 815 * Otherwise, we can try to continue our search with 816 * the next component. So try to look up the 817 * directory, e.g., "refs/foo/". If we come up empty, 818 * we know there is nothing under this whole prefix, 819 * but even in that case we still have to continue the 820 * search for conflicts with extras. 821 */ 822 strbuf_addch(&dirname, '/'); 823 if (dir) { 824 pos = search_ref_dir(dir, dirname.buf, dirname.len); 825 if (pos < 0) { 826 /* 827 * There was no directory "refs/foo/", 828 * so there is nothing under this 829 * whole prefix. So there is no need 830 * to continue looking for conflicting 831 * references. But we need to continue 832 * looking for conflicting extras. 833 */ 834 dir = NULL; 835 } else { 836 dir = get_ref_dir(dir->entries[pos]); 837 } 838 } 839 } 840 841 /* 842 * We are at the leaf of our refname (e.g., "refs/foo/bar"). 843 * There is no point in searching for a reference with that 844 * name, because a refname isn't considered to conflict with 845 * itself. But we still need to check for references whose 846 * names are in the "refs/foo/bar/" namespace, because they 847 * *do* conflict. 848 */ 849 strbuf_addstr(&dirname, refname + dirname.len); 850 strbuf_addch(&dirname, '/'); 851 852 if (dir) { 853 pos = search_ref_dir(dir, dirname.buf, dirname.len); 854 855 if (pos >= 0) { 856 /* 857 * We found a directory named "$refname/" 858 * (e.g., "refs/foo/bar/"). It is a problem 859 * iff it contains any ref that is not in 860 * "skip". 861 */ 862 struct nonmatching_ref_data data; 863 864 data.skip = skip; 865 data.conflicting_refname = NULL; 866 dir = get_ref_dir(dir->entries[pos]); 867 sort_ref_dir(dir); 868 if (do_for_each_entry_in_dir(dir, 0, nonmatching_ref_fn, &data)) { 869 strbuf_addf(err, "'%s' exists; cannot create '%s'", 870 data.conflicting_refname, refname); 871 goto cleanup; 872 } 873 } 874 } 875 876 extra_refname = find_descendant_ref(dirname.buf, extras, skip); 877 if (extra_refname) 878 strbuf_addf(err, "cannot process '%s' and '%s' at the same time", 879 refname, extra_refname); 880 else 881 ret = 0; 882 883cleanup: 884 strbuf_release(&dirname); 885 return ret; 886} 887 888struct packed_ref_cache { 889 struct ref_entry *root; 890 891 /* 892 * Count of references to the data structure in this instance, 893 * including the pointer from files_ref_store::packed if any. 894 * The data will not be freed as long as the reference count 895 * is nonzero. 896 */ 897 unsigned int referrers; 898 899 /* 900 * Iff the packed-refs file associated with this instance is 901 * currently locked for writing, this points at the associated 902 * lock (which is owned by somebody else). The referrer count 903 * is also incremented when the file is locked and decremented 904 * when it is unlocked. 905 */ 906 struct lock_file *lock; 907 908 /* The metadata from when this packed-refs cache was read */ 909 struct stat_validity validity; 910}; 911 912/* 913 * Future: need to be in "struct repository" 914 * when doing a full libification. 915 */ 916struct files_ref_store { 917 struct ref_store base; 918 919 /* 920 * The name of the submodule represented by this object, or 921 * NULL if it represents the main repository's reference 922 * store: 923 */ 924 const char *submodule; 925 926 char *packed_refs_path; 927 928 struct ref_entry *loose; 929 struct packed_ref_cache *packed; 930}; 931 932/* Lock used for the main packed-refs file: */ 933static struct lock_file packlock; 934 935/* 936 * Increment the reference count of *packed_refs. 937 */ 938static void acquire_packed_ref_cache(struct packed_ref_cache *packed_refs) 939{ 940 packed_refs->referrers++; 941} 942 943/* 944 * Decrease the reference count of *packed_refs. If it goes to zero, 945 * free *packed_refs and return true; otherwise return false. 946 */ 947static int release_packed_ref_cache(struct packed_ref_cache *packed_refs) 948{ 949 if (!--packed_refs->referrers) { 950 free_ref_entry(packed_refs->root); 951 stat_validity_clear(&packed_refs->validity); 952 free(packed_refs); 953 return 1; 954 } else { 955 return 0; 956 } 957} 958 959static void clear_packed_ref_cache(struct files_ref_store *refs) 960{ 961 if (refs->packed) { 962 struct packed_ref_cache *packed_refs = refs->packed; 963 964 if (packed_refs->lock) 965 die("internal error: packed-ref cache cleared while locked"); 966 refs->packed = NULL; 967 release_packed_ref_cache(packed_refs); 968 } 969} 970 971static void clear_loose_ref_cache(struct files_ref_store *refs) 972{ 973 if (refs->loose) { 974 free_ref_entry(refs->loose); 975 refs->loose = NULL; 976 } 977} 978 979/* 980 * Create a new submodule ref cache and add it to the internal 981 * set of caches. 982 */ 983static struct ref_store *files_ref_store_create(const char *submodule) 984{ 985 struct files_ref_store *refs = xcalloc(1, sizeof(*refs)); 986 struct ref_store *ref_store = (struct ref_store *)refs; 987 988 base_ref_store_init(ref_store, &refs_be_files); 989 990 if (submodule) { 991 refs->submodule = xstrdup(submodule); 992 refs->packed_refs_path = git_pathdup_submodule( 993 refs->submodule, "packed-refs"); 994 return ref_store; 995 } 996 997 refs->packed_refs_path = git_pathdup("packed-refs"); 998 999 return ref_store;1000}10011002/*1003 * Die if refs is for a submodule (i.e., not for the main repository).1004 * caller is used in any necessary error messages.1005 */1006static void files_assert_main_repository(struct files_ref_store *refs,1007 const char *caller)1008{1009 if (refs->submodule)1010 die("BUG: %s called for a submodule", caller);1011}10121013/*1014 * Downcast ref_store to files_ref_store. Die if ref_store is not a1015 * files_ref_store. If submodule_allowed is not true, then also die if1016 * files_ref_store is for a submodule (i.e., not for the main1017 * repository). caller is used in any necessary error messages.1018 */1019static struct files_ref_store *files_downcast(1020 struct ref_store *ref_store, int submodule_allowed,1021 const char *caller)1022{1023 struct files_ref_store *refs;10241025 if (ref_store->be != &refs_be_files)1026 die("BUG: ref_store is type \"%s\" not \"files\" in %s",1027 ref_store->be->name, caller);10281029 refs = (struct files_ref_store *)ref_store;10301031 if (!submodule_allowed)1032 files_assert_main_repository(refs, caller);10331034 return refs;1035}10361037/* The length of a peeled reference line in packed-refs, including EOL: */1038#define PEELED_LINE_LENGTH 4210391040/*1041 * The packed-refs header line that we write out. Perhaps other1042 * traits will be added later. The trailing space is required.1043 */1044static const char PACKED_REFS_HEADER[] =1045 "# pack-refs with: peeled fully-peeled \n";10461047/*1048 * Parse one line from a packed-refs file. Write the SHA1 to sha1.1049 * Return a pointer to the refname within the line (null-terminated),1050 * or NULL if there was a problem.1051 */1052static const char *parse_ref_line(struct strbuf *line, unsigned char *sha1)1053{1054 const char *ref;10551056 /*1057 * 42: the answer to everything.1058 *1059 * In this case, it happens to be the answer to1060 * 40 (length of sha1 hex representation)1061 * +1 (space in between hex and name)1062 * +1 (newline at the end of the line)1063 */1064 if (line->len <= 42)1065 return NULL;10661067 if (get_sha1_hex(line->buf, sha1) < 0)1068 return NULL;1069 if (!isspace(line->buf[40]))1070 return NULL;10711072 ref = line->buf + 41;1073 if (isspace(*ref))1074 return NULL;10751076 if (line->buf[line->len - 1] != '\n')1077 return NULL;1078 line->buf[--line->len] = 0;10791080 return ref;1081}10821083/*1084 * Read f, which is a packed-refs file, into dir.1085 *1086 * A comment line of the form "# pack-refs with: " may contain zero or1087 * more traits. We interpret the traits as follows:1088 *1089 * No traits:1090 *1091 * Probably no references are peeled. But if the file contains a1092 * peeled value for a reference, we will use it.1093 *1094 * peeled:1095 *1096 * References under "refs/tags/", if they *can* be peeled, *are*1097 * peeled in this file. References outside of "refs/tags/" are1098 * probably not peeled even if they could have been, but if we find1099 * a peeled value for such a reference we will use it.1100 *1101 * fully-peeled:1102 *1103 * All references in the file that can be peeled are peeled.1104 * Inversely (and this is more important), any references in the1105 * file for which no peeled value is recorded is not peelable. This1106 * trait should typically be written alongside "peeled" for1107 * compatibility with older clients, but we do not require it1108 * (i.e., "peeled" is a no-op if "fully-peeled" is set).1109 */1110static void read_packed_refs(FILE *f, struct ref_dir *dir)1111{1112 struct ref_entry *last = NULL;1113 struct strbuf line = STRBUF_INIT;1114 enum { PEELED_NONE, PEELED_TAGS, PEELED_FULLY } peeled = PEELED_NONE;11151116 while (strbuf_getwholeline(&line, f, '\n') != EOF) {1117 unsigned char sha1[20];1118 const char *refname;1119 const char *traits;11201121 if (skip_prefix(line.buf, "# pack-refs with:", &traits)) {1122 if (strstr(traits, " fully-peeled "))1123 peeled = PEELED_FULLY;1124 else if (strstr(traits, " peeled "))1125 peeled = PEELED_TAGS;1126 /* perhaps other traits later as well */1127 continue;1128 }11291130 refname = parse_ref_line(&line, sha1);1131 if (refname) {1132 int flag = REF_ISPACKED;11331134 if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL)) {1135 if (!refname_is_safe(refname))1136 die("packed refname is dangerous: %s", refname);1137 hashclr(sha1);1138 flag |= REF_BAD_NAME | REF_ISBROKEN;1139 }1140 last = create_ref_entry(refname, sha1, flag, 0);1141 if (peeled == PEELED_FULLY ||1142 (peeled == PEELED_TAGS && starts_with(refname, "refs/tags/")))1143 last->flag |= REF_KNOWS_PEELED;1144 add_ref(dir, last);1145 continue;1146 }1147 if (last &&1148 line.buf[0] == '^' &&1149 line.len == PEELED_LINE_LENGTH &&1150 line.buf[PEELED_LINE_LENGTH - 1] == '\n' &&1151 !get_sha1_hex(line.buf + 1, sha1)) {1152 hashcpy(last->u.value.peeled.hash, sha1);1153 /*1154 * Regardless of what the file header said,1155 * we definitely know the value of *this*1156 * reference:1157 */1158 last->flag |= REF_KNOWS_PEELED;1159 }1160 }11611162 strbuf_release(&line);1163}11641165static const char *files_packed_refs_path(struct files_ref_store *refs)1166{1167 return refs->packed_refs_path;1168}11691170/*1171 * Get the packed_ref_cache for the specified files_ref_store,1172 * creating it if necessary.1173 */1174static struct packed_ref_cache *get_packed_ref_cache(struct files_ref_store *refs)1175{1176 const char *packed_refs_file = files_packed_refs_path(refs);11771178 if (refs->packed &&1179 !stat_validity_check(&refs->packed->validity, packed_refs_file))1180 clear_packed_ref_cache(refs);11811182 if (!refs->packed) {1183 FILE *f;11841185 refs->packed = xcalloc(1, sizeof(*refs->packed));1186 acquire_packed_ref_cache(refs->packed);1187 refs->packed->root = create_dir_entry(refs, "", 0, 0);1188 f = fopen(packed_refs_file, "r");1189 if (f) {1190 stat_validity_update(&refs->packed->validity, fileno(f));1191 read_packed_refs(f, get_ref_dir(refs->packed->root));1192 fclose(f);1193 }1194 }1195 return refs->packed;1196}11971198static struct ref_dir *get_packed_ref_dir(struct packed_ref_cache *packed_ref_cache)1199{1200 return get_ref_dir(packed_ref_cache->root);1201}12021203static struct ref_dir *get_packed_refs(struct files_ref_store *refs)1204{1205 return get_packed_ref_dir(get_packed_ref_cache(refs));1206}12071208/*1209 * Add a reference to the in-memory packed reference cache. This may1210 * only be called while the packed-refs file is locked (see1211 * lock_packed_refs()). To actually write the packed-refs file, call1212 * commit_packed_refs().1213 */1214static void add_packed_ref(struct files_ref_store *refs,1215 const char *refname, const unsigned char *sha1)1216{1217 struct packed_ref_cache *packed_ref_cache = get_packed_ref_cache(refs);12181219 if (!packed_ref_cache->lock)1220 die("internal error: packed refs not locked");1221 add_ref(get_packed_ref_dir(packed_ref_cache),1222 create_ref_entry(refname, sha1, REF_ISPACKED, 1));1223}12241225/*1226 * Read the loose references from the namespace dirname into dir1227 * (without recursing). dirname must end with '/'. dir must be the1228 * directory entry corresponding to dirname.1229 */1230static void read_loose_refs(const char *dirname, struct ref_dir *dir)1231{1232 struct files_ref_store *refs = dir->ref_store;1233 DIR *d;1234 struct dirent *de;1235 int dirnamelen = strlen(dirname);1236 struct strbuf refname;1237 struct strbuf path = STRBUF_INIT;1238 size_t path_baselen;1239 int err = 0;12401241 if (refs->submodule)1242 err = strbuf_git_path_submodule(&path, refs->submodule, "%s", dirname);1243 else1244 strbuf_git_path(&path, "%s", dirname);1245 path_baselen = path.len;12461247 if (err) {1248 strbuf_release(&path);1249 return;1250 }12511252 d = opendir(path.buf);1253 if (!d) {1254 strbuf_release(&path);1255 return;1256 }12571258 strbuf_init(&refname, dirnamelen + 257);1259 strbuf_add(&refname, dirname, dirnamelen);12601261 while ((de = readdir(d)) != NULL) {1262 unsigned char sha1[20];1263 struct stat st;1264 int flag;12651266 if (de->d_name[0] == '.')1267 continue;1268 if (ends_with(de->d_name, ".lock"))1269 continue;1270 strbuf_addstr(&refname, de->d_name);1271 strbuf_addstr(&path, de->d_name);1272 if (stat(path.buf, &st) < 0) {1273 ; /* silently ignore */1274 } else if (S_ISDIR(st.st_mode)) {1275 strbuf_addch(&refname, '/');1276 add_entry_to_dir(dir,1277 create_dir_entry(refs, refname.buf,1278 refname.len, 1));1279 } else {1280 if (!resolve_ref_recursively(&refs->base,1281 refname.buf,1282 RESOLVE_REF_READING,1283 sha1, &flag)) {1284 hashclr(sha1);1285 flag |= REF_ISBROKEN;1286 } else if (is_null_sha1(sha1)) {1287 /*1288 * It is so astronomically unlikely1289 * that NULL_SHA1 is the SHA-1 of an1290 * actual object that we consider its1291 * appearance in a loose reference1292 * file to be repo corruption1293 * (probably due to a software bug).1294 */1295 flag |= REF_ISBROKEN;1296 }12971298 if (check_refname_format(refname.buf,1299 REFNAME_ALLOW_ONELEVEL)) {1300 if (!refname_is_safe(refname.buf))1301 die("loose refname is dangerous: %s", refname.buf);1302 hashclr(sha1);1303 flag |= REF_BAD_NAME | REF_ISBROKEN;1304 }1305 add_entry_to_dir(dir,1306 create_ref_entry(refname.buf, sha1, flag, 0));1307 }1308 strbuf_setlen(&refname, dirnamelen);1309 strbuf_setlen(&path, path_baselen);1310 }1311 strbuf_release(&refname);1312 strbuf_release(&path);1313 closedir(d);1314}13151316static struct ref_dir *get_loose_refs(struct files_ref_store *refs)1317{1318 if (!refs->loose) {1319 /*1320 * Mark the top-level directory complete because we1321 * are about to read the only subdirectory that can1322 * hold references:1323 */1324 refs->loose = create_dir_entry(refs, "", 0, 0);1325 /*1326 * Create an incomplete entry for "refs/":1327 */1328 add_entry_to_dir(get_ref_dir(refs->loose),1329 create_dir_entry(refs, "refs/", 5, 1));1330 }1331 return get_ref_dir(refs->loose);1332}13331334/*1335 * Return the ref_entry for the given refname from the packed1336 * references. If it does not exist, return NULL.1337 */1338static struct ref_entry *get_packed_ref(struct files_ref_store *refs,1339 const char *refname)1340{1341 return find_ref(get_packed_refs(refs), refname);1342}13431344/*1345 * A loose ref file doesn't exist; check for a packed ref.1346 */1347static int resolve_packed_ref(struct files_ref_store *refs,1348 const char *refname,1349 unsigned char *sha1, unsigned int *flags)1350{1351 struct ref_entry *entry;13521353 /*1354 * The loose reference file does not exist; check for a packed1355 * reference.1356 */1357 entry = get_packed_ref(refs, refname);1358 if (entry) {1359 hashcpy(sha1, entry->u.value.oid.hash);1360 *flags |= REF_ISPACKED;1361 return 0;1362 }1363 /* refname is not a packed reference. */1364 return -1;1365}13661367static int files_read_raw_ref(struct ref_store *ref_store,1368 const char *refname, unsigned char *sha1,1369 struct strbuf *referent, unsigned int *type)1370{1371 struct files_ref_store *refs =1372 files_downcast(ref_store, 1, "read_raw_ref");1373 struct strbuf sb_contents = STRBUF_INIT;1374 struct strbuf sb_path = STRBUF_INIT;1375 const char *path;1376 const char *buf;1377 struct stat st;1378 int fd;1379 int ret = -1;1380 int save_errno;1381 int remaining_retries = 3;13821383 *type = 0;1384 strbuf_reset(&sb_path);13851386 if (refs->submodule)1387 strbuf_git_path_submodule(&sb_path, refs->submodule, "%s", refname);1388 else1389 strbuf_git_path(&sb_path, "%s", refname);13901391 path = sb_path.buf;13921393stat_ref:1394 /*1395 * We might have to loop back here to avoid a race1396 * condition: first we lstat() the file, then we try1397 * to read it as a link or as a file. But if somebody1398 * changes the type of the file (file <-> directory1399 * <-> symlink) between the lstat() and reading, then1400 * we don't want to report that as an error but rather1401 * try again starting with the lstat().1402 *1403 * We'll keep a count of the retries, though, just to avoid1404 * any confusing situation sending us into an infinite loop.1405 */14061407 if (remaining_retries-- <= 0)1408 goto out;14091410 if (lstat(path, &st) < 0) {1411 if (errno != ENOENT)1412 goto out;1413 if (resolve_packed_ref(refs, refname, sha1, type)) {1414 errno = ENOENT;1415 goto out;1416 }1417 ret = 0;1418 goto out;1419 }14201421 /* Follow "normalized" - ie "refs/.." symlinks by hand */1422 if (S_ISLNK(st.st_mode)) {1423 strbuf_reset(&sb_contents);1424 if (strbuf_readlink(&sb_contents, path, 0) < 0) {1425 if (errno == ENOENT || errno == EINVAL)1426 /* inconsistent with lstat; retry */1427 goto stat_ref;1428 else1429 goto out;1430 }1431 if (starts_with(sb_contents.buf, "refs/") &&1432 !check_refname_format(sb_contents.buf, 0)) {1433 strbuf_swap(&sb_contents, referent);1434 *type |= REF_ISSYMREF;1435 ret = 0;1436 goto out;1437 }1438 /*1439 * It doesn't look like a refname; fall through to just1440 * treating it like a non-symlink, and reading whatever it1441 * points to.1442 */1443 }14441445 /* Is it a directory? */1446 if (S_ISDIR(st.st_mode)) {1447 /*1448 * Even though there is a directory where the loose1449 * ref is supposed to be, there could still be a1450 * packed ref:1451 */1452 if (resolve_packed_ref(refs, refname, sha1, type)) {1453 errno = EISDIR;1454 goto out;1455 }1456 ret = 0;1457 goto out;1458 }14591460 /*1461 * Anything else, just open it and try to use it as1462 * a ref1463 */1464 fd = open(path, O_RDONLY);1465 if (fd < 0) {1466 if (errno == ENOENT && !S_ISLNK(st.st_mode))1467 /* inconsistent with lstat; retry */1468 goto stat_ref;1469 else1470 goto out;1471 }1472 strbuf_reset(&sb_contents);1473 if (strbuf_read(&sb_contents, fd, 256) < 0) {1474 int save_errno = errno;1475 close(fd);1476 errno = save_errno;1477 goto out;1478 }1479 close(fd);1480 strbuf_rtrim(&sb_contents);1481 buf = sb_contents.buf;1482 if (starts_with(buf, "ref:")) {1483 buf += 4;1484 while (isspace(*buf))1485 buf++;14861487 strbuf_reset(referent);1488 strbuf_addstr(referent, buf);1489 *type |= REF_ISSYMREF;1490 ret = 0;1491 goto out;1492 }14931494 /*1495 * Please note that FETCH_HEAD has additional1496 * data after the sha.1497 */1498 if (get_sha1_hex(buf, sha1) ||1499 (buf[40] != '\0' && !isspace(buf[40]))) {1500 *type |= REF_ISBROKEN;1501 errno = EINVAL;1502 goto out;1503 }15041505 ret = 0;15061507out:1508 save_errno = errno;1509 strbuf_release(&sb_path);1510 strbuf_release(&sb_contents);1511 errno = save_errno;1512 return ret;1513}15141515static void unlock_ref(struct ref_lock *lock)1516{1517 /* Do not free lock->lk -- atexit() still looks at them */1518 if (lock->lk)1519 rollback_lock_file(lock->lk);1520 free(lock->ref_name);1521 free(lock);1522}15231524/*1525 * Lock refname, without following symrefs, and set *lock_p to point1526 * at a newly-allocated lock object. Fill in lock->old_oid, referent,1527 * and type similarly to read_raw_ref().1528 *1529 * The caller must verify that refname is a "safe" reference name (in1530 * the sense of refname_is_safe()) before calling this function.1531 *1532 * If the reference doesn't already exist, verify that refname doesn't1533 * have a D/F conflict with any existing references. extras and skip1534 * are passed to verify_refname_available_dir() for this check.1535 *1536 * If mustexist is not set and the reference is not found or is1537 * broken, lock the reference anyway but clear sha1.1538 *1539 * Return 0 on success. On failure, write an error message to err and1540 * return TRANSACTION_NAME_CONFLICT or TRANSACTION_GENERIC_ERROR.1541 *1542 * Implementation note: This function is basically1543 *1544 * lock reference1545 * read_raw_ref()1546 *1547 * but it includes a lot more code to1548 * - Deal with possible races with other processes1549 * - Avoid calling verify_refname_available_dir() when it can be1550 * avoided, namely if we were successfully able to read the ref1551 * - Generate informative error messages in the case of failure1552 */1553static int lock_raw_ref(struct files_ref_store *refs,1554 const char *refname, int mustexist,1555 const struct string_list *extras,1556 const struct string_list *skip,1557 struct ref_lock **lock_p,1558 struct strbuf *referent,1559 unsigned int *type,1560 struct strbuf *err)1561{1562 struct ref_lock *lock;1563 struct strbuf ref_file = STRBUF_INIT;1564 int attempts_remaining = 3;1565 int ret = TRANSACTION_GENERIC_ERROR;15661567 assert(err);1568 files_assert_main_repository(refs, "lock_raw_ref");15691570 *type = 0;15711572 /* First lock the file so it can't change out from under us. */15731574 *lock_p = lock = xcalloc(1, sizeof(*lock));15751576 lock->ref_name = xstrdup(refname);1577 strbuf_git_path(&ref_file, "%s", refname);15781579retry:1580 switch (safe_create_leading_directories(ref_file.buf)) {1581 case SCLD_OK:1582 break; /* success */1583 case SCLD_EXISTS:1584 /*1585 * Suppose refname is "refs/foo/bar". We just failed1586 * to create the containing directory, "refs/foo",1587 * because there was a non-directory in the way. This1588 * indicates a D/F conflict, probably because of1589 * another reference such as "refs/foo". There is no1590 * reason to expect this error to be transitory.1591 */1592 if (verify_refname_available(refname, extras, skip, err)) {1593 if (mustexist) {1594 /*1595 * To the user the relevant error is1596 * that the "mustexist" reference is1597 * missing:1598 */1599 strbuf_reset(err);1600 strbuf_addf(err, "unable to resolve reference '%s'",1601 refname);1602 } else {1603 /*1604 * The error message set by1605 * verify_refname_available_dir() is OK.1606 */1607 ret = TRANSACTION_NAME_CONFLICT;1608 }1609 } else {1610 /*1611 * The file that is in the way isn't a loose1612 * reference. Report it as a low-level1613 * failure.1614 */1615 strbuf_addf(err, "unable to create lock file %s.lock; "1616 "non-directory in the way",1617 ref_file.buf);1618 }1619 goto error_return;1620 case SCLD_VANISHED:1621 /* Maybe another process was tidying up. Try again. */1622 if (--attempts_remaining > 0)1623 goto retry;1624 /* fall through */1625 default:1626 strbuf_addf(err, "unable to create directory for %s",1627 ref_file.buf);1628 goto error_return;1629 }16301631 if (!lock->lk)1632 lock->lk = xcalloc(1, sizeof(struct lock_file));16331634 if (hold_lock_file_for_update(lock->lk, ref_file.buf, LOCK_NO_DEREF) < 0) {1635 if (errno == ENOENT && --attempts_remaining > 0) {1636 /*1637 * Maybe somebody just deleted one of the1638 * directories leading to ref_file. Try1639 * again:1640 */1641 goto retry;1642 } else {1643 unable_to_lock_message(ref_file.buf, errno, err);1644 goto error_return;1645 }1646 }16471648 /*1649 * Now we hold the lock and can read the reference without1650 * fear that its value will change.1651 */16521653 if (files_read_raw_ref(&refs->base, refname,1654 lock->old_oid.hash, referent, type)) {1655 if (errno == ENOENT) {1656 if (mustexist) {1657 /* Garden variety missing reference. */1658 strbuf_addf(err, "unable to resolve reference '%s'",1659 refname);1660 goto error_return;1661 } else {1662 /*1663 * Reference is missing, but that's OK. We1664 * know that there is not a conflict with1665 * another loose reference because1666 * (supposing that we are trying to lock1667 * reference "refs/foo/bar"):1668 *1669 * - We were successfully able to create1670 * the lockfile refs/foo/bar.lock, so we1671 * know there cannot be a loose reference1672 * named "refs/foo".1673 *1674 * - We got ENOENT and not EISDIR, so we1675 * know that there cannot be a loose1676 * reference named "refs/foo/bar/baz".1677 */1678 }1679 } else if (errno == EISDIR) {1680 /*1681 * There is a directory in the way. It might have1682 * contained references that have been deleted. If1683 * we don't require that the reference already1684 * exists, try to remove the directory so that it1685 * doesn't cause trouble when we want to rename the1686 * lockfile into place later.1687 */1688 if (mustexist) {1689 /* Garden variety missing reference. */1690 strbuf_addf(err, "unable to resolve reference '%s'",1691 refname);1692 goto error_return;1693 } else if (remove_dir_recursively(&ref_file,1694 REMOVE_DIR_EMPTY_ONLY)) {1695 if (verify_refname_available_dir(1696 refname, extras, skip,1697 get_loose_refs(refs),1698 err)) {1699 /*1700 * The error message set by1701 * verify_refname_available() is OK.1702 */1703 ret = TRANSACTION_NAME_CONFLICT;1704 goto error_return;1705 } else {1706 /*1707 * We can't delete the directory,1708 * but we also don't know of any1709 * references that it should1710 * contain.1711 */1712 strbuf_addf(err, "there is a non-empty directory '%s' "1713 "blocking reference '%s'",1714 ref_file.buf, refname);1715 goto error_return;1716 }1717 }1718 } else if (errno == EINVAL && (*type & REF_ISBROKEN)) {1719 strbuf_addf(err, "unable to resolve reference '%s': "1720 "reference broken", refname);1721 goto error_return;1722 } else {1723 strbuf_addf(err, "unable to resolve reference '%s': %s",1724 refname, strerror(errno));1725 goto error_return;1726 }17271728 /*1729 * If the ref did not exist and we are creating it,1730 * make sure there is no existing packed ref whose1731 * name begins with our refname, nor a packed ref1732 * whose name is a proper prefix of our refname.1733 */1734 if (verify_refname_available_dir(1735 refname, extras, skip,1736 get_packed_refs(refs),1737 err)) {1738 goto error_return;1739 }1740 }17411742 ret = 0;1743 goto out;17441745error_return:1746 unlock_ref(lock);1747 *lock_p = NULL;17481749out:1750 strbuf_release(&ref_file);1751 return ret;1752}17531754/*1755 * Peel the entry (if possible) and return its new peel_status. If1756 * repeel is true, re-peel the entry even if there is an old peeled1757 * value that is already stored in it.1758 *1759 * It is OK to call this function with a packed reference entry that1760 * might be stale and might even refer to an object that has since1761 * been garbage-collected. In such a case, if the entry has1762 * REF_KNOWS_PEELED then leave the status unchanged and return1763 * PEEL_PEELED or PEEL_NON_TAG; otherwise, return PEEL_INVALID.1764 */1765static enum peel_status peel_entry(struct ref_entry *entry, int repeel)1766{1767 enum peel_status status;17681769 if (entry->flag & REF_KNOWS_PEELED) {1770 if (repeel) {1771 entry->flag &= ~REF_KNOWS_PEELED;1772 oidclr(&entry->u.value.peeled);1773 } else {1774 return is_null_oid(&entry->u.value.peeled) ?1775 PEEL_NON_TAG : PEEL_PEELED;1776 }1777 }1778 if (entry->flag & REF_ISBROKEN)1779 return PEEL_BROKEN;1780 if (entry->flag & REF_ISSYMREF)1781 return PEEL_IS_SYMREF;17821783 status = peel_object(entry->u.value.oid.hash, entry->u.value.peeled.hash);1784 if (status == PEEL_PEELED || status == PEEL_NON_TAG)1785 entry->flag |= REF_KNOWS_PEELED;1786 return status;1787}17881789static int files_peel_ref(struct ref_store *ref_store,1790 const char *refname, unsigned char *sha1)1791{1792 struct files_ref_store *refs = files_downcast(ref_store, 0, "peel_ref");1793 int flag;1794 unsigned char base[20];17951796 if (current_ref_iter && current_ref_iter->refname == refname) {1797 struct object_id peeled;17981799 if (ref_iterator_peel(current_ref_iter, &peeled))1800 return -1;1801 hashcpy(sha1, peeled.hash);1802 return 0;1803 }18041805 if (read_ref_full(refname, RESOLVE_REF_READING, base, &flag))1806 return -1;18071808 /*1809 * If the reference is packed, read its ref_entry from the1810 * cache in the hope that we already know its peeled value.1811 * We only try this optimization on packed references because1812 * (a) forcing the filling of the loose reference cache could1813 * be expensive and (b) loose references anyway usually do not1814 * have REF_KNOWS_PEELED.1815 */1816 if (flag & REF_ISPACKED) {1817 struct ref_entry *r = get_packed_ref(refs, refname);1818 if (r) {1819 if (peel_entry(r, 0))1820 return -1;1821 hashcpy(sha1, r->u.value.peeled.hash);1822 return 0;1823 }1824 }18251826 return peel_object(base, sha1);1827}18281829struct files_ref_iterator {1830 struct ref_iterator base;18311832 struct packed_ref_cache *packed_ref_cache;1833 struct ref_iterator *iter0;1834 unsigned int flags;1835};18361837static int files_ref_iterator_advance(struct ref_iterator *ref_iterator)1838{1839 struct files_ref_iterator *iter =1840 (struct files_ref_iterator *)ref_iterator;1841 int ok;18421843 while ((ok = ref_iterator_advance(iter->iter0)) == ITER_OK) {1844 if (iter->flags & DO_FOR_EACH_PER_WORKTREE_ONLY &&1845 ref_type(iter->iter0->refname) != REF_TYPE_PER_WORKTREE)1846 continue;18471848 if (!(iter->flags & DO_FOR_EACH_INCLUDE_BROKEN) &&1849 !ref_resolves_to_object(iter->iter0->refname,1850 iter->iter0->oid,1851 iter->iter0->flags))1852 continue;18531854 iter->base.refname = iter->iter0->refname;1855 iter->base.oid = iter->iter0->oid;1856 iter->base.flags = iter->iter0->flags;1857 return ITER_OK;1858 }18591860 iter->iter0 = NULL;1861 if (ref_iterator_abort(ref_iterator) != ITER_DONE)1862 ok = ITER_ERROR;18631864 return ok;1865}18661867static int files_ref_iterator_peel(struct ref_iterator *ref_iterator,1868 struct object_id *peeled)1869{1870 struct files_ref_iterator *iter =1871 (struct files_ref_iterator *)ref_iterator;18721873 return ref_iterator_peel(iter->iter0, peeled);1874}18751876static int files_ref_iterator_abort(struct ref_iterator *ref_iterator)1877{1878 struct files_ref_iterator *iter =1879 (struct files_ref_iterator *)ref_iterator;1880 int ok = ITER_DONE;18811882 if (iter->iter0)1883 ok = ref_iterator_abort(iter->iter0);18841885 release_packed_ref_cache(iter->packed_ref_cache);1886 base_ref_iterator_free(ref_iterator);1887 return ok;1888}18891890static struct ref_iterator_vtable files_ref_iterator_vtable = {1891 files_ref_iterator_advance,1892 files_ref_iterator_peel,1893 files_ref_iterator_abort1894};18951896static struct ref_iterator *files_ref_iterator_begin(1897 struct ref_store *ref_store,1898 const char *prefix, unsigned int flags)1899{1900 struct files_ref_store *refs =1901 files_downcast(ref_store, 1, "ref_iterator_begin");1902 struct ref_dir *loose_dir, *packed_dir;1903 struct ref_iterator *loose_iter, *packed_iter;1904 struct files_ref_iterator *iter;1905 struct ref_iterator *ref_iterator;19061907 if (ref_paranoia < 0)1908 ref_paranoia = git_env_bool("GIT_REF_PARANOIA", 0);1909 if (ref_paranoia)1910 flags |= DO_FOR_EACH_INCLUDE_BROKEN;19111912 iter = xcalloc(1, sizeof(*iter));1913 ref_iterator = &iter->base;1914 base_ref_iterator_init(ref_iterator, &files_ref_iterator_vtable);19151916 /*1917 * We must make sure that all loose refs are read before1918 * accessing the packed-refs file; this avoids a race1919 * condition if loose refs are migrated to the packed-refs1920 * file by a simultaneous process, but our in-memory view is1921 * from before the migration. We ensure this as follows:1922 * First, we call prime_ref_dir(), which pre-reads the loose1923 * references for the subtree into the cache. (If they've1924 * already been read, that's OK; we only need to guarantee1925 * that they're read before the packed refs, not *how much*1926 * before.) After that, we call get_packed_ref_cache(), which1927 * internally checks whether the packed-ref cache is up to1928 * date with what is on disk, and re-reads it if not.1929 */19301931 loose_dir = get_loose_refs(refs);19321933 if (prefix && *prefix)1934 loose_dir = find_containing_dir(loose_dir, prefix, 0);19351936 if (loose_dir) {1937 prime_ref_dir(loose_dir);1938 loose_iter = cache_ref_iterator_begin(loose_dir);1939 } else {1940 /* There's nothing to iterate over. */1941 loose_iter = empty_ref_iterator_begin();1942 }19431944 iter->packed_ref_cache = get_packed_ref_cache(refs);1945 acquire_packed_ref_cache(iter->packed_ref_cache);1946 packed_dir = get_packed_ref_dir(iter->packed_ref_cache);19471948 if (prefix && *prefix)1949 packed_dir = find_containing_dir(packed_dir, prefix, 0);19501951 if (packed_dir) {1952 packed_iter = cache_ref_iterator_begin(packed_dir);1953 } else {1954 /* There's nothing to iterate over. */1955 packed_iter = empty_ref_iterator_begin();1956 }19571958 iter->iter0 = overlay_ref_iterator_begin(loose_iter, packed_iter);1959 iter->flags = flags;19601961 return ref_iterator;1962}19631964/*1965 * Verify that the reference locked by lock has the value old_sha1.1966 * Fail if the reference doesn't exist and mustexist is set. Return 01967 * on success. On error, write an error message to err, set errno, and1968 * return a negative value.1969 */1970static int verify_lock(struct ref_lock *lock,1971 const unsigned char *old_sha1, int mustexist,1972 struct strbuf *err)1973{1974 assert(err);19751976 if (read_ref_full(lock->ref_name,1977 mustexist ? RESOLVE_REF_READING : 0,1978 lock->old_oid.hash, NULL)) {1979 if (old_sha1) {1980 int save_errno = errno;1981 strbuf_addf(err, "can't verify ref '%s'", lock->ref_name);1982 errno = save_errno;1983 return -1;1984 } else {1985 oidclr(&lock->old_oid);1986 return 0;1987 }1988 }1989 if (old_sha1 && hashcmp(lock->old_oid.hash, old_sha1)) {1990 strbuf_addf(err, "ref '%s' is at %s but expected %s",1991 lock->ref_name,1992 oid_to_hex(&lock->old_oid),1993 sha1_to_hex(old_sha1));1994 errno = EBUSY;1995 return -1;1996 }1997 return 0;1998}19992000static int remove_empty_directories(struct strbuf *path)2001{2002 /*2003 * we want to create a file but there is a directory there;2004 * if that is an empty directory (or a directory that contains2005 * only empty directories), remove them.2006 */2007 return remove_dir_recursively(path, REMOVE_DIR_EMPTY_ONLY);2008}20092010static int create_reflock(const char *path, void *cb)2011{2012 struct lock_file *lk = cb;20132014 return hold_lock_file_for_update(lk, path, LOCK_NO_DEREF) < 0 ? -1 : 0;2015}20162017/*2018 * Locks a ref returning the lock on success and NULL on failure.2019 * On failure errno is set to something meaningful.2020 */2021static struct ref_lock *lock_ref_sha1_basic(struct files_ref_store *refs,2022 const char *refname,2023 const unsigned char *old_sha1,2024 const struct string_list *extras,2025 const struct string_list *skip,2026 unsigned int flags, int *type,2027 struct strbuf *err)2028{2029 struct strbuf ref_file = STRBUF_INIT;2030 struct ref_lock *lock;2031 int last_errno = 0;2032 int mustexist = (old_sha1 && !is_null_sha1(old_sha1));2033 int resolve_flags = RESOLVE_REF_NO_RECURSE;2034 int resolved;20352036 files_assert_main_repository(refs, "lock_ref_sha1_basic");2037 assert(err);20382039 lock = xcalloc(1, sizeof(struct ref_lock));20402041 if (mustexist)2042 resolve_flags |= RESOLVE_REF_READING;2043 if (flags & REF_DELETING)2044 resolve_flags |= RESOLVE_REF_ALLOW_BAD_NAME;20452046 strbuf_git_path(&ref_file, "%s", refname);2047 resolved = !!resolve_ref_unsafe(refname, resolve_flags,2048 lock->old_oid.hash, type);2049 if (!resolved && errno == EISDIR) {2050 /*2051 * we are trying to lock foo but we used to2052 * have foo/bar which now does not exist;2053 * it is normal for the empty directory 'foo'2054 * to remain.2055 */2056 if (remove_empty_directories(&ref_file)) {2057 last_errno = errno;2058 if (!verify_refname_available_dir(2059 refname, extras, skip,2060 get_loose_refs(refs), err))2061 strbuf_addf(err, "there are still refs under '%s'",2062 refname);2063 goto error_return;2064 }2065 resolved = !!resolve_ref_unsafe(refname, resolve_flags,2066 lock->old_oid.hash, type);2067 }2068 if (!resolved) {2069 last_errno = errno;2070 if (last_errno != ENOTDIR ||2071 !verify_refname_available_dir(2072 refname, extras, skip,2073 get_loose_refs(refs), err))2074 strbuf_addf(err, "unable to resolve reference '%s': %s",2075 refname, strerror(last_errno));20762077 goto error_return;2078 }20792080 /*2081 * If the ref did not exist and we are creating it, make sure2082 * there is no existing packed ref whose name begins with our2083 * refname, nor a packed ref whose name is a proper prefix of2084 * our refname.2085 */2086 if (is_null_oid(&lock->old_oid) &&2087 verify_refname_available_dir(refname, extras, skip,2088 get_packed_refs(refs),2089 err)) {2090 last_errno = ENOTDIR;2091 goto error_return;2092 }20932094 lock->lk = xcalloc(1, sizeof(struct lock_file));20952096 lock->ref_name = xstrdup(refname);20972098 if (raceproof_create_file(ref_file.buf, create_reflock, lock->lk)) {2099 last_errno = errno;2100 unable_to_lock_message(ref_file.buf, errno, err);2101 goto error_return;2102 }21032104 if (verify_lock(lock, old_sha1, mustexist, err)) {2105 last_errno = errno;2106 goto error_return;2107 }2108 goto out;21092110 error_return:2111 unlock_ref(lock);2112 lock = NULL;21132114 out:2115 strbuf_release(&ref_file);2116 errno = last_errno;2117 return lock;2118}21192120/*2121 * Write an entry to the packed-refs file for the specified refname.2122 * If peeled is non-NULL, write it as the entry's peeled value.2123 */2124static void write_packed_entry(FILE *fh, char *refname, unsigned char *sha1,2125 unsigned char *peeled)2126{2127 fprintf_or_die(fh, "%s %s\n", sha1_to_hex(sha1), refname);2128 if (peeled)2129 fprintf_or_die(fh, "^%s\n", sha1_to_hex(peeled));2130}21312132/*2133 * An each_ref_entry_fn that writes the entry to a packed-refs file.2134 */2135static int write_packed_entry_fn(struct ref_entry *entry, void *cb_data)2136{2137 enum peel_status peel_status = peel_entry(entry, 0);21382139 if (peel_status != PEEL_PEELED && peel_status != PEEL_NON_TAG)2140 error("internal error: %s is not a valid packed reference!",2141 entry->name);2142 write_packed_entry(cb_data, entry->name, entry->u.value.oid.hash,2143 peel_status == PEEL_PEELED ?2144 entry->u.value.peeled.hash : NULL);2145 return 0;2146}21472148/*2149 * Lock the packed-refs file for writing. Flags is passed to2150 * hold_lock_file_for_update(). Return 0 on success. On errors, set2151 * errno appropriately and return a nonzero value.2152 */2153static int lock_packed_refs(struct files_ref_store *refs, int flags)2154{2155 static int timeout_configured = 0;2156 static int timeout_value = 1000;2157 struct packed_ref_cache *packed_ref_cache;21582159 files_assert_main_repository(refs, "lock_packed_refs");21602161 if (!timeout_configured) {2162 git_config_get_int("core.packedrefstimeout", &timeout_value);2163 timeout_configured = 1;2164 }21652166 if (hold_lock_file_for_update_timeout(2167 &packlock, files_packed_refs_path(refs),2168 flags, timeout_value) < 0)2169 return -1;2170 /*2171 * Get the current packed-refs while holding the lock. If the2172 * packed-refs file has been modified since we last read it,2173 * this will automatically invalidate the cache and re-read2174 * the packed-refs file.2175 */2176 packed_ref_cache = get_packed_ref_cache(refs);2177 packed_ref_cache->lock = &packlock;2178 /* Increment the reference count to prevent it from being freed: */2179 acquire_packed_ref_cache(packed_ref_cache);2180 return 0;2181}21822183/*2184 * Write the current version of the packed refs cache from memory to2185 * disk. The packed-refs file must already be locked for writing (see2186 * lock_packed_refs()). Return zero on success. On errors, set errno2187 * and return a nonzero value2188 */2189static int commit_packed_refs(struct files_ref_store *refs)2190{2191 struct packed_ref_cache *packed_ref_cache =2192 get_packed_ref_cache(refs);2193 int error = 0;2194 int save_errno = 0;2195 FILE *out;21962197 files_assert_main_repository(refs, "commit_packed_refs");21982199 if (!packed_ref_cache->lock)2200 die("internal error: packed-refs not locked");22012202 out = fdopen_lock_file(packed_ref_cache->lock, "w");2203 if (!out)2204 die_errno("unable to fdopen packed-refs descriptor");22052206 fprintf_or_die(out, "%s", PACKED_REFS_HEADER);2207 do_for_each_entry_in_dir(get_packed_ref_dir(packed_ref_cache),2208 0, write_packed_entry_fn, out);22092210 if (commit_lock_file(packed_ref_cache->lock)) {2211 save_errno = errno;2212 error = -1;2213 }2214 packed_ref_cache->lock = NULL;2215 release_packed_ref_cache(packed_ref_cache);2216 errno = save_errno;2217 return error;2218}22192220/*2221 * Rollback the lockfile for the packed-refs file, and discard the2222 * in-memory packed reference cache. (The packed-refs file will be2223 * read anew if it is needed again after this function is called.)2224 */2225static void rollback_packed_refs(struct files_ref_store *refs)2226{2227 struct packed_ref_cache *packed_ref_cache =2228 get_packed_ref_cache(refs);22292230 files_assert_main_repository(refs, "rollback_packed_refs");22312232 if (!packed_ref_cache->lock)2233 die("internal error: packed-refs not locked");2234 rollback_lock_file(packed_ref_cache->lock);2235 packed_ref_cache->lock = NULL;2236 release_packed_ref_cache(packed_ref_cache);2237 clear_packed_ref_cache(refs);2238}22392240struct ref_to_prune {2241 struct ref_to_prune *next;2242 unsigned char sha1[20];2243 char name[FLEX_ARRAY];2244};22452246struct pack_refs_cb_data {2247 unsigned int flags;2248 struct ref_dir *packed_refs;2249 struct ref_to_prune *ref_to_prune;2250};22512252/*2253 * An each_ref_entry_fn that is run over loose references only. If2254 * the loose reference can be packed, add an entry in the packed ref2255 * cache. If the reference should be pruned, also add it to2256 * ref_to_prune in the pack_refs_cb_data.2257 */2258static int pack_if_possible_fn(struct ref_entry *entry, void *cb_data)2259{2260 struct pack_refs_cb_data *cb = cb_data;2261 enum peel_status peel_status;2262 struct ref_entry *packed_entry;2263 int is_tag_ref = starts_with(entry->name, "refs/tags/");22642265 /* Do not pack per-worktree refs: */2266 if (ref_type(entry->name) != REF_TYPE_NORMAL)2267 return 0;22682269 /* ALWAYS pack tags */2270 if (!(cb->flags & PACK_REFS_ALL) && !is_tag_ref)2271 return 0;22722273 /* Do not pack symbolic or broken refs: */2274 if ((entry->flag & REF_ISSYMREF) || !entry_resolves_to_object(entry))2275 return 0;22762277 /* Add a packed ref cache entry equivalent to the loose entry. */2278 peel_status = peel_entry(entry, 1);2279 if (peel_status != PEEL_PEELED && peel_status != PEEL_NON_TAG)2280 die("internal error peeling reference %s (%s)",2281 entry->name, oid_to_hex(&entry->u.value.oid));2282 packed_entry = find_ref(cb->packed_refs, entry->name);2283 if (packed_entry) {2284 /* Overwrite existing packed entry with info from loose entry */2285 packed_entry->flag = REF_ISPACKED | REF_KNOWS_PEELED;2286 oidcpy(&packed_entry->u.value.oid, &entry->u.value.oid);2287 } else {2288 packed_entry = create_ref_entry(entry->name, entry->u.value.oid.hash,2289 REF_ISPACKED | REF_KNOWS_PEELED, 0);2290 add_ref(cb->packed_refs, packed_entry);2291 }2292 oidcpy(&packed_entry->u.value.peeled, &entry->u.value.peeled);22932294 /* Schedule the loose reference for pruning if requested. */2295 if ((cb->flags & PACK_REFS_PRUNE)) {2296 struct ref_to_prune *n;2297 FLEX_ALLOC_STR(n, name, entry->name);2298 hashcpy(n->sha1, entry->u.value.oid.hash);2299 n->next = cb->ref_to_prune;2300 cb->ref_to_prune = n;2301 }2302 return 0;2303}23042305enum {2306 REMOVE_EMPTY_PARENTS_REF = 0x01,2307 REMOVE_EMPTY_PARENTS_REFLOG = 0x022308};23092310/*2311 * Remove empty parent directories associated with the specified2312 * reference and/or its reflog, but spare [logs/]refs/ and immediate2313 * subdirs. flags is a combination of REMOVE_EMPTY_PARENTS_REF and/or2314 * REMOVE_EMPTY_PARENTS_REFLOG.2315 */2316static void try_remove_empty_parents(const char *refname, unsigned int flags)2317{2318 struct strbuf buf = STRBUF_INIT;2319 char *p, *q;2320 int i;23212322 strbuf_addstr(&buf, refname);2323 p = buf.buf;2324 for (i = 0; i < 2; i++) { /* refs/{heads,tags,...}/ */2325 while (*p && *p != '/')2326 p++;2327 /* tolerate duplicate slashes; see check_refname_format() */2328 while (*p == '/')2329 p++;2330 }2331 q = buf.buf + buf.len;2332 while (flags & (REMOVE_EMPTY_PARENTS_REF | REMOVE_EMPTY_PARENTS_REFLOG)) {2333 while (q > p && *q != '/')2334 q--;2335 while (q > p && *(q-1) == '/')2336 q--;2337 if (q == p)2338 break;2339 strbuf_setlen(&buf, q - buf.buf);2340 if ((flags & REMOVE_EMPTY_PARENTS_REF) &&2341 rmdir(git_path("%s", buf.buf)))2342 flags &= ~REMOVE_EMPTY_PARENTS_REF;2343 if ((flags & REMOVE_EMPTY_PARENTS_REFLOG) &&2344 rmdir(git_path("logs/%s", buf.buf)))2345 flags &= ~REMOVE_EMPTY_PARENTS_REFLOG;2346 }2347 strbuf_release(&buf);2348}23492350/* make sure nobody touched the ref, and unlink */2351static void prune_ref(struct ref_to_prune *r)2352{2353 struct ref_transaction *transaction;2354 struct strbuf err = STRBUF_INIT;23552356 if (check_refname_format(r->name, 0))2357 return;23582359 transaction = ref_transaction_begin(&err);2360 if (!transaction ||2361 ref_transaction_delete(transaction, r->name, r->sha1,2362 REF_ISPRUNING | REF_NODEREF, NULL, &err) ||2363 ref_transaction_commit(transaction, &err)) {2364 ref_transaction_free(transaction);2365 error("%s", err.buf);2366 strbuf_release(&err);2367 return;2368 }2369 ref_transaction_free(transaction);2370 strbuf_release(&err);2371}23722373static void prune_refs(struct ref_to_prune *r)2374{2375 while (r) {2376 prune_ref(r);2377 r = r->next;2378 }2379}23802381static int files_pack_refs(struct ref_store *ref_store, unsigned int flags)2382{2383 struct files_ref_store *refs =2384 files_downcast(ref_store, 0, "pack_refs");2385 struct pack_refs_cb_data cbdata;23862387 memset(&cbdata, 0, sizeof(cbdata));2388 cbdata.flags = flags;23892390 lock_packed_refs(refs, LOCK_DIE_ON_ERROR);2391 cbdata.packed_refs = get_packed_refs(refs);23922393 do_for_each_entry_in_dir(get_loose_refs(refs), 0,2394 pack_if_possible_fn, &cbdata);23952396 if (commit_packed_refs(refs))2397 die_errno("unable to overwrite old ref-pack file");23982399 prune_refs(cbdata.ref_to_prune);2400 return 0;2401}24022403/*2404 * Rewrite the packed-refs file, omitting any refs listed in2405 * 'refnames'. On error, leave packed-refs unchanged, write an error2406 * message to 'err', and return a nonzero value.2407 *2408 * The refs in 'refnames' needn't be sorted. `err` must not be NULL.2409 */2410static int repack_without_refs(struct files_ref_store *refs,2411 struct string_list *refnames, struct strbuf *err)2412{2413 struct ref_dir *packed;2414 struct string_list_item *refname;2415 int ret, needs_repacking = 0, removed = 0;24162417 files_assert_main_repository(refs, "repack_without_refs");2418 assert(err);24192420 /* Look for a packed ref */2421 for_each_string_list_item(refname, refnames) {2422 if (get_packed_ref(refs, refname->string)) {2423 needs_repacking = 1;2424 break;2425 }2426 }24272428 /* Avoid locking if we have nothing to do */2429 if (!needs_repacking)2430 return 0; /* no refname exists in packed refs */24312432 if (lock_packed_refs(refs, 0)) {2433 unable_to_lock_message(files_packed_refs_path(refs), errno, err);2434 return -1;2435 }2436 packed = get_packed_refs(refs);24372438 /* Remove refnames from the cache */2439 for_each_string_list_item(refname, refnames)2440 if (remove_entry(packed, refname->string) != -1)2441 removed = 1;2442 if (!removed) {2443 /*2444 * All packed entries disappeared while we were2445 * acquiring the lock.2446 */2447 rollback_packed_refs(refs);2448 return 0;2449 }24502451 /* Write what remains */2452 ret = commit_packed_refs(refs);2453 if (ret)2454 strbuf_addf(err, "unable to overwrite old ref-pack file: %s",2455 strerror(errno));2456 return ret;2457}24582459static int files_delete_refs(struct ref_store *ref_store,2460 struct string_list *refnames, unsigned int flags)2461{2462 struct files_ref_store *refs =2463 files_downcast(ref_store, 0, "delete_refs");2464 struct strbuf err = STRBUF_INIT;2465 int i, result = 0;24662467 if (!refnames->nr)2468 return 0;24692470 result = repack_without_refs(refs, refnames, &err);2471 if (result) {2472 /*2473 * If we failed to rewrite the packed-refs file, then2474 * it is unsafe to try to remove loose refs, because2475 * doing so might expose an obsolete packed value for2476 * a reference that might even point at an object that2477 * has been garbage collected.2478 */2479 if (refnames->nr == 1)2480 error(_("could not delete reference %s: %s"),2481 refnames->items[0].string, err.buf);2482 else2483 error(_("could not delete references: %s"), err.buf);24842485 goto out;2486 }24872488 for (i = 0; i < refnames->nr; i++) {2489 const char *refname = refnames->items[i].string;24902491 if (delete_ref(NULL, refname, NULL, flags))2492 result |= error(_("could not remove reference %s"), refname);2493 }24942495out:2496 strbuf_release(&err);2497 return result;2498}24992500/*2501 * People using contrib's git-new-workdir have .git/logs/refs ->2502 * /some/other/path/.git/logs/refs, and that may live on another device.2503 *2504 * IOW, to avoid cross device rename errors, the temporary renamed log must2505 * live into logs/refs.2506 */2507#define TMP_RENAMED_LOG "logs/refs/.tmp-renamed-log"25082509static int rename_tmp_log_callback(const char *path, void *cb)2510{2511 int *true_errno = cb;25122513 if (rename(git_path(TMP_RENAMED_LOG), path)) {2514 /*2515 * rename(a, b) when b is an existing directory ought2516 * to result in ISDIR, but Solaris 5.8 gives ENOTDIR.2517 * Sheesh. Record the true errno for error reporting,2518 * but report EISDIR to raceproof_create_file() so2519 * that it knows to retry.2520 */2521 *true_errno = errno;2522 if (errno == ENOTDIR)2523 errno = EISDIR;2524 return -1;2525 } else {2526 return 0;2527 }2528}25292530static int rename_tmp_log(const char *newrefname)2531{2532 char *path = git_pathdup("logs/%s", newrefname);2533 int ret, true_errno;25342535 ret = raceproof_create_file(path, rename_tmp_log_callback, &true_errno);2536 if (ret) {2537 if (errno == EISDIR)2538 error("directory not empty: %s", path);2539 else2540 error("unable to move logfile %s to %s: %s",2541 git_path(TMP_RENAMED_LOG), path,2542 strerror(true_errno));2543 }25442545 free(path);2546 return ret;2547}25482549static int files_verify_refname_available(struct ref_store *ref_store,2550 const char *newname,2551 const struct string_list *extras,2552 const struct string_list *skip,2553 struct strbuf *err)2554{2555 struct files_ref_store *refs =2556 files_downcast(ref_store, 1, "verify_refname_available");2557 struct ref_dir *packed_refs = get_packed_refs(refs);2558 struct ref_dir *loose_refs = get_loose_refs(refs);25592560 if (verify_refname_available_dir(newname, extras, skip,2561 packed_refs, err) ||2562 verify_refname_available_dir(newname, extras, skip,2563 loose_refs, err))2564 return -1;25652566 return 0;2567}25682569static int write_ref_to_lockfile(struct ref_lock *lock,2570 const unsigned char *sha1, struct strbuf *err);2571static int commit_ref_update(struct files_ref_store *refs,2572 struct ref_lock *lock,2573 const unsigned char *sha1, const char *logmsg,2574 struct strbuf *err);25752576static int files_rename_ref(struct ref_store *ref_store,2577 const char *oldrefname, const char *newrefname,2578 const char *logmsg)2579{2580 struct files_ref_store *refs =2581 files_downcast(ref_store, 0, "rename_ref");2582 unsigned char sha1[20], orig_sha1[20];2583 int flag = 0, logmoved = 0;2584 struct ref_lock *lock;2585 struct stat loginfo;2586 int log = !lstat(git_path("logs/%s", oldrefname), &loginfo);2587 struct strbuf err = STRBUF_INIT;25882589 if (log && S_ISLNK(loginfo.st_mode))2590 return error("reflog for %s is a symlink", oldrefname);25912592 if (!resolve_ref_unsafe(oldrefname, RESOLVE_REF_READING | RESOLVE_REF_NO_RECURSE,2593 orig_sha1, &flag))2594 return error("refname %s not found", oldrefname);25952596 if (flag & REF_ISSYMREF)2597 return error("refname %s is a symbolic ref, renaming it is not supported",2598 oldrefname);2599 if (!rename_ref_available(oldrefname, newrefname))2600 return 1;26012602 if (log && rename(git_path("logs/%s", oldrefname), git_path(TMP_RENAMED_LOG)))2603 return error("unable to move logfile logs/%s to "TMP_RENAMED_LOG": %s",2604 oldrefname, strerror(errno));26052606 if (delete_ref(logmsg, oldrefname, orig_sha1, REF_NODEREF)) {2607 error("unable to delete old %s", oldrefname);2608 goto rollback;2609 }26102611 /*2612 * Since we are doing a shallow lookup, sha1 is not the2613 * correct value to pass to delete_ref as old_sha1. But that2614 * doesn't matter, because an old_sha1 check wouldn't add to2615 * the safety anyway; we want to delete the reference whatever2616 * its current value.2617 */2618 if (!read_ref_full(newrefname, RESOLVE_REF_READING | RESOLVE_REF_NO_RECURSE,2619 sha1, NULL) &&2620 delete_ref(NULL, newrefname, NULL, REF_NODEREF)) {2621 if (errno == EISDIR) {2622 struct strbuf path = STRBUF_INIT;2623 int result;26242625 strbuf_git_path(&path, "%s", newrefname);2626 result = remove_empty_directories(&path);2627 strbuf_release(&path);26282629 if (result) {2630 error("Directory not empty: %s", newrefname);2631 goto rollback;2632 }2633 } else {2634 error("unable to delete existing %s", newrefname);2635 goto rollback;2636 }2637 }26382639 if (log && rename_tmp_log(newrefname))2640 goto rollback;26412642 logmoved = log;26432644 lock = lock_ref_sha1_basic(refs, newrefname, NULL, NULL, NULL,2645 REF_NODEREF, NULL, &err);2646 if (!lock) {2647 error("unable to rename '%s' to '%s': %s", oldrefname, newrefname, err.buf);2648 strbuf_release(&err);2649 goto rollback;2650 }2651 hashcpy(lock->old_oid.hash, orig_sha1);26522653 if (write_ref_to_lockfile(lock, orig_sha1, &err) ||2654 commit_ref_update(refs, lock, orig_sha1, logmsg, &err)) {2655 error("unable to write current sha1 into %s: %s", newrefname, err.buf);2656 strbuf_release(&err);2657 goto rollback;2658 }26592660 return 0;26612662 rollback:2663 lock = lock_ref_sha1_basic(refs, oldrefname, NULL, NULL, NULL,2664 REF_NODEREF, NULL, &err);2665 if (!lock) {2666 error("unable to lock %s for rollback: %s", oldrefname, err.buf);2667 strbuf_release(&err);2668 goto rollbacklog;2669 }26702671 flag = log_all_ref_updates;2672 log_all_ref_updates = LOG_REFS_NONE;2673 if (write_ref_to_lockfile(lock, orig_sha1, &err) ||2674 commit_ref_update(refs, lock, orig_sha1, NULL, &err)) {2675 error("unable to write current sha1 into %s: %s", oldrefname, err.buf);2676 strbuf_release(&err);2677 }2678 log_all_ref_updates = flag;26792680 rollbacklog:2681 if (logmoved && rename(git_path("logs/%s", newrefname), git_path("logs/%s", oldrefname)))2682 error("unable to restore logfile %s from %s: %s",2683 oldrefname, newrefname, strerror(errno));2684 if (!logmoved && log &&2685 rename(git_path(TMP_RENAMED_LOG), git_path("logs/%s", oldrefname)))2686 error("unable to restore logfile %s from "TMP_RENAMED_LOG": %s",2687 oldrefname, strerror(errno));26882689 return 1;2690}26912692static int close_ref(struct ref_lock *lock)2693{2694 if (close_lock_file(lock->lk))2695 return -1;2696 return 0;2697}26982699static int commit_ref(struct ref_lock *lock)2700{2701 char *path = get_locked_file_path(lock->lk);2702 struct stat st;27032704 if (!lstat(path, &st) && S_ISDIR(st.st_mode)) {2705 /*2706 * There is a directory at the path we want to rename2707 * the lockfile to. Hopefully it is empty; try to2708 * delete it.2709 */2710 size_t len = strlen(path);2711 struct strbuf sb_path = STRBUF_INIT;27122713 strbuf_attach(&sb_path, path, len, len);27142715 /*2716 * If this fails, commit_lock_file() will also fail2717 * and will report the problem.2718 */2719 remove_empty_directories(&sb_path);2720 strbuf_release(&sb_path);2721 } else {2722 free(path);2723 }27242725 if (commit_lock_file(lock->lk))2726 return -1;2727 return 0;2728}27292730static int open_or_create_logfile(const char *path, void *cb)2731{2732 int *fd = cb;27332734 *fd = open(path, O_APPEND | O_WRONLY | O_CREAT, 0666);2735 return (*fd < 0) ? -1 : 0;2736}27372738/*2739 * Create a reflog for a ref. If force_create = 0, only create the2740 * reflog for certain refs (those for which should_autocreate_reflog2741 * returns non-zero). Otherwise, create it regardless of the reference2742 * name. If the logfile already existed or was created, return 0 and2743 * set *logfd to the file descriptor opened for appending to the file.2744 * If no logfile exists and we decided not to create one, return 0 and2745 * set *logfd to -1. On failure, fill in *err, set *logfd to -1, and2746 * return -1.2747 */2748static int log_ref_setup(const char *refname, int force_create,2749 int *logfd, struct strbuf *err)2750{2751 char *logfile = git_pathdup("logs/%s", refname);27522753 if (force_create || should_autocreate_reflog(refname)) {2754 if (raceproof_create_file(logfile, open_or_create_logfile, logfd)) {2755 if (errno == ENOENT)2756 strbuf_addf(err, "unable to create directory for '%s': "2757 "%s", logfile, strerror(errno));2758 else if (errno == EISDIR)2759 strbuf_addf(err, "there are still logs under '%s'",2760 logfile);2761 else2762 strbuf_addf(err, "unable to append to '%s': %s",2763 logfile, strerror(errno));27642765 goto error;2766 }2767 } else {2768 *logfd = open(logfile, O_APPEND | O_WRONLY, 0666);2769 if (*logfd < 0) {2770 if (errno == ENOENT || errno == EISDIR) {2771 /*2772 * The logfile doesn't already exist,2773 * but that is not an error; it only2774 * means that we won't write log2775 * entries to it.2776 */2777 ;2778 } else {2779 strbuf_addf(err, "unable to append to '%s': %s",2780 logfile, strerror(errno));2781 goto error;2782 }2783 }2784 }27852786 if (*logfd >= 0)2787 adjust_shared_perm(logfile);27882789 free(logfile);2790 return 0;27912792error:2793 free(logfile);2794 return -1;2795}27962797static int files_create_reflog(struct ref_store *ref_store,2798 const char *refname, int force_create,2799 struct strbuf *err)2800{2801 int fd;28022803 /* Check validity (but we don't need the result): */2804 files_downcast(ref_store, 0, "create_reflog");28052806 if (log_ref_setup(refname, force_create, &fd, err))2807 return -1;28082809 if (fd >= 0)2810 close(fd);28112812 return 0;2813}28142815static int log_ref_write_fd(int fd, const unsigned char *old_sha1,2816 const unsigned char *new_sha1,2817 const char *committer, const char *msg)2818{2819 int msglen, written;2820 unsigned maxlen, len;2821 char *logrec;28222823 msglen = msg ? strlen(msg) : 0;2824 maxlen = strlen(committer) + msglen + 100;2825 logrec = xmalloc(maxlen);2826 len = xsnprintf(logrec, maxlen, "%s %s %s\n",2827 sha1_to_hex(old_sha1),2828 sha1_to_hex(new_sha1),2829 committer);2830 if (msglen)2831 len += copy_reflog_msg(logrec + len - 1, msg) - 1;28322833 written = len <= maxlen ? write_in_full(fd, logrec, len) : -1;2834 free(logrec);2835 if (written != len)2836 return -1;28372838 return 0;2839}28402841static int files_log_ref_write(const char *refname, const unsigned char *old_sha1,2842 const unsigned char *new_sha1, const char *msg,2843 int flags, struct strbuf *err)2844{2845 int logfd, result;28462847 if (log_all_ref_updates == LOG_REFS_UNSET)2848 log_all_ref_updates = is_bare_repository() ? LOG_REFS_NONE : LOG_REFS_NORMAL;28492850 result = log_ref_setup(refname, flags & REF_FORCE_CREATE_REFLOG,2851 &logfd, err);28522853 if (result)2854 return result;28552856 if (logfd < 0)2857 return 0;2858 result = log_ref_write_fd(logfd, old_sha1, new_sha1,2859 git_committer_info(0), msg);2860 if (result) {2861 int save_errno = errno;28622863 strbuf_addf(err, "unable to append to '%s': %s",2864 git_path("logs/%s", refname), strerror(save_errno));2865 close(logfd);2866 return -1;2867 }2868 if (close(logfd)) {2869 int save_errno = errno;28702871 strbuf_addf(err, "unable to append to '%s': %s",2872 git_path("logs/%s", refname), strerror(save_errno));2873 return -1;2874 }2875 return 0;2876}28772878/*2879 * Write sha1 into the open lockfile, then close the lockfile. On2880 * errors, rollback the lockfile, fill in *err and2881 * return -1.2882 */2883static int write_ref_to_lockfile(struct ref_lock *lock,2884 const unsigned char *sha1, struct strbuf *err)2885{2886 static char term = '\n';2887 struct object *o;2888 int fd;28892890 o = parse_object(sha1);2891 if (!o) {2892 strbuf_addf(err,2893 "trying to write ref '%s' with nonexistent object %s",2894 lock->ref_name, sha1_to_hex(sha1));2895 unlock_ref(lock);2896 return -1;2897 }2898 if (o->type != OBJ_COMMIT && is_branch(lock->ref_name)) {2899 strbuf_addf(err,2900 "trying to write non-commit object %s to branch '%s'",2901 sha1_to_hex(sha1), lock->ref_name);2902 unlock_ref(lock);2903 return -1;2904 }2905 fd = get_lock_file_fd(lock->lk);2906 if (write_in_full(fd, sha1_to_hex(sha1), 40) != 40 ||2907 write_in_full(fd, &term, 1) != 1 ||2908 close_ref(lock) < 0) {2909 strbuf_addf(err,2910 "couldn't write '%s'", get_lock_file_path(lock->lk));2911 unlock_ref(lock);2912 return -1;2913 }2914 return 0;2915}29162917/*2918 * Commit a change to a loose reference that has already been written2919 * to the loose reference lockfile. Also update the reflogs if2920 * necessary, using the specified lockmsg (which can be NULL).2921 */2922static int commit_ref_update(struct files_ref_store *refs,2923 struct ref_lock *lock,2924 const unsigned char *sha1, const char *logmsg,2925 struct strbuf *err)2926{2927 files_assert_main_repository(refs, "commit_ref_update");29282929 clear_loose_ref_cache(refs);2930 if (files_log_ref_write(lock->ref_name, lock->old_oid.hash, sha1,2931 logmsg, 0, err)) {2932 char *old_msg = strbuf_detach(err, NULL);2933 strbuf_addf(err, "cannot update the ref '%s': %s",2934 lock->ref_name, old_msg);2935 free(old_msg);2936 unlock_ref(lock);2937 return -1;2938 }29392940 if (strcmp(lock->ref_name, "HEAD") != 0) {2941 /*2942 * Special hack: If a branch is updated directly and HEAD2943 * points to it (may happen on the remote side of a push2944 * for example) then logically the HEAD reflog should be2945 * updated too.2946 * A generic solution implies reverse symref information,2947 * but finding all symrefs pointing to the given branch2948 * would be rather costly for this rare event (the direct2949 * update of a branch) to be worth it. So let's cheat and2950 * check with HEAD only which should cover 99% of all usage2951 * scenarios (even 100% of the default ones).2952 */2953 unsigned char head_sha1[20];2954 int head_flag;2955 const char *head_ref;29562957 head_ref = resolve_ref_unsafe("HEAD", RESOLVE_REF_READING,2958 head_sha1, &head_flag);2959 if (head_ref && (head_flag & REF_ISSYMREF) &&2960 !strcmp(head_ref, lock->ref_name)) {2961 struct strbuf log_err = STRBUF_INIT;2962 if (files_log_ref_write("HEAD", lock->old_oid.hash, sha1,2963 logmsg, 0, &log_err)) {2964 error("%s", log_err.buf);2965 strbuf_release(&log_err);2966 }2967 }2968 }29692970 if (commit_ref(lock)) {2971 strbuf_addf(err, "couldn't set '%s'", lock->ref_name);2972 unlock_ref(lock);2973 return -1;2974 }29752976 unlock_ref(lock);2977 return 0;2978}29792980static int create_ref_symlink(struct ref_lock *lock, const char *target)2981{2982 int ret = -1;2983#ifndef NO_SYMLINK_HEAD2984 char *ref_path = get_locked_file_path(lock->lk);2985 unlink(ref_path);2986 ret = symlink(target, ref_path);2987 free(ref_path);29882989 if (ret)2990 fprintf(stderr, "no symlink - falling back to symbolic ref\n");2991#endif2992 return ret;2993}29942995static void update_symref_reflog(struct ref_lock *lock, const char *refname,2996 const char *target, const char *logmsg)2997{2998 struct strbuf err = STRBUF_INIT;2999 unsigned char new_sha1[20];3000 if (logmsg && !read_ref(target, new_sha1) &&3001 files_log_ref_write(refname, lock->old_oid.hash, new_sha1,3002 logmsg, 0, &err)) {3003 error("%s", err.buf);3004 strbuf_release(&err);3005 }3006}30073008static int create_symref_locked(struct ref_lock *lock, const char *refname,3009 const char *target, const char *logmsg)3010{3011 if (prefer_symlink_refs && !create_ref_symlink(lock, target)) {3012 update_symref_reflog(lock, refname, target, logmsg);3013 return 0;3014 }30153016 if (!fdopen_lock_file(lock->lk, "w"))3017 return error("unable to fdopen %s: %s",3018 lock->lk->tempfile.filename.buf, strerror(errno));30193020 update_symref_reflog(lock, refname, target, logmsg);30213022 /* no error check; commit_ref will check ferror */3023 fprintf(lock->lk->tempfile.fp, "ref: %s\n", target);3024 if (commit_ref(lock) < 0)3025 return error("unable to write symref for %s: %s", refname,3026 strerror(errno));3027 return 0;3028}30293030static int files_create_symref(struct ref_store *ref_store,3031 const char *refname, const char *target,3032 const char *logmsg)3033{3034 struct files_ref_store *refs =3035 files_downcast(ref_store, 0, "create_symref");3036 struct strbuf err = STRBUF_INIT;3037 struct ref_lock *lock;3038 int ret;30393040 lock = lock_ref_sha1_basic(refs, refname, NULL,3041 NULL, NULL, REF_NODEREF, NULL,3042 &err);3043 if (!lock) {3044 error("%s", err.buf);3045 strbuf_release(&err);3046 return -1;3047 }30483049 ret = create_symref_locked(lock, refname, target, logmsg);3050 unlock_ref(lock);3051 return ret;3052}30533054int set_worktree_head_symref(const char *gitdir, const char *target, const char *logmsg)3055{3056 static struct lock_file head_lock;3057 struct ref_lock *lock;3058 struct strbuf head_path = STRBUF_INIT;3059 const char *head_rel;3060 int ret;30613062 strbuf_addf(&head_path, "%s/HEAD", absolute_path(gitdir));3063 if (hold_lock_file_for_update(&head_lock, head_path.buf,3064 LOCK_NO_DEREF) < 0) {3065 struct strbuf err = STRBUF_INIT;3066 unable_to_lock_message(head_path.buf, errno, &err);3067 error("%s", err.buf);3068 strbuf_release(&err);3069 strbuf_release(&head_path);3070 return -1;3071 }30723073 /* head_rel will be "HEAD" for the main tree, "worktrees/wt/HEAD" for3074 linked trees */3075 head_rel = remove_leading_path(head_path.buf,3076 absolute_path(get_git_common_dir()));3077 /* to make use of create_symref_locked(), initialize ref_lock */3078 lock = xcalloc(1, sizeof(struct ref_lock));3079 lock->lk = &head_lock;3080 lock->ref_name = xstrdup(head_rel);30813082 ret = create_symref_locked(lock, head_rel, target, logmsg);30833084 unlock_ref(lock); /* will free lock */3085 strbuf_release(&head_path);3086 return ret;3087}30883089static int files_reflog_exists(struct ref_store *ref_store,3090 const char *refname)3091{3092 struct stat st;30933094 /* Check validity (but we don't need the result): */3095 files_downcast(ref_store, 0, "reflog_exists");30963097 return !lstat(git_path("logs/%s", refname), &st) &&3098 S_ISREG(st.st_mode);3099}31003101static int files_delete_reflog(struct ref_store *ref_store,3102 const char *refname)3103{3104 /* Check validity (but we don't need the result): */3105 files_downcast(ref_store, 0, "delete_reflog");31063107 return remove_path(git_path("logs/%s", refname));3108}31093110static int show_one_reflog_ent(struct strbuf *sb, each_reflog_ent_fn fn, void *cb_data)3111{3112 struct object_id ooid, noid;3113 char *email_end, *message;3114 unsigned long timestamp;3115 int tz;3116 const char *p = sb->buf;31173118 /* old SP new SP name <email> SP time TAB msg LF */3119 if (!sb->len || sb->buf[sb->len - 1] != '\n' ||3120 parse_oid_hex(p, &ooid, &p) || *p++ != ' ' ||3121 parse_oid_hex(p, &noid, &p) || *p++ != ' ' ||3122 !(email_end = strchr(p, '>')) ||3123 email_end[1] != ' ' ||3124 !(timestamp = strtoul(email_end + 2, &message, 10)) ||3125 !message || message[0] != ' ' ||3126 (message[1] != '+' && message[1] != '-') ||3127 !isdigit(message[2]) || !isdigit(message[3]) ||3128 !isdigit(message[4]) || !isdigit(message[5]))3129 return 0; /* corrupt? */3130 email_end[1] = '\0';3131 tz = strtol(message + 1, NULL, 10);3132 if (message[6] != '\t')3133 message += 6;3134 else3135 message += 7;3136 return fn(&ooid, &noid, p, timestamp, tz, message, cb_data);3137}31383139static char *find_beginning_of_line(char *bob, char *scan)3140{3141 while (bob < scan && *(--scan) != '\n')3142 ; /* keep scanning backwards */3143 /*3144 * Return either beginning of the buffer, or LF at the end of3145 * the previous line.3146 */3147 return scan;3148}31493150static int files_for_each_reflog_ent_reverse(struct ref_store *ref_store,3151 const char *refname,3152 each_reflog_ent_fn fn,3153 void *cb_data)3154{3155 struct strbuf sb = STRBUF_INIT;3156 FILE *logfp;3157 long pos;3158 int ret = 0, at_tail = 1;31593160 /* Check validity (but we don't need the result): */3161 files_downcast(ref_store, 0, "for_each_reflog_ent_reverse");31623163 logfp = fopen(git_path("logs/%s", refname), "r");3164 if (!logfp)3165 return -1;31663167 /* Jump to the end */3168 if (fseek(logfp, 0, SEEK_END) < 0)3169 return error("cannot seek back reflog for %s: %s",3170 refname, strerror(errno));3171 pos = ftell(logfp);3172 while (!ret && 0 < pos) {3173 int cnt;3174 size_t nread;3175 char buf[BUFSIZ];3176 char *endp, *scanp;31773178 /* Fill next block from the end */3179 cnt = (sizeof(buf) < pos) ? sizeof(buf) : pos;3180 if (fseek(logfp, pos - cnt, SEEK_SET))3181 return error("cannot seek back reflog for %s: %s",3182 refname, strerror(errno));3183 nread = fread(buf, cnt, 1, logfp);3184 if (nread != 1)3185 return error("cannot read %d bytes from reflog for %s: %s",3186 cnt, refname, strerror(errno));3187 pos -= cnt;31883189 scanp = endp = buf + cnt;3190 if (at_tail && scanp[-1] == '\n')3191 /* Looking at the final LF at the end of the file */3192 scanp--;3193 at_tail = 0;31943195 while (buf < scanp) {3196 /*3197 * terminating LF of the previous line, or the beginning3198 * of the buffer.3199 */3200 char *bp;32013202 bp = find_beginning_of_line(buf, scanp);32033204 if (*bp == '\n') {3205 /*3206 * The newline is the end of the previous line,3207 * so we know we have complete line starting3208 * at (bp + 1). Prefix it onto any prior data3209 * we collected for the line and process it.3210 */3211 strbuf_splice(&sb, 0, 0, bp + 1, endp - (bp + 1));3212 scanp = bp;3213 endp = bp + 1;3214 ret = show_one_reflog_ent(&sb, fn, cb_data);3215 strbuf_reset(&sb);3216 if (ret)3217 break;3218 } else if (!pos) {3219 /*3220 * We are at the start of the buffer, and the3221 * start of the file; there is no previous3222 * line, and we have everything for this one.3223 * Process it, and we can end the loop.3224 */3225 strbuf_splice(&sb, 0, 0, buf, endp - buf);3226 ret = show_one_reflog_ent(&sb, fn, cb_data);3227 strbuf_reset(&sb);3228 break;3229 }32303231 if (bp == buf) {3232 /*3233 * We are at the start of the buffer, and there3234 * is more file to read backwards. Which means3235 * we are in the middle of a line. Note that we3236 * may get here even if *bp was a newline; that3237 * just means we are at the exact end of the3238 * previous line, rather than some spot in the3239 * middle.3240 *3241 * Save away what we have to be combined with3242 * the data from the next read.3243 */3244 strbuf_splice(&sb, 0, 0, buf, endp - buf);3245 break;3246 }3247 }32483249 }3250 if (!ret && sb.len)3251 die("BUG: reverse reflog parser had leftover data");32523253 fclose(logfp);3254 strbuf_release(&sb);3255 return ret;3256}32573258static int files_for_each_reflog_ent(struct ref_store *ref_store,3259 const char *refname,3260 each_reflog_ent_fn fn, void *cb_data)3261{3262 FILE *logfp;3263 struct strbuf sb = STRBUF_INIT;3264 int ret = 0;32653266 /* Check validity (but we don't need the result): */3267 files_downcast(ref_store, 0, "for_each_reflog_ent");32683269 logfp = fopen(git_path("logs/%s", refname), "r");3270 if (!logfp)3271 return -1;32723273 while (!ret && !strbuf_getwholeline(&sb, logfp, '\n'))3274 ret = show_one_reflog_ent(&sb, fn, cb_data);3275 fclose(logfp);3276 strbuf_release(&sb);3277 return ret;3278}32793280struct files_reflog_iterator {3281 struct ref_iterator base;32823283 struct dir_iterator *dir_iterator;3284 struct object_id oid;3285};32863287static int files_reflog_iterator_advance(struct ref_iterator *ref_iterator)3288{3289 struct files_reflog_iterator *iter =3290 (struct files_reflog_iterator *)ref_iterator;3291 struct dir_iterator *diter = iter->dir_iterator;3292 int ok;32933294 while ((ok = dir_iterator_advance(diter)) == ITER_OK) {3295 int flags;32963297 if (!S_ISREG(diter->st.st_mode))3298 continue;3299 if (diter->basename[0] == '.')3300 continue;3301 if (ends_with(diter->basename, ".lock"))3302 continue;33033304 if (read_ref_full(diter->relative_path, 0,3305 iter->oid.hash, &flags)) {3306 error("bad ref for %s", diter->path.buf);3307 continue;3308 }33093310 iter->base.refname = diter->relative_path;3311 iter->base.oid = &iter->oid;3312 iter->base.flags = flags;3313 return ITER_OK;3314 }33153316 iter->dir_iterator = NULL;3317 if (ref_iterator_abort(ref_iterator) == ITER_ERROR)3318 ok = ITER_ERROR;3319 return ok;3320}33213322static int files_reflog_iterator_peel(struct ref_iterator *ref_iterator,3323 struct object_id *peeled)3324{3325 die("BUG: ref_iterator_peel() called for reflog_iterator");3326}33273328static int files_reflog_iterator_abort(struct ref_iterator *ref_iterator)3329{3330 struct files_reflog_iterator *iter =3331 (struct files_reflog_iterator *)ref_iterator;3332 int ok = ITER_DONE;33333334 if (iter->dir_iterator)3335 ok = dir_iterator_abort(iter->dir_iterator);33363337 base_ref_iterator_free(ref_iterator);3338 return ok;3339}33403341static struct ref_iterator_vtable files_reflog_iterator_vtable = {3342 files_reflog_iterator_advance,3343 files_reflog_iterator_peel,3344 files_reflog_iterator_abort3345};33463347static struct ref_iterator *files_reflog_iterator_begin(struct ref_store *ref_store)3348{3349 struct files_reflog_iterator *iter = xcalloc(1, sizeof(*iter));3350 struct ref_iterator *ref_iterator = &iter->base;33513352 /* Check validity (but we don't need the result): */3353 files_downcast(ref_store, 0, "reflog_iterator_begin");33543355 base_ref_iterator_init(ref_iterator, &files_reflog_iterator_vtable);3356 iter->dir_iterator = dir_iterator_begin(git_path("logs"));3357 return ref_iterator;3358}33593360static int ref_update_reject_duplicates(struct string_list *refnames,3361 struct strbuf *err)3362{3363 int i, n = refnames->nr;33643365 assert(err);33663367 for (i = 1; i < n; i++)3368 if (!strcmp(refnames->items[i - 1].string, refnames->items[i].string)) {3369 strbuf_addf(err,3370 "multiple updates for ref '%s' not allowed.",3371 refnames->items[i].string);3372 return 1;3373 }3374 return 0;3375}33763377/*3378 * If update is a direct update of head_ref (the reference pointed to3379 * by HEAD), then add an extra REF_LOG_ONLY update for HEAD.3380 */3381static int split_head_update(struct ref_update *update,3382 struct ref_transaction *transaction,3383 const char *head_ref,3384 struct string_list *affected_refnames,3385 struct strbuf *err)3386{3387 struct string_list_item *item;3388 struct ref_update *new_update;33893390 if ((update->flags & REF_LOG_ONLY) ||3391 (update->flags & REF_ISPRUNING) ||3392 (update->flags & REF_UPDATE_VIA_HEAD))3393 return 0;33943395 if (strcmp(update->refname, head_ref))3396 return 0;33973398 /*3399 * First make sure that HEAD is not already in the3400 * transaction. This insertion is O(N) in the transaction3401 * size, but it happens at most once per transaction.3402 */3403 item = string_list_insert(affected_refnames, "HEAD");3404 if (item->util) {3405 /* An entry already existed */3406 strbuf_addf(err,3407 "multiple updates for 'HEAD' (including one "3408 "via its referent '%s') are not allowed",3409 update->refname);3410 return TRANSACTION_NAME_CONFLICT;3411 }34123413 new_update = ref_transaction_add_update(3414 transaction, "HEAD",3415 update->flags | REF_LOG_ONLY | REF_NODEREF,3416 update->new_sha1, update->old_sha1,3417 update->msg);34183419 item->util = new_update;34203421 return 0;3422}34233424/*3425 * update is for a symref that points at referent and doesn't have3426 * REF_NODEREF set. Split it into two updates:3427 * - The original update, but with REF_LOG_ONLY and REF_NODEREF set3428 * - A new, separate update for the referent reference3429 * Note that the new update will itself be subject to splitting when3430 * the iteration gets to it.3431 */3432static int split_symref_update(struct files_ref_store *refs,3433 struct ref_update *update,3434 const char *referent,3435 struct ref_transaction *transaction,3436 struct string_list *affected_refnames,3437 struct strbuf *err)3438{3439 struct string_list_item *item;3440 struct ref_update *new_update;3441 unsigned int new_flags;34423443 /*3444 * First make sure that referent is not already in the3445 * transaction. This insertion is O(N) in the transaction3446 * size, but it happens at most once per symref in a3447 * transaction.3448 */3449 item = string_list_insert(affected_refnames, referent);3450 if (item->util) {3451 /* An entry already existed */3452 strbuf_addf(err,3453 "multiple updates for '%s' (including one "3454 "via symref '%s') are not allowed",3455 referent, update->refname);3456 return TRANSACTION_NAME_CONFLICT;3457 }34583459 new_flags = update->flags;3460 if (!strcmp(update->refname, "HEAD")) {3461 /*3462 * Record that the new update came via HEAD, so that3463 * when we process it, split_head_update() doesn't try3464 * to add another reflog update for HEAD. Note that3465 * this bit will be propagated if the new_update3466 * itself needs to be split.3467 */3468 new_flags |= REF_UPDATE_VIA_HEAD;3469 }34703471 new_update = ref_transaction_add_update(3472 transaction, referent, new_flags,3473 update->new_sha1, update->old_sha1,3474 update->msg);34753476 new_update->parent_update = update;34773478 /*3479 * Change the symbolic ref update to log only. Also, it3480 * doesn't need to check its old SHA-1 value, as that will be3481 * done when new_update is processed.3482 */3483 update->flags |= REF_LOG_ONLY | REF_NODEREF;3484 update->flags &= ~REF_HAVE_OLD;34853486 item->util = new_update;34873488 return 0;3489}34903491/*3492 * Return the refname under which update was originally requested.3493 */3494static const char *original_update_refname(struct ref_update *update)3495{3496 while (update->parent_update)3497 update = update->parent_update;34983499 return update->refname;3500}35013502/*3503 * Check whether the REF_HAVE_OLD and old_oid values stored in update3504 * are consistent with oid, which is the reference's current value. If3505 * everything is OK, return 0; otherwise, write an error message to3506 * err and return -1.3507 */3508static int check_old_oid(struct ref_update *update, struct object_id *oid,3509 struct strbuf *err)3510{3511 if (!(update->flags & REF_HAVE_OLD) ||3512 !hashcmp(oid->hash, update->old_sha1))3513 return 0;35143515 if (is_null_sha1(update->old_sha1))3516 strbuf_addf(err, "cannot lock ref '%s': "3517 "reference already exists",3518 original_update_refname(update));3519 else if (is_null_oid(oid))3520 strbuf_addf(err, "cannot lock ref '%s': "3521 "reference is missing but expected %s",3522 original_update_refname(update),3523 sha1_to_hex(update->old_sha1));3524 else3525 strbuf_addf(err, "cannot lock ref '%s': "3526 "is at %s but expected %s",3527 original_update_refname(update),3528 oid_to_hex(oid),3529 sha1_to_hex(update->old_sha1));35303531 return -1;3532}35333534/*3535 * Prepare for carrying out update:3536 * - Lock the reference referred to by update.3537 * - Read the reference under lock.3538 * - Check that its old SHA-1 value (if specified) is correct, and in3539 * any case record it in update->lock->old_oid for later use when3540 * writing the reflog.3541 * - If it is a symref update without REF_NODEREF, split it up into a3542 * REF_LOG_ONLY update of the symref and add a separate update for3543 * the referent to transaction.3544 * - If it is an update of head_ref, add a corresponding REF_LOG_ONLY3545 * update of HEAD.3546 */3547static int lock_ref_for_update(struct files_ref_store *refs,3548 struct ref_update *update,3549 struct ref_transaction *transaction,3550 const char *head_ref,3551 struct string_list *affected_refnames,3552 struct strbuf *err)3553{3554 struct strbuf referent = STRBUF_INIT;3555 int mustexist = (update->flags & REF_HAVE_OLD) &&3556 !is_null_sha1(update->old_sha1);3557 int ret;3558 struct ref_lock *lock;35593560 files_assert_main_repository(refs, "lock_ref_for_update");35613562 if ((update->flags & REF_HAVE_NEW) && is_null_sha1(update->new_sha1))3563 update->flags |= REF_DELETING;35643565 if (head_ref) {3566 ret = split_head_update(update, transaction, head_ref,3567 affected_refnames, err);3568 if (ret)3569 return ret;3570 }35713572 ret = lock_raw_ref(refs, update->refname, mustexist,3573 affected_refnames, NULL,3574 &lock, &referent,3575 &update->type, err);3576 if (ret) {3577 char *reason;35783579 reason = strbuf_detach(err, NULL);3580 strbuf_addf(err, "cannot lock ref '%s': %s",3581 original_update_refname(update), reason);3582 free(reason);3583 return ret;3584 }35853586 update->backend_data = lock;35873588 if (update->type & REF_ISSYMREF) {3589 if (update->flags & REF_NODEREF) {3590 /*3591 * We won't be reading the referent as part of3592 * the transaction, so we have to read it here3593 * to record and possibly check old_sha1:3594 */3595 if (read_ref_full(referent.buf, 0,3596 lock->old_oid.hash, NULL)) {3597 if (update->flags & REF_HAVE_OLD) {3598 strbuf_addf(err, "cannot lock ref '%s': "3599 "error reading reference",3600 original_update_refname(update));3601 return -1;3602 }3603 } else if (check_old_oid(update, &lock->old_oid, err)) {3604 return TRANSACTION_GENERIC_ERROR;3605 }3606 } else {3607 /*3608 * Create a new update for the reference this3609 * symref is pointing at. Also, we will record3610 * and verify old_sha1 for this update as part3611 * of processing the split-off update, so we3612 * don't have to do it here.3613 */3614 ret = split_symref_update(refs, update,3615 referent.buf, transaction,3616 affected_refnames, err);3617 if (ret)3618 return ret;3619 }3620 } else {3621 struct ref_update *parent_update;36223623 if (check_old_oid(update, &lock->old_oid, err))3624 return TRANSACTION_GENERIC_ERROR;36253626 /*3627 * If this update is happening indirectly because of a3628 * symref update, record the old SHA-1 in the parent3629 * update:3630 */3631 for (parent_update = update->parent_update;3632 parent_update;3633 parent_update = parent_update->parent_update) {3634 struct ref_lock *parent_lock = parent_update->backend_data;3635 oidcpy(&parent_lock->old_oid, &lock->old_oid);3636 }3637 }36383639 if ((update->flags & REF_HAVE_NEW) &&3640 !(update->flags & REF_DELETING) &&3641 !(update->flags & REF_LOG_ONLY)) {3642 if (!(update->type & REF_ISSYMREF) &&3643 !hashcmp(lock->old_oid.hash, update->new_sha1)) {3644 /*3645 * The reference already has the desired3646 * value, so we don't need to write it.3647 */3648 } else if (write_ref_to_lockfile(lock, update->new_sha1,3649 err)) {3650 char *write_err = strbuf_detach(err, NULL);36513652 /*3653 * The lock was freed upon failure of3654 * write_ref_to_lockfile():3655 */3656 update->backend_data = NULL;3657 strbuf_addf(err,3658 "cannot update ref '%s': %s",3659 update->refname, write_err);3660 free(write_err);3661 return TRANSACTION_GENERIC_ERROR;3662 } else {3663 update->flags |= REF_NEEDS_COMMIT;3664 }3665 }3666 if (!(update->flags & REF_NEEDS_COMMIT)) {3667 /*3668 * We didn't call write_ref_to_lockfile(), so3669 * the lockfile is still open. Close it to3670 * free up the file descriptor:3671 */3672 if (close_ref(lock)) {3673 strbuf_addf(err, "couldn't close '%s.lock'",3674 update->refname);3675 return TRANSACTION_GENERIC_ERROR;3676 }3677 }3678 return 0;3679}36803681static int files_transaction_commit(struct ref_store *ref_store,3682 struct ref_transaction *transaction,3683 struct strbuf *err)3684{3685 struct files_ref_store *refs =3686 files_downcast(ref_store, 0, "ref_transaction_commit");3687 int ret = 0, i;3688 struct string_list refs_to_delete = STRING_LIST_INIT_NODUP;3689 struct string_list_item *ref_to_delete;3690 struct string_list affected_refnames = STRING_LIST_INIT_NODUP;3691 char *head_ref = NULL;3692 int head_type;3693 struct object_id head_oid;36943695 assert(err);36963697 if (transaction->state != REF_TRANSACTION_OPEN)3698 die("BUG: commit called for transaction that is not open");36993700 if (!transaction->nr) {3701 transaction->state = REF_TRANSACTION_CLOSED;3702 return 0;3703 }37043705 /*3706 * Fail if a refname appears more than once in the3707 * transaction. (If we end up splitting up any updates using3708 * split_symref_update() or split_head_update(), those3709 * functions will check that the new updates don't have the3710 * same refname as any existing ones.)3711 */3712 for (i = 0; i < transaction->nr; i++) {3713 struct ref_update *update = transaction->updates[i];3714 struct string_list_item *item =3715 string_list_append(&affected_refnames, update->refname);37163717 /*3718 * We store a pointer to update in item->util, but at3719 * the moment we never use the value of this field3720 * except to check whether it is non-NULL.3721 */3722 item->util = update;3723 }3724 string_list_sort(&affected_refnames);3725 if (ref_update_reject_duplicates(&affected_refnames, err)) {3726 ret = TRANSACTION_GENERIC_ERROR;3727 goto cleanup;3728 }37293730 /*3731 * Special hack: If a branch is updated directly and HEAD3732 * points to it (may happen on the remote side of a push3733 * for example) then logically the HEAD reflog should be3734 * updated too.3735 *3736 * A generic solution would require reverse symref lookups,3737 * but finding all symrefs pointing to a given branch would be3738 * rather costly for this rare event (the direct update of a3739 * branch) to be worth it. So let's cheat and check with HEAD3740 * only, which should cover 99% of all usage scenarios (even3741 * 100% of the default ones).3742 *3743 * So if HEAD is a symbolic reference, then record the name of3744 * the reference that it points to. If we see an update of3745 * head_ref within the transaction, then split_head_update()3746 * arranges for the reflog of HEAD to be updated, too.3747 */3748 head_ref = resolve_refdup("HEAD", RESOLVE_REF_NO_RECURSE,3749 head_oid.hash, &head_type);37503751 if (head_ref && !(head_type & REF_ISSYMREF)) {3752 free(head_ref);3753 head_ref = NULL;3754 }37553756 /*3757 * Acquire all locks, verify old values if provided, check3758 * that new values are valid, and write new values to the3759 * lockfiles, ready to be activated. Only keep one lockfile3760 * open at a time to avoid running out of file descriptors.3761 */3762 for (i = 0; i < transaction->nr; i++) {3763 struct ref_update *update = transaction->updates[i];37643765 ret = lock_ref_for_update(refs, update, transaction,3766 head_ref, &affected_refnames, err);3767 if (ret)3768 goto cleanup;3769 }37703771 /* Perform updates first so live commits remain referenced */3772 for (i = 0; i < transaction->nr; i++) {3773 struct ref_update *update = transaction->updates[i];3774 struct ref_lock *lock = update->backend_data;37753776 if (update->flags & REF_NEEDS_COMMIT ||3777 update->flags & REF_LOG_ONLY) {3778 if (files_log_ref_write(lock->ref_name,3779 lock->old_oid.hash,3780 update->new_sha1,3781 update->msg, update->flags,3782 err)) {3783 char *old_msg = strbuf_detach(err, NULL);37843785 strbuf_addf(err, "cannot update the ref '%s': %s",3786 lock->ref_name, old_msg);3787 free(old_msg);3788 unlock_ref(lock);3789 update->backend_data = NULL;3790 ret = TRANSACTION_GENERIC_ERROR;3791 goto cleanup;3792 }3793 }3794 if (update->flags & REF_NEEDS_COMMIT) {3795 clear_loose_ref_cache(refs);3796 if (commit_ref(lock)) {3797 strbuf_addf(err, "couldn't set '%s'", lock->ref_name);3798 unlock_ref(lock);3799 update->backend_data = NULL;3800 ret = TRANSACTION_GENERIC_ERROR;3801 goto cleanup;3802 }3803 }3804 }3805 /* Perform deletes now that updates are safely completed */3806 for (i = 0; i < transaction->nr; i++) {3807 struct ref_update *update = transaction->updates[i];3808 struct ref_lock *lock = update->backend_data;38093810 if (update->flags & REF_DELETING &&3811 !(update->flags & REF_LOG_ONLY)) {3812 if (!(update->type & REF_ISPACKED) ||3813 update->type & REF_ISSYMREF) {3814 /* It is a loose reference. */3815 if (unlink_or_msg(git_path("%s", lock->ref_name), err)) {3816 ret = TRANSACTION_GENERIC_ERROR;3817 goto cleanup;3818 }3819 update->flags |= REF_DELETED_LOOSE;3820 }38213822 if (!(update->flags & REF_ISPRUNING))3823 string_list_append(&refs_to_delete,3824 lock->ref_name);3825 }3826 }38273828 if (repack_without_refs(refs, &refs_to_delete, err)) {3829 ret = TRANSACTION_GENERIC_ERROR;3830 goto cleanup;3831 }38323833 /* Delete the reflogs of any references that were deleted: */3834 for_each_string_list_item(ref_to_delete, &refs_to_delete) {3835 if (!unlink_or_warn(git_path("logs/%s", ref_to_delete->string)))3836 try_remove_empty_parents(ref_to_delete->string,3837 REMOVE_EMPTY_PARENTS_REFLOG);3838 }38393840 clear_loose_ref_cache(refs);38413842cleanup:3843 transaction->state = REF_TRANSACTION_CLOSED;38443845 for (i = 0; i < transaction->nr; i++) {3846 struct ref_update *update = transaction->updates[i];3847 struct ref_lock *lock = update->backend_data;38483849 if (lock)3850 unlock_ref(lock);38513852 if (update->flags & REF_DELETED_LOOSE) {3853 /*3854 * The loose reference was deleted. Delete any3855 * empty parent directories. (Note that this3856 * can only work because we have already3857 * removed the lockfile.)3858 */3859 try_remove_empty_parents(update->refname,3860 REMOVE_EMPTY_PARENTS_REF);3861 }3862 }38633864 string_list_clear(&refs_to_delete, 0);3865 free(head_ref);3866 string_list_clear(&affected_refnames, 0);38673868 return ret;3869}38703871static int ref_present(const char *refname,3872 const struct object_id *oid, int flags, void *cb_data)3873{3874 struct string_list *affected_refnames = cb_data;38753876 return string_list_has_string(affected_refnames, refname);3877}38783879static int files_initial_transaction_commit(struct ref_store *ref_store,3880 struct ref_transaction *transaction,3881 struct strbuf *err)3882{3883 struct files_ref_store *refs =3884 files_downcast(ref_store, 0, "initial_ref_transaction_commit");3885 int ret = 0, i;3886 struct string_list affected_refnames = STRING_LIST_INIT_NODUP;38873888 assert(err);38893890 if (transaction->state != REF_TRANSACTION_OPEN)3891 die("BUG: commit called for transaction that is not open");38923893 /* Fail if a refname appears more than once in the transaction: */3894 for (i = 0; i < transaction->nr; i++)3895 string_list_append(&affected_refnames,3896 transaction->updates[i]->refname);3897 string_list_sort(&affected_refnames);3898 if (ref_update_reject_duplicates(&affected_refnames, err)) {3899 ret = TRANSACTION_GENERIC_ERROR;3900 goto cleanup;3901 }39023903 /*3904 * It's really undefined to call this function in an active3905 * repository or when there are existing references: we are3906 * only locking and changing packed-refs, so (1) any3907 * simultaneous processes might try to change a reference at3908 * the same time we do, and (2) any existing loose versions of3909 * the references that we are setting would have precedence3910 * over our values. But some remote helpers create the remote3911 * "HEAD" and "master" branches before calling this function,3912 * so here we really only check that none of the references3913 * that we are creating already exists.3914 */3915 if (for_each_rawref(ref_present, &affected_refnames))3916 die("BUG: initial ref transaction called with existing refs");39173918 for (i = 0; i < transaction->nr; i++) {3919 struct ref_update *update = transaction->updates[i];39203921 if ((update->flags & REF_HAVE_OLD) &&3922 !is_null_sha1(update->old_sha1))3923 die("BUG: initial ref transaction with old_sha1 set");3924 if (verify_refname_available(update->refname,3925 &affected_refnames, NULL,3926 err)) {3927 ret = TRANSACTION_NAME_CONFLICT;3928 goto cleanup;3929 }3930 }39313932 if (lock_packed_refs(refs, 0)) {3933 strbuf_addf(err, "unable to lock packed-refs file: %s",3934 strerror(errno));3935 ret = TRANSACTION_GENERIC_ERROR;3936 goto cleanup;3937 }39383939 for (i = 0; i < transaction->nr; i++) {3940 struct ref_update *update = transaction->updates[i];39413942 if ((update->flags & REF_HAVE_NEW) &&3943 !is_null_sha1(update->new_sha1))3944 add_packed_ref(refs, update->refname, update->new_sha1);3945 }39463947 if (commit_packed_refs(refs)) {3948 strbuf_addf(err, "unable to commit packed-refs file: %s",3949 strerror(errno));3950 ret = TRANSACTION_GENERIC_ERROR;3951 goto cleanup;3952 }39533954cleanup:3955 transaction->state = REF_TRANSACTION_CLOSED;3956 string_list_clear(&affected_refnames, 0);3957 return ret;3958}39593960struct expire_reflog_cb {3961 unsigned int flags;3962 reflog_expiry_should_prune_fn *should_prune_fn;3963 void *policy_cb;3964 FILE *newlog;3965 struct object_id last_kept_oid;3966};39673968static int expire_reflog_ent(struct object_id *ooid, struct object_id *noid,3969 const char *email, unsigned long timestamp, int tz,3970 const char *message, void *cb_data)3971{3972 struct expire_reflog_cb *cb = cb_data;3973 struct expire_reflog_policy_cb *policy_cb = cb->policy_cb;39743975 if (cb->flags & EXPIRE_REFLOGS_REWRITE)3976 ooid = &cb->last_kept_oid;39773978 if ((*cb->should_prune_fn)(ooid->hash, noid->hash, email, timestamp, tz,3979 message, policy_cb)) {3980 if (!cb->newlog)3981 printf("would prune %s", message);3982 else if (cb->flags & EXPIRE_REFLOGS_VERBOSE)3983 printf("prune %s", message);3984 } else {3985 if (cb->newlog) {3986 fprintf(cb->newlog, "%s %s %s %lu %+05d\t%s",3987 oid_to_hex(ooid), oid_to_hex(noid),3988 email, timestamp, tz, message);3989 oidcpy(&cb->last_kept_oid, noid);3990 }3991 if (cb->flags & EXPIRE_REFLOGS_VERBOSE)3992 printf("keep %s", message);3993 }3994 return 0;3995}39963997static int files_reflog_expire(struct ref_store *ref_store,3998 const char *refname, const unsigned char *sha1,3999 unsigned int flags,4000 reflog_expiry_prepare_fn prepare_fn,4001 reflog_expiry_should_prune_fn should_prune_fn,4002 reflog_expiry_cleanup_fn cleanup_fn,4003 void *policy_cb_data)4004{4005 struct files_ref_store *refs =4006 files_downcast(ref_store, 0, "reflog_expire");4007 static struct lock_file reflog_lock;4008 struct expire_reflog_cb cb;4009 struct ref_lock *lock;4010 char *log_file;4011 int status = 0;4012 int type;4013 struct strbuf err = STRBUF_INIT;40144015 memset(&cb, 0, sizeof(cb));4016 cb.flags = flags;4017 cb.policy_cb = policy_cb_data;4018 cb.should_prune_fn = should_prune_fn;40194020 /*4021 * The reflog file is locked by holding the lock on the4022 * reference itself, plus we might need to update the4023 * reference if --updateref was specified:4024 */4025 lock = lock_ref_sha1_basic(refs, refname, sha1,4026 NULL, NULL, REF_NODEREF,4027 &type, &err);4028 if (!lock) {4029 error("cannot lock ref '%s': %s", refname, err.buf);4030 strbuf_release(&err);4031 return -1;4032 }4033 if (!reflog_exists(refname)) {4034 unlock_ref(lock);4035 return 0;4036 }40374038 log_file = git_pathdup("logs/%s", refname);4039 if (!(flags & EXPIRE_REFLOGS_DRY_RUN)) {4040 /*4041 * Even though holding $GIT_DIR/logs/$reflog.lock has4042 * no locking implications, we use the lock_file4043 * machinery here anyway because it does a lot of the4044 * work we need, including cleaning up if the program4045 * exits unexpectedly.4046 */4047 if (hold_lock_file_for_update(&reflog_lock, log_file, 0) < 0) {4048 struct strbuf err = STRBUF_INIT;4049 unable_to_lock_message(log_file, errno, &err);4050 error("%s", err.buf);4051 strbuf_release(&err);4052 goto failure;4053 }4054 cb.newlog = fdopen_lock_file(&reflog_lock, "w");4055 if (!cb.newlog) {4056 error("cannot fdopen %s (%s)",4057 get_lock_file_path(&reflog_lock), strerror(errno));4058 goto failure;4059 }4060 }40614062 (*prepare_fn)(refname, sha1, cb.policy_cb);4063 for_each_reflog_ent(refname, expire_reflog_ent, &cb);4064 (*cleanup_fn)(cb.policy_cb);40654066 if (!(flags & EXPIRE_REFLOGS_DRY_RUN)) {4067 /*4068 * It doesn't make sense to adjust a reference pointed4069 * to by a symbolic ref based on expiring entries in4070 * the symbolic reference's reflog. Nor can we update4071 * a reference if there are no remaining reflog4072 * entries.4073 */4074 int update = (flags & EXPIRE_REFLOGS_UPDATE_REF) &&4075 !(type & REF_ISSYMREF) &&4076 !is_null_oid(&cb.last_kept_oid);40774078 if (close_lock_file(&reflog_lock)) {4079 status |= error("couldn't write %s: %s", log_file,4080 strerror(errno));4081 } else if (update &&4082 (write_in_full(get_lock_file_fd(lock->lk),4083 oid_to_hex(&cb.last_kept_oid), GIT_SHA1_HEXSZ) != GIT_SHA1_HEXSZ ||4084 write_str_in_full(get_lock_file_fd(lock->lk), "\n") != 1 ||4085 close_ref(lock) < 0)) {4086 status |= error("couldn't write %s",4087 get_lock_file_path(lock->lk));4088 rollback_lock_file(&reflog_lock);4089 } else if (commit_lock_file(&reflog_lock)) {4090 status |= error("unable to write reflog '%s' (%s)",4091 log_file, strerror(errno));4092 } else if (update && commit_ref(lock)) {4093 status |= error("couldn't set %s", lock->ref_name);4094 }4095 }4096 free(log_file);4097 unlock_ref(lock);4098 return status;40994100 failure:4101 rollback_lock_file(&reflog_lock);4102 free(log_file);4103 unlock_ref(lock);4104 return -1;4105}41064107static int files_init_db(struct ref_store *ref_store, struct strbuf *err)4108{4109 /* Check validity (but we don't need the result): */4110 files_downcast(ref_store, 0, "init_db");41114112 /*4113 * Create .git/refs/{heads,tags}4114 */4115 safe_create_dir(git_path("refs/heads"), 1);4116 safe_create_dir(git_path("refs/tags"), 1);4117 return 0;4118}41194120struct ref_storage_be refs_be_files = {4121 NULL,4122 "files",4123 files_ref_store_create,4124 files_init_db,4125 files_transaction_commit,4126 files_initial_transaction_commit,41274128 files_pack_refs,4129 files_peel_ref,4130 files_create_symref,4131 files_delete_refs,4132 files_rename_ref,41334134 files_ref_iterator_begin,4135 files_read_raw_ref,4136 files_verify_refname_available,41374138 files_reflog_iterator_begin,4139 files_for_each_reflog_ent,4140 files_for_each_reflog_ent_reverse,4141 files_reflog_exists,4142 files_create_reflog,4143 files_delete_reflog,4144 files_reflog_expire4145};