Merge pull request #918 from yuriks/romfs
Do not load entire RomFS to memory, read from the file as needed instead (rebased)
This commit is contained in:
commit
946f0ee2f4
17 changed files with 112 additions and 98 deletions
|
@ -17,16 +17,15 @@
|
||||||
|
|
||||||
namespace FileSys {
|
namespace FileSys {
|
||||||
|
|
||||||
ArchiveFactory_RomFS::ArchiveFactory_RomFS(const Loader::AppLoader& app_loader)
|
ArchiveFactory_RomFS::ArchiveFactory_RomFS(Loader::AppLoader& app_loader) {
|
||||||
: romfs_data(std::make_shared<std::vector<u8>>()) {
|
|
||||||
// Load the RomFS from the app
|
// Load the RomFS from the app
|
||||||
if (Loader::ResultStatus::Success != app_loader.ReadRomFS(*romfs_data)) {
|
if (Loader::ResultStatus::Success != app_loader.ReadRomFS(romfs_file, data_offset, data_size)) {
|
||||||
LOG_ERROR(Service_FS, "Unable to read RomFS!");
|
LOG_ERROR(Service_FS, "Unable to read RomFS!");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultVal<std::unique_ptr<ArchiveBackend>> ArchiveFactory_RomFS::Open(const Path& path) {
|
ResultVal<std::unique_ptr<ArchiveBackend>> ArchiveFactory_RomFS::Open(const Path& path) {
|
||||||
auto archive = Common::make_unique<IVFCArchive>(romfs_data);
|
auto archive = Common::make_unique<IVFCArchive>(romfs_file, data_offset, data_size);
|
||||||
return MakeResult<std::unique_ptr<ArchiveBackend>>(std::move(archive));
|
return MakeResult<std::unique_ptr<ArchiveBackend>>(std::move(archive));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -22,14 +22,16 @@ namespace FileSys {
|
||||||
/// File system interface to the RomFS archive
|
/// File system interface to the RomFS archive
|
||||||
class ArchiveFactory_RomFS final : public ArchiveFactory {
|
class ArchiveFactory_RomFS final : public ArchiveFactory {
|
||||||
public:
|
public:
|
||||||
ArchiveFactory_RomFS(const Loader::AppLoader& app_loader);
|
ArchiveFactory_RomFS(Loader::AppLoader& app_loader);
|
||||||
|
|
||||||
std::string GetName() const override { return "RomFS"; }
|
std::string GetName() const override { return "RomFS"; }
|
||||||
ResultVal<std::unique_ptr<ArchiveBackend>> Open(const Path& path) override;
|
ResultVal<std::unique_ptr<ArchiveBackend>> Open(const Path& path) override;
|
||||||
ResultCode Format(const Path& path) override;
|
ResultCode Format(const Path& path) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::shared_ptr<std::vector<u8>> romfs_data;
|
std::shared_ptr<FileUtil::IOFile> romfs_file;
|
||||||
|
u64 data_offset;
|
||||||
|
u64 data_size;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace FileSys
|
} // namespace FileSys
|
||||||
|
|
|
@ -37,17 +37,14 @@ ResultVal<std::unique_ptr<ArchiveBackend>> ArchiveFactory_SaveDataCheck::Open(co
|
||||||
auto vec = path.AsBinary();
|
auto vec = path.AsBinary();
|
||||||
const u32* data = reinterpret_cast<u32*>(vec.data());
|
const u32* data = reinterpret_cast<u32*>(vec.data());
|
||||||
std::string file_path = GetSaveDataCheckPath(mount_point, data[1], data[0]);
|
std::string file_path = GetSaveDataCheckPath(mount_point, data[1], data[0]);
|
||||||
FileUtil::IOFile file(file_path, "rb");
|
auto file = std::make_shared<FileUtil::IOFile>(file_path, "rb");
|
||||||
|
|
||||||
if (!file.IsOpen()) {
|
if (!file->IsOpen()) {
|
||||||
return ResultCode(-1); // TODO(Subv): Find the right error code
|
return ResultCode(-1); // TODO(Subv): Find the right error code
|
||||||
}
|
}
|
||||||
auto size = file.GetSize();
|
auto size = file->GetSize();
|
||||||
auto raw_data = std::make_shared<std::vector<u8>>(size);
|
|
||||||
file.ReadBytes(raw_data->data(), size);
|
|
||||||
file.Close();
|
|
||||||
|
|
||||||
auto archive = Common::make_unique<IVFCArchive>(std::move(raw_data));
|
auto archive = Common::make_unique<IVFCArchive>(file, 0, size);
|
||||||
return MakeResult<std::unique_ptr<ArchiveBackend>>(std::move(archive));
|
return MakeResult<std::unique_ptr<ArchiveBackend>>(std::move(archive));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -105,12 +105,12 @@ bool DiskFile::Open() {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
size_t DiskFile::Read(const u64 offset, const u32 length, u8* buffer) const {
|
size_t DiskFile::Read(const u64 offset, const size_t length, u8* buffer) const {
|
||||||
file->Seek(offset, SEEK_SET);
|
file->Seek(offset, SEEK_SET);
|
||||||
return file->ReadBytes(buffer, length);
|
return file->ReadBytes(buffer, length);
|
||||||
}
|
}
|
||||||
|
|
||||||
size_t DiskFile::Write(const u64 offset, const u32 length, const u32 flush, const u8* buffer) const {
|
size_t DiskFile::Write(const u64 offset, const size_t length, const bool flush, const u8* buffer) const {
|
||||||
file->Seek(offset, SEEK_SET);
|
file->Seek(offset, SEEK_SET);
|
||||||
size_t written = file->WriteBytes(buffer, length);
|
size_t written = file->WriteBytes(buffer, length);
|
||||||
if (flush)
|
if (flush)
|
||||||
|
@ -118,8 +118,8 @@ size_t DiskFile::Write(const u64 offset, const u32 length, const u32 flush, cons
|
||||||
return written;
|
return written;
|
||||||
}
|
}
|
||||||
|
|
||||||
size_t DiskFile::GetSize() const {
|
u64 DiskFile::GetSize() const {
|
||||||
return static_cast<size_t>(file->GetSize());
|
return file->GetSize();
|
||||||
}
|
}
|
||||||
|
|
||||||
bool DiskFile::SetSize(const u64 size) const {
|
bool DiskFile::SetSize(const u64 size) const {
|
||||||
|
|
|
@ -55,10 +55,10 @@ public:
|
||||||
DiskFile(const DiskArchive& archive, const Path& path, const Mode mode);
|
DiskFile(const DiskArchive& archive, const Path& path, const Mode mode);
|
||||||
|
|
||||||
bool Open() override;
|
bool Open() override;
|
||||||
size_t Read(const u64 offset, const u32 length, u8* buffer) const override;
|
size_t Read(u64 offset, size_t length, u8* buffer) const override;
|
||||||
size_t Write(const u64 offset, const u32 length, const u32 flush, const u8* buffer) const override;
|
size_t Write(u64 offset, size_t length, bool flush, const u8* buffer) const override;
|
||||||
size_t GetSize() const override;
|
u64 GetSize() const override;
|
||||||
bool SetSize(const u64 size) const override;
|
bool SetSize(u64 size) const override;
|
||||||
bool Close() const override;
|
bool Close() const override;
|
||||||
|
|
||||||
void Flush() const override {
|
void Flush() const override {
|
||||||
|
|
|
@ -31,7 +31,7 @@ public:
|
||||||
* @param buffer Buffer to read data into
|
* @param buffer Buffer to read data into
|
||||||
* @return Number of bytes read
|
* @return Number of bytes read
|
||||||
*/
|
*/
|
||||||
virtual size_t Read(const u64 offset, const u32 length, u8* buffer) const = 0;
|
virtual size_t Read(u64 offset, size_t length, u8* buffer) const = 0;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Write data to the file
|
* Write data to the file
|
||||||
|
@ -41,20 +41,20 @@ public:
|
||||||
* @param buffer Buffer to read data from
|
* @param buffer Buffer to read data from
|
||||||
* @return Number of bytes written
|
* @return Number of bytes written
|
||||||
*/
|
*/
|
||||||
virtual size_t Write(const u64 offset, const u32 length, const u32 flush, const u8* buffer) const = 0;
|
virtual size_t Write(u64 offset, size_t length, bool flush, const u8* buffer) const = 0;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the size of the file in bytes
|
* Get the size of the file in bytes
|
||||||
* @return Size of the file in bytes
|
* @return Size of the file in bytes
|
||||||
*/
|
*/
|
||||||
virtual size_t GetSize() const = 0;
|
virtual u64 GetSize() const = 0;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Set the size of the file in bytes
|
* Set the size of the file in bytes
|
||||||
* @param size New size of the file
|
* @param size New size of the file
|
||||||
* @return true if successful
|
* @return true if successful
|
||||||
*/
|
*/
|
||||||
virtual bool SetSize(const u64 size) const = 0;
|
virtual bool SetSize(u64 size) const = 0;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Close the file
|
* Close the file
|
||||||
|
|
|
@ -16,15 +16,12 @@
|
||||||
|
|
||||||
namespace FileSys {
|
namespace FileSys {
|
||||||
|
|
||||||
IVFCArchive::IVFCArchive(std::shared_ptr<const std::vector<u8>> data) : data(data) {
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string IVFCArchive::GetName() const {
|
std::string IVFCArchive::GetName() const {
|
||||||
return "IVFC";
|
return "IVFC";
|
||||||
}
|
}
|
||||||
|
|
||||||
std::unique_ptr<FileBackend> IVFCArchive::OpenFile(const Path& path, const Mode mode) const {
|
std::unique_ptr<FileBackend> IVFCArchive::OpenFile(const Path& path, const Mode mode) const {
|
||||||
return Common::make_unique<IVFCFile>(data);
|
return Common::make_unique<IVFCFile>(romfs_file, data_offset, data_size);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool IVFCArchive::DeleteFile(const Path& path) const {
|
bool IVFCArchive::DeleteFile(const Path& path) const {
|
||||||
|
@ -64,19 +61,21 @@ std::unique_ptr<DirectoryBackend> IVFCArchive::OpenDirectory(const Path& path) c
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
size_t IVFCFile::Read(const u64 offset, const u32 length, u8* buffer) const {
|
size_t IVFCFile::Read(const u64 offset, const size_t length, u8* buffer) const {
|
||||||
LOG_TRACE(Service_FS, "called offset=%llu, length=%d", offset, length);
|
LOG_TRACE(Service_FS, "called offset=%llu, length=%d", offset, length);
|
||||||
memcpy(buffer, data->data() + offset, length);
|
romfs_file->Seek(data_offset + offset, SEEK_SET);
|
||||||
return length;
|
size_t read_length = (size_t)std::min((u64)length, data_size - offset);
|
||||||
|
|
||||||
|
return romfs_file->ReadBytes(buffer, read_length);
|
||||||
}
|
}
|
||||||
|
|
||||||
size_t IVFCFile::Write(const u64 offset, const u32 length, const u32 flush, const u8* buffer) const {
|
size_t IVFCFile::Write(const u64 offset, const size_t length, const bool flush, const u8* buffer) const {
|
||||||
LOG_ERROR(Service_FS, "Attempted to write to IVFC file");
|
LOG_ERROR(Service_FS, "Attempted to write to IVFC file");
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
size_t IVFCFile::GetSize() const {
|
u64 IVFCFile::GetSize() const {
|
||||||
return sizeof(u8) * data->size();
|
return data_size;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool IVFCFile::SetSize(const u64 size) const {
|
bool IVFCFile::SetSize(const u64 size) const {
|
||||||
|
|
|
@ -10,6 +10,7 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "common/common_types.h"
|
#include "common/common_types.h"
|
||||||
|
#include "common/file_util.h"
|
||||||
|
|
||||||
#include "core/file_sys/archive_backend.h"
|
#include "core/file_sys/archive_backend.h"
|
||||||
#include "core/file_sys/directory_backend.h"
|
#include "core/file_sys/directory_backend.h"
|
||||||
|
@ -28,7 +29,8 @@ namespace FileSys {
|
||||||
*/
|
*/
|
||||||
class IVFCArchive : public ArchiveBackend {
|
class IVFCArchive : public ArchiveBackend {
|
||||||
public:
|
public:
|
||||||
IVFCArchive(std::shared_ptr<const std::vector<u8>> data);
|
IVFCArchive(std::shared_ptr<FileUtil::IOFile> file, u64 offset, u64 size)
|
||||||
|
: romfs_file(file), data_offset(offset), data_size(size) {}
|
||||||
|
|
||||||
std::string GetName() const override;
|
std::string GetName() const override;
|
||||||
|
|
||||||
|
@ -42,23 +44,28 @@ public:
|
||||||
std::unique_ptr<DirectoryBackend> OpenDirectory(const Path& path) const override;
|
std::unique_ptr<DirectoryBackend> OpenDirectory(const Path& path) const override;
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
std::shared_ptr<const std::vector<u8>> data;
|
std::shared_ptr<FileUtil::IOFile> romfs_file;
|
||||||
|
u64 data_offset;
|
||||||
|
u64 data_size;
|
||||||
};
|
};
|
||||||
|
|
||||||
class IVFCFile : public FileBackend {
|
class IVFCFile : public FileBackend {
|
||||||
public:
|
public:
|
||||||
IVFCFile(std::shared_ptr<const std::vector<u8>> data) : data(data) {}
|
IVFCFile(std::shared_ptr<FileUtil::IOFile> file, u64 offset, u64 size)
|
||||||
|
: romfs_file(file), data_offset(offset), data_size(size) {}
|
||||||
|
|
||||||
bool Open() override { return true; }
|
bool Open() override { return true; }
|
||||||
size_t Read(const u64 offset, const u32 length, u8* buffer) const override;
|
size_t Read(u64 offset, size_t length, u8* buffer) const override;
|
||||||
size_t Write(const u64 offset, const u32 length, const u32 flush, const u8* buffer) const override;
|
size_t Write(u64 offset, size_t length, bool flush, const u8* buffer) const override;
|
||||||
size_t GetSize() const override;
|
u64 GetSize() const override;
|
||||||
bool SetSize(const u64 size) const override;
|
bool SetSize(u64 size) const override;
|
||||||
bool Close() const override { return false; }
|
bool Close() const override { return false; }
|
||||||
void Flush() const override { }
|
void Flush() const override { }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::shared_ptr<const std::vector<u8>> data;
|
std::shared_ptr<FileUtil::IOFile> romfs_file;
|
||||||
|
u64 data_offset;
|
||||||
|
u64 data_size;
|
||||||
};
|
};
|
||||||
|
|
||||||
class IVFCDirectory : public DirectoryBackend {
|
class IVFCDirectory : public DirectoryBackend {
|
||||||
|
|
|
@ -116,7 +116,7 @@ ResultVal<bool> File::SyncRequest() {
|
||||||
u32 address = cmd_buff[6];
|
u32 address = cmd_buff[6];
|
||||||
LOG_TRACE(Service_FS, "Write %s %s: offset=0x%llx length=%d address=0x%x, flush=0x%x",
|
LOG_TRACE(Service_FS, "Write %s %s: offset=0x%llx length=%d address=0x%x, flush=0x%x",
|
||||||
GetTypeName().c_str(), GetName().c_str(), offset, length, address, flush);
|
GetTypeName().c_str(), GetName().c_str(), offset, length, address, flush);
|
||||||
cmd_buff[2] = static_cast<u32>(backend->Write(offset, length, flush, Memory::GetPointer(address)));
|
cmd_buff[2] = static_cast<u32>(backend->Write(offset, length, flush != 0, Memory::GetPointer(address)));
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -246,11 +246,11 @@ ResultStatus AppLoader_THREEDSX::Load() {
|
||||||
if (is_loaded)
|
if (is_loaded)
|
||||||
return ResultStatus::ErrorAlreadyLoaded;
|
return ResultStatus::ErrorAlreadyLoaded;
|
||||||
|
|
||||||
if (!file->IsOpen())
|
if (!file.IsOpen())
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
SharedPtr<CodeSet> codeset;
|
SharedPtr<CodeSet> codeset;
|
||||||
if (Load3DSXFile(*file, Memory::PROCESS_IMAGE_VADDR, &codeset) != ERROR_NONE)
|
if (Load3DSXFile(file, Memory::PROCESS_IMAGE_VADDR, &codeset) != ERROR_NONE)
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
codeset->name = filename;
|
codeset->name = filename;
|
||||||
|
|
||||||
|
|
|
@ -17,7 +17,7 @@ namespace Loader {
|
||||||
/// Loads an 3DSX file
|
/// Loads an 3DSX file
|
||||||
class AppLoader_THREEDSX final : public AppLoader {
|
class AppLoader_THREEDSX final : public AppLoader {
|
||||||
public:
|
public:
|
||||||
AppLoader_THREEDSX(std::unique_ptr<FileUtil::IOFile>&& file, std::string filename)
|
AppLoader_THREEDSX(FileUtil::IOFile&& file, std::string filename)
|
||||||
: AppLoader(std::move(file)), filename(std::move(filename)) {}
|
: AppLoader(std::move(file)), filename(std::move(filename)) {}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -273,7 +273,6 @@ SharedPtr<CodeSet> ElfReader::LoadInto(u32 vaddr) {
|
||||||
LOG_DEBUG(Loader, "%i segments:", header->e_phnum);
|
LOG_DEBUG(Loader, "%i segments:", header->e_phnum);
|
||||||
|
|
||||||
// First pass : Get the bits into RAM
|
// First pass : Get the bits into RAM
|
||||||
u32 segment_addr[32];
|
|
||||||
u32 base_addr = relocate ? vaddr : 0;
|
u32 base_addr = relocate ? vaddr : 0;
|
||||||
|
|
||||||
u32 total_image_size = 0;
|
u32 total_image_size = 0;
|
||||||
|
@ -392,15 +391,15 @@ ResultStatus AppLoader_ELF::Load() {
|
||||||
if (is_loaded)
|
if (is_loaded)
|
||||||
return ResultStatus::ErrorAlreadyLoaded;
|
return ResultStatus::ErrorAlreadyLoaded;
|
||||||
|
|
||||||
if (!file->IsOpen())
|
if (!file.IsOpen())
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
// Reset read pointer in case this file has been read before.
|
// Reset read pointer in case this file has been read before.
|
||||||
file->Seek(0, SEEK_SET);
|
file.Seek(0, SEEK_SET);
|
||||||
|
|
||||||
u32 size = static_cast<u32>(file->GetSize());
|
size_t size = file.GetSize();
|
||||||
std::unique_ptr<u8[]> buffer(new u8[size]);
|
std::unique_ptr<u8[]> buffer(new u8[size]);
|
||||||
if (file->ReadBytes(&buffer[0], size) != size)
|
if (file.ReadBytes(&buffer[0], size) != size)
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
ElfReader elf_reader(&buffer[0]);
|
ElfReader elf_reader(&buffer[0]);
|
||||||
|
|
|
@ -17,7 +17,7 @@ namespace Loader {
|
||||||
/// Loads an ELF/AXF file
|
/// Loads an ELF/AXF file
|
||||||
class AppLoader_ELF final : public AppLoader {
|
class AppLoader_ELF final : public AppLoader {
|
||||||
public:
|
public:
|
||||||
AppLoader_ELF(std::unique_ptr<FileUtil::IOFile>&& file, std::string filename)
|
AppLoader_ELF(FileUtil::IOFile&& file, std::string filename)
|
||||||
: AppLoader(std::move(file)), filename(std::move(filename)) { }
|
: AppLoader(std::move(file)), filename(std::move(filename)) { }
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -90,8 +90,8 @@ static const char* GetFileTypeString(FileType type) {
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultStatus LoadFile(const std::string& filename) {
|
ResultStatus LoadFile(const std::string& filename) {
|
||||||
std::unique_ptr<FileUtil::IOFile> file(new FileUtil::IOFile(filename, "rb"));
|
FileUtil::IOFile file(filename, "rb");
|
||||||
if (!file->IsOpen()) {
|
if (!file.IsOpen()) {
|
||||||
LOG_ERROR(Loader, "Failed to load file %s", filename.c_str());
|
LOG_ERROR(Loader, "Failed to load file %s", filename.c_str());
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
}
|
}
|
||||||
|
@ -99,7 +99,7 @@ ResultStatus LoadFile(const std::string& filename) {
|
||||||
std::string filename_filename, filename_extension;
|
std::string filename_filename, filename_extension;
|
||||||
Common::SplitPath(filename, nullptr, &filename_filename, &filename_extension);
|
Common::SplitPath(filename, nullptr, &filename_filename, &filename_extension);
|
||||||
|
|
||||||
FileType type = IdentifyFile(*file);
|
FileType type = IdentifyFile(file);
|
||||||
FileType filename_type = GuessFromExtension(filename_extension);
|
FileType filename_type = GuessFromExtension(filename_extension);
|
||||||
|
|
||||||
if (type != filename_type) {
|
if (type != filename_type) {
|
||||||
|
@ -124,7 +124,7 @@ ResultStatus LoadFile(const std::string& filename) {
|
||||||
case FileType::CXI:
|
case FileType::CXI:
|
||||||
case FileType::CCI:
|
case FileType::CCI:
|
||||||
{
|
{
|
||||||
AppLoader_NCCH app_loader(std::move(file));
|
AppLoader_NCCH app_loader(std::move(file), filename);
|
||||||
|
|
||||||
// Load application and RomFS
|
// Load application and RomFS
|
||||||
if (ResultStatus::Success == app_loader.Load()) {
|
if (ResultStatus::Success == app_loader.Load()) {
|
||||||
|
|
|
@ -52,7 +52,7 @@ static inline u32 MakeMagic(char a, char b, char c, char d) {
|
||||||
/// Interface for loading an application
|
/// Interface for loading an application
|
||||||
class AppLoader : NonCopyable {
|
class AppLoader : NonCopyable {
|
||||||
public:
|
public:
|
||||||
AppLoader(std::unique_ptr<FileUtil::IOFile>&& file) : file(std::move(file)) { }
|
AppLoader(FileUtil::IOFile&& file) : file(std::move(file)) { }
|
||||||
virtual ~AppLoader() { }
|
virtual ~AppLoader() { }
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -66,7 +66,7 @@ public:
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
virtual ResultStatus ReadCode(std::vector<u8>& buffer) const {
|
virtual ResultStatus ReadCode(std::vector<u8>& buffer) {
|
||||||
return ResultStatus::ErrorNotImplemented;
|
return ResultStatus::ErrorNotImplemented;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -75,7 +75,7 @@ public:
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
virtual ResultStatus ReadIcon(std::vector<u8>& buffer) const {
|
virtual ResultStatus ReadIcon(std::vector<u8>& buffer) {
|
||||||
return ResultStatus::ErrorNotImplemented;
|
return ResultStatus::ErrorNotImplemented;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -84,7 +84,7 @@ public:
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
virtual ResultStatus ReadBanner(std::vector<u8>& buffer) const {
|
virtual ResultStatus ReadBanner(std::vector<u8>& buffer) {
|
||||||
return ResultStatus::ErrorNotImplemented;
|
return ResultStatus::ErrorNotImplemented;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -93,22 +93,25 @@ public:
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
virtual ResultStatus ReadLogo(std::vector<u8>& buffer) const {
|
virtual ResultStatus ReadLogo(std::vector<u8>& buffer) {
|
||||||
return ResultStatus::ErrorNotImplemented;
|
return ResultStatus::ErrorNotImplemented;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the RomFS of the application
|
* Get the RomFS of the application
|
||||||
* @param buffer Reference to buffer to store data
|
* Since the RomFS can be huge, we return a file reference instead of copying to a buffer
|
||||||
|
* @param romfs_file The file containing the RomFS
|
||||||
|
* @param offset The offset the romfs begins on
|
||||||
|
* @param size The size of the romfs
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
virtual ResultStatus ReadRomFS(std::vector<u8>& buffer) const {
|
virtual ResultStatus ReadRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset, u64& size) {
|
||||||
return ResultStatus::ErrorNotImplemented;
|
return ResultStatus::ErrorNotImplemented;
|
||||||
}
|
}
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
std::unique_ptr<FileUtil::IOFile> file;
|
FileUtil::IOFile file;
|
||||||
bool is_loaded = false;
|
bool is_loaded = false;
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -117,7 +117,7 @@ FileType AppLoader_NCCH::IdentifyType(FileUtil::IOFile& file) {
|
||||||
return FileType::Error;
|
return FileType::Error;
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultStatus AppLoader_NCCH::LoadExec() const {
|
ResultStatus AppLoader_NCCH::LoadExec() {
|
||||||
using Kernel::SharedPtr;
|
using Kernel::SharedPtr;
|
||||||
using Kernel::CodeSet;
|
using Kernel::CodeSet;
|
||||||
|
|
||||||
|
@ -171,8 +171,8 @@ ResultStatus AppLoader_NCCH::LoadExec() const {
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>& buffer) const {
|
ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>& buffer) {
|
||||||
if (!file->IsOpen())
|
if (!file.IsOpen())
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
LOG_DEBUG(Loader, "%d sections:", kMaxSections);
|
LOG_DEBUG(Loader, "%d sections:", kMaxSections);
|
||||||
|
@ -186,7 +186,7 @@ ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>&
|
||||||
section.offset, section.size, section.name);
|
section.offset, section.size, section.name);
|
||||||
|
|
||||||
s64 section_offset = (section.offset + exefs_offset + sizeof(ExeFs_Header) + ncch_offset);
|
s64 section_offset = (section.offset + exefs_offset + sizeof(ExeFs_Header) + ncch_offset);
|
||||||
file->Seek(section_offset, SEEK_SET);
|
file.Seek(section_offset, SEEK_SET);
|
||||||
|
|
||||||
if (is_compressed) {
|
if (is_compressed) {
|
||||||
// Section is compressed, read compressed .code section...
|
// Section is compressed, read compressed .code section...
|
||||||
|
@ -197,7 +197,7 @@ ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>&
|
||||||
return ResultStatus::ErrorMemoryAllocationFailed;
|
return ResultStatus::ErrorMemoryAllocationFailed;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (file->ReadBytes(&temp_buffer[0], section.size) != section.size)
|
if (file.ReadBytes(&temp_buffer[0], section.size) != section.size)
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
// Decompress .code section...
|
// Decompress .code section...
|
||||||
|
@ -208,7 +208,7 @@ ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>&
|
||||||
} else {
|
} else {
|
||||||
// Section is uncompressed...
|
// Section is uncompressed...
|
||||||
buffer.resize(section.size);
|
buffer.resize(section.size);
|
||||||
if (file->ReadBytes(&buffer[0], section.size) != section.size)
|
if (file.ReadBytes(&buffer[0], section.size) != section.size)
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
}
|
}
|
||||||
return ResultStatus::Success;
|
return ResultStatus::Success;
|
||||||
|
@ -221,21 +221,21 @@ ResultStatus AppLoader_NCCH::Load() {
|
||||||
if (is_loaded)
|
if (is_loaded)
|
||||||
return ResultStatus::ErrorAlreadyLoaded;
|
return ResultStatus::ErrorAlreadyLoaded;
|
||||||
|
|
||||||
if (!file->IsOpen())
|
if (!file.IsOpen())
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
// Reset read pointer in case this file has been read before.
|
// Reset read pointer in case this file has been read before.
|
||||||
file->Seek(0, SEEK_SET);
|
file.Seek(0, SEEK_SET);
|
||||||
|
|
||||||
if (file->ReadBytes(&ncch_header, sizeof(NCCH_Header)) != sizeof(NCCH_Header))
|
if (file.ReadBytes(&ncch_header, sizeof(NCCH_Header)) != sizeof(NCCH_Header))
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
// Skip NCSD header and load first NCCH (NCSD is just a container of NCCH files)...
|
// Skip NCSD header and load first NCCH (NCSD is just a container of NCCH files)...
|
||||||
if (MakeMagic('N', 'C', 'S', 'D') == ncch_header.magic) {
|
if (MakeMagic('N', 'C', 'S', 'D') == ncch_header.magic) {
|
||||||
LOG_WARNING(Loader, "Only loading the first (bootable) NCCH within the NCSD file!");
|
LOG_WARNING(Loader, "Only loading the first (bootable) NCCH within the NCSD file!");
|
||||||
ncch_offset = 0x4000;
|
ncch_offset = 0x4000;
|
||||||
file->Seek(ncch_offset, SEEK_SET);
|
file.Seek(ncch_offset, SEEK_SET);
|
||||||
file->ReadBytes(&ncch_header, sizeof(NCCH_Header));
|
file.ReadBytes(&ncch_header, sizeof(NCCH_Header));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Verify we are loading the correct file type...
|
// Verify we are loading the correct file type...
|
||||||
|
@ -244,7 +244,7 @@ ResultStatus AppLoader_NCCH::Load() {
|
||||||
|
|
||||||
// Read ExHeader...
|
// Read ExHeader...
|
||||||
|
|
||||||
if (file->ReadBytes(&exheader_header, sizeof(ExHeader_Header)) != sizeof(ExHeader_Header))
|
if (file.ReadBytes(&exheader_header, sizeof(ExHeader_Header)) != sizeof(ExHeader_Header))
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
is_compressed = (exheader_header.codeset_info.flags.flag & 1) == 1;
|
is_compressed = (exheader_header.codeset_info.flags.flag & 1) == 1;
|
||||||
|
@ -274,8 +274,8 @@ ResultStatus AppLoader_NCCH::Load() {
|
||||||
LOG_DEBUG(Loader, "ExeFS offset: 0x%08X", exefs_offset);
|
LOG_DEBUG(Loader, "ExeFS offset: 0x%08X", exefs_offset);
|
||||||
LOG_DEBUG(Loader, "ExeFS size: 0x%08X", exefs_size);
|
LOG_DEBUG(Loader, "ExeFS size: 0x%08X", exefs_size);
|
||||||
|
|
||||||
file->Seek(exefs_offset + ncch_offset, SEEK_SET);
|
file.Seek(exefs_offset + ncch_offset, SEEK_SET);
|
||||||
if (file->ReadBytes(&exefs_header, sizeof(ExeFs_Header)) != sizeof(ExeFs_Header))
|
if (file.ReadBytes(&exefs_header, sizeof(ExeFs_Header)) != sizeof(ExeFs_Header))
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
is_loaded = true; // Set state to loaded
|
is_loaded = true; // Set state to loaded
|
||||||
|
@ -283,24 +283,24 @@ ResultStatus AppLoader_NCCH::Load() {
|
||||||
return LoadExec(); // Load the executable into memory for booting
|
return LoadExec(); // Load the executable into memory for booting
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultStatus AppLoader_NCCH::ReadCode(std::vector<u8>& buffer) const {
|
ResultStatus AppLoader_NCCH::ReadCode(std::vector<u8>& buffer) {
|
||||||
return LoadSectionExeFS(".code", buffer);
|
return LoadSectionExeFS(".code", buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultStatus AppLoader_NCCH::ReadIcon(std::vector<u8>& buffer) const {
|
ResultStatus AppLoader_NCCH::ReadIcon(std::vector<u8>& buffer) {
|
||||||
return LoadSectionExeFS("icon", buffer);
|
return LoadSectionExeFS("icon", buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultStatus AppLoader_NCCH::ReadBanner(std::vector<u8>& buffer) const {
|
ResultStatus AppLoader_NCCH::ReadBanner(std::vector<u8>& buffer) {
|
||||||
return LoadSectionExeFS("banner", buffer);
|
return LoadSectionExeFS("banner", buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultStatus AppLoader_NCCH::ReadLogo(std::vector<u8>& buffer) const {
|
ResultStatus AppLoader_NCCH::ReadLogo(std::vector<u8>& buffer) {
|
||||||
return LoadSectionExeFS("logo", buffer);
|
return LoadSectionExeFS("logo", buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
ResultStatus AppLoader_NCCH::ReadRomFS(std::vector<u8>& buffer) const {
|
ResultStatus AppLoader_NCCH::ReadRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset, u64& size) {
|
||||||
if (!file->IsOpen())
|
if (!file.IsOpen())
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
// Check if the NCCH has a RomFS...
|
// Check if the NCCH has a RomFS...
|
||||||
|
@ -311,12 +311,17 @@ ResultStatus AppLoader_NCCH::ReadRomFS(std::vector<u8>& buffer) const {
|
||||||
LOG_DEBUG(Loader, "RomFS offset: 0x%08X", romfs_offset);
|
LOG_DEBUG(Loader, "RomFS offset: 0x%08X", romfs_offset);
|
||||||
LOG_DEBUG(Loader, "RomFS size: 0x%08X", romfs_size);
|
LOG_DEBUG(Loader, "RomFS size: 0x%08X", romfs_size);
|
||||||
|
|
||||||
buffer.resize(romfs_size);
|
if (file.GetSize () < romfs_offset + romfs_size)
|
||||||
|
|
||||||
file->Seek(romfs_offset, SEEK_SET);
|
|
||||||
if (file->ReadBytes(&buffer[0], romfs_size) != romfs_size)
|
|
||||||
return ResultStatus::Error;
|
return ResultStatus::Error;
|
||||||
|
|
||||||
|
// We reopen the file, to allow its position to be independent from file's
|
||||||
|
romfs_file = std::make_shared<FileUtil::IOFile>(filepath, "rb");
|
||||||
|
if (!romfs_file->IsOpen())
|
||||||
|
return ResultStatus::Error;
|
||||||
|
|
||||||
|
offset = romfs_offset;
|
||||||
|
size = romfs_size;
|
||||||
|
|
||||||
return ResultStatus::Success;
|
return ResultStatus::Success;
|
||||||
}
|
}
|
||||||
LOG_DEBUG(Loader, "NCCH has no RomFS");
|
LOG_DEBUG(Loader, "NCCH has no RomFS");
|
||||||
|
|
|
@ -163,7 +163,8 @@ namespace Loader {
|
||||||
/// Loads an NCCH file (e.g. from a CCI, or the first NCCH in a CXI)
|
/// Loads an NCCH file (e.g. from a CCI, or the first NCCH in a CXI)
|
||||||
class AppLoader_NCCH final : public AppLoader {
|
class AppLoader_NCCH final : public AppLoader {
|
||||||
public:
|
public:
|
||||||
AppLoader_NCCH(std::unique_ptr<FileUtil::IOFile>&& file) : AppLoader(std::move(file)) { }
|
AppLoader_NCCH(FileUtil::IOFile&& file, const std::string& filepath)
|
||||||
|
: AppLoader(std::move(file)), filepath(filepath) { }
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns the type of the file
|
* Returns the type of the file
|
||||||
|
@ -183,35 +184,35 @@ public:
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
ResultStatus ReadCode(std::vector<u8>& buffer) const override;
|
ResultStatus ReadCode(std::vector<u8>& buffer) override;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the icon (typically icon section) of the application
|
* Get the icon (typically icon section) of the application
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
ResultStatus ReadIcon(std::vector<u8>& buffer) const override;
|
ResultStatus ReadIcon(std::vector<u8>& buffer) override;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the banner (typically banner section) of the application
|
* Get the banner (typically banner section) of the application
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
ResultStatus ReadBanner(std::vector<u8>& buffer) const override;
|
ResultStatus ReadBanner(std::vector<u8>& buffer) override;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the logo (typically logo section) of the application
|
* Get the logo (typically logo section) of the application
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
ResultStatus ReadLogo(std::vector<u8>& buffer) const override;
|
ResultStatus ReadLogo(std::vector<u8>& buffer) override;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the RomFS of the application
|
* Get the RomFS of the application
|
||||||
* @param buffer Reference to buffer to store data
|
* @param buffer Reference to buffer to store data
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
ResultStatus ReadRomFS(std::vector<u8>& buffer) const override;
|
ResultStatus ReadRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset, u64& size) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|
||||||
|
@ -221,13 +222,13 @@ private:
|
||||||
* @param buffer Vector to read data into
|
* @param buffer Vector to read data into
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
ResultStatus LoadSectionExeFS(const char* name, std::vector<u8>& buffer) const;
|
ResultStatus LoadSectionExeFS(const char* name, std::vector<u8>& buffer);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Loads .code section into memory for booting
|
* Loads .code section into memory for booting
|
||||||
* @return ResultStatus result of function
|
* @return ResultStatus result of function
|
||||||
*/
|
*/
|
||||||
ResultStatus LoadExec() const;
|
ResultStatus LoadExec();
|
||||||
|
|
||||||
bool is_compressed = false;
|
bool is_compressed = false;
|
||||||
|
|
||||||
|
@ -244,6 +245,8 @@ private:
|
||||||
NCCH_Header ncch_header;
|
NCCH_Header ncch_header;
|
||||||
ExeFs_Header exefs_header;
|
ExeFs_Header exefs_header;
|
||||||
ExHeader_Header exheader_header;
|
ExHeader_Header exheader_header;
|
||||||
|
|
||||||
|
std::string filepath;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace Loader
|
} // namespace Loader
|
||||||
|
|
Loading…
Reference in a new issue