mirror of
https://github.com/RPCS3/rpcs3.git
synced 2024-11-22 02:32:36 +01:00
1290 lines
26 KiB
C++
1290 lines
26 KiB
C++
#include "File.h"
|
|
#include "StrFmt.h"
|
|
#include "SharedMutex.h"
|
|
#include "BEType.h"
|
|
#include "Crypto/sha1.h"
|
|
|
|
#include <unordered_map>
|
|
#include <algorithm>
|
|
#include <cerrno>
|
|
|
|
#ifdef _WIN32
|
|
|
|
#include <cwchar>
|
|
#include <Windows.h>
|
|
|
|
static std::unique_ptr<wchar_t[]> to_wchar(const std::string& source)
|
|
{
|
|
// String size + null terminator
|
|
const std::size_t buf_size = source.size() + 1;
|
|
|
|
// Safe size
|
|
const int size = narrow<int>(buf_size, "to_wchar" HERE);
|
|
|
|
// Buffer for max possible output length
|
|
std::unique_ptr<wchar_t[]> buffer(new wchar_t[buf_size]);
|
|
|
|
verify("to_wchar" HERE), MultiByteToWideChar(CP_UTF8, 0, source.c_str(), size, buffer.get(), size);
|
|
|
|
return buffer;
|
|
}
|
|
|
|
static void to_utf8(std::string& out, const wchar_t* source)
|
|
{
|
|
// String size
|
|
const std::size_t length = std::wcslen(source);
|
|
|
|
// Safe buffer size for max possible output length (including null terminator)
|
|
const int buf_size = narrow<int>(length * 3 + 1, "to_utf8" HERE);
|
|
|
|
// Resize buffer
|
|
out.resize(buf_size - 1);
|
|
|
|
const int result = WideCharToMultiByte(CP_UTF8, 0, source, static_cast<int>(length) + 1, &out.front(), buf_size, NULL, NULL);
|
|
|
|
// Fix the size
|
|
out.resize(verify("to_utf8" HERE, result) - 1);
|
|
}
|
|
|
|
static time_t to_time(const ULARGE_INTEGER& ft)
|
|
{
|
|
return ft.QuadPart / 10000000ULL - 11644473600ULL;
|
|
}
|
|
|
|
static time_t to_time(const LARGE_INTEGER& ft)
|
|
{
|
|
ULARGE_INTEGER v;
|
|
v.LowPart = ft.LowPart;
|
|
v.HighPart = ft.HighPart;
|
|
|
|
return to_time(v);
|
|
}
|
|
|
|
static time_t to_time(const FILETIME& ft)
|
|
{
|
|
ULARGE_INTEGER v;
|
|
v.LowPart = ft.dwLowDateTime;
|
|
v.HighPart = ft.dwHighDateTime;
|
|
|
|
return to_time(v);
|
|
}
|
|
|
|
static fs::error to_error(DWORD e)
|
|
{
|
|
switch (e)
|
|
{
|
|
case ERROR_FILE_NOT_FOUND: return fs::error::noent;
|
|
case ERROR_PATH_NOT_FOUND: return fs::error::noent;
|
|
case ERROR_ALREADY_EXISTS: return fs::error::exist;
|
|
case ERROR_FILE_EXISTS: return fs::error::exist;
|
|
case ERROR_NEGATIVE_SEEK: return fs::error::inval;
|
|
case ERROR_DIRECTORY: return fs::error::inval;
|
|
case ERROR_INVALID_NAME: return fs::error::inval;
|
|
default: fmt::throw_exception("Unknown Win32 error: %u.", e);
|
|
}
|
|
}
|
|
|
|
#else
|
|
|
|
#include <sys/mman.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <dirent.h>
|
|
#include <fcntl.h>
|
|
#include <libgen.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#if defined(__APPLE__) || defined(__FreeBSD__)
|
|
#include <copyfile.h>
|
|
#include <mach-o/dyld.h>
|
|
#else
|
|
#include <sys/sendfile.h>
|
|
#endif
|
|
|
|
static fs::error to_error(int e)
|
|
{
|
|
switch (e)
|
|
{
|
|
case ENOENT: return fs::error::noent;
|
|
case EEXIST: return fs::error::exist;
|
|
case EINVAL: return fs::error::inval;
|
|
default: fmt::throw_exception("Unknown system error: %d.", e);
|
|
}
|
|
}
|
|
|
|
#endif
|
|
|
|
namespace fs
|
|
{
|
|
thread_local error g_tls_error = error::ok;
|
|
|
|
class device_manager final
|
|
{
|
|
mutable shared_mutex m_mutex;
|
|
|
|
std::unordered_map<std::string, std::shared_ptr<device_base>> m_map;
|
|
|
|
public:
|
|
std::shared_ptr<device_base> get_device(const std::string& path);
|
|
std::shared_ptr<device_base> set_device(const std::string& name, const std::shared_ptr<device_base>&);
|
|
};
|
|
|
|
static device_manager& get_device_manager()
|
|
{
|
|
// Use magic static
|
|
static device_manager instance;
|
|
return instance;
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<fs::device_base> fs::device_manager::get_device(const std::string& path)
|
|
{
|
|
reader_lock lock(m_mutex);
|
|
|
|
const auto found = m_map.find(path.substr(0, path.find_first_of('/', 2)));
|
|
|
|
if (found == m_map.end())
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
return found->second;
|
|
}
|
|
|
|
std::shared_ptr<fs::device_base> fs::device_manager::set_device(const std::string& name, const std::shared_ptr<device_base>& device)
|
|
{
|
|
writer_lock lock(m_mutex);
|
|
|
|
return m_map[name] = device;
|
|
}
|
|
|
|
std::shared_ptr<fs::device_base> fs::get_virtual_device(const std::string& path)
|
|
{
|
|
// Every virtual device path must have "//" at the beginning
|
|
if (path.size() > 2 && reinterpret_cast<const u16&>(path.front()) == "//"_u16)
|
|
{
|
|
return get_device_manager().get_device(path);
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
std::shared_ptr<fs::device_base> fs::set_virtual_device(const std::string& name, const std::shared_ptr<device_base>& device)
|
|
{
|
|
verify(HERE), name.size() > 2, name[0] == '/', name[1] == '/', name.find('/', 2) == -1;
|
|
|
|
return get_device_manager().set_device(name, device);
|
|
}
|
|
|
|
std::string fs::get_parent_dir(const std::string& path)
|
|
{
|
|
// Search upper bound (set to the last character, npos for empty string)
|
|
auto last = path.size() - 1;
|
|
|
|
#ifdef _WIN32
|
|
const auto& delim = "/\\";
|
|
#else
|
|
const auto& delim = "/";
|
|
#endif
|
|
|
|
while (true)
|
|
{
|
|
const auto pos = path.find_last_of(delim, last, sizeof(delim) - 1);
|
|
|
|
// Contiguous slashes are ignored at the end
|
|
if (std::exchange(last, pos - 1) != pos)
|
|
{
|
|
// Return empty string if the path doesn't contain at least 2 elements
|
|
return path.substr(0, pos != -1 && path.find_last_not_of(delim, pos, sizeof(delim) - 1) != -1 ? pos : 0);
|
|
}
|
|
}
|
|
}
|
|
|
|
bool fs::stat(const std::string& path, stat_t& info)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
return device->stat(path, info);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
WIN32_FILE_ATTRIBUTE_DATA attrs;
|
|
if (!GetFileAttributesExW(to_wchar(path).get(), GetFileExInfoStandard, &attrs))
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
info.is_directory = (attrs.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;
|
|
info.is_writable = (attrs.dwFileAttributes & FILE_ATTRIBUTE_READONLY) == 0;
|
|
info.size = (u64)attrs.nFileSizeLow | ((u64)attrs.nFileSizeHigh << 32);
|
|
info.atime = to_time(attrs.ftLastAccessTime);
|
|
info.mtime = to_time(attrs.ftLastWriteTime);
|
|
info.ctime = to_time(attrs.ftCreationTime);
|
|
#else
|
|
struct ::stat file_info;
|
|
if (::stat(path.c_str(), &file_info) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
info.is_directory = S_ISDIR(file_info.st_mode);
|
|
info.is_writable = file_info.st_mode & 0200; // HACK: approximation
|
|
info.size = file_info.st_size;
|
|
info.atime = file_info.st_atime;
|
|
info.mtime = file_info.st_mtime;
|
|
info.ctime = file_info.st_ctime;
|
|
#endif
|
|
|
|
return true;
|
|
}
|
|
|
|
bool fs::exists(const std::string& path)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
stat_t info;
|
|
return device->stat(path, info);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
if (GetFileAttributesW(to_wchar(path).get()) == INVALID_FILE_ATTRIBUTES)
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#else
|
|
struct ::stat file_info;
|
|
if (::stat(path.c_str(), &file_info) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool fs::is_file(const std::string& path)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
stat_t info;
|
|
if (!device->stat(path, info))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (info.is_directory)
|
|
{
|
|
g_tls_error = error::exist;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
const DWORD attrs = GetFileAttributesW(to_wchar(path).get());
|
|
if (attrs == INVALID_FILE_ATTRIBUTES)
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
#else
|
|
struct ::stat file_info;
|
|
if (::stat(path.c_str(), &file_info) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
#endif
|
|
|
|
// TODO: correct file type check
|
|
#ifdef _WIN32
|
|
if ((attrs & FILE_ATTRIBUTE_DIRECTORY) != 0)
|
|
#else
|
|
if (S_ISDIR(file_info.st_mode))
|
|
#endif
|
|
{
|
|
g_tls_error = error::exist;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool fs::is_dir(const std::string& path)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
stat_t info;
|
|
if (!device->stat(path, info))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (info.is_directory == false)
|
|
{
|
|
g_tls_error = error::exist;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
const DWORD attrs = GetFileAttributesW(to_wchar(path).get());
|
|
if (attrs == INVALID_FILE_ATTRIBUTES)
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
#else
|
|
struct ::stat file_info;
|
|
if (::stat(path.c_str(), &file_info) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
if ((attrs & FILE_ATTRIBUTE_DIRECTORY) == 0)
|
|
#else
|
|
if (!S_ISDIR(file_info.st_mode))
|
|
#endif
|
|
{
|
|
g_tls_error = error::exist;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool fs::create_dir(const std::string& path)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
return device->create_dir(path);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
if (!CreateDirectoryW(to_wchar(path).get(), NULL))
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#else
|
|
if (::mkdir(path.c_str(), S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool fs::create_path(const std::string& path)
|
|
{
|
|
const auto& parent = get_parent_dir(path);
|
|
|
|
if (!parent.empty() && !is_dir(parent) && !create_path(parent))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return create_dir(path);
|
|
}
|
|
|
|
bool fs::remove_dir(const std::string& path)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
return device->remove_dir(path);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
if (!RemoveDirectoryW(to_wchar(path).get()))
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#else
|
|
if (::rmdir(path.c_str()) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool fs::rename(const std::string& from, const std::string& to)
|
|
{
|
|
const auto device = get_virtual_device(from);
|
|
|
|
if (device != get_virtual_device(to))
|
|
{
|
|
fmt::throw_exception("fs::rename() between different devices not implemented.\nFrom: %s\nTo: %s", from, to);
|
|
}
|
|
|
|
if (device)
|
|
{
|
|
return device->rename(from, to);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
if (!MoveFileW(to_wchar(from).get(), to_wchar(to).get()))
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#else
|
|
if (::rename(from.c_str(), to.c_str()) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool fs::copy_file(const std::string& from, const std::string& to, bool overwrite)
|
|
{
|
|
const auto device = get_virtual_device(from);
|
|
|
|
if (device != get_virtual_device(to) || device) // TODO
|
|
{
|
|
fmt::throw_exception("fs::copy_file() for virtual devices not implemented.\nFrom: %s\nTo: %s", from, to);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
if (!CopyFileW(to_wchar(from).get(), to_wchar(to).get(), !overwrite))
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#else
|
|
/* Source: http://stackoverflow.com/questions/2180079/how-can-i-copy-a-file-on-unix-using-c */
|
|
|
|
const int input = ::open(from.c_str(), O_RDONLY);
|
|
if (input == -1)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
const int output = ::open(to.c_str(), O_WRONLY | O_CREAT | (overwrite ? O_TRUNC : O_EXCL), S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
|
|
if (output == -1)
|
|
{
|
|
const int err = errno;
|
|
|
|
::close(input);
|
|
g_tls_error = to_error(err);
|
|
return false;
|
|
}
|
|
|
|
// Here we use kernel-space copying for performance reasons
|
|
#if defined(__APPLE__) || defined(__FreeBSD__)
|
|
// fcopyfile works on FreeBSD and OS X 10.5+
|
|
if (::fcopyfile(input, output, 0, COPYFILE_ALL))
|
|
#else
|
|
// sendfile will work with non-socket output (i.e. regular file) on Linux 2.6.33+
|
|
off_t bytes_copied = 0;
|
|
struct ::stat fileinfo = { 0 };
|
|
if (::fstat(input, &fileinfo) || ::sendfile(output, input, &bytes_copied, fileinfo.st_size))
|
|
#endif
|
|
{
|
|
const int err = errno;
|
|
|
|
::close(input);
|
|
::close(output);
|
|
g_tls_error = to_error(err);
|
|
return false;
|
|
}
|
|
|
|
::close(input);
|
|
::close(output);
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool fs::remove_file(const std::string& path)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
return device->remove(path);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
if (!DeleteFileW(to_wchar(path).get()))
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#else
|
|
if (::unlink(path.c_str()) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool fs::truncate_file(const std::string& path, u64 length)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
return device->trunc(path, length);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
// Open the file
|
|
const auto handle = CreateFileW(to_wchar(path).get(), GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (handle == INVALID_HANDLE_VALUE)
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
LARGE_INTEGER distance;
|
|
distance.QuadPart = length;
|
|
|
|
// Seek and truncate
|
|
if (!SetFilePointerEx(handle, distance, NULL, FILE_BEGIN) || !SetEndOfFile(handle))
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
CloseHandle(handle);
|
|
return false;
|
|
}
|
|
|
|
CloseHandle(handle);
|
|
return true;
|
|
#else
|
|
if (::truncate(path.c_str(), length) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
void fs::file::xnull() const
|
|
{
|
|
fmt::throw_exception<std::logic_error>("fs::file is null");
|
|
}
|
|
|
|
void fs::file::xfail() const
|
|
{
|
|
fmt::throw_exception("Unexpected fs::error %s", g_tls_error);
|
|
}
|
|
|
|
bool fs::file::open(const std::string& path, bs_t<open_mode> mode)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
if (auto&& _file = device->open(path, mode))
|
|
{
|
|
m_file = std::move(_file);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
DWORD access = 0;
|
|
if (test(mode & fs::read)) access |= GENERIC_READ;
|
|
if (test(mode & fs::write)) access |= test(mode & fs::append) ? FILE_APPEND_DATA : GENERIC_WRITE;
|
|
|
|
DWORD disp = 0;
|
|
if (test(mode & fs::create))
|
|
{
|
|
disp =
|
|
test(mode & fs::excl) ? CREATE_NEW :
|
|
test(mode & fs::trunc) ? CREATE_ALWAYS : OPEN_ALWAYS;
|
|
}
|
|
else
|
|
{
|
|
if (test(mode & fs::excl))
|
|
{
|
|
g_tls_error = error::inval;
|
|
return false;
|
|
}
|
|
|
|
disp = test(mode & fs::trunc) ? TRUNCATE_EXISTING : OPEN_EXISTING;
|
|
}
|
|
|
|
const HANDLE handle = CreateFileW(to_wchar(path).get(), access, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, disp, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE)
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
class windows_file final : public file_base
|
|
{
|
|
const HANDLE m_handle;
|
|
|
|
public:
|
|
windows_file(HANDLE handle)
|
|
: m_handle(handle)
|
|
{
|
|
}
|
|
|
|
~windows_file() override
|
|
{
|
|
CloseHandle(m_handle);
|
|
}
|
|
|
|
stat_t stat() override
|
|
{
|
|
FILE_BASIC_INFO basic_info;
|
|
verify("file::stat" HERE), GetFileInformationByHandleEx(m_handle, FileBasicInfo, &basic_info, sizeof(FILE_BASIC_INFO));
|
|
|
|
stat_t info;
|
|
info.is_directory = (basic_info.FileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;
|
|
info.is_writable = (basic_info.FileAttributes & FILE_ATTRIBUTE_READONLY) == 0;
|
|
info.size = this->size();
|
|
info.atime = to_time(basic_info.LastAccessTime);
|
|
info.mtime = to_time(basic_info.ChangeTime);
|
|
info.ctime = to_time(basic_info.CreationTime);
|
|
|
|
return info;
|
|
}
|
|
|
|
bool trunc(u64 length) override
|
|
{
|
|
LARGE_INTEGER old, pos;
|
|
|
|
pos.QuadPart = 0;
|
|
if (!SetFilePointerEx(m_handle, pos, &old, FILE_CURRENT)) // get old position
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
pos.QuadPart = length;
|
|
if (!SetFilePointerEx(m_handle, pos, NULL, FILE_BEGIN)) // set new position
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
verify("file::trunc" HERE), SetEndOfFile(m_handle), SetFilePointerEx(m_handle, old, NULL, FILE_BEGIN);
|
|
return true;
|
|
}
|
|
|
|
u64 read(void* buffer, u64 count) override
|
|
{
|
|
// TODO (call ReadFile multiple times if count is too big)
|
|
const int size = narrow<int>(count, "file::read" HERE);
|
|
|
|
DWORD nread;
|
|
verify("file::read" HERE), ReadFile(m_handle, buffer, size, &nread, NULL);
|
|
|
|
return nread;
|
|
}
|
|
|
|
u64 write(const void* buffer, u64 count) override
|
|
{
|
|
// TODO (call WriteFile multiple times if count is too big)
|
|
const int size = narrow<int>(count, "file::write" HERE);
|
|
|
|
DWORD nwritten;
|
|
verify("file::write" HERE), WriteFile(m_handle, buffer, size, &nwritten, NULL);
|
|
|
|
return nwritten;
|
|
}
|
|
|
|
u64 seek(s64 offset, seek_mode whence) override
|
|
{
|
|
LARGE_INTEGER pos;
|
|
pos.QuadPart = offset;
|
|
|
|
const DWORD mode =
|
|
whence == seek_set ? FILE_BEGIN :
|
|
whence == seek_cur ? FILE_CURRENT :
|
|
whence == seek_end ? FILE_END :
|
|
(fmt::throw_exception("Invalid whence (0x%x)" HERE, whence), 0);
|
|
|
|
verify("file::seek" HERE), SetFilePointerEx(m_handle, pos, &pos, mode);
|
|
|
|
return pos.QuadPart;
|
|
}
|
|
|
|
u64 size() override
|
|
{
|
|
LARGE_INTEGER size;
|
|
verify("file::size" HERE), GetFileSizeEx(m_handle, &size);
|
|
|
|
return size.QuadPart;
|
|
}
|
|
};
|
|
|
|
m_file = std::make_unique<windows_file>(handle);
|
|
#else
|
|
int flags = 0;
|
|
|
|
if (test(mode & fs::read) && test(mode & fs::write)) flags |= O_RDWR;
|
|
else if (test(mode & fs::read)) flags |= O_RDONLY;
|
|
else if (test(mode & fs::write)) flags |= O_WRONLY;
|
|
|
|
if (test(mode & fs::append)) flags |= O_APPEND;
|
|
if (test(mode & fs::create)) flags |= O_CREAT;
|
|
if (test(mode & fs::trunc)) flags |= O_TRUNC;
|
|
if (test(mode & fs::excl)) flags |= O_EXCL;
|
|
|
|
const int fd = ::open(path.c_str(), flags, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
|
|
|
|
if (fd == -1)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
class unix_file final : public file_base
|
|
{
|
|
const int m_fd;
|
|
|
|
public:
|
|
unix_file(int fd)
|
|
: m_fd(fd)
|
|
{
|
|
}
|
|
|
|
~unix_file() override
|
|
{
|
|
::close(m_fd);
|
|
}
|
|
|
|
stat_t stat() override
|
|
{
|
|
struct ::stat file_info;
|
|
verify("file::stat" HERE), ::fstat(m_fd, &file_info) == 0;
|
|
|
|
stat_t info;
|
|
info.is_directory = S_ISDIR(file_info.st_mode);
|
|
info.is_writable = file_info.st_mode & 0200; // HACK: approximation
|
|
info.size = file_info.st_size;
|
|
info.atime = file_info.st_atime;
|
|
info.mtime = file_info.st_mtime;
|
|
info.ctime = file_info.st_ctime;
|
|
|
|
return info;
|
|
}
|
|
|
|
bool trunc(u64 length) override
|
|
{
|
|
if (::ftruncate(m_fd, length) != 0)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
u64 read(void* buffer, u64 count) override
|
|
{
|
|
const auto result = ::read(m_fd, buffer, count);
|
|
verify("file::read" HERE), result != -1;
|
|
|
|
return result;
|
|
}
|
|
|
|
u64 write(const void* buffer, u64 count) override
|
|
{
|
|
const auto result = ::write(m_fd, buffer, count);
|
|
verify("file::write" HERE), result != -1;
|
|
|
|
return result;
|
|
}
|
|
|
|
u64 seek(s64 offset, seek_mode whence) override
|
|
{
|
|
const int mode =
|
|
whence == seek_set ? SEEK_SET :
|
|
whence == seek_cur ? SEEK_CUR :
|
|
whence == seek_end ? SEEK_END :
|
|
(fmt::throw_exception("Invalid whence (0x%x)" HERE, whence), 0);
|
|
|
|
const auto result = ::lseek(m_fd, offset, mode);
|
|
verify("file::seek" HERE), result != -1;
|
|
|
|
return result;
|
|
}
|
|
|
|
u64 size() override
|
|
{
|
|
struct ::stat file_info;
|
|
verify("file::size" HERE), ::fstat(m_fd, &file_info) == 0;
|
|
|
|
return file_info.st_size;
|
|
}
|
|
};
|
|
|
|
m_file = std::make_unique<unix_file>(fd);
|
|
#endif
|
|
|
|
return true;
|
|
}
|
|
|
|
fs::file::file(const void* ptr, std::size_t size)
|
|
{
|
|
class memory_stream : public file_base
|
|
{
|
|
u64 m_pos{};
|
|
|
|
const char* const m_ptr;
|
|
const u64 m_size;
|
|
|
|
public:
|
|
memory_stream(const void* ptr, u64 size)
|
|
: m_ptr(static_cast<const char*>(ptr))
|
|
, m_size(size)
|
|
{
|
|
}
|
|
|
|
fs::stat_t stat() override
|
|
{
|
|
fmt::throw_exception<std::logic_error>("Not supported" HERE);
|
|
}
|
|
|
|
bool trunc(u64 length) override
|
|
{
|
|
fmt::throw_exception<std::logic_error>("Not allowed" HERE);
|
|
}
|
|
|
|
u64 read(void* buffer, u64 count) override
|
|
{
|
|
const u64 start = m_pos;
|
|
const u64 end = seek(count, fs::seek_cur);
|
|
if (end < start) fmt::throw_exception<std::logic_error>("Stream overflow" HERE);
|
|
const u64 read_size = end - start;
|
|
std::memcpy(buffer, m_ptr + start, read_size);
|
|
return read_size;
|
|
}
|
|
|
|
u64 write(const void* buffer, u64 count) override
|
|
{
|
|
fmt::throw_exception<std::logic_error>("Not allowed" HERE);
|
|
}
|
|
|
|
u64 seek(s64 offset, fs::seek_mode whence) override
|
|
{
|
|
return
|
|
whence == fs::seek_set ? m_pos = std::min<u64>(offset, m_size) :
|
|
whence == fs::seek_cur ? m_pos = std::min<u64>(offset + m_pos, m_size) :
|
|
whence == fs::seek_end ? m_pos = std::min<u64>(offset + m_size, m_size) :
|
|
(fmt::throw_exception("Invalid whence (0x%x)" HERE, whence), 0);
|
|
}
|
|
|
|
u64 size() override
|
|
{
|
|
return m_size;
|
|
}
|
|
};
|
|
|
|
m_file = std::make_unique<memory_stream>(ptr, size);
|
|
}
|
|
|
|
void fs::dir::xnull() const
|
|
{
|
|
fmt::throw_exception<std::logic_error>("fs::dir is null");
|
|
}
|
|
|
|
bool fs::dir::open(const std::string& path)
|
|
{
|
|
if (auto device = get_virtual_device(path))
|
|
{
|
|
if (auto&& _dir = device->open_dir(path))
|
|
{
|
|
m_dir = std::move(_dir);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
WIN32_FIND_DATAW found;
|
|
const auto handle = FindFirstFileExW(to_wchar(path + "/*").get(), FindExInfoBasic, &found, FindExSearchNameMatch, NULL, FIND_FIRST_EX_CASE_SENSITIVE | FIND_FIRST_EX_LARGE_FETCH);
|
|
|
|
if (handle == INVALID_HANDLE_VALUE)
|
|
{
|
|
g_tls_error = to_error(GetLastError());
|
|
return false;
|
|
}
|
|
|
|
class windows_dir final : public dir_base
|
|
{
|
|
std::vector<dir_entry> m_entries;
|
|
std::size_t m_pos = 0;
|
|
|
|
void add_entry(const WIN32_FIND_DATAW& found)
|
|
{
|
|
dir_entry info;
|
|
|
|
to_utf8(info.name, found.cFileName);
|
|
info.is_directory = (found.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;
|
|
info.is_writable = (found.dwFileAttributes & FILE_ATTRIBUTE_READONLY) == 0;
|
|
info.size = ((u64)found.nFileSizeHigh << 32) | (u64)found.nFileSizeLow;
|
|
info.atime = to_time(found.ftLastAccessTime);
|
|
info.mtime = to_time(found.ftLastWriteTime);
|
|
info.ctime = to_time(found.ftCreationTime);
|
|
|
|
m_entries.emplace_back(std::move(info));
|
|
}
|
|
|
|
public:
|
|
windows_dir(HANDLE handle, WIN32_FIND_DATAW& found)
|
|
{
|
|
add_entry(found);
|
|
|
|
while (FindNextFileW(handle, &found))
|
|
{
|
|
add_entry(found);
|
|
}
|
|
|
|
verify("dir::read" HERE), ERROR_NO_MORE_FILES == GetLastError();
|
|
FindClose(handle);
|
|
}
|
|
|
|
bool read(dir_entry& out) override
|
|
{
|
|
if (m_pos >= m_entries.size())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
out = m_entries[m_pos++];
|
|
return true;
|
|
}
|
|
|
|
void rewind() override
|
|
{
|
|
m_pos = 0;
|
|
}
|
|
};
|
|
|
|
m_dir = std::make_unique<windows_dir>(handle, found);
|
|
#else
|
|
::DIR* const ptr = ::opendir(path.c_str());
|
|
|
|
if (!ptr)
|
|
{
|
|
g_tls_error = to_error(errno);
|
|
return false;
|
|
}
|
|
|
|
class unix_dir final : public dir_base
|
|
{
|
|
::DIR* m_dd;
|
|
|
|
public:
|
|
unix_dir(::DIR* dd)
|
|
: m_dd(dd)
|
|
{
|
|
}
|
|
|
|
~unix_dir() override
|
|
{
|
|
::closedir(m_dd);
|
|
}
|
|
|
|
bool read(dir_entry& info) override
|
|
{
|
|
const auto found = ::readdir(m_dd);
|
|
if (!found)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
struct ::stat file_info;
|
|
verify("dir::read" HERE), ::fstatat(::dirfd(m_dd), found->d_name, &file_info, 0) == 0;
|
|
|
|
info.name = found->d_name;
|
|
info.is_directory = S_ISDIR(file_info.st_mode);
|
|
info.is_writable = file_info.st_mode & 0200; // HACK: approximation
|
|
info.size = file_info.st_size;
|
|
info.atime = file_info.st_atime;
|
|
info.mtime = file_info.st_mtime;
|
|
info.ctime = file_info.st_ctime;
|
|
|
|
return true;
|
|
}
|
|
|
|
void rewind() override
|
|
{
|
|
::rewinddir(m_dd);
|
|
}
|
|
};
|
|
|
|
m_dir = std::make_unique<unix_dir>(ptr);
|
|
#endif
|
|
|
|
return true;
|
|
}
|
|
|
|
const std::string& fs::get_config_dir()
|
|
{
|
|
// Use magic static
|
|
static const std::string s_dir = []
|
|
{
|
|
#ifdef _WIN32
|
|
return get_executable_dir(); // ?
|
|
#else
|
|
std::string dir;
|
|
|
|
if (const char* home = ::getenv("XDG_CONFIG_HOME"))
|
|
dir = home;
|
|
else if (const char* home = ::getenv("HOME"))
|
|
dir = home + "/.config"s;
|
|
else // Just in case
|
|
dir = "./config";
|
|
|
|
dir += "/rpcs3/";
|
|
|
|
if (!is_dir(dir) && !create_path(dir))
|
|
{
|
|
std::printf("Failed to create configuration directory '%s' (%d).\n", dir.c_str(), errno);
|
|
return get_executable_dir();
|
|
}
|
|
|
|
return dir;
|
|
#endif
|
|
}();
|
|
|
|
return s_dir;
|
|
}
|
|
|
|
const std::string& fs::get_executable_dir()
|
|
{
|
|
// Use magic static
|
|
static const std::string s_dir = []
|
|
{
|
|
std::string dir;
|
|
|
|
#ifdef _WIN32
|
|
wchar_t buf[2048];
|
|
if (GetModuleFileName(NULL, buf, ::size32(buf)) - 1 >= ::size32(buf) - 1)
|
|
{
|
|
MessageBoxA(0, fmt::format("GetModuleFileName() failed: error %u.", GetLastError()).c_str(), "fs::get_executable_dir()", MB_ICONERROR);
|
|
return dir; // empty
|
|
}
|
|
|
|
to_utf8(dir, buf); // Convert to UTF-8
|
|
|
|
std::replace(dir.begin(), dir.end(), '\\', '/');
|
|
|
|
#elif __APPLE__
|
|
char buf[4096];
|
|
u32 size = sizeof(buf);
|
|
if (_NSGetExecutablePath(buf, &size))
|
|
{
|
|
std::printf("_NSGetExecutablePath() failed (size=0x%x).\n", size);
|
|
return dir; // empty
|
|
}
|
|
|
|
dir = buf;
|
|
#else
|
|
char buf[4096];
|
|
const auto size = ::readlink("/proc/self/exe", buf, sizeof(buf));
|
|
if (size <= 0 || size >= sizeof(buf))
|
|
{
|
|
std::printf("readlink(/proc/self/exe) failed (%d).\n", errno);
|
|
return dir; // empty
|
|
}
|
|
|
|
dir.assign(buf, size);
|
|
#endif
|
|
|
|
// Leave only path
|
|
dir.resize(dir.rfind('/') + 1);
|
|
return dir;
|
|
}();
|
|
|
|
return s_dir;
|
|
}
|
|
|
|
std::string fs::get_data_dir(const std::string& prefix, const std::string& location, const std::string& suffix)
|
|
{
|
|
static const std::string s_dir = []
|
|
{
|
|
const std::string& dir = get_config_dir() + "/data/";
|
|
|
|
if (!is_dir(dir) && !create_path(dir))
|
|
{
|
|
return get_config_dir();
|
|
}
|
|
|
|
return dir;
|
|
}();
|
|
|
|
std::vector<u8> buf;
|
|
buf.reserve(location.size() + 1);
|
|
|
|
// Normalize location
|
|
for (char c : location)
|
|
{
|
|
#ifdef _WIN32
|
|
if (c == '/' || c == '\\')
|
|
#else
|
|
if (c == '/')
|
|
#endif
|
|
{
|
|
if (buf.empty() || buf.back() != '/')
|
|
{
|
|
buf.push_back('/');
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
buf.push_back(c);
|
|
}
|
|
|
|
// Calculate hash
|
|
u8 hash[20];
|
|
sha1(buf.data(), buf.size(), hash);
|
|
|
|
// Concatenate
|
|
std::string&& result = fmt::format("%s%s/%016llx%08x-%s/", s_dir, prefix, reinterpret_cast<be_t<u64>&>(hash[0]), reinterpret_cast<be_t<u32>&>(hash[8]), suffix);
|
|
|
|
if (!is_dir(result))
|
|
{
|
|
// Create dir if necessary
|
|
if (create_path(result))
|
|
{
|
|
// Acknowledge original location
|
|
file(result + ".location", rewrite).write(buf);
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
std::string fs::get_data_dir(const std::string& prefix, const std::string& path)
|
|
{
|
|
#ifdef _WIN32
|
|
const auto& delim = "/\\";
|
|
#else
|
|
const auto& delim = "/";
|
|
#endif
|
|
|
|
// Extract file name and location
|
|
const std::string& location = fs::get_parent_dir(path);
|
|
const std::size_t name_pos = path.find_first_not_of(delim, location.size());
|
|
|
|
return fs::get_data_dir(prefix, location, name_pos == -1 ? std::string{} : path.substr(name_pos));
|
|
}
|
|
|
|
void fs::remove_all(const std::string& path, bool remove_root)
|
|
{
|
|
for (const auto& entry : dir(path))
|
|
{
|
|
if (entry.name == "." || entry.name == "..")
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (entry.is_directory == false)
|
|
{
|
|
remove_file(path + '/' + entry.name);
|
|
}
|
|
|
|
if (entry.is_directory == true)
|
|
{
|
|
remove_all(path + '/' + entry.name);
|
|
}
|
|
}
|
|
|
|
if (remove_root)
|
|
{
|
|
remove_dir(path);
|
|
}
|
|
}
|
|
|
|
u64 fs::get_dir_size(const std::string& path)
|
|
{
|
|
u64 result = 0;
|
|
|
|
for (const auto entry : dir(path))
|
|
{
|
|
if (entry.name == "." || entry.name == "..")
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (entry.is_directory == false)
|
|
{
|
|
result += entry.size;
|
|
}
|
|
|
|
if (entry.is_directory == true)
|
|
{
|
|
result += get_dir_size(path + '/' + entry.name);
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
template<>
|
|
void fmt_class_string<fs::seek_mode>::format(std::string& out, u64 arg)
|
|
{
|
|
format_enum(out, arg, [](auto arg)
|
|
{
|
|
switch (arg)
|
|
{
|
|
STR_CASE(fs::seek_mode::seek_set);
|
|
STR_CASE(fs::seek_mode::seek_cur);
|
|
STR_CASE(fs::seek_mode::seek_end);
|
|
}
|
|
|
|
return unknown;
|
|
});
|
|
}
|
|
|
|
template<>
|
|
void fmt_class_string<fs::error>::format(std::string& out, u64 arg)
|
|
{
|
|
format_enum(out, arg, [](auto arg)
|
|
{
|
|
switch (arg)
|
|
{
|
|
case fs::error::ok: return "OK";
|
|
|
|
case fs::error::inval: return "Invalid arguments";
|
|
case fs::error::noent: return "Not found";
|
|
case fs::error::exist: return "Already exists";
|
|
}
|
|
|
|
return unknown;
|
|
});
|
|
}
|