Fix segfault in Filereader if directory does not exist.

Signed-off-by: ptredak <ptredak@nvidia.com>
NVIDIA · Aug 17, 2018 · b5dd3e3 · b5dd3e3
1 parent a66396e
commit b5dd3e3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dali/pipeline/operators/reader/loader/file_loader.h b/dali/pipeline/operators/reader/loader/file_loader.h
@@ -53,6 +53,9 @@ vector<std::pair<string, int>> traverse_directories(const std::string& path) {
   // open the root
   DIR *dir = opendir(path.c_str());
 
+  DALI_ENFORCE(dir != nullptr,
+      "Directory " + path + " could not be opened.");
+
   struct dirent *entry;
   int dir_count = 0;