mirror of
https://github.com/rakshasa/rtorrent.git
synced 2026-10-04 21:29:21 +00:00
xirvik
a5c39566c8
Also make directory_entry::is_file() report the real type so symlinks are skipped.
100 lines
1.8 KiB
C++
100 lines
1.8 KiB
C++
#include "config.h"
|
|
|
|
#include "utils/directory.h"
|
|
|
|
#include <algorithm>
|
|
#include <cstdlib>
|
|
#include <dirent.h>
|
|
#include <fcntl.h>
|
|
#include <functional>
|
|
#include <sys/stat.h>
|
|
#include <torrent/exceptions.h>
|
|
|
|
#include "globals.h"
|
|
|
|
namespace utils {
|
|
|
|
namespace {
|
|
|
|
uint8_t
|
|
entry_type_from_mode(mode_t mode) {
|
|
if (S_ISREG(mode))
|
|
return DT_REG;
|
|
|
|
if (S_ISDIR(mode))
|
|
return DT_DIR;
|
|
|
|
if (S_ISLNK(mode))
|
|
return DT_LNK;
|
|
|
|
return DT_UNKNOWN;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
// Keep this?
|
|
bool
|
|
Directory::is_valid() const {
|
|
if (m_path.empty())
|
|
return false;
|
|
|
|
DIR* d = opendir(expand_path(m_path).c_str());
|
|
|
|
if (d != NULL)
|
|
closedir(d);
|
|
|
|
return d;
|
|
}
|
|
|
|
bool
|
|
Directory::update(int flags) {
|
|
if (m_path.empty())
|
|
throw torrent::input_error("Directory::update() tried to open an empty path.");
|
|
|
|
DIR* d = opendir(expand_path(m_path).c_str());
|
|
|
|
if (d == NULL)
|
|
return false;
|
|
|
|
struct dirent* entry;
|
|
|
|
while ((entry = readdir(d)) != NULL) {
|
|
if ((flags & update_hide_dot) && entry->d_name[0] == '.')
|
|
continue;
|
|
|
|
iterator itr = base_type::insert(end(), value_type());
|
|
|
|
#ifdef __sun__
|
|
itr->s_fileno = entry->d_ino;
|
|
itr->s_reclen = 0;
|
|
itr->s_type = DT_UNKNOWN;
|
|
#else
|
|
itr->s_fileno = entry->d_fileno;
|
|
itr->s_reclen = entry->d_reclen;
|
|
itr->s_type = entry->d_type;
|
|
#endif
|
|
|
|
if (itr->s_type == DT_UNKNOWN) {
|
|
struct stat st;
|
|
|
|
if (fstatat(dirfd(d), entry->d_name, &st, AT_SYMLINK_NOFOLLOW) == 0)
|
|
itr->s_type = entry_type_from_mode(st.st_mode);
|
|
}
|
|
|
|
#ifdef DIRENT_NAMLEN_EXISTS_FOOBAR
|
|
itr->s_name = std::string(entry->d_name, entry->d_name + entry->d_namlen);
|
|
#else
|
|
itr->s_name = std::string(entry->d_name);
|
|
#endif
|
|
}
|
|
|
|
closedir(d);
|
|
|
|
if (flags & update_sort)
|
|
std::sort(begin(), end());
|
|
|
|
return true;
|
|
}
|
|
|
|
}
|