RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-12-14 23:29:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support/PathV2: Implement directory iteration on POSIX.

Browse Source 
llvm-svn: 122879
This commit is contained in:
Michael J. Spencer 2011-01-05 16:38:57 +00:00
parent a7b9603f24
commit e369cc8053
2 changed files with 54 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
lib/Support/Unix/PathV2.inc
View File

@ -23,6 +23,22 @@
#if HAVE_FCNTL_H
#include <fcntl.h>
#endif
#if HAVE_DIRENT_H
# include <dirent.h>
# define NAMLEN(dirent) strlen((dirent)->d_name)
#else
# define dirent direct
# define NAMLEN(dirent) (dirent)->d_namlen
# if HAVE_SYS_NDIR_H
# include <sys/ndir.h>
# endif
# if HAVE_SYS_DIR_H
# include <sys/dir.h>
# endif
# if HAVE_NDIR_H
# include <ndir.h>
# endif
#endif
#if HAVE_STDIO_H
#include <stdio.h>
#endif
@ -408,6 +424,44 @@ rety_open_create:
return success;
}
error_code directory_iterator_construct(directory_iterator &it, StringRef path){
SmallString<128> path_null(path);
DIR *directory = ::opendir(path_null.c_str());
if (directory == 0)
return error_code(errno, system_category());
it.IterationHandle = reinterpret_cast<intptr_t>(directory);
// Add something for replace_filename to replace.
path::append(path_null, ".");
it.CurrentEntry = directory_entry(path_null.str());
return directory_iterator_increment(it);
}
error_code directory_iterator_destruct(directory_iterator& it) {
if (it.IterationHandle)
::closedir(reinterpret_cast<DIR *>(it.IterationHandle));
it.IterationHandle = 0;
it.CurrentEntry = directory_entry();
return success;
}
error_code directory_iterator_increment(directory_iterator& it) {
errno = 0;
dirent *cur_dir = ::readdir(reinterpret_cast<DIR *>(it.IterationHandle));
if (cur_dir == 0 && errno != 0) {
return error_code(errno, system_category());
} else if (cur_dir != 0) {
StringRef name(cur_dir->d_name, NAMLEN(cur_dir));
if ((name.size() == 1 && name[0] == '.') ||
(name.size() == 2 && name[0] == '.' && name[1] == '.'))
return directory_iterator_increment(it);
it.CurrentEntry.replace_filename(name);
} else
return directory_iterator_destruct(it);
return success;
}
} // end namespace fs
} // end namespace sys
} // end namespace llvm

3
unittests/Support/Path.cpp
View File

@ -176,8 +176,6 @@ TEST(Support, Path) {
ASSERT_NO_ERROR(fs::exists(Twine(TempPath), TempFileExists));
EXPECT_FALSE(TempFileExists);
// I've yet to do directory iteration on Unix.
#ifdef LLVM_ON_WIN32
error_code ec;
for (fs::directory_iterator i(".", ec), e; i != e; i.increment(ec)) {
if (ec) {
@ -186,7 +184,6 @@ TEST(Support, Path) {
report_fatal_error("Directory iteration failed!");
}
}
#endif
}
} // anonymous namespace