mirror of
				https://github.com/PabloMK7/citra.git
				synced 2025-10-31 05:40:04 +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 { | 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…
	
	Add table
		Add a link
		
	
		Reference in a new issue