forked from ShuriZma/suyu
1
0
Fork 0

file_util: Migrate remaining file handling functions over to std::filesystem

Converts creation and deletion functions over to std::filesystem,
simplifying our file-handling code.

Notably with this, CopyDir will now function on Windows.
This commit is contained in:
Lioncash 2020-12-08 18:51:04 -05:00
parent 0e54aa17e6
commit 20aad9e01a
2 changed files with 111 additions and 351 deletions

View File

@ -68,13 +68,6 @@
#include <algorithm> #include <algorithm>
#include <sys/stat.h> #include <sys/stat.h>
#ifndef S_ISDIR
#define S_ISDIR(m) (((m)&S_IFMT) == S_IFDIR)
#endif
// This namespace has various generic functions related to files and paths.
// The code still needs a ton of cleanup.
// REMEMBER: strdup considered harmful!
namespace Common::FS { namespace Common::FS {
namespace fs = std::filesystem; namespace fs = std::filesystem;
@ -88,224 +81,89 @@ bool IsDirectory(const fs::path& path) {
return fs::is_directory(path, ec); return fs::is_directory(path, ec);
} }
bool Delete(const std::string& filename) { bool Delete(const fs::path& path) {
LOG_TRACE(Common_Filesystem, "file {}", filename); LOG_TRACE(Common_Filesystem, "file {}", path.string());
// Return true because we care about the file no // Return true because we care about the file no
// being there, not the actual delete. // being there, not the actual delete.
if (!Exists(filename)) { if (!Exists(path)) {
LOG_DEBUG(Common_Filesystem, "{} does not exist", filename); LOG_DEBUG(Common_Filesystem, "{} does not exist", path.string());
return true; return true;
} }
// We can't delete a directory std::error_code ec;
if (IsDirectory(filename)) { return fs::remove(path, ec);
LOG_ERROR(Common_Filesystem, "Failed: {} is a directory", filename); }
return false;
}
#ifdef _WIN32 bool CreateDir(const fs::path& path) {
if (!DeleteFileW(Common::UTF8ToUTF16W(filename).c_str())) { LOG_TRACE(Common_Filesystem, "directory {}", path.string());
LOG_ERROR(Common_Filesystem, "DeleteFile failed on {}: {}", filename, GetLastErrorMsg());
std::error_code ec;
const bool success = fs::create_directory(path, ec);
if (!success) {
LOG_ERROR(Common_Filesystem, "Unable to create directory: {}", ec.message());
return false; return false;
} }
#else
if (unlink(filename.c_str()) == -1) {
LOG_ERROR(Common_Filesystem, "unlink failed on {}: {}", filename, GetLastErrorMsg());
return false;
}
#endif
return true; return true;
} }
bool CreateDir(const std::string& path) { bool CreateFullPath(const fs::path& path) {
LOG_TRACE(Common_Filesystem, "directory {}", path); LOG_TRACE(Common_Filesystem, "path {}", path.string());
#ifdef _WIN32
if (::CreateDirectoryW(Common::UTF8ToUTF16W(path).c_str(), nullptr)) std::error_code ec;
return true; const bool success = fs::create_directories(path, ec);
DWORD error = GetLastError();
if (error == ERROR_ALREADY_EXISTS) { if (!success) {
LOG_DEBUG(Common_Filesystem, "CreateDirectory failed on {}: already exists", path); LOG_ERROR(Common_Filesystem, "Unable to create full path: {}", ec.message());
return true;
}
LOG_ERROR(Common_Filesystem, "CreateDirectory failed on {}: {}", path, error);
return false; return false;
#else
if (mkdir(path.c_str(), 0755) == 0)
return true;
int err = errno;
if (err == EEXIST) {
LOG_DEBUG(Common_Filesystem, "mkdir failed on {}: already exists", path);
return true;
} }
LOG_ERROR(Common_Filesystem, "mkdir failed on {}: {}", path, strerror(err)); return true;
return false;
#endif
} }
bool CreateFullPath(const std::string& fullPath) { bool Rename(const fs::path& src, const fs::path& dst) {
int panicCounter = 100; LOG_TRACE(Common_Filesystem, "{} --> {}", src.string(), dst.string());
LOG_TRACE(Common_Filesystem, "path {}", fullPath);
if (Exists(fullPath)) { std::error_code ec;
LOG_DEBUG(Common_Filesystem, "path exists {}", fullPath); fs::rename(src, dst, ec);
return true;
}
std::size_t position = 0; if (ec) {
while (true) { LOG_ERROR(Common_Filesystem, "Unable to rename file from {} to {}: {}", src.string(),
// Find next sub path dst.string(), ec.message());
position = fullPath.find(DIR_SEP_CHR, position);
// we're done, yay!
if (position == fullPath.npos)
return true;
// Include the '/' so the first call is CreateDir("/") rather than CreateDir("")
std::string const subPath(fullPath.substr(0, position + 1));
if (!IsDirectory(subPath) && !CreateDir(subPath)) {
LOG_ERROR(Common, "CreateFullPath: directory creation failed");
return false; return false;
} }
// A safety check return true;
panicCounter--;
if (panicCounter <= 0) {
LOG_ERROR(Common, "CreateFullPath: directory structure is too deep");
return false;
}
position++;
}
} }
bool DeleteDir(const std::string& filename) { bool Copy(const fs::path& src, const fs::path& dst) {
LOG_TRACE(Common_Filesystem, "directory {}", filename); LOG_TRACE(Common_Filesystem, "{} --> {}", src.string(), dst.string());
// check if a directory std::error_code ec;
if (!IsDirectory(filename)) { const bool success = fs::copy_file(src, dst, fs::copy_options::overwrite_existing, ec);
LOG_ERROR(Common_Filesystem, "Not a directory {}", filename);
if (!success) {
LOG_ERROR(Common_Filesystem, "Unable to copy file {} to {}: {}", src.string(), dst.string(),
ec.message());
return false; return false;
} }
#ifdef _WIN32
if (::RemoveDirectoryW(Common::UTF8ToUTF16W(filename).c_str()))
return true; return true;
#else
if (rmdir(filename.c_str()) == 0)
return true;
#endif
LOG_ERROR(Common_Filesystem, "failed {}: {}", filename, GetLastErrorMsg());
return false;
} }
bool Rename(const std::string& srcFilename, const std::string& destFilename) { u64 GetSize(const fs::path& path) {
LOG_TRACE(Common_Filesystem, "{} --> {}", srcFilename, destFilename); std::error_code ec;
#ifdef _WIN32 const auto size = fs::file_size(path, ec);
if (_wrename(Common::UTF8ToUTF16W(srcFilename).c_str(),
Common::UTF8ToUTF16W(destFilename).c_str()) == 0)
return true;
#else
if (rename(srcFilename.c_str(), destFilename.c_str()) == 0)
return true;
#endif
LOG_ERROR(Common_Filesystem, "failed {} --> {}: {}", srcFilename, destFilename,
GetLastErrorMsg());
return false;
}
bool Copy(const std::string& srcFilename, const std::string& destFilename) { if (ec) {
LOG_TRACE(Common_Filesystem, "{} --> {}", srcFilename, destFilename); LOG_ERROR(Common_Filesystem, "Unable to retrieve file size ({}): {}", path.string(),
#ifdef _WIN32 ec.message());
if (CopyFileW(Common::UTF8ToUTF16W(srcFilename).c_str(),
Common::UTF8ToUTF16W(destFilename).c_str(), FALSE))
return true;
LOG_ERROR(Common_Filesystem, "failed {} --> {}: {}", srcFilename, destFilename,
GetLastErrorMsg());
return false;
#else
using CFilePointer = std::unique_ptr<FILE, decltype(&std::fclose)>;
// Open input file
CFilePointer input{fopen(srcFilename.c_str(), "rb"), std::fclose};
if (!input) {
LOG_ERROR(Common_Filesystem, "opening input failed {} --> {}: {}", srcFilename,
destFilename, GetLastErrorMsg());
return false;
}
// open output file
CFilePointer output{fopen(destFilename.c_str(), "wb"), std::fclose};
if (!output) {
LOG_ERROR(Common_Filesystem, "opening output failed {} --> {}: {}", srcFilename,
destFilename, GetLastErrorMsg());
return false;
}
// copy loop
std::array<char, 1024> buffer;
while (!feof(input.get())) {
// read input
std::size_t rnum = fread(buffer.data(), sizeof(char), buffer.size(), input.get());
if (rnum != buffer.size()) {
if (ferror(input.get()) != 0) {
LOG_ERROR(Common_Filesystem, "failed reading from source, {} --> {}: {}",
srcFilename, destFilename, GetLastErrorMsg());
return false;
}
}
// write output
std::size_t wnum = fwrite(buffer.data(), sizeof(char), rnum, output.get());
if (wnum != rnum) {
LOG_ERROR(Common_Filesystem, "failed writing to output, {} --> {}: {}", srcFilename,
destFilename, GetLastErrorMsg());
return false;
}
}
return true;
#endif
}
u64 GetSize(const std::string& filename) {
if (!Exists(filename)) {
LOG_ERROR(Common_Filesystem, "failed {}: No such file", filename);
return 0; return 0;
} }
if (IsDirectory(filename)) { return size;
LOG_ERROR(Common_Filesystem, "failed {}: is a directory", filename);
return 0;
}
struct stat buf;
#ifdef _WIN32
if (_wstat64(Common::UTF8ToUTF16W(filename).c_str(), &buf) == 0)
#else
if (stat(filename.c_str(), &buf) == 0)
#endif
{
LOG_TRACE(Common_Filesystem, "{}: {}", filename, buf.st_size);
return buf.st_size;
}
LOG_ERROR(Common_Filesystem, "Stat failed {}: {}", filename, GetLastErrorMsg());
return 0;
}
u64 GetSize(const int fd) {
struct stat buf;
if (fstat(fd, &buf) != 0) {
LOG_ERROR(Common_Filesystem, "GetSize: stat failed {}: {}", fd, GetLastErrorMsg());
return 0;
}
return buf.st_size;
} }
u64 GetSize(FILE* f) { u64 GetSize(FILE* f) {
@ -393,132 +251,58 @@ bool ForeachDirectoryEntry(u64* num_entries_out, const std::string& directory,
return true; return true;
} }
u64 ScanDirectoryTree(const std::string& directory, FSTEntry& parent_entry, bool DeleteDirRecursively(const fs::path& path) {
unsigned int recursion) { std::error_code ec;
const auto callback = [recursion, &parent_entry](u64* num_entries_out, fs::remove_all(path, ec);
const std::string& directory,
const std::string& virtual_name) -> bool {
FSTEntry entry;
entry.virtualName = virtual_name;
entry.physicalName = directory + DIR_SEP + virtual_name;
if (IsDirectory(entry.physicalName)) { if (ec) {
entry.isDirectory = true; LOG_ERROR(Common_Filesystem, "Unable to completely delete directory {}: {}", path.string(),
// is a directory, lets go inside if we didn't recurse to often ec.message());
if (recursion > 0) {
entry.size = ScanDirectoryTree(entry.physicalName, entry, recursion - 1);
*num_entries_out += entry.size;
} else {
entry.size = 0;
}
} else { // is a file
entry.isDirectory = false;
entry.size = GetSize(entry.physicalName);
}
(*num_entries_out)++;
// Push into the tree
parent_entry.children.push_back(std::move(entry));
return true;
};
u64 num_entries;
return ForeachDirectoryEntry(&num_entries, directory, callback) ? num_entries : 0;
}
bool DeleteDirRecursively(const std::string& directory, unsigned int recursion) {
const auto callback = [recursion](u64*, const std::string& directory,
const std::string& virtual_name) {
const std::string new_path = directory + DIR_SEP_CHR + virtual_name;
if (IsDirectory(new_path)) {
if (recursion == 0) {
return false; return false;
} }
return DeleteDirRecursively(new_path, recursion - 1);
}
return Delete(new_path);
};
if (!ForeachDirectoryEntry(nullptr, directory, callback))
return false;
// Delete the outermost directory
DeleteDir(directory);
return true; return true;
} }
void CopyDir([[maybe_unused]] const std::string& source_path, void CopyDir(const fs::path& src, const fs::path& dst) {
[[maybe_unused]] const std::string& dest_path) { constexpr auto copy_flags = fs::copy_options::skip_existing | fs::copy_options::recursive;
#ifndef _WIN32
if (source_path == dest_path) {
return;
}
if (!Exists(source_path)) {
return;
}
if (!Exists(dest_path)) {
CreateFullPath(dest_path);
}
DIR* dirp = opendir(source_path.c_str()); std::error_code ec;
if (!dirp) { fs::copy(src, dst, copy_flags, ec);
if (ec) {
LOG_ERROR(Common_Filesystem, "Error copying directory {} to {}: {}", src.string(),
dst.string(), ec.message());
return; return;
} }
while (struct dirent* result = readdir(dirp)) { LOG_TRACE(Common_Filesystem, "Successfully copied directory.");
const std::string virtualName(result->d_name);
// check for "." and ".."
if (((virtualName[0] == '.') && (virtualName[1] == '\0')) ||
((virtualName[0] == '.') && (virtualName[1] == '.') && (virtualName[2] == '\0'))) {
continue;
}
std::string source, dest;
source = source_path + virtualName;
dest = dest_path + virtualName;
if (IsDirectory(source)) {
source += '/';
dest += '/';
if (!Exists(dest)) {
CreateFullPath(dest);
}
CopyDir(source, dest);
} else if (!Exists(dest)) {
Copy(source, dest);
}
}
closedir(dirp);
#endif
} }
std::optional<std::string> GetCurrentDir() { std::optional<fs::path> GetCurrentDir() {
// Get the current working directory (getcwd uses malloc) std::error_code ec;
#ifdef _WIN32 auto path = fs::current_path(ec);
wchar_t* dir = _wgetcwd(nullptr, 0);
if (!dir) { if (ec) {
#else LOG_ERROR(Common_Filesystem, "Unable to retrieve current working directory: {}",
char* dir = getcwd(nullptr, 0); ec.message());
if (!dir) {
#endif
LOG_ERROR(Common_Filesystem, "GetCurrentDirectory failed: {}", GetLastErrorMsg());
return std::nullopt; return std::nullopt;
} }
#ifdef _WIN32
std::string strDir = Common::UTF16ToUTF8(dir); return {std::move(path)};
#else
std::string strDir = dir;
#endif
free(dir);
return strDir;
} }
bool SetCurrentDir(const std::string& directory) { bool SetCurrentDir(const fs::path& path) {
#ifdef _WIN32 std::error_code ec;
return _wchdir(Common::UTF8ToUTF16W(directory).c_str()) == 0; fs::current_path(path, ec);
#else
return chdir(directory.c_str()) == 0; if (ec) {
#endif LOG_ERROR(Common_Filesystem, "Unable to set {} as working directory: {}", path.string(),
ec.message());
return false;
}
return true;
} }
#if defined(__APPLE__) #if defined(__APPLE__)

View File

@ -39,48 +39,34 @@ enum class UserPath {
UserDir, UserDir,
}; };
// FileSystem tree node/ // Returns true if the path exists
struct FSTEntry {
bool isDirectory;
u64 size; // file length or number of entries from children
std::string physicalName; // name on disk
std::string virtualName; // name in FST names table
std::vector<FSTEntry> children;
};
// Returns true if the exists
[[nodiscard]] bool Exists(const std::filesystem::path& path); [[nodiscard]] bool Exists(const std::filesystem::path& path);
// Returns true if path is a directory // Returns true if path is a directory
[[nodiscard]] bool IsDirectory(const std::filesystem::path& path); [[nodiscard]] bool IsDirectory(const std::filesystem::path& path);
// Returns the size of filename (64bit) // Returns the size of filename (64bit)
[[nodiscard]] u64 GetSize(const std::string& filename); [[nodiscard]] u64 GetSize(const std::filesystem::path& path);
// Overloaded GetSize, accepts file descriptor
[[nodiscard]] u64 GetSize(int fd);
// Overloaded GetSize, accepts FILE* // Overloaded GetSize, accepts FILE*
[[nodiscard]] u64 GetSize(FILE* f); [[nodiscard]] u64 GetSize(FILE* f);
// Returns true if successful, or path already exists. // Returns true if successful, or path already exists.
bool CreateDir(const std::string& filename); bool CreateDir(const std::filesystem::path& path);
// Creates the full path of fullPath returns true on success // Creates the full path of path. Returns true on success
bool CreateFullPath(const std::string& fullPath); bool CreateFullPath(const std::filesystem::path& path);
// Deletes a given filename, return true on success // Deletes a given file at the path.
// Doesn't supports deleting a directory // This will also delete empty directories.
bool Delete(const std::string& filename); // Return true on success
bool Delete(const std::filesystem::path& path);
// Deletes a directory filename, returns true on success // Renames file src to dst, returns true on success
bool DeleteDir(const std::string& filename); bool Rename(const std::filesystem::path& src, const std::filesystem::path& dst);
// renames file srcFilename to destFilename, returns true on success // copies file src to dst, returns true on success
bool Rename(const std::string& srcFilename, const std::string& destFilename); bool Copy(const std::filesystem::path& src, const std::filesystem::path& dst);
// copies file srcFilename to destFilename, returns true on success
bool Copy(const std::string& srcFilename, const std::string& destFilename);
// creates an empty file filename, returns true on success // creates an empty file filename, returns true on success
bool CreateEmptyFile(const std::string& filename); bool CreateEmptyFile(const std::string& filename);
@ -107,27 +93,17 @@ using DirectoryEntryCallable = std::function<bool(
bool ForeachDirectoryEntry(u64* num_entries_out, const std::string& directory, bool ForeachDirectoryEntry(u64* num_entries_out, const std::string& directory,
DirectoryEntryCallable callback); DirectoryEntryCallable callback);
/** // Deletes the given path and anything under it. Returns true on success.
* Scans the directory tree, storing the results. bool DeleteDirRecursively(const std::filesystem::path& path);
* @param directory the parent directory to start scanning from
* @param parent_entry FSTEntry where the filesystem tree results will be stored.
* @param recursion Number of children directories to read before giving up.
* @return the total number of files/directories found
*/
u64 ScanDirectoryTree(const std::string& directory, FSTEntry& parent_entry,
unsigned int recursion = 0);
// deletes the given directory and anything under it. Returns true on success.
bool DeleteDirRecursively(const std::string& directory, unsigned int recursion = 256);
// Returns the current directory // Returns the current directory
[[nodiscard]] std::optional<std::string> GetCurrentDir(); [[nodiscard]] std::optional<std::filesystem::path> GetCurrentDir();
// Create directory and copy contents (does not overwrite existing files) // Create directory and copy contents (does not overwrite existing files)
void CopyDir(const std::string& source_path, const std::string& dest_path); void CopyDir(const std::filesystem::path& src, const std::filesystem::path& dst);
// Set the current directory to given directory // Set the current directory to given path
bool SetCurrentDir(const std::string& directory); bool SetCurrentDir(const std::filesystem::path& path);
// Returns a pointer to a string with a yuzu data dir in the user's home // Returns a pointer to a string with a yuzu data dir in the user's home
// directory. To be used in "multi-user" mode (that is, installed). // directory. To be used in "multi-user" mode (that is, installed).