| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011, 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 <dirent.h> | |
| 6 #include <libgen.h> | |
| 7 #include <string.h> | |
| 8 #include <sys/param.h> | |
| 9 #include <sys/stat.h> | |
| 10 #include <sys/types.h> | |
| 11 #include <unistd.h> | |
| 12 | |
| 13 #include "bin/dartutils.h" | |
| 14 #include "bin/directory.h" | |
| 15 #include "bin/file.h" | |
| 16 | |
| 17 // Forward declaration. | |
| 18 static bool ListRecursively(const char* dir_name, | |
| 19 bool recursive, | |
| 20 Dart_Port dir_port, | |
| 21 Dart_Port file_port, | |
| 22 Dart_Port done_port, | |
| 23 Dart_Port error_port); | |
| 24 | |
| 25 | |
| 26 static void ComputeFullPath(const char* dir_name, | |
| 27 char* path, | |
| 28 int* path_length) { | |
| 29 size_t written = 0; | |
| 30 | |
| 31 if (!File::IsAbsolutePath(dir_name)) { | |
| 32 ASSERT(getcwd(path, PATH_MAX) != NULL); | |
| 33 *path_length = strlen(path); | |
| 34 written = snprintf(path + *path_length, | |
| 35 PATH_MAX - *path_length, | |
| 36 "%s", | |
| 37 File::PathSeparator()); | |
| 38 ASSERT(written == strlen(File::PathSeparator())); | |
| 39 *path_length += written; | |
| 40 } | |
| 41 | |
| 42 // Use dirname and basename to canonicalize the provided directory | |
| 43 // name. | |
| 44 char* dir_name_copy = strdup(dir_name); | |
| 45 char* dir = dirname(dir_name_copy); | |
| 46 if (strcmp(dir, ".") != 0) { | |
| 47 written = snprintf(path + *path_length, | |
| 48 PATH_MAX - *path_length, | |
| 49 "%s%s", | |
| 50 dir, | |
| 51 File::PathSeparator()); | |
| 52 ASSERT(written == (strlen(dir) + strlen(File::PathSeparator()))); | |
| 53 *path_length += written; | |
| 54 } | |
| 55 char* base_name_copy = strdup(dir_name); | |
| 56 char* base = basename(base_name_copy); | |
| 57 if (strcmp(base, ".") != 0) { | |
| 58 written = snprintf(path + *path_length, | |
| 59 PATH_MAX - *path_length, | |
| 60 "%s%s", | |
| 61 base, | |
| 62 File::PathSeparator()); | |
| 63 ASSERT(written == (strlen(base) + strlen(File::PathSeparator()))); | |
| 64 *path_length += written; | |
| 65 } | |
| 66 | |
| 67 free(dir_name_copy); | |
| 68 free(base_name_copy); | |
| 69 } | |
| 70 | |
| 71 | |
| 72 static bool HandleDir(char* dir_name, | |
| 73 char* path, | |
| 74 int path_length, | |
| 75 bool recursive, | |
| 76 Dart_Port dir_port, | |
| 77 Dart_Port file_port, | |
| 78 Dart_Port done_port, | |
| 79 Dart_Port error_port) { | |
| 80 if (strcmp(dir_name, ".") != 0 && | |
| 81 strcmp(dir_name, "..") != 0) { | |
| 82 size_t written = snprintf(path + path_length, | |
| 83 PATH_MAX - path_length, | |
| 84 "%s", | |
| 85 dir_name); | |
| 86 ASSERT(written == strlen(dir_name)); | |
| 87 if (dir_port != 0) { | |
| 88 Dart_Handle name = Dart_NewString(path); | |
| 89 Dart_Post(dir_port, name); | |
| 90 } | |
| 91 if (recursive) { | |
| 92 return ListRecursively(path, | |
| 93 recursive, | |
| 94 dir_port, | |
| 95 file_port, | |
| 96 done_port, | |
| 97 error_port); | |
| 98 } | |
| 99 } | |
| 100 return true; | |
| 101 } | |
| 102 | |
| 103 | |
| 104 static void HandleFile(char* file_name, | |
| 105 char* path, | |
| 106 int path_length, | |
| 107 Dart_Port file_port) { | |
| 108 if (file_port != 0) { | |
| 109 size_t written = snprintf(path + path_length, | |
| 110 PATH_MAX - path_length, | |
| 111 "%s", | |
| 112 file_name); | |
| 113 ASSERT(written == strlen(file_name)); | |
| 114 Dart_Handle name = Dart_NewString(path); | |
| 115 Dart_Post(file_port, name); | |
| 116 } | |
| 117 } | |
| 118 | |
| 119 | |
| 120 static bool ListRecursively(const char* dir_name, | |
| 121 bool recursive, | |
| 122 Dart_Port dir_port, | |
| 123 Dart_Port file_port, | |
| 124 Dart_Port done_port, | |
| 125 Dart_Port error_port) { | |
| 126 DIR* dir_pointer = opendir(dir_name); | |
| 127 if (dir_pointer == NULL) { | |
| 128 // TODO(ager): post something on the error port. | |
| 129 return false; | |
| 130 } | |
| 131 | |
| 132 // Compute full path for the directory currently being listed. | |
| 133 char *path = static_cast<char*>(malloc(PATH_MAX)); | |
| 134 ASSERT(path != NULL); | |
| 135 int path_length = 0; | |
| 136 ComputeFullPath(dir_name, path, &path_length); | |
| 137 | |
| 138 // Iterated the directory and post the directories and files to the | |
| 139 // ports. | |
| 140 int success = 0; | |
| 141 bool completed = true; | |
| 142 dirent entry; | |
| 143 dirent* result; | |
| 144 while ((success = readdir_r(dir_pointer, &entry, &result)) == 0 && | |
| 145 result != NULL) { | |
| 146 switch (entry.d_type) { | |
| 147 case DT_DIR: | |
| 148 completed = completed && HandleDir(entry.d_name, | |
| 149 path, | |
| 150 path_length, | |
| 151 recursive, | |
| 152 dir_port, | |
| 153 file_port, | |
| 154 done_port, | |
| 155 error_port); | |
| 156 break; | |
| 157 case DT_REG: | |
| 158 HandleFile(entry.d_name, path, path_length, file_port); | |
| 159 break; | |
| 160 case DT_UNKNOWN: { | |
| 161 // On some file systems the entry type is not determined by | |
| 162 // readdir_r. For those we use lstat to determine the entry | |
| 163 // type. | |
| 164 struct stat entry_info; | |
| 165 size_t written = snprintf(path + path_length, | |
| 166 PATH_MAX - path_length, | |
| 167 "%s", | |
| 168 entry.d_name); | |
| 169 ASSERT(written == strlen(entry.d_name)); | |
| 170 int lstat_success = lstat(path, &entry_info); | |
| 171 if (lstat_success != 0) { | |
| 172 completed = false; | |
| 173 break; | |
| 174 } | |
| 175 if ((entry_info.st_mode & S_IFMT) == S_IFDIR) { | |
| 176 HandleDir(entry.d_name, | |
| 177 path, | |
| 178 path_length, | |
| 179 recursive, | |
| 180 dir_port, | |
| 181 file_port, | |
| 182 done_port, | |
| 183 error_port); | |
| 184 } else if ((entry_info.st_mode & S_IFMT) == S_IFREG) { | |
| 185 HandleFile(entry.d_name, path, path_length, file_port); | |
| 186 } | |
| 187 break; | |
| 188 } | |
| 189 default: | |
| 190 break; | |
| 191 } | |
| 192 } | |
| 193 completed = completed && (success == 0); | |
| 194 | |
| 195 // TODO(ager): Post on error port if closing fails. | |
| 196 closedir(dir_pointer); | |
| 197 free(path); | |
| 198 | |
| 199 return completed; | |
| 200 } | |
| 201 | |
| 202 | |
| 203 void Directory::List(const char* dir_name, | |
| 204 bool recursive, | |
| 205 Dart_Port dir_port, | |
| 206 Dart_Port file_port, | |
| 207 Dart_Port done_port, | |
| 208 Dart_Port error_port) { | |
| 209 bool completed = ListRecursively(dir_name, | |
| 210 recursive, | |
| 211 dir_port, | |
| 212 file_port, | |
| 213 done_port, | |
| 214 error_port); | |
| 215 if (done_port != 0) { | |
| 216 Dart_Handle value = Dart_NewBoolean(completed); | |
| 217 Dart_Post(done_port, value); | |
| 218 } | |
| 219 } | |
| OLD | NEW |