write_entry: untangle symlink and regular-file cases
[git] / entry.c
1 #include "cache.h"
2 #include "blob.h"
3 #include "dir.h"
4 #include "streaming.h"
5 #include "submodule.h"
6
7 static void create_directories(const char *path, int path_len,
8                                const struct checkout *state)
9 {
10         char *buf = xmallocz(path_len);
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;
20                 buf[len] = 0;
21
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                  */
29                 if (has_dirs_only_path(buf, len, state->base_dir_len))
30                         continue; /* ok, it is already a directory. */
31
32                 /*
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.
37                  */
38                 if (mkdir(buf, 0777)) {
39                         if (errno == EEXIST && state->force &&
40                             !unlink_or_warn(buf) && !mkdir(buf, 0777))
41                                 continue;
42                         die_errno("cannot create directory at '%s'", buf);
43                 }
44         }
45         free(buf);
46 }
47
48 static void remove_subtree(struct strbuf *path)
49 {
50         DIR *dir = opendir(path->buf);
51         struct dirent *de;
52         int origlen = path->len;
53
54         if (!dir)
55                 die_errno("cannot opendir '%s'", path->buf);
56         while ((de = readdir(dir)) != NULL) {
57                 struct stat st;
58
59                 if (is_dot_or_dotdot(de->d_name))
60                         continue;
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);
66                 if (S_ISDIR(st.st_mode))
67                         remove_subtree(path);
68                 else if (unlink(path->buf))
69                         die_errno("cannot unlink '%s'", path->buf);
70                 strbuf_setlen(path, origlen);
71         }
72         closedir(dir);
73         if (rmdir(path->buf))
74                 die_errno("cannot rmdir '%s'", path->buf);
75 }
76
77 static int create_file(const char *path, unsigned int mode)
78 {
79         mode = (mode & 0100) ? 0777 : 0666;
80         return open(path, O_WRONLY | O_CREAT | O_EXCL, mode);
81 }
82
83 static void *read_blob_entry(const struct cache_entry *ce, unsigned long *size)
84 {
85         enum object_type type;
86         void *new = read_sha1_file(ce->oid.hash, &type, size);
87
88         if (new) {
89                 if (type == OBJ_BLOB)
90                         return new;
91                 free(new);
92         }
93         return NULL;
94 }
95
96 static int open_output_fd(char *path, const struct cache_entry *ce, int to_tempfile)
97 {
98         int symlink = (ce->ce_mode & S_IFMT) != S_IFREG;
99         if (to_tempfile) {
100                 xsnprintf(path, TEMPORARY_FILENAME_LENGTH, "%s",
101                           symlink ? ".merge_link_XXXXXX" : ".merge_file_XXXXXX");
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
119 static int streaming_write_entry(const struct cache_entry *ce, char *path,
120                                  struct stream_filter *filter,
121                                  const struct checkout *state, int to_tempfile,
122                                  int *fstat_done, struct stat *statbuf)
123 {
124         int result = 0;
125         int fd;
126
127         fd = open_output_fd(path, ce, to_tempfile);
128         if (fd < 0)
129                 return -1;
130
131         result |= stream_blob_to_fd(fd, &ce->oid, filter, 1);
132         *fstat_done = fstat_output(fd, state, statbuf);
133         result |= close(fd);
134
135         if (result)
136                 unlink(path);
137         return result;
138 }
139
140 void enable_delayed_checkout(struct checkout *state)
141 {
142         if (!state->delayed_checkout) {
143                 state->delayed_checkout = xmalloc(sizeof(*state->delayed_checkout));
144                 state->delayed_checkout->state = CE_CAN_DELAY;
145                 string_list_init(&state->delayed_checkout->filters, 0);
146                 string_list_init(&state->delayed_checkout->paths, 0);
147         }
148 }
149
150 static int remove_available_paths(struct string_list_item *item, void *cb_data)
151 {
152         struct string_list *available_paths = cb_data;
153         struct string_list_item *available;
154
155         available = string_list_lookup(available_paths, item->string);
156         if (available)
157                 available->util = (void *)item->string;
158         return !available;
159 }
160
161 int finish_delayed_checkout(struct checkout *state)
162 {
163         int errs = 0;
164         struct string_list_item *filter, *path;
165         struct delayed_checkout *dco = state->delayed_checkout;
166
167         if (!state->delayed_checkout)
168                 return errs;
169
170         dco->state = CE_RETRY;
171         while (dco->filters.nr > 0) {
172                 for_each_string_list_item(filter, &dco->filters) {
173                         struct string_list available_paths = STRING_LIST_INIT_NODUP;
174
175                         if (!async_query_available_blobs(filter->string, &available_paths)) {
176                                 /* Filter reported an error */
177                                 errs = 1;
178                                 filter->string = "";
179                                 continue;
180                         }
181                         if (available_paths.nr <= 0) {
182                                 /*
183                                  * Filter responded with no entries. That means
184                                  * the filter is done and we can remove the
185                                  * filter from the list (see
186                                  * "string_list_remove_empty_items" call below).
187                                  */
188                                 filter->string = "";
189                                 continue;
190                         }
191
192                         /*
193                          * In dco->paths we store a list of all delayed paths.
194                          * The filter just send us a list of available paths.
195                          * Remove them from the list.
196                          */
197                         filter_string_list(&dco->paths, 0,
198                                 &remove_available_paths, &available_paths);
199
200                         for_each_string_list_item(path, &available_paths) {
201                                 struct cache_entry* ce;
202
203                                 if (!path->util) {
204                                         error("external filter '%s' signaled that '%s' "
205                                               "is now available although it has not been "
206                                               "delayed earlier",
207                                               filter->string, path->string);
208                                         errs |= 1;
209
210                                         /*
211                                          * Do not ask the filter for available blobs,
212                                          * again, as the filter is likely buggy.
213                                          */
214                                         filter->string = "";
215                                         continue;
216                                 }
217                                 ce = index_file_exists(state->istate, path->string,
218                                                        strlen(path->string), 0);
219                                 errs |= (ce ? checkout_entry(ce, state, NULL) : 1);
220                         }
221                 }
222                 string_list_remove_empty_items(&dco->filters, 0);
223         }
224         string_list_clear(&dco->filters, 0);
225
226         /* At this point we should not have any delayed paths anymore. */
227         errs |= dco->paths.nr;
228         for_each_string_list_item(path, &dco->paths) {
229                 error("'%s' was not filtered properly", path->string);
230         }
231         string_list_clear(&dco->paths, 0);
232
233         free(dco);
234         state->delayed_checkout = NULL;
235
236         return errs;
237 }
238
239 static int write_entry(struct cache_entry *ce,
240                        char *path, const struct checkout *state, int to_tempfile)
241 {
242         unsigned int ce_mode_s_ifmt = ce->ce_mode & S_IFMT;
243         struct delayed_checkout *dco = state->delayed_checkout;
244         int fd, ret, fstat_done = 0;
245         char *new;
246         struct strbuf buf = STRBUF_INIT;
247         unsigned long size;
248         size_t wrote, newsize = 0;
249         struct stat st;
250         const struct submodule *sub;
251
252         if (ce_mode_s_ifmt == S_IFREG) {
253                 struct stream_filter *filter = get_stream_filter(ce->name,
254                                                                  ce->oid.hash);
255                 if (filter &&
256                     !streaming_write_entry(ce, path, filter,
257                                            state, to_tempfile,
258                                            &fstat_done, &st))
259                         goto finish;
260         }
261
262         switch (ce_mode_s_ifmt) {
263         case S_IFLNK:
264                 new = read_blob_entry(ce, &size);
265                 if (!new)
266                         return error("unable to read sha1 file of %s (%s)",
267                                      path, oid_to_hex(&ce->oid));
268
269                 /*
270                  * We can't make a real symlink; write out a regular file entry
271                  * with the symlink destination as its contents.
272                  */
273                 if (!has_symlinks || to_tempfile)
274                         goto write_file_entry;
275
276                 ret = symlink(new, path);
277                 free(new);
278                 if (ret)
279                         return error_errno("unable to create symlink %s", path);
280                 break;
281
282         case S_IFREG:
283                 /*
284                  * We do not send the blob in case of a retry, so do not
285                  * bother reading it at all.
286                  */
287                 if (dco && dco->state == CE_RETRY) {
288                         new = NULL;
289                         size = 0;
290                 } else {
291                         new = read_blob_entry(ce, &size);
292                         if (!new)
293                                 return error("unable to read sha1 file of %s (%s)",
294                                              path, oid_to_hex(&ce->oid));
295                 }
296
297                 /*
298                  * Convert from git internal format to working tree format
299                  */
300                 if (dco && dco->state != CE_NO_DELAY) {
301                         ret = async_convert_to_working_tree(ce->name, new,
302                                                             size, &buf, dco);
303                         if (ret && string_list_has_string(&dco->paths, ce->name)) {
304                                 free(new);
305                                 goto delayed;
306                         }
307                 } else
308                         ret = convert_to_working_tree(ce->name, new, size, &buf);
309
310                 if (ret) {
311                         free(new);
312                         new = strbuf_detach(&buf, &newsize);
313                         size = newsize;
314                 }
315                 /*
316                  * No "else" here as errors from convert are OK at this
317                  * point. If the error would have been fatal (e.g.
318                  * filter is required), then we would have died already.
319                  */
320
321         write_file_entry:
322                 fd = open_output_fd(path, ce, to_tempfile);
323                 if (fd < 0) {
324                         free(new);
325                         return error_errno("unable to create file %s", path);
326                 }
327
328                 wrote = write_in_full(fd, new, size);
329                 if (!to_tempfile)
330                         fstat_done = fstat_output(fd, state, &st);
331                 close(fd);
332                 free(new);
333                 if (wrote != size)
334                         return error("unable to write file %s", path);
335                 break;
336
337         case S_IFGITLINK:
338                 if (to_tempfile)
339                         return error("cannot create temporary submodule %s", path);
340                 if (mkdir(path, 0777) < 0)
341                         return error("cannot create submodule directory %s", path);
342                 sub = submodule_from_ce(ce);
343                 if (sub)
344                         return submodule_move_head(ce->name,
345                                 NULL, oid_to_hex(&ce->oid),
346                                 state->force ? SUBMODULE_MOVE_HEAD_FORCE : 0);
347                 break;
348
349         default:
350                 return error("unknown file mode for %s in index", path);
351         }
352
353 finish:
354         if (state->refresh_cache) {
355                 assert(state->istate);
356                 if (!fstat_done)
357                         if (lstat(ce->name, &st) < 0)
358                                 return error_errno("unable to stat just-written file %s",
359                                                    ce->name);
360                 fill_stat_cache_info(ce, &st);
361                 ce->ce_flags |= CE_UPDATE_IN_BASE;
362                 state->istate->cache_changed |= CE_ENTRY_CHANGED;
363         }
364 delayed:
365         return 0;
366 }
367
368 /*
369  * This is like 'lstat()', except it refuses to follow symlinks
370  * in the path, after skipping "skiplen".
371  */
372 static int check_path(const char *path, int len, struct stat *st, int skiplen)
373 {
374         const char *slash = path + len;
375
376         while (path < slash && *slash != '/')
377                 slash--;
378         if (!has_dirs_only_path(path, slash - path, skiplen)) {
379                 errno = ENOENT;
380                 return -1;
381         }
382         return lstat(path, st);
383 }
384
385 /*
386  * Write the contents from ce out to the working tree.
387  *
388  * When topath[] is not NULL, instead of writing to the working tree
389  * file named by ce, a temporary file is created by this function and
390  * its name is returned in topath[], which must be able to hold at
391  * least TEMPORARY_FILENAME_LENGTH bytes long.
392  */
393 int checkout_entry(struct cache_entry *ce,
394                    const struct checkout *state, char *topath)
395 {
396         static struct strbuf path = STRBUF_INIT;
397         struct stat st;
398
399         if (topath)
400                 return write_entry(ce, topath, state, 1);
401
402         strbuf_reset(&path);
403         strbuf_add(&path, state->base_dir, state->base_dir_len);
404         strbuf_add(&path, ce->name, ce_namelen(ce));
405
406         if (!check_path(path.buf, path.len, &st, state->base_dir_len)) {
407                 const struct submodule *sub;
408                 unsigned changed = ce_match_stat(ce, &st, CE_MATCH_IGNORE_VALID|CE_MATCH_IGNORE_SKIP_WORKTREE);
409                 /*
410                  * Needs to be checked before !changed returns early,
411                  * as the possibly empty directory was not changed
412                  */
413                 sub = submodule_from_ce(ce);
414                 if (sub) {
415                         int err;
416                         if (!is_submodule_populated_gently(ce->name, &err)) {
417                                 struct stat sb;
418                                 if (lstat(ce->name, &sb))
419                                         die(_("could not stat file '%s'"), ce->name);
420                                 if (!(st.st_mode & S_IFDIR))
421                                         unlink_or_warn(ce->name);
422
423                                 return submodule_move_head(ce->name,
424                                         NULL, oid_to_hex(&ce->oid), 0);
425                         } else
426                                 return submodule_move_head(ce->name,
427                                         "HEAD", oid_to_hex(&ce->oid),
428                                         state->force ? SUBMODULE_MOVE_HEAD_FORCE : 0);
429                 }
430
431                 if (!changed)
432                         return 0;
433                 if (!state->force) {
434                         if (!state->quiet)
435                                 fprintf(stderr,
436                                         "%s already exists, no checkout\n",
437                                         path.buf);
438                         return -1;
439                 }
440
441                 /*
442                  * We unlink the old file, to get the new one with the
443                  * right permissions (including umask, which is nasty
444                  * to emulate by hand - much easier to let the system
445                  * just do the right thing)
446                  */
447                 if (S_ISDIR(st.st_mode)) {
448                         /* If it is a gitlink, leave it alone! */
449                         if (S_ISGITLINK(ce->ce_mode))
450                                 return 0;
451                         if (!state->force)
452                                 return error("%s is a directory", path.buf);
453                         remove_subtree(&path);
454                 } else if (unlink(path.buf))
455                         return error_errno("unable to unlink old '%s'", path.buf);
456         } else if (state->not_new)
457                 return 0;
458
459         create_directories(path.buf, path.len, state);
460         return write_entry(ce, path.buf, state, 0);
461 }