]> git.ipfire.org Git - thirdparty/git.git/blob - refs.c
Merge branch 'jk/parse-object-type-mismatch'
[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 for_each_fullref_in_prefixes(const char *namespace,
1727 const char **patterns,
1728 each_ref_fn fn, void *cb_data)
1729 {
1730 struct string_list prefixes = STRING_LIST_INIT_DUP;
1731 struct string_list_item *prefix;
1732 struct strbuf buf = STRBUF_INIT;
1733 int ret = 0, namespace_len;
1734
1735 find_longest_prefixes(&prefixes, patterns);
1736
1737 if (namespace)
1738 strbuf_addstr(&buf, namespace);
1739 namespace_len = buf.len;
1740
1741 for_each_string_list_item(prefix, &prefixes) {
1742 strbuf_addstr(&buf, prefix->string);
1743 ret = for_each_fullref_in(buf.buf, fn, cb_data);
1744 if (ret)
1745 break;
1746 strbuf_setlen(&buf, namespace_len);
1747 }
1748
1749 string_list_clear(&prefixes, 0);
1750 strbuf_release(&buf);
1751 return ret;
1752 }
1753
1754 static int refs_read_special_head(struct ref_store *ref_store,
1755 const char *refname, struct object_id *oid,
1756 struct strbuf *referent, unsigned int *type,
1757 int *failure_errno)
1758 {
1759 struct strbuf full_path = STRBUF_INIT;
1760 struct strbuf content = STRBUF_INIT;
1761 int result = -1;
1762 strbuf_addf(&full_path, "%s/%s", ref_store->gitdir, refname);
1763
1764 if (strbuf_read_file(&content, full_path.buf, 0) < 0)
1765 goto done;
1766
1767 result = parse_loose_ref_contents(content.buf, oid, referent, type,
1768 failure_errno);
1769
1770 done:
1771 strbuf_release(&full_path);
1772 strbuf_release(&content);
1773 return result;
1774 }
1775
1776 int refs_read_raw_ref(struct ref_store *ref_store, const char *refname,
1777 struct object_id *oid, struct strbuf *referent,
1778 unsigned int *type, int *failure_errno)
1779 {
1780 assert(failure_errno);
1781 if (!strcmp(refname, "FETCH_HEAD") || !strcmp(refname, "MERGE_HEAD")) {
1782 return refs_read_special_head(ref_store, refname, oid, referent,
1783 type, failure_errno);
1784 }
1785
1786 return ref_store->be->read_raw_ref(ref_store, refname, oid, referent,
1787 type, failure_errno);
1788 }
1789
1790 int refs_read_symbolic_ref(struct ref_store *ref_store, const char *refname,
1791 struct strbuf *referent)
1792 {
1793 return ref_store->be->read_symbolic_ref(ref_store, refname, referent);
1794 }
1795
1796 const char *refs_resolve_ref_unsafe(struct ref_store *refs,
1797 const char *refname,
1798 int resolve_flags,
1799 struct object_id *oid,
1800 int *flags)
1801 {
1802 static struct strbuf sb_refname = STRBUF_INIT;
1803 struct object_id unused_oid;
1804 int unused_flags;
1805 int symref_count;
1806
1807 if (!oid)
1808 oid = &unused_oid;
1809 if (!flags)
1810 flags = &unused_flags;
1811
1812 *flags = 0;
1813
1814 if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL)) {
1815 if (!(resolve_flags & RESOLVE_REF_ALLOW_BAD_NAME) ||
1816 !refname_is_safe(refname))
1817 return NULL;
1818
1819 /*
1820 * dwim_ref() uses REF_ISBROKEN to distinguish between
1821 * missing refs and refs that were present but invalid,
1822 * to complain about the latter to stderr.
1823 *
1824 * We don't know whether the ref exists, so don't set
1825 * REF_ISBROKEN yet.
1826 */
1827 *flags |= REF_BAD_NAME;
1828 }
1829
1830 for (symref_count = 0; symref_count < SYMREF_MAXDEPTH; symref_count++) {
1831 unsigned int read_flags = 0;
1832 int failure_errno;
1833
1834 if (refs_read_raw_ref(refs, refname, oid, &sb_refname,
1835 &read_flags, &failure_errno)) {
1836 *flags |= read_flags;
1837
1838 /* In reading mode, refs must eventually resolve */
1839 if (resolve_flags & RESOLVE_REF_READING)
1840 return NULL;
1841
1842 /*
1843 * Otherwise a missing ref is OK. But the files backend
1844 * may show errors besides ENOENT if there are
1845 * similarly-named refs.
1846 */
1847 if (failure_errno != ENOENT &&
1848 failure_errno != EISDIR &&
1849 failure_errno != ENOTDIR)
1850 return NULL;
1851
1852 oidclr(oid);
1853 if (*flags & REF_BAD_NAME)
1854 *flags |= REF_ISBROKEN;
1855 return refname;
1856 }
1857
1858 *flags |= read_flags;
1859
1860 if (!(read_flags & REF_ISSYMREF)) {
1861 if (*flags & REF_BAD_NAME) {
1862 oidclr(oid);
1863 *flags |= REF_ISBROKEN;
1864 }
1865 return refname;
1866 }
1867
1868 refname = sb_refname.buf;
1869 if (resolve_flags & RESOLVE_REF_NO_RECURSE) {
1870 oidclr(oid);
1871 return refname;
1872 }
1873 if (check_refname_format(refname, REFNAME_ALLOW_ONELEVEL)) {
1874 if (!(resolve_flags & RESOLVE_REF_ALLOW_BAD_NAME) ||
1875 !refname_is_safe(refname))
1876 return NULL;
1877
1878 *flags |= REF_ISBROKEN | REF_BAD_NAME;
1879 }
1880 }
1881
1882 return NULL;
1883 }
1884
1885 /* backend functions */
1886 int refs_init_db(struct strbuf *err)
1887 {
1888 struct ref_store *refs = get_main_ref_store(the_repository);
1889
1890 return refs->be->init_db(refs, err);
1891 }
1892
1893 const char *resolve_ref_unsafe(const char *refname, int resolve_flags,
1894 struct object_id *oid, int *flags)
1895 {
1896 return refs_resolve_ref_unsafe(get_main_ref_store(the_repository), refname,
1897 resolve_flags, oid, flags);
1898 }
1899
1900 int resolve_gitlink_ref(const char *submodule, const char *refname,
1901 struct object_id *oid)
1902 {
1903 struct ref_store *refs;
1904 int flags;
1905
1906 refs = get_submodule_ref_store(submodule);
1907
1908 if (!refs)
1909 return -1;
1910
1911 if (!refs_resolve_ref_unsafe(refs, refname, 0, oid, &flags) ||
1912 is_null_oid(oid))
1913 return -1;
1914 return 0;
1915 }
1916
1917 struct ref_store_hash_entry
1918 {
1919 struct hashmap_entry ent;
1920
1921 struct ref_store *refs;
1922
1923 /* NUL-terminated identifier of the ref store: */
1924 char name[FLEX_ARRAY];
1925 };
1926
1927 static int ref_store_hash_cmp(const void *cmp_data UNUSED,
1928 const struct hashmap_entry *eptr,
1929 const struct hashmap_entry *entry_or_key,
1930 const void *keydata)
1931 {
1932 const struct ref_store_hash_entry *e1, *e2;
1933 const char *name;
1934
1935 e1 = container_of(eptr, const struct ref_store_hash_entry, ent);
1936 e2 = container_of(entry_or_key, const struct ref_store_hash_entry, ent);
1937 name = keydata ? keydata : e2->name;
1938
1939 return strcmp(e1->name, name);
1940 }
1941
1942 static struct ref_store_hash_entry *alloc_ref_store_hash_entry(
1943 const char *name, struct ref_store *refs)
1944 {
1945 struct ref_store_hash_entry *entry;
1946
1947 FLEX_ALLOC_STR(entry, name, name);
1948 hashmap_entry_init(&entry->ent, strhash(name));
1949 entry->refs = refs;
1950 return entry;
1951 }
1952
1953 /* A hashmap of ref_stores, stored by submodule name: */
1954 static struct hashmap submodule_ref_stores;
1955
1956 /* A hashmap of ref_stores, stored by worktree id: */
1957 static struct hashmap worktree_ref_stores;
1958
1959 /*
1960 * Look up a ref store by name. If that ref_store hasn't been
1961 * registered yet, return NULL.
1962 */
1963 static struct ref_store *lookup_ref_store_map(struct hashmap *map,
1964 const char *name)
1965 {
1966 struct ref_store_hash_entry *entry;
1967 unsigned int hash;
1968
1969 if (!map->tablesize)
1970 /* It's initialized on demand in register_ref_store(). */
1971 return NULL;
1972
1973 hash = strhash(name);
1974 entry = hashmap_get_entry_from_hash(map, hash, name,
1975 struct ref_store_hash_entry, ent);
1976 return entry ? entry->refs : NULL;
1977 }
1978
1979 /*
1980 * Create, record, and return a ref_store instance for the specified
1981 * gitdir.
1982 */
1983 static struct ref_store *ref_store_init(struct repository *repo,
1984 const char *gitdir,
1985 unsigned int flags)
1986 {
1987 const char *be_name = "files";
1988 struct ref_storage_be *be = find_ref_storage_backend(be_name);
1989 struct ref_store *refs;
1990
1991 if (!be)
1992 BUG("reference backend %s is unknown", be_name);
1993
1994 refs = be->init(repo, gitdir, flags);
1995 return refs;
1996 }
1997
1998 struct ref_store *get_main_ref_store(struct repository *r)
1999 {
2000 if (r->refs_private)
2001 return r->refs_private;
2002
2003 if (!r->gitdir)
2004 BUG("attempting to get main_ref_store outside of repository");
2005
2006 r->refs_private = ref_store_init(r, r->gitdir, REF_STORE_ALL_CAPS);
2007 r->refs_private = maybe_debug_wrap_ref_store(r->gitdir, r->refs_private);
2008 return r->refs_private;
2009 }
2010
2011 /*
2012 * Associate a ref store with a name. It is a fatal error to call this
2013 * function twice for the same name.
2014 */
2015 static void register_ref_store_map(struct hashmap *map,
2016 const char *type,
2017 struct ref_store *refs,
2018 const char *name)
2019 {
2020 struct ref_store_hash_entry *entry;
2021
2022 if (!map->tablesize)
2023 hashmap_init(map, ref_store_hash_cmp, NULL, 0);
2024
2025 entry = alloc_ref_store_hash_entry(name, refs);
2026 if (hashmap_put(map, &entry->ent))
2027 BUG("%s ref_store '%s' initialized twice", type, name);
2028 }
2029
2030 struct ref_store *get_submodule_ref_store(const char *submodule)
2031 {
2032 struct strbuf submodule_sb = STRBUF_INIT;
2033 struct ref_store *refs;
2034 char *to_free = NULL;
2035 size_t len;
2036 struct repository *subrepo;
2037
2038 if (!submodule)
2039 return NULL;
2040
2041 len = strlen(submodule);
2042 while (len && is_dir_sep(submodule[len - 1]))
2043 len--;
2044 if (!len)
2045 return NULL;
2046
2047 if (submodule[len])
2048 /* We need to strip off one or more trailing slashes */
2049 submodule = to_free = xmemdupz(submodule, len);
2050
2051 refs = lookup_ref_store_map(&submodule_ref_stores, submodule);
2052 if (refs)
2053 goto done;
2054
2055 strbuf_addstr(&submodule_sb, submodule);
2056 if (!is_nonbare_repository_dir(&submodule_sb))
2057 goto done;
2058
2059 if (submodule_to_gitdir(&submodule_sb, submodule))
2060 goto done;
2061
2062 subrepo = xmalloc(sizeof(*subrepo));
2063 /*
2064 * NEEDSWORK: Make get_submodule_ref_store() work with arbitrary
2065 * superprojects other than the_repository. This probably should be
2066 * done by making it take a struct repository * parameter instead of a
2067 * submodule path.
2068 */
2069 if (repo_submodule_init(subrepo, the_repository, submodule,
2070 null_oid())) {
2071 free(subrepo);
2072 goto done;
2073 }
2074 refs = ref_store_init(subrepo, submodule_sb.buf,
2075 REF_STORE_READ | REF_STORE_ODB);
2076 register_ref_store_map(&submodule_ref_stores, "submodule",
2077 refs, submodule);
2078
2079 done:
2080 strbuf_release(&submodule_sb);
2081 free(to_free);
2082
2083 return refs;
2084 }
2085
2086 struct ref_store *get_worktree_ref_store(const struct worktree *wt)
2087 {
2088 struct ref_store *refs;
2089 const char *id;
2090
2091 if (wt->is_current)
2092 return get_main_ref_store(the_repository);
2093
2094 id = wt->id ? wt->id : "/";
2095 refs = lookup_ref_store_map(&worktree_ref_stores, id);
2096 if (refs)
2097 return refs;
2098
2099 if (wt->id)
2100 refs = ref_store_init(the_repository,
2101 git_common_path("worktrees/%s", wt->id),
2102 REF_STORE_ALL_CAPS);
2103 else
2104 refs = ref_store_init(the_repository,
2105 get_git_common_dir(),
2106 REF_STORE_ALL_CAPS);
2107
2108 if (refs)
2109 register_ref_store_map(&worktree_ref_stores, "worktree",
2110 refs, id);
2111 return refs;
2112 }
2113
2114 void base_ref_store_init(struct ref_store *refs, struct repository *repo,
2115 const char *path, const struct ref_storage_be *be)
2116 {
2117 refs->be = be;
2118 refs->repo = repo;
2119 refs->gitdir = xstrdup(path);
2120 }
2121
2122 /* backend functions */
2123 int refs_pack_refs(struct ref_store *refs, unsigned int flags)
2124 {
2125 return refs->be->pack_refs(refs, flags);
2126 }
2127
2128 int peel_iterated_oid(const struct object_id *base, struct object_id *peeled)
2129 {
2130 if (current_ref_iter &&
2131 (current_ref_iter->oid == base ||
2132 oideq(current_ref_iter->oid, base)))
2133 return ref_iterator_peel(current_ref_iter, peeled);
2134
2135 return peel_object(base, peeled) ? -1 : 0;
2136 }
2137
2138 int refs_create_symref(struct ref_store *refs,
2139 const char *ref_target,
2140 const char *refs_heads_master,
2141 const char *logmsg)
2142 {
2143 char *msg;
2144 int retval;
2145
2146 msg = normalize_reflog_message(logmsg);
2147 retval = refs->be->create_symref(refs, ref_target, refs_heads_master,
2148 msg);
2149 free(msg);
2150 return retval;
2151 }
2152
2153 int create_symref(const char *ref_target, const char *refs_heads_master,
2154 const char *logmsg)
2155 {
2156 return refs_create_symref(get_main_ref_store(the_repository), ref_target,
2157 refs_heads_master, logmsg);
2158 }
2159
2160 int ref_update_reject_duplicates(struct string_list *refnames,
2161 struct strbuf *err)
2162 {
2163 size_t i, n = refnames->nr;
2164
2165 assert(err);
2166
2167 for (i = 1; i < n; i++) {
2168 int cmp = strcmp(refnames->items[i - 1].string,
2169 refnames->items[i].string);
2170
2171 if (!cmp) {
2172 strbuf_addf(err,
2173 _("multiple updates for ref '%s' not allowed"),
2174 refnames->items[i].string);
2175 return 1;
2176 } else if (cmp > 0) {
2177 BUG("ref_update_reject_duplicates() received unsorted list");
2178 }
2179 }
2180 return 0;
2181 }
2182
2183 static int run_transaction_hook(struct ref_transaction *transaction,
2184 const char *state)
2185 {
2186 struct child_process proc = CHILD_PROCESS_INIT;
2187 struct strbuf buf = STRBUF_INIT;
2188 const char *hook;
2189 int ret = 0, i;
2190
2191 hook = find_hook("reference-transaction");
2192 if (!hook)
2193 return ret;
2194
2195 strvec_pushl(&proc.args, hook, state, NULL);
2196 proc.in = -1;
2197 proc.stdout_to_stderr = 1;
2198 proc.trace2_hook_name = "reference-transaction";
2199
2200 ret = start_command(&proc);
2201 if (ret)
2202 return ret;
2203
2204 sigchain_push(SIGPIPE, SIG_IGN);
2205
2206 for (i = 0; i < transaction->nr; i++) {
2207 struct ref_update *update = transaction->updates[i];
2208
2209 strbuf_reset(&buf);
2210 strbuf_addf(&buf, "%s %s %s\n",
2211 oid_to_hex(&update->old_oid),
2212 oid_to_hex(&update->new_oid),
2213 update->refname);
2214
2215 if (write_in_full(proc.in, buf.buf, buf.len) < 0) {
2216 if (errno != EPIPE) {
2217 /* Don't leak errno outside this API */
2218 errno = 0;
2219 ret = -1;
2220 }
2221 break;
2222 }
2223 }
2224
2225 close(proc.in);
2226 sigchain_pop(SIGPIPE);
2227 strbuf_release(&buf);
2228
2229 ret |= finish_command(&proc);
2230 return ret;
2231 }
2232
2233 int ref_transaction_prepare(struct ref_transaction *transaction,
2234 struct strbuf *err)
2235 {
2236 struct ref_store *refs = transaction->ref_store;
2237 int ret;
2238
2239 switch (transaction->state) {
2240 case REF_TRANSACTION_OPEN:
2241 /* Good. */
2242 break;
2243 case REF_TRANSACTION_PREPARED:
2244 BUG("prepare called twice on reference transaction");
2245 break;
2246 case REF_TRANSACTION_CLOSED:
2247 BUG("prepare called on a closed reference transaction");
2248 break;
2249 default:
2250 BUG("unexpected reference transaction state");
2251 break;
2252 }
2253
2254 if (refs->repo->objects->odb->disable_ref_updates) {
2255 strbuf_addstr(err,
2256 _("ref updates forbidden inside quarantine environment"));
2257 return -1;
2258 }
2259
2260 ret = refs->be->transaction_prepare(refs, transaction, err);
2261 if (ret)
2262 return ret;
2263
2264 ret = run_transaction_hook(transaction, "prepared");
2265 if (ret) {
2266 ref_transaction_abort(transaction, err);
2267 die(_("ref updates aborted by hook"));
2268 }
2269
2270 return 0;
2271 }
2272
2273 int ref_transaction_abort(struct ref_transaction *transaction,
2274 struct strbuf *err)
2275 {
2276 struct ref_store *refs = transaction->ref_store;
2277 int ret = 0;
2278
2279 switch (transaction->state) {
2280 case REF_TRANSACTION_OPEN:
2281 /* No need to abort explicitly. */
2282 break;
2283 case REF_TRANSACTION_PREPARED:
2284 ret = refs->be->transaction_abort(refs, transaction, err);
2285 break;
2286 case REF_TRANSACTION_CLOSED:
2287 BUG("abort called on a closed reference transaction");
2288 break;
2289 default:
2290 BUG("unexpected reference transaction state");
2291 break;
2292 }
2293
2294 run_transaction_hook(transaction, "aborted");
2295
2296 ref_transaction_free(transaction);
2297 return ret;
2298 }
2299
2300 int ref_transaction_commit(struct ref_transaction *transaction,
2301 struct strbuf *err)
2302 {
2303 struct ref_store *refs = transaction->ref_store;
2304 int ret;
2305
2306 switch (transaction->state) {
2307 case REF_TRANSACTION_OPEN:
2308 /* Need to prepare first. */
2309 ret = ref_transaction_prepare(transaction, err);
2310 if (ret)
2311 return ret;
2312 break;
2313 case REF_TRANSACTION_PREPARED:
2314 /* Fall through to finish. */
2315 break;
2316 case REF_TRANSACTION_CLOSED:
2317 BUG("commit called on a closed reference transaction");
2318 break;
2319 default:
2320 BUG("unexpected reference transaction state");
2321 break;
2322 }
2323
2324 ret = refs->be->transaction_finish(refs, transaction, err);
2325 if (!ret)
2326 run_transaction_hook(transaction, "committed");
2327 return ret;
2328 }
2329
2330 int refs_verify_refname_available(struct ref_store *refs,
2331 const char *refname,
2332 const struct string_list *extras,
2333 const struct string_list *skip,
2334 struct strbuf *err)
2335 {
2336 const char *slash;
2337 const char *extra_refname;
2338 struct strbuf dirname = STRBUF_INIT;
2339 struct strbuf referent = STRBUF_INIT;
2340 struct object_id oid;
2341 unsigned int type;
2342 struct ref_iterator *iter;
2343 int ok;
2344 int ret = -1;
2345
2346 /*
2347 * For the sake of comments in this function, suppose that
2348 * refname is "refs/foo/bar".
2349 */
2350
2351 assert(err);
2352
2353 strbuf_grow(&dirname, strlen(refname) + 1);
2354 for (slash = strchr(refname, '/'); slash; slash = strchr(slash + 1, '/')) {
2355 /*
2356 * Just saying "Is a directory" when we e.g. can't
2357 * lock some multi-level ref isn't very informative,
2358 * the user won't be told *what* is a directory, so
2359 * let's not use strerror() below.
2360 */
2361 int ignore_errno;
2362 /* Expand dirname to the new prefix, not including the trailing slash: */
2363 strbuf_add(&dirname, refname + dirname.len, slash - refname - dirname.len);
2364
2365 /*
2366 * We are still at a leading dir of the refname (e.g.,
2367 * "refs/foo"; if there is a reference with that name,
2368 * it is a conflict, *unless* it is in skip.
2369 */
2370 if (skip && string_list_has_string(skip, dirname.buf))
2371 continue;
2372
2373 if (!refs_read_raw_ref(refs, dirname.buf, &oid, &referent,
2374 &type, &ignore_errno)) {
2375 strbuf_addf(err, _("'%s' exists; cannot create '%s'"),
2376 dirname.buf, refname);
2377 goto cleanup;
2378 }
2379
2380 if (extras && string_list_has_string(extras, dirname.buf)) {
2381 strbuf_addf(err, _("cannot process '%s' and '%s' at the same time"),
2382 refname, dirname.buf);
2383 goto cleanup;
2384 }
2385 }
2386
2387 /*
2388 * We are at the leaf of our refname (e.g., "refs/foo/bar").
2389 * There is no point in searching for a reference with that
2390 * name, because a refname isn't considered to conflict with
2391 * itself. But we still need to check for references whose
2392 * names are in the "refs/foo/bar/" namespace, because they
2393 * *do* conflict.
2394 */
2395 strbuf_addstr(&dirname, refname + dirname.len);
2396 strbuf_addch(&dirname, '/');
2397
2398 iter = refs_ref_iterator_begin(refs, dirname.buf, 0,
2399 DO_FOR_EACH_INCLUDE_BROKEN);
2400 while ((ok = ref_iterator_advance(iter)) == ITER_OK) {
2401 if (skip &&
2402 string_list_has_string(skip, iter->refname))
2403 continue;
2404
2405 strbuf_addf(err, _("'%s' exists; cannot create '%s'"),
2406 iter->refname, refname);
2407 ref_iterator_abort(iter);
2408 goto cleanup;
2409 }
2410
2411 if (ok != ITER_DONE)
2412 BUG("error while iterating over references");
2413
2414 extra_refname = find_descendant_ref(dirname.buf, extras, skip);
2415 if (extra_refname)
2416 strbuf_addf(err, _("cannot process '%s' and '%s' at the same time"),
2417 refname, extra_refname);
2418 else
2419 ret = 0;
2420
2421 cleanup:
2422 strbuf_release(&referent);
2423 strbuf_release(&dirname);
2424 return ret;
2425 }
2426
2427 int refs_for_each_reflog(struct ref_store *refs, each_ref_fn fn, void *cb_data)
2428 {
2429 struct ref_iterator *iter;
2430 struct do_for_each_ref_help hp = { fn, cb_data };
2431
2432 iter = refs->be->reflog_iterator_begin(refs);
2433
2434 return do_for_each_repo_ref_iterator(the_repository, iter,
2435 do_for_each_ref_helper, &hp);
2436 }
2437
2438 int for_each_reflog(each_ref_fn fn, void *cb_data)
2439 {
2440 return refs_for_each_reflog(get_main_ref_store(the_repository), fn, cb_data);
2441 }
2442
2443 int refs_for_each_reflog_ent_reverse(struct ref_store *refs,
2444 const char *refname,
2445 each_reflog_ent_fn fn,
2446 void *cb_data)
2447 {
2448 return refs->be->for_each_reflog_ent_reverse(refs, refname,
2449 fn, cb_data);
2450 }
2451
2452 int for_each_reflog_ent_reverse(const char *refname, each_reflog_ent_fn fn,
2453 void *cb_data)
2454 {
2455 return refs_for_each_reflog_ent_reverse(get_main_ref_store(the_repository),
2456 refname, fn, cb_data);
2457 }
2458
2459 int refs_for_each_reflog_ent(struct ref_store *refs, const char *refname,
2460 each_reflog_ent_fn fn, void *cb_data)
2461 {
2462 return refs->be->for_each_reflog_ent(refs, refname, fn, cb_data);
2463 }
2464
2465 int for_each_reflog_ent(const char *refname, each_reflog_ent_fn fn,
2466 void *cb_data)
2467 {
2468 return refs_for_each_reflog_ent(get_main_ref_store(the_repository), refname,
2469 fn, cb_data);
2470 }
2471
2472 int refs_reflog_exists(struct ref_store *refs, const char *refname)
2473 {
2474 return refs->be->reflog_exists(refs, refname);
2475 }
2476
2477 int reflog_exists(const char *refname)
2478 {
2479 return refs_reflog_exists(get_main_ref_store(the_repository), refname);
2480 }
2481
2482 int refs_create_reflog(struct ref_store *refs, const char *refname,
2483 struct strbuf *err)
2484 {
2485 return refs->be->create_reflog(refs, refname, err);
2486 }
2487
2488 int safe_create_reflog(const char *refname, struct strbuf *err)
2489 {
2490 return refs_create_reflog(get_main_ref_store(the_repository), refname,
2491 err);
2492 }
2493
2494 int refs_delete_reflog(struct ref_store *refs, const char *refname)
2495 {
2496 return refs->be->delete_reflog(refs, refname);
2497 }
2498
2499 int delete_reflog(const char *refname)
2500 {
2501 return refs_delete_reflog(get_main_ref_store(the_repository), refname);
2502 }
2503
2504 int refs_reflog_expire(struct ref_store *refs,
2505 const char *refname,
2506 unsigned int flags,
2507 reflog_expiry_prepare_fn prepare_fn,
2508 reflog_expiry_should_prune_fn should_prune_fn,
2509 reflog_expiry_cleanup_fn cleanup_fn,
2510 void *policy_cb_data)
2511 {
2512 return refs->be->reflog_expire(refs, refname, flags,
2513 prepare_fn, should_prune_fn,
2514 cleanup_fn, policy_cb_data);
2515 }
2516
2517 int reflog_expire(const char *refname,
2518 unsigned int flags,
2519 reflog_expiry_prepare_fn prepare_fn,
2520 reflog_expiry_should_prune_fn should_prune_fn,
2521 reflog_expiry_cleanup_fn cleanup_fn,
2522 void *policy_cb_data)
2523 {
2524 return refs_reflog_expire(get_main_ref_store(the_repository),
2525 refname, flags,
2526 prepare_fn, should_prune_fn,
2527 cleanup_fn, policy_cb_data);
2528 }
2529
2530 int initial_ref_transaction_commit(struct ref_transaction *transaction,
2531 struct strbuf *err)
2532 {
2533 struct ref_store *refs = transaction->ref_store;
2534
2535 return refs->be->initial_transaction_commit(refs, transaction, err);
2536 }
2537
2538 void ref_transaction_for_each_queued_update(struct ref_transaction *transaction,
2539 ref_transaction_for_each_queued_update_fn cb,
2540 void *cb_data)
2541 {
2542 int i;
2543
2544 for (i = 0; i < transaction->nr; i++) {
2545 struct ref_update *update = transaction->updates[i];
2546
2547 cb(update->refname,
2548 (update->flags & REF_HAVE_OLD) ? &update->old_oid : NULL,
2549 (update->flags & REF_HAVE_NEW) ? &update->new_oid : NULL,
2550 cb_data);
2551 }
2552 }
2553
2554 int refs_delete_refs(struct ref_store *refs, const char *logmsg,
2555 struct string_list *refnames, unsigned int flags)
2556 {
2557 char *msg;
2558 int retval;
2559
2560 msg = normalize_reflog_message(logmsg);
2561 retval = refs->be->delete_refs(refs, msg, refnames, flags);
2562 free(msg);
2563 return retval;
2564 }
2565
2566 int delete_refs(const char *msg, struct string_list *refnames,
2567 unsigned int flags)
2568 {
2569 return refs_delete_refs(get_main_ref_store(the_repository), msg, refnames, flags);
2570 }
2571
2572 int refs_rename_ref(struct ref_store *refs, const char *oldref,
2573 const char *newref, const char *logmsg)
2574 {
2575 char *msg;
2576 int retval;
2577
2578 msg = normalize_reflog_message(logmsg);
2579 retval = refs->be->rename_ref(refs, oldref, newref, msg);
2580 free(msg);
2581 return retval;
2582 }
2583
2584 int rename_ref(const char *oldref, const char *newref, const char *logmsg)
2585 {
2586 return refs_rename_ref(get_main_ref_store(the_repository), oldref, newref, logmsg);
2587 }
2588
2589 int refs_copy_existing_ref(struct ref_store *refs, const char *oldref,
2590 const char *newref, const char *logmsg)
2591 {
2592 char *msg;
2593 int retval;
2594
2595 msg = normalize_reflog_message(logmsg);
2596 retval = refs->be->copy_ref(refs, oldref, newref, msg);
2597 free(msg);
2598 return retval;
2599 }
2600
2601 int copy_existing_ref(const char *oldref, const char *newref, const char *logmsg)
2602 {
2603 return refs_copy_existing_ref(get_main_ref_store(the_repository), oldref, newref, logmsg);
2604 }