]> git.ipfire.org Git - thirdparty/git.git/blob - refs.c
Merge branch 'rs/name-rev-with-mempool'
[thirdparty/git.git] / refs.c
1 /*
2 * The backend-independent part of the reference module.
3 */
4
5 #include "git-compat-util.h"
6 #include "advice.h"
7 #include "config.h"
8 #include "environment.h"
9 #include "hashmap.h"
10 #include "gettext.h"
11 #include "hex.h"
12 #include "lockfile.h"
13 #include "iterator.h"
14 #include "refs.h"
15 #include "refs/refs-internal.h"
16 #include "run-command.h"
17 #include "hook.h"
18 #include "object-name.h"
19 #include "object-store-ll.h"
20 #include "object.h"
21 #include "path.h"
22 #include "tag.h"
23 #include "submodule.h"
24 #include "worktree.h"
25 #include "strvec.h"
26 #include "repository.h"
27 #include "setup.h"
28 #include "sigchain.h"
29 #include "date.h"
30 #include "commit.h"
31 #include "wildmatch.h"
32
33 /*
34 * List of all available backends
35 */
36 static const struct ref_storage_be *refs_backends[] = {
37 [REF_STORAGE_FORMAT_FILES] = &refs_be_files,
38 [REF_STORAGE_FORMAT_REFTABLE] = &refs_be_reftable,
39 };
40
41 static const struct ref_storage_be *find_ref_storage_backend(unsigned int ref_storage_format)
42 {
43 if (ref_storage_format < ARRAY_SIZE(refs_backends))
44 return refs_backends[ref_storage_format];
45 return NULL;
46 }
47
48 unsigned int ref_storage_format_by_name(const char *name)
49 {
50 for (unsigned int i = 0; i < ARRAY_SIZE(refs_backends); i++)
51 if (refs_backends[i] && !strcmp(refs_backends[i]->name, name))
52 return i;
53 return REF_STORAGE_FORMAT_UNKNOWN;
54 }
55
56 const char *ref_storage_format_to_name(unsigned int ref_storage_format)
57 {
58 const struct ref_storage_be *be = find_ref_storage_backend(ref_storage_format);
59 if (!be)
60 return "unknown";
61 return be->name;
62 }
63
64 /*
65 * How to handle various characters in refnames:
66 * 0: An acceptable character for refs
67 * 1: End-of-component
68 * 2: ., look for a preceding . to reject .. in refs
69 * 3: {, look for a preceding @ to reject @{ in refs
70 * 4: A bad character: ASCII control characters, and
71 * ":", "?", "[", "\", "^", "~", SP, or TAB
72 * 5: *, reject unless REFNAME_REFSPEC_PATTERN is set
73 */
74 static unsigned char refname_disposition[256] = {
75 1, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
76 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
77 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 2, 1,
78 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 4,
79 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
80 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4, 4, 0, 4, 0,
81 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
82 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 0, 0, 4, 4
83 };
84
85 struct ref_namespace_info ref_namespace[] = {
86 [NAMESPACE_HEAD] = {
87 .ref = "HEAD",
88 .decoration = DECORATION_REF_HEAD,
89 .exact = 1,
90 },
91 [NAMESPACE_BRANCHES] = {
92 .ref = "refs/heads/",
93 .decoration = DECORATION_REF_LOCAL,
94 },
95 [NAMESPACE_TAGS] = {
96 .ref = "refs/tags/",
97 .decoration = DECORATION_REF_TAG,
98 },
99 [NAMESPACE_REMOTE_REFS] = {
100 /*
101 * The default refspec for new remotes copies refs from
102 * refs/heads/ on the remote into refs/remotes/<remote>/.
103 * As such, "refs/remotes/" has special handling.
104 */
105 .ref = "refs/remotes/",
106 .decoration = DECORATION_REF_REMOTE,
107 },
108 [NAMESPACE_STASH] = {
109 /*
110 * The single ref "refs/stash" stores the latest stash.
111 * Older stashes can be found in the reflog.
112 */
113 .ref = "refs/stash",
114 .exact = 1,
115 .decoration = DECORATION_REF_STASH,
116 },
117 [NAMESPACE_REPLACE] = {
118 /*
119 * This namespace allows Git to act as if one object ID
120 * points to the content of another. Unlike the other
121 * ref namespaces, this one can be changed by the
122 * GIT_REPLACE_REF_BASE environment variable. This
123 * .namespace value will be overwritten in setup_git_env().
124 */
125 .ref = "refs/replace/",
126 .decoration = DECORATION_GRAFTED,
127 },
128 [NAMESPACE_NOTES] = {
129 /*
130 * The refs/notes/commit ref points to the tip of a
131 * parallel commit history that adds metadata to commits
132 * in the normal history. This ref can be overwritten
133 * by the core.notesRef config variable or the
134 * GIT_NOTES_REFS environment variable.
135 */
136 .ref = "refs/notes/commit",
137 .exact = 1,
138 },
139 [NAMESPACE_PREFETCH] = {
140 /*
141 * Prefetch refs are written by the background 'fetch'
142 * maintenance task. It allows faster foreground fetches
143 * by advertising these previously-downloaded tips without
144 * updating refs/remotes/ without user intervention.
145 */
146 .ref = "refs/prefetch/",
147 },
148 [NAMESPACE_REWRITTEN] = {
149 /*
150 * Rewritten refs are used by the 'label' command in the
151 * sequencer. These are particularly useful during an
152 * interactive rebase that uses the 'merge' command.
153 */
154 .ref = "refs/rewritten/",
155 },
156 };
157
158 void update_ref_namespace(enum ref_namespace namespace, char *ref)
159 {
160 struct ref_namespace_info *info = &ref_namespace[namespace];
161 if (info->ref_updated)
162 free(info->ref);
163 info->ref = ref;
164 info->ref_updated = 1;
165 }
166
167 /*
168 * Try to read one refname component from the front of refname.
169 * Return the length of the component found, or -1 if the component is
170 * not legal. It is legal if it is something reasonable to have under
171 * ".git/refs/"; We do not like it if:
172 *
173 * - it begins with ".", or
174 * - it has double dots "..", or
175 * - it has ASCII control characters, or
176 * - it has ":", "?", "[", "\", "^", "~", SP, or TAB anywhere, or
177 * - it has "*" anywhere unless REFNAME_REFSPEC_PATTERN is set, or
178 * - it ends with a "/", or
179 * - it ends with ".lock", or
180 * - it contains a "@{" portion
181 *
182 * When sanitized is not NULL, instead of rejecting the input refname
183 * as an error, try to come up with a usable replacement for the input
184 * refname in it.
185 */
186 static int check_refname_component(const char *refname, int *flags,
187 struct strbuf *sanitized)
188 {
189 const char *cp;
190 char last = '\0';
191 size_t component_start = 0; /* garbage - not a reasonable initial value */
192
193 if (sanitized)
194 component_start = sanitized->len;
195
196 for (cp = refname; ; cp++) {
197 int ch = *cp & 255;
198 unsigned char disp = refname_disposition[ch];
199
200 if (sanitized && disp != 1)
201 strbuf_addch(sanitized, ch);
202
203 switch (disp) {
204 case 1:
205 goto out;
206 case 2:
207 if (last == '.') { /* Refname contains "..". */
208 if (sanitized)
209 /* collapse ".." to single "." */
210 strbuf_setlen(sanitized, sanitized->len - 1);
211 else
212 return -1;
213 }
214 break;
215 case 3:
216 if (last == '@') { /* Refname contains "@{". */
217 if (sanitized)
218 sanitized->buf[sanitized->len-1] = '-';
219 else
220 return -1;
221 }
222 break;
223 case 4:
224 /* forbidden char */
225 if (sanitized)
226 sanitized->buf[sanitized->len-1] = '-';
227 else
228 return -1;
229 break;
230 case 5:
231 if (!(*flags & REFNAME_REFSPEC_PATTERN)) {
232 /* refspec can't be a pattern */
233 if (sanitized)
234 sanitized->buf[sanitized->len-1] = '-';
235 else
236 return -1;
237 }
238
239 /*
240 * Unset the pattern flag so that we only accept
241 * a single asterisk for one side of refspec.
242 */
243 *flags &= ~ REFNAME_REFSPEC_PATTERN;
244 break;
245 }
246 last = ch;
247 }
248 out:
249 if (cp == refname)
250 return 0; /* Component has zero length. */
251
252 if (refname[0] == '.') { /* Component starts with '.'. */
253 if (sanitized)
254 sanitized->buf[component_start] = '-';
255 else
256 return -1;
257 }
258 if (cp - refname >= LOCK_SUFFIX_LEN &&
259 !memcmp(cp - LOCK_SUFFIX_LEN, LOCK_SUFFIX, LOCK_SUFFIX_LEN)) {
260 if (!sanitized)
261 return -1;
262 /* Refname ends with ".lock". */
263 while (strbuf_strip_suffix(sanitized, LOCK_SUFFIX)) {
264 /* try again in case we have .lock.lock */
265 }
266 }
267 return cp - refname;
268 }
269
270 static int check_or_sanitize_refname(const char *refname, int flags,
271 struct strbuf *sanitized)
272 {
273 int component_len, component_count = 0;
274
275 if (!strcmp(refname, "@")) {
276 /* Refname is a single character '@'. */
277 if (sanitized)
278 strbuf_addch(sanitized, '-');
279 else
280 return -1;
281 }
282
283 while (1) {
284 if (sanitized && sanitized->len)
285 strbuf_complete(sanitized, '/');
286
287 /* We are at the start of a path component. */
288 component_len = check_refname_component(refname, &flags,
289 sanitized);
290 if (sanitized && component_len == 0)
291 ; /* OK, omit empty component */
292 else if (component_len <= 0)
293 return -1;
294
295 component_count++;
296 if (refname[component_len] == '\0')
297 break;
298 /* Skip to next component. */
299 refname += component_len + 1;
300 }
301
302 if (refname[component_len - 1] == '.') {
303 /* Refname ends with '.'. */
304 if (sanitized)
305 ; /* omit ending dot */
306 else
307 return -1;
308 }
309 if (!(flags & REFNAME_ALLOW_ONELEVEL) && component_count < 2)
310 return -1; /* Refname has only one component. */
311 return 0;
312 }
313
314 int check_refname_format(const char *refname, int flags)
315 {
316 return check_or_sanitize_refname(refname, flags, NULL);
317 }
318
319 void sanitize_refname_component(const char *refname, struct strbuf *out)
320 {
321 if (check_or_sanitize_refname(refname, REFNAME_ALLOW_ONELEVEL, out))
322 BUG("sanitizing refname '%s' check returned error", refname);
323 }
324
325 int refname_is_safe(const char *refname)
326 {
327 const char *rest;
328
329 if (skip_prefix(refname, "refs/", &rest)) {
330 char *buf;
331 int result;
332 size_t restlen = strlen(rest);
333
334 /* rest must not be empty, or start or end with "/" */
335 if (!restlen || *rest == '/' || rest[restlen - 1] == '/')
336 return 0;
337
338 /*
339 * Does the refname try to escape refs/?
340 * For example: refs/foo/../bar is safe but refs/foo/../../bar
341 * is not.
342 */
343 buf = xmallocz(restlen);
344 result = !normalize_path_copy(buf, rest) && !strcmp(buf, rest);
345 free(buf);
346 return result;
347 }
348
349 do {
350 if (!isupper(*refname) && *refname != '_')
351 return 0;
352 refname++;
353 } while (*refname);
354 return 1;
355 }
356
357 /*
358 * Return true if refname, which has the specified oid and flags, can
359 * be resolved to an object in the database. If the referred-to object
360 * does not exist, emit a warning and return false.
361 */
362 int ref_resolves_to_object(const char *refname,
363 struct repository *repo,
364 const struct object_id *oid,
365 unsigned int flags)
366 {
367 if (flags & REF_ISBROKEN)
368 return 0;
369 if (!repo_has_object_file(repo, oid)) {
370 error(_("%s does not point to a valid object!"), refname);
371 return 0;
372 }
373 return 1;
374 }
375
376 char *refs_resolve_refdup(struct ref_store *refs,
377 const char *refname, int resolve_flags,
378 struct object_id *oid, int *flags)
379 {
380 const char *result;
381
382 result = refs_resolve_ref_unsafe(refs, refname, resolve_flags,
383 oid, flags);
384 return xstrdup_or_null(result);
385 }
386
387 char *resolve_refdup(const char *refname, int resolve_flags,
388 struct object_id *oid, int *flags)
389 {
390 return refs_resolve_refdup(get_main_ref_store(the_repository),
391 refname, resolve_flags,
392 oid, flags);
393 }
394
395 /* The argument to for_each_filter_refs */
396 struct for_each_ref_filter {
397 const char *pattern;
398 const char *prefix;
399 each_ref_fn *fn;
400 void *cb_data;
401 };
402
403 int read_ref_full(const char *refname, int resolve_flags, struct object_id *oid, int *flags)
404 {
405 struct ref_store *refs = get_main_ref_store(the_repository);
406
407 if (refs_resolve_ref_unsafe(refs, refname, resolve_flags,
408 oid, flags))
409 return 0;
410 return -1;
411 }
412
413 int read_ref(const char *refname, struct object_id *oid)
414 {
415 return read_ref_full(refname, RESOLVE_REF_READING, oid, NULL);
416 }
417
418 int refs_ref_exists(struct ref_store *refs, const char *refname)
419 {
420 return !!refs_resolve_ref_unsafe(refs, refname, RESOLVE_REF_READING,
421 NULL, NULL);
422 }
423
424 int ref_exists(const char *refname)
425 {
426 return refs_ref_exists(get_main_ref_store(the_repository), refname);
427 }
428
429 static int for_each_filter_refs(const char *refname,
430 const struct object_id *oid,
431 int flags, void *data)
432 {
433 struct for_each_ref_filter *filter = data;
434
435 if (wildmatch(filter->pattern, refname, 0))
436 return 0;
437 if (filter->prefix)
438 skip_prefix(refname, filter->prefix, &refname);
439 return filter->fn(refname, oid, flags, filter->cb_data);
440 }
441
442 enum peel_status peel_object(const struct object_id *name, struct object_id *oid)
443 {
444 struct object *o = lookup_unknown_object(the_repository, name);
445
446 if (o->type == OBJ_NONE) {
447 int type = oid_object_info(the_repository, name, NULL);
448 if (type < 0 || !object_as_type(o, type, 0))
449 return PEEL_INVALID;
450 }
451
452 if (o->type != OBJ_TAG)
453 return PEEL_NON_TAG;
454
455 o = deref_tag_noverify(o);
456 if (!o)
457 return PEEL_INVALID;
458
459 oidcpy(oid, &o->oid);
460 return PEEL_PEELED;
461 }
462
463 struct warn_if_dangling_data {
464 FILE *fp;
465 const char *refname;
466 const struct string_list *refnames;
467 const char *msg_fmt;
468 };
469
470 static int warn_if_dangling_symref(const char *refname,
471 const struct object_id *oid UNUSED,
472 int flags, void *cb_data)
473 {
474 struct warn_if_dangling_data *d = cb_data;
475 const char *resolves_to;
476
477 if (!(flags & REF_ISSYMREF))
478 return 0;
479
480 resolves_to = resolve_ref_unsafe(refname, 0, NULL, NULL);
481 if (!resolves_to
482 || (d->refname
483 ? strcmp(resolves_to, d->refname)
484 : !string_list_has_string(d->refnames, resolves_to))) {
485 return 0;
486 }
487
488 fprintf(d->fp, d->msg_fmt, refname);
489 fputc('\n', d->fp);
490 return 0;
491 }
492
493 void warn_dangling_symref(FILE *fp, const char *msg_fmt, const char *refname)
494 {
495 struct warn_if_dangling_data data;
496
497 data.fp = fp;
498 data.refname = refname;
499 data.refnames = NULL;
500 data.msg_fmt = msg_fmt;
501 for_each_rawref(warn_if_dangling_symref, &data);
502 }
503
504 void warn_dangling_symrefs(FILE *fp, const char *msg_fmt, const struct string_list *refnames)
505 {
506 struct warn_if_dangling_data data;
507
508 data.fp = fp;
509 data.refname = NULL;
510 data.refnames = refnames;
511 data.msg_fmt = msg_fmt;
512 for_each_rawref(warn_if_dangling_symref, &data);
513 }
514
515 int refs_for_each_tag_ref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
516 {
517 return refs_for_each_ref_in(refs, "refs/tags/", fn, cb_data);
518 }
519
520 int for_each_tag_ref(each_ref_fn fn, void *cb_data)
521 {
522 return refs_for_each_tag_ref(get_main_ref_store(the_repository), fn, cb_data);
523 }
524
525 int refs_for_each_branch_ref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
526 {
527 return refs_for_each_ref_in(refs, "refs/heads/", fn, cb_data);
528 }
529
530 int for_each_branch_ref(each_ref_fn fn, void *cb_data)
531 {
532 return refs_for_each_branch_ref(get_main_ref_store(the_repository), fn, cb_data);
533 }
534
535 int refs_for_each_remote_ref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
536 {
537 return refs_for_each_ref_in(refs, "refs/remotes/", fn, cb_data);
538 }
539
540 int for_each_remote_ref(each_ref_fn fn, void *cb_data)
541 {
542 return refs_for_each_remote_ref(get_main_ref_store(the_repository), fn, cb_data);
543 }
544
545 int head_ref_namespaced(each_ref_fn fn, void *cb_data)
546 {
547 struct strbuf buf = STRBUF_INIT;
548 int ret = 0;
549 struct object_id oid;
550 int flag;
551
552 strbuf_addf(&buf, "%sHEAD", get_git_namespace());
553 if (!read_ref_full(buf.buf, RESOLVE_REF_READING, &oid, &flag))
554 ret = fn(buf.buf, &oid, flag, cb_data);
555 strbuf_release(&buf);
556
557 return ret;
558 }
559
560 void normalize_glob_ref(struct string_list_item *item, const char *prefix,
561 const char *pattern)
562 {
563 struct strbuf normalized_pattern = STRBUF_INIT;
564
565 if (*pattern == '/')
566 BUG("pattern must not start with '/'");
567
568 if (prefix)
569 strbuf_addstr(&normalized_pattern, prefix);
570 else if (!starts_with(pattern, "refs/") &&
571 strcmp(pattern, "HEAD"))
572 strbuf_addstr(&normalized_pattern, "refs/");
573 /*
574 * NEEDSWORK: Special case other symrefs such as REBASE_HEAD,
575 * MERGE_HEAD, etc.
576 */
577
578 strbuf_addstr(&normalized_pattern, pattern);
579 strbuf_strip_suffix(&normalized_pattern, "/");
580
581 item->string = strbuf_detach(&normalized_pattern, NULL);
582 item->util = has_glob_specials(pattern) ? NULL : item->string;
583 strbuf_release(&normalized_pattern);
584 }
585
586 int for_each_glob_ref_in(each_ref_fn fn, const char *pattern,
587 const char *prefix, void *cb_data)
588 {
589 struct strbuf real_pattern = STRBUF_INIT;
590 struct for_each_ref_filter filter;
591 int ret;
592
593 if (!prefix && !starts_with(pattern, "refs/"))
594 strbuf_addstr(&real_pattern, "refs/");
595 else if (prefix)
596 strbuf_addstr(&real_pattern, prefix);
597 strbuf_addstr(&real_pattern, pattern);
598
599 if (!has_glob_specials(pattern)) {
600 /* Append implied '/' '*' if not present. */
601 strbuf_complete(&real_pattern, '/');
602 /* No need to check for '*', there is none. */
603 strbuf_addch(&real_pattern, '*');
604 }
605
606 filter.pattern = real_pattern.buf;
607 filter.prefix = prefix;
608 filter.fn = fn;
609 filter.cb_data = cb_data;
610 ret = for_each_ref(for_each_filter_refs, &filter);
611
612 strbuf_release(&real_pattern);
613 return ret;
614 }
615
616 int for_each_glob_ref(each_ref_fn fn, const char *pattern, void *cb_data)
617 {
618 return for_each_glob_ref_in(fn, pattern, NULL, cb_data);
619 }
620
621 const char *prettify_refname(const char *name)
622 {
623 if (skip_prefix(name, "refs/heads/", &name) ||
624 skip_prefix(name, "refs/tags/", &name) ||
625 skip_prefix(name, "refs/remotes/", &name))
626 ; /* nothing */
627 return name;
628 }
629
630 static const char *ref_rev_parse_rules[] = {
631 "%.*s",
632 "refs/%.*s",
633 "refs/tags/%.*s",
634 "refs/heads/%.*s",
635 "refs/remotes/%.*s",
636 "refs/remotes/%.*s/HEAD",
637 NULL
638 };
639
640 #define NUM_REV_PARSE_RULES (ARRAY_SIZE(ref_rev_parse_rules) - 1)
641
642 /*
643 * Is it possible that the caller meant full_name with abbrev_name?
644 * If so return a non-zero value to signal "yes"; the magnitude of
645 * the returned value gives the precedence used for disambiguation.
646 *
647 * If abbrev_name cannot mean full_name, return 0.
648 */
649 int refname_match(const char *abbrev_name, const char *full_name)
650 {
651 const char **p;
652 const int abbrev_name_len = strlen(abbrev_name);
653 const int num_rules = NUM_REV_PARSE_RULES;
654
655 for (p = ref_rev_parse_rules; *p; p++)
656 if (!strcmp(full_name, mkpath(*p, abbrev_name_len, abbrev_name)))
657 return &ref_rev_parse_rules[num_rules] - p;
658
659 return 0;
660 }
661
662 /*
663 * Given a 'prefix' expand it by the rules in 'ref_rev_parse_rules' and add
664 * the results to 'prefixes'
665 */
666 void expand_ref_prefix(struct strvec *prefixes, const char *prefix)
667 {
668 const char **p;
669 int len = strlen(prefix);
670
671 for (p = ref_rev_parse_rules; *p; p++)
672 strvec_pushf(prefixes, *p, len, prefix);
673 }
674
675 static const char default_branch_name_advice[] = N_(
676 "Using '%s' as the name for the initial branch. This default branch name\n"
677 "is subject to change. To configure the initial branch name to use in all\n"
678 "of your new repositories, which will suppress this warning, call:\n"
679 "\n"
680 "\tgit config --global init.defaultBranch <name>\n"
681 "\n"
682 "Names commonly chosen instead of 'master' are 'main', 'trunk' and\n"
683 "'development'. The just-created branch can be renamed via this command:\n"
684 "\n"
685 "\tgit branch -m <name>\n"
686 );
687
688 char *repo_default_branch_name(struct repository *r, int quiet)
689 {
690 const char *config_key = "init.defaultbranch";
691 const char *config_display_key = "init.defaultBranch";
692 char *ret = NULL, *full_ref;
693 const char *env = getenv("GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME");
694
695 if (env && *env)
696 ret = xstrdup(env);
697 else if (repo_config_get_string(r, config_key, &ret) < 0)
698 die(_("could not retrieve `%s`"), config_display_key);
699
700 if (!ret) {
701 ret = xstrdup("master");
702 if (!quiet)
703 advise(_(default_branch_name_advice), ret);
704 }
705
706 full_ref = xstrfmt("refs/heads/%s", ret);
707 if (check_refname_format(full_ref, 0))
708 die(_("invalid branch name: %s = %s"), config_display_key, ret);
709 free(full_ref);
710
711 return ret;
712 }
713
714 const char *git_default_branch_name(int quiet)
715 {
716 static char *ret;
717
718 if (!ret)
719 ret = repo_default_branch_name(the_repository, quiet);
720
721 return ret;
722 }
723
724 /*
725 * *string and *len will only be substituted, and *string returned (for
726 * later free()ing) if the string passed in is a magic short-hand form
727 * to name a branch.
728 */
729 static char *substitute_branch_name(struct repository *r,
730 const char **string, int *len,
731 int nonfatal_dangling_mark)
732 {
733 struct strbuf buf = STRBUF_INIT;
734 struct interpret_branch_name_options options = {
735 .nonfatal_dangling_mark = nonfatal_dangling_mark
736 };
737 int ret = repo_interpret_branch_name(r, *string, *len, &buf, &options);
738
739 if (ret == *len) {
740 size_t size;
741 *string = strbuf_detach(&buf, &size);
742 *len = size;
743 return (char *)*string;
744 }
745
746 return NULL;
747 }
748
749 int repo_dwim_ref(struct repository *r, const char *str, int len,
750 struct object_id *oid, char **ref, int nonfatal_dangling_mark)
751 {
752 char *last_branch = substitute_branch_name(r, &str, &len,
753 nonfatal_dangling_mark);
754 int refs_found = expand_ref(r, str, len, oid, ref);
755 free(last_branch);
756 return refs_found;
757 }
758
759 int expand_ref(struct repository *repo, const char *str, int len,
760 struct object_id *oid, char **ref)
761 {
762 const char **p, *r;
763 int refs_found = 0;
764 struct strbuf fullref = STRBUF_INIT;
765
766 *ref = NULL;
767 for (p = ref_rev_parse_rules; *p; p++) {
768 struct object_id oid_from_ref;
769 struct object_id *this_result;
770 int flag;
771 struct ref_store *refs = get_main_ref_store(repo);
772
773 this_result = refs_found ? &oid_from_ref : oid;
774 strbuf_reset(&fullref);
775 strbuf_addf(&fullref, *p, len, str);
776 r = refs_resolve_ref_unsafe(refs, fullref.buf,
777 RESOLVE_REF_READING,
778 this_result, &flag);
779 if (r) {
780 if (!refs_found++)
781 *ref = xstrdup(r);
782 if (!warn_ambiguous_refs)
783 break;
784 } else if ((flag & REF_ISSYMREF) && strcmp(fullref.buf, "HEAD")) {
785 warning(_("ignoring dangling symref %s"), fullref.buf);
786 } else if ((flag & REF_ISBROKEN) && strchr(fullref.buf, '/')) {
787 warning(_("ignoring broken ref %s"), fullref.buf);
788 }
789 }
790 strbuf_release(&fullref);
791 return refs_found;
792 }
793
794 int repo_dwim_log(struct repository *r, const char *str, int len,
795 struct object_id *oid, char **log)
796 {
797 struct ref_store *refs = get_main_ref_store(r);
798 char *last_branch = substitute_branch_name(r, &str, &len, 0);
799 const char **p;
800 int logs_found = 0;
801 struct strbuf path = STRBUF_INIT;
802
803 *log = NULL;
804 for (p = ref_rev_parse_rules; *p; p++) {
805 struct object_id hash;
806 const char *ref, *it;
807
808 strbuf_reset(&path);
809 strbuf_addf(&path, *p, len, str);
810 ref = refs_resolve_ref_unsafe(refs, path.buf,
811 RESOLVE_REF_READING,
812 oid ? &hash : NULL, NULL);
813 if (!ref)
814 continue;
815 if (refs_reflog_exists(refs, path.buf))
816 it = path.buf;
817 else if (strcmp(ref, path.buf) &&
818 refs_reflog_exists(refs, ref))
819 it = ref;
820 else
821 continue;
822 if (!logs_found++) {
823 *log = xstrdup(it);
824 if (oid)
825 oidcpy(oid, &hash);
826 }
827 if (!warn_ambiguous_refs)
828 break;
829 }
830 strbuf_release(&path);
831 free(last_branch);
832 return logs_found;
833 }
834
835 int dwim_log(const char *str, int len, struct object_id *oid, char **log)
836 {
837 return repo_dwim_log(the_repository, str, len, oid, log);
838 }
839
840 int is_per_worktree_ref(const char *refname)
841 {
842 return starts_with(refname, "refs/worktree/") ||
843 starts_with(refname, "refs/bisect/") ||
844 starts_with(refname, "refs/rewritten/");
845 }
846
847 static int is_pseudoref_syntax(const char *refname)
848 {
849 const char *c;
850
851 for (c = refname; *c; c++) {
852 if (!isupper(*c) && *c != '-' && *c != '_')
853 return 0;
854 }
855
856 /*
857 * HEAD is not a pseudoref, but it certainly uses the
858 * pseudoref syntax.
859 */
860 return 1;
861 }
862
863 static int is_current_worktree_ref(const char *ref) {
864 return is_pseudoref_syntax(ref) || is_per_worktree_ref(ref);
865 }
866
867 enum ref_worktree_type parse_worktree_ref(const char *maybe_worktree_ref,
868 const char **worktree_name, int *worktree_name_length,
869 const char **bare_refname)
870 {
871 const char *name_dummy;
872 int name_length_dummy;
873 const char *ref_dummy;
874
875 if (!worktree_name)
876 worktree_name = &name_dummy;
877 if (!worktree_name_length)
878 worktree_name_length = &name_length_dummy;
879 if (!bare_refname)
880 bare_refname = &ref_dummy;
881
882 if (skip_prefix(maybe_worktree_ref, "worktrees/", bare_refname)) {
883 const char *slash = strchr(*bare_refname, '/');
884
885 *worktree_name = *bare_refname;
886 if (!slash) {
887 *worktree_name_length = strlen(*worktree_name);
888
889 /* This is an error condition, and the caller tell because the bare_refname is "" */
890 *bare_refname = *worktree_name + *worktree_name_length;
891 return REF_WORKTREE_OTHER;
892 }
893
894 *worktree_name_length = slash - *bare_refname;
895 *bare_refname = slash + 1;
896
897 if (is_current_worktree_ref(*bare_refname))
898 return REF_WORKTREE_OTHER;
899 }
900
901 *worktree_name = NULL;
902 *worktree_name_length = 0;
903
904 if (skip_prefix(maybe_worktree_ref, "main-worktree/", bare_refname)
905 && is_current_worktree_ref(*bare_refname))
906 return REF_WORKTREE_MAIN;
907
908 *bare_refname = maybe_worktree_ref;
909 if (is_current_worktree_ref(maybe_worktree_ref))
910 return REF_WORKTREE_CURRENT;
911
912 return REF_WORKTREE_SHARED;
913 }
914
915 long get_files_ref_lock_timeout_ms(void)
916 {
917 static int configured = 0;
918
919 /* The default timeout is 100 ms: */
920 static int timeout_ms = 100;
921
922 if (!configured) {
923 git_config_get_int("core.filesreflocktimeout", &timeout_ms);
924 configured = 1;
925 }
926
927 return timeout_ms;
928 }
929
930 int refs_delete_ref(struct ref_store *refs, const char *msg,
931 const char *refname,
932 const struct object_id *old_oid,
933 unsigned int flags)
934 {
935 struct ref_transaction *transaction;
936 struct strbuf err = STRBUF_INIT;
937
938 transaction = ref_store_transaction_begin(refs, &err);
939 if (!transaction ||
940 ref_transaction_delete(transaction, refname, old_oid,
941 flags, msg, &err) ||
942 ref_transaction_commit(transaction, &err)) {
943 error("%s", err.buf);
944 ref_transaction_free(transaction);
945 strbuf_release(&err);
946 return 1;
947 }
948 ref_transaction_free(transaction);
949 strbuf_release(&err);
950 return 0;
951 }
952
953 int delete_ref(const char *msg, const char *refname,
954 const struct object_id *old_oid, unsigned int flags)
955 {
956 return refs_delete_ref(get_main_ref_store(the_repository), msg, refname,
957 old_oid, flags);
958 }
959
960 static void copy_reflog_msg(struct strbuf *sb, const char *msg)
961 {
962 char c;
963 int wasspace = 1;
964
965 while ((c = *msg++)) {
966 if (wasspace && isspace(c))
967 continue;
968 wasspace = isspace(c);
969 if (wasspace)
970 c = ' ';
971 strbuf_addch(sb, c);
972 }
973 strbuf_rtrim(sb);
974 }
975
976 static char *normalize_reflog_message(const char *msg)
977 {
978 struct strbuf sb = STRBUF_INIT;
979
980 if (msg && *msg)
981 copy_reflog_msg(&sb, msg);
982 return strbuf_detach(&sb, NULL);
983 }
984
985 int should_autocreate_reflog(const char *refname)
986 {
987 switch (log_all_ref_updates) {
988 case LOG_REFS_ALWAYS:
989 return 1;
990 case LOG_REFS_NORMAL:
991 return starts_with(refname, "refs/heads/") ||
992 starts_with(refname, "refs/remotes/") ||
993 starts_with(refname, "refs/notes/") ||
994 !strcmp(refname, "HEAD");
995 default:
996 return 0;
997 }
998 }
999
1000 int is_branch(const char *refname)
1001 {
1002 return !strcmp(refname, "HEAD") || starts_with(refname, "refs/heads/");
1003 }
1004
1005 struct read_ref_at_cb {
1006 const char *refname;
1007 timestamp_t at_time;
1008 int cnt;
1009 int reccnt;
1010 struct object_id *oid;
1011 int found_it;
1012
1013 struct object_id ooid;
1014 struct object_id noid;
1015 int tz;
1016 timestamp_t date;
1017 char **msg;
1018 timestamp_t *cutoff_time;
1019 int *cutoff_tz;
1020 int *cutoff_cnt;
1021 };
1022
1023 static void set_read_ref_cutoffs(struct read_ref_at_cb *cb,
1024 timestamp_t timestamp, int tz, const char *message)
1025 {
1026 if (cb->msg)
1027 *cb->msg = xstrdup(message);
1028 if (cb->cutoff_time)
1029 *cb->cutoff_time = timestamp;
1030 if (cb->cutoff_tz)
1031 *cb->cutoff_tz = tz;
1032 if (cb->cutoff_cnt)
1033 *cb->cutoff_cnt = cb->reccnt;
1034 }
1035
1036 static int read_ref_at_ent(struct object_id *ooid, struct object_id *noid,
1037 const char *email UNUSED,
1038 timestamp_t timestamp, int tz,
1039 const char *message, void *cb_data)
1040 {
1041 struct read_ref_at_cb *cb = cb_data;
1042
1043 cb->tz = tz;
1044 cb->date = timestamp;
1045
1046 if (timestamp <= cb->at_time || cb->cnt == 0) {
1047 set_read_ref_cutoffs(cb, timestamp, tz, message);
1048 /*
1049 * we have not yet updated cb->[n|o]oid so they still
1050 * hold the values for the previous record.
1051 */
1052 if (!is_null_oid(&cb->ooid)) {
1053 oidcpy(cb->oid, noid);
1054 if (!oideq(&cb->ooid, noid))
1055 warning(_("log for ref %s has gap after %s"),
1056 cb->refname, show_date(cb->date, cb->tz, DATE_MODE(RFC2822)));
1057 }
1058 else if (cb->date == cb->at_time)
1059 oidcpy(cb->oid, noid);
1060 else if (!oideq(noid, cb->oid))
1061 warning(_("log for ref %s unexpectedly ended on %s"),
1062 cb->refname, show_date(cb->date, cb->tz,
1063 DATE_MODE(RFC2822)));
1064 cb->reccnt++;
1065 oidcpy(&cb->ooid, ooid);
1066 oidcpy(&cb->noid, noid);
1067 cb->found_it = 1;
1068 return 1;
1069 }
1070 cb->reccnt++;
1071 oidcpy(&cb->ooid, ooid);
1072 oidcpy(&cb->noid, noid);
1073 if (cb->cnt > 0)
1074 cb->cnt--;
1075 return 0;
1076 }
1077
1078 static int read_ref_at_ent_oldest(struct object_id *ooid, struct object_id *noid,
1079 const char *email UNUSED,
1080 timestamp_t timestamp, int tz,
1081 const char *message, void *cb_data)
1082 {
1083 struct read_ref_at_cb *cb = cb_data;
1084
1085 set_read_ref_cutoffs(cb, timestamp, tz, message);
1086 oidcpy(cb->oid, ooid);
1087 if (cb->at_time && is_null_oid(cb->oid))
1088 oidcpy(cb->oid, noid);
1089 /* We just want the first entry */
1090 return 1;
1091 }
1092
1093 int read_ref_at(struct ref_store *refs, const char *refname,
1094 unsigned int flags, timestamp_t at_time, int cnt,
1095 struct object_id *oid, char **msg,
1096 timestamp_t *cutoff_time, int *cutoff_tz, int *cutoff_cnt)
1097 {
1098 struct read_ref_at_cb cb;
1099
1100 memset(&cb, 0, sizeof(cb));
1101 cb.refname = refname;
1102 cb.at_time = at_time;
1103 cb.cnt = cnt;
1104 cb.msg = msg;
1105 cb.cutoff_time = cutoff_time;
1106 cb.cutoff_tz = cutoff_tz;
1107 cb.cutoff_cnt = cutoff_cnt;
1108 cb.oid = oid;
1109
1110 refs_for_each_reflog_ent_reverse(refs, refname, read_ref_at_ent, &cb);
1111
1112 if (!cb.reccnt) {
1113 if (cnt == 0) {
1114 /*
1115 * The caller asked for ref@{0}, and we had no entries.
1116 * It's a bit subtle, but in practice all callers have
1117 * prepped the "oid" field with the current value of
1118 * the ref, which is the most reasonable fallback.
1119 *
1120 * We'll put dummy values into the out-parameters (so
1121 * they're not just uninitialized garbage), and the
1122 * caller can take our return value as a hint that
1123 * we did not find any such reflog.
1124 */
1125 set_read_ref_cutoffs(&cb, 0, 0, "empty reflog");
1126 return 1;
1127 }
1128 if (flags & GET_OID_QUIETLY)
1129 exit(128);
1130 else
1131 die(_("log for %s is empty"), refname);
1132 }
1133 if (cb.found_it)
1134 return 0;
1135
1136 refs_for_each_reflog_ent(refs, refname, read_ref_at_ent_oldest, &cb);
1137
1138 return 1;
1139 }
1140
1141 struct ref_transaction *ref_store_transaction_begin(struct ref_store *refs,
1142 struct strbuf *err)
1143 {
1144 struct ref_transaction *tr;
1145 assert(err);
1146
1147 CALLOC_ARRAY(tr, 1);
1148 tr->ref_store = refs;
1149 return tr;
1150 }
1151
1152 struct ref_transaction *ref_transaction_begin(struct strbuf *err)
1153 {
1154 return ref_store_transaction_begin(get_main_ref_store(the_repository), err);
1155 }
1156
1157 void ref_transaction_free(struct ref_transaction *transaction)
1158 {
1159 size_t i;
1160
1161 if (!transaction)
1162 return;
1163
1164 switch (transaction->state) {
1165 case REF_TRANSACTION_OPEN:
1166 case REF_TRANSACTION_CLOSED:
1167 /* OK */
1168 break;
1169 case REF_TRANSACTION_PREPARED:
1170 BUG("free called on a prepared reference transaction");
1171 break;
1172 default:
1173 BUG("unexpected reference transaction state");
1174 break;
1175 }
1176
1177 for (i = 0; i < transaction->nr; i++) {
1178 free(transaction->updates[i]->msg);
1179 free(transaction->updates[i]);
1180 }
1181 free(transaction->updates);
1182 free(transaction);
1183 }
1184
1185 struct ref_update *ref_transaction_add_update(
1186 struct ref_transaction *transaction,
1187 const char *refname, unsigned int flags,
1188 const struct object_id *new_oid,
1189 const struct object_id *old_oid,
1190 const char *msg)
1191 {
1192 struct ref_update *update;
1193
1194 if (transaction->state != REF_TRANSACTION_OPEN)
1195 BUG("update called for transaction that is not open");
1196
1197 FLEX_ALLOC_STR(update, refname, refname);
1198 ALLOC_GROW(transaction->updates, transaction->nr + 1, transaction->alloc);
1199 transaction->updates[transaction->nr++] = update;
1200
1201 update->flags = flags;
1202
1203 if (flags & REF_HAVE_NEW)
1204 oidcpy(&update->new_oid, new_oid);
1205 if (flags & REF_HAVE_OLD)
1206 oidcpy(&update->old_oid, old_oid);
1207 update->msg = normalize_reflog_message(msg);
1208 return update;
1209 }
1210
1211 int ref_transaction_update(struct ref_transaction *transaction,
1212 const char *refname,
1213 const struct object_id *new_oid,
1214 const struct object_id *old_oid,
1215 unsigned int flags, const char *msg,
1216 struct strbuf *err)
1217 {
1218 assert(err);
1219
1220 if (!(flags & REF_SKIP_REFNAME_VERIFICATION) &&
1221 ((new_oid && !is_null_oid(new_oid)) ?
1222 check_refname_format(refname, REFNAME_ALLOW_ONELEVEL) :
1223 !refname_is_safe(refname))) {
1224 strbuf_addf(err, _("refusing to update ref with bad name '%s'"),
1225 refname);
1226 return -1;
1227 }
1228
1229 if (flags & ~REF_TRANSACTION_UPDATE_ALLOWED_FLAGS)
1230 BUG("illegal flags 0x%x passed to ref_transaction_update()", flags);
1231
1232 /*
1233 * Clear flags outside the allowed set; this should be a noop because
1234 * of the BUG() check above, but it works around a -Wnonnull warning
1235 * with some versions of "gcc -O3".
1236 */
1237 flags &= REF_TRANSACTION_UPDATE_ALLOWED_FLAGS;
1238
1239 flags |= (new_oid ? REF_HAVE_NEW : 0) | (old_oid ? REF_HAVE_OLD : 0);
1240
1241 ref_transaction_add_update(transaction, refname, flags,
1242 new_oid, old_oid, msg);
1243 return 0;
1244 }
1245
1246 int ref_transaction_create(struct ref_transaction *transaction,
1247 const char *refname,
1248 const struct object_id *new_oid,
1249 unsigned int flags, const char *msg,
1250 struct strbuf *err)
1251 {
1252 if (!new_oid || is_null_oid(new_oid)) {
1253 strbuf_addf(err, "'%s' has a null OID", refname);
1254 return 1;
1255 }
1256 return ref_transaction_update(transaction, refname, new_oid,
1257 null_oid(), flags, msg, err);
1258 }
1259
1260 int ref_transaction_delete(struct ref_transaction *transaction,
1261 const char *refname,
1262 const struct object_id *old_oid,
1263 unsigned int flags, const char *msg,
1264 struct strbuf *err)
1265 {
1266 if (old_oid && is_null_oid(old_oid))
1267 BUG("delete called with old_oid set to zeros");
1268 return ref_transaction_update(transaction, refname,
1269 null_oid(), old_oid,
1270 flags, msg, err);
1271 }
1272
1273 int ref_transaction_verify(struct ref_transaction *transaction,
1274 const char *refname,
1275 const struct object_id *old_oid,
1276 unsigned int flags,
1277 struct strbuf *err)
1278 {
1279 if (!old_oid)
1280 BUG("verify called with old_oid set to NULL");
1281 return ref_transaction_update(transaction, refname,
1282 NULL, old_oid,
1283 flags, NULL, err);
1284 }
1285
1286 int refs_update_ref(struct ref_store *refs, const char *msg,
1287 const char *refname, const struct object_id *new_oid,
1288 const struct object_id *old_oid, unsigned int flags,
1289 enum action_on_err onerr)
1290 {
1291 struct ref_transaction *t = NULL;
1292 struct strbuf err = STRBUF_INIT;
1293 int ret = 0;
1294
1295 t = ref_store_transaction_begin(refs, &err);
1296 if (!t ||
1297 ref_transaction_update(t, refname, new_oid, old_oid, flags, msg,
1298 &err) ||
1299 ref_transaction_commit(t, &err)) {
1300 ret = 1;
1301 ref_transaction_free(t);
1302 }
1303 if (ret) {
1304 const char *str = _("update_ref failed for ref '%s': %s");
1305
1306 switch (onerr) {
1307 case UPDATE_REFS_MSG_ON_ERR:
1308 error(str, refname, err.buf);
1309 break;
1310 case UPDATE_REFS_DIE_ON_ERR:
1311 die(str, refname, err.buf);
1312 break;
1313 case UPDATE_REFS_QUIET_ON_ERR:
1314 break;
1315 }
1316 strbuf_release(&err);
1317 return 1;
1318 }
1319 strbuf_release(&err);
1320 if (t)
1321 ref_transaction_free(t);
1322 return 0;
1323 }
1324
1325 int update_ref(const char *msg, const char *refname,
1326 const struct object_id *new_oid,
1327 const struct object_id *old_oid,
1328 unsigned int flags, enum action_on_err onerr)
1329 {
1330 return refs_update_ref(get_main_ref_store(the_repository), msg, refname, new_oid,
1331 old_oid, flags, onerr);
1332 }
1333
1334 /*
1335 * Check that the string refname matches a rule of the form
1336 * "{prefix}%.*s{suffix}". So "foo/bar/baz" would match the rule
1337 * "foo/%.*s/baz", and return the string "bar".
1338 */
1339 static const char *match_parse_rule(const char *refname, const char *rule,
1340 size_t *len)
1341 {
1342 /*
1343 * Check that rule matches refname up to the first percent in the rule.
1344 * We can bail immediately if not, but otherwise we leave "rule" at the
1345 * %-placeholder, and "refname" at the start of the potential matched
1346 * name.
1347 */
1348 while (*rule != '%') {
1349 if (!*rule)
1350 BUG("rev-parse rule did not have percent");
1351 if (*refname++ != *rule++)
1352 return NULL;
1353 }
1354
1355 /*
1356 * Check that our "%" is the expected placeholder. This assumes there
1357 * are no other percents (placeholder or quoted) in the string, but
1358 * that is sufficient for our rev-parse rules.
1359 */
1360 if (!skip_prefix(rule, "%.*s", &rule))
1361 return NULL;
1362
1363 /*
1364 * And now check that our suffix (if any) matches.
1365 */
1366 if (!strip_suffix(refname, rule, len))
1367 return NULL;
1368
1369 return refname; /* len set by strip_suffix() */
1370 }
1371
1372 char *refs_shorten_unambiguous_ref(struct ref_store *refs,
1373 const char *refname, int strict)
1374 {
1375 int i;
1376 struct strbuf resolved_buf = STRBUF_INIT;
1377
1378 /* skip first rule, it will always match */
1379 for (i = NUM_REV_PARSE_RULES - 1; i > 0 ; --i) {
1380 int j;
1381 int rules_to_fail = i;
1382 const char *short_name;
1383 size_t short_name_len;
1384
1385 short_name = match_parse_rule(refname, ref_rev_parse_rules[i],
1386 &short_name_len);
1387 if (!short_name)
1388 continue;
1389
1390 /*
1391 * in strict mode, all (except the matched one) rules
1392 * must fail to resolve to a valid non-ambiguous ref
1393 */
1394 if (strict)
1395 rules_to_fail = NUM_REV_PARSE_RULES;
1396
1397 /*
1398 * check if the short name resolves to a valid ref,
1399 * but use only rules prior to the matched one
1400 */
1401 for (j = 0; j < rules_to_fail; j++) {
1402 const char *rule = ref_rev_parse_rules[j];
1403
1404 /* skip matched rule */
1405 if (i == j)
1406 continue;
1407
1408 /*
1409 * the short name is ambiguous, if it resolves
1410 * (with this previous rule) to a valid ref
1411 * read_ref() returns 0 on success
1412 */
1413 strbuf_reset(&resolved_buf);
1414 strbuf_addf(&resolved_buf, rule,
1415 cast_size_t_to_int(short_name_len),
1416 short_name);
1417 if (refs_ref_exists(refs, resolved_buf.buf))
1418 break;
1419 }
1420
1421 /*
1422 * short name is non-ambiguous if all previous rules
1423 * haven't resolved to a valid ref
1424 */
1425 if (j == rules_to_fail) {
1426 strbuf_release(&resolved_buf);
1427 return xmemdupz(short_name, short_name_len);
1428 }
1429 }
1430
1431 strbuf_release(&resolved_buf);
1432 return xstrdup(refname);
1433 }
1434
1435 char *shorten_unambiguous_ref(const char *refname, int strict)
1436 {
1437 return refs_shorten_unambiguous_ref(get_main_ref_store(the_repository),
1438 refname, strict);
1439 }
1440
1441 int parse_hide_refs_config(const char *var, const char *value, const char *section,
1442 struct strvec *hide_refs)
1443 {
1444 const char *key;
1445 if (!strcmp("transfer.hiderefs", var) ||
1446 (!parse_config_key(var, section, NULL, NULL, &key) &&
1447 !strcmp(key, "hiderefs"))) {
1448 char *ref;
1449 int len;
1450
1451 if (!value)
1452 return config_error_nonbool(var);
1453
1454 /* drop const to remove trailing '/' characters */
1455 ref = (char *)strvec_push(hide_refs, value);
1456 len = strlen(ref);
1457 while (len && ref[len - 1] == '/')
1458 ref[--len] = '\0';
1459 }
1460 return 0;
1461 }
1462
1463 int ref_is_hidden(const char *refname, const char *refname_full,
1464 const struct strvec *hide_refs)
1465 {
1466 int i;
1467
1468 for (i = hide_refs->nr - 1; i >= 0; i--) {
1469 const char *match = hide_refs->v[i];
1470 const char *subject;
1471 int neg = 0;
1472 const char *p;
1473
1474 if (*match == '!') {
1475 neg = 1;
1476 match++;
1477 }
1478
1479 if (*match == '^') {
1480 subject = refname_full;
1481 match++;
1482 } else {
1483 subject = refname;
1484 }
1485
1486 /* refname can be NULL when namespaces are used. */
1487 if (subject &&
1488 skip_prefix(subject, match, &p) &&
1489 (!*p || *p == '/'))
1490 return !neg;
1491 }
1492 return 0;
1493 }
1494
1495 const char **hidden_refs_to_excludes(const struct strvec *hide_refs)
1496 {
1497 const char **pattern;
1498 for (pattern = hide_refs->v; *pattern; pattern++) {
1499 /*
1500 * We can't feed any excludes from hidden refs config
1501 * sections, since later rules may override previous
1502 * ones. For example, with rules "refs/foo" and
1503 * "!refs/foo/bar", we should show "refs/foo/bar" (and
1504 * everything underneath it), but the earlier exclusion
1505 * would cause us to skip all of "refs/foo". We
1506 * likewise don't implement the namespace stripping
1507 * required for '^' rules.
1508 *
1509 * Both are possible to do, but complicated, so avoid
1510 * populating the jump list at all if we see either of
1511 * these patterns.
1512 */
1513 if (**pattern == '!' || **pattern == '^')
1514 return NULL;
1515 }
1516 return hide_refs->v;
1517 }
1518
1519 const char *find_descendant_ref(const char *dirname,
1520 const struct string_list *extras,
1521 const struct string_list *skip)
1522 {
1523 int pos;
1524
1525 if (!extras)
1526 return NULL;
1527
1528 /*
1529 * Look at the place where dirname would be inserted into
1530 * extras. If there is an entry at that position that starts
1531 * with dirname (remember, dirname includes the trailing
1532 * slash) and is not in skip, then we have a conflict.
1533 */
1534 for (pos = string_list_find_insert_index(extras, dirname, 0);
1535 pos < extras->nr; pos++) {
1536 const char *extra_refname = extras->items[pos].string;
1537
1538 if (!starts_with(extra_refname, dirname))
1539 break;
1540
1541 if (!skip || !string_list_has_string(skip, extra_refname))
1542 return extra_refname;
1543 }
1544 return NULL;
1545 }
1546
1547 int refs_head_ref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
1548 {
1549 struct object_id oid;
1550 int flag;
1551
1552 if (refs_resolve_ref_unsafe(refs, "HEAD", RESOLVE_REF_READING,
1553 &oid, &flag))
1554 return fn("HEAD", &oid, flag, cb_data);
1555
1556 return 0;
1557 }
1558
1559 int head_ref(each_ref_fn fn, void *cb_data)
1560 {
1561 return refs_head_ref(get_main_ref_store(the_repository), fn, cb_data);
1562 }
1563
1564 struct ref_iterator *refs_ref_iterator_begin(
1565 struct ref_store *refs,
1566 const char *prefix,
1567 const char **exclude_patterns,
1568 int trim,
1569 enum do_for_each_ref_flags flags)
1570 {
1571 struct ref_iterator *iter;
1572
1573 if (!(flags & DO_FOR_EACH_INCLUDE_BROKEN)) {
1574 static int ref_paranoia = -1;
1575
1576 if (ref_paranoia < 0)
1577 ref_paranoia = git_env_bool("GIT_REF_PARANOIA", 1);
1578 if (ref_paranoia) {
1579 flags |= DO_FOR_EACH_INCLUDE_BROKEN;
1580 flags |= DO_FOR_EACH_OMIT_DANGLING_SYMREFS;
1581 }
1582 }
1583
1584 iter = refs->be->iterator_begin(refs, prefix, exclude_patterns, flags);
1585 /*
1586 * `iterator_begin()` already takes care of prefix, but we
1587 * might need to do some trimming:
1588 */
1589 if (trim)
1590 iter = prefix_ref_iterator_begin(iter, "", trim);
1591
1592 return iter;
1593 }
1594
1595 /*
1596 * Call fn for each reference in the specified submodule for which the
1597 * refname begins with prefix. If trim is non-zero, then trim that
1598 * many characters off the beginning of each refname before passing
1599 * the refname to fn. flags can be DO_FOR_EACH_INCLUDE_BROKEN to
1600 * include broken references in the iteration. If fn ever returns a
1601 * non-zero value, stop the iteration and return that value;
1602 * otherwise, return 0.
1603 */
1604 static int do_for_each_repo_ref(struct repository *r, const char *prefix,
1605 each_repo_ref_fn fn, int trim, int flags,
1606 void *cb_data)
1607 {
1608 struct ref_iterator *iter;
1609 struct ref_store *refs = get_main_ref_store(r);
1610
1611 if (!refs)
1612 return 0;
1613
1614 iter = refs_ref_iterator_begin(refs, prefix, NULL, trim, flags);
1615
1616 return do_for_each_repo_ref_iterator(r, iter, fn, cb_data);
1617 }
1618
1619 struct do_for_each_ref_help {
1620 each_ref_fn *fn;
1621 void *cb_data;
1622 };
1623
1624 static int do_for_each_ref_helper(struct repository *r UNUSED,
1625 const char *refname,
1626 const struct object_id *oid,
1627 int flags,
1628 void *cb_data)
1629 {
1630 struct do_for_each_ref_help *hp = cb_data;
1631
1632 return hp->fn(refname, oid, flags, hp->cb_data);
1633 }
1634
1635 static int do_for_each_ref(struct ref_store *refs, const char *prefix,
1636 const char **exclude_patterns,
1637 each_ref_fn fn, int trim,
1638 enum do_for_each_ref_flags flags, void *cb_data)
1639 {
1640 struct ref_iterator *iter;
1641 struct do_for_each_ref_help hp = { fn, cb_data };
1642
1643 if (!refs)
1644 return 0;
1645
1646 iter = refs_ref_iterator_begin(refs, prefix, exclude_patterns, trim,
1647 flags);
1648
1649 return do_for_each_repo_ref_iterator(the_repository, iter,
1650 do_for_each_ref_helper, &hp);
1651 }
1652
1653 int refs_for_each_ref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
1654 {
1655 return do_for_each_ref(refs, "", NULL, fn, 0, 0, cb_data);
1656 }
1657
1658 int for_each_ref(each_ref_fn fn, void *cb_data)
1659 {
1660 return refs_for_each_ref(get_main_ref_store(the_repository), fn, cb_data);
1661 }
1662
1663 int refs_for_each_ref_in(struct ref_store *refs, const char *prefix,
1664 each_ref_fn fn, void *cb_data)
1665 {
1666 return do_for_each_ref(refs, prefix, NULL, fn, strlen(prefix), 0, cb_data);
1667 }
1668
1669 int for_each_ref_in(const char *prefix, each_ref_fn fn, void *cb_data)
1670 {
1671 return refs_for_each_ref_in(get_main_ref_store(the_repository), prefix, fn, cb_data);
1672 }
1673
1674 int for_each_fullref_in(const char *prefix, each_ref_fn fn, void *cb_data)
1675 {
1676 return do_for_each_ref(get_main_ref_store(the_repository),
1677 prefix, NULL, fn, 0, 0, cb_data);
1678 }
1679
1680 int refs_for_each_fullref_in(struct ref_store *refs, const char *prefix,
1681 const char **exclude_patterns,
1682 each_ref_fn fn, void *cb_data)
1683 {
1684 return do_for_each_ref(refs, prefix, exclude_patterns, fn, 0, 0, cb_data);
1685 }
1686
1687 int for_each_replace_ref(struct repository *r, each_repo_ref_fn fn, void *cb_data)
1688 {
1689 const char *git_replace_ref_base = ref_namespace[NAMESPACE_REPLACE].ref;
1690 return do_for_each_repo_ref(r, git_replace_ref_base, fn,
1691 strlen(git_replace_ref_base),
1692 DO_FOR_EACH_INCLUDE_BROKEN, cb_data);
1693 }
1694
1695 int for_each_namespaced_ref(const char **exclude_patterns,
1696 each_ref_fn fn, void *cb_data)
1697 {
1698 struct strbuf buf = STRBUF_INIT;
1699 int ret;
1700 strbuf_addf(&buf, "%srefs/", get_git_namespace());
1701 ret = do_for_each_ref(get_main_ref_store(the_repository),
1702 buf.buf, exclude_patterns, fn, 0, 0, cb_data);
1703 strbuf_release(&buf);
1704 return ret;
1705 }
1706
1707 int refs_for_each_rawref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
1708 {
1709 return do_for_each_ref(refs, "", NULL, fn, 0,
1710 DO_FOR_EACH_INCLUDE_BROKEN, cb_data);
1711 }
1712
1713 int for_each_rawref(each_ref_fn fn, void *cb_data)
1714 {
1715 return refs_for_each_rawref(get_main_ref_store(the_repository), fn, cb_data);
1716 }
1717
1718 static int qsort_strcmp(const void *va, const void *vb)
1719 {
1720 const char *a = *(const char **)va;
1721 const char *b = *(const char **)vb;
1722
1723 return strcmp(a, b);
1724 }
1725
1726 static void find_longest_prefixes_1(struct string_list *out,
1727 struct strbuf *prefix,
1728 const char **patterns, size_t nr)
1729 {
1730 size_t i;
1731
1732 for (i = 0; i < nr; i++) {
1733 char c = patterns[i][prefix->len];
1734 if (!c || is_glob_special(c)) {
1735 string_list_append(out, prefix->buf);
1736 return;
1737 }
1738 }
1739
1740 i = 0;
1741 while (i < nr) {
1742 size_t end;
1743
1744 /*
1745 * Set "end" to the index of the element _after_ the last one
1746 * in our group.
1747 */
1748 for (end = i + 1; end < nr; end++) {
1749 if (patterns[i][prefix->len] != patterns[end][prefix->len])
1750 break;
1751 }
1752
1753 strbuf_addch(prefix, patterns[i][prefix->len]);
1754 find_longest_prefixes_1(out, prefix, patterns + i, end - i);
1755 strbuf_setlen(prefix, prefix->len - 1);
1756
1757 i = end;
1758 }
1759 }
1760
1761 static void find_longest_prefixes(struct string_list *out,
1762 const char **patterns)
1763 {
1764 struct strvec sorted = STRVEC_INIT;
1765 struct strbuf prefix = STRBUF_INIT;
1766
1767 strvec_pushv(&sorted, patterns);
1768 QSORT(sorted.v, sorted.nr, qsort_strcmp);
1769
1770 find_longest_prefixes_1(out, &prefix, sorted.v, sorted.nr);
1771
1772 strvec_clear(&sorted);
1773 strbuf_release(&prefix);
1774 }
1775
1776 int refs_for_each_fullref_in_prefixes(struct ref_store *ref_store,
1777 const char *namespace,
1778 const char **patterns,
1779 const char **exclude_patterns,
1780 each_ref_fn fn, void *cb_data)
1781 {
1782 struct string_list prefixes = STRING_LIST_INIT_DUP;
1783 struct string_list_item *prefix;
1784 struct strbuf buf = STRBUF_INIT;
1785 int ret = 0, namespace_len;
1786
1787 find_longest_prefixes(&prefixes, patterns);
1788
1789 if (namespace)
1790 strbuf_addstr(&buf, namespace);
1791 namespace_len = buf.len;
1792
1793 for_each_string_list_item(prefix, &prefixes) {
1794 strbuf_addstr(&buf, prefix->string);
1795 ret = refs_for_each_fullref_in(ref_store, buf.buf,
1796 exclude_patterns, fn, cb_data);
1797 if (ret)
1798 break;
1799 strbuf_setlen(&buf, namespace_len);
1800 }
1801
1802 string_list_clear(&prefixes, 0);
1803 strbuf_release(&buf);
1804 return ret;
1805 }
1806
1807 static int refs_read_special_head(struct ref_store *ref_store,
1808 const char *refname, struct object_id *oid,
1809 struct strbuf *referent, unsigned int *type,
1810 int *failure_errno)
1811 {
1812 struct strbuf full_path = STRBUF_INIT;
1813 struct strbuf content = STRBUF_INIT;
1814 int result = -1;
1815 strbuf_addf(&full_path, "%s/%s", ref_store->gitdir, refname);
1816
1817 if (strbuf_read_file(&content, full_path.buf, 0) < 0) {
1818 *failure_errno = errno;
1819 goto done;
1820 }
1821
1822 result = parse_loose_ref_contents(content.buf, oid, referent, type,
1823 failure_errno);
1824
1825 done:
1826 strbuf_release(&full_path);
1827 strbuf_release(&content);
1828 return result;
1829 }
1830
1831 static int is_special_ref(const char *refname)
1832 {
1833 /*
1834 * Special references are refs that have different semantics compared
1835 * to "normal" refs. These refs can thus not be stored in the ref
1836 * backend, but must always be accessed via the filesystem. The
1837 * following refs are special:
1838 *
1839 * - FETCH_HEAD may contain multiple object IDs, and each one of them
1840 * carries additional metadata like where it came from.
1841 *
1842 * - MERGE_HEAD may contain multiple object IDs when merging multiple
1843 * heads.
1844 *
1845 * Reading, writing or deleting references must consistently go either
1846 * through the filesystem (special refs) or through the reference
1847 * backend (normal ones).
1848 */
1849 static const char * const special_refs[] = {
1850 "FETCH_HEAD",
1851 "MERGE_HEAD",
1852 };
1853 size_t i;
1854
1855 for (i = 0; i < ARRAY_SIZE(special_refs); i++)
1856 if (!strcmp(refname, special_refs[i]))
1857 return 1;
1858
1859 return 0;
1860 }
1861
1862 int refs_read_raw_ref(struct ref_store *ref_store, const char *refname,
1863 struct object_id *oid, struct strbuf *referent,
1864 unsigned int *type, int *failure_errno)
1865 {
1866 assert(failure_errno);
1867 if (is_special_ref(refname))
1868 return refs_read_special_head(ref_store, refname, oid, referent,
1869 type, failure_errno);
1870
1871 return ref_store->be->read_raw_ref(ref_store, refname, oid, referent,
1872 type, failure_errno);
1873 }
1874
1875 int refs_read_symbolic_ref(struct ref_store *ref_store, const char *refname,
1876 struct strbuf *referent)
1877 {
1878 return ref_store->be->read_symbolic_ref(ref_store, refname, referent);
1879 }
1880
1881 const char *refs_resolve_ref_unsafe(struct ref_store *refs,
1882 const char *refname,
1883 int resolve_flags,
1884 struct object_id *oid,
1885 int *flags)
1886 {
1887 static struct strbuf sb_refname = STRBUF_INIT;
1888 struct object_id unused_oid;
1889 int unused_flags;
1890 int symref_count;
1891
1892 if (!oid)
1893 oid = &unused_oid;
1894 if (!flags)
1895 flags = &unused_flags;
1896
1897 *flags = 0;
1898
1899 if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL)) {
1900 if (!(resolve_flags & RESOLVE_REF_ALLOW_BAD_NAME) ||
1901 !refname_is_safe(refname))
1902 return NULL;
1903
1904 /*
1905 * repo_dwim_ref() uses REF_ISBROKEN to distinguish between
1906 * missing refs and refs that were present but invalid,
1907 * to complain about the latter to stderr.
1908 *
1909 * We don't know whether the ref exists, so don't set
1910 * REF_ISBROKEN yet.
1911 */
1912 *flags |= REF_BAD_NAME;
1913 }
1914
1915 for (symref_count = 0; symref_count < SYMREF_MAXDEPTH; symref_count++) {
1916 unsigned int read_flags = 0;
1917 int failure_errno;
1918
1919 if (refs_read_raw_ref(refs, refname, oid, &sb_refname,
1920 &read_flags, &failure_errno)) {
1921 *flags |= read_flags;
1922
1923 /* In reading mode, refs must eventually resolve */
1924 if (resolve_flags & RESOLVE_REF_READING)
1925 return NULL;
1926
1927 /*
1928 * Otherwise a missing ref is OK. But the files backend
1929 * may show errors besides ENOENT if there are
1930 * similarly-named refs.
1931 */
1932 if (failure_errno != ENOENT &&
1933 failure_errno != EISDIR &&
1934 failure_errno != ENOTDIR)
1935 return NULL;
1936
1937 oidclr(oid);
1938 if (*flags & REF_BAD_NAME)
1939 *flags |= REF_ISBROKEN;
1940 return refname;
1941 }
1942
1943 *flags |= read_flags;
1944
1945 if (!(read_flags & REF_ISSYMREF)) {
1946 if (*flags & REF_BAD_NAME) {
1947 oidclr(oid);
1948 *flags |= REF_ISBROKEN;
1949 }
1950 return refname;
1951 }
1952
1953 refname = sb_refname.buf;
1954 if (resolve_flags & RESOLVE_REF_NO_RECURSE) {
1955 oidclr(oid);
1956 return refname;
1957 }
1958 if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL)) {
1959 if (!(resolve_flags & RESOLVE_REF_ALLOW_BAD_NAME) ||
1960 !refname_is_safe(refname))
1961 return NULL;
1962
1963 *flags |= REF_ISBROKEN | REF_BAD_NAME;
1964 }
1965 }
1966
1967 return NULL;
1968 }
1969
1970 /* backend functions */
1971 int refs_init_db(struct ref_store *refs, int flags, struct strbuf *err)
1972 {
1973 return refs->be->init_db(refs, flags, err);
1974 }
1975
1976 const char *resolve_ref_unsafe(const char *refname, int resolve_flags,
1977 struct object_id *oid, int *flags)
1978 {
1979 return refs_resolve_ref_unsafe(get_main_ref_store(the_repository), refname,
1980 resolve_flags, oid, flags);
1981 }
1982
1983 int resolve_gitlink_ref(const char *submodule, const char *refname,
1984 struct object_id *oid)
1985 {
1986 struct ref_store *refs;
1987 int flags;
1988
1989 refs = get_submodule_ref_store(submodule);
1990
1991 if (!refs)
1992 return -1;
1993
1994 if (!refs_resolve_ref_unsafe(refs, refname, 0, oid, &flags) ||
1995 is_null_oid(oid))
1996 return -1;
1997 return 0;
1998 }
1999
2000 struct ref_store_hash_entry
2001 {
2002 struct hashmap_entry ent;
2003
2004 struct ref_store *refs;
2005
2006 /* NUL-terminated identifier of the ref store: */
2007 char name[FLEX_ARRAY];
2008 };
2009
2010 static int ref_store_hash_cmp(const void *cmp_data UNUSED,
2011 const struct hashmap_entry *eptr,
2012 const struct hashmap_entry *entry_or_key,
2013 const void *keydata)
2014 {
2015 const struct ref_store_hash_entry *e1, *e2;
2016 const char *name;
2017
2018 e1 = container_of(eptr, const struct ref_store_hash_entry, ent);
2019 e2 = container_of(entry_or_key, const struct ref_store_hash_entry, ent);
2020 name = keydata ? keydata : e2->name;
2021
2022 return strcmp(e1->name, name);
2023 }
2024
2025 static struct ref_store_hash_entry *alloc_ref_store_hash_entry(
2026 const char *name, struct ref_store *refs)
2027 {
2028 struct ref_store_hash_entry *entry;
2029
2030 FLEX_ALLOC_STR(entry, name, name);
2031 hashmap_entry_init(&entry->ent, strhash(name));
2032 entry->refs = refs;
2033 return entry;
2034 }
2035
2036 /* A hashmap of ref_stores, stored by submodule name: */
2037 static struct hashmap submodule_ref_stores;
2038
2039 /* A hashmap of ref_stores, stored by worktree id: */
2040 static struct hashmap worktree_ref_stores;
2041
2042 /*
2043 * Look up a ref store by name. If that ref_store hasn't been
2044 * registered yet, return NULL.
2045 */
2046 static struct ref_store *lookup_ref_store_map(struct hashmap *map,
2047 const char *name)
2048 {
2049 struct ref_store_hash_entry *entry;
2050 unsigned int hash;
2051
2052 if (!map->tablesize)
2053 /* It's initialized on demand in register_ref_store(). */
2054 return NULL;
2055
2056 hash = strhash(name);
2057 entry = hashmap_get_entry_from_hash(map, hash, name,
2058 struct ref_store_hash_entry, ent);
2059 return entry ? entry->refs : NULL;
2060 }
2061
2062 /*
2063 * Create, record, and return a ref_store instance for the specified
2064 * gitdir.
2065 */
2066 static struct ref_store *ref_store_init(struct repository *repo,
2067 const char *gitdir,
2068 unsigned int flags)
2069 {
2070 const struct ref_storage_be *be;
2071 struct ref_store *refs;
2072
2073 be = find_ref_storage_backend(repo->ref_storage_format);
2074 if (!be)
2075 BUG("reference backend is unknown");
2076
2077 refs = be->init(repo, gitdir, flags);
2078 return refs;
2079 }
2080
2081 struct ref_store *get_main_ref_store(struct repository *r)
2082 {
2083 if (r->refs_private)
2084 return r->refs_private;
2085
2086 if (!r->gitdir)
2087 BUG("attempting to get main_ref_store outside of repository");
2088
2089 r->refs_private = ref_store_init(r, r->gitdir, REF_STORE_ALL_CAPS);
2090 r->refs_private = maybe_debug_wrap_ref_store(r->gitdir, r->refs_private);
2091 return r->refs_private;
2092 }
2093
2094 /*
2095 * Associate a ref store with a name. It is a fatal error to call this
2096 * function twice for the same name.
2097 */
2098 static void register_ref_store_map(struct hashmap *map,
2099 const char *type,
2100 struct ref_store *refs,
2101 const char *name)
2102 {
2103 struct ref_store_hash_entry *entry;
2104
2105 if (!map->tablesize)
2106 hashmap_init(map, ref_store_hash_cmp, NULL, 0);
2107
2108 entry = alloc_ref_store_hash_entry(name, refs);
2109 if (hashmap_put(map, &entry->ent))
2110 BUG("%s ref_store '%s' initialized twice", type, name);
2111 }
2112
2113 struct ref_store *get_submodule_ref_store(const char *submodule)
2114 {
2115 struct strbuf submodule_sb = STRBUF_INIT;
2116 struct ref_store *refs;
2117 char *to_free = NULL;
2118 size_t len;
2119 struct repository *subrepo;
2120
2121 if (!submodule)
2122 return NULL;
2123
2124 len = strlen(submodule);
2125 while (len && is_dir_sep(submodule[len - 1]))
2126 len--;
2127 if (!len)
2128 return NULL;
2129
2130 if (submodule[len])
2131 /* We need to strip off one or more trailing slashes */
2132 submodule = to_free = xmemdupz(submodule, len);
2133
2134 refs = lookup_ref_store_map(&submodule_ref_stores, submodule);
2135 if (refs)
2136 goto done;
2137
2138 strbuf_addstr(&submodule_sb, submodule);
2139 if (!is_nonbare_repository_dir(&submodule_sb))
2140 goto done;
2141
2142 if (submodule_to_gitdir(&submodule_sb, submodule))
2143 goto done;
2144
2145 subrepo = xmalloc(sizeof(*subrepo));
2146 /*
2147 * NEEDSWORK: Make get_submodule_ref_store() work with arbitrary
2148 * superprojects other than the_repository. This probably should be
2149 * done by making it take a struct repository * parameter instead of a
2150 * submodule path.
2151 */
2152 if (repo_submodule_init(subrepo, the_repository, submodule,
2153 null_oid())) {
2154 free(subrepo);
2155 goto done;
2156 }
2157 refs = ref_store_init(subrepo, submodule_sb.buf,
2158 REF_STORE_READ | REF_STORE_ODB);
2159 register_ref_store_map(&submodule_ref_stores, "submodule",
2160 refs, submodule);
2161
2162 done:
2163 strbuf_release(&submodule_sb);
2164 free(to_free);
2165
2166 return refs;
2167 }
2168
2169 struct ref_store *get_worktree_ref_store(const struct worktree *wt)
2170 {
2171 struct ref_store *refs;
2172 const char *id;
2173
2174 if (wt->is_current)
2175 return get_main_ref_store(the_repository);
2176
2177 id = wt->id ? wt->id : "/";
2178 refs = lookup_ref_store_map(&worktree_ref_stores, id);
2179 if (refs)
2180 return refs;
2181
2182 if (wt->id)
2183 refs = ref_store_init(the_repository,
2184 git_common_path("worktrees/%s", wt->id),
2185 REF_STORE_ALL_CAPS);
2186 else
2187 refs = ref_store_init(the_repository,
2188 get_git_common_dir(),
2189 REF_STORE_ALL_CAPS);
2190
2191 if (refs)
2192 register_ref_store_map(&worktree_ref_stores, "worktree",
2193 refs, id);
2194 return refs;
2195 }
2196
2197 void base_ref_store_init(struct ref_store *refs, struct repository *repo,
2198 const char *path, const struct ref_storage_be *be)
2199 {
2200 refs->be = be;
2201 refs->repo = repo;
2202 refs->gitdir = xstrdup(path);
2203 }
2204
2205 /* backend functions */
2206 int refs_pack_refs(struct ref_store *refs, struct pack_refs_opts *opts)
2207 {
2208 return refs->be->pack_refs(refs, opts);
2209 }
2210
2211 int peel_iterated_oid(const struct object_id *base, struct object_id *peeled)
2212 {
2213 if (current_ref_iter &&
2214 (current_ref_iter->oid == base ||
2215 oideq(current_ref_iter->oid, base)))
2216 return ref_iterator_peel(current_ref_iter, peeled);
2217
2218 return peel_object(base, peeled) ? -1 : 0;
2219 }
2220
2221 int refs_create_symref(struct ref_store *refs,
2222 const char *ref_target,
2223 const char *refs_heads_master,
2224 const char *logmsg)
2225 {
2226 char *msg;
2227 int retval;
2228
2229 msg = normalize_reflog_message(logmsg);
2230 retval = refs->be->create_symref(refs, ref_target, refs_heads_master,
2231 msg);
2232 free(msg);
2233 return retval;
2234 }
2235
2236 int create_symref(const char *ref_target, const char *refs_heads_master,
2237 const char *logmsg)
2238 {
2239 return refs_create_symref(get_main_ref_store(the_repository), ref_target,
2240 refs_heads_master, logmsg);
2241 }
2242
2243 int ref_update_reject_duplicates(struct string_list *refnames,
2244 struct strbuf *err)
2245 {
2246 size_t i, n = refnames->nr;
2247
2248 assert(err);
2249
2250 for (i = 1; i < n; i++) {
2251 int cmp = strcmp(refnames->items[i - 1].string,
2252 refnames->items[i].string);
2253
2254 if (!cmp) {
2255 strbuf_addf(err,
2256 _("multiple updates for ref '%s' not allowed"),
2257 refnames->items[i].string);
2258 return 1;
2259 } else if (cmp > 0) {
2260 BUG("ref_update_reject_duplicates() received unsorted list");
2261 }
2262 }
2263 return 0;
2264 }
2265
2266 static int run_transaction_hook(struct ref_transaction *transaction,
2267 const char *state)
2268 {
2269 struct child_process proc = CHILD_PROCESS_INIT;
2270 struct strbuf buf = STRBUF_INIT;
2271 const char *hook;
2272 int ret = 0, i;
2273
2274 hook = find_hook("reference-transaction");
2275 if (!hook)
2276 return ret;
2277
2278 strvec_pushl(&proc.args, hook, state, NULL);
2279 proc.in = -1;
2280 proc.stdout_to_stderr = 1;
2281 proc.trace2_hook_name = "reference-transaction";
2282
2283 ret = start_command(&proc);
2284 if (ret)
2285 return ret;
2286
2287 sigchain_push(SIGPIPE, SIG_IGN);
2288
2289 for (i = 0; i < transaction->nr; i++) {
2290 struct ref_update *update = transaction->updates[i];
2291
2292 strbuf_reset(&buf);
2293 strbuf_addf(&buf, "%s %s %s\n",
2294 oid_to_hex(&update->old_oid),
2295 oid_to_hex(&update->new_oid),
2296 update->refname);
2297
2298 if (write_in_full(proc.in, buf.buf, buf.len) < 0) {
2299 if (errno != EPIPE) {
2300 /* Don't leak errno outside this API */
2301 errno = 0;
2302 ret = -1;
2303 }
2304 break;
2305 }
2306 }
2307
2308 close(proc.in);
2309 sigchain_pop(SIGPIPE);
2310 strbuf_release(&buf);
2311
2312 ret |= finish_command(&proc);
2313 return ret;
2314 }
2315
2316 int ref_transaction_prepare(struct ref_transaction *transaction,
2317 struct strbuf *err)
2318 {
2319 struct ref_store *refs = transaction->ref_store;
2320 int ret;
2321
2322 switch (transaction->state) {
2323 case REF_TRANSACTION_OPEN:
2324 /* Good. */
2325 break;
2326 case REF_TRANSACTION_PREPARED:
2327 BUG("prepare called twice on reference transaction");
2328 break;
2329 case REF_TRANSACTION_CLOSED:
2330 BUG("prepare called on a closed reference transaction");
2331 break;
2332 default:
2333 BUG("unexpected reference transaction state");
2334 break;
2335 }
2336
2337 if (refs->repo->objects->odb->disable_ref_updates) {
2338 strbuf_addstr(err,
2339 _("ref updates forbidden inside quarantine environment"));
2340 return -1;
2341 }
2342
2343 ret = refs->be->transaction_prepare(refs, transaction, err);
2344 if (ret)
2345 return ret;
2346
2347 ret = run_transaction_hook(transaction, "prepared");
2348 if (ret) {
2349 ref_transaction_abort(transaction, err);
2350 die(_("ref updates aborted by hook"));
2351 }
2352
2353 return 0;
2354 }
2355
2356 int ref_transaction_abort(struct ref_transaction *transaction,
2357 struct strbuf *err)
2358 {
2359 struct ref_store *refs = transaction->ref_store;
2360 int ret = 0;
2361
2362 switch (transaction->state) {
2363 case REF_TRANSACTION_OPEN:
2364 /* No need to abort explicitly. */
2365 break;
2366 case REF_TRANSACTION_PREPARED:
2367 ret = refs->be->transaction_abort(refs, transaction, err);
2368 break;
2369 case REF_TRANSACTION_CLOSED:
2370 BUG("abort called on a closed reference transaction");
2371 break;
2372 default:
2373 BUG("unexpected reference transaction state");
2374 break;
2375 }
2376
2377 run_transaction_hook(transaction, "aborted");
2378
2379 ref_transaction_free(transaction);
2380 return ret;
2381 }
2382
2383 int ref_transaction_commit(struct ref_transaction *transaction,
2384 struct strbuf *err)
2385 {
2386 struct ref_store *refs = transaction->ref_store;
2387 int ret;
2388
2389 switch (transaction->state) {
2390 case REF_TRANSACTION_OPEN:
2391 /* Need to prepare first. */
2392 ret = ref_transaction_prepare(transaction, err);
2393 if (ret)
2394 return ret;
2395 break;
2396 case REF_TRANSACTION_PREPARED:
2397 /* Fall through to finish. */
2398 break;
2399 case REF_TRANSACTION_CLOSED:
2400 BUG("commit called on a closed reference transaction");
2401 break;
2402 default:
2403 BUG("unexpected reference transaction state");
2404 break;
2405 }
2406
2407 ret = refs->be->transaction_finish(refs, transaction, err);
2408 if (!ret)
2409 run_transaction_hook(transaction, "committed");
2410 return ret;
2411 }
2412
2413 int refs_verify_refname_available(struct ref_store *refs,
2414 const char *refname,
2415 const struct string_list *extras,
2416 const struct string_list *skip,
2417 struct strbuf *err)
2418 {
2419 const char *slash;
2420 const char *extra_refname;
2421 struct strbuf dirname = STRBUF_INIT;
2422 struct strbuf referent = STRBUF_INIT;
2423 struct object_id oid;
2424 unsigned int type;
2425 struct ref_iterator *iter;
2426 int ok;
2427 int ret = -1;
2428
2429 /*
2430 * For the sake of comments in this function, suppose that
2431 * refname is "refs/foo/bar".
2432 */
2433
2434 assert(err);
2435
2436 strbuf_grow(&dirname, strlen(refname) + 1);
2437 for (slash = strchr(refname, '/'); slash; slash = strchr(slash + 1, '/')) {
2438 /*
2439 * Just saying "Is a directory" when we e.g. can't
2440 * lock some multi-level ref isn't very informative,
2441 * the user won't be told *what* is a directory, so
2442 * let's not use strerror() below.
2443 */
2444 int ignore_errno;
2445 /* Expand dirname to the new prefix, not including the trailing slash: */
2446 strbuf_add(&dirname, refname + dirname.len, slash - refname - dirname.len);
2447
2448 /*
2449 * We are still at a leading dir of the refname (e.g.,
2450 * "refs/foo"; if there is a reference with that name,
2451 * it is a conflict, *unless* it is in skip.
2452 */
2453 if (skip && string_list_has_string(skip, dirname.buf))
2454 continue;
2455
2456 if (!refs_read_raw_ref(refs, dirname.buf, &oid, &referent,
2457 &type, &ignore_errno)) {
2458 strbuf_addf(err, _("'%s' exists; cannot create '%s'"),
2459 dirname.buf, refname);
2460 goto cleanup;
2461 }
2462
2463 if (extras && string_list_has_string(extras, dirname.buf)) {
2464 strbuf_addf(err, _("cannot process '%s' and '%s' at the same time"),
2465 refname, dirname.buf);
2466 goto cleanup;
2467 }
2468 }
2469
2470 /*
2471 * We are at the leaf of our refname (e.g., "refs/foo/bar").
2472 * There is no point in searching for a reference with that
2473 * name, because a refname isn't considered to conflict with
2474 * itself. But we still need to check for references whose
2475 * names are in the "refs/foo/bar/" namespace, because they
2476 * *do* conflict.
2477 */
2478 strbuf_addstr(&dirname, refname + dirname.len);
2479 strbuf_addch(&dirname, '/');
2480
2481 iter = refs_ref_iterator_begin(refs, dirname.buf, NULL, 0,
2482 DO_FOR_EACH_INCLUDE_BROKEN);
2483 while ((ok = ref_iterator_advance(iter)) == ITER_OK) {
2484 if (skip &&
2485 string_list_has_string(skip, iter->refname))
2486 continue;
2487
2488 strbuf_addf(err, _("'%s' exists; cannot create '%s'"),
2489 iter->refname, refname);
2490 ref_iterator_abort(iter);
2491 goto cleanup;
2492 }
2493
2494 if (ok != ITER_DONE)
2495 BUG("error while iterating over references");
2496
2497 extra_refname = find_descendant_ref(dirname.buf, extras, skip);
2498 if (extra_refname)
2499 strbuf_addf(err, _("cannot process '%s' and '%s' at the same time"),
2500 refname, extra_refname);
2501 else
2502 ret = 0;
2503
2504 cleanup:
2505 strbuf_release(&referent);
2506 strbuf_release(&dirname);
2507 return ret;
2508 }
2509
2510 struct do_for_each_reflog_help {
2511 each_reflog_fn *fn;
2512 void *cb_data;
2513 };
2514
2515 static int do_for_each_reflog_helper(struct repository *r UNUSED,
2516 const char *refname,
2517 const struct object_id *oid UNUSED,
2518 int flags,
2519 void *cb_data)
2520 {
2521 struct do_for_each_reflog_help *hp = cb_data;
2522 return hp->fn(refname, hp->cb_data);
2523 }
2524
2525 int refs_for_each_reflog(struct ref_store *refs, each_reflog_fn fn, void *cb_data)
2526 {
2527 struct ref_iterator *iter;
2528 struct do_for_each_reflog_help hp = { fn, cb_data };
2529
2530 iter = refs->be->reflog_iterator_begin(refs);
2531
2532 return do_for_each_repo_ref_iterator(the_repository, iter,
2533 do_for_each_reflog_helper, &hp);
2534 }
2535
2536 int for_each_reflog(each_reflog_fn fn, void *cb_data)
2537 {
2538 return refs_for_each_reflog(get_main_ref_store(the_repository), fn, cb_data);
2539 }
2540
2541 int refs_for_each_reflog_ent_reverse(struct ref_store *refs,
2542 const char *refname,
2543 each_reflog_ent_fn fn,
2544 void *cb_data)
2545 {
2546 return refs->be->for_each_reflog_ent_reverse(refs, refname,
2547 fn, cb_data);
2548 }
2549
2550 int for_each_reflog_ent_reverse(const char *refname, each_reflog_ent_fn fn,
2551 void *cb_data)
2552 {
2553 return refs_for_each_reflog_ent_reverse(get_main_ref_store(the_repository),
2554 refname, fn, cb_data);
2555 }
2556
2557 int refs_for_each_reflog_ent(struct ref_store *refs, const char *refname,
2558 each_reflog_ent_fn fn, void *cb_data)
2559 {
2560 return refs->be->for_each_reflog_ent(refs, refname, fn, cb_data);
2561 }
2562
2563 int for_each_reflog_ent(const char *refname, each_reflog_ent_fn fn,
2564 void *cb_data)
2565 {
2566 return refs_for_each_reflog_ent(get_main_ref_store(the_repository), refname,
2567 fn, cb_data);
2568 }
2569
2570 int refs_reflog_exists(struct ref_store *refs, const char *refname)
2571 {
2572 return refs->be->reflog_exists(refs, refname);
2573 }
2574
2575 int reflog_exists(const char *refname)
2576 {
2577 return refs_reflog_exists(get_main_ref_store(the_repository), refname);
2578 }
2579
2580 int refs_create_reflog(struct ref_store *refs, const char *refname,
2581 struct strbuf *err)
2582 {
2583 return refs->be->create_reflog(refs, refname, err);
2584 }
2585
2586 int safe_create_reflog(const char *refname, struct strbuf *err)
2587 {
2588 return refs_create_reflog(get_main_ref_store(the_repository), refname,
2589 err);
2590 }
2591
2592 int refs_delete_reflog(struct ref_store *refs, const char *refname)
2593 {
2594 return refs->be->delete_reflog(refs, refname);
2595 }
2596
2597 int delete_reflog(const char *refname)
2598 {
2599 return refs_delete_reflog(get_main_ref_store(the_repository), refname);
2600 }
2601
2602 int refs_reflog_expire(struct ref_store *refs,
2603 const char *refname,
2604 unsigned int flags,
2605 reflog_expiry_prepare_fn prepare_fn,
2606 reflog_expiry_should_prune_fn should_prune_fn,
2607 reflog_expiry_cleanup_fn cleanup_fn,
2608 void *policy_cb_data)
2609 {
2610 return refs->be->reflog_expire(refs, refname, flags,
2611 prepare_fn, should_prune_fn,
2612 cleanup_fn, policy_cb_data);
2613 }
2614
2615 int reflog_expire(const char *refname,
2616 unsigned int flags,
2617 reflog_expiry_prepare_fn prepare_fn,
2618 reflog_expiry_should_prune_fn should_prune_fn,
2619 reflog_expiry_cleanup_fn cleanup_fn,
2620 void *policy_cb_data)
2621 {
2622 return refs_reflog_expire(get_main_ref_store(the_repository),
2623 refname, flags,
2624 prepare_fn, should_prune_fn,
2625 cleanup_fn, policy_cb_data);
2626 }
2627
2628 int initial_ref_transaction_commit(struct ref_transaction *transaction,
2629 struct strbuf *err)
2630 {
2631 struct ref_store *refs = transaction->ref_store;
2632
2633 return refs->be->initial_transaction_commit(refs, transaction, err);
2634 }
2635
2636 void ref_transaction_for_each_queued_update(struct ref_transaction *transaction,
2637 ref_transaction_for_each_queued_update_fn cb,
2638 void *cb_data)
2639 {
2640 int i;
2641
2642 for (i = 0; i < transaction->nr; i++) {
2643 struct ref_update *update = transaction->updates[i];
2644
2645 cb(update->refname,
2646 (update->flags & REF_HAVE_OLD) ? &update->old_oid : NULL,
2647 (update->flags & REF_HAVE_NEW) ? &update->new_oid : NULL,
2648 cb_data);
2649 }
2650 }
2651
2652 int refs_delete_refs(struct ref_store *refs, const char *logmsg,
2653 struct string_list *refnames, unsigned int flags)
2654 {
2655 struct ref_transaction *transaction;
2656 struct strbuf err = STRBUF_INIT;
2657 struct string_list_item *item;
2658 int ret = 0, failures = 0;
2659 char *msg;
2660
2661 if (!refnames->nr)
2662 return 0;
2663
2664 msg = normalize_reflog_message(logmsg);
2665
2666 /*
2667 * Since we don't check the references' old_oids, the
2668 * individual updates can't fail, so we can pack all of the
2669 * updates into a single transaction.
2670 */
2671 transaction = ref_store_transaction_begin(refs, &err);
2672 if (!transaction) {
2673 ret = error("%s", err.buf);
2674 goto out;
2675 }
2676
2677 for_each_string_list_item(item, refnames) {
2678 ret = ref_transaction_delete(transaction, item->string,
2679 NULL, flags, msg, &err);
2680 if (ret) {
2681 warning(_("could not delete reference %s: %s"),
2682 item->string, err.buf);
2683 strbuf_reset(&err);
2684 failures = 1;
2685 }
2686 }
2687
2688 ret = ref_transaction_commit(transaction, &err);
2689 if (ret) {
2690 if (refnames->nr == 1)
2691 error(_("could not delete reference %s: %s"),
2692 refnames->items[0].string, err.buf);
2693 else
2694 error(_("could not delete references: %s"), err.buf);
2695 }
2696
2697 out:
2698 if (!ret && failures)
2699 ret = -1;
2700 ref_transaction_free(transaction);
2701 strbuf_release(&err);
2702 free(msg);
2703 return ret;
2704 }
2705
2706 int delete_refs(const char *msg, struct string_list *refnames,
2707 unsigned int flags)
2708 {
2709 return refs_delete_refs(get_main_ref_store(the_repository), msg, refnames, flags);
2710 }
2711
2712 int refs_rename_ref(struct ref_store *refs, const char *oldref,
2713 const char *newref, const char *logmsg)
2714 {
2715 char *msg;
2716 int retval;
2717
2718 msg = normalize_reflog_message(logmsg);
2719 retval = refs->be->rename_ref(refs, oldref, newref, msg);
2720 free(msg);
2721 return retval;
2722 }
2723
2724 int rename_ref(const char *oldref, const char *newref, const char *logmsg)
2725 {
2726 return refs_rename_ref(get_main_ref_store(the_repository), oldref, newref, logmsg);
2727 }
2728
2729 int refs_copy_existing_ref(struct ref_store *refs, const char *oldref,
2730 const char *newref, const char *logmsg)
2731 {
2732 char *msg;
2733 int retval;
2734
2735 msg = normalize_reflog_message(logmsg);
2736 retval = refs->be->copy_ref(refs, oldref, newref, msg);
2737 free(msg);
2738 return retval;
2739 }
2740
2741 int copy_existing_ref(const char *oldref, const char *newref, const char *logmsg)
2742 {
2743 return refs_copy_existing_ref(get_main_ref_store(the_repository), oldref, newref, logmsg);
2744 }