refs.c 76.0 KB
Newer Older
D
Daniel Barkalow 已提交
1
#include "cache.h"
2
#include "refs.h"
3 4
#include "object.h"
#include "tag.h"
5
#include "dir.h"
D
Daniel Barkalow 已提交
6

7 8 9 10 11 12 13 14 15 16 17
/*
 * Make sure "ref" is something reasonable to have under ".git/refs/";
 * We do not like it if:
 *
 * - any path component of it begins with ".", or
 * - it has double dots "..", or
 * - it has ASCII control character, "~", "^", ":" or SP, anywhere, or
 * - it ends with a "/".
 * - it ends with ".lock"
 * - it contains a "\" (backslash)
 */
18

19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
/* Return true iff ch is not allowed in reference names. */
static inline int bad_ref_char(int ch)
{
	if (((unsigned) ch) <= ' ' || ch == 0x7f ||
	    ch == '~' || ch == '^' || ch == ':' || ch == '\\')
		return 1;
	/* 2.13 Pattern Matching Notation */
	if (ch == '*' || ch == '?' || ch == '[') /* Unsupported */
		return 1;
	return 0;
}

/*
 * Try to read one refname component from the front of refname.  Return
 * the length of the component found, or -1 if the component is not
 * legal.
 */
static int check_refname_component(const char *refname, int flags)
{
	const char *cp;
	char last = '\0';

	for (cp = refname; ; cp++) {
		char ch = *cp;
		if (ch == '\0' || ch == '/')
			break;
		if (bad_ref_char(ch))
			return -1; /* Illegal character in refname. */
		if (last == '.' && ch == '.')
			return -1; /* Refname contains "..". */
		if (last == '@' && ch == '{')
			return -1; /* Refname contains "@{". */
		last = ch;
	}
	if (cp == refname)
54
		return 0; /* Component has zero length. */
55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
	if (refname[0] == '.') {
		if (!(flags & REFNAME_DOT_COMPONENT))
			return -1; /* Component starts with '.'. */
		/*
		 * Even if leading dots are allowed, don't allow "."
		 * as a component (".." is prevented by a rule above).
		 */
		if (refname[1] == '\0')
			return -1; /* Component equals ".". */
	}
	if (cp - refname >= 5 && !memcmp(cp - 5, ".lock", 5))
		return -1; /* Refname ends with ".lock". */
	return cp - refname;
}

int check_refname_format(const char *refname, int flags)
{
	int component_len, component_count = 0;

	while (1) {
		/* We are at the start of a path component. */
		component_len = check_refname_component(refname, flags);
77
		if (component_len <= 0) {
78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102
			if ((flags & REFNAME_REFSPEC_PATTERN) &&
					refname[0] == '*' &&
					(refname[1] == '\0' || refname[1] == '/')) {
				/* Accept one wildcard as a full refname component. */
				flags &= ~REFNAME_REFSPEC_PATTERN;
				component_len = 1;
			} else {
				return -1;
			}
		}
		component_count++;
		if (refname[component_len] == '\0')
			break;
		/* Skip to next component. */
		refname += component_len + 1;
	}

	if (refname[component_len - 1] == '.')
		return -1; /* Refname ends with '.'. */
	if (!(flags & REFNAME_ALLOW_ONELEVEL) && component_count < 2)
		return -1; /* Refname has only one component. */
	return 0;
}

struct ref_entry;
103

104 105 106 107 108 109
/*
 * Information used (along with the information in ref_entry) to
 * describe a single cached reference.  This data structure only
 * occurs embedded in a union in struct ref_entry, and only when
 * (ref_entry->flag & REF_DIR) is zero.
 */
110
struct ref_value {
111 112 113 114 115 116
	/*
	 * The name of the object to which this reference resolves
	 * (which may be a tag object).  If REF_ISBROKEN, this is
	 * null.  If REF_ISSYMREF, then this is the name of the object
	 * referred to by the last reference in the symlink chain.
	 */
117
	unsigned char sha1[20];
118 119 120 121

	/*
	 * If REF_KNOWS_PEELED, then this field holds the peeled value
	 * of this reference, or null if the reference is known not to
122 123
	 * be peelable.  See the documentation for peel_ref() for an
	 * exact definition of "peelable".
124
	 */
125 126 127
	unsigned char peeled[20];
};

128 129
struct ref_cache;

130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
/*
 * Information used (along with the information in ref_entry) to
 * describe a level in the hierarchy of references.  This data
 * structure only occurs embedded in a union in struct ref_entry, and
 * only when (ref_entry.flag & REF_DIR) is set.  In that case,
 * (ref_entry.flag & REF_INCOMPLETE) determines whether the references
 * in the directory have already been read:
 *
 *     (ref_entry.flag & REF_INCOMPLETE) unset -- a directory of loose
 *         or packed references, already read.
 *
 *     (ref_entry.flag & REF_INCOMPLETE) set -- a directory of loose
 *         references that hasn't been read yet (nor has any of its
 *         subdirectories).
 *
 * Entries within a directory are stored within a growable array of
 * pointers to ref_entries (entries, nr, alloc).  Entries 0 <= i <
 * sorted are sorted by their component name in strcmp() order and the
 * remaining entries are unsorted.
 *
 * Loose references are read lazily, one directory at a time.  When a
 * directory of loose references is read, then all of the references
 * in that directory are stored, and REF_INCOMPLETE stubs are created
 * for any subdirectories, but the subdirectories themselves are not
 * read.  The reading is triggered by get_ref_dir().
 */
156
struct ref_dir {
157
	int nr, alloc;
158 159 160 161 162 163 164 165 166

	/*
	 * Entries with index 0 <= i < sorted are sorted by name.  New
	 * entries are appended to the list unsorted, and are sorted
	 * only when required; thus we avoid the need to sort the list
	 * after the addition of every reference.
	 */
	int sorted;

167 168 169
	/* A pointer to the ref_cache that contains this ref_dir. */
	struct ref_cache *ref_cache;

170
	struct ref_entry **entries;
171 172
};

173 174 175 176 177 178 179 180 181 182 183
/*
 * Bit values for ref_entry::flag.  REF_ISSYMREF=0x01,
 * REF_ISPACKED=0x02, and REF_ISBROKEN=0x04 are public values; see
 * refs.h.
 */

/*
 * The field ref_entry->u.value.peeled of this value entry contains
 * the correct peeled value for the reference, which might be
 * null_sha1 if the reference is not a tag or if it is broken.
 */
184
#define REF_KNOWS_PEELED 0x08
185 186

/* ref_entry represents a directory of references */
187
#define REF_DIR 0x10
188

189 190 191 192 193 194
/*
 * Entry has not yet been read from disk (used only for REF_DIR
 * entries representing loose references)
 */
#define REF_INCOMPLETE 0x20

195 196
/*
 * A ref_entry represents either a reference or a "subdirectory" of
197 198 199 200 201 202 203 204 205 206 207 208 209 210 211
 * references.
 *
 * Each directory in the reference namespace is represented by a
 * ref_entry with (flags & REF_DIR) set and containing a subdir member
 * that holds the entries in that directory that have been read so
 * far.  If (flags & REF_INCOMPLETE) is set, then the directory and
 * its subdirectories haven't been read yet.  REF_INCOMPLETE is only
 * used for loose reference directories.
 *
 * References are represented by a ref_entry with (flags & REF_DIR)
 * unset and a value member that describes the reference's value.  The
 * flag member is at the ref_entry level, but it is also needed to
 * interpret the contents of the value field (in other words, a
 * ref_value object is not very much use without the enclosing
 * ref_entry).
212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
 *
 * Reference names cannot end with slash and directories' names are
 * always stored with a trailing slash (except for the top-level
 * directory, which is always denoted by "").  This has two nice
 * consequences: (1) when the entries in each subdir are sorted
 * lexicographically by name (as they usually are), the references in
 * a whole tree can be generated in lexicographic order by traversing
 * the tree in left-to-right, depth-first order; (2) the names of
 * references and subdirectories cannot conflict, and therefore the
 * presence of an empty subdirectory does not block the creation of a
 * similarly-named reference.  (The fact that reference names with the
 * same leading components can conflict *with each other* is a
 * separate issue that is regulated by is_refname_available().)
 *
 * Please note that the name field contains the fully-qualified
 * reference (or subdirectory) name.  Space could be saved by only
 * storing the relative names.  But that would require the full names
 * to be generated on the fly when iterating in do_for_each_ref(), and
 * would break callback functions, who have always been able to assume
 * that the name strings that they are passed will not be freed during
 * the iteration.
 */
234 235
struct ref_entry {
	unsigned char flag; /* ISSYMREF? ISPACKED? */
236
	union {
237 238
		struct ref_value value; /* if not (flags&REF_DIR) */
		struct ref_dir subdir; /* if (flags&REF_DIR) */
239
	} u;
240 241 242 243 244
	/*
	 * The full name of the reference (e.g., "refs/heads/master")
	 * or the full name of the directory with a trailing slash
	 * (e.g., "refs/heads/"):
	 */
245 246
	char name[FLEX_ARRAY];
};
247

248 249
static void read_loose_refs(const char *dirname, struct ref_dir *dir);

250 251
static struct ref_dir *get_ref_dir(struct ref_entry *entry)
{
252
	struct ref_dir *dir;
253
	assert(entry->flag & REF_DIR);
254 255 256 257 258 259
	dir = &entry->u.subdir;
	if (entry->flag & REF_INCOMPLETE) {
		read_loose_refs(entry->name, dir);
		entry->flag &= ~REF_INCOMPLETE;
	}
	return dir;
260 261
}

262 263 264
static struct ref_entry *create_ref_entry(const char *refname,
					  const unsigned char *sha1, int flag,
					  int check_name)
265 266
{
	int len;
267
	struct ref_entry *ref;
268

269
	if (check_name &&
270 271
	    check_refname_format(refname, REFNAME_ALLOW_ONELEVEL|REFNAME_DOT_COMPONENT))
		die("Reference has invalid format: '%s'", refname);
272 273
	len = strlen(refname) + 1;
	ref = xmalloc(sizeof(struct ref_entry) + len);
274 275
	hashcpy(ref->u.value.sha1, sha1);
	hashclr(ref->u.value.peeled);
276 277 278 279 280
	memcpy(ref->name, refname, len);
	ref->flag = flag;
	return ref;
}

281 282
static void clear_ref_dir(struct ref_dir *dir);

283 284
static void free_ref_entry(struct ref_entry *entry)
{
285 286 287 288 289 290 291
	if (entry->flag & REF_DIR) {
		/*
		 * Do not use get_ref_dir() here, as that might
		 * trigger the reading of loose refs.
		 */
		clear_ref_dir(&entry->u.subdir);
	}
292 293 294
	free(entry);
}

295 296 297 298 299 300
/*
 * Add a ref_entry to the end of dir (unsorted).  Entry is always
 * stored directly in dir; no recursion into subdirectories is
 * done.
 */
static void add_entry_to_dir(struct ref_dir *dir, struct ref_entry *entry)
301
{
302 303
	ALLOC_GROW(dir->entries, dir->nr + 1, dir->alloc);
	dir->entries[dir->nr++] = entry;
304 305 306 307 308 309
	/* optimize for the case that entries are added in order */
	if (dir->nr == 1 ||
	    (dir->nr == dir->sorted + 1 &&
	     strcmp(dir->entries[dir->nr - 2]->name,
		    dir->entries[dir->nr - 1]->name) < 0))
		dir->sorted = dir->nr;
310 311
}

312 313 314
/*
 * Clear and free all entries in dir, recursively.
 */
315
static void clear_ref_dir(struct ref_dir *dir)
316 317
{
	int i;
318 319 320 321 322
	for (i = 0; i < dir->nr; i++)
		free_ref_entry(dir->entries[i]);
	free(dir->entries);
	dir->sorted = dir->nr = dir->alloc = 0;
	dir->entries = NULL;
323 324
}

325 326 327 328 329
/*
 * Create a struct ref_entry object for the specified dirname.
 * dirname is the name of the directory with a trailing slash (e.g.,
 * "refs/heads/") or "" for the top-level directory.
 */
330
static struct ref_entry *create_dir_entry(struct ref_cache *ref_cache,
331 332
					  const char *dirname, size_t len,
					  int incomplete)
333 334 335
{
	struct ref_entry *direntry;
	direntry = xcalloc(1, sizeof(struct ref_entry) + len + 1);
336 337
	memcpy(direntry->name, dirname, len);
	direntry->name[len] = '\0';
338
	direntry->u.subdir.ref_cache = ref_cache;
339
	direntry->flag = REF_DIR | (incomplete ? REF_INCOMPLETE : 0);
340 341 342
	return direntry;
}

343
static int ref_entry_cmp(const void *a, const void *b)
344
{
345 346 347 348
	struct ref_entry *one = *(struct ref_entry **)a;
	struct ref_entry *two = *(struct ref_entry **)b;
	return strcmp(one->name, two->name);
}
349

350
static void sort_ref_dir(struct ref_dir *dir);
351

352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368
struct string_slice {
	size_t len;
	const char *str;
};

static int ref_entry_cmp_sslice(const void *key_, const void *ent_)
{
	struct string_slice *key = (struct string_slice *)key_;
	struct ref_entry *ent = *(struct ref_entry **)ent_;
	int entlen = strlen(ent->name);
	int cmplen = key->len < entlen ? key->len : entlen;
	int cmp = memcmp(key->str, ent->name, cmplen);
	if (cmp)
		return cmp;
	return key->len - entlen;
}

369
/*
370 371 372
 * Return the index of the entry with the given refname from the
 * ref_dir (non-recursively), sorting dir if necessary.  Return -1 if
 * no such entry is found.  dir must already be complete.
373
 */
374
static int search_ref_dir(struct ref_dir *dir, const char *refname, size_t len)
375
{
376 377
	struct ref_entry **r;
	struct string_slice key;
378

379
	if (refname == NULL || !dir->nr)
380
		return -1;
381

382
	sort_ref_dir(dir);
383 384 385 386
	key.len = len;
	key.str = refname;
	r = bsearch(&key, dir->entries, dir->nr, sizeof(*dir->entries),
		    ref_entry_cmp_sslice);
387 388

	if (r == NULL)
389
		return -1;
390

391
	return r - dir->entries;
392 393
}

394 395 396 397 398
/*
 * Search for a directory entry directly within dir (without
 * recursing).  Sort dir if necessary.  subdirname must be a directory
 * name (i.e., end in '/').  If mkdir is set, then create the
 * directory if it is missing; otherwise, return NULL if the desired
399
 * directory cannot be found.  dir must already be complete.
400
 */
401
static struct ref_dir *search_for_subdir(struct ref_dir *dir,
402 403
					 const char *subdirname, size_t len,
					 int mkdir)
404
{
405 406 407
	int entry_index = search_ref_dir(dir, subdirname, len);
	struct ref_entry *entry;
	if (entry_index == -1) {
408 409
		if (!mkdir)
			return NULL;
410 411 412 413 414 415
		/*
		 * Since dir is complete, the absence of a subdir
		 * means that the subdir really doesn't exist;
		 * therefore, create an empty record for it but mark
		 * the record complete.
		 */
416
		entry = create_dir_entry(dir->ref_cache, subdirname, len, 0);
417
		add_entry_to_dir(dir, entry);
418 419
	} else {
		entry = dir->entries[entry_index];
420
	}
421
	return get_ref_dir(entry);
422 423
}

424 425 426 427
/*
 * If refname is a reference name, find the ref_dir within the dir
 * tree that should hold refname.  If refname is a directory name
 * (i.e., ends in '/'), then return that ref_dir itself.  dir must
428 429 430 431
 * represent the top-level directory and must already be complete.
 * Sort ref_dirs and recurse into subdirectories as necessary.  If
 * mkdir is set, then create any missing directories; otherwise,
 * return NULL if the desired directory cannot be found.
432 433 434 435
 */
static struct ref_dir *find_containing_dir(struct ref_dir *dir,
					   const char *refname, int mkdir)
{
436 437
	const char *slash;
	for (slash = strchr(refname, '/'); slash; slash = strchr(slash + 1, '/')) {
438
		size_t dirnamelen = slash - refname + 1;
439
		struct ref_dir *subdir;
440
		subdir = search_for_subdir(dir, refname, dirnamelen, mkdir);
441 442
		if (!subdir) {
			dir = NULL;
443
			break;
444
		}
445
		dir = subdir;
446 447 448 449 450 451 452 453 454 455 456 457
	}

	return dir;
}

/*
 * Find the value entry with the given name in dir, sorting ref_dirs
 * and recursing into subdirectories as necessary.  If the name is not
 * found or it corresponds to a directory entry, return NULL.
 */
static struct ref_entry *find_ref(struct ref_dir *dir, const char *refname)
{
458
	int entry_index;
459 460 461 462
	struct ref_entry *entry;
	dir = find_containing_dir(dir, refname, 0);
	if (!dir)
		return NULL;
463 464 465 466 467
	entry_index = search_ref_dir(dir, refname, strlen(refname));
	if (entry_index == -1)
		return NULL;
	entry = dir->entries[entry_index];
	return (entry->flag & REF_DIR) ? NULL : entry;
468 469
}

470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520
/*
 * Remove the entry with the given name from dir, recursing into
 * subdirectories as necessary.  If refname is the name of a directory
 * (i.e., ends with '/'), then remove the directory and its contents.
 * If the removal was successful, return the number of entries
 * remaining in the directory entry that contained the deleted entry.
 * If the name was not found, return -1.  Please note that this
 * function only deletes the entry from the cache; it does not delete
 * it from the filesystem or ensure that other cache entries (which
 * might be symbolic references to the removed entry) are updated.
 * Nor does it remove any containing dir entries that might be made
 * empty by the removal.  dir must represent the top-level directory
 * and must already be complete.
 */
static int remove_entry(struct ref_dir *dir, const char *refname)
{
	int refname_len = strlen(refname);
	int entry_index;
	struct ref_entry *entry;
	int is_dir = refname[refname_len - 1] == '/';
	if (is_dir) {
		/*
		 * refname represents a reference directory.  Remove
		 * the trailing slash; otherwise we will get the
		 * directory *representing* refname rather than the
		 * one *containing* it.
		 */
		char *dirname = xmemdupz(refname, refname_len - 1);
		dir = find_containing_dir(dir, dirname, 0);
		free(dirname);
	} else {
		dir = find_containing_dir(dir, refname, 0);
	}
	if (!dir)
		return -1;
	entry_index = search_ref_dir(dir, refname, refname_len);
	if (entry_index == -1)
		return -1;
	entry = dir->entries[entry_index];

	memmove(&dir->entries[entry_index],
		&dir->entries[entry_index + 1],
		(dir->nr - entry_index - 1) * sizeof(*dir->entries)
		);
	dir->nr--;
	if (dir->sorted > entry_index)
		dir->sorted--;
	free_ref_entry(entry);
	return dir->nr;
}

521 522 523 524 525 526 527 528 529 530 531 532 533 534
/*
 * Add a ref_entry to the ref_dir (unsorted), recursing into
 * subdirectories as necessary.  dir must represent the top-level
 * directory.  Return 0 on success.
 */
static int add_ref(struct ref_dir *dir, struct ref_entry *ref)
{
	dir = find_containing_dir(dir, ref->name, 1);
	if (!dir)
		return -1;
	add_entry_to_dir(dir, ref);
	return 0;
}

535 536 537 538 539 540 541
/*
 * Emit a warning and return true iff ref1 and ref2 have the same name
 * and the same sha1.  Die if they have the same name but different
 * sha1s.
 */
static int is_dup_ref(const struct ref_entry *ref1, const struct ref_entry *ref2)
{
542
	if (strcmp(ref1->name, ref2->name))
543
		return 0;
544 545 546 547 548 549 550 551 552 553 554 555

	/* Duplicate name; make sure that they don't conflict: */

	if ((ref1->flag & REF_DIR) || (ref2->flag & REF_DIR))
		/* This is impossible by construction */
		die("Reference directory conflict: %s", ref1->name);

	if (hashcmp(ref1->u.value.sha1, ref2->u.value.sha1))
		die("Duplicated ref, and SHA1s don't match: %s", ref1->name);

	warning("Duplicated ref: %s", ref1->name);
	return 1;
556 557
}

558
/*
559 560
 * Sort the entries in dir non-recursively (if they are not already
 * sorted) and remove any duplicate entries.
561
 */
562
static void sort_ref_dir(struct ref_dir *dir)
563
{
564
	int i, j;
565
	struct ref_entry *last = NULL;
566

567 568 569 570
	/*
	 * This check also prevents passing a zero-length array to qsort(),
	 * which is a problem on some platforms.
	 */
571
	if (dir->sorted == dir->nr)
572
		return;
573

574
	qsort(dir->entries, dir->nr, sizeof(*dir->entries), ref_entry_cmp);
575

576 577 578 579 580 581 582
	/* Remove any duplicates: */
	for (i = 0, j = 0; j < dir->nr; j++) {
		struct ref_entry *entry = dir->entries[j];
		if (last && is_dup_ref(last, entry))
			free_ref_entry(entry);
		else
			last = dir->entries[i++] = entry;
583
	}
584
	dir->sorted = dir->nr = i;
585
}
586

587 588
/* Include broken references in a do_for_each_ref*() iteration: */
#define DO_FOR_EACH_INCLUDE_BROKEN 0x01
589

590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605
/*
 * Return true iff the reference described by entry can be resolved to
 * an object in the database.  Emit a warning if the referred-to
 * object does not exist.
 */
static int ref_resolves_to_object(struct ref_entry *entry)
{
	if (entry->flag & REF_ISBROKEN)
		return 0;
	if (!has_sha1_file(entry->u.value.sha1)) {
		error("%s does not point to a valid object!", entry->name);
		return 0;
	}
	return 1;
}

606 607 608 609 610 611 612 613 614
/*
 * current_ref is a performance hack: when iterating over references
 * using the for_each_ref*() functions, current_ref is set to the
 * current reference's entry before calling the callback function.  If
 * the callback function calls peel_ref(), then peel_ref() first
 * checks whether the reference to be peeled is the current reference
 * (it usually is) and if so, returns that reference's peeled version
 * if it is available.  This avoids a refname lookup in a common case.
 */
615
static struct ref_entry *current_ref;
616

617 618 619 620 621 622 623 624 625 626
typedef int each_ref_entry_fn(struct ref_entry *entry, void *cb_data);

struct ref_entry_cb {
	const char *base;
	int trim;
	int flags;
	each_ref_fn *fn;
	void *cb_data;
};

627
/*
628 629
 * Handle one reference in a do_for_each_ref*()-style iteration,
 * calling an each_ref_fn for each entry.
630
 */
631
static int do_one_ref(struct ref_entry *entry, void *cb_data)
632
{
633
	struct ref_entry_cb *data = cb_data;
634
	int retval;
635
	if (prefixcmp(entry->name, data->base))
636
		return 0;
637

638
	if (!(data->flags & DO_FOR_EACH_INCLUDE_BROKEN) &&
639 640 641
	      !ref_resolves_to_object(entry))
		return 0;

642
	current_ref = entry;
643 644
	retval = data->fn(entry->name + data->trim, entry->u.value.sha1,
			  entry->flag, data->cb_data);
645 646
	current_ref = NULL;
	return retval;
647
}
648

649
/*
650
 * Call fn for each reference in dir that has index in the range
651 652
 * offset <= index < dir->nr.  Recurse into subdirectories that are in
 * that index range, sorting them before iterating.  This function
653 654
 * does not sort dir itself; it should be sorted beforehand.  fn is
 * called for all references, including broken ones.
655
 */
656 657
static int do_for_each_entry_in_dir(struct ref_dir *dir, int offset,
				    each_ref_entry_fn fn, void *cb_data)
658 659
{
	int i;
660 661
	assert(dir->sorted == dir->nr);
	for (i = offset; i < dir->nr; i++) {
662 663 664
		struct ref_entry *entry = dir->entries[i];
		int retval;
		if (entry->flag & REF_DIR) {
665 666
			struct ref_dir *subdir = get_ref_dir(entry);
			sort_ref_dir(subdir);
667
			retval = do_for_each_entry_in_dir(subdir, 0, fn, cb_data);
668
		} else {
669
			retval = fn(entry, cb_data);
670
		}
671 672 673 674 675 676
		if (retval)
			return retval;
	}
	return 0;
}

677
/*
678
 * Call fn for each reference in the union of dir1 and dir2, in order
679 680 681
 * by refname.  Recurse into subdirectories.  If a value entry appears
 * in both dir1 and dir2, then only process the version that is in
 * dir2.  The input dirs must already be sorted, but subdirs will be
682 683
 * sorted as needed.  fn is called for all references, including
 * broken ones.
684
 */
685 686 687
static int do_for_each_entry_in_dirs(struct ref_dir *dir1,
				     struct ref_dir *dir2,
				     each_ref_entry_fn fn, void *cb_data)
688 689 690 691
{
	int retval;
	int i1 = 0, i2 = 0;

692 693
	assert(dir1->sorted == dir1->nr);
	assert(dir2->sorted == dir2->nr);
694 695 696 697
	while (1) {
		struct ref_entry *e1, *e2;
		int cmp;
		if (i1 == dir1->nr) {
698
			return do_for_each_entry_in_dir(dir2, i2, fn, cb_data);
699 700
		}
		if (i2 == dir2->nr) {
701
			return do_for_each_entry_in_dir(dir1, i1, fn, cb_data);
702 703 704 705 706 707 708
		}
		e1 = dir1->entries[i1];
		e2 = dir2->entries[i2];
		cmp = strcmp(e1->name, e2->name);
		if (cmp == 0) {
			if ((e1->flag & REF_DIR) && (e2->flag & REF_DIR)) {
				/* Both are directories; descend them in parallel. */
709 710 711 712
				struct ref_dir *subdir1 = get_ref_dir(e1);
				struct ref_dir *subdir2 = get_ref_dir(e2);
				sort_ref_dir(subdir1);
				sort_ref_dir(subdir2);
713 714
				retval = do_for_each_entry_in_dirs(
						subdir1, subdir2, fn, cb_data);
715 716 717 718
				i1++;
				i2++;
			} else if (!(e1->flag & REF_DIR) && !(e2->flag & REF_DIR)) {
				/* Both are references; ignore the one from dir1. */
719
				retval = fn(e2, cb_data);
720 721 722 723 724 725
				i1++;
				i2++;
			} else {
				die("conflict between reference and directory: %s",
				    e1->name);
			}
726
		} else {
727 728 729
			struct ref_entry *e;
			if (cmp < 0) {
				e = e1;
730
				i1++;
731 732 733 734 735
			} else {
				e = e2;
				i2++;
			}
			if (e->flag & REF_DIR) {
736 737
				struct ref_dir *subdir = get_ref_dir(e);
				sort_ref_dir(subdir);
738 739
				retval = do_for_each_entry_in_dir(
						subdir, 0, fn, cb_data);
740
			} else {
741
				retval = fn(e, cb_data);
742 743 744 745 746 747 748
			}
		}
		if (retval)
			return retval;
	}
}

749 750 751 752 753 754 755 756 757
/*
 * Return true iff refname1 and refname2 conflict with each other.
 * Two reference names conflict if one of them exactly matches the
 * leading components of the other; e.g., "foo/bar" conflicts with
 * both "foo" and with "foo/bar/baz" but not with "foo/bar" or
 * "foo/barbados".
 */
static int names_conflict(const char *refname1, const char *refname2)
{
758 759 760 761 762 763 764 765 766 767 768 769
	for (; *refname1 && *refname1 == *refname2; refname1++, refname2++)
		;
	return (*refname1 == '\0' && *refname2 == '/')
		|| (*refname1 == '/' && *refname2 == '\0');
}

struct name_conflict_cb {
	const char *refname;
	const char *oldrefname;
	const char *conflicting_refname;
};

770
static int name_conflict_fn(struct ref_entry *entry, void *cb_data)
771 772
{
	struct name_conflict_cb *data = (struct name_conflict_cb *)cb_data;
773
	if (data->oldrefname && !strcmp(data->oldrefname, entry->name))
774
		return 0;
775 776
	if (names_conflict(data->refname, entry->name)) {
		data->conflicting_refname = entry->name;
777
		return 1;
778
	}
779
	return 0;
780 781
}

782 783
/*
 * Return true iff a reference named refname could be created without
784
 * conflicting with the name of an existing reference in dir.  If
785 786
 * oldrefname is non-NULL, ignore potential conflicts with oldrefname
 * (e.g., because oldrefname is scheduled for deletion in the same
787 788 789
 * operation).
 */
static int is_refname_available(const char *refname, const char *oldrefname,
790
				struct ref_dir *dir)
791
{
792 793 794 795 796
	struct name_conflict_cb data;
	data.refname = refname;
	data.oldrefname = oldrefname;
	data.conflicting_refname = NULL;

797
	sort_ref_dir(dir);
798
	if (do_for_each_entry_in_dir(dir, 0, name_conflict_fn, &data)) {
799 800 801
		error("'%s' exists; cannot create '%s'",
		      data.conflicting_refname, refname);
		return 0;
802 803
	}
	return 1;
804 805
}

806 807 808 809
/*
 * Future: need to be in "struct repository"
 * when doing a full libification.
 */
810 811
static struct ref_cache {
	struct ref_cache *next;
812 813
	struct ref_entry *loose;
	struct ref_entry *packed;
814 815
	/* The submodule name, or "" for the main repo. */
	char name[FLEX_ARRAY];
816
} *ref_cache;
817

818
static void clear_packed_ref_cache(struct ref_cache *refs)
819
{
820 821 822 823
	if (refs->packed) {
		free_ref_entry(refs->packed);
		refs->packed = NULL;
	}
824
}
825

826 827
static void clear_loose_ref_cache(struct ref_cache *refs)
{
828 829 830 831
	if (refs->loose) {
		free_ref_entry(refs->loose);
		refs->loose = NULL;
	}
832 833
}

834
static struct ref_cache *create_ref_cache(const char *submodule)
835
{
836
	int len;
837
	struct ref_cache *refs;
838 839 840
	if (!submodule)
		submodule = "";
	len = strlen(submodule) + 1;
841
	refs = xcalloc(1, sizeof(struct ref_cache) + len);
842
	memcpy(refs->name, submodule, len);
843 844 845
	return refs;
}

846
/*
847
 * Return a pointer to a ref_cache for the specified submodule. For
848 849 850 851
 * the main repository, use submodule==NULL. The returned structure
 * will be allocated and initialized but not necessarily populated; it
 * should not be freed.
 */
852
static struct ref_cache *get_ref_cache(const char *submodule)
853
{
854
	struct ref_cache *refs = ref_cache;
855 856 857 858 859 860
	if (!submodule)
		submodule = "";
	while (refs) {
		if (!strcmp(submodule, refs->name))
			return refs;
		refs = refs->next;
861
	}
862

863 864 865
	refs = create_ref_cache(submodule);
	refs->next = ref_cache;
	ref_cache = refs;
866
	return refs;
867 868
}

869
void invalidate_ref_cache(const char *submodule)
870
{
871 872 873
	struct ref_cache *refs = get_ref_cache(submodule);
	clear_packed_ref_cache(refs);
	clear_loose_ref_cache(refs);
874
}
875

876 877 878
/* The length of a peeled reference line in packed-refs, including EOL: */
#define PEELED_LINE_LENGTH 42

879 880 881 882 883 884 885
/*
 * The packed-refs header line that we write out.  Perhaps other
 * traits will be added later.  The trailing space is required.
 */
static const char PACKED_REFS_HEADER[] =
	"# pack-refs with: peeled fully-peeled \n";

886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918
/*
 * Parse one line from a packed-refs file.  Write the SHA1 to sha1.
 * Return a pointer to the refname within the line (null-terminated),
 * or NULL if there was a problem.
 */
static const char *parse_ref_line(char *line, unsigned char *sha1)
{
	/*
	 * 42: the answer to everything.
	 *
	 * In this case, it happens to be the answer to
	 *  40 (length of sha1 hex representation)
	 *  +1 (space in between hex and name)
	 *  +1 (newline at the end of the line)
	 */
	int len = strlen(line) - 42;

	if (len <= 0)
		return NULL;
	if (get_sha1_hex(line, sha1) < 0)
		return NULL;
	if (!isspace(line[40]))
		return NULL;
	line += 41;
	if (isspace(*line))
		return NULL;
	if (line[len] != '\n')
		return NULL;
	line[len] = 0;

	return line;
}

919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945
/*
 * Read f, which is a packed-refs file, into dir.
 *
 * A comment line of the form "# pack-refs with: " may contain zero or
 * more traits. We interpret the traits as follows:
 *
 *   No traits:
 *
 *      Probably no references are peeled. But if the file contains a
 *      peeled value for a reference, we will use it.
 *
 *   peeled:
 *
 *      References under "refs/tags/", if they *can* be peeled, *are*
 *      peeled in this file. References outside of "refs/tags/" are
 *      probably not peeled even if they could have been, but if we find
 *      a peeled value for such a reference we will use it.
 *
 *   fully-peeled:
 *
 *      All references in the file that can be peeled are peeled.
 *      Inversely (and this is more important), any references in the
 *      file for which no peeled value is recorded is not peelable. This
 *      trait should typically be written alongside "peeled" for
 *      compatibility with older clients, but we do not require it
 *      (i.e., "peeled" is a no-op if "fully-peeled" is set).
 */
946
static void read_packed_refs(FILE *f, struct ref_dir *dir)
947
{
948
	struct ref_entry *last = NULL;
949
	char refline[PATH_MAX];
950
	enum { PEELED_NONE, PEELED_TAGS, PEELED_FULLY } peeled = PEELED_NONE;
951 952 953

	while (fgets(refline, sizeof(refline), f)) {
		unsigned char sha1[20];
954
		const char *refname;
955 956 957 958
		static const char header[] = "# pack-refs with:";

		if (!strncmp(refline, header, sizeof(header)-1)) {
			const char *traits = refline + sizeof(header) - 1;
959 960 961 962
			if (strstr(traits, " fully-peeled "))
				peeled = PEELED_FULLY;
			else if (strstr(traits, " peeled "))
				peeled = PEELED_TAGS;
963 964 965 966
			/* perhaps other traits later as well */
			continue;
		}

967 968
		refname = parse_ref_line(refline, sha1);
		if (refname) {
969 970 971 972
			last = create_ref_entry(refname, sha1, REF_ISPACKED, 1);
			if (peeled == PEELED_FULLY ||
			    (peeled == PEELED_TAGS && !prefixcmp(refname, "refs/tags/")))
				last->flag |= REF_KNOWS_PEELED;
973
			add_ref(dir, last);
974 975 976 977
			continue;
		}
		if (last &&
		    refline[0] == '^' &&
978 979
		    strlen(refline) == PEELED_LINE_LENGTH &&
		    refline[PEELED_LINE_LENGTH - 1] == '\n' &&
980
		    !get_sha1_hex(refline + 1, sha1)) {
981
			hashcpy(last->u.value.peeled, sha1);
982 983 984 985 986 987 988
			/*
			 * Regardless of what the file header said,
			 * we definitely know the value of *this*
			 * reference:
			 */
			last->flag |= REF_KNOWS_PEELED;
		}
989 990 991
	}
}

992
static struct ref_dir *get_packed_refs(struct ref_cache *refs)
993
{
994
	if (!refs->packed) {
995 996
		const char *packed_refs_file;
		FILE *f;
997

998
		refs->packed = create_dir_entry(refs, "", 0, 0);
999 1000
		if (*refs->name)
			packed_refs_file = git_path_submodule(refs->name, "packed-refs");
1001 1002 1003
		else
			packed_refs_file = git_path("packed-refs");
		f = fopen(packed_refs_file, "r");
1004
		if (f) {
1005
			read_packed_refs(f, get_ref_dir(refs->packed));
1006 1007 1008
			fclose(f);
		}
	}
1009
	return get_ref_dir(refs->packed);
1010 1011
}

1012 1013 1014 1015 1016 1017
void add_packed_ref(const char *refname, const unsigned char *sha1)
{
	add_ref(get_packed_refs(get_ref_cache(NULL)),
			create_ref_entry(refname, sha1, REF_ISPACKED, 1));
}

1018
/*
1019 1020 1021
 * Read the loose references from the namespace dirname into dir
 * (without recursing).  dirname must end with '/'.  dir must be the
 * directory entry corresponding to dirname.
1022
 */
1023
static void read_loose_refs(const char *dirname, struct ref_dir *dir)
1024
{
1025
	struct ref_cache *refs = dir->ref_cache;
1026
	DIR *d;
1027
	const char *path;
1028
	struct dirent *de;
1029
	int dirnamelen = strlen(dirname);
1030
	struct strbuf refname;
1031

M
Michael Haggerty 已提交
1032
	if (*refs->name)
1033
		path = git_path_submodule(refs->name, "%s", dirname);
1034
	else
1035
		path = git_path("%s", dirname);
1036

1037
	d = opendir(path);
1038 1039 1040
	if (!d)
		return;

1041 1042
	strbuf_init(&refname, dirnamelen + 257);
	strbuf_add(&refname, dirname, dirnamelen);
1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053

	while ((de = readdir(d)) != NULL) {
		unsigned char sha1[20];
		struct stat st;
		int flag;
		const char *refdir;

		if (de->d_name[0] == '.')
			continue;
		if (has_extension(de->d_name, ".lock"))
			continue;
1054
		strbuf_addstr(&refname, de->d_name);
1055
		refdir = *refs->name
1056 1057 1058 1059 1060
			? git_path_submodule(refs->name, "%s", refname.buf)
			: git_path("%s", refname.buf);
		if (stat(refdir, &st) < 0) {
			; /* silently ignore */
		} else if (S_ISDIR(st.st_mode)) {
1061
			strbuf_addch(&refname, '/');
1062
			add_entry_to_dir(dir,
1063 1064
					 create_dir_entry(refs, refname.buf,
							  refname.len, 1));
1065
		} else {
M
Michael Haggerty 已提交
1066
			if (*refs->name) {
1067
				hashclr(sha1);
1068
				flag = 0;
1069
				if (resolve_gitlink_ref(refs->name, refname.buf, sha1) < 0) {
1070
					hashclr(sha1);
1071
					flag |= REF_ISBROKEN;
1072
				}
1073
			} else if (read_ref_full(refname.buf, sha1, 1, &flag)) {
1074 1075 1076
				hashclr(sha1);
				flag |= REF_ISBROKEN;
			}
1077 1078
			add_entry_to_dir(dir,
					 create_ref_entry(refname.buf, sha1, flag, 1));
1079
		}
1080
		strbuf_setlen(&refname, dirnamelen);
1081
	}
1082
	strbuf_release(&refname);
1083
	closedir(d);
1084 1085
}

1086
static struct ref_dir *get_loose_refs(struct ref_cache *refs)
1087
{
1088
	if (!refs->loose) {
1089 1090 1091 1092 1093
		/*
		 * Mark the top-level directory complete because we
		 * are about to read the only subdirectory that can
		 * hold references:
		 */
1094
		refs->loose = create_dir_entry(refs, "", 0, 0);
1095 1096 1097 1098
		/*
		 * Create an incomplete entry for "refs/":
		 */
		add_entry_to_dir(get_ref_dir(refs->loose),
1099
				 create_dir_entry(refs, "refs/", 5, 1));
1100
	}
1101
	return get_ref_dir(refs->loose);
1102 1103
}

1104 1105
/* We allow "recursive" symbolic refs. Only within reason, though */
#define MAXDEPTH 5
1106 1107
#define MAXREFLEN (1024)

1108 1109
/*
 * Called by resolve_gitlink_ref_recursive() after it failed to read
1110 1111
 * from the loose refs in ref_cache refs. Find <refname> in the
 * packed-refs file for the submodule.
1112
 */
1113
static int resolve_gitlink_packed_ref(struct ref_cache *refs,
1114
				      const char *refname, unsigned char *sha1)
1115
{
1116
	struct ref_entry *ref;
1117
	struct ref_dir *dir = get_packed_refs(refs);
1118

1119
	ref = find_ref(dir, refname);
1120 1121 1122
	if (ref == NULL)
		return -1;

1123
	memcpy(sha1, ref->u.value.sha1, 20);
1124
	return 0;
1125 1126
}

1127
static int resolve_gitlink_ref_recursive(struct ref_cache *refs,
1128
					 const char *refname, unsigned char *sha1,
1129
					 int recursion)
1130
{
1131
	int fd, len;
1132
	char buffer[128], *p;
1133
	char *path;
1134

1135
	if (recursion > MAXDEPTH || strlen(refname) > MAXREFLEN)
1136
		return -1;
1137 1138 1139 1140
	path = *refs->name
		? git_path_submodule(refs->name, "%s", refname)
		: git_path("%s", refname);
	fd = open(path, O_RDONLY);
1141
	if (fd < 0)
1142
		return resolve_gitlink_packed_ref(refs, refname, sha1);
1143 1144 1145 1146 1147 1148 1149 1150 1151 1152

	len = read(fd, buffer, sizeof(buffer)-1);
	close(fd);
	if (len < 0)
		return -1;
	while (len && isspace(buffer[len-1]))
		len--;
	buffer[len] = 0;

	/* Was it a detached head or an old-fashioned symlink? */
1153
	if (!get_sha1_hex(buffer, sha1))
1154 1155 1156 1157 1158 1159 1160 1161 1162
		return 0;

	/* Symref? */
	if (strncmp(buffer, "ref:", 4))
		return -1;
	p = buffer + 4;
	while (isspace(*p))
		p++;

1163
	return resolve_gitlink_ref_recursive(refs, p, sha1, recursion+1);
1164 1165
}

1166
int resolve_gitlink_ref(const char *path, const char *refname, unsigned char *sha1)
1167 1168
{
	int len = strlen(path), retval;
1169
	char *submodule;
1170
	struct ref_cache *refs;
1171 1172 1173 1174 1175

	while (len && path[len-1] == '/')
		len--;
	if (!len)
		return -1;
1176 1177 1178 1179
	submodule = xstrndup(path, len);
	refs = get_ref_cache(submodule);
	free(submodule);

1180
	retval = resolve_gitlink_ref_recursive(refs, refname, sha1, 0);
1181 1182
	return retval;
}
1183

1184
/*
1185 1186
 * Return the ref_entry for the given refname from the packed
 * references.  If it does not exist, return NULL.
1187
 */
1188
static struct ref_entry *get_packed_ref(const char *refname)
1189
{
1190
	return find_ref(get_packed_refs(get_ref_cache(NULL)), refname);
1191 1192
}

J
Junio C Hamano 已提交
1193
const char *resolve_ref_unsafe(const char *refname, unsigned char *sha1, int reading, int *flag)
1194
{
1195 1196
	int depth = MAXDEPTH;
	ssize_t len;
1197
	char buffer[256];
1198
	static char refname_buffer[256];
1199

1200 1201 1202
	if (flag)
		*flag = 0;

1203
	if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL))
1204 1205
		return NULL;

1206
	for (;;) {
1207
		char path[PATH_MAX];
1208 1209 1210
		struct stat st;
		char *buf;
		int fd;
1211

1212 1213
		if (--depth < 0)
			return NULL;
1214

1215
		git_snpath(path, sizeof(path), "%s", refname);
1216

1217
		if (lstat(path, &st) < 0) {
1218 1219
			struct ref_entry *entry;

1220 1221 1222 1223 1224 1225
			if (errno != ENOENT)
				return NULL;
			/*
			 * The loose reference file does not exist;
			 * check for a packed reference.
			 */
1226 1227 1228
			entry = get_packed_ref(refname);
			if (entry) {
				hashcpy(sha1, entry->u.value.sha1);
1229 1230
				if (flag)
					*flag |= REF_ISPACKED;
1231
				return refname;
1232
			}
1233 1234
			/* The reference is not a packed reference, either. */
			if (reading) {
1235
				return NULL;
1236 1237
			} else {
				hashclr(sha1);
1238
				return refname;
1239
			}
1240
		}
1241

1242 1243 1244
		/* Follow "normalized" - ie "refs/.." symlinks by hand */
		if (S_ISLNK(st.st_mode)) {
			len = readlink(path, buffer, sizeof(buffer)-1);
1245 1246
			if (len < 0)
				return NULL;
M
Michael Haggerty 已提交
1247
			buffer[len] = 0;
1248 1249
			if (!prefixcmp(buffer, "refs/") &&
					!check_refname_format(buffer, 0)) {
1250 1251
				strcpy(refname_buffer, buffer);
				refname = refname_buffer;
1252 1253
				if (flag)
					*flag |= REF_ISSYMREF;
1254 1255
				continue;
			}
1256
		}
1257

1258 1259 1260 1261 1262 1263
		/* Is it a directory? */
		if (S_ISDIR(st.st_mode)) {
			errno = EISDIR;
			return NULL;
		}

1264 1265 1266 1267 1268 1269 1270
		/*
		 * Anything else, just open it and try to use it as
		 * a ref
		 */
		fd = open(path, O_RDONLY);
		if (fd < 0)
			return NULL;
1271
		len = read_in_full(fd, buffer, sizeof(buffer)-1);
1272
		close(fd);
1273 1274 1275 1276 1277
		if (len < 0)
			return NULL;
		while (len && isspace(buffer[len-1]))
			len--;
		buffer[len] = '\0';
1278 1279 1280 1281

		/*
		 * Is it a symbolic ref?
		 */
1282
		if (prefixcmp(buffer, "ref:"))
1283
			break;
1284 1285
		if (flag)
			*flag |= REF_ISSYMREF;
1286
		buf = buffer + 4;
1287 1288
		while (isspace(*buf))
			buf++;
1289
		if (check_refname_format(buf, REFNAME_ALLOW_ONELEVEL)) {
1290 1291
			if (flag)
				*flag |= REF_ISBROKEN;
1292 1293
			return NULL;
		}
1294
		refname = strcpy(refname_buffer, buf);
1295
	}
1296 1297
	/* Please note that FETCH_HEAD has a second line containing other data. */
	if (get_sha1_hex(buffer, sha1) || (buffer[40] != '\0' && !isspace(buffer[40]))) {
1298 1299
		if (flag)
			*flag |= REF_ISBROKEN;
1300
		return NULL;
1301
	}
1302
	return refname;
1303 1304
}

1305 1306
char *resolve_refdup(const char *ref, unsigned char *sha1, int reading, int *flag)
{
1307
	const char *ret = resolve_ref_unsafe(ref, sha1, reading, flag);
1308 1309 1310
	return ret ? xstrdup(ret) : NULL;
}

I
Ilari Liusvaara 已提交
1311 1312 1313 1314 1315 1316 1317
/* The argument to filter_refs */
struct ref_filter {
	const char *pattern;
	each_ref_fn *fn;
	void *cb_data;
};

1318
int read_ref_full(const char *refname, unsigned char *sha1, int reading, int *flags)
1319
{
J
Junio C Hamano 已提交
1320
	if (resolve_ref_unsafe(refname, sha1, reading, flags))
1321 1322
		return 0;
	return -1;
1323 1324
}

1325
int read_ref(const char *refname, unsigned char *sha1)
1326
{
1327
	return read_ref_full(refname, sha1, 1, NULL);
1328 1329
}

1330
int ref_exists(const char *refname)
1331
{
1332 1333
	unsigned char sha1[20];
	return !!resolve_ref_unsafe(refname, sha1, 1, NULL);
1334 1335
}

1336
static int filter_refs(const char *refname, const unsigned char *sha1, int flags,
1337
		       void *data)
I
Ilari Liusvaara 已提交
1338 1339
{
	struct ref_filter *filter = (struct ref_filter *)data;
1340
	if (fnmatch(filter->pattern, refname, 0))
I
Ilari Liusvaara 已提交
1341
		return 0;
1342
	return filter->fn(refname, sha1, flags, filter->cb_data);
I
Ilari Liusvaara 已提交
1343 1344
}

1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356
enum peel_status {
	/* object was peeled successfully: */
	PEEL_PEELED = 0,

	/*
	 * object cannot be peeled because the named object (or an
	 * object referred to by a tag in the peel chain), does not
	 * exist.
	 */
	PEEL_INVALID = -1,

	/* object cannot be peeled because it is not a tag: */
1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367
	PEEL_NON_TAG = -2,

	/* ref_entry contains no peeled value because it is a symref: */
	PEEL_IS_SYMREF = -3,

	/*
	 * ref_entry cannot be peeled because it is broken (i.e., the
	 * symbolic reference cannot even be resolved to an object
	 * name):
	 */
	PEEL_BROKEN = -4
1368 1369
};

1370 1371
/*
 * Peel the named object; i.e., if the object is a tag, resolve the
1372 1373 1374 1375
 * tag recursively until a non-tag is found.  If successful, store the
 * result to sha1 and return PEEL_PEELED.  If the object is not a tag
 * or is not valid, return PEEL_NON_TAG or PEEL_INVALID, respectively,
 * and leave sha1 unchanged.
1376
 */
1377
static enum peel_status peel_object(const unsigned char *name, unsigned char *sha1)
1378 1379 1380 1381 1382 1383
{
	struct object *o = lookup_unknown_object(name);

	if (o->type == OBJ_NONE) {
		int type = sha1_object_info(name, NULL);
		if (type < 0)
1384
			return PEEL_INVALID;
1385 1386 1387 1388
		o->type = type;
	}

	if (o->type != OBJ_TAG)
1389
		return PEEL_NON_TAG;
1390 1391 1392

	o = deref_tag_noverify(o);
	if (!o)
1393
		return PEEL_INVALID;
1394 1395

	hashcpy(sha1, o->sha1);
1396
	return PEEL_PEELED;
1397 1398
}

1399 1400
/*
 * Peel the entry (if possible) and return its new peel_status.
1401 1402 1403 1404 1405 1406
 *
 * It is OK to call this function with a packed reference entry that
 * might be stale and might even refer to an object that has since
 * been garbage-collected.  In such a case, if the entry has
 * REF_KNOWS_PEELED then leave the status unchanged and return
 * PEEL_PEELED or PEEL_NON_TAG; otherwise, return PEEL_INVALID.
1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425
 */
static enum peel_status peel_entry(struct ref_entry *entry)
{
	enum peel_status status;

	if (entry->flag & REF_KNOWS_PEELED)
		return is_null_sha1(entry->u.value.peeled) ?
			PEEL_NON_TAG : PEEL_PEELED;
	if (entry->flag & REF_ISBROKEN)
		return PEEL_BROKEN;
	if (entry->flag & REF_ISSYMREF)
		return PEEL_IS_SYMREF;

	status = peel_object(entry->u.value.sha1, entry->u.value.peeled);
	if (status == PEEL_PEELED || status == PEEL_NON_TAG)
		entry->flag |= REF_KNOWS_PEELED;
	return status;
}

1426
int peel_ref(const char *refname, unsigned char *sha1)
1427 1428 1429 1430
{
	int flag;
	unsigned char base[20];

1431
	if (current_ref && (current_ref->name == refname
1432 1433 1434 1435 1436
			    || !strcmp(current_ref->name, refname))) {
		if (peel_entry(current_ref))
			return -1;
		hashcpy(sha1, current_ref->u.value.peeled);
		return 0;
1437 1438
	}

1439
	if (read_ref_full(refname, base, 1, &flag))
1440 1441
		return -1;

1442 1443 1444 1445 1446 1447 1448 1449 1450
	/*
	 * If the reference is packed, read its ref_entry from the
	 * cache in the hope that we already know its peeled value.
	 * We only try this optimization on packed references because
	 * (a) forcing the filling of the loose reference cache could
	 * be expensive and (b) loose references anyway usually do not
	 * have REF_KNOWS_PEELED.
	 */
	if (flag & REF_ISPACKED) {
1451
		struct ref_entry *r = get_packed_ref(refname);
1452 1453 1454
		if (r) {
			if (peel_entry(r))
				return -1;
1455
			hashcpy(sha1, r->u.value.peeled);
1456
			return 0;
1457 1458 1459
		}
	}

1460
	return peel_object(base, sha1);
1461 1462
}

1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483
struct warn_if_dangling_data {
	FILE *fp;
	const char *refname;
	const char *msg_fmt;
};

static int warn_if_dangling_symref(const char *refname, const unsigned char *sha1,
				   int flags, void *cb_data)
{
	struct warn_if_dangling_data *d = cb_data;
	const char *resolves_to;
	unsigned char junk[20];

	if (!(flags & REF_ISSYMREF))
		return 0;

	resolves_to = resolve_ref_unsafe(refname, junk, 0, NULL);
	if (!resolves_to || strcmp(resolves_to, d->refname))
		return 0;

	fprintf(d->fp, d->msg_fmt, refname);
J
Junio C Hamano 已提交
1484
	fputc('\n', d->fp);
1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497
	return 0;
}

void warn_dangling_symref(FILE *fp, const char *msg_fmt, const char *refname)
{
	struct warn_if_dangling_data data;

	data.fp = fp;
	data.refname = refname;
	data.msg_fmt = msg_fmt;
	for_each_rawref(warn_if_dangling_symref, &data);
}

1498
/*
1499 1500 1501
 * Call fn for each reference in the specified submodule, omitting
 * references not in the containing_dir of base.  fn is called for all
 * references, including broken ones.  If fn ever returns a non-zero
1502 1503 1504
 * value, stop the iteration and return that value; otherwise, return
 * 0.
 */
1505 1506
static int do_for_each_entry(const char *submodule, const char *base,
			     each_ref_entry_fn fn, void *cb_data)
1507
{
1508
	struct ref_cache *refs = get_ref_cache(submodule);
1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520
	struct ref_dir *packed_dir = get_packed_refs(refs);
	struct ref_dir *loose_dir = get_loose_refs(refs);
	int retval = 0;

	if (base && *base) {
		packed_dir = find_containing_dir(packed_dir, base, 0);
		loose_dir = find_containing_dir(loose_dir, base, 0);
	}

	if (packed_dir && loose_dir) {
		sort_ref_dir(packed_dir);
		sort_ref_dir(loose_dir);
1521 1522
		retval = do_for_each_entry_in_dirs(
				packed_dir, loose_dir, fn, cb_data);
1523 1524
	} else if (packed_dir) {
		sort_ref_dir(packed_dir);
1525 1526
		retval = do_for_each_entry_in_dir(
				packed_dir, 0, fn, cb_data);
1527 1528
	} else if (loose_dir) {
		sort_ref_dir(loose_dir);
1529 1530
		retval = do_for_each_entry_in_dir(
				loose_dir, 0, fn, cb_data);
1531 1532 1533
	}

	return retval;
1534 1535
}

1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
/*
 * Call fn for each reference in the specified submodule for which the
 * refname begins with base.  If trim is non-zero, then trim that many
 * characters off the beginning of each refname before passing the
 * refname to fn.  flags can be DO_FOR_EACH_INCLUDE_BROKEN to include
 * broken references in the iteration.  If fn ever returns a non-zero
 * value, stop the iteration and return that value; otherwise, return
 * 0.
 */
static int do_for_each_ref(const char *submodule, const char *base, each_ref_fn fn,
			   int trim, int flags, void *cb_data)
{
	struct ref_entry_cb data;
	data.base = base;
	data.trim = trim;
	data.flags = flags;
	data.fn = fn;
	data.cb_data = cb_data;

	return do_for_each_entry(submodule, base, do_one_ref, &data);
}

1558
static int do_head_ref(const char *submodule, each_ref_fn fn, void *cb_data)
1559 1560
{
	unsigned char sha1[20];
1561 1562
	int flag;

1563 1564 1565 1566 1567 1568 1569
	if (submodule) {
		if (resolve_gitlink_ref(submodule, "HEAD", sha1) == 0)
			return fn("HEAD", sha1, 0, cb_data);

		return 0;
	}

1570
	if (!read_ref_full("HEAD", sha1, 1, &flag))
1571
		return fn("HEAD", sha1, flag, cb_data);
1572

1573
	return 0;
1574 1575
}

1576 1577 1578 1579 1580
int head_ref(each_ref_fn fn, void *cb_data)
{
	return do_head_ref(NULL, fn, cb_data);
}

1581 1582 1583 1584 1585
int head_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
{
	return do_head_ref(submodule, fn, cb_data);
}

1586
int for_each_ref(each_ref_fn fn, void *cb_data)
1587
{
1588
	return do_for_each_ref(NULL, "", fn, 0, 0, cb_data);
1589 1590
}

1591 1592
int for_each_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
{
1593
	return do_for_each_ref(submodule, "", fn, 0, 0, cb_data);
1594 1595
}

1596 1597
int for_each_ref_in(const char *prefix, each_ref_fn fn, void *cb_data)
{
1598
	return do_for_each_ref(NULL, prefix, fn, strlen(prefix), 0, cb_data);
1599 1600
}

1601 1602 1603 1604
int for_each_ref_in_submodule(const char *submodule, const char *prefix,
		each_ref_fn fn, void *cb_data)
{
	return do_for_each_ref(submodule, prefix, fn, strlen(prefix), 0, cb_data);
1605 1606
}

1607
int for_each_tag_ref(each_ref_fn fn, void *cb_data)
1608
{
1609
	return for_each_ref_in("refs/tags/", fn, cb_data);
1610 1611
}

1612 1613 1614 1615 1616
int for_each_tag_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
{
	return for_each_ref_in_submodule(submodule, "refs/tags/", fn, cb_data);
}

1617
int for_each_branch_ref(each_ref_fn fn, void *cb_data)
1618
{
1619
	return for_each_ref_in("refs/heads/", fn, cb_data);
1620 1621
}

1622 1623 1624 1625 1626
int for_each_branch_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
{
	return for_each_ref_in_submodule(submodule, "refs/heads/", fn, cb_data);
}

1627
int for_each_remote_ref(each_ref_fn fn, void *cb_data)
1628
{
1629
	return for_each_ref_in("refs/remotes/", fn, cb_data);
1630 1631
}

1632 1633 1634 1635 1636
int for_each_remote_ref_submodule(const char *submodule, each_ref_fn fn, void *cb_data)
{
	return for_each_ref_in_submodule(submodule, "refs/remotes/", fn, cb_data);
}

1637 1638
int for_each_replace_ref(each_ref_fn fn, void *cb_data)
{
1639
	return do_for_each_ref(NULL, "refs/replace/", fn, 13, 0, cb_data);
1640 1641
}

J
Josh Triplett 已提交
1642 1643 1644 1645 1646 1647 1648 1649
int head_ref_namespaced(each_ref_fn fn, void *cb_data)
{
	struct strbuf buf = STRBUF_INIT;
	int ret = 0;
	unsigned char sha1[20];
	int flag;

	strbuf_addf(&buf, "%sHEAD", get_git_namespace());
1650
	if (!read_ref_full(buf.buf, sha1, 1, &flag))
J
Josh Triplett 已提交
1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666
		ret = fn(buf.buf, sha1, flag, cb_data);
	strbuf_release(&buf);

	return ret;
}

int for_each_namespaced_ref(each_ref_fn fn, void *cb_data)
{
	struct strbuf buf = STRBUF_INIT;
	int ret;
	strbuf_addf(&buf, "%srefs/", get_git_namespace());
	ret = do_for_each_ref(NULL, buf.buf, fn, 0, 0, cb_data);
	strbuf_release(&buf);
	return ret;
}

1667 1668
int for_each_glob_ref_in(each_ref_fn fn, const char *pattern,
	const char *prefix, void *cb_data)
I
Ilari Liusvaara 已提交
1669 1670 1671 1672 1673
{
	struct strbuf real_pattern = STRBUF_INIT;
	struct ref_filter filter;
	int ret;

1674
	if (!prefix && prefixcmp(pattern, "refs/"))
I
Ilari Liusvaara 已提交
1675
		strbuf_addstr(&real_pattern, "refs/");
1676 1677
	else if (prefix)
		strbuf_addstr(&real_pattern, prefix);
I
Ilari Liusvaara 已提交
1678 1679
	strbuf_addstr(&real_pattern, pattern);

1680
	if (!has_glob_specials(pattern)) {
1681
		/* Append implied '/' '*' if not present. */
I
Ilari Liusvaara 已提交
1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696
		if (real_pattern.buf[real_pattern.len - 1] != '/')
			strbuf_addch(&real_pattern, '/');
		/* No need to check for '*', there is none. */
		strbuf_addch(&real_pattern, '*');
	}

	filter.pattern = real_pattern.buf;
	filter.fn = fn;
	filter.cb_data = cb_data;
	ret = for_each_ref(filter_refs, &filter);

	strbuf_release(&real_pattern);
	return ret;
}

1697 1698 1699 1700 1701
int for_each_glob_ref(each_ref_fn fn, const char *pattern, void *cb_data)
{
	return for_each_glob_ref_in(fn, pattern, NULL, cb_data);
}

1702 1703
int for_each_rawref(each_ref_fn fn, void *cb_data)
{
1704
	return do_for_each_ref(NULL, "", fn, 0,
1705
			       DO_FOR_EACH_INCLUDE_BROKEN, cb_data);
1706 1707
}

1708
const char *prettify_refname(const char *name)
1709 1710 1711 1712 1713 1714 1715 1716
{
	return name + (
		!prefixcmp(name, "refs/heads/") ? 11 :
		!prefixcmp(name, "refs/tags/") ? 10 :
		!prefixcmp(name, "refs/remotes/") ? 13 :
		0);
}

S
Steffen Prohaska 已提交
1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740
const char *ref_rev_parse_rules[] = {
	"%.*s",
	"refs/%.*s",
	"refs/tags/%.*s",
	"refs/heads/%.*s",
	"refs/remotes/%.*s",
	"refs/remotes/%.*s/HEAD",
	NULL
};

int refname_match(const char *abbrev_name, const char *full_name, const char **rules)
{
	const char **p;
	const int abbrev_name_len = strlen(abbrev_name);

	for (p = rules; *p; p++) {
		if (!strcmp(full_name, mkpath(*p, abbrev_name_len, abbrev_name))) {
			return 1;
		}
	}

	return 0;
}

J
Junio C Hamano 已提交
1741
static struct ref_lock *verify_lock(struct ref_lock *lock,
1742 1743
	const unsigned char *old_sha1, int mustexist)
{
1744
	if (read_ref_full(lock->ref_name, lock->old_sha1, mustexist, NULL)) {
1745
		error("Can't verify ref %s", lock->ref_name);
1746 1747 1748
		unlock_ref(lock);
		return NULL;
	}
1749
	if (hashcmp(lock->old_sha1, old_sha1)) {
1750
		error("Ref %s is at %s but expected %s", lock->ref_name,
1751 1752 1753 1754 1755 1756 1757
			sha1_to_hex(lock->old_sha1), sha1_to_hex(old_sha1));
		unlock_ref(lock);
		return NULL;
	}
	return lock;
}

1758
static int remove_empty_directories(const char *file)
1759 1760 1761 1762 1763
{
	/* we want to create a file but there is a directory there;
	 * if that is an empty directory (or a directory that contains
	 * only empty directories), remove them.
	 */
1764 1765
	struct strbuf path;
	int result;
1766

1767 1768 1769
	strbuf_init(&path, 20);
	strbuf_addstr(&path, file);

1770
	result = remove_dir_recursively(&path, REMOVE_DIR_EMPTY_ONLY);
1771 1772 1773 1774

	strbuf_release(&path);

	return result;
1775 1776
}

1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811
/*
 * *string and *len will only be substituted, and *string returned (for
 * later free()ing) if the string passed in is a magic short-hand form
 * to name a branch.
 */
static char *substitute_branch_name(const char **string, int *len)
{
	struct strbuf buf = STRBUF_INIT;
	int ret = interpret_branch_name(*string, &buf);

	if (ret == *len) {
		size_t size;
		*string = strbuf_detach(&buf, &size);
		*len = size;
		return (char *)*string;
	}

	return NULL;
}

int dwim_ref(const char *str, int len, unsigned char *sha1, char **ref)
{
	char *last_branch = substitute_branch_name(&str, &len);
	const char **p, *r;
	int refs_found = 0;

	*ref = NULL;
	for (p = ref_rev_parse_rules; *p; p++) {
		char fullref[PATH_MAX];
		unsigned char sha1_from_ref[20];
		unsigned char *this_result;
		int flag;

		this_result = refs_found ? sha1_from_ref : sha1;
		mksnpath(fullref, sizeof(fullref), *p, len, str);
1812
		r = resolve_ref_unsafe(fullref, this_result, 1, &flag);
1813 1814 1815 1816 1817
		if (r) {
			if (!refs_found++)
				*ref = xstrdup(r);
			if (!warn_ambiguous_refs)
				break;
1818
		} else if ((flag & REF_ISSYMREF) && strcmp(fullref, "HEAD")) {
1819
			warning("ignoring dangling symref %s.", fullref);
1820 1821 1822
		} else if ((flag & REF_ISBROKEN) && strchr(fullref, '/')) {
			warning("ignoring broken ref %s.", fullref);
		}
1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841
	}
	free(last_branch);
	return refs_found;
}

int dwim_log(const char *str, int len, unsigned char *sha1, char **log)
{
	char *last_branch = substitute_branch_name(&str, &len);
	const char **p;
	int logs_found = 0;

	*log = NULL;
	for (p = ref_rev_parse_rules; *p; p++) {
		struct stat st;
		unsigned char hash[20];
		char path[PATH_MAX];
		const char *ref, *it;

		mksnpath(path, sizeof(path), *p, len, str);
1842
		ref = resolve_ref_unsafe(path, hash, 1, NULL);
1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864
		if (!ref)
			continue;
		if (!stat(git_path("logs/%s", path), &st) &&
		    S_ISREG(st.st_mode))
			it = path;
		else if (strcmp(ref, path) &&
			 !stat(git_path("logs/%s", ref), &st) &&
			 S_ISREG(st.st_mode))
			it = ref;
		else
			continue;
		if (!logs_found++) {
			*log = xstrdup(it);
			hashcpy(sha1, hash);
		}
		if (!warn_ambiguous_refs)
			break;
	}
	free(last_branch);
	return logs_found;
}

1865 1866 1867
static struct ref_lock *lock_ref_sha1_basic(const char *refname,
					    const unsigned char *old_sha1,
					    int flags, int *type_p)
1868
{
1869
	char *ref_file;
1870
	const char *orig_refname = refname;
1871
	struct ref_lock *lock;
1872
	int last_errno = 0;
1873
	int type, lflags;
1874
	int mustexist = (old_sha1 && !is_null_sha1(old_sha1));
1875
	int missing = 0;
1876 1877 1878 1879

	lock = xcalloc(1, sizeof(struct ref_lock));
	lock->lock_fd = -1;

J
Junio C Hamano 已提交
1880
	refname = resolve_ref_unsafe(refname, lock->old_sha1, mustexist, &type);
1881
	if (!refname && errno == EISDIR) {
1882 1883 1884 1885 1886
		/* we are trying to lock foo but we used to
		 * have foo/bar which now does not exist;
		 * it is normal for the empty directory 'foo'
		 * to remain.
		 */
1887
		ref_file = git_path("%s", orig_refname);
1888 1889
		if (remove_empty_directories(ref_file)) {
			last_errno = errno;
1890
			error("there are still refs under '%s'", orig_refname);
1891 1892
			goto error_return;
		}
J
Junio C Hamano 已提交
1893
		refname = resolve_ref_unsafe(orig_refname, lock->old_sha1, mustexist, &type);
1894
	}
1895 1896
	if (type_p)
	    *type_p = type;
1897
	if (!refname) {
1898
		last_errno = errno;
1899
		error("unable to resolve reference %s: %s",
1900
			orig_refname, strerror(errno));
1901
		goto error_return;
1902
	}
1903
	missing = is_null_sha1(lock->old_sha1);
1904 1905 1906 1907 1908
	/* When the ref did not exist and we are creating it,
	 * make sure there is no existing ref that is packed
	 * whose name begins with our refname, nor a ref whose
	 * name is a proper prefix of our refname.
	 */
1909
	if (missing &&
1910
	     !is_refname_available(refname, NULL, get_packed_refs(get_ref_cache(NULL)))) {
1911
		last_errno = ENOTDIR;
1912
		goto error_return;
1913
	}
1914

1915
	lock->lk = xcalloc(1, sizeof(struct lock_file));
1916

1917 1918
	lflags = LOCK_DIE_ON_ERROR;
	if (flags & REF_NODEREF) {
1919
		refname = orig_refname;
1920 1921
		lflags |= LOCK_NODEREF;
	}
1922 1923 1924
	lock->ref_name = xstrdup(refname);
	lock->orig_ref_name = xstrdup(orig_refname);
	ref_file = git_path("%s", refname);
1925
	if (missing)
1926 1927 1928
		lock->force_write = 1;
	if ((flags & REF_NODEREF) && (type & REF_ISSYMREF))
		lock->force_write = 1;
1929

1930 1931 1932 1933 1934
	if (safe_create_leading_directories(ref_file)) {
		last_errno = errno;
		error("unable to create directory for %s", ref_file);
		goto error_return;
	}
1935

1936
	lock->lock_fd = hold_lock_file_for_update(lock->lk, ref_file, lflags);
1937
	return old_sha1 ? verify_lock(lock, old_sha1, mustexist) : lock;
1938 1939 1940 1941 1942

 error_return:
	unlock_ref(lock);
	errno = last_errno;
	return NULL;
1943 1944
}

1945
struct ref_lock *lock_ref_sha1(const char *refname, const unsigned char *old_sha1)
D
Daniel Barkalow 已提交
1946
{
P
Petr Baudis 已提交
1947
	char refpath[PATH_MAX];
1948
	if (check_refname_format(refname, 0))
1949
		return NULL;
1950
	strcpy(refpath, mkpath("refs/%s", refname));
1951
	return lock_ref_sha1_basic(refpath, old_sha1, 0, NULL);
1952 1953
}

1954 1955
struct ref_lock *lock_any_ref_for_update(const char *refname,
					 const unsigned char *old_sha1, int flags)
1956
{
1957
	if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL))
1958
		return NULL;
1959
	return lock_ref_sha1_basic(refname, old_sha1, flags, NULL);
J
Junio C Hamano 已提交
1960 1961
}

1962 1963 1964 1965 1966 1967
/*
 * Write an entry to the packed-refs file for the specified refname.
 * If peeled is non-NULL, write it as the entry's peeled value.
 */
static void write_packed_entry(int fd, char *refname, unsigned char *sha1,
			       unsigned char *peeled)
1968 1969 1970 1971
{
	char line[PATH_MAX + 100];
	int len;

1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991
	len = snprintf(line, sizeof(line), "%s %s\n",
		       sha1_to_hex(sha1), refname);
	/* this should not happen but just being defensive */
	if (len > sizeof(line))
		die("too long a refname '%s'", refname);
	write_or_die(fd, line, len);

	if (peeled) {
		if (snprintf(line, sizeof(line), "^%s\n",
			     sha1_to_hex(peeled)) != PEELED_LINE_LENGTH)
			die("internal error");
		write_or_die(fd, line, PEELED_LINE_LENGTH);
	}
}

static int repack_ref_fn(struct ref_entry *entry, void *cb_data)
{
	int *fd = cb_data;
	enum peel_status peel_status;

1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026
	if (entry->flag & REF_ISBROKEN) {
		/* This shouldn't happen to packed refs. */
		error("%s is broken!", entry->name);
		return 0;
	}
	if (!has_sha1_file(entry->u.value.sha1)) {
		unsigned char sha1[20];
		int flags;

		if (read_ref_full(entry->name, sha1, 0, &flags))
			/* We should at least have found the packed ref. */
			die("Internal error");
		if ((flags & REF_ISSYMREF) || !(flags & REF_ISPACKED))
			/*
			 * This packed reference is overridden by a
			 * loose reference, so it is OK that its value
			 * is no longer valid; for example, it might
			 * refer to an object that has been garbage
			 * collected.  For this purpose we don't even
			 * care whether the loose reference itself is
			 * invalid, broken, symbolic, etc.  Silently
			 * omit the packed reference from the output.
			 */
			return 0;
		/*
		 * There is no overriding loose reference, so the fact
		 * that this reference doesn't refer to a valid object
		 * indicates some kind of repository corruption.
		 * Report the problem, then omit the reference from
		 * the output.
		 */
		error("%s does not point to a valid object!", entry->name);
		return 0;
	}

2027 2028 2029 2030
	peel_status = peel_entry(entry);
	write_packed_entry(*fd, entry->name, entry->u.value.sha1,
			   peel_status == PEEL_PEELED ?
			   entry->u.value.peeled : NULL);
2031

2032 2033 2034
	return 0;
}

2035 2036
static struct lock_file packlock;

J
Junio C Hamano 已提交
2037 2038
static int repack_without_ref(const char *refname)
{
2039
	int fd;
2040
	struct ref_cache *refs = get_ref_cache(NULL);
2041 2042 2043 2044 2045
	struct ref_dir *packed;

	if (!get_packed_ref(refname))
		return 0; /* refname does not exist in packed refs */

2046 2047
	fd = hold_lock_file_for_update(&packlock, git_path("packed-refs"), 0);
	if (fd < 0) {
2048
		unable_to_lock_error(git_path("packed-refs"), errno);
J
Junio C Hamano 已提交
2049
		return error("cannot delete '%s' from packed refs", refname);
2050
	}
2051 2052
	clear_packed_ref_cache(refs);
	packed = get_packed_refs(refs);
2053 2054 2055 2056 2057 2058 2059 2060 2061
	/* Remove refname from the cache. */
	if (remove_entry(packed, refname) == -1) {
		/*
		 * The packed entry disappeared while we were
		 * acquiring the lock.
		 */
		rollback_lock_file(&packlock);
		return 0;
	}
2062
	write_or_die(fd, PACKED_REFS_HEADER, strlen(PACKED_REFS_HEADER));
2063
	do_for_each_entry_in_dir(packed, 0, repack_ref_fn, &fd);
J
Junio C Hamano 已提交
2064 2065 2066
	return commit_lock_file(&packlock);
}

M
Miklos Vajna 已提交
2067
int delete_ref(const char *refname, const unsigned char *sha1, int delopt)
J
Junio C Hamano 已提交
2068 2069
{
	struct ref_lock *lock;
M
Miklos Vajna 已提交
2070
	int err, i = 0, ret = 0, flag = 0;
J
Junio C Hamano 已提交
2071

2072
	lock = lock_ref_sha1_basic(refname, sha1, delopt, &flag);
J
Junio C Hamano 已提交
2073 2074
	if (!lock)
		return 1;
2075
	if (!(flag & REF_ISPACKED) || flag & REF_ISSYMREF) {
J
Junio C Hamano 已提交
2076
		/* loose */
2077 2078 2079
		i = strlen(lock->lk->filename) - 5; /* .lock */
		lock->lk->filename[i] = 0;
		err = unlink_or_warn(lock->lk->filename);
2080
		if (err && errno != ENOENT)
J
Junio C Hamano 已提交
2081
			ret = 1;
2082

2083
		lock->lk->filename[i] = '.';
J
Junio C Hamano 已提交
2084 2085 2086 2087 2088
	}
	/* removing the loose one could have resurrected an earlier
	 * packed one.  Also, if it was not loose we need to repack
	 * without it.
	 */
2089
	ret |= repack_without_ref(lock->ref_name);
J
Junio C Hamano 已提交
2090

2091
	unlink_or_warn(git_path("logs/%s", lock->ref_name));
2092
	clear_loose_ref_cache(get_ref_cache(NULL));
J
Junio C Hamano 已提交
2093 2094
	unlock_ref(lock);
	return ret;
2095 2096
}

2097 2098 2099 2100 2101 2102 2103 2104 2105
/*
 * People using contrib's git-new-workdir have .git/logs/refs ->
 * /some/other/path/.git/logs/refs, and that may live on another device.
 *
 * IOW, to avoid cross device rename errors, the temporary renamed log must
 * live into logs/refs.
 */
#define TMP_RENAMED_LOG  "logs/refs/.tmp-renamed-log"

2106
int rename_ref(const char *oldrefname, const char *newrefname, const char *logmsg)
2107 2108 2109 2110 2111
{
	unsigned char sha1[20], orig_sha1[20];
	int flag = 0, logmoved = 0;
	struct ref_lock *lock;
	struct stat loginfo;
2112
	int log = !lstat(git_path("logs/%s", oldrefname), &loginfo);
M
Miklos Vajna 已提交
2113
	const char *symref = NULL;
2114
	struct ref_cache *refs = get_ref_cache(NULL);
2115

2116
	if (log && S_ISLNK(loginfo.st_mode))
2117
		return error("reflog for %s is a symlink", oldrefname);
2118

J
Junio C Hamano 已提交
2119
	symref = resolve_ref_unsafe(oldrefname, orig_sha1, 1, &flag);
M
Miklos Vajna 已提交
2120
	if (flag & REF_ISSYMREF)
2121
		return error("refname %s is a symbolic ref, renaming it is not supported",
2122
			oldrefname);
M
Miklos Vajna 已提交
2123
	if (!symref)
2124
		return error("refname %s not found", oldrefname);
2125

2126
	if (!is_refname_available(newrefname, oldrefname, get_packed_refs(refs)))
2127 2128
		return 1;

2129
	if (!is_refname_available(newrefname, oldrefname, get_loose_refs(refs)))
2130 2131
		return 1;

2132
	if (log && rename(git_path("logs/%s", oldrefname), git_path(TMP_RENAMED_LOG)))
2133
		return error("unable to move logfile logs/%s to "TMP_RENAMED_LOG": %s",
2134
			oldrefname, strerror(errno));
2135

2136 2137
	if (delete_ref(oldrefname, orig_sha1, REF_NODEREF)) {
		error("unable to delete old %s", oldrefname);
2138 2139 2140
		goto rollback;
	}

2141 2142
	if (!read_ref_full(newrefname, sha1, 1, &flag) &&
	    delete_ref(newrefname, sha1, REF_NODEREF)) {
2143
		if (errno==EISDIR) {
2144 2145
			if (remove_empty_directories(git_path("%s", newrefname))) {
				error("Directory not empty: %s", newrefname);
2146 2147 2148
				goto rollback;
			}
		} else {
2149
			error("unable to delete existing %s", newrefname);
2150 2151 2152 2153
			goto rollback;
		}
	}

2154 2155
	if (log && safe_create_leading_directories(git_path("logs/%s", newrefname))) {
		error("unable to create directory for %s", newrefname);
2156 2157 2158 2159
		goto rollback;
	}

 retry:
2160
	if (log && rename(git_path(TMP_RENAMED_LOG), git_path("logs/%s", newrefname))) {
2161 2162 2163 2164 2165 2166
		if (errno==EISDIR || errno==ENOTDIR) {
			/*
			 * rename(a, b) when b is an existing
			 * directory ought to result in ISDIR, but
			 * Solaris 5.8 gives ENOTDIR.  Sheesh.
			 */
2167 2168
			if (remove_empty_directories(git_path("logs/%s", newrefname))) {
				error("Directory not empty: logs/%s", newrefname);
2169 2170 2171 2172
				goto rollback;
			}
			goto retry;
		} else {
2173
			error("unable to move logfile "TMP_RENAMED_LOG" to logs/%s: %s",
2174
				newrefname, strerror(errno));
2175 2176 2177 2178 2179
			goto rollback;
		}
	}
	logmoved = log;

2180
	lock = lock_ref_sha1_basic(newrefname, NULL, 0, NULL);
2181
	if (!lock) {
2182
		error("unable to lock %s for update", newrefname);
2183 2184 2185 2186
		goto rollback;
	}
	lock->force_write = 1;
	hashcpy(lock->old_sha1, orig_sha1);
2187
	if (write_ref_sha1(lock, orig_sha1, logmsg)) {
2188
		error("unable to write current sha1 into %s", newrefname);
2189 2190 2191 2192 2193 2194
		goto rollback;
	}

	return 0;

 rollback:
2195
	lock = lock_ref_sha1_basic(oldrefname, NULL, 0, NULL);
2196
	if (!lock) {
2197
		error("unable to lock %s for rollback", oldrefname);
2198 2199 2200 2201 2202 2203 2204
		goto rollbacklog;
	}

	lock->force_write = 1;
	flag = log_all_ref_updates;
	log_all_ref_updates = 0;
	if (write_ref_sha1(lock, orig_sha1, NULL))
2205
		error("unable to write current sha1 into %s", oldrefname);
2206 2207 2208
	log_all_ref_updates = flag;

 rollbacklog:
2209
	if (logmoved && rename(git_path("logs/%s", newrefname), git_path("logs/%s", oldrefname)))
2210
		error("unable to restore logfile %s from %s: %s",
2211
			oldrefname, newrefname, strerror(errno));
2212
	if (!logmoved && log &&
2213
	    rename(git_path(TMP_RENAMED_LOG), git_path("logs/%s", oldrefname)))
2214
		error("unable to restore logfile %s from "TMP_RENAMED_LOG": %s",
2215
			oldrefname, strerror(errno));
2216 2217 2218 2219

	return 1;
}

2220
int close_ref(struct ref_lock *lock)
2221 2222 2223 2224 2225 2226 2227
{
	if (close_lock_file(lock->lk))
		return -1;
	lock->lock_fd = -1;
	return 0;
}

2228
int commit_ref(struct ref_lock *lock)
2229 2230 2231 2232 2233 2234 2235
{
	if (commit_lock_file(lock->lk))
		return -1;
	lock->lock_fd = -1;
	return 0;
}

J
Junio C Hamano 已提交
2236
void unlock_ref(struct ref_lock *lock)
2237
{
B
Brandon Casey 已提交
2238 2239 2240
	/* Do not free lock->lk -- atexit() still looks at them */
	if (lock->lk)
		rollback_lock_file(lock->lk);
2241
	free(lock->ref_name);
2242
	free(lock->orig_ref_name);
2243 2244 2245
	free(lock);
}

2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271
/*
 * copy the reflog message msg to buf, which has been allocated sufficiently
 * large, while cleaning up the whitespaces.  Especially, convert LF to space,
 * because reflog file is one line per entry.
 */
static int copy_msg(char *buf, const char *msg)
{
	char *cp = buf;
	char c;
	int wasspace = 1;

	*cp++ = '\t';
	while ((c = *msg++)) {
		if (wasspace && isspace(c))
			continue;
		wasspace = isspace(c);
		if (wasspace)
			c = ' ';
		*cp++ = c;
	}
	while (buf < cp && isspace(cp[-1]))
		cp--;
	*cp++ = '\n';
	return cp - buf;
}

2272
int log_ref_setup(const char *refname, char *logfile, int bufsize)
2273
{
2274
	int logfd, oflags = O_APPEND | O_WRONLY;
2275

2276
	git_snpath(logfile, bufsize, "logs/%s", refname);
2277
	if (log_all_ref_updates &&
2278 2279 2280 2281
	    (!prefixcmp(refname, "refs/heads/") ||
	     !prefixcmp(refname, "refs/remotes/") ||
	     !prefixcmp(refname, "refs/notes/") ||
	     !strcmp(refname, "HEAD"))) {
2282
		if (safe_create_leading_directories(logfile) < 0)
2283
			return error("unable to create directory for %s",
2284
				     logfile);
2285 2286 2287
		oflags |= O_CREAT;
	}

2288
	logfd = open(logfile, oflags, 0666);
2289
	if (logfd < 0) {
J
Junio C Hamano 已提交
2290
		if (!(oflags & O_CREAT) && errno == ENOENT)
2291
			return 0;
2292 2293

		if ((oflags & O_CREAT) && errno == EISDIR) {
2294
			if (remove_empty_directories(logfile)) {
2295
				return error("There are still logs under '%s'",
2296
					     logfile);
2297
			}
2298
			logfd = open(logfile, oflags, 0666);
2299 2300 2301 2302
		}

		if (logfd < 0)
			return error("Unable to append to %s: %s",
2303
				     logfile, strerror(errno));
2304 2305
	}

2306
	adjust_shared_perm(logfile);
2307 2308 2309
	close(logfd);
	return 0;
}
2310

2311
static int log_ref_write(const char *refname, const unsigned char *old_sha1,
2312 2313 2314 2315 2316
			 const unsigned char *new_sha1, const char *msg)
{
	int logfd, result, written, oflags = O_APPEND | O_WRONLY;
	unsigned maxlen, len;
	int msglen;
2317
	char log_file[PATH_MAX];
2318 2319 2320 2321 2322 2323
	char *logrec;
	const char *committer;

	if (log_all_ref_updates < 0)
		log_all_ref_updates = !is_bare_repository();

2324
	result = log_ref_setup(refname, log_file, sizeof(log_file));
2325 2326 2327 2328 2329 2330
	if (result)
		return result;

	logfd = open(log_file, oflags);
	if (logfd < 0)
		return 0;
2331
	msglen = msg ? strlen(msg) : 0;
2332
	committer = git_committer_info(0);
2333 2334 2335
	maxlen = strlen(committer) + msglen + 100;
	logrec = xmalloc(maxlen);
	len = sprintf(logrec, "%s %s %s\n",
2336 2337
		      sha1_to_hex(old_sha1),
		      sha1_to_hex(new_sha1),
2338 2339
		      committer);
	if (msglen)
2340
		len += copy_msg(logrec + len - 1, msg) - 1;
2341
	written = len <= maxlen ? write_in_full(logfd, logrec, len) : -1;
2342
	free(logrec);
2343
	if (close(logfd) != 0 || written != len)
2344
		return error("Unable to append to %s", log_file);
2345 2346 2347
	return 0;
}

2348 2349 2350 2351 2352
static int is_branch(const char *refname)
{
	return !strcmp(refname, "HEAD") || !prefixcmp(refname, "refs/heads/");
}

2353 2354 2355 2356
int write_ref_sha1(struct ref_lock *lock,
	const unsigned char *sha1, const char *logmsg)
{
	static char term = '\n';
2357
	struct object *o;
2358 2359

	if (!lock)
D
Daniel Barkalow 已提交
2360
		return -1;
2361
	if (!lock->force_write && !hashcmp(lock->old_sha1, sha1)) {
2362 2363
		unlock_ref(lock);
		return 0;
D
Daniel Barkalow 已提交
2364
	}
2365 2366
	o = parse_object(sha1);
	if (!o) {
D
Dmitry Ivankov 已提交
2367
		error("Trying to write ref %s with nonexistent object %s",
2368 2369 2370 2371 2372 2373 2374 2375 2376 2377
			lock->ref_name, sha1_to_hex(sha1));
		unlock_ref(lock);
		return -1;
	}
	if (o->type != OBJ_COMMIT && is_branch(lock->ref_name)) {
		error("Trying to write non-commit object %s to branch %s",
			sha1_to_hex(sha1), lock->ref_name);
		unlock_ref(lock);
		return -1;
	}
2378 2379
	if (write_in_full(lock->lock_fd, sha1_to_hex(sha1), 40) != 40 ||
	    write_in_full(lock->lock_fd, &term, 1) != 1
2380
		|| close_ref(lock) < 0) {
2381
		error("Couldn't write %s", lock->lk->filename);
2382 2383 2384
		unlock_ref(lock);
		return -1;
	}
2385
	clear_loose_ref_cache(get_ref_cache(NULL));
N
Nicolas Pitre 已提交
2386 2387 2388
	if (log_ref_write(lock->ref_name, lock->old_sha1, sha1, logmsg) < 0 ||
	    (strcmp(lock->ref_name, lock->orig_ref_name) &&
	     log_ref_write(lock->orig_ref_name, lock->old_sha1, sha1, logmsg) < 0)) {
2389 2390 2391
		unlock_ref(lock);
		return -1;
	}
2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407
	if (strcmp(lock->orig_ref_name, "HEAD") != 0) {
		/*
		 * Special hack: If a branch is updated directly and HEAD
		 * points to it (may happen on the remote side of a push
		 * for example) then logically the HEAD reflog should be
		 * updated too.
		 * A generic solution implies reverse symref information,
		 * but finding all symrefs pointing to the given branch
		 * would be rather costly for this rare event (the direct
		 * update of a branch) to be worth it.  So let's cheat and
		 * check with HEAD only which should cover 99% of all usage
		 * scenarios (even 100% of the default ones).
		 */
		unsigned char head_sha1[20];
		int head_flag;
		const char *head_ref;
2408
		head_ref = resolve_ref_unsafe("HEAD", head_sha1, 1, &head_flag);
2409 2410 2411 2412
		if (head_ref && (head_flag & REF_ISSYMREF) &&
		    !strcmp(head_ref, lock->ref_name))
			log_ref_write("HEAD", lock->old_sha1, sha1, logmsg);
	}
2413
	if (commit_ref(lock)) {
2414
		error("Couldn't set %s", lock->ref_name);
2415 2416 2417 2418 2419
		unlock_ref(lock);
		return -1;
	}
	unlock_ref(lock);
	return 0;
D
Daniel Barkalow 已提交
2420
}
2421

2422 2423
int create_symref(const char *ref_target, const char *refs_heads_master,
		  const char *logmsg)
2424 2425 2426 2427
{
	const char *lockpath;
	char ref[1000];
	int fd, len, written;
2428
	char *git_HEAD = git_pathdup("%s", ref_target);
2429 2430 2431 2432
	unsigned char old_sha1[20], new_sha1[20];

	if (logmsg && read_ref(ref_target, old_sha1))
		hashclr(old_sha1);
2433

2434 2435 2436
	if (safe_create_leading_directories(git_HEAD) < 0)
		return error("unable to create directory for %s", git_HEAD);

2437 2438 2439 2440
#ifndef NO_SYMLINK_HEAD
	if (prefer_symlink_refs) {
		unlink(git_HEAD);
		if (!symlink(refs_heads_master, git_HEAD))
2441
			goto done;
2442 2443 2444 2445 2446 2447 2448
		fprintf(stderr, "no symlink - falling back to symbolic ref\n");
	}
#endif

	len = snprintf(ref, sizeof(ref), "ref: %s\n", refs_heads_master);
	if (sizeof(ref) <= len) {
		error("refname too long: %s", refs_heads_master);
2449
		goto error_free_return;
2450 2451 2452 2453 2454
	}
	lockpath = mkpath("%s.lock", git_HEAD);
	fd = open(lockpath, O_CREAT | O_EXCL | O_WRONLY, 0666);
	if (fd < 0) {
		error("Unable to open %s for writing", lockpath);
2455
		goto error_free_return;
2456 2457
	}
	written = write_in_full(fd, ref, len);
2458
	if (close(fd) != 0 || written != len) {
2459
		error("Unable to write to %s", lockpath);
2460
		goto error_unlink_return;
2461 2462 2463
	}
	if (rename(lockpath, git_HEAD) < 0) {
		error("Unable to create %s", git_HEAD);
2464
		goto error_unlink_return;
2465 2466 2467
	}
	if (adjust_shared_perm(git_HEAD)) {
		error("Unable to fix permissions on %s", lockpath);
2468
	error_unlink_return:
2469
		unlink_or_warn(lockpath);
2470 2471 2472
	error_free_return:
		free(git_HEAD);
		return -1;
2473
	}
2474

2475
#ifndef NO_SYMLINK_HEAD
2476
	done:
2477
#endif
2478 2479 2480
	if (logmsg && !read_ref(refs_heads_master, new_sha1))
		log_ref_write(ref_target, old_sha1, new_sha1, logmsg);

2481
	free(git_HEAD);
2482 2483 2484
	return 0;
}

2485 2486 2487 2488
static char *ref_msg(const char *line, const char *endp)
{
	const char *ep;
	line += 82;
P
Pierre Habouzit 已提交
2489 2490 2491 2492
	ep = memchr(line, '\n', endp - line);
	if (!ep)
		ep = endp;
	return xmemdupz(line, ep - line);
2493 2494
}

2495 2496 2497
int read_ref_at(const char *refname, unsigned long at_time, int cnt,
		unsigned char *sha1, char **msg,
		unsigned long *cutoff_time, int *cutoff_tz, int *cutoff_cnt)
2498
{
2499
	const char *logfile, *logdata, *logend, *rec, *lastgt, *lastrec;
2500
	char *tz_c;
J
Junio C Hamano 已提交
2501
	int logfd, tz, reccnt = 0;
2502 2503
	struct stat st;
	unsigned long date;
2504
	unsigned char logged_sha1[20];
2505
	void *log_mapped;
2506
	size_t mapsz;
2507

2508
	logfile = git_path("logs/%s", refname);
2509 2510
	logfd = open(logfile, O_RDONLY, 0);
	if (logfd < 0)
2511
		die_errno("Unable to read log '%s'", logfile);
2512 2513 2514
	fstat(logfd, &st);
	if (!st.st_size)
		die("Log %s is empty.", logfile);
2515 2516
	mapsz = xsize_t(st.st_size);
	log_mapped = xmmap(NULL, mapsz, PROT_READ, MAP_PRIVATE, logfd, 0);
2517
	logdata = log_mapped;
2518 2519
	close(logfd);

2520
	lastrec = NULL;
2521 2522
	rec = logend = logdata + st.st_size;
	while (logdata < rec) {
J
Junio C Hamano 已提交
2523
		reccnt++;
2524 2525
		if (logdata < rec && *(rec-1) == '\n')
			rec--;
2526 2527
		lastgt = NULL;
		while (logdata < rec && *(rec-1) != '\n') {
2528
			rec--;
2529 2530 2531 2532
			if (*rec == '>')
				lastgt = rec;
		}
		if (!lastgt)
2533
			die("Log %s is corrupt.", logfile);
2534
		date = strtoul(lastgt + 1, &tz_c, 10);
2535
		if (date <= at_time || cnt == 0) {
2536
			tz = strtoul(tz_c, NULL, 10);
2537 2538 2539 2540 2541 2542 2543
			if (msg)
				*msg = ref_msg(rec, logend);
			if (cutoff_time)
				*cutoff_time = date;
			if (cutoff_tz)
				*cutoff_tz = tz;
			if (cutoff_cnt)
2544
				*cutoff_cnt = reccnt - 1;
2545 2546 2547 2548 2549
			if (lastrec) {
				if (get_sha1_hex(lastrec, logged_sha1))
					die("Log %s is corrupt.", logfile);
				if (get_sha1_hex(rec + 41, sha1))
					die("Log %s is corrupt.", logfile);
2550
				if (hashcmp(logged_sha1, sha1)) {
2551
					warning("Log %s has gap after %s.",
2552
						logfile, show_date(date, tz, DATE_RFC2822));
2553
				}
J
Junio C Hamano 已提交
2554 2555
			}
			else if (date == at_time) {
2556 2557
				if (get_sha1_hex(rec + 41, sha1))
					die("Log %s is corrupt.", logfile);
J
Junio C Hamano 已提交
2558 2559
			}
			else {
2560 2561
				if (get_sha1_hex(rec + 41, logged_sha1))
					die("Log %s is corrupt.", logfile);
2562
				if (hashcmp(logged_sha1, sha1)) {
2563
					warning("Log %s unexpectedly ended on %s.",
2564
						logfile, show_date(date, tz, DATE_RFC2822));
2565 2566
				}
			}
2567
			munmap(log_mapped, mapsz);
2568 2569
			return 0;
		}
2570
		lastrec = rec;
2571 2572
		if (cnt > 0)
			cnt--;
2573 2574
	}

2575 2576 2577 2578
	rec = logdata;
	while (rec < logend && *rec != '>' && *rec != '\n')
		rec++;
	if (rec == logend || *rec == '\n')
2579
		die("Log %s is corrupt.", logfile);
2580
	date = strtoul(rec + 1, &tz_c, 10);
2581 2582 2583
	tz = strtoul(tz_c, NULL, 10);
	if (get_sha1_hex(logdata, sha1))
		die("Log %s is corrupt.", logfile);
J
Jeff King 已提交
2584 2585 2586 2587
	if (is_null_sha1(sha1)) {
		if (get_sha1_hex(logdata + 41, sha1))
			die("Log %s is corrupt.", logfile);
	}
2588 2589
	if (msg)
		*msg = ref_msg(logdata, logend);
2590
	munmap(log_mapped, mapsz);
2591 2592 2593 2594 2595 2596 2597 2598

	if (cutoff_time)
		*cutoff_time = date;
	if (cutoff_tz)
		*cutoff_tz = tz;
	if (cutoff_cnt)
		*cutoff_cnt = reccnt;
	return 1;
2599
}
2600

2601
int for_each_recent_reflog_ent(const char *refname, each_reflog_ent_fn fn, long ofs, void *cb_data)
2602 2603 2604
{
	const char *logfile;
	FILE *logfp;
2605
	struct strbuf sb = STRBUF_INIT;
2606
	int ret = 0;
2607

2608
	logfile = git_path("logs/%s", refname);
2609 2610
	logfp = fopen(logfile, "r");
	if (!logfp)
2611
		return -1;
2612 2613 2614 2615 2616 2617

	if (ofs) {
		struct stat statbuf;
		if (fstat(fileno(logfp), &statbuf) ||
		    statbuf.st_size < ofs ||
		    fseek(logfp, -ofs, SEEK_END) ||
2618
		    strbuf_getwholeline(&sb, logfp, '\n')) {
2619
			fclose(logfp);
2620
			strbuf_release(&sb);
2621
			return -1;
2622
		}
2623 2624
	}

2625
	while (!strbuf_getwholeline(&sb, logfp, '\n')) {
2626
		unsigned char osha1[20], nsha1[20];
2627 2628
		char *email_end, *message;
		unsigned long timestamp;
2629
		int tz;
2630 2631

		/* old SP new SP name <email> SP time TAB msg LF */
2632 2633 2634 2635
		if (sb.len < 83 || sb.buf[sb.len - 1] != '\n' ||
		    get_sha1_hex(sb.buf, osha1) || sb.buf[40] != ' ' ||
		    get_sha1_hex(sb.buf + 41, nsha1) || sb.buf[81] != ' ' ||
		    !(email_end = strchr(sb.buf + 82, '>')) ||
2636 2637 2638 2639 2640
		    email_end[1] != ' ' ||
		    !(timestamp = strtoul(email_end + 2, &message, 10)) ||
		    !message || message[0] != ' ' ||
		    (message[1] != '+' && message[1] != '-') ||
		    !isdigit(message[2]) || !isdigit(message[3]) ||
2641
		    !isdigit(message[4]) || !isdigit(message[5]))
2642
			continue; /* corrupt? */
2643 2644
		email_end[1] = '\0';
		tz = strtol(message + 1, NULL, 10);
2645 2646 2647 2648
		if (message[6] != '\t')
			message += 6;
		else
			message += 7;
2649 2650
		ret = fn(osha1, nsha1, sb.buf + 82, timestamp, tz, message,
			 cb_data);
2651
		if (ret)
2652
			break;
2653 2654
	}
	fclose(logfp);
2655
	strbuf_release(&sb);
2656
	return ret;
2657
}
J
Junio C Hamano 已提交
2658

2659
int for_each_reflog_ent(const char *refname, each_reflog_ent_fn fn, void *cb_data)
2660
{
2661
	return for_each_recent_reflog_ent(refname, fn, 0, cb_data);
2662 2663
}

2664 2665 2666 2667 2668 2669
/*
 * Call fn for each reflog in the namespace indicated by name.  name
 * must be empty or end with '/'.  Name will be used as a scratch
 * space, but its contents will be restored before return.
 */
static int do_for_each_reflog(struct strbuf *name, each_ref_fn fn, void *cb_data)
2670
{
2671
	DIR *d = opendir(git_path("logs/%s", name->buf));
2672
	int retval = 0;
2673
	struct dirent *de;
2674
	int oldlen = name->len;
2675

2676
	if (!d)
2677
		return name->len ? errno : 0;
2678

2679 2680
	while ((de = readdir(d)) != NULL) {
		struct stat st;
2681

2682 2683 2684 2685
		if (de->d_name[0] == '.')
			continue;
		if (has_extension(de->d_name, ".lock"))
			continue;
2686 2687 2688
		strbuf_addstr(name, de->d_name);
		if (stat(git_path("logs/%s", name->buf), &st) < 0) {
			; /* silently ignore */
2689
		} else {
2690
			if (S_ISDIR(st.st_mode)) {
2691 2692
				strbuf_addch(name, '/');
				retval = do_for_each_reflog(name, fn, cb_data);
2693 2694
			} else {
				unsigned char sha1[20];
2695 2696
				if (read_ref_full(name->buf, sha1, 0, NULL))
					retval = error("bad ref for %s", name->buf);
2697
				else
2698
					retval = fn(name->buf, sha1, 0, cb_data);
2699 2700 2701 2702
			}
			if (retval)
				break;
		}
2703
		strbuf_setlen(name, oldlen);
2704
	}
2705
	closedir(d);
2706 2707 2708 2709 2710
	return retval;
}

int for_each_reflog(each_ref_fn fn, void *cb_data)
{
2711 2712 2713 2714 2715 2716
	int retval;
	struct strbuf name;
	strbuf_init(&name, PATH_MAX);
	retval = do_for_each_reflog(&name, fn, cb_data);
	strbuf_release(&name);
	return retval;
2717
}
C
Carlos Rica 已提交
2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744

int update_ref(const char *action, const char *refname,
		const unsigned char *sha1, const unsigned char *oldval,
		int flags, enum action_on_err onerr)
{
	static struct ref_lock *lock;
	lock = lock_any_ref_for_update(refname, oldval, flags);
	if (!lock) {
		const char *str = "Cannot lock the ref '%s'.";
		switch (onerr) {
		case MSG_ON_ERR: error(str, refname); break;
		case DIE_ON_ERR: die(str, refname); break;
		case QUIET_ON_ERR: break;
		}
		return 1;
	}
	if (write_ref_sha1(lock, sha1, action) < 0) {
		const char *str = "Cannot update the ref '%s'.";
		switch (onerr) {
		case MSG_ON_ERR: error(str, refname); break;
		case DIE_ON_ERR: die(str, refname); break;
		case QUIET_ON_ERR: break;
		}
		return 1;
	}
	return 0;
}
2745

2746
struct ref *find_ref_by_name(const struct ref *list, const char *name)
2747 2748 2749
{
	for ( ; list; list = list->next)
		if (!strcmp(list->name, name))
2750
			return (struct ref *)list;
2751 2752
	return NULL;
}
J
Jeff King 已提交
2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776

/*
 * generate a format suitable for scanf from a ref_rev_parse_rules
 * rule, that is replace the "%.*s" spec with a "%s" spec
 */
static void gen_scanf_fmt(char *scanf_fmt, const char *rule)
{
	char *spec;

	spec = strstr(rule, "%.*s");
	if (!spec || strstr(spec + 4, "%.*s"))
		die("invalid rule in ref_rev_parse_rules: %s", rule);

	/* copy all until spec */
	strncpy(scanf_fmt, rule, spec - rule);
	scanf_fmt[spec - rule] = '\0';
	/* copy new spec */
	strcat(scanf_fmt, "%s");
	/* copy remaining rule */
	strcat(scanf_fmt, spec + 4);

	return;
}

2777
char *shorten_unambiguous_ref(const char *refname, int strict)
J
Jeff King 已提交
2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805
{
	int i;
	static char **scanf_fmts;
	static int nr_rules;
	char *short_name;

	/* pre generate scanf formats from ref_rev_parse_rules[] */
	if (!nr_rules) {
		size_t total_len = 0;

		/* the rule list is NULL terminated, count them first */
		for (; ref_rev_parse_rules[nr_rules]; nr_rules++)
			/* no +1 because strlen("%s") < strlen("%.*s") */
			total_len += strlen(ref_rev_parse_rules[nr_rules]);

		scanf_fmts = xmalloc(nr_rules * sizeof(char *) + total_len);

		total_len = 0;
		for (i = 0; i < nr_rules; i++) {
			scanf_fmts[i] = (char *)&scanf_fmts[nr_rules]
					+ total_len;
			gen_scanf_fmt(scanf_fmts[i], ref_rev_parse_rules[i]);
			total_len += strlen(ref_rev_parse_rules[i]);
		}
	}

	/* bail out if there are no rules */
	if (!nr_rules)
2806
		return xstrdup(refname);
J
Jeff King 已提交
2807

2808 2809
	/* buffer for scanf result, at most refname must fit */
	short_name = xstrdup(refname);
J
Jeff King 已提交
2810 2811 2812 2813

	/* skip first rule, it will always match */
	for (i = nr_rules - 1; i > 0 ; --i) {
		int j;
2814
		int rules_to_fail = i;
J
Jeff King 已提交
2815 2816
		int short_name_len;

2817
		if (1 != sscanf(refname, scanf_fmts[i], short_name))
J
Jeff King 已提交
2818 2819 2820 2821
			continue;

		short_name_len = strlen(short_name);

2822 2823 2824 2825 2826 2827 2828
		/*
		 * in strict mode, all (except the matched one) rules
		 * must fail to resolve to a valid non-ambiguous ref
		 */
		if (strict)
			rules_to_fail = nr_rules;

J
Jeff King 已提交
2829 2830 2831 2832
		/*
		 * check if the short name resolves to a valid ref,
		 * but use only rules prior to the matched one
		 */
2833
		for (j = 0; j < rules_to_fail; j++) {
J
Jeff King 已提交
2834 2835 2836
			const char *rule = ref_rev_parse_rules[j];
			char refname[PATH_MAX];

2837 2838 2839 2840
			/* skip matched rule */
			if (i == j)
				continue;

J
Jeff King 已提交
2841 2842 2843 2844 2845 2846 2847
			/*
			 * the short name is ambiguous, if it resolves
			 * (with this previous rule) to a valid ref
			 * read_ref() returns 0 on success
			 */
			mksnpath(refname, sizeof(refname),
				 rule, short_name_len, short_name);
2848
			if (ref_exists(refname))
J
Jeff King 已提交
2849 2850 2851 2852 2853 2854 2855
				break;
		}

		/*
		 * short name is non-ambiguous if all previous rules
		 * haven't resolved to a valid ref
		 */
2856
		if (j == rules_to_fail)
J
Jeff King 已提交
2857 2858 2859 2860
			return short_name;
	}

	free(short_name);
2861
	return xstrdup(refname);
J
Jeff King 已提交
2862
}