Merge branch 'ao/check-resolve-ref-unsafe-result' into maint
[git] / builtin / check-ref-format.c
1 /*
2  * GIT - The information manager from hell
3  */
4
5 #include "cache.h"
6 #include "refs.h"
7 #include "builtin.h"
8 #include "strbuf.h"
9
10 static const char builtin_check_ref_format_usage[] =
11 "git check-ref-format [--normalize] [<options>] <refname>\n"
12 "   or: git check-ref-format --branch <branchname-shorthand>";
13
14 /*
15  * Return a copy of refname but with leading slashes removed and runs
16  * of adjacent slashes replaced with single slashes.
17  *
18  * This function is similar to normalize_path_copy(), but stripped down
19  * to meet check_ref_format's simpler needs.
20  */
21 static char *collapse_slashes(const char *refname)
22 {
23         char *ret = xmallocz(strlen(refname));
24         char ch;
25         char prev = '/';
26         char *cp = ret;
27
28         while ((ch = *refname++) != '\0') {
29                 if (prev == '/' && ch == prev)
30                         continue;
31
32                 *cp++ = ch;
33                 prev = ch;
34         }
35         *cp = '\0';
36         return ret;
37 }
38
39 static int check_ref_format_branch(const char *arg)
40 {
41         struct strbuf sb = STRBUF_INIT;
42         int nongit;
43
44         setup_git_directory_gently(&nongit);
45         if (strbuf_check_branch_ref(&sb, arg))
46                 die("'%s' is not a valid branch name", arg);
47         printf("%s\n", sb.buf + 11);
48         strbuf_release(&sb);
49         return 0;
50 }
51
52 int cmd_check_ref_format(int argc, const char **argv, const char *prefix)
53 {
54         int i;
55         int normalize = 0;
56         int flags = 0;
57         const char *refname;
58
59         if (argc == 2 && !strcmp(argv[1], "-h"))
60                 usage(builtin_check_ref_format_usage);
61
62         if (argc == 3 && !strcmp(argv[1], "--branch"))
63                 return check_ref_format_branch(argv[2]);
64
65         for (i = 1; i < argc && argv[i][0] == '-'; i++) {
66                 if (!strcmp(argv[i], "--normalize") || !strcmp(argv[i], "--print"))
67                         normalize = 1;
68                 else if (!strcmp(argv[i], "--allow-onelevel"))
69                         flags |= REFNAME_ALLOW_ONELEVEL;
70                 else if (!strcmp(argv[i], "--no-allow-onelevel"))
71                         flags &= ~REFNAME_ALLOW_ONELEVEL;
72                 else if (!strcmp(argv[i], "--refspec-pattern"))
73                         flags |= REFNAME_REFSPEC_PATTERN;
74                 else
75                         usage(builtin_check_ref_format_usage);
76         }
77         if (! (i == argc - 1))
78                 usage(builtin_check_ref_format_usage);
79
80         refname = argv[i];
81         if (normalize)
82                 refname = collapse_slashes(refname);
83         if (check_refname_format(refname, flags))
84                 return 1;
85         if (normalize)
86                 printf("%s\n", refname);
87
88         return 0;
89 }