ReadDirectory() added in webrtc/test/testsupport/fileutils.h

webrtc/test/testsupport/fileutils.cc

 /*
  *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
@@ skipping 22 matching lines @@
 #ifndef S_ISDIR  // Not defined in stat.h on Windows.
 #define S_ISDIR(mode) (((mode) & S_IFMT) == S_IFDIR)
 #endif
 
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 
 #include <memory>
 
+#include "webrtc/base/checks.h"
 #include "webrtc/typedefs.h"  // For architecture defines
 
 namespace webrtc {
 namespace test {
 
 #if defined(WEBRTC_IOS)
 // Defined in iosfileutils.mm.  No header file to discourage use elsewhere.
 std::string IOSOutputPath();
 std::string IOSRootPath();
 std::string IOSResourcePath(std::string name, std::string extension);
@@ skipping 66 matching lines @@
 }
 
 std::string OutputPath() {
   return kRootDirName;
 }
 
 std::string WorkingDir() {
   return kRootDirName;
 }
 
-#else // WEBRTC_ANDROID
+#else  // WEBRTC_ANDROID
 
 std::string ProjectRootPath() {
 #if defined(WEBRTC_IOS)
   return IOSRootPath();
 #else
   std::string path = WorkingDir();
   if (path == kFallbackPath) {
     return kCannotFindProjectRootDir;
   }
   if (relative_dir_path_set) {
@@ skipping 85 matching lines @@
   } else {
 #ifdef WIN32
     return _mkdir(directory_name.c_str()) == 0;
 #else
     return mkdir(directory_name.c_str(),  S_IRWXU | S_IRWXG | S_IRWXO) == 0;
 #endif
   }
   return true;
 }
 
+size_t RemoveDirRecursively(const rtc::Pathname& dir) {
+  if (!DirExists(dir.pathname())) { return false; }
+  rtc::DirectoryIterator it;
+  it.Iterate(dir);
+  size_t deleted_entries = 0;
+  do {
+    const std::string entry_name = it.Name();
+    if (entry_name == "." || entry_name == "..") { continue; }
+
+    const rtc::Pathname entry = it.Path();
+    if (it.IsDirectory()) {
+      deleted_entries += RemoveDirRecursively(entry);
+    } else {
+      if (RemoveFile(entry.pathname())) { deleted_entries++; }
+    }
+  } while (it.Next());
+  RemoveDir(dir.pathname());
+  return deleted_entries + 1;
+}
+
 bool RemoveDir(const std::string& directory_name) {
 #ifdef WIN32
     return RemoveDirectoryA(directory_name.c_str()) != FALSE;
 #else
     return rmdir(directory_name.c_str()) == 0;
 #endif
 }
 
 bool RemoveFile(const std::string& file_name) {
 #ifdef WIN32
@@ skipping 55 matching lines @@
     if (fseek(f, 0, SEEK_END) == 0) {
       size = ftell(f);
     }
     fclose(f);
   }
   return size;
 }
 
 }  // namespace test
 }  // namespace webrtc