1f53588a11aee0debef472b6cce09c8798a3e3af
1#include "cache.h"
2#include "blob.h"
3#include "dir.h"
4
5static void create_directories(const char *path, int path_len,
6 const struct checkout *state)
7{
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;
18 buf[len] = 0;
19
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 */
27 if (has_dirs_only_path(buf, len, state->base_dir_len))
28 continue; /* ok, it is already a directory. */
29
30 /*
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.
35 */
36 if (mkdir(buf, 0777)) {
37 if (errno == EEXIST && state->force &&
38 !unlink(buf) && !mkdir(buf, 0777))
39 continue;
40 die("cannot create directory at %s", buf);
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;
52
53 if (!dir)
54 die("cannot opendir %s (%s)", path, strerror(errno));
55 strcpy(pathbuf, path);
56 name = pathbuf + strlen(path);
57 *name++ = '/';
58 while ((de = readdir(dir)) != NULL) {
59 struct stat st;
60 if (is_dot_or_dotdot(de->d_name))
61 continue;
62 strcpy(name, de->d_name);
63 if (lstat(pathbuf, &st))
64 die("cannot lstat %s (%s)", pathbuf, strerror(errno));
65 if (S_ISDIR(st.st_mode))
66 remove_subtree(pathbuf);
67 else if (unlink(pathbuf))
68 die("cannot unlink %s (%s)", pathbuf, strerror(errno));
69 }
70 closedir(dir);
71 if (rmdir(path))
72 die("cannot rmdir %s (%s)", path, strerror(errno));
73}
74
75static int create_file(const char *path, unsigned int mode)
76{
77 mode = (mode & 0100) ? 0777 : 0666;
78 return open(path, O_WRONLY | O_CREAT | O_EXCL, mode);
79}
80
81static void *read_blob_entry(struct cache_entry *ce, unsigned long *size)
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
94static int write_entry(struct cache_entry *ce, char *path, const struct checkout *state, int to_tempfile)
95{
96 unsigned int ce_mode_s_ifmt = ce->ce_mode & S_IFMT;
97 int fd, ret;
98 char *new;
99 struct strbuf buf = STRBUF_INIT;
100 unsigned long size;
101 size_t wrote, newsize = 0;
102
103 switch (ce_mode_s_ifmt) {
104 case S_IFREG:
105 case S_IFLNK:
106 new = read_blob_entry(ce, &size);
107 if (!new)
108 return error("git checkout-index: unable to read sha1 file of %s (%s)",
109 path, sha1_to_hex(ce->sha1));
110
111 if (ce_mode_s_ifmt == S_IFLNK && has_symlinks && !to_tempfile) {
112 ret = symlink(new, path);
113 free(new);
114 if (ret)
115 return error("git checkout-index: unable to create symlink %s (%s)",
116 path, strerror(errno));
117 break;
118 }
119
120 /*
121 * Convert from git internal format to working tree format
122 */
123 if (ce_mode_s_ifmt == S_IFREG &&
124 convert_to_working_tree(ce->name, new, size, &buf)) {
125 free(new);
126 new = strbuf_detach(&buf, &newsize);
127 size = newsize;
128 }
129
130 if (to_tempfile) {
131 if (ce_mode_s_ifmt == S_IFREG)
132 strcpy(path, ".merge_file_XXXXXX");
133 else
134 strcpy(path, ".merge_link_XXXXXX");
135 fd = mkstemp(path);
136 } else if (ce_mode_s_ifmt == S_IFREG) {
137 fd = create_file(path, ce->ce_mode);
138 } else {
139 fd = create_file(path, 0666);
140 }
141 if (fd < 0) {
142 free(new);
143 return error("git checkout-index: unable to create file %s (%s)",
144 path, strerror(errno));
145 }
146
147 wrote = write_in_full(fd, new, size);
148 close(fd);
149 free(new);
150 if (wrote != size)
151 return error("git checkout-index: unable to write file %s", path);
152 break;
153 case S_IFGITLINK:
154 if (to_tempfile)
155 return error("git checkout-index: cannot create temporary subproject %s", path);
156 if (mkdir(path, 0777) < 0)
157 return error("git checkout-index: cannot create subproject directory %s", path);
158 break;
159 default:
160 return error("git checkout-index: unknown file mode for %s", path);
161 }
162
163 if (state->refresh_cache) {
164 struct stat st;
165 lstat(ce->name, &st);
166 fill_stat_cache_info(ce, &st);
167 }
168 return 0;
169}
170
171int checkout_entry(struct cache_entry *ce, const struct checkout *state, char *topath)
172{
173 static char path[PATH_MAX + 1];
174 struct stat st;
175 int len = state->base_dir_len;
176
177 if (topath)
178 return write_entry(ce, topath, state, 1);
179
180 memcpy(path, state->base_dir, len);
181 strcpy(path + len, ce->name);
182 len += ce_namelen(ce);
183
184 if (!lstat(path, &st)) {
185 unsigned changed = ce_match_stat(ce, &st, CE_MATCH_IGNORE_VALID);
186 if (!changed)
187 return 0;
188 if (!state->force) {
189 if (!state->quiet)
190 fprintf(stderr, "git-checkout-index: %s already exists\n", path);
191 return -1;
192 }
193
194 /*
195 * We unlink the old file, to get the new one with the
196 * right permissions (including umask, which is nasty
197 * to emulate by hand - much easier to let the system
198 * just do the right thing)
199 */
200 if (S_ISDIR(st.st_mode)) {
201 /* If it is a gitlink, leave it alone! */
202 if (S_ISGITLINK(ce->ce_mode))
203 return 0;
204 if (!state->force)
205 return error("%s is a directory", path);
206 remove_subtree(path);
207 } else if (unlink(path))
208 return error("unable to unlink old '%s' (%s)", path, strerror(errno));
209 } else if (state->not_new)
210 return 0;
211 create_directories(path, len, state);
212 return write_entry(ce, path, state, 0);
213}