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