lstat_cache: guard against full match of length of 'name' parameter
[git/git.git] / entry.c
CommitLineData
12dccc16 1#include "cache.h"
8e440259 2#include "blob.h"
8ca12c0d 3#include "dir.h"
12dccc16 4
81a9aa60
KB
5static void create_directories(const char *path, int path_len,
6 const struct checkout *state)
12dccc16 7{
81a9aa60
KB
8 char *buf = xmalloc(path_len + 1);
9 int len = 0;
10
11 while (len < path_len) {
12 do {
13 buf[len] = path[len];
14 len++;
15 } while (len < path_len && path[len] != '/');
16 if (len >= path_len)
17 break;
12dccc16 18 buf[len] = 0;
fa2e71c9 19
bad4a54f
KB
20 /*
21 * For 'checkout-index --prefix=<dir>', <dir> is
22 * allowed to be a symlink to an existing directory,
23 * and we set 'state->base_dir_len' below, such that
24 * we test the path components of the prefix with the
25 * stat() function instead of the lstat() function.
26 */
57199892 27 if (has_dirs_only_path(buf, len, state->base_dir_len))
fa2e71c9
JH
28 continue; /* ok, it is already a directory. */
29
30 /*
bad4a54f
KB
31 * If this mkdir() would fail, it could be that there
32 * is already a symlink or something else exists
33 * there, therefore we then try to unlink it and try
34 * one more time to create the directory.
fa2e71c9 35 */
f312de01 36 if (mkdir(buf, 0777)) {
fa2e71c9 37 if (errno == EEXIST && state->force &&
691f1a28 38 !unlink_or_warn(buf) && !mkdir(buf, 0777))
fa2e71c9 39 continue;
0721c314 40 die_errno("cannot create directory at '%s'", buf);
12dccc16
LT
41 }
42 }
43 free(buf);
44}
45
46static void remove_subtree(const char *path)
47{
48 DIR *dir = opendir(path);
49 struct dirent *de;
50 char pathbuf[PATH_MAX];
51 char *name;
a6080a0a 52
12dccc16 53 if (!dir)
d824cbba 54 die_errno("cannot opendir '%s'", path);
12dccc16
LT
55 strcpy(pathbuf, path);
56 name = pathbuf + strlen(path);
57 *name++ = '/';
58 while ((de = readdir(dir)) != NULL) {
59 struct stat st;
8ca12c0d 60 if (is_dot_or_dotdot(de->d_name))
12dccc16
LT
61 continue;
62 strcpy(name, de->d_name);
63 if (lstat(pathbuf, &st))
d824cbba 64 die_errno("cannot lstat '%s'", pathbuf);
12dccc16
LT
65 if (S_ISDIR(st.st_mode))
66 remove_subtree(pathbuf);
67 else if (unlink(pathbuf))
d824cbba 68 die_errno("cannot unlink '%s'", pathbuf);
12dccc16
LT
69 }
70 closedir(dir);
71 if (rmdir(path))
d824cbba 72 die_errno("cannot rmdir '%s'", path);
12dccc16
LT
73}
74
d48a72f3 75static int create_file(const char *path, unsigned int mode)
12dccc16 76{
12dccc16 77 mode = (mode & 0100) ? 0777 : 0666;
781411ed 78 return open(path, O_WRONLY | O_CREAT | O_EXCL, mode);
12dccc16
LT
79}
80
4857c761 81static void *read_blob_entry(struct cache_entry *ce, unsigned long *size)
f0807e62
LT
82{
83 enum object_type type;
84 void *new = read_sha1_file(ce->sha1, &type, size);
85
86 if (new) {
87 if (type == OBJ_BLOB)
88 return new;
89 free(new);
90 }
91 return NULL;
92}
93
efbc5831 94static int write_entry(struct cache_entry *ce, char *path, const struct checkout *state, int to_tempfile)
12dccc16 95{
4857c761 96 unsigned int ce_mode_s_ifmt = ce->ce_mode & S_IFMT;
e4c72923 97 int fd, ret, fstat_done = 0;
4857c761
KB
98 char *new;
99 struct strbuf buf = STRBUF_INIT;
100 unsigned long size;
101 size_t wrote, newsize = 0;
e4c72923 102 struct stat st;
4857c761
KB
103
104 switch (ce_mode_s_ifmt) {
12dccc16 105 case S_IFREG:
4857c761
KB
106 case S_IFLNK:
107 new = read_blob_entry(ce, &size);
f0807e62 108 if (!new)
7e44c935 109 return error("git checkout-index: unable to read sha1 file of %s (%s)",
f0807e62 110 path, sha1_to_hex(ce->sha1));
1a9d7e9b 111
4857c761
KB
112 if (ce_mode_s_ifmt == S_IFLNK && has_symlinks && !to_tempfile) {
113 ret = symlink(new, path);
114 free(new);
115 if (ret)
116 return error("git checkout-index: unable to create symlink %s (%s)",
117 path, strerror(errno));
118 break;
119 }
120
1a9d7e9b
JH
121 /*
122 * Convert from git internal format to working tree format
123 */
4857c761
KB
124 if (ce_mode_s_ifmt == S_IFREG &&
125 convert_to_working_tree(ce->name, new, size, &buf)) {
1a9d7e9b 126 free(new);
c32f749f
RS
127 new = strbuf_detach(&buf, &newsize);
128 size = newsize;
1a9d7e9b
JH
129 }
130
de84f99c 131 if (to_tempfile) {
4857c761
KB
132 if (ce_mode_s_ifmt == S_IFREG)
133 strcpy(path, ".merge_file_XXXXXX");
134 else
135 strcpy(path, ".merge_link_XXXXXX");
de84f99c 136 fd = mkstemp(path);
4857c761 137 } else if (ce_mode_s_ifmt == S_IFREG) {
7a51ed66 138 fd = create_file(path, ce->ce_mode);
4857c761
KB
139 } else {
140 fd = create_file(path, 0666);
141 }
12dccc16
LT
142 if (fd < 0) {
143 free(new);
7e44c935 144 return error("git checkout-index: unable to create file %s (%s)",
12dccc16
LT
145 path, strerror(errno));
146 }
6c510bee 147
93822c22 148 wrote = write_in_full(fd, new, size);
e4c72923 149 /* use fstat() only when path == ce->name */
34779c53
JS
150 if (fstat_is_reliable() &&
151 state->refresh_cache && !to_tempfile && !state->base_dir_len) {
e4c72923
KB
152 fstat(fd, &st);
153 fstat_done = 1;
154 }
12dccc16
LT
155 close(fd);
156 free(new);
157 if (wrote != size)
7e44c935 158 return error("git checkout-index: unable to write file %s", path);
12dccc16 159 break;
302b9282 160 case S_IFGITLINK:
f0807e62 161 if (to_tempfile)
7e44c935 162 return error("git checkout-index: cannot create temporary subproject %s", path);
f0807e62 163 if (mkdir(path, 0777) < 0)
7e44c935 164 return error("git checkout-index: cannot create subproject directory %s", path);
f0807e62 165 break;
12dccc16 166 default:
7e44c935 167 return error("git checkout-index: unknown file mode for %s", path);
12dccc16
LT
168 }
169
6ee67f26 170 if (state->refresh_cache) {
e4c72923
KB
171 if (!fstat_done)
172 lstat(ce->name, &st);
12dccc16
LT
173 fill_stat_cache_info(ce, &st);
174 }
175 return 0;
176}
177
efbc5831 178int checkout_entry(struct cache_entry *ce, const struct checkout *state, char *topath)
12dccc16 179{
095c424d 180 static char path[PATH_MAX + 1];
de84f99c 181 struct stat st;
12dccc16
LT
182 int len = state->base_dir_len;
183
de84f99c
SP
184 if (topath)
185 return write_entry(ce, topath, state, 1);
186
12dccc16
LT
187 memcpy(path, state->base_dir, len);
188 strcpy(path + len, ce->name);
81a9aa60 189 len += ce_namelen(ce);
12dccc16
LT
190
191 if (!lstat(path, &st)) {
4bd5b7da 192 unsigned changed = ce_match_stat(ce, &st, CE_MATCH_IGNORE_VALID);
12dccc16
LT
193 if (!changed)
194 return 0;
195 if (!state->force) {
196 if (!state->quiet)
215a7ad1 197 fprintf(stderr, "git-checkout-index: %s already exists\n", path);
4b12dae6 198 return -1;
12dccc16
LT
199 }
200
201 /*
202 * We unlink the old file, to get the new one with the
203 * right permissions (including umask, which is nasty
204 * to emulate by hand - much easier to let the system
205 * just do the right thing)
206 */
d48a72f3 207 if (S_ISDIR(st.st_mode)) {
f0807e62 208 /* If it is a gitlink, leave it alone! */
7a51ed66 209 if (S_ISGITLINK(ce->ce_mode))
f0807e62 210 return 0;
d48a72f3
LT
211 if (!state->force)
212 return error("%s is a directory", path);
213 remove_subtree(path);
971f229c
LT
214 } else if (unlink(path))
215 return error("unable to unlink old '%s' (%s)", path, strerror(errno));
de84f99c 216 } else if (state->not_new)
12dccc16 217 return 0;
81a9aa60 218 create_directories(path, len, state);
de84f99c 219 return write_entry(ce, path, state, 0);
12dccc16 220}