]>
Commit | Line | Data |
---|---|---|
12dccc16 | 1 | #include "cache.h" |
8e440259 | 2 | #include "blob.h" |
8ca12c0d | 3 | #include "dir.h" |
dd8e9121 | 4 | #include "streaming.h" |
6d14eac3 | 5 | #include "submodule.h" |
12dccc16 | 6 | |
81a9aa60 KB |
7 | static void create_directories(const char *path, int path_len, |
8 | const struct checkout *state) | |
12dccc16 | 9 | { |
3733e694 | 10 | char *buf = xmallocz(path_len); |
81a9aa60 KB |
11 | int len = 0; |
12 | ||
13 | while (len < path_len) { | |
14 | do { | |
15 | buf[len] = path[len]; | |
16 | len++; | |
17 | } while (len < path_len && path[len] != '/'); | |
18 | if (len >= path_len) | |
19 | break; | |
12dccc16 | 20 | buf[len] = 0; |
fa2e71c9 | 21 | |
bad4a54f KB |
22 | /* |
23 | * For 'checkout-index --prefix=<dir>', <dir> is | |
24 | * allowed to be a symlink to an existing directory, | |
25 | * and we set 'state->base_dir_len' below, such that | |
26 | * we test the path components of the prefix with the | |
27 | * stat() function instead of the lstat() function. | |
28 | */ | |
57199892 | 29 | if (has_dirs_only_path(buf, len, state->base_dir_len)) |
fa2e71c9 JH |
30 | continue; /* ok, it is already a directory. */ |
31 | ||
32 | /* | |
bad4a54f KB |
33 | * If this mkdir() would fail, it could be that there |
34 | * is already a symlink or something else exists | |
35 | * there, therefore we then try to unlink it and try | |
36 | * one more time to create the directory. | |
fa2e71c9 | 37 | */ |
f312de01 | 38 | if (mkdir(buf, 0777)) { |
fa2e71c9 | 39 | if (errno == EEXIST && state->force && |
691f1a28 | 40 | !unlink_or_warn(buf) && !mkdir(buf, 0777)) |
fa2e71c9 | 41 | continue; |
0721c314 | 42 | die_errno("cannot create directory at '%s'", buf); |
12dccc16 LT |
43 | } |
44 | } | |
45 | free(buf); | |
46 | } | |
47 | ||
2f29e0c6 | 48 | static void remove_subtree(struct strbuf *path) |
12dccc16 | 49 | { |
2f29e0c6 | 50 | DIR *dir = opendir(path->buf); |
12dccc16 | 51 | struct dirent *de; |
2f29e0c6 | 52 | int origlen = path->len; |
a6080a0a | 53 | |
12dccc16 | 54 | if (!dir) |
2f29e0c6 | 55 | die_errno("cannot opendir '%s'", path->buf); |
12dccc16 LT |
56 | while ((de = readdir(dir)) != NULL) { |
57 | struct stat st; | |
2f29e0c6 | 58 | |
8ca12c0d | 59 | if (is_dot_or_dotdot(de->d_name)) |
12dccc16 | 60 | continue; |
2f29e0c6 MH |
61 | |
62 | strbuf_addch(path, '/'); | |
63 | strbuf_addstr(path, de->d_name); | |
64 | if (lstat(path->buf, &st)) | |
65 | die_errno("cannot lstat '%s'", path->buf); | |
12dccc16 | 66 | if (S_ISDIR(st.st_mode)) |
2f29e0c6 MH |
67 | remove_subtree(path); |
68 | else if (unlink(path->buf)) | |
69 | die_errno("cannot unlink '%s'", path->buf); | |
70 | strbuf_setlen(path, origlen); | |
12dccc16 LT |
71 | } |
72 | closedir(dir); | |
2f29e0c6 MH |
73 | if (rmdir(path->buf)) |
74 | die_errno("cannot rmdir '%s'", path->buf); | |
12dccc16 LT |
75 | } |
76 | ||
d48a72f3 | 77 | static int create_file(const char *path, unsigned int mode) |
12dccc16 | 78 | { |
12dccc16 | 79 | mode = (mode & 0100) ? 0777 : 0666; |
781411ed | 80 | return open(path, O_WRONLY | O_CREAT | O_EXCL, mode); |
12dccc16 LT |
81 | } |
82 | ||
9c5e6c80 | 83 | static void *read_blob_entry(const struct cache_entry *ce, unsigned long *size) |
f0807e62 LT |
84 | { |
85 | enum object_type type; | |
99d1a986 | 86 | void *new = read_sha1_file(ce->oid.hash, &type, size); |
f0807e62 LT |
87 | |
88 | if (new) { | |
89 | if (type == OBJ_BLOB) | |
90 | return new; | |
91 | free(new); | |
92 | } | |
93 | return NULL; | |
94 | } | |
95 | ||
9c5e6c80 | 96 | static int open_output_fd(char *path, const struct cache_entry *ce, int to_tempfile) |
fd5db55d JH |
97 | { |
98 | int symlink = (ce->ce_mode & S_IFMT) != S_IFREG; | |
99 | if (to_tempfile) { | |
330c8e26 JK |
100 | xsnprintf(path, TEMPORARY_FILENAME_LENGTH, "%s", |
101 | symlink ? ".merge_link_XXXXXX" : ".merge_file_XXXXXX"); | |
fd5db55d JH |
102 | return mkstemp(path); |
103 | } else { | |
104 | return create_file(path, !symlink ? ce->ce_mode : 0666); | |
105 | } | |
106 | } | |
107 | ||
108 | static int fstat_output(int fd, const struct checkout *state, struct stat *st) | |
109 | { | |
110 | /* use fstat() only when path == ce->name */ | |
111 | if (fstat_is_reliable() && | |
112 | state->refresh_cache && !state->base_dir_len) { | |
113 | fstat(fd, st); | |
114 | return 1; | |
115 | } | |
116 | return 0; | |
117 | } | |
118 | ||
9c5e6c80 | 119 | static int streaming_write_entry(const struct cache_entry *ce, char *path, |
b6691092 | 120 | struct stream_filter *filter, |
dd8e9121 JH |
121 | const struct checkout *state, int to_tempfile, |
122 | int *fstat_done, struct stat *statbuf) | |
123 | { | |
d9c31e14 | 124 | int result = 0; |
47a02ff2 | 125 | int fd; |
dd8e9121 JH |
126 | |
127 | fd = open_output_fd(path, ce, to_tempfile); | |
d9c31e14 JK |
128 | if (fd < 0) |
129 | return -1; | |
130 | ||
7eda0e4f | 131 | result |= stream_blob_to_fd(fd, &ce->oid, filter, 1); |
d9c31e14 JK |
132 | *fstat_done = fstat_output(fd, state, statbuf); |
133 | result |= close(fd); | |
134 | ||
135 | if (result) | |
dd8e9121 JH |
136 | unlink(path); |
137 | return result; | |
138 | } | |
139 | ||
9c5e6c80 NTND |
140 | static int write_entry(struct cache_entry *ce, |
141 | char *path, const struct checkout *state, int to_tempfile) | |
12dccc16 | 142 | { |
4857c761 | 143 | unsigned int ce_mode_s_ifmt = ce->ce_mode & S_IFMT; |
e4c72923 | 144 | int fd, ret, fstat_done = 0; |
4857c761 KB |
145 | char *new; |
146 | struct strbuf buf = STRBUF_INIT; | |
147 | unsigned long size; | |
148 | size_t wrote, newsize = 0; | |
e4c72923 | 149 | struct stat st; |
6d14eac3 | 150 | const struct submodule *sub; |
4857c761 | 151 | |
b6691092 | 152 | if (ce_mode_s_ifmt == S_IFREG) { |
99d1a986 | 153 | struct stream_filter *filter = get_stream_filter(ce->name, |
154 | ce->oid.hash); | |
b6691092 JH |
155 | if (filter && |
156 | !streaming_write_entry(ce, path, filter, | |
157 | state, to_tempfile, | |
158 | &fstat_done, &st)) | |
159 | goto finish; | |
160 | } | |
dd8e9121 | 161 | |
4857c761 | 162 | switch (ce_mode_s_ifmt) { |
12dccc16 | 163 | case S_IFREG: |
4857c761 KB |
164 | case S_IFLNK: |
165 | new = read_blob_entry(ce, &size); | |
f0807e62 | 166 | if (!new) |
d43e9073 | 167 | return error("unable to read sha1 file of %s (%s)", |
99d1a986 | 168 | path, oid_to_hex(&ce->oid)); |
1a9d7e9b | 169 | |
4857c761 KB |
170 | if (ce_mode_s_ifmt == S_IFLNK && has_symlinks && !to_tempfile) { |
171 | ret = symlink(new, path); | |
172 | free(new); | |
173 | if (ret) | |
e1ebb3c2 NTND |
174 | return error_errno("unable to create symlink %s", |
175 | path); | |
4857c761 KB |
176 | break; |
177 | } | |
178 | ||
1a9d7e9b JH |
179 | /* |
180 | * Convert from git internal format to working tree format | |
181 | */ | |
4857c761 KB |
182 | if (ce_mode_s_ifmt == S_IFREG && |
183 | convert_to_working_tree(ce->name, new, size, &buf)) { | |
1a9d7e9b | 184 | free(new); |
c32f749f RS |
185 | new = strbuf_detach(&buf, &newsize); |
186 | size = newsize; | |
1a9d7e9b JH |
187 | } |
188 | ||
fd5db55d | 189 | fd = open_output_fd(path, ce, to_tempfile); |
12dccc16 LT |
190 | if (fd < 0) { |
191 | free(new); | |
e1ebb3c2 | 192 | return error_errno("unable to create file %s", path); |
12dccc16 | 193 | } |
6c510bee | 194 | |
93822c22 | 195 | wrote = write_in_full(fd, new, size); |
fd5db55d JH |
196 | if (!to_tempfile) |
197 | fstat_done = fstat_output(fd, state, &st); | |
12dccc16 LT |
198 | close(fd); |
199 | free(new); | |
200 | if (wrote != size) | |
d43e9073 | 201 | return error("unable to write file %s", path); |
12dccc16 | 202 | break; |
302b9282 | 203 | case S_IFGITLINK: |
f0807e62 | 204 | if (to_tempfile) |
42063f95 | 205 | return error("cannot create temporary submodule %s", path); |
f0807e62 | 206 | if (mkdir(path, 0777) < 0) |
42063f95 | 207 | return error("cannot create submodule directory %s", path); |
6d14eac3 SB |
208 | sub = submodule_from_ce(ce); |
209 | if (sub) | |
210 | return submodule_move_head(ce->name, | |
211 | NULL, oid_to_hex(&ce->oid), SUBMODULE_MOVE_HEAD_FORCE); | |
f0807e62 | 212 | break; |
12dccc16 | 213 | default: |
d43e9073 | 214 | return error("unknown file mode for %s in index", path); |
12dccc16 LT |
215 | } |
216 | ||
dd8e9121 | 217 | finish: |
6ee67f26 | 218 | if (state->refresh_cache) { |
d4a2024a | 219 | assert(state->istate); |
e4c72923 KB |
220 | if (!fstat_done) |
221 | lstat(ce->name, &st); | |
12dccc16 | 222 | fill_stat_cache_info(ce, &st); |
078a58e8 | 223 | ce->ce_flags |= CE_UPDATE_IN_BASE; |
d4a2024a | 224 | state->istate->cache_changed |= CE_ENTRY_CHANGED; |
12dccc16 LT |
225 | } |
226 | return 0; | |
227 | } | |
228 | ||
b6986d8a LT |
229 | /* |
230 | * This is like 'lstat()', except it refuses to follow symlinks | |
da02ca50 | 231 | * in the path, after skipping "skiplen". |
b6986d8a | 232 | */ |
61b97df7 | 233 | static int check_path(const char *path, int len, struct stat *st, int skiplen) |
b6986d8a | 234 | { |
da02ca50 JH |
235 | const char *slash = path + len; |
236 | ||
237 | while (path < slash && *slash != '/') | |
238 | slash--; | |
239 | if (!has_dirs_only_path(path, slash - path, skiplen)) { | |
b6986d8a LT |
240 | errno = ENOENT; |
241 | return -1; | |
242 | } | |
243 | return lstat(path, st); | |
244 | } | |
245 | ||
af2a651d JH |
246 | /* |
247 | * Write the contents from ce out to the working tree. | |
248 | * | |
249 | * When topath[] is not NULL, instead of writing to the working tree | |
250 | * file named by ce, a temporary file is created by this function and | |
251 | * its name is returned in topath[], which must be able to hold at | |
252 | * least TEMPORARY_FILENAME_LENGTH bytes long. | |
253 | */ | |
9c5e6c80 NTND |
254 | int checkout_entry(struct cache_entry *ce, |
255 | const struct checkout *state, char *topath) | |
12dccc16 | 256 | { |
f63272a3 | 257 | static struct strbuf path = STRBUF_INIT; |
de84f99c | 258 | struct stat st; |
12dccc16 | 259 | |
de84f99c SP |
260 | if (topath) |
261 | return write_entry(ce, topath, state, 1); | |
262 | ||
f63272a3 MH |
263 | strbuf_reset(&path); |
264 | strbuf_add(&path, state->base_dir, state->base_dir_len); | |
265 | strbuf_add(&path, ce->name, ce_namelen(ce)); | |
12dccc16 | 266 | |
f63272a3 | 267 | if (!check_path(path.buf, path.len, &st, state->base_dir_len)) { |
6d14eac3 | 268 | const struct submodule *sub; |
56cac48c | 269 | unsigned changed = ce_match_stat(ce, &st, CE_MATCH_IGNORE_VALID|CE_MATCH_IGNORE_SKIP_WORKTREE); |
6d14eac3 SB |
270 | /* |
271 | * Needs to be checked before !changed returns early, | |
272 | * as the possibly empty directory was not changed | |
273 | */ | |
274 | sub = submodule_from_ce(ce); | |
275 | if (sub) { | |
276 | int err; | |
277 | if (!is_submodule_populated_gently(ce->name, &err)) { | |
278 | struct stat sb; | |
279 | if (lstat(ce->name, &sb)) | |
280 | die(_("could not stat file '%s'"), ce->name); | |
281 | if (!(st.st_mode & S_IFDIR)) | |
282 | unlink_or_warn(ce->name); | |
283 | ||
284 | return submodule_move_head(ce->name, | |
285 | NULL, oid_to_hex(&ce->oid), | |
286 | SUBMODULE_MOVE_HEAD_FORCE); | |
287 | } else | |
288 | return submodule_move_head(ce->name, | |
289 | "HEAD", oid_to_hex(&ce->oid), | |
290 | SUBMODULE_MOVE_HEAD_FORCE); | |
291 | } | |
292 | ||
12dccc16 LT |
293 | if (!changed) |
294 | return 0; | |
295 | if (!state->force) { | |
296 | if (!state->quiet) | |
f63272a3 MH |
297 | fprintf(stderr, |
298 | "%s already exists, no checkout\n", | |
299 | path.buf); | |
4b12dae6 | 300 | return -1; |
12dccc16 LT |
301 | } |
302 | ||
303 | /* | |
304 | * We unlink the old file, to get the new one with the | |
305 | * right permissions (including umask, which is nasty | |
306 | * to emulate by hand - much easier to let the system | |
307 | * just do the right thing) | |
308 | */ | |
d48a72f3 | 309 | if (S_ISDIR(st.st_mode)) { |
f0807e62 | 310 | /* If it is a gitlink, leave it alone! */ |
7a51ed66 | 311 | if (S_ISGITLINK(ce->ce_mode)) |
f0807e62 | 312 | return 0; |
d48a72f3 | 313 | if (!state->force) |
f63272a3 | 314 | return error("%s is a directory", path.buf); |
2f29e0c6 | 315 | remove_subtree(&path); |
f63272a3 | 316 | } else if (unlink(path.buf)) |
e1ebb3c2 | 317 | return error_errno("unable to unlink old '%s'", path.buf); |
de84f99c | 318 | } else if (state->not_new) |
12dccc16 | 319 | return 0; |
f63272a3 MH |
320 | |
321 | create_directories(path.buf, path.len, state); | |
322 | return write_entry(ce, path.buf, state, 0); | |
12dccc16 | 323 | } |