//
// $Id$
+#include <simgear_config.h>
+#include <simgear/compiler.h>
#include <simgear/misc/sg_dir.hxx>
+#include <simgear/structure/exception.hxx>
+#include <math.h>
+#include <stdlib.h>
+#include <cstdio>
#ifdef _WIN32
# define WIN32_LEAN_AND_MEAN
# include <windows.h>
+# include <direct.h>
+# include <Shlwapi.h>
#else
# include <sys/types.h>
# include <dirent.h>
+# include <sys/stat.h>
+# include <unistd.h>
+# include <errno.h>
#endif
-#include <sys/stat.h>
-
+#include <simgear/misc/strutils.hxx>
#include <simgear/debug/logstream.hxx>
+#include <boost/foreach.hpp>
#include <cstring>
+#include <cstdlib>
#include <iostream>
+#include <algorithm> // for std::sort
+
+using std::string;
namespace simgear
{
+Dir::Dir() :
+ _removeOnDestroy(false)
+{
+}
+
Dir::Dir(const SGPath& path) :
- _path(path)
+ _path(path),
+ _removeOnDestroy(false)
{
+ _path.set_cached(false); // disable caching, so create/remove work
}
Dir::Dir(const Dir& rel, const SGPath& relPath) :
- _path(rel.file(relPath.str()))
+ _path(rel.file(relPath.utf8Str())),
+ _removeOnDestroy(false)
+{
+ _path.set_cached(false); // disable caching, so create/remove work
+}
+
+Dir::~Dir()
+{
+ if (_removeOnDestroy) {
+ remove(true);
+ }
+}
+
+void Dir::setRemoveOnDestroy()
+{
+ _removeOnDestroy = true;
+}
+
+Dir Dir::current()
+{
+#if defined(SG_WINDOWS)
+ wchar_t* buf = _wgetcwd(NULL, 0);
+#else
+ char *buf = ::getcwd(NULL, 0);
+#endif
+ if (!buf) {
+ if (errno == 2) throw sg_exception("The current directory is invalid");
+ else throw sg_exception(strerror(errno));
+ }
+
+ SGPath p(buf);
+ free(buf);
+ return Dir(p);
+}
+
+Dir Dir::tempDir(const std::string& templ)
+{
+#ifdef HAVE_MKDTEMP
+ char buf[1024];
+ const char* tempPath = ::getenv("TMPDIR");
+ if (!tempPath) {
+ tempPath = "/tmp";
+ }
+ SGPath p(tempPath);
+ p.append(templ);
+ // Mac OS-X / BSD manual says any number of 'X's, but GLibc manual
+ // says exactly six, so that's what I'm going with
+ p.concat("-XXXXXX");
+ std::string s = p.local8BitStr();
+ ::snprintf(buf, 1024, "%s", s.c_str());
+ if (!mkdtemp(buf)) {
+ SG_LOG(SG_IO, SG_WARN, "mkdtemp failed:" << strerror(errno));
+ return Dir();
+ }
+
+ return Dir(SGPath(buf));
+#else
+#if defined(SG_WINDOWS)
+ std::wstring wideTemplate = simgear::strutils::convertUtf8ToWString(templ);
+ wchar_t* buf = _wtempnam(0, wideTemplate.c_str());
+ SGPath p(buf);
+ free(buf); // unlike tempnam(), _wtempnam mallocs its result buffer
+#else
+ SGPath p(tempnam(0, templ.c_str()));
+#endif
+ Dir t(p);
+ if (!t.create(0700)) {
+ SG_LOG(SG_IO, SG_WARN, "failed to create temporary directory at " << p);
+ }
+
+ return t;
+#endif
+}
+
+static bool pathSortPredicate(const SGPath& p1, const SGPath& p2)
{
+ return p1.file() < p2.file();
}
PathList Dir::children(int types, const std::string& nameFilter) const
types = TYPE_FILE | TYPE_DIR | NO_DOT_OR_DOTDOT;
}
-#ifdef _WIN32
- std::string search(_path.str());
+#if defined(SG_WINDOWS)
+ std::wstring search(_path.wstr());
if (nameFilter.empty()) {
- search += "\\*"; // everything
+ search += simgear::strutils::convertUtf8ToWString("\\*"); // everything
} else {
- search += "\\*" + nameFilter;
+ search += simgear::strutils::convertUtf8ToWString("\\*" + nameFilter);
}
- WIN32_FIND_DATA fData;
- HANDLE find = FindFirstFile(search.c_str(), &fData);
+ WIN32_FIND_DATAW fData;
+ HANDLE find = FindFirstFileW(search.c_str(), &fData);
if (find == INVALID_HANDLE_VALUE) {
- SG_LOG(SG_GENERAL, SG_WARN, "Dir::children: FindFirstFile failed:" << _path.str());
+ int err = GetLastError();
+ if (err != ERROR_FILE_NOT_FOUND) {
+ SG_LOG(SG_GENERAL, SG_WARN, "Dir::children: FindFirstFile failed:" <<
+ _path << " with error:" << err);
+ }
return result;
}
bool done = false;
- for (bool done = false; !done; done = (FindNextFile(find, &fData) == 0)) {
+ for (bool done = false; !done; done = (FindNextFileW(find, &fData) == 0)) {
if (fData.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) {
if (!(types & INCLUDE_HIDDEN)) {
continue;
}
}
+ std::string utf8File = simgear::strutils::convertWStringToUtf8(fData.cFileName);
if (fData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
+ if (types & NO_DOT_OR_DOTDOT) {
+ if ((utf8File == ".") || (utf8File == "..")) {
+ continue;
+ }
+ }
+
if (!(types & TYPE_DIR)) {
continue;
}
continue;
}
- result.push_back(file(fData.cFileName));
+ result.push_back(file(utf8File));
}
FindClose(find);
#else
- DIR* dp = opendir(_path.c_str());
+ std::string ps = _path.local8BitStr();
+ DIR* dp = opendir(ps.c_str());
if (!dp) {
- SG_LOG(SG_GENERAL, SG_WARN, "Dir::children: opendir failed:" << _path.str());
+ SG_LOG(SG_GENERAL, SG_WARN, "Dir::children: opendir failed:" << _path);
return result;
}
+ int filterLen = nameFilter.size();
+
while (true) {
struct dirent* entry = readdir(dp);
if (!entry) {
}
// skip hidden files (names beginning with '.') unless requested
- if (!(types & INCLUDE_HIDDEN) && (entry->d_name[0] == '.')) {
+ if (!(types & INCLUDE_HIDDEN) && (entry->d_name[0] == '.') &&
+ strcmp(entry->d_name, ".") && strcmp(entry->d_name, "..")) {
continue;
}
- int type = entry->d_type;
- if (type == DT_LNK) {
- // find symlink target type using stat()
- struct stat s;
- if (stat(file(entry->d_name).c_str(), &s)) {
- continue; // stat() failed
- }
-
- if (S_ISDIR(s.st_mode)) {
- type = DT_DIR;
- } else if (S_ISREG(s.st_mode)) {
- type = DT_REG;
- } else {
- // symlink to block/fifo/char file, ignore
- continue;
- }
- } // of symlink look-through
+ SGPath f = file(entry->d_name);
+ if (!f.exists()) {
+ continue; // stat() failed
+ }
- if (type == DT_DIR) {
+ if (f.isDir()) {
+ // directory handling
if (!(types & TYPE_DIR)) {
continue;
}
continue;
}
}
- } else if (type == DT_REG) {
+ } else if (f.isFile()) {
+ // regular file handling
if (!(types & TYPE_FILE)) {
continue;
}
} else {
- continue; // ignore char/block devices, fifos, etc
+ // block device /fifo/char file, ignore
+ continue;
}
-
+
if (!nameFilter.empty()) {
- if (strstr(entry->d_name, nameFilter.c_str()) == NULL) {
+ int nameLen = strlen(entry->d_name);
+ if (nameLen < filterLen) {
+ continue; // name is shorter than the filter
+ }
+
+ char* nameSuffix = entry->d_name + (nameLen - filterLen);
+ if (strcmp(nameSuffix, nameFilter.c_str())) {
continue;
}
}
closedir(dp);
#endif
+
+ // File system order is random. Make things deterministic,
+ // so it's the same for every user.
+ std::sort(result.begin(), result.end(), pathSortPredicate);
+
return result;
}
+bool Dir::isEmpty() const
+{
+#if defined(SG_WINDOWS)
+ std::wstring ps = _path.wstr();
+ return PathIsDirectoryEmptyW( ps.c_str() );
+#else
+ std::string ps = _path.local8BitStr();
+ DIR* dp = opendir( ps.c_str() );
+ if (!dp) return true;
+
+ int n = 0;
+ dirent* d;
+ while( (d = readdir(dp)) !=NULL && (n < 4) ) n++;
+ closedir(dp);
+
+ return (n == 2); // '.' and '..' always exist
+#endif
+}
+
+bool Dir::exists() const
+{
+ return _path.isDir();
+}
+
SGPath Dir::file(const std::string& name) const
{
SGPath childPath = _path;
+ childPath.set_cached(true);
childPath.append(name);
return childPath;
}
+bool Dir::create(mode_t mode)
+{
+ if (exists()) {
+ return false; // already exists
+ }
+
+// recursively create parent directories
+ Dir pr(parent());
+ if (!pr.path().isNull() && !pr.exists()) {
+ bool ok = pr.create(mode);
+ if (!ok) {
+ return false;
+ }
+ }
+
+// finally, create ourselves
+#if defined(SG_WINDOWS)
+ std::wstring ps = _path.wstr();
+ int err = _wmkdir(ps.c_str());
+#else
+ std::string ps = _path.local8BitStr();
+ int err = mkdir(ps.c_str(), mode);
+#endif
+ if (err) {
+ SG_LOG(SG_IO, SG_WARN, "directory creation failed: (" << _path << ") " << strerror(errno) );
+ }
+
+ return (err == 0);
+}
+
+bool Dir::removeChildren() const
+{
+ bool ok;
+ PathList cs = children(NO_DOT_OR_DOTDOT | INCLUDE_HIDDEN | TYPE_FILE | TYPE_DIR);
+ BOOST_FOREACH(SGPath path, cs) {
+ if (path.isDir()) {
+ Dir childDir(path);
+ ok = childDir.remove(true);
+ } else {
+ ok = path.remove();
+ }
+
+ if (!ok) {
+ SG_LOG(SG_IO, SG_WARN, "failed to remove:" << path);
+ return false;
+ }
+ } // of child iteration
+
+ return true;
+}
+
+bool Dir::remove(bool recursive)
+{
+ if (!exists()) {
+ SG_LOG(SG_IO, SG_WARN, "attempt to remove non-existant dir:" << _path);
+ return false;
+ }
+
+ if (recursive) {
+ if (!removeChildren()) {
+ SG_LOG(SG_IO, SG_WARN, "Dir at:" << _path << " failed to remove children");
+ return false;
+ }
+ } // of recursive deletion
+
+#if defined(SG_WINDOWS)
+ std::wstring ps = _path.wstr();
+ int err = _wrmdir(ps.c_str());
+#else
+ std::string ps = _path.local8BitStr();
+ int err = rmdir(ps.c_str());
+#endif
+ if (err) {
+ SG_LOG(SG_IO, SG_WARN, "rmdir failed:" << _path << ":" << strerror(errno));
+ }
+ return (err == 0);
+}
+
+Dir Dir::parent() const
+{
+ return Dir(_path.dir());
+}
+
} // of namespace simgear