Fix buffer overflow in git-grep
[git/trast.git] / builtin-mv.c
blob5530e11b89c2c05c95c8ab1e82999eec0fdcc4c5
1 /*
2 * "git mv" builtin command
4 * Copyright (C) 2006 Johannes Schindelin
5 */
6 #include "cache.h"
7 #include "builtin.h"
8 #include "dir.h"
9 #include "cache-tree.h"
10 #include "path-list.h"
11 #include "parse-options.h"
13 static const char * const builtin_mv_usage[] = {
14 "git-mv [options] <source>... <destination>",
15 NULL
18 static const char **copy_pathspec(const char *prefix, const char **pathspec,
19 int count, int base_name)
21 int i;
22 const char **result = xmalloc((count + 1) * sizeof(const char *));
23 memcpy(result, pathspec, count * sizeof(const char *));
24 result[count] = NULL;
25 for (i = 0; i < count; i++) {
26 int length = strlen(result[i]);
27 if (length > 0 && result[i][length - 1] == '/') {
28 result[i] = xmemdupz(result[i], length - 1);
30 if (base_name) {
31 const char *last_slash = strrchr(result[i], '/');
32 if (last_slash)
33 result[i] = last_slash + 1;
36 return get_pathspec(prefix, result);
39 static void show_list(const char *label, struct path_list *list)
41 if (list->nr > 0) {
42 int i;
43 printf("%s", label);
44 for (i = 0; i < list->nr; i++)
45 printf("%s%s", i > 0 ? ", " : "", list->items[i].path);
46 putchar('\n');
50 static const char *add_slash(const char *path)
52 int len = strlen(path);
53 if (path[len - 1] != '/') {
54 char *with_slash = xmalloc(len + 2);
55 memcpy(with_slash, path, len);
56 with_slash[len++] = '/';
57 with_slash[len] = 0;
58 return with_slash;
60 return path;
63 static struct lock_file lock_file;
65 int cmd_mv(int argc, const char **argv, const char *prefix)
67 int i, newfd;
68 int verbose = 0, show_only = 0, force = 0, ignore_errors = 0;
69 struct option builtin_mv_options[] = {
70 OPT__DRY_RUN(&show_only),
71 OPT_BOOLEAN('f', NULL, &force, "force move/rename even if target exists"),
72 OPT_BOOLEAN('k', NULL, &ignore_errors, "skip move/rename errors"),
73 OPT_END(),
75 const char **source, **destination, **dest_path;
76 enum update_mode { BOTH = 0, WORKING_DIRECTORY, INDEX } *modes;
77 struct stat st;
78 struct path_list overwritten = {NULL, 0, 0, 0};
79 struct path_list src_for_dst = {NULL, 0, 0, 0};
80 struct path_list added = {NULL, 0, 0, 0};
81 struct path_list deleted = {NULL, 0, 0, 0};
82 struct path_list changed = {NULL, 0, 0, 0};
84 git_config(git_default_config, NULL);
86 newfd = hold_locked_index(&lock_file, 1);
87 if (read_cache() < 0)
88 die("index file corrupt");
90 argc = parse_options(argc, argv, builtin_mv_options, builtin_mv_usage, 0);
91 if (--argc < 1)
92 usage_with_options(builtin_mv_usage, builtin_mv_options);
94 source = copy_pathspec(prefix, argv, argc, 0);
95 modes = xcalloc(argc, sizeof(enum update_mode));
96 dest_path = copy_pathspec(prefix, argv + argc, 1, 0);
98 if (dest_path[0][0] == '\0')
99 /* special case: "." was normalized to "" */
100 destination = copy_pathspec(dest_path[0], argv, argc, 1);
101 else if (!lstat(dest_path[0], &st) &&
102 S_ISDIR(st.st_mode)) {
103 dest_path[0] = add_slash(dest_path[0]);
104 destination = copy_pathspec(dest_path[0], argv, argc, 1);
105 } else {
106 if (argc != 1)
107 usage_with_options(builtin_mv_usage, builtin_mv_options);
108 destination = dest_path;
111 /* Checking */
112 for (i = 0; i < argc; i++) {
113 const char *src = source[i], *dst = destination[i];
114 int length, src_is_dir;
115 const char *bad = NULL;
117 if (show_only)
118 printf("Checking rename of '%s' to '%s'\n", src, dst);
120 length = strlen(src);
121 if (lstat(src, &st) < 0)
122 bad = "bad source";
123 else if (!strncmp(src, dst, length) &&
124 (dst[length] == 0 || dst[length] == '/')) {
125 bad = "can not move directory into itself";
126 } else if ((src_is_dir = S_ISDIR(st.st_mode))
127 && lstat(dst, &st) == 0)
128 bad = "cannot move directory over file";
129 else if (src_is_dir) {
130 const char *src_w_slash = add_slash(src);
131 int len_w_slash = length + 1;
132 int first, last;
134 modes[i] = WORKING_DIRECTORY;
136 first = cache_name_pos(src_w_slash, len_w_slash);
137 if (first >= 0)
138 die ("Huh? %.*s is in index?",
139 len_w_slash, src_w_slash);
141 first = -1 - first;
142 for (last = first; last < active_nr; last++) {
143 const char *path = active_cache[last]->name;
144 if (strncmp(path, src_w_slash, len_w_slash))
145 break;
147 free((char *)src_w_slash);
149 if (last - first < 1)
150 bad = "source directory is empty";
151 else {
152 int j, dst_len;
154 if (last - first > 0) {
155 source = xrealloc(source,
156 (argc + last - first)
157 * sizeof(char *));
158 destination = xrealloc(destination,
159 (argc + last - first)
160 * sizeof(char *));
161 modes = xrealloc(modes,
162 (argc + last - first)
163 * sizeof(enum update_mode));
166 dst = add_slash(dst);
167 dst_len = strlen(dst);
169 for (j = 0; j < last - first; j++) {
170 const char *path =
171 active_cache[first + j]->name;
172 source[argc + j] = path;
173 destination[argc + j] =
174 prefix_path(dst, dst_len,
175 path + length + 1);
176 modes[argc + j] = INDEX;
178 argc += last - first;
180 } else if (lstat(dst, &st) == 0) {
181 bad = "destination exists";
182 if (force) {
184 * only files can overwrite each other:
185 * check both source and destination
187 if (S_ISREG(st.st_mode)) {
188 fprintf(stderr, "Warning: %s;"
189 " will overwrite!\n",
190 bad);
191 bad = NULL;
192 path_list_insert(dst, &overwritten);
193 } else
194 bad = "Cannot overwrite";
196 } else if (cache_name_pos(src, length) < 0)
197 bad = "not under version control";
198 else if (path_list_has_path(&src_for_dst, dst))
199 bad = "multiple sources for the same target";
200 else
201 path_list_insert(dst, &src_for_dst);
203 if (bad) {
204 if (ignore_errors) {
205 if (--argc > 0) {
206 memmove(source + i, source + i + 1,
207 (argc - i) * sizeof(char *));
208 memmove(destination + i,
209 destination + i + 1,
210 (argc - i) * sizeof(char *));
212 } else
213 die ("%s, source=%s, destination=%s",
214 bad, src, dst);
218 for (i = 0; i < argc; i++) {
219 const char *src = source[i], *dst = destination[i];
220 enum update_mode mode = modes[i];
221 if (show_only || verbose)
222 printf("Renaming %s to %s\n", src, dst);
223 if (!show_only && mode != INDEX &&
224 rename(src, dst) < 0 && !ignore_errors)
225 die ("renaming %s failed: %s", src, strerror(errno));
227 if (mode == WORKING_DIRECTORY)
228 continue;
230 if (cache_name_pos(src, strlen(src)) >= 0) {
231 path_list_insert(src, &deleted);
233 /* destination can be a directory with 1 file inside */
234 if (path_list_has_path(&overwritten, dst))
235 path_list_insert(dst, &changed);
236 else
237 path_list_insert(dst, &added);
238 } else
239 path_list_insert(dst, &added);
242 if (show_only) {
243 show_list("Changed : ", &changed);
244 show_list("Adding : ", &added);
245 show_list("Deleting : ", &deleted);
246 } else {
247 for (i = 0; i < changed.nr; i++) {
248 const char *path = changed.items[i].path;
249 int j = cache_name_pos(path, strlen(path));
250 struct cache_entry *ce = active_cache[j];
252 if (j < 0)
253 die ("Huh? Cache entry for %s unknown?", path);
254 refresh_cache_entry(ce, 0);
257 for (i = 0; i < added.nr; i++) {
258 const char *path = added.items[i].path;
259 if (add_file_to_cache(path, verbose ? ADD_CACHE_VERBOSE : 0))
260 die("updating index entries failed");
263 for (i = 0; i < deleted.nr; i++)
264 remove_file_from_cache(deleted.items[i].path);
266 if (active_cache_changed) {
267 if (write_cache(newfd, active_cache, active_nr) ||
268 commit_locked_index(&lock_file))
269 die("Unable to write new index file");
273 return 0;