]> git.ipfire.org Git - thirdparty/git.git/blob - refs.c
refs: accept symref values in `ref_transaction_update()`
[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 int is_pseudoref(struct ref_store *refs, const char *refname)
864 {
865 static const char *const irregular_pseudorefs[] = {
866 "AUTO_MERGE",
867 "BISECT_EXPECTED_REV",
868 "NOTES_MERGE_PARTIAL",
869 "NOTES_MERGE_REF",
870 "MERGE_AUTOSTASH",
871 };
872 struct object_id oid;
873 size_t i;
874
875 if (!is_pseudoref_syntax(refname))
876 return 0;
877
878 if (ends_with(refname, "_HEAD")) {
879 refs_resolve_ref_unsafe(refs, refname,
880 RESOLVE_REF_READING | RESOLVE_REF_NO_RECURSE,
881 &oid, NULL);
882 return !is_null_oid(&oid);
883 }
884
885 for (i = 0; i < ARRAY_SIZE(irregular_pseudorefs); i++)
886 if (!strcmp(refname, irregular_pseudorefs[i])) {
887 refs_resolve_ref_unsafe(refs, refname,
888 RESOLVE_REF_READING | RESOLVE_REF_NO_RECURSE,
889 &oid, NULL);
890 return !is_null_oid(&oid);
891 }
892
893 return 0;
894 }
895
896 int is_headref(struct ref_store *refs, const char *refname)
897 {
898 if (!strcmp(refname, "HEAD"))
899 return refs_ref_exists(refs, refname);
900
901 return 0;
902 }
903
904 static int is_current_worktree_ref(const char *ref) {
905 return is_pseudoref_syntax(ref) || is_per_worktree_ref(ref);
906 }
907
908 enum ref_worktree_type parse_worktree_ref(const char *maybe_worktree_ref,
909 const char **worktree_name, int *worktree_name_length,
910 const char **bare_refname)
911 {
912 const char *name_dummy;
913 int name_length_dummy;
914 const char *ref_dummy;
915
916 if (!worktree_name)
917 worktree_name = &name_dummy;
918 if (!worktree_name_length)
919 worktree_name_length = &name_length_dummy;
920 if (!bare_refname)
921 bare_refname = &ref_dummy;
922
923 if (skip_prefix(maybe_worktree_ref, "worktrees/", bare_refname)) {
924 const char *slash = strchr(*bare_refname, '/');
925
926 *worktree_name = *bare_refname;
927 if (!slash) {
928 *worktree_name_length = strlen(*worktree_name);
929
930 /* This is an error condition, and the caller tell because the bare_refname is "" */
931 *bare_refname = *worktree_name + *worktree_name_length;
932 return REF_WORKTREE_OTHER;
933 }
934
935 *worktree_name_length = slash - *bare_refname;
936 *bare_refname = slash + 1;
937
938 if (is_current_worktree_ref(*bare_refname))
939 return REF_WORKTREE_OTHER;
940 }
941
942 *worktree_name = NULL;
943 *worktree_name_length = 0;
944
945 if (skip_prefix(maybe_worktree_ref, "main-worktree/", bare_refname)
946 && is_current_worktree_ref(*bare_refname))
947 return REF_WORKTREE_MAIN;
948
949 *bare_refname = maybe_worktree_ref;
950 if (is_current_worktree_ref(maybe_worktree_ref))
951 return REF_WORKTREE_CURRENT;
952
953 return REF_WORKTREE_SHARED;
954 }
955
956 long get_files_ref_lock_timeout_ms(void)
957 {
958 static int configured = 0;
959
960 /* The default timeout is 100 ms: */
961 static int timeout_ms = 100;
962
963 if (!configured) {
964 git_config_get_int("core.filesreflocktimeout", &timeout_ms);
965 configured = 1;
966 }
967
968 return timeout_ms;
969 }
970
971 int refs_delete_ref(struct ref_store *refs, const char *msg,
972 const char *refname,
973 const struct object_id *old_oid,
974 unsigned int flags)
975 {
976 struct ref_transaction *transaction;
977 struct strbuf err = STRBUF_INIT;
978
979 transaction = ref_store_transaction_begin(refs, &err);
980 if (!transaction ||
981 ref_transaction_delete(transaction, refname, old_oid,
982 flags, msg, &err) ||
983 ref_transaction_commit(transaction, &err)) {
984 error("%s", err.buf);
985 ref_transaction_free(transaction);
986 strbuf_release(&err);
987 return 1;
988 }
989 ref_transaction_free(transaction);
990 strbuf_release(&err);
991 return 0;
992 }
993
994 int delete_ref(const char *msg, const char *refname,
995 const struct object_id *old_oid, unsigned int flags)
996 {
997 return refs_delete_ref(get_main_ref_store(the_repository), msg, refname,
998 old_oid, flags);
999 }
1000
1001 static void copy_reflog_msg(struct strbuf *sb, const char *msg)
1002 {
1003 char c;
1004 int wasspace = 1;
1005
1006 while ((c = *msg++)) {
1007 if (wasspace && isspace(c))
1008 continue;
1009 wasspace = isspace(c);
1010 if (wasspace)
1011 c = ' ';
1012 strbuf_addch(sb, c);
1013 }
1014 strbuf_rtrim(sb);
1015 }
1016
1017 static char *normalize_reflog_message(const char *msg)
1018 {
1019 struct strbuf sb = STRBUF_INIT;
1020
1021 if (msg && *msg)
1022 copy_reflog_msg(&sb, msg);
1023 return strbuf_detach(&sb, NULL);
1024 }
1025
1026 int should_autocreate_reflog(const char *refname)
1027 {
1028 switch (log_all_ref_updates) {
1029 case LOG_REFS_ALWAYS:
1030 return 1;
1031 case LOG_REFS_NORMAL:
1032 return starts_with(refname, "refs/heads/") ||
1033 starts_with(refname, "refs/remotes/") ||
1034 starts_with(refname, "refs/notes/") ||
1035 !strcmp(refname, "HEAD");
1036 default:
1037 return 0;
1038 }
1039 }
1040
1041 int is_branch(const char *refname)
1042 {
1043 return !strcmp(refname, "HEAD") || starts_with(refname, "refs/heads/");
1044 }
1045
1046 struct read_ref_at_cb {
1047 const char *refname;
1048 timestamp_t at_time;
1049 int cnt;
1050 int reccnt;
1051 struct object_id *oid;
1052 int found_it;
1053
1054 struct object_id ooid;
1055 struct object_id noid;
1056 int tz;
1057 timestamp_t date;
1058 char **msg;
1059 timestamp_t *cutoff_time;
1060 int *cutoff_tz;
1061 int *cutoff_cnt;
1062 };
1063
1064 static void set_read_ref_cutoffs(struct read_ref_at_cb *cb,
1065 timestamp_t timestamp, int tz, const char *message)
1066 {
1067 if (cb->msg)
1068 *cb->msg = xstrdup(message);
1069 if (cb->cutoff_time)
1070 *cb->cutoff_time = timestamp;
1071 if (cb->cutoff_tz)
1072 *cb->cutoff_tz = tz;
1073 if (cb->cutoff_cnt)
1074 *cb->cutoff_cnt = cb->reccnt;
1075 }
1076
1077 static int read_ref_at_ent(struct object_id *ooid, struct object_id *noid,
1078 const char *email UNUSED,
1079 timestamp_t timestamp, int tz,
1080 const char *message, void *cb_data)
1081 {
1082 struct read_ref_at_cb *cb = cb_data;
1083
1084 cb->tz = tz;
1085 cb->date = timestamp;
1086
1087 if (timestamp <= cb->at_time || cb->cnt == 0) {
1088 set_read_ref_cutoffs(cb, timestamp, tz, message);
1089 /*
1090 * we have not yet updated cb->[n|o]oid so they still
1091 * hold the values for the previous record.
1092 */
1093 if (!is_null_oid(&cb->ooid)) {
1094 oidcpy(cb->oid, noid);
1095 if (!oideq(&cb->ooid, noid))
1096 warning(_("log for ref %s has gap after %s"),
1097 cb->refname, show_date(cb->date, cb->tz, DATE_MODE(RFC2822)));
1098 }
1099 else if (cb->date == cb->at_time)
1100 oidcpy(cb->oid, noid);
1101 else if (!oideq(noid, cb->oid))
1102 warning(_("log for ref %s unexpectedly ended on %s"),
1103 cb->refname, show_date(cb->date, cb->tz,
1104 DATE_MODE(RFC2822)));
1105 cb->reccnt++;
1106 oidcpy(&cb->ooid, ooid);
1107 oidcpy(&cb->noid, noid);
1108 cb->found_it = 1;
1109 return 1;
1110 }
1111 cb->reccnt++;
1112 oidcpy(&cb->ooid, ooid);
1113 oidcpy(&cb->noid, noid);
1114 if (cb->cnt > 0)
1115 cb->cnt--;
1116 return 0;
1117 }
1118
1119 static int read_ref_at_ent_oldest(struct object_id *ooid, struct object_id *noid,
1120 const char *email UNUSED,
1121 timestamp_t timestamp, int tz,
1122 const char *message, void *cb_data)
1123 {
1124 struct read_ref_at_cb *cb = cb_data;
1125
1126 set_read_ref_cutoffs(cb, timestamp, tz, message);
1127 oidcpy(cb->oid, ooid);
1128 if (cb->at_time && is_null_oid(cb->oid))
1129 oidcpy(cb->oid, noid);
1130 /* We just want the first entry */
1131 return 1;
1132 }
1133
1134 int read_ref_at(struct ref_store *refs, const char *refname,
1135 unsigned int flags, timestamp_t at_time, int cnt,
1136 struct object_id *oid, char **msg,
1137 timestamp_t *cutoff_time, int *cutoff_tz, int *cutoff_cnt)
1138 {
1139 struct read_ref_at_cb cb;
1140
1141 memset(&cb, 0, sizeof(cb));
1142 cb.refname = refname;
1143 cb.at_time = at_time;
1144 cb.cnt = cnt;
1145 cb.msg = msg;
1146 cb.cutoff_time = cutoff_time;
1147 cb.cutoff_tz = cutoff_tz;
1148 cb.cutoff_cnt = cutoff_cnt;
1149 cb.oid = oid;
1150
1151 refs_for_each_reflog_ent_reverse(refs, refname, read_ref_at_ent, &cb);
1152
1153 if (!cb.reccnt) {
1154 if (cnt == 0) {
1155 /*
1156 * The caller asked for ref@{0}, and we had no entries.
1157 * It's a bit subtle, but in practice all callers have
1158 * prepped the "oid" field with the current value of
1159 * the ref, which is the most reasonable fallback.
1160 *
1161 * We'll put dummy values into the out-parameters (so
1162 * they're not just uninitialized garbage), and the
1163 * caller can take our return value as a hint that
1164 * we did not find any such reflog.
1165 */
1166 set_read_ref_cutoffs(&cb, 0, 0, "empty reflog");
1167 return 1;
1168 }
1169 if (flags & GET_OID_QUIETLY)
1170 exit(128);
1171 else
1172 die(_("log for %s is empty"), refname);
1173 }
1174 if (cb.found_it)
1175 return 0;
1176
1177 refs_for_each_reflog_ent(refs, refname, read_ref_at_ent_oldest, &cb);
1178
1179 return 1;
1180 }
1181
1182 struct ref_transaction *ref_store_transaction_begin(struct ref_store *refs,
1183 struct strbuf *err)
1184 {
1185 struct ref_transaction *tr;
1186 assert(err);
1187
1188 CALLOC_ARRAY(tr, 1);
1189 tr->ref_store = refs;
1190 return tr;
1191 }
1192
1193 struct ref_transaction *ref_transaction_begin(struct strbuf *err)
1194 {
1195 return ref_store_transaction_begin(get_main_ref_store(the_repository), err);
1196 }
1197
1198 void ref_transaction_free(struct ref_transaction *transaction)
1199 {
1200 size_t i;
1201
1202 if (!transaction)
1203 return;
1204
1205 switch (transaction->state) {
1206 case REF_TRANSACTION_OPEN:
1207 case REF_TRANSACTION_CLOSED:
1208 /* OK */
1209 break;
1210 case REF_TRANSACTION_PREPARED:
1211 BUG("free called on a prepared reference transaction");
1212 break;
1213 default:
1214 BUG("unexpected reference transaction state");
1215 break;
1216 }
1217
1218 for (i = 0; i < transaction->nr; i++) {
1219 free(transaction->updates[i]->msg);
1220 free(transaction->updates[i]);
1221 }
1222 free(transaction->updates);
1223 free(transaction);
1224 }
1225
1226 struct ref_update *ref_transaction_add_update(
1227 struct ref_transaction *transaction,
1228 const char *refname, unsigned int flags,
1229 const struct object_id *new_oid,
1230 const struct object_id *old_oid,
1231 const char *new_target, const char *old_target,
1232 const char *msg)
1233 {
1234 struct ref_update *update;
1235
1236 if (transaction->state != REF_TRANSACTION_OPEN)
1237 BUG("update called for transaction that is not open");
1238
1239 if (old_oid && old_target)
1240 BUG("only one of old_oid and old_target should be non NULL");
1241 if (new_oid && new_target)
1242 BUG("only one of new_oid and new_target should be non NULL");
1243
1244 FLEX_ALLOC_STR(update, refname, refname);
1245 ALLOC_GROW(transaction->updates, transaction->nr + 1, transaction->alloc);
1246 transaction->updates[transaction->nr++] = update;
1247
1248 update->flags = flags;
1249
1250 if (flags & REF_HAVE_NEW)
1251 oidcpy(&update->new_oid, new_oid);
1252 if (flags & REF_HAVE_OLD)
1253 oidcpy(&update->old_oid, old_oid);
1254 update->msg = normalize_reflog_message(msg);
1255 return update;
1256 }
1257
1258 int ref_transaction_update(struct ref_transaction *transaction,
1259 const char *refname,
1260 const struct object_id *new_oid,
1261 const struct object_id *old_oid,
1262 const char *new_target,
1263 const char *old_target,
1264 unsigned int flags, const char *msg,
1265 struct strbuf *err)
1266 {
1267 assert(err);
1268
1269 if (!(flags & REF_SKIP_REFNAME_VERIFICATION) &&
1270 ((new_oid && !is_null_oid(new_oid)) ?
1271 check_refname_format(refname, REFNAME_ALLOW_ONELEVEL) :
1272 !refname_is_safe(refname))) {
1273 strbuf_addf(err, _("refusing to update ref with bad name '%s'"),
1274 refname);
1275 return -1;
1276 }
1277
1278 if (flags & ~REF_TRANSACTION_UPDATE_ALLOWED_FLAGS)
1279 BUG("illegal flags 0x%x passed to ref_transaction_update()", flags);
1280
1281 /*
1282 * Clear flags outside the allowed set; this should be a noop because
1283 * of the BUG() check above, but it works around a -Wnonnull warning
1284 * with some versions of "gcc -O3".
1285 */
1286 flags &= REF_TRANSACTION_UPDATE_ALLOWED_FLAGS;
1287
1288 flags |= (new_oid ? REF_HAVE_NEW : 0) | (old_oid ? REF_HAVE_OLD : 0);
1289
1290 ref_transaction_add_update(transaction, refname, flags,
1291 new_oid, old_oid, new_target,
1292 old_target, msg);
1293 return 0;
1294 }
1295
1296 int ref_transaction_create(struct ref_transaction *transaction,
1297 const char *refname,
1298 const struct object_id *new_oid,
1299 unsigned int flags, const char *msg,
1300 struct strbuf *err)
1301 {
1302 if (!new_oid || is_null_oid(new_oid)) {
1303 strbuf_addf(err, "'%s' has a null OID", refname);
1304 return 1;
1305 }
1306 return ref_transaction_update(transaction, refname, new_oid,
1307 null_oid(), NULL, NULL, flags,
1308 msg, err);
1309 }
1310
1311 int ref_transaction_delete(struct ref_transaction *transaction,
1312 const char *refname,
1313 const struct object_id *old_oid,
1314 unsigned int flags, const char *msg,
1315 struct strbuf *err)
1316 {
1317 if (old_oid && is_null_oid(old_oid))
1318 BUG("delete called with old_oid set to zeros");
1319 return ref_transaction_update(transaction, refname,
1320 null_oid(), old_oid,
1321 NULL, NULL, flags,
1322 msg, err);
1323 }
1324
1325 int ref_transaction_verify(struct ref_transaction *transaction,
1326 const char *refname,
1327 const struct object_id *old_oid,
1328 unsigned int flags,
1329 struct strbuf *err)
1330 {
1331 if (!old_oid)
1332 BUG("verify called with old_oid set to NULL");
1333 return ref_transaction_update(transaction, refname,
1334 NULL, old_oid,
1335 NULL, NULL,
1336 flags, NULL, err);
1337 }
1338
1339 int refs_update_ref(struct ref_store *refs, const char *msg,
1340 const char *refname, const struct object_id *new_oid,
1341 const struct object_id *old_oid, unsigned int flags,
1342 enum action_on_err onerr)
1343 {
1344 struct ref_transaction *t = NULL;
1345 struct strbuf err = STRBUF_INIT;
1346 int ret = 0;
1347
1348 t = ref_store_transaction_begin(refs, &err);
1349 if (!t ||
1350 ref_transaction_update(t, refname, new_oid, old_oid, NULL, NULL,
1351 flags, msg, &err) ||
1352 ref_transaction_commit(t, &err)) {
1353 ret = 1;
1354 ref_transaction_free(t);
1355 }
1356 if (ret) {
1357 const char *str = _("update_ref failed for ref '%s': %s");
1358
1359 switch (onerr) {
1360 case UPDATE_REFS_MSG_ON_ERR:
1361 error(str, refname, err.buf);
1362 break;
1363 case UPDATE_REFS_DIE_ON_ERR:
1364 die(str, refname, err.buf);
1365 break;
1366 case UPDATE_REFS_QUIET_ON_ERR:
1367 break;
1368 }
1369 strbuf_release(&err);
1370 return 1;
1371 }
1372 strbuf_release(&err);
1373 if (t)
1374 ref_transaction_free(t);
1375 return 0;
1376 }
1377
1378 int update_ref(const char *msg, const char *refname,
1379 const struct object_id *new_oid,
1380 const struct object_id *old_oid,
1381 unsigned int flags, enum action_on_err onerr)
1382 {
1383 return refs_update_ref(get_main_ref_store(the_repository), msg, refname, new_oid,
1384 old_oid, flags, onerr);
1385 }
1386
1387 /*
1388 * Check that the string refname matches a rule of the form
1389 * "{prefix}%.*s{suffix}". So "foo/bar/baz" would match the rule
1390 * "foo/%.*s/baz", and return the string "bar".
1391 */
1392 static const char *match_parse_rule(const char *refname, const char *rule,
1393 size_t *len)
1394 {
1395 /*
1396 * Check that rule matches refname up to the first percent in the rule.
1397 * We can bail immediately if not, but otherwise we leave "rule" at the
1398 * %-placeholder, and "refname" at the start of the potential matched
1399 * name.
1400 */
1401 while (*rule != '%') {
1402 if (!*rule)
1403 BUG("rev-parse rule did not have percent");
1404 if (*refname++ != *rule++)
1405 return NULL;
1406 }
1407
1408 /*
1409 * Check that our "%" is the expected placeholder. This assumes there
1410 * are no other percents (placeholder or quoted) in the string, but
1411 * that is sufficient for our rev-parse rules.
1412 */
1413 if (!skip_prefix(rule, "%.*s", &rule))
1414 return NULL;
1415
1416 /*
1417 * And now check that our suffix (if any) matches.
1418 */
1419 if (!strip_suffix(refname, rule, len))
1420 return NULL;
1421
1422 return refname; /* len set by strip_suffix() */
1423 }
1424
1425 char *refs_shorten_unambiguous_ref(struct ref_store *refs,
1426 const char *refname, int strict)
1427 {
1428 int i;
1429 struct strbuf resolved_buf = STRBUF_INIT;
1430
1431 /* skip first rule, it will always match */
1432 for (i = NUM_REV_PARSE_RULES - 1; i > 0 ; --i) {
1433 int j;
1434 int rules_to_fail = i;
1435 const char *short_name;
1436 size_t short_name_len;
1437
1438 short_name = match_parse_rule(refname, ref_rev_parse_rules[i],
1439 &short_name_len);
1440 if (!short_name)
1441 continue;
1442
1443 /*
1444 * in strict mode, all (except the matched one) rules
1445 * must fail to resolve to a valid non-ambiguous ref
1446 */
1447 if (strict)
1448 rules_to_fail = NUM_REV_PARSE_RULES;
1449
1450 /*
1451 * check if the short name resolves to a valid ref,
1452 * but use only rules prior to the matched one
1453 */
1454 for (j = 0; j < rules_to_fail; j++) {
1455 const char *rule = ref_rev_parse_rules[j];
1456
1457 /* skip matched rule */
1458 if (i == j)
1459 continue;
1460
1461 /*
1462 * the short name is ambiguous, if it resolves
1463 * (with this previous rule) to a valid ref
1464 * read_ref() returns 0 on success
1465 */
1466 strbuf_reset(&resolved_buf);
1467 strbuf_addf(&resolved_buf, rule,
1468 cast_size_t_to_int(short_name_len),
1469 short_name);
1470 if (refs_ref_exists(refs, resolved_buf.buf))
1471 break;
1472 }
1473
1474 /*
1475 * short name is non-ambiguous if all previous rules
1476 * haven't resolved to a valid ref
1477 */
1478 if (j == rules_to_fail) {
1479 strbuf_release(&resolved_buf);
1480 return xmemdupz(short_name, short_name_len);
1481 }
1482 }
1483
1484 strbuf_release(&resolved_buf);
1485 return xstrdup(refname);
1486 }
1487
1488 char *shorten_unambiguous_ref(const char *refname, int strict)
1489 {
1490 return refs_shorten_unambiguous_ref(get_main_ref_store(the_repository),
1491 refname, strict);
1492 }
1493
1494 int parse_hide_refs_config(const char *var, const char *value, const char *section,
1495 struct strvec *hide_refs)
1496 {
1497 const char *key;
1498 if (!strcmp("transfer.hiderefs", var) ||
1499 (!parse_config_key(var, section, NULL, NULL, &key) &&
1500 !strcmp(key, "hiderefs"))) {
1501 char *ref;
1502 int len;
1503
1504 if (!value)
1505 return config_error_nonbool(var);
1506
1507 /* drop const to remove trailing '/' characters */
1508 ref = (char *)strvec_push(hide_refs, value);
1509 len = strlen(ref);
1510 while (len && ref[len - 1] == '/')
1511 ref[--len] = '\0';
1512 }
1513 return 0;
1514 }
1515
1516 int ref_is_hidden(const char *refname, const char *refname_full,
1517 const struct strvec *hide_refs)
1518 {
1519 int i;
1520
1521 for (i = hide_refs->nr - 1; i >= 0; i--) {
1522 const char *match = hide_refs->v[i];
1523 const char *subject;
1524 int neg = 0;
1525 const char *p;
1526
1527 if (*match == '!') {
1528 neg = 1;
1529 match++;
1530 }
1531
1532 if (*match == '^') {
1533 subject = refname_full;
1534 match++;
1535 } else {
1536 subject = refname;
1537 }
1538
1539 /* refname can be NULL when namespaces are used. */
1540 if (subject &&
1541 skip_prefix(subject, match, &p) &&
1542 (!*p || *p == '/'))
1543 return !neg;
1544 }
1545 return 0;
1546 }
1547
1548 const char **hidden_refs_to_excludes(const struct strvec *hide_refs)
1549 {
1550 const char **pattern;
1551 for (pattern = hide_refs->v; *pattern; pattern++) {
1552 /*
1553 * We can't feed any excludes from hidden refs config
1554 * sections, since later rules may override previous
1555 * ones. For example, with rules "refs/foo" and
1556 * "!refs/foo/bar", we should show "refs/foo/bar" (and
1557 * everything underneath it), but the earlier exclusion
1558 * would cause us to skip all of "refs/foo". We
1559 * likewise don't implement the namespace stripping
1560 * required for '^' rules.
1561 *
1562 * Both are possible to do, but complicated, so avoid
1563 * populating the jump list at all if we see either of
1564 * these patterns.
1565 */
1566 if (**pattern == '!' || **pattern == '^')
1567 return NULL;
1568 }
1569 return hide_refs->v;
1570 }
1571
1572 const char *find_descendant_ref(const char *dirname,
1573 const struct string_list *extras,
1574 const struct string_list *skip)
1575 {
1576 int pos;
1577
1578 if (!extras)
1579 return NULL;
1580
1581 /*
1582 * Look at the place where dirname would be inserted into
1583 * extras. If there is an entry at that position that starts
1584 * with dirname (remember, dirname includes the trailing
1585 * slash) and is not in skip, then we have a conflict.
1586 */
1587 for (pos = string_list_find_insert_index(extras, dirname, 0);
1588 pos < extras->nr; pos++) {
1589 const char *extra_refname = extras->items[pos].string;
1590
1591 if (!starts_with(extra_refname, dirname))
1592 break;
1593
1594 if (!skip || !string_list_has_string(skip, extra_refname))
1595 return extra_refname;
1596 }
1597 return NULL;
1598 }
1599
1600 int refs_head_ref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
1601 {
1602 struct object_id oid;
1603 int flag;
1604
1605 if (refs_resolve_ref_unsafe(refs, "HEAD", RESOLVE_REF_READING,
1606 &oid, &flag))
1607 return fn("HEAD", &oid, flag, cb_data);
1608
1609 return 0;
1610 }
1611
1612 int head_ref(each_ref_fn fn, void *cb_data)
1613 {
1614 return refs_head_ref(get_main_ref_store(the_repository), fn, cb_data);
1615 }
1616
1617 struct ref_iterator *refs_ref_iterator_begin(
1618 struct ref_store *refs,
1619 const char *prefix,
1620 const char **exclude_patterns,
1621 int trim,
1622 enum do_for_each_ref_flags flags)
1623 {
1624 struct ref_iterator *iter;
1625
1626 if (!(flags & DO_FOR_EACH_INCLUDE_BROKEN)) {
1627 static int ref_paranoia = -1;
1628
1629 if (ref_paranoia < 0)
1630 ref_paranoia = git_env_bool("GIT_REF_PARANOIA", 1);
1631 if (ref_paranoia) {
1632 flags |= DO_FOR_EACH_INCLUDE_BROKEN;
1633 flags |= DO_FOR_EACH_OMIT_DANGLING_SYMREFS;
1634 }
1635 }
1636
1637 iter = refs->be->iterator_begin(refs, prefix, exclude_patterns, flags);
1638 /*
1639 * `iterator_begin()` already takes care of prefix, but we
1640 * might need to do some trimming:
1641 */
1642 if (trim)
1643 iter = prefix_ref_iterator_begin(iter, "", trim);
1644
1645 return iter;
1646 }
1647
1648 /*
1649 * Call fn for each reference in the specified submodule for which the
1650 * refname begins with prefix. If trim is non-zero, then trim that
1651 * many characters off the beginning of each refname before passing
1652 * the refname to fn. flags can be DO_FOR_EACH_INCLUDE_BROKEN to
1653 * include broken references in the iteration. If fn ever returns a
1654 * non-zero value, stop the iteration and return that value;
1655 * otherwise, return 0.
1656 */
1657 static int do_for_each_repo_ref(struct repository *r, const char *prefix,
1658 each_repo_ref_fn fn, int trim, int flags,
1659 void *cb_data)
1660 {
1661 struct ref_iterator *iter;
1662 struct ref_store *refs = get_main_ref_store(r);
1663
1664 if (!refs)
1665 return 0;
1666
1667 iter = refs_ref_iterator_begin(refs, prefix, NULL, trim, flags);
1668
1669 return do_for_each_repo_ref_iterator(r, iter, fn, cb_data);
1670 }
1671
1672 struct do_for_each_ref_help {
1673 each_ref_fn *fn;
1674 void *cb_data;
1675 };
1676
1677 static int do_for_each_ref_helper(struct repository *r UNUSED,
1678 const char *refname,
1679 const struct object_id *oid,
1680 int flags,
1681 void *cb_data)
1682 {
1683 struct do_for_each_ref_help *hp = cb_data;
1684
1685 return hp->fn(refname, oid, flags, hp->cb_data);
1686 }
1687
1688 static int do_for_each_ref(struct ref_store *refs, const char *prefix,
1689 const char **exclude_patterns,
1690 each_ref_fn fn, int trim,
1691 enum do_for_each_ref_flags flags, void *cb_data)
1692 {
1693 struct ref_iterator *iter;
1694 struct do_for_each_ref_help hp = { fn, cb_data };
1695
1696 if (!refs)
1697 return 0;
1698
1699 iter = refs_ref_iterator_begin(refs, prefix, exclude_patterns, trim,
1700 flags);
1701
1702 return do_for_each_repo_ref_iterator(the_repository, iter,
1703 do_for_each_ref_helper, &hp);
1704 }
1705
1706 int refs_for_each_ref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
1707 {
1708 return do_for_each_ref(refs, "", NULL, fn, 0, 0, cb_data);
1709 }
1710
1711 int for_each_ref(each_ref_fn fn, void *cb_data)
1712 {
1713 return refs_for_each_ref(get_main_ref_store(the_repository), fn, cb_data);
1714 }
1715
1716 int refs_for_each_ref_in(struct ref_store *refs, const char *prefix,
1717 each_ref_fn fn, void *cb_data)
1718 {
1719 return do_for_each_ref(refs, prefix, NULL, fn, strlen(prefix), 0, cb_data);
1720 }
1721
1722 int for_each_ref_in(const char *prefix, each_ref_fn fn, void *cb_data)
1723 {
1724 return refs_for_each_ref_in(get_main_ref_store(the_repository), prefix, fn, cb_data);
1725 }
1726
1727 int for_each_fullref_in(const char *prefix, each_ref_fn fn, void *cb_data)
1728 {
1729 return do_for_each_ref(get_main_ref_store(the_repository),
1730 prefix, NULL, fn, 0, 0, cb_data);
1731 }
1732
1733 int refs_for_each_fullref_in(struct ref_store *refs, const char *prefix,
1734 const char **exclude_patterns,
1735 each_ref_fn fn, void *cb_data)
1736 {
1737 return do_for_each_ref(refs, prefix, exclude_patterns, fn, 0, 0, cb_data);
1738 }
1739
1740 int for_each_replace_ref(struct repository *r, each_repo_ref_fn fn, void *cb_data)
1741 {
1742 const char *git_replace_ref_base = ref_namespace[NAMESPACE_REPLACE].ref;
1743 return do_for_each_repo_ref(r, git_replace_ref_base, fn,
1744 strlen(git_replace_ref_base),
1745 DO_FOR_EACH_INCLUDE_BROKEN, cb_data);
1746 }
1747
1748 int for_each_namespaced_ref(const char **exclude_patterns,
1749 each_ref_fn fn, void *cb_data)
1750 {
1751 struct strbuf buf = STRBUF_INIT;
1752 int ret;
1753 strbuf_addf(&buf, "%srefs/", get_git_namespace());
1754 ret = do_for_each_ref(get_main_ref_store(the_repository),
1755 buf.buf, exclude_patterns, fn, 0, 0, cb_data);
1756 strbuf_release(&buf);
1757 return ret;
1758 }
1759
1760 int refs_for_each_rawref(struct ref_store *refs, each_ref_fn fn, void *cb_data)
1761 {
1762 return do_for_each_ref(refs, "", NULL, fn, 0,
1763 DO_FOR_EACH_INCLUDE_BROKEN, cb_data);
1764 }
1765
1766 int for_each_rawref(each_ref_fn fn, void *cb_data)
1767 {
1768 return refs_for_each_rawref(get_main_ref_store(the_repository), fn, cb_data);
1769 }
1770
1771 int refs_for_each_include_root_refs(struct ref_store *refs, each_ref_fn fn,
1772 void *cb_data)
1773 {
1774 return do_for_each_ref(refs, "", NULL, fn, 0,
1775 DO_FOR_EACH_INCLUDE_ROOT_REFS, cb_data);
1776 }
1777
1778 static int qsort_strcmp(const void *va, const void *vb)
1779 {
1780 const char *a = *(const char **)va;
1781 const char *b = *(const char **)vb;
1782
1783 return strcmp(a, b);
1784 }
1785
1786 static void find_longest_prefixes_1(struct string_list *out,
1787 struct strbuf *prefix,
1788 const char **patterns, size_t nr)
1789 {
1790 size_t i;
1791
1792 for (i = 0; i < nr; i++) {
1793 char c = patterns[i][prefix->len];
1794 if (!c || is_glob_special(c)) {
1795 string_list_append(out, prefix->buf);
1796 return;
1797 }
1798 }
1799
1800 i = 0;
1801 while (i < nr) {
1802 size_t end;
1803
1804 /*
1805 * Set "end" to the index of the element _after_ the last one
1806 * in our group.
1807 */
1808 for (end = i + 1; end < nr; end++) {
1809 if (patterns[i][prefix->len] != patterns[end][prefix->len])
1810 break;
1811 }
1812
1813 strbuf_addch(prefix, patterns[i][prefix->len]);
1814 find_longest_prefixes_1(out, prefix, patterns + i, end - i);
1815 strbuf_setlen(prefix, prefix->len - 1);
1816
1817 i = end;
1818 }
1819 }
1820
1821 static void find_longest_prefixes(struct string_list *out,
1822 const char **patterns)
1823 {
1824 struct strvec sorted = STRVEC_INIT;
1825 struct strbuf prefix = STRBUF_INIT;
1826
1827 strvec_pushv(&sorted, patterns);
1828 QSORT(sorted.v, sorted.nr, qsort_strcmp);
1829
1830 find_longest_prefixes_1(out, &prefix, sorted.v, sorted.nr);
1831
1832 strvec_clear(&sorted);
1833 strbuf_release(&prefix);
1834 }
1835
1836 int refs_for_each_fullref_in_prefixes(struct ref_store *ref_store,
1837 const char *namespace,
1838 const char **patterns,
1839 const char **exclude_patterns,
1840 each_ref_fn fn, void *cb_data)
1841 {
1842 struct string_list prefixes = STRING_LIST_INIT_DUP;
1843 struct string_list_item *prefix;
1844 struct strbuf buf = STRBUF_INIT;
1845 int ret = 0, namespace_len;
1846
1847 find_longest_prefixes(&prefixes, patterns);
1848
1849 if (namespace)
1850 strbuf_addstr(&buf, namespace);
1851 namespace_len = buf.len;
1852
1853 for_each_string_list_item(prefix, &prefixes) {
1854 strbuf_addstr(&buf, prefix->string);
1855 ret = refs_for_each_fullref_in(ref_store, buf.buf,
1856 exclude_patterns, fn, cb_data);
1857 if (ret)
1858 break;
1859 strbuf_setlen(&buf, namespace_len);
1860 }
1861
1862 string_list_clear(&prefixes, 0);
1863 strbuf_release(&buf);
1864 return ret;
1865 }
1866
1867 static int refs_read_special_head(struct ref_store *ref_store,
1868 const char *refname, struct object_id *oid,
1869 struct strbuf *referent, unsigned int *type,
1870 int *failure_errno)
1871 {
1872 struct strbuf full_path = STRBUF_INIT;
1873 struct strbuf content = STRBUF_INIT;
1874 int result = -1;
1875 strbuf_addf(&full_path, "%s/%s", ref_store->gitdir, refname);
1876
1877 if (strbuf_read_file(&content, full_path.buf, 0) < 0) {
1878 *failure_errno = errno;
1879 goto done;
1880 }
1881
1882 result = parse_loose_ref_contents(content.buf, oid, referent, type,
1883 failure_errno);
1884
1885 done:
1886 strbuf_release(&full_path);
1887 strbuf_release(&content);
1888 return result;
1889 }
1890
1891 static int is_special_ref(const char *refname)
1892 {
1893 /*
1894 * Special references are refs that have different semantics compared
1895 * to "normal" refs. These refs can thus not be stored in the ref
1896 * backend, but must always be accessed via the filesystem. The
1897 * following refs are special:
1898 *
1899 * - FETCH_HEAD may contain multiple object IDs, and each one of them
1900 * carries additional metadata like where it came from.
1901 *
1902 * - MERGE_HEAD may contain multiple object IDs when merging multiple
1903 * heads.
1904 *
1905 * Reading, writing or deleting references must consistently go either
1906 * through the filesystem (special refs) or through the reference
1907 * backend (normal ones).
1908 */
1909 static const char * const special_refs[] = {
1910 "FETCH_HEAD",
1911 "MERGE_HEAD",
1912 };
1913 size_t i;
1914
1915 for (i = 0; i < ARRAY_SIZE(special_refs); i++)
1916 if (!strcmp(refname, special_refs[i]))
1917 return 1;
1918
1919 return 0;
1920 }
1921
1922 int refs_read_raw_ref(struct ref_store *ref_store, const char *refname,
1923 struct object_id *oid, struct strbuf *referent,
1924 unsigned int *type, int *failure_errno)
1925 {
1926 assert(failure_errno);
1927 if (is_special_ref(refname))
1928 return refs_read_special_head(ref_store, refname, oid, referent,
1929 type, failure_errno);
1930
1931 return ref_store->be->read_raw_ref(ref_store, refname, oid, referent,
1932 type, failure_errno);
1933 }
1934
1935 int refs_read_symbolic_ref(struct ref_store *ref_store, const char *refname,
1936 struct strbuf *referent)
1937 {
1938 return ref_store->be->read_symbolic_ref(ref_store, refname, referent);
1939 }
1940
1941 const char *refs_resolve_ref_unsafe(struct ref_store *refs,
1942 const char *refname,
1943 int resolve_flags,
1944 struct object_id *oid,
1945 int *flags)
1946 {
1947 static struct strbuf sb_refname = STRBUF_INIT;
1948 struct object_id unused_oid;
1949 int unused_flags;
1950 int symref_count;
1951
1952 if (!oid)
1953 oid = &unused_oid;
1954 if (!flags)
1955 flags = &unused_flags;
1956
1957 *flags = 0;
1958
1959 if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL)) {
1960 if (!(resolve_flags & RESOLVE_REF_ALLOW_BAD_NAME) ||
1961 !refname_is_safe(refname))
1962 return NULL;
1963
1964 /*
1965 * repo_dwim_ref() uses REF_ISBROKEN to distinguish between
1966 * missing refs and refs that were present but invalid,
1967 * to complain about the latter to stderr.
1968 *
1969 * We don't know whether the ref exists, so don't set
1970 * REF_ISBROKEN yet.
1971 */
1972 *flags |= REF_BAD_NAME;
1973 }
1974
1975 for (symref_count = 0; symref_count < SYMREF_MAXDEPTH; symref_count++) {
1976 unsigned int read_flags = 0;
1977 int failure_errno;
1978
1979 if (refs_read_raw_ref(refs, refname, oid, &sb_refname,
1980 &read_flags, &failure_errno)) {
1981 *flags |= read_flags;
1982
1983 /* In reading mode, refs must eventually resolve */
1984 if (resolve_flags & RESOLVE_REF_READING)
1985 return NULL;
1986
1987 /*
1988 * Otherwise a missing ref is OK. But the files backend
1989 * may show errors besides ENOENT if there are
1990 * similarly-named refs.
1991 */
1992 if (failure_errno != ENOENT &&
1993 failure_errno != EISDIR &&
1994 failure_errno != ENOTDIR)
1995 return NULL;
1996
1997 oidclr(oid);
1998 if (*flags & REF_BAD_NAME)
1999 *flags |= REF_ISBROKEN;
2000 return refname;
2001 }
2002
2003 *flags |= read_flags;
2004
2005 if (!(read_flags & REF_ISSYMREF)) {
2006 if (*flags & REF_BAD_NAME) {
2007 oidclr(oid);
2008 *flags |= REF_ISBROKEN;
2009 }
2010 return refname;
2011 }
2012
2013 refname = sb_refname.buf;
2014 if (resolve_flags & RESOLVE_REF_NO_RECURSE) {
2015 oidclr(oid);
2016 return refname;
2017 }
2018 if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL)) {
2019 if (!(resolve_flags & RESOLVE_REF_ALLOW_BAD_NAME) ||
2020 !refname_is_safe(refname))
2021 return NULL;
2022
2023 *flags |= REF_ISBROKEN | REF_BAD_NAME;
2024 }
2025 }
2026
2027 return NULL;
2028 }
2029
2030 /* backend functions */
2031 int refs_init_db(struct ref_store *refs, int flags, struct strbuf *err)
2032 {
2033 return refs->be->init_db(refs, flags, err);
2034 }
2035
2036 const char *resolve_ref_unsafe(const char *refname, int resolve_flags,
2037 struct object_id *oid, int *flags)
2038 {
2039 return refs_resolve_ref_unsafe(get_main_ref_store(the_repository), refname,
2040 resolve_flags, oid, flags);
2041 }
2042
2043 int resolve_gitlink_ref(const char *submodule, const char *refname,
2044 struct object_id *oid)
2045 {
2046 struct ref_store *refs;
2047 int flags;
2048
2049 refs = get_submodule_ref_store(submodule);
2050
2051 if (!refs)
2052 return -1;
2053
2054 if (!refs_resolve_ref_unsafe(refs, refname, 0, oid, &flags) ||
2055 is_null_oid(oid))
2056 return -1;
2057 return 0;
2058 }
2059
2060 struct ref_store_hash_entry
2061 {
2062 struct hashmap_entry ent;
2063
2064 struct ref_store *refs;
2065
2066 /* NUL-terminated identifier of the ref store: */
2067 char name[FLEX_ARRAY];
2068 };
2069
2070 static int ref_store_hash_cmp(const void *cmp_data UNUSED,
2071 const struct hashmap_entry *eptr,
2072 const struct hashmap_entry *entry_or_key,
2073 const void *keydata)
2074 {
2075 const struct ref_store_hash_entry *e1, *e2;
2076 const char *name;
2077
2078 e1 = container_of(eptr, const struct ref_store_hash_entry, ent);
2079 e2 = container_of(entry_or_key, const struct ref_store_hash_entry, ent);
2080 name = keydata ? keydata : e2->name;
2081
2082 return strcmp(e1->name, name);
2083 }
2084
2085 static struct ref_store_hash_entry *alloc_ref_store_hash_entry(
2086 const char *name, struct ref_store *refs)
2087 {
2088 struct ref_store_hash_entry *entry;
2089
2090 FLEX_ALLOC_STR(entry, name, name);
2091 hashmap_entry_init(&entry->ent, strhash(name));
2092 entry->refs = refs;
2093 return entry;
2094 }
2095
2096 /* A hashmap of ref_stores, stored by submodule name: */
2097 static struct hashmap submodule_ref_stores;
2098
2099 /* A hashmap of ref_stores, stored by worktree id: */
2100 static struct hashmap worktree_ref_stores;
2101
2102 /*
2103 * Look up a ref store by name. If that ref_store hasn't been
2104 * registered yet, return NULL.
2105 */
2106 static struct ref_store *lookup_ref_store_map(struct hashmap *map,
2107 const char *name)
2108 {
2109 struct ref_store_hash_entry *entry;
2110 unsigned int hash;
2111
2112 if (!map->tablesize)
2113 /* It's initialized on demand in register_ref_store(). */
2114 return NULL;
2115
2116 hash = strhash(name);
2117 entry = hashmap_get_entry_from_hash(map, hash, name,
2118 struct ref_store_hash_entry, ent);
2119 return entry ? entry->refs : NULL;
2120 }
2121
2122 /*
2123 * Create, record, and return a ref_store instance for the specified
2124 * gitdir.
2125 */
2126 static struct ref_store *ref_store_init(struct repository *repo,
2127 const char *gitdir,
2128 unsigned int flags)
2129 {
2130 const struct ref_storage_be *be;
2131 struct ref_store *refs;
2132
2133 be = find_ref_storage_backend(repo->ref_storage_format);
2134 if (!be)
2135 BUG("reference backend is unknown");
2136
2137 refs = be->init(repo, gitdir, flags);
2138 return refs;
2139 }
2140
2141 struct ref_store *get_main_ref_store(struct repository *r)
2142 {
2143 if (r->refs_private)
2144 return r->refs_private;
2145
2146 if (!r->gitdir)
2147 BUG("attempting to get main_ref_store outside of repository");
2148
2149 r->refs_private = ref_store_init(r, r->gitdir, REF_STORE_ALL_CAPS);
2150 r->refs_private = maybe_debug_wrap_ref_store(r->gitdir, r->refs_private);
2151 return r->refs_private;
2152 }
2153
2154 /*
2155 * Associate a ref store with a name. It is a fatal error to call this
2156 * function twice for the same name.
2157 */
2158 static void register_ref_store_map(struct hashmap *map,
2159 const char *type,
2160 struct ref_store *refs,
2161 const char *name)
2162 {
2163 struct ref_store_hash_entry *entry;
2164
2165 if (!map->tablesize)
2166 hashmap_init(map, ref_store_hash_cmp, NULL, 0);
2167
2168 entry = alloc_ref_store_hash_entry(name, refs);
2169 if (hashmap_put(map, &entry->ent))
2170 BUG("%s ref_store '%s' initialized twice", type, name);
2171 }
2172
2173 struct ref_store *get_submodule_ref_store(const char *submodule)
2174 {
2175 struct strbuf submodule_sb = STRBUF_INIT;
2176 struct ref_store *refs;
2177 char *to_free = NULL;
2178 size_t len;
2179 struct repository *subrepo;
2180
2181 if (!submodule)
2182 return NULL;
2183
2184 len = strlen(submodule);
2185 while (len && is_dir_sep(submodule[len - 1]))
2186 len--;
2187 if (!len)
2188 return NULL;
2189
2190 if (submodule[len])
2191 /* We need to strip off one or more trailing slashes */
2192 submodule = to_free = xmemdupz(submodule, len);
2193
2194 refs = lookup_ref_store_map(&submodule_ref_stores, submodule);
2195 if (refs)
2196 goto done;
2197
2198 strbuf_addstr(&submodule_sb, submodule);
2199 if (!is_nonbare_repository_dir(&submodule_sb))
2200 goto done;
2201
2202 if (submodule_to_gitdir(&submodule_sb, submodule))
2203 goto done;
2204
2205 subrepo = xmalloc(sizeof(*subrepo));
2206 /*
2207 * NEEDSWORK: Make get_submodule_ref_store() work with arbitrary
2208 * superprojects other than the_repository. This probably should be
2209 * done by making it take a struct repository * parameter instead of a
2210 * submodule path.
2211 */
2212 if (repo_submodule_init(subrepo, the_repository, submodule,
2213 null_oid())) {
2214 free(subrepo);
2215 goto done;
2216 }
2217 refs = ref_store_init(subrepo, submodule_sb.buf,
2218 REF_STORE_READ | REF_STORE_ODB);
2219 register_ref_store_map(&submodule_ref_stores, "submodule",
2220 refs, submodule);
2221
2222 done:
2223 strbuf_release(&submodule_sb);
2224 free(to_free);
2225
2226 return refs;
2227 }
2228
2229 struct ref_store *get_worktree_ref_store(const struct worktree *wt)
2230 {
2231 struct ref_store *refs;
2232 const char *id;
2233
2234 if (wt->is_current)
2235 return get_main_ref_store(the_repository);
2236
2237 id = wt->id ? wt->id : "/";
2238 refs = lookup_ref_store_map(&worktree_ref_stores, id);
2239 if (refs)
2240 return refs;
2241
2242 if (wt->id)
2243 refs = ref_store_init(the_repository,
2244 git_common_path("worktrees/%s", wt->id),
2245 REF_STORE_ALL_CAPS);
2246 else
2247 refs = ref_store_init(the_repository,
2248 get_git_common_dir(),
2249 REF_STORE_ALL_CAPS);
2250
2251 if (refs)
2252 register_ref_store_map(&worktree_ref_stores, "worktree",
2253 refs, id);
2254 return refs;
2255 }
2256
2257 void base_ref_store_init(struct ref_store *refs, struct repository *repo,
2258 const char *path, const struct ref_storage_be *be)
2259 {
2260 refs->be = be;
2261 refs->repo = repo;
2262 refs->gitdir = xstrdup(path);
2263 }
2264
2265 /* backend functions */
2266 int refs_pack_refs(struct ref_store *refs, struct pack_refs_opts *opts)
2267 {
2268 return refs->be->pack_refs(refs, opts);
2269 }
2270
2271 int peel_iterated_oid(const struct object_id *base, struct object_id *peeled)
2272 {
2273 if (current_ref_iter &&
2274 (current_ref_iter->oid == base ||
2275 oideq(current_ref_iter->oid, base)))
2276 return ref_iterator_peel(current_ref_iter, peeled);
2277
2278 return peel_object(base, peeled) ? -1 : 0;
2279 }
2280
2281 int refs_create_symref(struct ref_store *refs,
2282 const char *ref_target,
2283 const char *refs_heads_master,
2284 const char *logmsg)
2285 {
2286 char *msg;
2287 int retval;
2288
2289 msg = normalize_reflog_message(logmsg);
2290 retval = refs->be->create_symref(refs, ref_target, refs_heads_master,
2291 msg);
2292 free(msg);
2293 return retval;
2294 }
2295
2296 int create_symref(const char *ref_target, const char *refs_heads_master,
2297 const char *logmsg)
2298 {
2299 return refs_create_symref(get_main_ref_store(the_repository), ref_target,
2300 refs_heads_master, logmsg);
2301 }
2302
2303 int ref_update_reject_duplicates(struct string_list *refnames,
2304 struct strbuf *err)
2305 {
2306 size_t i, n = refnames->nr;
2307
2308 assert(err);
2309
2310 for (i = 1; i < n; i++) {
2311 int cmp = strcmp(refnames->items[i - 1].string,
2312 refnames->items[i].string);
2313
2314 if (!cmp) {
2315 strbuf_addf(err,
2316 _("multiple updates for ref '%s' not allowed"),
2317 refnames->items[i].string);
2318 return 1;
2319 } else if (cmp > 0) {
2320 BUG("ref_update_reject_duplicates() received unsorted list");
2321 }
2322 }
2323 return 0;
2324 }
2325
2326 static int run_transaction_hook(struct ref_transaction *transaction,
2327 const char *state)
2328 {
2329 struct child_process proc = CHILD_PROCESS_INIT;
2330 struct strbuf buf = STRBUF_INIT;
2331 const char *hook;
2332 int ret = 0, i;
2333
2334 hook = find_hook("reference-transaction");
2335 if (!hook)
2336 return ret;
2337
2338 strvec_pushl(&proc.args, hook, state, NULL);
2339 proc.in = -1;
2340 proc.stdout_to_stderr = 1;
2341 proc.trace2_hook_name = "reference-transaction";
2342
2343 ret = start_command(&proc);
2344 if (ret)
2345 return ret;
2346
2347 sigchain_push(SIGPIPE, SIG_IGN);
2348
2349 for (i = 0; i < transaction->nr; i++) {
2350 struct ref_update *update = transaction->updates[i];
2351
2352 strbuf_reset(&buf);
2353 strbuf_addf(&buf, "%s %s %s\n",
2354 oid_to_hex(&update->old_oid),
2355 oid_to_hex(&update->new_oid),
2356 update->refname);
2357
2358 if (write_in_full(proc.in, buf.buf, buf.len) < 0) {
2359 if (errno != EPIPE) {
2360 /* Don't leak errno outside this API */
2361 errno = 0;
2362 ret = -1;
2363 }
2364 break;
2365 }
2366 }
2367
2368 close(proc.in);
2369 sigchain_pop(SIGPIPE);
2370 strbuf_release(&buf);
2371
2372 ret |= finish_command(&proc);
2373 return ret;
2374 }
2375
2376 int ref_transaction_prepare(struct ref_transaction *transaction,
2377 struct strbuf *err)
2378 {
2379 struct ref_store *refs = transaction->ref_store;
2380 int ret;
2381
2382 switch (transaction->state) {
2383 case REF_TRANSACTION_OPEN:
2384 /* Good. */
2385 break;
2386 case REF_TRANSACTION_PREPARED:
2387 BUG("prepare called twice on reference transaction");
2388 break;
2389 case REF_TRANSACTION_CLOSED:
2390 BUG("prepare called on a closed reference transaction");
2391 break;
2392 default:
2393 BUG("unexpected reference transaction state");
2394 break;
2395 }
2396
2397 if (refs->repo->objects->odb->disable_ref_updates) {
2398 strbuf_addstr(err,
2399 _("ref updates forbidden inside quarantine environment"));
2400 return -1;
2401 }
2402
2403 ret = refs->be->transaction_prepare(refs, transaction, err);
2404 if (ret)
2405 return ret;
2406
2407 ret = run_transaction_hook(transaction, "prepared");
2408 if (ret) {
2409 ref_transaction_abort(transaction, err);
2410 die(_("ref updates aborted by hook"));
2411 }
2412
2413 return 0;
2414 }
2415
2416 int ref_transaction_abort(struct ref_transaction *transaction,
2417 struct strbuf *err)
2418 {
2419 struct ref_store *refs = transaction->ref_store;
2420 int ret = 0;
2421
2422 switch (transaction->state) {
2423 case REF_TRANSACTION_OPEN:
2424 /* No need to abort explicitly. */
2425 break;
2426 case REF_TRANSACTION_PREPARED:
2427 ret = refs->be->transaction_abort(refs, transaction, err);
2428 break;
2429 case REF_TRANSACTION_CLOSED:
2430 BUG("abort called on a closed reference transaction");
2431 break;
2432 default:
2433 BUG("unexpected reference transaction state");
2434 break;
2435 }
2436
2437 run_transaction_hook(transaction, "aborted");
2438
2439 ref_transaction_free(transaction);
2440 return ret;
2441 }
2442
2443 int ref_transaction_commit(struct ref_transaction *transaction,
2444 struct strbuf *err)
2445 {
2446 struct ref_store *refs = transaction->ref_store;
2447 int ret;
2448
2449 switch (transaction->state) {
2450 case REF_TRANSACTION_OPEN:
2451 /* Need to prepare first. */
2452 ret = ref_transaction_prepare(transaction, err);
2453 if (ret)
2454 return ret;
2455 break;
2456 case REF_TRANSACTION_PREPARED:
2457 /* Fall through to finish. */
2458 break;
2459 case REF_TRANSACTION_CLOSED:
2460 BUG("commit called on a closed reference transaction");
2461 break;
2462 default:
2463 BUG("unexpected reference transaction state");
2464 break;
2465 }
2466
2467 ret = refs->be->transaction_finish(refs, transaction, err);
2468 if (!ret)
2469 run_transaction_hook(transaction, "committed");
2470 return ret;
2471 }
2472
2473 int refs_verify_refname_available(struct ref_store *refs,
2474 const char *refname,
2475 const struct string_list *extras,
2476 const struct string_list *skip,
2477 struct strbuf *err)
2478 {
2479 const char *slash;
2480 const char *extra_refname;
2481 struct strbuf dirname = STRBUF_INIT;
2482 struct strbuf referent = STRBUF_INIT;
2483 struct object_id oid;
2484 unsigned int type;
2485 struct ref_iterator *iter;
2486 int ok;
2487 int ret = -1;
2488
2489 /*
2490 * For the sake of comments in this function, suppose that
2491 * refname is "refs/foo/bar".
2492 */
2493
2494 assert(err);
2495
2496 strbuf_grow(&dirname, strlen(refname) + 1);
2497 for (slash = strchr(refname, '/'); slash; slash = strchr(slash + 1, '/')) {
2498 /*
2499 * Just saying "Is a directory" when we e.g. can't
2500 * lock some multi-level ref isn't very informative,
2501 * the user won't be told *what* is a directory, so
2502 * let's not use strerror() below.
2503 */
2504 int ignore_errno;
2505 /* Expand dirname to the new prefix, not including the trailing slash: */
2506 strbuf_add(&dirname, refname + dirname.len, slash - refname - dirname.len);
2507
2508 /*
2509 * We are still at a leading dir of the refname (e.g.,
2510 * "refs/foo"; if there is a reference with that name,
2511 * it is a conflict, *unless* it is in skip.
2512 */
2513 if (skip && string_list_has_string(skip, dirname.buf))
2514 continue;
2515
2516 if (!refs_read_raw_ref(refs, dirname.buf, &oid, &referent,
2517 &type, &ignore_errno)) {
2518 strbuf_addf(err, _("'%s' exists; cannot create '%s'"),
2519 dirname.buf, refname);
2520 goto cleanup;
2521 }
2522
2523 if (extras && string_list_has_string(extras, dirname.buf)) {
2524 strbuf_addf(err, _("cannot process '%s' and '%s' at the same time"),
2525 refname, dirname.buf);
2526 goto cleanup;
2527 }
2528 }
2529
2530 /*
2531 * We are at the leaf of our refname (e.g., "refs/foo/bar").
2532 * There is no point in searching for a reference with that
2533 * name, because a refname isn't considered to conflict with
2534 * itself. But we still need to check for references whose
2535 * names are in the "refs/foo/bar/" namespace, because they
2536 * *do* conflict.
2537 */
2538 strbuf_addstr(&dirname, refname + dirname.len);
2539 strbuf_addch(&dirname, '/');
2540
2541 iter = refs_ref_iterator_begin(refs, dirname.buf, NULL, 0,
2542 DO_FOR_EACH_INCLUDE_BROKEN);
2543 while ((ok = ref_iterator_advance(iter)) == ITER_OK) {
2544 if (skip &&
2545 string_list_has_string(skip, iter->refname))
2546 continue;
2547
2548 strbuf_addf(err, _("'%s' exists; cannot create '%s'"),
2549 iter->refname, refname);
2550 ref_iterator_abort(iter);
2551 goto cleanup;
2552 }
2553
2554 if (ok != ITER_DONE)
2555 BUG("error while iterating over references");
2556
2557 extra_refname = find_descendant_ref(dirname.buf, extras, skip);
2558 if (extra_refname)
2559 strbuf_addf(err, _("cannot process '%s' and '%s' at the same time"),
2560 refname, extra_refname);
2561 else
2562 ret = 0;
2563
2564 cleanup:
2565 strbuf_release(&referent);
2566 strbuf_release(&dirname);
2567 return ret;
2568 }
2569
2570 struct do_for_each_reflog_help {
2571 each_reflog_fn *fn;
2572 void *cb_data;
2573 };
2574
2575 static int do_for_each_reflog_helper(struct repository *r UNUSED,
2576 const char *refname,
2577 const struct object_id *oid UNUSED,
2578 int flags,
2579 void *cb_data)
2580 {
2581 struct do_for_each_reflog_help *hp = cb_data;
2582 return hp->fn(refname, hp->cb_data);
2583 }
2584
2585 int refs_for_each_reflog(struct ref_store *refs, each_reflog_fn fn, void *cb_data)
2586 {
2587 struct ref_iterator *iter;
2588 struct do_for_each_reflog_help hp = { fn, cb_data };
2589
2590 iter = refs->be->reflog_iterator_begin(refs);
2591
2592 return do_for_each_repo_ref_iterator(the_repository, iter,
2593 do_for_each_reflog_helper, &hp);
2594 }
2595
2596 int for_each_reflog(each_reflog_fn fn, void *cb_data)
2597 {
2598 return refs_for_each_reflog(get_main_ref_store(the_repository), fn, cb_data);
2599 }
2600
2601 int refs_for_each_reflog_ent_reverse(struct ref_store *refs,
2602 const char *refname,
2603 each_reflog_ent_fn fn,
2604 void *cb_data)
2605 {
2606 return refs->be->for_each_reflog_ent_reverse(refs, refname,
2607 fn, cb_data);
2608 }
2609
2610 int for_each_reflog_ent_reverse(const char *refname, each_reflog_ent_fn fn,
2611 void *cb_data)
2612 {
2613 return refs_for_each_reflog_ent_reverse(get_main_ref_store(the_repository),
2614 refname, fn, cb_data);
2615 }
2616
2617 int refs_for_each_reflog_ent(struct ref_store *refs, const char *refname,
2618 each_reflog_ent_fn fn, void *cb_data)
2619 {
2620 return refs->be->for_each_reflog_ent(refs, refname, fn, cb_data);
2621 }
2622
2623 int for_each_reflog_ent(const char *refname, each_reflog_ent_fn fn,
2624 void *cb_data)
2625 {
2626 return refs_for_each_reflog_ent(get_main_ref_store(the_repository), refname,
2627 fn, cb_data);
2628 }
2629
2630 int refs_reflog_exists(struct ref_store *refs, const char *refname)
2631 {
2632 return refs->be->reflog_exists(refs, refname);
2633 }
2634
2635 int reflog_exists(const char *refname)
2636 {
2637 return refs_reflog_exists(get_main_ref_store(the_repository), refname);
2638 }
2639
2640 int refs_create_reflog(struct ref_store *refs, const char *refname,
2641 struct strbuf *err)
2642 {
2643 return refs->be->create_reflog(refs, refname, err);
2644 }
2645
2646 int safe_create_reflog(const char *refname, struct strbuf *err)
2647 {
2648 return refs_create_reflog(get_main_ref_store(the_repository), refname,
2649 err);
2650 }
2651
2652 int refs_delete_reflog(struct ref_store *refs, const char *refname)
2653 {
2654 return refs->be->delete_reflog(refs, refname);
2655 }
2656
2657 int delete_reflog(const char *refname)
2658 {
2659 return refs_delete_reflog(get_main_ref_store(the_repository), refname);
2660 }
2661
2662 int refs_reflog_expire(struct ref_store *refs,
2663 const char *refname,
2664 unsigned int flags,
2665 reflog_expiry_prepare_fn prepare_fn,
2666 reflog_expiry_should_prune_fn should_prune_fn,
2667 reflog_expiry_cleanup_fn cleanup_fn,
2668 void *policy_cb_data)
2669 {
2670 return refs->be->reflog_expire(refs, refname, flags,
2671 prepare_fn, should_prune_fn,
2672 cleanup_fn, policy_cb_data);
2673 }
2674
2675 int reflog_expire(const char *refname,
2676 unsigned int flags,
2677 reflog_expiry_prepare_fn prepare_fn,
2678 reflog_expiry_should_prune_fn should_prune_fn,
2679 reflog_expiry_cleanup_fn cleanup_fn,
2680 void *policy_cb_data)
2681 {
2682 return refs_reflog_expire(get_main_ref_store(the_repository),
2683 refname, flags,
2684 prepare_fn, should_prune_fn,
2685 cleanup_fn, policy_cb_data);
2686 }
2687
2688 int initial_ref_transaction_commit(struct ref_transaction *transaction,
2689 struct strbuf *err)
2690 {
2691 struct ref_store *refs = transaction->ref_store;
2692
2693 return refs->be->initial_transaction_commit(refs, transaction, err);
2694 }
2695
2696 void ref_transaction_for_each_queued_update(struct ref_transaction *transaction,
2697 ref_transaction_for_each_queued_update_fn cb,
2698 void *cb_data)
2699 {
2700 int i;
2701
2702 for (i = 0; i < transaction->nr; i++) {
2703 struct ref_update *update = transaction->updates[i];
2704
2705 cb(update->refname,
2706 (update->flags & REF_HAVE_OLD) ? &update->old_oid : NULL,
2707 (update->flags & REF_HAVE_NEW) ? &update->new_oid : NULL,
2708 cb_data);
2709 }
2710 }
2711
2712 int refs_delete_refs(struct ref_store *refs, const char *logmsg,
2713 struct string_list *refnames, unsigned int flags)
2714 {
2715 struct ref_transaction *transaction;
2716 struct strbuf err = STRBUF_INIT;
2717 struct string_list_item *item;
2718 int ret = 0, failures = 0;
2719 char *msg;
2720
2721 if (!refnames->nr)
2722 return 0;
2723
2724 msg = normalize_reflog_message(logmsg);
2725
2726 /*
2727 * Since we don't check the references' old_oids, the
2728 * individual updates can't fail, so we can pack all of the
2729 * updates into a single transaction.
2730 */
2731 transaction = ref_store_transaction_begin(refs, &err);
2732 if (!transaction) {
2733 ret = error("%s", err.buf);
2734 goto out;
2735 }
2736
2737 for_each_string_list_item(item, refnames) {
2738 ret = ref_transaction_delete(transaction, item->string,
2739 NULL, flags, msg, &err);
2740 if (ret) {
2741 warning(_("could not delete reference %s: %s"),
2742 item->string, err.buf);
2743 strbuf_reset(&err);
2744 failures = 1;
2745 }
2746 }
2747
2748 ret = ref_transaction_commit(transaction, &err);
2749 if (ret) {
2750 if (refnames->nr == 1)
2751 error(_("could not delete reference %s: %s"),
2752 refnames->items[0].string, err.buf);
2753 else
2754 error(_("could not delete references: %s"), err.buf);
2755 }
2756
2757 out:
2758 if (!ret && failures)
2759 ret = -1;
2760 ref_transaction_free(transaction);
2761 strbuf_release(&err);
2762 free(msg);
2763 return ret;
2764 }
2765
2766 int delete_refs(const char *msg, struct string_list *refnames,
2767 unsigned int flags)
2768 {
2769 return refs_delete_refs(get_main_ref_store(the_repository), msg, refnames, flags);
2770 }
2771
2772 int refs_rename_ref(struct ref_store *refs, const char *oldref,
2773 const char *newref, const char *logmsg)
2774 {
2775 char *msg;
2776 int retval;
2777
2778 msg = normalize_reflog_message(logmsg);
2779 retval = refs->be->rename_ref(refs, oldref, newref, msg);
2780 free(msg);
2781 return retval;
2782 }
2783
2784 int rename_ref(const char *oldref, const char *newref, const char *logmsg)
2785 {
2786 return refs_rename_ref(get_main_ref_store(the_repository), oldref, newref, logmsg);
2787 }
2788
2789 int refs_copy_existing_ref(struct ref_store *refs, const char *oldref,
2790 const char *newref, const char *logmsg)
2791 {
2792 char *msg;
2793 int retval;
2794
2795 msg = normalize_reflog_message(logmsg);
2796 retval = refs->be->copy_ref(refs, oldref, newref, msg);
2797 free(msg);
2798 return retval;
2799 }
2800
2801 int copy_existing_ref(const char *oldref, const char *newref, const char *logmsg)
2802 {
2803 return refs_copy_existing_ref(get_main_ref_store(the_repository), oldref, newref, logmsg);
2804 }