mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-12-23 21:01:13 +00:00
259 lines
8.7 KiB
C++
259 lines
8.7 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 "fatal_fs_api.hpp"
|
|
|
|
namespace ams::fs {
|
|
|
|
namespace {
|
|
|
|
constexpr size_t MaxFiles = 8;
|
|
|
|
constinit bool g_is_sd_mounted = false;
|
|
|
|
FATFS g_sd_fs;
|
|
|
|
constinit FIL g_files[MaxFiles] = {};
|
|
constinit bool g_files_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);
|
|
}
|
|
|
|
ALWAYS_INLINE size_t GetFileIndex(FIL *fp) {
|
|
const size_t file_index = (fp - g_files);
|
|
AMS_ASSERT(file_index < MaxFiles);
|
|
return file_index;
|
|
}
|
|
|
|
}
|
|
|
|
bool MountSdCard() {
|
|
AMS_ASSERT(!g_is_sd_mounted);
|
|
g_is_sd_mounted = f_mount(std::addressof(g_sd_fs), "", 1) == FR_OK;
|
|
return g_is_sd_mounted;
|
|
}
|
|
|
|
void UnmountSdCard() {
|
|
AMS_ASSERT(g_is_sd_mounted);
|
|
f_unmount("");
|
|
g_is_sd_mounted = false;
|
|
}
|
|
|
|
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 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. */
|
|
u32 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. */
|
|
u32 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. */
|
|
u32 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;
|
|
}
|
|
|
|
}
|