Code

Don't dereference a strdup-returned NULL
[git.git] / builtin-rm.c
1 /*
2  * "git rm" builtin command
3  *
4  * Copyright (C) Linus Torvalds 2006
5  */
6 #include "cache.h"
7 #include "builtin.h"
8 #include "dir.h"
9 #include "cache-tree.h"
10 #include "tree-walk.h"
12 static const char builtin_rm_usage[] =
13 "git-rm [-f] [-n] [-r] [--cached] [--ignore-unmatch] [--quiet] [--] <file>...";
15 static struct {
16         int nr, alloc;
17         const char **name;
18 } list;
20 static void add_list(const char *name)
21 {
22         if (list.nr >= list.alloc) {
23                 list.alloc = alloc_nr(list.alloc);
24                 list.name = xrealloc(list.name, list.alloc * sizeof(const char *));
25         }
26         list.name[list.nr++] = name;
27 }
29 static int remove_file(const char *name)
30 {
31         int ret;
32         char *slash;
34         ret = unlink(name);
35         if (ret && errno == ENOENT)
36                 /* The user has removed it from the filesystem by hand */
37                 ret = errno = 0;
39         if (!ret && (slash = strrchr(name, '/'))) {
40                 char *n = xstrdup(name);
41                 do {
42                         n[slash - name] = 0;
43                         name = n;
44                 } while (!rmdir(name) && (slash = strrchr(name, '/')));
45         }
46         return ret;
47 }
49 static int check_local_mod(unsigned char *head)
50 {
51         /* items in list are already sorted in the cache order,
52          * so we could do this a lot more efficiently by using
53          * tree_desc based traversal if we wanted to, but I am
54          * lazy, and who cares if removal of files is a tad
55          * slower than the theoretical maximum speed?
56          */
57         int i, no_head;
58         int errs = 0;
60         no_head = is_null_sha1(head);
61         for (i = 0; i < list.nr; i++) {
62                 struct stat st;
63                 int pos;
64                 struct cache_entry *ce;
65                 const char *name = list.name[i];
66                 unsigned char sha1[20];
67                 unsigned mode;
69                 pos = cache_name_pos(name, strlen(name));
70                 if (pos < 0)
71                         continue; /* removing unmerged entry */
72                 ce = active_cache[pos];
74                 if (lstat(ce->name, &st) < 0) {
75                         if (errno != ENOENT)
76                                 fprintf(stderr, "warning: '%s': %s",
77                                         ce->name, strerror(errno));
78                         /* It already vanished from the working tree */
79                         continue;
80                 }
81                 else if (S_ISDIR(st.st_mode)) {
82                         /* if a file was removed and it is now a
83                          * directory, that is the same as ENOENT as
84                          * far as git is concerned; we do not track
85                          * directories.
86                          */
87                         continue;
88                 }
89                 if (ce_match_stat(ce, &st, 0))
90                         errs = error("'%s' has local modifications "
91                                      "(hint: try -f)", ce->name);
92                 if (no_head
93                      || get_tree_entry(head, name, sha1, &mode)
94                      || ce->ce_mode != create_ce_mode(mode)
95                      || hashcmp(ce->sha1, sha1))
96                         errs = error("'%s' has changes staged in the index "
97                                      "(hint: try -f)", name);
98         }
99         return errs;
102 static struct lock_file lock_file;
104 int cmd_rm(int argc, const char **argv, const char *prefix)
106         int i, newfd;
107         int show_only = 0, force = 0, index_only = 0, recursive = 0, quiet = 0;
108         int ignore_unmatch = 0;
109         const char **pathspec;
110         char *seen;
112         git_config(git_default_config);
114         newfd = hold_locked_index(&lock_file, 1);
116         if (read_cache() < 0)
117                 die("index file corrupt");
119         for (i = 1 ; i < argc ; i++) {
120                 const char *arg = argv[i];
122                 if (*arg != '-')
123                         break;
124                 else if (!strcmp(arg, "--")) {
125                         i++;
126                         break;
127                 }
128                 else if (!strcmp(arg, "-n"))
129                         show_only = 1;
130                 else if (!strcmp(arg, "--cached"))
131                         index_only = 1;
132                 else if (!strcmp(arg, "-f"))
133                         force = 1;
134                 else if (!strcmp(arg, "-r"))
135                         recursive = 1;
136                 else if (!strcmp(arg, "--quiet"))
137                         quiet = 1;
138                 else if (!strcmp(arg, "--ignore-unmatch"))
139                         ignore_unmatch = 1;
140                 else
141                         usage(builtin_rm_usage);
142         }
143         if (argc <= i)
144                 usage(builtin_rm_usage);
146         pathspec = get_pathspec(prefix, argv + i);
147         seen = NULL;
148         for (i = 0; pathspec[i] ; i++)
149                 /* nothing */;
150         seen = xcalloc(i, 1);
152         for (i = 0; i < active_nr; i++) {
153                 struct cache_entry *ce = active_cache[i];
154                 if (!match_pathspec(pathspec, ce->name, ce_namelen(ce), 0, seen))
155                         continue;
156                 add_list(ce->name);
157         }
159         if (pathspec) {
160                 const char *match;
161                 int seen_any = 0;
162                 for (i = 0; (match = pathspec[i]) != NULL ; i++) {
163                         if (!seen[i]) {
164                                 if (!ignore_unmatch) {
165                                         die("pathspec '%s' did not match any files",
166                                             match);
167                                 }
168                         }
169                         else {
170                                 seen_any = 1;
171                         }
172                         if (!recursive && seen[i] == MATCHED_RECURSIVELY)
173                                 die("not removing '%s' recursively without -r",
174                                     *match ? match : ".");
175                 }
177                 if (! seen_any)
178                         exit(0);
179         }
181         /*
182          * If not forced, the file, the index and the HEAD (if exists)
183          * must match; but the file can already been removed, since
184          * this sequence is a natural "novice" way:
185          *
186          *      rm F; git rm F
187          *
188          * Further, if HEAD commit exists, "diff-index --cached" must
189          * report no changes unless forced.
190          */
191         if (!force) {
192                 unsigned char sha1[20];
193                 if (get_sha1("HEAD", sha1))
194                         hashclr(sha1);
195                 if (check_local_mod(sha1))
196                         exit(1);
197         }
199         /*
200          * First remove the names from the index: we won't commit
201          * the index unless all of them succeed.
202          */
203         for (i = 0; i < list.nr; i++) {
204                 const char *path = list.name[i];
205                 if (!quiet)
206                         printf("rm '%s'\n", path);
208                 if (remove_file_from_cache(path))
209                         die("git-rm: unable to remove %s", path);
210                 cache_tree_invalidate_path(active_cache_tree, path);
211         }
213         if (show_only)
214                 return 0;
216         /*
217          * Then, unless we used "--cached", remove the filenames from
218          * the workspace. If we fail to remove the first one, we
219          * abort the "git rm" (but once we've successfully removed
220          * any file at all, we'll go ahead and commit to it all:
221          * by then we've already committed ourselves and can't fail
222          * in the middle)
223          */
224         if (!index_only) {
225                 int removed = 0;
226                 for (i = 0; i < list.nr; i++) {
227                         const char *path = list.name[i];
228                         if (!remove_file(path)) {
229                                 removed = 1;
230                                 continue;
231                         }
232                         if (!removed)
233                                 die("git-rm: %s: %s", path, strerror(errno));
234                 }
235         }
237         if (active_cache_changed) {
238                 if (write_cache(newfd, active_cache, active_nr) ||
239                     close(newfd) || commit_locked_index(&lock_file))
240                         die("Unable to write new index file");
241         }
243         return 0;