OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file | |
2 // for details. All rights reserved. Use of this source code is governed by a | |
3 // BSD-style license that can be found in the LICENSE file. | |
4 | |
5 #include "bin/directory.h" | |
6 | |
7 #include <dirent.h> | |
8 #include <errno.h> | |
9 #include <sys/param.h> | |
10 #include <sys/stat.h> | |
11 #include <unistd.h> | |
12 | |
13 #include "bin/file.h" | |
14 #include "bin/platform.h" | |
15 | |
16 | |
17 static char* SafeStrNCpy(char* dest, const char* src, size_t n) { | |
18 strncpy(dest, src, n); | |
19 dest[n - 1] = '\0'; | |
20 return dest; | |
21 } | |
22 | |
23 | |
24 // Forward declarations. | |
25 static bool ListRecursively(const char* dir_name, | |
26 bool recursive, | |
27 DirectoryListing* listing); | |
28 static bool DeleteRecursively(const char* dir_name); | |
29 | |
30 | |
31 static bool ComputeFullPath(const char* dir_name, | |
32 char* path, | |
33 int* path_length) { | |
34 char* abs_path; | |
35 do { | |
36 abs_path = realpath(dir_name, path); | |
37 } while (abs_path == NULL && errno == EINTR); | |
38 if (abs_path == NULL) { | |
39 return false; | |
40 } | |
41 *path_length = strlen(path); | |
42 size_t written = snprintf(path + *path_length, | |
43 PATH_MAX - *path_length, | |
44 "%s", | |
45 File::PathSeparator()); | |
46 if (written != strlen(File::PathSeparator())) { | |
47 return false; | |
48 } | |
49 *path_length += written; | |
50 return true; | |
51 } | |
52 | |
53 | |
54 static bool HandleDir(char* dir_name, | |
55 char* path, | |
56 int path_length, | |
57 bool recursive, | |
58 DirectoryListing *listing) { | |
59 if (strcmp(dir_name, ".") != 0 && | |
60 strcmp(dir_name, "..") != 0) { | |
61 size_t written = snprintf(path + path_length, | |
62 PATH_MAX - path_length, | |
63 "%s", | |
64 dir_name); | |
65 if (written != strlen(dir_name)) { | |
66 return false; | |
67 } | |
68 bool ok = listing->HandleDirectory(path); | |
69 if (!ok) return ok; | |
70 if (recursive) { | |
71 return ListRecursively(path, recursive, listing); | |
72 } | |
73 } | |
74 return true; | |
75 } | |
76 | |
77 | |
78 static bool HandleFile(char* file_name, | |
79 char* path, | |
80 int path_length, | |
81 DirectoryListing *listing) { | |
82 // TODO(sgjesse): Pass flags to indicate whether file responses are | |
83 // needed. | |
84 size_t written = snprintf(path + path_length, | |
85 PATH_MAX - path_length, | |
86 "%s", | |
87 file_name); | |
88 if (written != strlen(file_name)) { | |
89 return false; | |
90 } | |
91 return listing->HandleFile(path); | |
92 } | |
93 | |
94 | |
95 static void PostError(DirectoryListing *listing, | |
96 const char* dir_name) { | |
97 listing->HandleError(dir_name); | |
98 } | |
99 | |
100 | |
101 static bool ListRecursively(const char* dir_name, | |
102 bool recursive, | |
103 DirectoryListing *listing) { | |
104 DIR* dir_pointer; | |
105 do { | |
106 dir_pointer = opendir(dir_name); | |
107 } while (dir_pointer == NULL && errno == EINTR); | |
108 if (dir_pointer == NULL) { | |
109 PostError(listing, dir_name); | |
110 return false; | |
111 } | |
112 | |
113 // Compute full path for the directory currently being listed. The | |
114 // path buffer will be used to construct the current path in the | |
115 // recursive traversal. path_length does not always equal | |
116 // strlen(path) but indicates the current prefix of path that is the | |
117 // path of the current directory in the traversal. | |
118 char *path = static_cast<char*>(malloc(PATH_MAX)); | |
119 ASSERT(path != NULL); | |
120 int path_length = 0; | |
121 bool valid = ComputeFullPath(dir_name, path, &path_length); | |
122 if (!valid) { | |
123 free(path); | |
124 PostError(listing, dir_name); | |
125 return false; | |
126 } | |
127 | |
128 // Iterated the directory and post the directories and files to the | |
129 // ports. | |
130 int read = 0; | |
131 bool success = true; | |
132 dirent entry; | |
133 dirent* result; | |
134 while ((read = TEMP_FAILURE_RETRY(readdir_r(dir_pointer, | |
135 &entry, | |
136 &result))) == 0 && | |
137 result != NULL) { | |
138 switch (entry.d_type) { | |
139 case DT_DIR: | |
140 success = HandleDir(entry.d_name, | |
141 path, | |
142 path_length, | |
143 recursive, | |
144 listing) && success; | |
145 break; | |
146 case DT_REG: | |
147 success = HandleFile(entry.d_name, | |
148 path, | |
149 path_length, | |
150 listing) && success; | |
151 break; | |
152 case DT_LNK: | |
153 case DT_UNKNOWN: { | |
154 // On some file systems the entry type is not determined by | |
155 // readdir_r. For those and for links we use stat to determine | |
156 // the actual entry type. Notice that stat returns the type of | |
157 // the file pointed to. | |
158 struct stat entry_info; | |
159 size_t written = snprintf(path + path_length, | |
160 PATH_MAX - path_length, | |
161 "%s", | |
162 entry.d_name); | |
163 if (written != strlen(entry.d_name)) { | |
164 success = false; | |
165 break; | |
166 } | |
167 int stat_success = TEMP_FAILURE_RETRY(stat(path, &entry_info)); | |
168 if (stat_success == -1) { | |
169 success = false; | |
170 PostError(listing, path); | |
171 break; | |
172 } | |
173 if (S_ISDIR(entry_info.st_mode)) { | |
174 success = HandleDir(entry.d_name, | |
175 path, | |
176 path_length, | |
177 recursive, | |
178 listing) && success; | |
179 } else if (S_ISREG(entry_info.st_mode)) { | |
180 success = HandleFile(entry.d_name, | |
181 path, | |
182 path_length, | |
183 listing) && success; | |
184 } | |
185 ASSERT(!S_ISLNK(entry_info.st_mode)); | |
186 break; | |
187 } | |
188 default: | |
189 break; | |
190 } | |
191 } | |
192 | |
193 if (read != 0) { | |
194 errno = read; | |
195 success = false; | |
196 PostError(listing, dir_name); | |
197 } | |
198 | |
199 if (closedir(dir_pointer) == -1) { | |
200 success = false; | |
201 PostError(listing, dir_name); | |
202 } | |
203 free(path); | |
204 | |
205 return success; | |
206 } | |
207 | |
208 | |
209 static bool DeleteFile(char* file_name, | |
210 char* path, | |
211 int path_length) { | |
212 size_t written = snprintf(path + path_length, | |
213 PATH_MAX - path_length, | |
214 "%s", | |
215 file_name); | |
216 if (written != strlen(file_name)) { | |
217 return false; | |
218 } | |
219 return (remove(path) == 0); | |
220 } | |
221 | |
222 | |
223 static bool DeleteDir(char* dir_name, | |
224 char* path, | |
225 int path_length) { | |
226 if (strcmp(dir_name, ".") != 0 && | |
227 strcmp(dir_name, "..") != 0) { | |
228 size_t written = snprintf(path + path_length, | |
229 PATH_MAX - path_length, | |
230 "%s", | |
231 dir_name); | |
232 if (written != strlen(dir_name)) { | |
233 return false; | |
234 } | |
235 return DeleteRecursively(path); | |
236 } | |
237 return true; | |
238 } | |
239 | |
240 | |
241 static bool DeleteRecursively(const char* dir_name) { | |
242 // Do not recurse into links for deletion. Instead delete the link. | |
243 struct stat st; | |
244 if (TEMP_FAILURE_RETRY(lstat(dir_name, &st)) == -1) { | |
245 return false; | |
246 } else if (S_ISLNK(st.st_mode)) { | |
247 return (remove(dir_name) == 0); | |
248 } | |
249 | |
250 // Not a link. Attempt to open as a directory and recurse into the | |
251 // directory. | |
252 DIR* dir_pointer; | |
253 do { | |
254 dir_pointer = opendir(dir_name); | |
255 } while (dir_pointer == NULL && errno == EINTR); | |
256 | |
257 if (dir_pointer == NULL) { | |
258 return false; | |
259 } | |
260 | |
261 // Compute full path for the directory currently being deleted. The | |
262 // path buffer will be used to construct the current path in the | |
263 // recursive traversal. path_length does not always equal | |
264 // strlen(path) but indicates the current prefix of path that is the | |
265 // path of the current directory in the traversal. | |
266 char *path = static_cast<char*>(malloc(PATH_MAX)); | |
267 ASSERT(path != NULL); | |
268 int path_length = 0; | |
269 bool valid = ComputeFullPath(dir_name, path, &path_length); | |
270 if (!valid) { | |
271 free(path); | |
272 return false; | |
273 } | |
274 | |
275 // Iterate the directory and delete all files and directories. | |
276 int read = 0; | |
277 bool success = true; | |
278 dirent entry; | |
279 dirent* result; | |
280 while ((read = TEMP_FAILURE_RETRY(readdir_r(dir_pointer, | |
281 &entry, | |
282 &result))) == 0 && | |
283 result != NULL && | |
284 success) { | |
285 switch (entry.d_type) { | |
286 case DT_DIR: | |
287 success = success && DeleteDir(entry.d_name, path, path_length); | |
288 break; | |
289 case DT_REG: | |
290 case DT_LNK: | |
291 // Treat all links as files. This will delete the link which | |
292 // is what we want no matter if the link target is a file or a | |
293 // directory. | |
294 success = success && DeleteFile(entry.d_name, path, path_length); | |
295 break; | |
296 case DT_UNKNOWN: { | |
297 // On some file systems the entry type is not determined by | |
298 // readdir_r. For those we use lstat to determine the entry | |
299 // type. | |
300 struct stat entry_info; | |
301 size_t written = snprintf(path + path_length, | |
302 PATH_MAX - path_length, | |
303 "%s", | |
304 entry.d_name); | |
305 if (written != strlen(entry.d_name)) { | |
306 success = false; | |
307 break; | |
308 } | |
309 int lstat_success = TEMP_FAILURE_RETRY(lstat(path, &entry_info)); | |
310 if (lstat_success == -1) { | |
311 success = false; | |
312 break; | |
313 } | |
314 if (S_ISDIR(entry_info.st_mode)) { | |
315 success = success && DeleteDir(entry.d_name, path, path_length); | |
316 } else if (S_ISREG(entry_info.st_mode) || S_ISLNK(entry_info.st_mode)) { | |
317 // Treat links as files. This will delete the link which is | |
318 // what we want no matter if the link target is a file or a | |
319 // directory. | |
320 success = success && DeleteFile(entry.d_name, path, path_length); | |
321 } | |
322 break; | |
323 } | |
324 default: | |
325 break; | |
326 } | |
327 } | |
328 | |
329 free(path); | |
330 | |
331 if ((read != 0) || | |
332 (closedir(dir_pointer) == -1) || | |
333 (remove(dir_name) == -1)) { | |
334 return false; | |
335 } | |
336 | |
337 return success; | |
338 } | |
339 | |
340 | |
341 bool Directory::List(const char* dir_name, | |
342 bool recursive, | |
343 DirectoryListing *listing) { | |
344 bool completed = ListRecursively(dir_name, recursive, listing); | |
345 return completed; | |
346 } | |
347 | |
348 | |
349 Directory::ExistsResult Directory::Exists(const char* dir_name) { | |
350 struct stat entry_info; | |
351 int success = TEMP_FAILURE_RETRY(stat(dir_name, &entry_info)); | |
352 if (success == 0) { | |
353 if (S_ISDIR(entry_info.st_mode)) { | |
354 return EXISTS; | |
355 } else { | |
356 return DOES_NOT_EXIST; | |
357 } | |
358 } else { | |
359 if (errno == EACCES || | |
360 errno == EBADF || | |
361 errno == EFAULT || | |
362 errno == ENOMEM || | |
363 errno == EOVERFLOW) { | |
364 // Search permissions denied for one of the directories in the | |
365 // path or a low level error occured. We do not know if the | |
366 // directory exists. | |
367 return UNKNOWN; | |
368 } | |
369 ASSERT(errno == ELOOP || | |
370 errno == ENAMETOOLONG || | |
371 errno == ENOENT || | |
372 errno == ENOTDIR); | |
373 return DOES_NOT_EXIST; | |
374 } | |
375 } | |
376 | |
377 | |
378 char* Directory::Current() { | |
379 return getcwd(NULL, 0); | |
380 } | |
381 | |
382 | |
383 bool Directory::Create(const char* dir_name) { | |
384 // Create the directory with the permissions specified by the | |
385 // process umask. | |
386 return (TEMP_FAILURE_RETRY(mkdir(dir_name, 0777)) == 0); | |
387 } | |
388 | |
389 | |
390 char* Directory::CreateTemp(const char* const_template) { | |
391 // Returns a new, unused directory name, modifying the contents of | |
392 // dir_template. Creates the directory with the permissions specified | |
393 // by the process umask. | |
394 // The return value must be freed by the caller. | |
395 char* path = static_cast<char*>(malloc(PATH_MAX + 1)); | |
396 SafeStrNCpy(path, const_template, PATH_MAX + 1); | |
397 int path_length = strlen(path); | |
398 if (path_length > 0) { | |
399 if ((path)[path_length - 1] == '/') { | |
400 snprintf(path + path_length, PATH_MAX - path_length, "temp_dir_XXXXXX"); | |
401 } else { | |
402 snprintf(path + path_length, PATH_MAX - path_length, "XXXXXX"); | |
403 } | |
404 } else { | |
405 snprintf(path, PATH_MAX, "/tmp/temp_dir1_XXXXXX"); | |
406 } | |
407 char* result; | |
408 do { | |
409 result = mkdtemp(path); | |
410 } while (result == NULL && errno == EINTR); | |
411 if (result == NULL) { | |
412 free(path); | |
413 return NULL; | |
414 } | |
415 return path; | |
416 } | |
417 | |
418 | |
419 bool Directory::Delete(const char* dir_name, bool recursive) { | |
420 if (!recursive) { | |
421 return (TEMP_FAILURE_RETRY(remove(dir_name)) == 0); | |
422 } else { | |
423 return DeleteRecursively(dir_name); | |
424 } | |
425 } | |
426 | |
427 | |
428 bool Directory::Rename(const char* path, const char* new_path) { | |
429 ExistsResult exists = Exists(path); | |
430 if (exists != EXISTS) return false; | |
431 return (TEMP_FAILURE_RETRY(rename(path, new_path)) == 0); | |
432 } | |
OLD | NEW |