Directory: implement recursive directory listing on Linux and Mac.

runtime/bin/directory_linux.cc

 // Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include <dirent.h>
 #include <libgen.h>
 #include <string.h>
 #include <sys/param.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <unistd.h>
 
 #include "bin/dartutils.h"
 #include "bin/directory.h"
 #include "bin/file.h"
 
+// Forward declaration.
+static bool ListRecursively(const char* dir_name,
+                            bool recursive,
+                            Dart_Port dir_port,
+                            Dart_Port file_port,
+                            Dart_Port done_port,
+                            Dart_Port error_port);
+
 
 static void ComputeFullPath(const char* dir_name,
                             char* path,
                             int* path_length) {
   size_t written = 0;
 
   if (!File::IsAbsolutePath(dir_name)) {
     ASSERT(getcwd(path, PATH_MAX) != NULL);
     *path_length = strlen(path);
     written = snprintf(path + *path_length,
@@ skipping 10 matching lines @@
   char* dir = dirname(dir_name_copy);
   if (strcmp(dir, ".") != 0) {
     written = snprintf(path + *path_length,
                        PATH_MAX - *path_length,
                        "%s%s",
                        dir,
                        File::PathSeparator());
     ASSERT(written == (strlen(dir) + strlen(File::PathSeparator())));
     *path_length += written;
   }
-  char* base = basename(dir_name_copy);
+  char* base_name_copy = strdup(dir_name);
+  char* base = basename(base_name_copy);
   if (strcmp(base, ".") != 0) {
     written = snprintf(path + *path_length,
                        PATH_MAX - *path_length,
                        "%s%s",
                        base,
                        File::PathSeparator());
     ASSERT(written == (strlen(base) + strlen(File::PathSeparator())));
     *path_length += written;
   }
+
   free(dir_name_copy);
+  free(base_name_copy);
 }
 
 
-static void HandleDir(char* dir_name,
+static bool HandleDir(char* dir_name,
                       char* path,
                       int path_length,
-                      Dart_Port dir_port) {
-  if (dir_port != 0 &&
-      strcmp(dir_name, ".") != 0 &&
+                      bool recursive,
+                      Dart_Port dir_port,
+                      Dart_Port file_port,
+                      Dart_Port done_port,
+                      Dart_Port error_port) {
+  if (strcmp(dir_name, ".") != 0 &&
       strcmp(dir_name, "..") != 0) {
     size_t written = snprintf(path + path_length,
                               PATH_MAX - path_length,
                               "%s",
                               dir_name);
     ASSERT(written == strlen(dir_name));
-    Dart_Handle name = Dart_NewString(path);
-    Dart_Post(dir_port, name);
+    if (dir_port != 0) {
+      Dart_Handle name = Dart_NewString(path);
+      Dart_Post(dir_port, name);
+    }
+    if (recursive) {
+      return ListRecursively(path,
+                             recursive,
+                             dir_port,
+                             file_port,
+                             done_port,
+                             error_port);
+    }
   }
+  return true;
 }
 
 
 static void HandleFile(char* file_name,
                        char* path,
                        int path_length,
                        Dart_Port file_port) {
   if (file_port != 0) {
     size_t written = snprintf(path + path_length,
                               PATH_MAX - path_length,
                               "%s",
                               file_name);
     ASSERT(written == strlen(file_name));
     Dart_Handle name = Dart_NewString(path);
     Dart_Post(file_port, name);
   }
 }
 
 
-void Directory::List(const char* dir_name,
-                     bool recursive,
-                     Dart_Port dir_port,
-                     Dart_Port file_port,
-                     Dart_Port done_port,
-                     Dart_Port dir_error_port) {
+static bool ListRecursively(const char* dir_name,
+                            bool recursive,
+                            Dart_Port dir_port,
+                            Dart_Port file_port,
+                            Dart_Port done_port,
+                            Dart_Port error_port) {
   DIR* dir_pointer = opendir(dir_name);
   if (dir_pointer == NULL) {
     // TODO(ager): post something on the error port.
-    Dart_Handle value = Dart_NewBoolean(false);
-    Dart_Post(done_port, value);
-    return;
+    return false;
   }
 
   // Compute full path for the directory currently being listed.
-  char path[PATH_MAX];
+  char *path = static_cast<char*>(malloc(PATH_MAX));
+  ASSERT(path != NULL);
   int path_length = 0;
   ComputeFullPath(dir_name, path, &path_length);
 
   // Iterated the directory and post the directories and files to the
   // ports.
-  //
-  // TODO(ager): Handle recursion and errors caused by recursion.
   int success = 0;
-  bool lstat_error = false;
+  bool completed = true;
   dirent entry;
   dirent* result;
   while ((success = readdir_r(dir_pointer, &entry, &result)) == 0 &&
          result != NULL) {
     switch (entry.d_type) {
       case DT_DIR:
-        HandleDir(entry.d_name, path, path_length, dir_port);
+        completed = completed && HandleDir(entry.d_name,
+                                           path,
+                                           path_length,
+                                           recursive,
+                                           dir_port,
+                                           file_port,
+                                           done_port,
+                                           error_port);
         break;
       case DT_REG:
         HandleFile(entry.d_name, path, path_length, file_port);
         break;
       case DT_UNKNOWN: {
         // On some file systems the entry type is not determined by
         // readdir_r. For those we use lstat to determine the entry
         // type.
         struct stat entry_info;
         size_t written = snprintf(path + path_length,
                                   PATH_MAX - path_length,
                                   "%s",
                                   entry.d_name);
         ASSERT(written == strlen(entry.d_name));
         int lstat_success = lstat(path, &entry_info);
         if (lstat_success != 0) {
-          lstat_error = true;
+          completed = false;
           break;
         }
         if ((entry_info.st_mode & S_IFMT) == S_IFDIR) {
-          HandleDir(entry.d_name, path, path_length, dir_port);
+          HandleDir(entry.d_name,
+                    path,
+                    path_length,
+                    recursive,
+                    dir_port,
+                    file_port,
+                    done_port,
+                    error_port);
         } else if ((entry_info.st_mode & S_IFMT) == S_IFREG) {
           HandleFile(entry.d_name, path, path_length, file_port);
         }
         break;
       }
       default:
         break;
     }
   }
+  completed = completed && (success == 0);
+
+  // TODO(ager): Post on error port if closing fails.
+  closedir(dir_pointer);
+  free(path);
+
+  return completed;
+}
+
+
+void Directory::List(const char* dir_name,
+                     bool recursive,
+                     Dart_Port dir_port,
+                     Dart_Port file_port,
+                     Dart_Port done_port,
+                     Dart_Port error_port) {
+  bool completed = ListRecursively(dir_name,
+                                   recursive,
+                                   dir_port,
+                                   file_port,
+                                   done_port,
+                                   error_port);
   if (done_port != 0) {
-    if (success != 0 || lstat_error) {
-      Dart_Handle value = Dart_NewBoolean(false);
-      Dart_Post(done_port, value);
-    } else {
-      Dart_Handle value = Dart_NewBoolean(true);
-      Dart_Post(done_port, value);
-    }
+    Dart_Handle value = Dart_NewBoolean(completed);
+    Dart_Post(done_port, value);
   }
-
-  // TODO(ager): Post on error port.
-  closedir(dir_pointer);
 }