avoid NULL dereference on failed malloc
[git/mjg.git] / builtin-mv.c
blob8b81d4b51d78bb2aa49430102ab585afce9cccdf
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 "string-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 const char *add_slash(const char *path)
41 int len = strlen(path);
42 if (path[len - 1] != '/') {
43 char *with_slash = xmalloc(len + 2);
44 memcpy(with_slash, path, len);
45 with_slash[len++] = '/';
46 with_slash[len] = 0;
47 return with_slash;
49 return path;
52 static struct lock_file lock_file;
54 int cmd_mv(int argc, const char **argv, const char *prefix)
56 int i, newfd;
57 int verbose = 0, show_only = 0, force = 0, ignore_errors = 0;
58 struct option builtin_mv_options[] = {
59 OPT__DRY_RUN(&show_only),
60 OPT_BOOLEAN('f', NULL, &force, "force move/rename even if target exists"),
61 OPT_BOOLEAN('k', NULL, &ignore_errors, "skip move/rename errors"),
62 OPT_END(),
64 const char **source, **destination, **dest_path;
65 enum update_mode { BOTH = 0, WORKING_DIRECTORY, INDEX } *modes;
66 struct stat st;
67 struct string_list src_for_dst = {NULL, 0, 0, 0};
69 git_config(git_default_config, NULL);
71 newfd = hold_locked_index(&lock_file, 1);
72 if (read_cache() < 0)
73 die("index file corrupt");
75 argc = parse_options(argc, argv, prefix, builtin_mv_options,
76 builtin_mv_usage, 0);
77 if (--argc < 1)
78 usage_with_options(builtin_mv_usage, builtin_mv_options);
80 source = copy_pathspec(prefix, argv, argc, 0);
81 modes = xcalloc(argc, sizeof(enum update_mode));
82 dest_path = copy_pathspec(prefix, argv + argc, 1, 0);
84 if (dest_path[0][0] == '\0')
85 /* special case: "." was normalized to "" */
86 destination = copy_pathspec(dest_path[0], argv, argc, 1);
87 else if (!lstat(dest_path[0], &st) &&
88 S_ISDIR(st.st_mode)) {
89 dest_path[0] = add_slash(dest_path[0]);
90 destination = copy_pathspec(dest_path[0], argv, argc, 1);
91 } else {
92 if (argc != 1)
93 usage_with_options(builtin_mv_usage, builtin_mv_options);
94 destination = dest_path;
97 /* Checking */
98 for (i = 0; i < argc; i++) {
99 const char *src = source[i], *dst = destination[i];
100 int length, src_is_dir;
101 const char *bad = NULL;
103 if (show_only)
104 printf("Checking rename of '%s' to '%s'\n", src, dst);
106 length = strlen(src);
107 if (lstat(src, &st) < 0)
108 bad = "bad source";
109 else if (!strncmp(src, dst, length) &&
110 (dst[length] == 0 || dst[length] == '/')) {
111 bad = "can not move directory into itself";
112 } else if ((src_is_dir = S_ISDIR(st.st_mode))
113 && lstat(dst, &st) == 0)
114 bad = "cannot move directory over file";
115 else if (src_is_dir) {
116 const char *src_w_slash = add_slash(src);
117 int len_w_slash = length + 1;
118 int first, last;
120 modes[i] = WORKING_DIRECTORY;
122 first = cache_name_pos(src_w_slash, len_w_slash);
123 if (first >= 0)
124 die ("Huh? %.*s is in index?",
125 len_w_slash, src_w_slash);
127 first = -1 - first;
128 for (last = first; last < active_nr; last++) {
129 const char *path = active_cache[last]->name;
130 if (strncmp(path, src_w_slash, len_w_slash))
131 break;
133 free((char *)src_w_slash);
135 if (last - first < 1)
136 bad = "source directory is empty";
137 else {
138 int j, dst_len;
140 if (last - first > 0) {
141 source = xrealloc(source,
142 (argc + last - first)
143 * sizeof(char *));
144 destination = xrealloc(destination,
145 (argc + last - first)
146 * sizeof(char *));
147 modes = xrealloc(modes,
148 (argc + last - first)
149 * sizeof(enum update_mode));
152 dst = add_slash(dst);
153 dst_len = strlen(dst);
155 for (j = 0; j < last - first; j++) {
156 const char *path =
157 active_cache[first + j]->name;
158 source[argc + j] = path;
159 destination[argc + j] =
160 prefix_path(dst, dst_len,
161 path + length + 1);
162 modes[argc + j] = INDEX;
164 argc += last - first;
166 } else if (cache_name_pos(src, length) < 0)
167 bad = "not under version control";
168 else if (lstat(dst, &st) == 0) {
169 bad = "destination exists";
170 if (force) {
172 * only files can overwrite each other:
173 * check both source and destination
175 if (S_ISREG(st.st_mode) || S_ISLNK(st.st_mode)) {
176 fprintf(stderr, "Warning: %s;"
177 " will overwrite!\n",
178 bad);
179 bad = NULL;
180 } else
181 bad = "Cannot overwrite";
183 } else if (string_list_has_string(&src_for_dst, dst))
184 bad = "multiple sources for the same target";
185 else
186 string_list_insert(dst, &src_for_dst);
188 if (bad) {
189 if (ignore_errors) {
190 if (--argc > 0) {
191 memmove(source + i, source + i + 1,
192 (argc - i) * sizeof(char *));
193 memmove(destination + i,
194 destination + i + 1,
195 (argc - i) * sizeof(char *));
196 i--;
198 } else
199 die ("%s, source=%s, destination=%s",
200 bad, src, dst);
204 for (i = 0; i < argc; i++) {
205 const char *src = source[i], *dst = destination[i];
206 enum update_mode mode = modes[i];
207 int pos;
208 if (show_only || verbose)
209 printf("Renaming %s to %s\n", src, dst);
210 if (!show_only && mode != INDEX &&
211 rename(src, dst) < 0 && !ignore_errors)
212 die ("renaming %s failed: %s", src, strerror(errno));
214 if (mode == WORKING_DIRECTORY)
215 continue;
217 pos = cache_name_pos(src, strlen(src));
218 assert(pos >= 0);
219 if (!show_only)
220 rename_cache_entry_at(pos, dst);
223 if (active_cache_changed) {
224 if (write_cache(newfd, active_cache, active_nr) ||
225 commit_locked_index(&lock_file))
226 die("Unable to write new index file");
229 return 0;