mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-11-27 06:12:15 +00:00
341 lines
11 KiB
C++
341 lines
11 KiB
C++
/*
|
|
* Copyright (c) 2018-2020 Atmosphère-NX
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#include <exosphere.hpp>
|
|
#include "../fatfs/ff.h"
|
|
#include "fusee_fs_api.hpp"
|
|
|
|
namespace ams::fs {
|
|
|
|
static_assert(sizeof(DirectoryEntry) == sizeof(FILINFO));
|
|
|
|
namespace {
|
|
|
|
constexpr size_t MaxFiles = 8 + 64;
|
|
constexpr size_t MaxDirectories = 2;
|
|
|
|
constinit bool g_is_sd_mounted = false;
|
|
constinit bool g_is_sys_mounted = false;
|
|
|
|
alignas(0x10) constinit FATFS g_sd_fs = {};
|
|
alignas(0x10) constinit FATFS g_sys_fs = {};
|
|
|
|
alignas(0x10) constinit FIL g_files[MaxFiles] = {};
|
|
alignas(0x10) constinit DIR g_dirs[MaxDirectories] = {};
|
|
constinit bool g_files_opened[MaxFiles] = {};
|
|
constinit bool g_dirs_opened[MaxFiles] = {};
|
|
constinit int g_open_modes[MaxFiles] = {};
|
|
|
|
Result TranslateFatFsError(FRESULT res) {
|
|
switch (res) {
|
|
case FR_OK:
|
|
return ResultSuccess();
|
|
case FR_DISK_ERR:
|
|
return fs::ResultMmcAccessFailed();
|
|
case FR_INT_ERR:
|
|
return fs::ResultPreconditionViolation();
|
|
case FR_NOT_READY:
|
|
return fs::ResultMmcAccessFailed();
|
|
case FR_NO_FILE:
|
|
return fs::ResultPathNotFound();
|
|
case FR_NO_PATH:
|
|
return fs::ResultPathNotFound();
|
|
case FR_INVALID_NAME:
|
|
return fs::ResultInvalidPath();
|
|
case FR_DENIED:
|
|
return fs::ResultPermissionDenied();
|
|
case FR_EXIST:
|
|
return fs::ResultPathAlreadyExists();
|
|
case FR_INVALID_OBJECT:
|
|
return fs::ResultInvalidArgument();
|
|
case FR_WRITE_PROTECTED:
|
|
return fs::ResultWriteNotPermitted();
|
|
case FR_INVALID_DRIVE:
|
|
return fs::ResultInvalidMountName();
|
|
case FR_NOT_ENABLED:
|
|
return fs::ResultInvalidMountName(); /* BAD/TODO */
|
|
case FR_NO_FILESYSTEM:
|
|
return fs::ResultInvalidMountName(); /* BAD/TODO */
|
|
case FR_TIMEOUT:
|
|
return fs::ResultTargetLocked(); /* BAD/TODO */
|
|
case FR_LOCKED:
|
|
return fs::ResultTargetLocked();
|
|
case FR_NOT_ENOUGH_CORE:
|
|
return fs::ResultPreconditionViolation(); /* BAD/TODO */
|
|
case FR_TOO_MANY_OPEN_FILES:
|
|
return fs::ResultPreconditionViolation(); /* BAD/TODO */
|
|
case FR_INVALID_PARAMETER:
|
|
return fs::ResultInvalidArgument();
|
|
default:
|
|
return fs::ResultInternal();
|
|
}
|
|
}
|
|
|
|
int TranslateToFatFsMode(int mode) {
|
|
int fmode = FA_OPEN_EXISTING;
|
|
if ((mode & OpenMode_Read) != 0) {
|
|
fmode |= FA_READ;
|
|
}
|
|
if ((mode & OpenMode_Write) != 0) {
|
|
fmode |= FA_WRITE;
|
|
}
|
|
if ((mode & OpenMode_AllowAppend) != 0) {
|
|
fmode |= FA_OPEN_APPEND;
|
|
}
|
|
return fmode;
|
|
}
|
|
|
|
FIL *GetInternalFile(FileHandle handle) {
|
|
return static_cast<FIL *>(handle._handle);
|
|
}
|
|
|
|
DIR *GetInternalDirectory(DirectoryHandle handle) {
|
|
return static_cast<DIR *>(handle._handle);
|
|
}
|
|
|
|
ALWAYS_INLINE size_t GetFileIndex(FIL *fp) {
|
|
const size_t file_index = (fp - g_files);
|
|
AMS_ASSERT(file_index < MaxFiles);
|
|
return file_index;
|
|
}
|
|
|
|
ALWAYS_INLINE size_t GetDirectoryIndex(DIR *dp) {
|
|
const size_t dir_index = (dp - g_dirs);
|
|
AMS_ASSERT(dir_index < MaxDirectories);
|
|
return dir_index;
|
|
}
|
|
|
|
}
|
|
|
|
bool MountSdCard() {
|
|
AMS_ASSERT(!g_is_sd_mounted);
|
|
g_is_sd_mounted = f_mount(std::addressof(g_sd_fs), "sdmc:", 1) == FR_OK;
|
|
return g_is_sd_mounted;
|
|
}
|
|
|
|
void UnmountSdCard() {
|
|
AMS_ASSERT(g_is_sd_mounted);
|
|
f_unmount("sdmc:");
|
|
g_is_sd_mounted = false;
|
|
}
|
|
|
|
bool MountSystem() {
|
|
AMS_ASSERT(!g_is_sys_mounted);
|
|
g_is_sys_mounted = f_mount(std::addressof(g_sys_fs), "sys:", 1) == FR_OK;
|
|
return g_is_sys_mounted;
|
|
}
|
|
|
|
void UnmountSystem() {
|
|
AMS_ASSERT(g_is_sys_mounted);
|
|
f_unmount("sys:");
|
|
g_is_sys_mounted = false;
|
|
}
|
|
|
|
Result GetEntryType(DirectoryEntryType *out_entry_type, bool *out_archive, const char *path) {
|
|
/* Get the file info. */
|
|
FILINFO info;
|
|
R_TRY(TranslateFatFsError(f_stat(path, std::addressof(info))));
|
|
|
|
/* Handle the file. */
|
|
*out_entry_type = (info.fattrib & AM_DIR) ? DirectoryEntryType_Directory : DirectoryEntryType_File;
|
|
*out_archive = (info.fattrib & AM_ARC);
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result CreateFile(const char *path, s64 size) {
|
|
/* Create the file. */
|
|
FIL fp;
|
|
R_TRY(TranslateFatFsError(f_open(std::addressof(fp), path, FA_CREATE_NEW | FA_READ | FA_WRITE)));
|
|
|
|
/* Ensure that we close the file when we're done with it. */
|
|
ON_SCOPE_EXIT { f_close(std::addressof(fp)); };
|
|
|
|
/* Expand the file. */
|
|
R_TRY(TranslateFatFsError(f_expand(std::addressof(fp), size, 1)));
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result CreateDirectory(const char *path) {
|
|
return TranslateFatFsError(f_mkdir(path));
|
|
}
|
|
|
|
Result OpenFile(FileHandle *out_file, const char *path, int mode) {
|
|
/* Find a free file. */
|
|
for (size_t i = 0; i < MaxFiles; ++i) {
|
|
if (!g_files_opened[i]) {
|
|
/* Open the file. */
|
|
FIL *fp = std::addressof(g_files[i]);
|
|
R_TRY(TranslateFatFsError(f_open(fp, path, TranslateToFatFsMode(mode))));
|
|
|
|
/* Set the output. */
|
|
out_file->_handle = fp;
|
|
g_files_opened[i] = true;
|
|
g_open_modes[i] = mode;
|
|
return ResultSuccess();
|
|
}
|
|
}
|
|
return fs::ResultOpenCountLimit();
|
|
}
|
|
|
|
Result OpenDirectory(DirectoryHandle *out_dir, const char *path) {
|
|
/* Find a free directory. */
|
|
for (size_t i = 0; i < MaxDirectories; ++i) {
|
|
if (!g_dirs_opened[i]) {
|
|
/* Open the file. */
|
|
DIR *dp = std::addressof(g_dirs[i]);
|
|
R_TRY(TranslateFatFsError(f_opendir(dp, path)));
|
|
|
|
/* Set the output. */
|
|
out_dir->_handle = dp;
|
|
g_dirs_opened[i] = true;
|
|
return ResultSuccess();
|
|
}
|
|
}
|
|
return fs::ResultOpenCountLimit();
|
|
}
|
|
|
|
Result ReadDirectory(s64 *out_count, DirectoryEntry *out_entries, DirectoryHandle handle, s64 max_entries) {
|
|
DIR * const dp = GetInternalDirectory(handle);
|
|
|
|
s64 count = 0;
|
|
while (count < max_entries) {
|
|
R_TRY(TranslateFatFsError(f_readdir(dp, reinterpret_cast<FILINFO *>(out_entries + count))));
|
|
|
|
if (out_entries[count].file_name[0] == '\x00') {
|
|
break;
|
|
}
|
|
|
|
++count;
|
|
}
|
|
|
|
*out_count = count;
|
|
return ResultSuccess();
|
|
}
|
|
|
|
void CloseDirectory(DirectoryHandle handle) {
|
|
const size_t index = GetDirectoryIndex(GetInternalDirectory(handle));
|
|
f_closedir(std::addressof(g_dirs[index]));
|
|
g_dirs_opened[index] = false;
|
|
}
|
|
|
|
Result ReadFile(FileHandle handle, s64 offset, void *buffer, size_t size, const fs::ReadOption &option) {
|
|
/* Option is unused. */
|
|
AMS_UNUSED(option);
|
|
|
|
/* Seek to the offset we're reading at. */
|
|
R_TRY(TranslateFatFsError(f_lseek(GetInternalFile(handle), offset)));
|
|
|
|
/* Read the data. */
|
|
UINT br;
|
|
R_TRY(TranslateFatFsError(f_read(GetInternalFile(handle), buffer, size, std::addressof(br))));
|
|
|
|
/* Check that we read the correct amount. */
|
|
R_UNLESS(br == size, fs::ResultOutOfRange());
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ReadFile(FileHandle handle, s64 offset, void *buffer, size_t size) {
|
|
return ReadFile(handle, offset, buffer, size, fs::ReadOption::None);
|
|
}
|
|
|
|
Result ReadFile(size_t *out, FileHandle handle, s64 offset, void *buffer, size_t size, const fs::ReadOption &option) {
|
|
/* Option is unused. */
|
|
AMS_UNUSED(option);
|
|
|
|
/* Seek to the offset we're reading at. */
|
|
R_TRY(TranslateFatFsError(f_lseek(GetInternalFile(handle), offset)));
|
|
|
|
/* Read the data. */
|
|
UINT br;
|
|
R_TRY(TranslateFatFsError(f_read(GetInternalFile(handle), buffer, size, std::addressof(br))));
|
|
|
|
/* Set the output size. */
|
|
*out = br;
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ReadFile(size_t *out, FileHandle handle, s64 offset, void *buffer, size_t size) {
|
|
return ReadFile(out, handle, offset, buffer, size, fs::ReadOption::None);
|
|
}
|
|
|
|
Result GetFileSize(s64 *out, FileHandle handle) {
|
|
FIL *fp = GetInternalFile(handle);
|
|
*out = f_size(fp);
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result FlushFile(FileHandle handle) {
|
|
return TranslateFatFsError(f_sync(GetInternalFile(handle)));
|
|
}
|
|
|
|
Result WriteFile(FileHandle handle, s64 offset, const void *buffer, size_t size, const fs::WriteOption &option) {
|
|
/* Seek to the offset we're writing at. */
|
|
R_TRY(TranslateFatFsError(f_lseek(GetInternalFile(handle), offset)));
|
|
|
|
/* Write the data. */
|
|
UINT bw;
|
|
R_TRY(TranslateFatFsError(f_write(GetInternalFile(handle), buffer, size, std::addressof(bw))));
|
|
|
|
/* Check that we wrote the correct amount. */
|
|
R_UNLESS(bw == size, fs::ResultOutOfRange());
|
|
|
|
/* If we should, flush the file. */
|
|
if (option.HasFlushFlag()) {
|
|
R_TRY(FlushFile(handle));
|
|
}
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result SetFileSize(FileHandle handle, s64 size) {
|
|
FIL *fp = GetInternalFile(handle);
|
|
|
|
/* Check if we have nothing to do. */
|
|
const size_t fsize = f_size(fp);
|
|
R_SUCCEED_IF(static_cast<FSIZE_t>(size) == fsize);
|
|
|
|
/* NOTE/TODO: This may not preserve file data. Do this in a way that does? */
|
|
|
|
/* Truncate the file. */
|
|
R_TRY(TranslateFatFsError(f_truncate(fp)));
|
|
|
|
/* Expand the file. */
|
|
R_TRY(TranslateFatFsError(f_expand(fp, size, 1)));
|
|
|
|
/* Ensure the file is synchronized. */
|
|
R_TRY(FlushFile(handle));
|
|
|
|
/* Check that our expansion succeeded. */
|
|
AMS_ASSERT(f_size(fp) == static_cast<FSIZE_t>(size));
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
int GetFileOpenMode(FileHandle handle) {
|
|
return g_open_modes[GetFileIndex(GetInternalFile(handle))];
|
|
}
|
|
|
|
void CloseFile(FileHandle handle) {
|
|
const size_t index = GetFileIndex(GetInternalFile(handle));
|
|
f_close(std::addressof(g_files[index]));
|
|
g_open_modes[index] = 0;
|
|
g_files_opened[index] = false;
|
|
}
|
|
|
|
}
|