fs/DirectoryReader: use FindFirstFile(), FindNextFile() on WIN32

This commit is contained in:
Max Kellermann 2015-02-26 18:11:02 +01:00
parent e3908c8de6
commit 4f0f81a047
1 changed files with 85 additions and 0 deletions

View File

@ -23,6 +23,89 @@
#include "check.h"
#include "Path.hxx"
#ifdef WIN32
#include <windows.h>
#include <tchar.h>
/**
* Reader for directory entries.
*/
class DirectoryReader {
const HANDLE handle;
WIN32_FIND_DATA data;
bool first;
class MakeWildcardPath {
PathTraitsFS::pointer path;
public:
MakeWildcardPath(PathTraitsFS::const_pointer _path) {
auto l = _tcslen(_path);
path = new PathTraitsFS::value_type[l + 3];
_tcscpy(path, _path);
path[l] = _T('\\');
path[l + 1] = _T('*');
path[l + 2] = 0;
}
~MakeWildcardPath() {
delete[] path;
}
operator PathTraitsFS::const_pointer() const {
return path;
}
};
public:
/**
* Creates new directory reader for the specified #dir.
*/
explicit DirectoryReader(Path dir)
:handle(FindFirstFile(MakeWildcardPath(dir.c_str()), &data)),
first(true) {}
DirectoryReader(const DirectoryReader &other) = delete;
DirectoryReader &operator=(const DirectoryReader &other) = delete;
/**
* Destroys this instance.
*/
~DirectoryReader() {
if (!HasFailed())
FindClose(handle);
}
/**
* Checks if directory failed to open.
*/
bool HasFailed() const {
return handle == INVALID_HANDLE_VALUE;
}
/**
* Reads next directory entry.
*/
bool ReadEntry() {
if (first) {
first = false;
return true;
}
return FindNextFile(handle, &data) != 0;
}
/**
* Extracts directory entry that was previously read by #ReadEntry.
*/
Path GetEntry() const {
return Path::FromFS(data.cFileName);
}
};
#else
#include <dirent.h>
/**
@ -85,3 +168,5 @@ public:
};
#endif
#endif