mirror of
				https://github.com/PabloMK7/citra.git
				synced 2025-10-31 13:50:03 +00:00 
			
		
		
		
	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 { | ||||
| 
 | ||||
| ArchiveFactory_RomFS::ArchiveFactory_RomFS(const Loader::AppLoader& app_loader) | ||||
|         : romfs_data(std::make_shared<std::vector<u8>>()) { | ||||
| ArchiveFactory_RomFS::ArchiveFactory_RomFS(Loader::AppLoader& app_loader) { | ||||
|     // 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!"); | ||||
|     } | ||||
| } | ||||
| 
 | ||||
| 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)); | ||||
| } | ||||
| 
 | ||||
|  |  | |||
|  | @ -22,14 +22,16 @@ namespace FileSys { | |||
| /// File system interface to the RomFS archive
 | ||||
| class ArchiveFactory_RomFS final : public ArchiveFactory { | ||||
| public: | ||||
|     ArchiveFactory_RomFS(const Loader::AppLoader& app_loader); | ||||
|     ArchiveFactory_RomFS(Loader::AppLoader& app_loader); | ||||
| 
 | ||||
|     std::string GetName() const override { return "RomFS"; } | ||||
|     ResultVal<std::unique_ptr<ArchiveBackend>> Open(const Path& path) override; | ||||
|     ResultCode Format(const Path& path) override; | ||||
| 
 | ||||
| private: | ||||
|     std::shared_ptr<std::vector<u8>> romfs_data; | ||||
|     std::shared_ptr<FileUtil::IOFile> romfs_file; | ||||
|     u64 data_offset; | ||||
|     u64 data_size; | ||||
| }; | ||||
| 
 | ||||
| } // namespace FileSys
 | ||||
|  |  | |||
|  | @ -37,17 +37,14 @@ ResultVal<std::unique_ptr<ArchiveBackend>> ArchiveFactory_SaveDataCheck::Open(co | |||
|     auto vec = path.AsBinary(); | ||||
|     const u32* data = reinterpret_cast<u32*>(vec.data()); | ||||
|     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
 | ||||
|     } | ||||
|     auto size = file.GetSize(); | ||||
|     auto raw_data = std::make_shared<std::vector<u8>>(size); | ||||
|     file.ReadBytes(raw_data->data(), size); | ||||
|     file.Close(); | ||||
|     auto size = file->GetSize(); | ||||
| 
 | ||||
|     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)); | ||||
| } | ||||
| 
 | ||||
|  |  | |||
|  | @ -105,12 +105,12 @@ bool DiskFile::Open() { | |||
|     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); | ||||
|     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); | ||||
|     size_t written = file->WriteBytes(buffer, length); | ||||
|     if (flush) | ||||
|  | @ -118,8 +118,8 @@ size_t DiskFile::Write(const u64 offset, const u32 length, const u32 flush, cons | |||
|     return written; | ||||
| } | ||||
| 
 | ||||
| size_t DiskFile::GetSize() const { | ||||
|     return static_cast<size_t>(file->GetSize()); | ||||
| u64 DiskFile::GetSize() const { | ||||
|     return file->GetSize(); | ||||
| } | ||||
| 
 | ||||
| bool DiskFile::SetSize(const u64 size) const { | ||||
|  |  | |||
|  | @ -55,10 +55,10 @@ public: | |||
|     DiskFile(const DiskArchive& archive, const Path& path, const Mode mode); | ||||
| 
 | ||||
|     bool Open() override; | ||||
|     size_t Read(const u64 offset, const u32 length, u8* buffer) const override; | ||||
|     size_t Write(const u64 offset, const u32 length, const u32 flush, const u8* buffer) const override; | ||||
|     size_t GetSize() const override; | ||||
|     bool SetSize(const u64 size) const override; | ||||
|     size_t Read(u64 offset, size_t length, u8* buffer) const override; | ||||
|     size_t Write(u64 offset, size_t length, bool flush, const u8* buffer) const override; | ||||
|     u64 GetSize() const override; | ||||
|     bool SetSize(u64 size) const override; | ||||
|     bool Close() const override; | ||||
| 
 | ||||
|     void Flush() const override { | ||||
|  |  | |||
|  | @ -31,7 +31,7 @@ public: | |||
|      * @param buffer Buffer to read data into | ||||
|      * @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 | ||||
|  | @ -41,20 +41,20 @@ public: | |||
|      * @param buffer Buffer to read data from | ||||
|      * @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 | ||||
|      * @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 | ||||
|      * @param size New size of the file | ||||
|      * @return true if successful | ||||
|      */ | ||||
|     virtual bool SetSize(const u64 size) const = 0; | ||||
|     virtual bool SetSize(u64 size) const = 0; | ||||
| 
 | ||||
|     /**
 | ||||
|      * Close the file | ||||
|  |  | |||
|  | @ -16,15 +16,12 @@ | |||
| 
 | ||||
| namespace FileSys { | ||||
| 
 | ||||
| IVFCArchive::IVFCArchive(std::shared_ptr<const std::vector<u8>> data) : data(data) { | ||||
| } | ||||
| 
 | ||||
| std::string IVFCArchive::GetName() const { | ||||
|     return "IVFC"; | ||||
| } | ||||
| 
 | ||||
| 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 { | ||||
|  | @ -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); | ||||
|     memcpy(buffer, data->data() + offset, length); | ||||
|     return length; | ||||
|     romfs_file->Seek(data_offset + offset, SEEK_SET); | ||||
|     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"); | ||||
|     return 0; | ||||
| } | ||||
| 
 | ||||
| size_t IVFCFile::GetSize() const { | ||||
|     return sizeof(u8) * data->size(); | ||||
| u64 IVFCFile::GetSize() const { | ||||
|     return data_size; | ||||
| } | ||||
| 
 | ||||
| bool IVFCFile::SetSize(const u64 size) const { | ||||
|  |  | |||
|  | @ -10,6 +10,7 @@ | |||
| #include <vector> | ||||
| 
 | ||||
| #include "common/common_types.h" | ||||
| #include "common/file_util.h" | ||||
| 
 | ||||
| #include "core/file_sys/archive_backend.h" | ||||
| #include "core/file_sys/directory_backend.h" | ||||
|  | @ -28,7 +29,8 @@ namespace FileSys { | |||
|  */ | ||||
| class IVFCArchive : public ArchiveBackend { | ||||
| 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; | ||||
| 
 | ||||
|  | @ -42,23 +44,28 @@ public: | |||
|     std::unique_ptr<DirectoryBackend> OpenDirectory(const Path& path) const override; | ||||
| 
 | ||||
| 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 { | ||||
| 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; } | ||||
|     size_t Read(const u64 offset, const u32 length, u8* buffer) const override; | ||||
|     size_t Write(const u64 offset, const u32 length, const u32 flush, const u8* buffer) const override; | ||||
|     size_t GetSize() const override; | ||||
|     bool SetSize(const u64 size) const override; | ||||
|     size_t Read(u64 offset, size_t length, u8* buffer) const override; | ||||
|     size_t Write(u64 offset, size_t length, bool flush, const u8* buffer) const override; | ||||
|     u64 GetSize() const override; | ||||
|     bool SetSize(u64 size) const override; | ||||
|     bool Close() const override { return false; } | ||||
|     void Flush() const override { } | ||||
| 
 | ||||
| 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 { | ||||
|  |  | |||
|  | @ -116,7 +116,7 @@ ResultVal<bool> File::SyncRequest() { | |||
|             u32 address = cmd_buff[6]; | ||||
|             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); | ||||
|             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; | ||||
|         } | ||||
| 
 | ||||
|  |  | |||
|  | @ -246,11 +246,11 @@ ResultStatus AppLoader_THREEDSX::Load() { | |||
|     if (is_loaded) | ||||
|         return ResultStatus::ErrorAlreadyLoaded; | ||||
| 
 | ||||
|     if (!file->IsOpen()) | ||||
|     if (!file.IsOpen()) | ||||
|         return ResultStatus::Error; | ||||
| 
 | ||||
|     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; | ||||
|     codeset->name = filename; | ||||
| 
 | ||||
|  |  | |||
|  | @ -17,7 +17,7 @@ namespace Loader { | |||
| /// Loads an 3DSX file
 | ||||
| class AppLoader_THREEDSX final : public AppLoader { | ||||
| 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)) {} | ||||
| 
 | ||||
|     /**
 | ||||
|  |  | |||
|  | @ -273,7 +273,6 @@ SharedPtr<CodeSet> ElfReader::LoadInto(u32 vaddr) { | |||
|     LOG_DEBUG(Loader, "%i segments:", header->e_phnum); | ||||
| 
 | ||||
|     // First pass : Get the bits into RAM
 | ||||
|     u32 segment_addr[32]; | ||||
|     u32 base_addr = relocate ? vaddr : 0; | ||||
| 
 | ||||
|     u32 total_image_size = 0; | ||||
|  | @ -392,15 +391,15 @@ ResultStatus AppLoader_ELF::Load() { | |||
|     if (is_loaded) | ||||
|         return ResultStatus::ErrorAlreadyLoaded; | ||||
| 
 | ||||
|     if (!file->IsOpen()) | ||||
|     if (!file.IsOpen()) | ||||
|         return ResultStatus::Error; | ||||
| 
 | ||||
|     // 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]); | ||||
|     if (file->ReadBytes(&buffer[0], size) != size) | ||||
|     if (file.ReadBytes(&buffer[0], size) != size) | ||||
|         return ResultStatus::Error; | ||||
| 
 | ||||
|     ElfReader elf_reader(&buffer[0]); | ||||
|  |  | |||
|  | @ -17,7 +17,7 @@ namespace Loader { | |||
| /// Loads an ELF/AXF file
 | ||||
| class AppLoader_ELF final : public AppLoader { | ||||
| 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)) { } | ||||
| 
 | ||||
|     /**
 | ||||
|  |  | |||
|  | @ -90,8 +90,8 @@ static const char* GetFileTypeString(FileType type) { | |||
| } | ||||
| 
 | ||||
| ResultStatus LoadFile(const std::string& filename) { | ||||
|     std::unique_ptr<FileUtil::IOFile> file(new FileUtil::IOFile(filename, "rb")); | ||||
|     if (!file->IsOpen()) { | ||||
|     FileUtil::IOFile file(filename, "rb"); | ||||
|     if (!file.IsOpen()) { | ||||
|         LOG_ERROR(Loader, "Failed to load file %s", filename.c_str()); | ||||
|         return ResultStatus::Error; | ||||
|     } | ||||
|  | @ -99,7 +99,7 @@ ResultStatus LoadFile(const std::string& filename) { | |||
|     std::string 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); | ||||
| 
 | ||||
|     if (type != filename_type) { | ||||
|  | @ -124,7 +124,7 @@ ResultStatus LoadFile(const std::string& filename) { | |||
|     case FileType::CXI: | ||||
|     case FileType::CCI: | ||||
|     { | ||||
|         AppLoader_NCCH app_loader(std::move(file)); | ||||
|         AppLoader_NCCH app_loader(std::move(file), filename); | ||||
| 
 | ||||
|         // Load application and RomFS
 | ||||
|         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
 | ||||
| class AppLoader : NonCopyable { | ||||
| public: | ||||
|     AppLoader(std::unique_ptr<FileUtil::IOFile>&& file) : file(std::move(file)) { } | ||||
|     AppLoader(FileUtil::IOFile&& file) : file(std::move(file)) { } | ||||
|     virtual ~AppLoader() { } | ||||
| 
 | ||||
|     /**
 | ||||
|  | @ -66,7 +66,7 @@ public: | |||
|      * @param buffer Reference to buffer to store data | ||||
|      * @return ResultStatus result of function | ||||
|      */ | ||||
|     virtual ResultStatus ReadCode(std::vector<u8>& buffer) const { | ||||
|     virtual ResultStatus ReadCode(std::vector<u8>& buffer) { | ||||
|         return ResultStatus::ErrorNotImplemented; | ||||
|     } | ||||
| 
 | ||||
|  | @ -75,7 +75,7 @@ public: | |||
|      * @param buffer Reference to buffer to store data | ||||
|      * @return ResultStatus result of function | ||||
|      */ | ||||
|     virtual ResultStatus ReadIcon(std::vector<u8>& buffer) const { | ||||
|     virtual ResultStatus ReadIcon(std::vector<u8>& buffer) { | ||||
|         return ResultStatus::ErrorNotImplemented; | ||||
|     } | ||||
| 
 | ||||
|  | @ -84,7 +84,7 @@ public: | |||
|      * @param buffer Reference to buffer to store data | ||||
|      * @return ResultStatus result of function | ||||
|      */ | ||||
|     virtual ResultStatus ReadBanner(std::vector<u8>& buffer) const { | ||||
|     virtual ResultStatus ReadBanner(std::vector<u8>& buffer) { | ||||
|         return ResultStatus::ErrorNotImplemented; | ||||
|     } | ||||
| 
 | ||||
|  | @ -93,22 +93,25 @@ public: | |||
|      * @param buffer Reference to buffer to store data | ||||
|      * @return ResultStatus result of function | ||||
|      */ | ||||
|     virtual ResultStatus ReadLogo(std::vector<u8>& buffer) const { | ||||
|     virtual ResultStatus ReadLogo(std::vector<u8>& buffer) { | ||||
|         return ResultStatus::ErrorNotImplemented; | ||||
|     } | ||||
| 
 | ||||
|     /**
 | ||||
|      * 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 | ||||
|      */ | ||||
|     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; | ||||
|     } | ||||
| 
 | ||||
| protected: | ||||
|     std::unique_ptr<FileUtil::IOFile> file; | ||||
|     bool                              is_loaded = false; | ||||
|     FileUtil::IOFile file; | ||||
|     bool is_loaded = false; | ||||
| }; | ||||
| 
 | ||||
| /**
 | ||||
|  |  | |||
|  | @ -117,7 +117,7 @@ FileType AppLoader_NCCH::IdentifyType(FileUtil::IOFile& file) { | |||
|     return FileType::Error; | ||||
| } | ||||
| 
 | ||||
| ResultStatus AppLoader_NCCH::LoadExec() const { | ||||
| ResultStatus AppLoader_NCCH::LoadExec() { | ||||
|     using Kernel::SharedPtr; | ||||
|     using Kernel::CodeSet; | ||||
| 
 | ||||
|  | @ -171,8 +171,8 @@ ResultStatus AppLoader_NCCH::LoadExec() const { | |||
|     return ResultStatus::Error; | ||||
| } | ||||
| 
 | ||||
| ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>& buffer) const { | ||||
|     if (!file->IsOpen()) | ||||
| ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>& buffer) { | ||||
|     if (!file.IsOpen()) | ||||
|         return ResultStatus::Error; | ||||
| 
 | ||||
|     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); | ||||
| 
 | ||||
|             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) { | ||||
|                 // Section is compressed, read compressed .code section...
 | ||||
|  | @ -197,7 +197,7 @@ ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>& | |||
|                     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; | ||||
| 
 | ||||
|                 // Decompress .code section...
 | ||||
|  | @ -208,7 +208,7 @@ ResultStatus AppLoader_NCCH::LoadSectionExeFS(const char* name, std::vector<u8>& | |||
|             } else { | ||||
|                 // Section is uncompressed...
 | ||||
|                 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::Success; | ||||
|  | @ -221,21 +221,21 @@ ResultStatus AppLoader_NCCH::Load() { | |||
|     if (is_loaded) | ||||
|         return ResultStatus::ErrorAlreadyLoaded; | ||||
| 
 | ||||
|     if (!file->IsOpen()) | ||||
|     if (!file.IsOpen()) | ||||
|         return ResultStatus::Error; | ||||
| 
 | ||||
|     // 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; | ||||
| 
 | ||||
|     // Skip NCSD header and load first NCCH (NCSD is just a container of NCCH files)...
 | ||||
|     if (MakeMagic('N', 'C', 'S', 'D') == ncch_header.magic) { | ||||
|         LOG_WARNING(Loader, "Only loading the first (bootable) NCCH within the NCSD file!"); | ||||
|         ncch_offset = 0x4000; | ||||
|         file->Seek(ncch_offset, SEEK_SET); | ||||
|         file->ReadBytes(&ncch_header, sizeof(NCCH_Header)); | ||||
|         file.Seek(ncch_offset, SEEK_SET); | ||||
|         file.ReadBytes(&ncch_header, sizeof(NCCH_Header)); | ||||
|     } | ||||
| 
 | ||||
|     // Verify we are loading the correct file type...
 | ||||
|  | @ -244,7 +244,7 @@ ResultStatus AppLoader_NCCH::Load() { | |||
| 
 | ||||
|     // 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; | ||||
| 
 | ||||
|     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 size:                  0x%08X", exefs_size); | ||||
| 
 | ||||
|     file->Seek(exefs_offset + ncch_offset, SEEK_SET); | ||||
|     if (file->ReadBytes(&exefs_header, sizeof(ExeFs_Header)) != sizeof(ExeFs_Header)) | ||||
|     file.Seek(exefs_offset + ncch_offset, SEEK_SET); | ||||
|     if (file.ReadBytes(&exefs_header, sizeof(ExeFs_Header)) != sizeof(ExeFs_Header)) | ||||
|         return ResultStatus::Error; | ||||
| 
 | ||||
|     is_loaded = true; // Set state to loaded
 | ||||
|  | @ -283,24 +283,24 @@ ResultStatus AppLoader_NCCH::Load() { | |||
|     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); | ||||
| } | ||||
| 
 | ||||
| ResultStatus AppLoader_NCCH::ReadIcon(std::vector<u8>& buffer) const { | ||||
| ResultStatus AppLoader_NCCH::ReadIcon(std::vector<u8>& 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); | ||||
| } | ||||
| 
 | ||||
| ResultStatus AppLoader_NCCH::ReadLogo(std::vector<u8>& buffer) const { | ||||
| ResultStatus AppLoader_NCCH::ReadLogo(std::vector<u8>& buffer) { | ||||
|     return LoadSectionExeFS("logo", buffer); | ||||
| } | ||||
| 
 | ||||
| ResultStatus AppLoader_NCCH::ReadRomFS(std::vector<u8>& buffer) const { | ||||
|     if (!file->IsOpen()) | ||||
| ResultStatus AppLoader_NCCH::ReadRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset, u64& size) { | ||||
|     if (!file.IsOpen()) | ||||
|         return ResultStatus::Error; | ||||
| 
 | ||||
|     // 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 size:             0x%08X", romfs_size); | ||||
| 
 | ||||
|         buffer.resize(romfs_size); | ||||
| 
 | ||||
|         file->Seek(romfs_offset, SEEK_SET); | ||||
|         if (file->ReadBytes(&buffer[0], romfs_size) != romfs_size) | ||||
|         if (file.GetSize () < romfs_offset + romfs_size) | ||||
|             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; | ||||
|     } | ||||
|     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)
 | ||||
| class AppLoader_NCCH final : public AppLoader { | ||||
| 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 | ||||
|  | @ -183,35 +184,35 @@ public: | |||
|      * @param buffer Reference to buffer to store data | ||||
|      * @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 | ||||
|      * @param buffer Reference to buffer to store data | ||||
|      * @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 | ||||
|      * @param buffer Reference to buffer to store data | ||||
|      * @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 | ||||
|      * @param buffer Reference to buffer to store data | ||||
|      * @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 | ||||
|      * @param buffer Reference to buffer to store data | ||||
|      * @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: | ||||
| 
 | ||||
|  | @ -221,13 +222,13 @@ private: | |||
|      * @param buffer Vector to read data into | ||||
|      * @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 | ||||
|      * @return ResultStatus result of function | ||||
|      */ | ||||
|     ResultStatus LoadExec() const; | ||||
|     ResultStatus LoadExec(); | ||||
| 
 | ||||
|     bool            is_compressed = false; | ||||
| 
 | ||||
|  | @ -244,6 +245,8 @@ private: | |||
|     NCCH_Header     ncch_header; | ||||
|     ExeFs_Header    exefs_header; | ||||
|     ExHeader_Header exheader_header; | ||||
| 
 | ||||
|     std::string     filepath; | ||||
| }; | ||||
| 
 | ||||
| } // namespace Loader
 | ||||
|  |  | |||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue