summaryrefslogtreecommitdiff
path: root/src/core/loader
diff options
context:
space:
mode:
Diffstat (limited to 'src/core/loader')
-rw-r--r--src/core/loader/deconstructed_rom_directory.cpp113
-rw-r--r--src/core/loader/deconstructed_rom_directory.h14
-rw-r--r--src/core/loader/elf.cpp24
-rw-r--r--src/core/loader/elf.h12
-rw-r--r--src/core/loader/loader.cpp63
-rw-r--r--src/core/loader/loader.h30
-rw-r--r--src/core/loader/nca.cpp252
-rw-r--r--src/core/loader/nca.h19
-rw-r--r--src/core/loader/nro.cpp32
-rw-r--r--src/core/loader/nro.h13
-rw-r--r--src/core/loader/nso.cpp91
-rw-r--r--src/core/loader/nso.h17
12 files changed, 140 insertions, 540 deletions
diff --git a/src/core/loader/deconstructed_rom_directory.cpp b/src/core/loader/deconstructed_rom_directory.cpp
index eb7feb617..0b3b4cd73 100644
--- a/src/core/loader/deconstructed_rom_directory.cpp
+++ b/src/core/loader/deconstructed_rom_directory.cpp
@@ -4,11 +4,10 @@
4 4
5#include <cinttypes> 5#include <cinttypes>
6#include "common/common_funcs.h" 6#include "common/common_funcs.h"
7#include "common/common_paths.h"
8#include "common/file_util.h" 7#include "common/file_util.h"
9#include "common/logging/log.h" 8#include "common/logging/log.h"
10#include "common/string_util.h" 9#include "common/string_util.h"
11#include "core/file_sys/romfs_factory.h" 10#include "core/file_sys/content_archive.h"
12#include "core/hle/kernel/process.h" 11#include "core/hle/kernel/process.h"
13#include "core/hle/kernel/resource_limit.h" 12#include "core/hle/kernel/resource_limit.h"
14#include "core/hle/service/filesystem/filesystem.h" 13#include "core/hle/service/filesystem/filesystem.h"
@@ -46,55 +45,11 @@ static std::string FindRomFS(const std::string& directory) {
46 return filepath_romfs; 45 return filepath_romfs;
47} 46}
48 47
49AppLoader_DeconstructedRomDirectory::AppLoader_DeconstructedRomDirectory(FileUtil::IOFile&& file, 48AppLoader_DeconstructedRomDirectory::AppLoader_DeconstructedRomDirectory(FileSys::VirtualFile file)
50 std::string filepath) 49 : AppLoader(std::move(file)) {}
51 : AppLoader(std::move(file)), filepath(std::move(filepath)) {}
52
53FileType AppLoader_DeconstructedRomDirectory::IdentifyType(FileUtil::IOFile& file,
54 const std::string& filepath) {
55 bool is_main_found{};
56 bool is_npdm_found{};
57 bool is_rtld_found{};
58 bool is_sdk_found{};
59
60 const auto callback = [&](unsigned* num_entries_out, const std::string& directory,
61 const std::string& virtual_name) -> bool {
62 // Skip directories
63 std::string physical_name = directory + virtual_name;
64 if (FileUtil::IsDirectory(physical_name)) {
65 return true;
66 }
67
68 // Verify filename
69 if (Common::ToLower(virtual_name) == "main") {
70 is_main_found = true;
71 } else if (Common::ToLower(virtual_name) == "main.npdm") {
72 is_npdm_found = true;
73 return true;
74 } else if (Common::ToLower(virtual_name) == "rtld") {
75 is_rtld_found = true;
76 } else if (Common::ToLower(virtual_name) == "sdk") {
77 is_sdk_found = true;
78 } else {
79 // Continue searching
80 return true;
81 }
82
83 // Verify file is an NSO
84 FileUtil::IOFile file(physical_name, "rb");
85 if (AppLoader_NSO::IdentifyType(file, physical_name) != FileType::NSO) {
86 return false;
87 }
88
89 // We are done if we've found and verified all required NSOs
90 return !(is_main_found && is_npdm_found && is_rtld_found && is_sdk_found);
91 };
92
93 // Search the directory recursively, looking for the required modules
94 const std::string directory = filepath.substr(0, filepath.find_last_of("/\\")) + DIR_SEP;
95 FileUtil::ForeachDirectoryEntry(nullptr, directory, callback);
96 50
97 if (is_main_found && is_npdm_found && is_rtld_found && is_sdk_found) { 51FileType AppLoader_DeconstructedRomDirectory::IdentifyType(const FileSys::VirtualFile& file) {
52 if (FileSys::IsDirectoryExeFS(file->GetContainingDirectory())) {
98 return FileType::DeconstructedRomDirectory; 53 return FileType::DeconstructedRomDirectory;
99 } 54 }
100 55
@@ -106,14 +61,13 @@ ResultStatus AppLoader_DeconstructedRomDirectory::Load(
106 if (is_loaded) { 61 if (is_loaded) {
107 return ResultStatus::ErrorAlreadyLoaded; 62 return ResultStatus::ErrorAlreadyLoaded;
108 } 63 }
109 if (!file.IsOpen()) {
110 return ResultStatus::Error;
111 }
112 64
113 const std::string directory = filepath.substr(0, filepath.find_last_of("/\\")) + DIR_SEP; 65 const FileSys::VirtualDir dir = file->GetContainingDirectory();
114 const std::string npdm_path = directory + DIR_SEP + "main.npdm"; 66 const FileSys::VirtualFile npdm = dir->GetFile("main.npdm");
67 if (npdm == nullptr)
68 return ResultStatus::ErrorInvalidFormat;
115 69
116 ResultStatus result = metadata.Load(npdm_path); 70 ResultStatus result = metadata.Load(npdm);
117 if (result != ResultStatus::Success) { 71 if (result != ResultStatus::Success) {
118 return result; 72 return result;
119 } 73 }
@@ -128,9 +82,10 @@ ResultStatus AppLoader_DeconstructedRomDirectory::Load(
128 VAddr next_load_addr{Memory::PROCESS_IMAGE_VADDR}; 82 VAddr next_load_addr{Memory::PROCESS_IMAGE_VADDR};
129 for (const auto& module : {"rtld", "main", "subsdk0", "subsdk1", "subsdk2", "subsdk3", 83 for (const auto& module : {"rtld", "main", "subsdk0", "subsdk1", "subsdk2", "subsdk3",
130 "subsdk4", "subsdk5", "subsdk6", "subsdk7", "sdk"}) { 84 "subsdk4", "subsdk5", "subsdk6", "subsdk7", "sdk"}) {
131 const std::string path = directory + DIR_SEP + module;
132 const VAddr load_addr = next_load_addr; 85 const VAddr load_addr = next_load_addr;
133 next_load_addr = AppLoader_NSO::LoadModule(path, load_addr); 86 const FileSys::VirtualFile module_file = dir->GetFile(module);
87 if (module_file != nullptr)
88 next_load_addr = AppLoader_NSO::LoadModule(module_file, load_addr);
134 if (next_load_addr) { 89 if (next_load_addr) {
135 LOG_DEBUG(Loader, "loaded module {} @ 0x{:X}", module, load_addr); 90 LOG_DEBUG(Loader, "loaded module {} @ 0x{:X}", module, load_addr);
136 } else { 91 } else {
@@ -147,41 +102,19 @@ ResultStatus AppLoader_DeconstructedRomDirectory::Load(
147 metadata.GetMainThreadStackSize()); 102 metadata.GetMainThreadStackSize());
148 103
149 // Find the RomFS by searching for a ".romfs" file in this directory 104 // Find the RomFS by searching for a ".romfs" file in this directory
150 filepath_romfs = FindRomFS(directory); 105 const auto& files = dir->GetFiles();
151 106 const auto romfs_iter =
152 // Register the RomFS if a ".romfs" file was found 107 std::find_if(files.begin(), files.end(), [](const FileSys::VirtualFile& file) {
153 if (!filepath_romfs.empty()) { 108 return file->GetName().find(".romfs") != std::string::npos;
154 Service::FileSystem::RegisterFileSystem(std::make_unique<FileSys::RomFS_Factory>(*this), 109 });
155 Service::FileSystem::Type::RomFS);
156 }
157
158 is_loaded = true;
159 return ResultStatus::Success;
160}
161 110
162ResultStatus AppLoader_DeconstructedRomDirectory::ReadRomFS( 111 // TODO(DarkLordZach): Identify RomFS if its a subdirectory.
163 std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset, u64& size) { 112 const auto romfs = (romfs_iter == files.end()) ? nullptr : *romfs_iter;
164 113
165 if (filepath_romfs.empty()) { 114 if (romfs != nullptr)
166 LOG_DEBUG(Loader, "No RomFS available"); 115 Service::FileSystem::RegisterRomFS(romfs);
167 return ResultStatus::ErrorNotUsed;
168 }
169
170 // We reopen the file, to allow its position to be independent
171 romfs_file = std::make_shared<FileUtil::IOFile>(filepath_romfs, "rb");
172 if (!romfs_file->IsOpen()) {
173 return ResultStatus::Error;
174 }
175
176 offset = 0;
177 size = romfs_file->GetSize();
178
179 LOG_DEBUG(Loader, "RomFS offset: 0x{:016X}", offset);
180 LOG_DEBUG(Loader, "RomFS size: 0x{:016X}", size);
181
182 // Reset read pointer
183 file.Seek(0, SEEK_SET);
184 116
117 is_loaded = true;
185 return ResultStatus::Success; 118 return ResultStatus::Success;
186} 119}
187 120
diff --git a/src/core/loader/deconstructed_rom_directory.h b/src/core/loader/deconstructed_rom_directory.h
index 23295d911..494220756 100644
--- a/src/core/loader/deconstructed_rom_directory.h
+++ b/src/core/loader/deconstructed_rom_directory.h
@@ -20,28 +20,22 @@ namespace Loader {
20 */ 20 */
21class AppLoader_DeconstructedRomDirectory final : public AppLoader { 21class AppLoader_DeconstructedRomDirectory final : public AppLoader {
22public: 22public:
23 AppLoader_DeconstructedRomDirectory(FileUtil::IOFile&& file, std::string filepath); 23 explicit AppLoader_DeconstructedRomDirectory(FileSys::VirtualFile main_file);
24 24
25 /** 25 /**
26 * Returns the type of the file 26 * Returns the type of the file
27 * @param file FileUtil::IOFile open file 27 * @param file std::shared_ptr<VfsFile> open file
28 * @param filepath Path of the file that we are opening.
29 * @return FileType found, or FileType::Error if this loader doesn't know it 28 * @return FileType found, or FileType::Error if this loader doesn't know it
30 */ 29 */
31 static FileType IdentifyType(FileUtil::IOFile& file, const std::string& filepath); 30 static FileType IdentifyType(const FileSys::VirtualFile& file);
32 31
33 FileType GetFileType() override { 32 FileType GetFileType() override {
34 return IdentifyType(file, filepath); 33 return IdentifyType(file);
35 } 34 }
36 35
37 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override; 36 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override;
38 37
39 ResultStatus ReadRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset,
40 u64& size) override;
41
42private: 38private:
43 std::string filepath_romfs;
44 std::string filepath;
45 FileSys::ProgramMetadata metadata; 39 FileSys::ProgramMetadata metadata;
46}; 40};
47 41
diff --git a/src/core/loader/elf.cpp b/src/core/loader/elf.cpp
index b69e5c6ef..4bfd5f536 100644
--- a/src/core/loader/elf.cpp
+++ b/src/core/loader/elf.cpp
@@ -365,20 +365,17 @@ SectionID ElfReader::GetSectionByName(const char* name, int firstSection) const
365 365
366namespace Loader { 366namespace Loader {
367 367
368AppLoader_ELF::AppLoader_ELF(FileUtil::IOFile&& file, std::string filename) 368AppLoader_ELF::AppLoader_ELF(FileSys::VirtualFile file) : AppLoader(std::move(file)) {}
369 : AppLoader(std::move(file)), filename(std::move(filename)) {}
370 369
371FileType AppLoader_ELF::IdentifyType(FileUtil::IOFile& file, const std::string&) { 370FileType AppLoader_ELF::IdentifyType(const FileSys::VirtualFile& file) {
372 static constexpr u16 ELF_MACHINE_ARM{0x28}; 371 static constexpr u16 ELF_MACHINE_ARM{0x28};
373 372
374 u32 magic = 0; 373 u32 magic = 0;
375 file.Seek(0, SEEK_SET); 374 if (4 != file->ReadObject(&magic))
376 if (1 != file.ReadArray<u32>(&magic, 1))
377 return FileType::Error; 375 return FileType::Error;
378 376
379 u16 machine = 0; 377 u16 machine = 0;
380 file.Seek(18, SEEK_SET); 378 if (2 != file->ReadObject(&machine, 18))
381 if (1 != file.ReadArray<u16>(&machine, 1))
382 return FileType::Error; 379 return FileType::Error;
383 380
384 if (Common::MakeMagic('\x7f', 'E', 'L', 'F') == magic && ELF_MACHINE_ARM == machine) 381 if (Common::MakeMagic('\x7f', 'E', 'L', 'F') == magic && ELF_MACHINE_ARM == machine)
@@ -391,20 +388,13 @@ ResultStatus AppLoader_ELF::Load(Kernel::SharedPtr<Kernel::Process>& process) {
391 if (is_loaded) 388 if (is_loaded)
392 return ResultStatus::ErrorAlreadyLoaded; 389 return ResultStatus::ErrorAlreadyLoaded;
393 390
394 if (!file.IsOpen()) 391 std::vector<u8> buffer = file->ReadAllBytes();
395 return ResultStatus::Error; 392 if (buffer.size() != file->GetSize())
396
397 // Reset read pointer in case this file has been read before.
398 file.Seek(0, SEEK_SET);
399
400 size_t size = file.GetSize();
401 std::unique_ptr<u8[]> buffer(new u8[size]);
402 if (file.ReadBytes(&buffer[0], size) != size)
403 return ResultStatus::Error; 393 return ResultStatus::Error;
404 394
405 ElfReader elf_reader(&buffer[0]); 395 ElfReader elf_reader(&buffer[0]);
406 SharedPtr<CodeSet> codeset = elf_reader.LoadInto(Memory::PROCESS_IMAGE_VADDR); 396 SharedPtr<CodeSet> codeset = elf_reader.LoadInto(Memory::PROCESS_IMAGE_VADDR);
407 codeset->name = filename; 397 codeset->name = file->GetName();
408 398
409 process->LoadModule(codeset, codeset->entrypoint); 399 process->LoadModule(codeset, codeset->entrypoint);
410 process->svc_access_mask.set(); 400 process->svc_access_mask.set();
diff --git a/src/core/loader/elf.h b/src/core/loader/elf.h
index ee741a789..b8fb982d0 100644
--- a/src/core/loader/elf.h
+++ b/src/core/loader/elf.h
@@ -16,24 +16,20 @@ namespace Loader {
16/// Loads an ELF/AXF file 16/// Loads an ELF/AXF file
17class AppLoader_ELF final : public AppLoader { 17class AppLoader_ELF final : public AppLoader {
18public: 18public:
19 AppLoader_ELF(FileUtil::IOFile&& file, std::string filename); 19 explicit AppLoader_ELF(FileSys::VirtualFile file);
20 20
21 /** 21 /**
22 * Returns the type of the file 22 * Returns the type of the file
23 * @param file FileUtil::IOFile open file 23 * @param file std::shared_ptr<VfsFile> open file
24 * @param filepath Path of the file that we are opening.
25 * @return FileType found, or FileType::Error if this loader doesn't know it 24 * @return FileType found, or FileType::Error if this loader doesn't know it
26 */ 25 */
27 static FileType IdentifyType(FileUtil::IOFile& file, const std::string& filepath); 26 static FileType IdentifyType(const FileSys::VirtualFile& file);
28 27
29 FileType GetFileType() override { 28 FileType GetFileType() override {
30 return IdentifyType(file, filename); 29 return IdentifyType(file);
31 } 30 }
32 31
33 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override; 32 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override;
34
35private:
36 std::string filename;
37}; 33};
38 34
39} // namespace Loader 35} // namespace Loader
diff --git a/src/core/loader/loader.cpp b/src/core/loader/loader.cpp
index 8831d8e83..1574345a1 100644
--- a/src/core/loader/loader.cpp
+++ b/src/core/loader/loader.cpp
@@ -6,6 +6,7 @@
6#include <string> 6#include <string>
7#include "common/logging/log.h" 7#include "common/logging/log.h"
8#include "common/string_util.h" 8#include "common/string_util.h"
9#include "core/file_sys/vfs_real.h"
9#include "core/hle/kernel/process.h" 10#include "core/hle/kernel/process.h"
10#include "core/loader/deconstructed_rom_directory.h" 11#include "core/loader/deconstructed_rom_directory.h"
11#include "core/loader/elf.h" 12#include "core/loader/elf.h"
@@ -21,11 +22,11 @@ const std::initializer_list<Kernel::AddressMapping> default_address_mappings = {
21 {0x1F000000, 0x600000, false}, // entire VRAM 22 {0x1F000000, 0x600000, false}, // entire VRAM
22}; 23};
23 24
24FileType IdentifyFile(FileUtil::IOFile& file, const std::string& filepath) { 25FileType IdentifyFile(FileSys::VirtualFile file) {
25 FileType type; 26 FileType type;
26 27
27#define CHECK_TYPE(loader) \ 28#define CHECK_TYPE(loader) \
28 type = AppLoader_##loader::IdentifyType(file, filepath); \ 29 type = AppLoader_##loader::IdentifyType(file); \
29 if (FileType::Error != type) \ 30 if (FileType::Error != type) \
30 return type; 31 return type;
31 32
@@ -41,25 +42,22 @@ FileType IdentifyFile(FileUtil::IOFile& file, const std::string& filepath) {
41} 42}
42 43
43FileType IdentifyFile(const std::string& file_name) { 44FileType IdentifyFile(const std::string& file_name) {
44 FileUtil::IOFile file(file_name, "rb"); 45 return IdentifyFile(FileSys::VirtualFile(std::make_shared<FileSys::RealVfsFile>(file_name)));
45 if (!file.IsOpen()) {
46 LOG_ERROR(Loader, "Failed to load file {}", file_name);
47 return FileType::Unknown;
48 }
49
50 return IdentifyFile(file, file_name);
51} 46}
52 47
53FileType GuessFromExtension(const std::string& extension_) { 48FileType GuessFromFilename(const std::string& name) {
54 std::string extension = Common::ToLower(extension_); 49 if (name == "main")
50 return FileType::DeconstructedRomDirectory;
55 51
56 if (extension == ".elf") 52 const std::string extension = Common::ToLower(FileUtil::GetExtensionFromFilename(name));
53
54 if (extension == "elf")
57 return FileType::ELF; 55 return FileType::ELF;
58 else if (extension == ".nro") 56 if (extension == "nro")
59 return FileType::NRO; 57 return FileType::NRO;
60 else if (extension == ".nso") 58 if (extension == "nso")
61 return FileType::NSO; 59 return FileType::NSO;
62 else if (extension == ".nca") 60 if (extension == "nca")
63 return FileType::NCA; 61 return FileType::NCA;
64 62
65 return FileType::Unknown; 63 return FileType::Unknown;
@@ -93,58 +91,47 @@ const char* GetFileTypeString(FileType type) {
93 * @param filepath the file full path (with name) 91 * @param filepath the file full path (with name)
94 * @return std::unique_ptr<AppLoader> a pointer to a loader object; nullptr for unsupported type 92 * @return std::unique_ptr<AppLoader> a pointer to a loader object; nullptr for unsupported type
95 */ 93 */
96static std::unique_ptr<AppLoader> GetFileLoader(FileUtil::IOFile&& file, FileType type, 94static std::unique_ptr<AppLoader> GetFileLoader(FileSys::VirtualFile file, FileType type) {
97 const std::string& filename,
98 const std::string& filepath) {
99 switch (type) { 95 switch (type) {
100 96
101 // Standard ELF file format. 97 // Standard ELF file format.
102 case FileType::ELF: 98 case FileType::ELF:
103 return std::make_unique<AppLoader_ELF>(std::move(file), filename); 99 return std::make_unique<AppLoader_ELF>(std::move(file));
104 100
105 // NX NSO file format. 101 // NX NSO file format.
106 case FileType::NSO: 102 case FileType::NSO:
107 return std::make_unique<AppLoader_NSO>(std::move(file), filepath); 103 return std::make_unique<AppLoader_NSO>(std::move(file));
108 104
109 // NX NRO file format. 105 // NX NRO file format.
110 case FileType::NRO: 106 case FileType::NRO:
111 return std::make_unique<AppLoader_NRO>(std::move(file), filepath); 107 return std::make_unique<AppLoader_NRO>(std::move(file));
112 108
113 // NX NCA file format. 109 // NX NCA file format.
114 case FileType::NCA: 110 case FileType::NCA:
115 return std::make_unique<AppLoader_NCA>(std::move(file), filepath); 111 return std::make_unique<AppLoader_NCA>(std::move(file));
116 112
117 // NX deconstructed ROM directory. 113 // NX deconstructed ROM directory.
118 case FileType::DeconstructedRomDirectory: 114 case FileType::DeconstructedRomDirectory:
119 return std::make_unique<AppLoader_DeconstructedRomDirectory>(std::move(file), filepath); 115 return std::make_unique<AppLoader_DeconstructedRomDirectory>(std::move(file));
120 116
121 default: 117 default:
122 return nullptr; 118 return nullptr;
123 } 119 }
124} 120}
125 121
126std::unique_ptr<AppLoader> GetLoader(const std::string& filename) { 122std::unique_ptr<AppLoader> GetLoader(FileSys::VirtualFile file) {
127 FileUtil::IOFile file(filename, "rb"); 123 FileType type = IdentifyFile(file);
128 if (!file.IsOpen()) { 124 FileType filename_type = GuessFromFilename(file->GetName());
129 LOG_ERROR(Loader, "Failed to load file {}", filename);
130 return nullptr;
131 }
132
133 std::string filename_filename, filename_extension;
134 Common::SplitPath(filename, nullptr, &filename_filename, &filename_extension);
135
136 FileType type = IdentifyFile(file, filename);
137 FileType filename_type = GuessFromExtension(filename_extension);
138 125
139 if (type != filename_type) { 126 if (type != filename_type) {
140 LOG_WARNING(Loader, "File {} has a different type than its extension.", filename); 127 LOG_WARNING(Loader, "File {} has a different type than its extension.", file->GetName());
141 if (FileType::Unknown == type) 128 if (FileType::Unknown == type)
142 type = filename_type; 129 type = filename_type;
143 } 130 }
144 131
145 LOG_DEBUG(Loader, "Loading file {} as {}...", filename, GetFileTypeString(type)); 132 LOG_DEBUG(Loader, "Loading file {} as {}...", file->GetName(), GetFileTypeString(type));
146 133
147 return GetFileLoader(std::move(file), type, filename_filename, filename); 134 return GetFileLoader(std::move(file), type);
148} 135}
149 136
150} // namespace Loader 137} // namespace Loader
diff --git a/src/core/loader/loader.h b/src/core/loader/loader.h
index b76f7b13d..1da9e8099 100644
--- a/src/core/loader/loader.h
+++ b/src/core/loader/loader.h
@@ -13,6 +13,7 @@
13#include <boost/optional.hpp> 13#include <boost/optional.hpp>
14#include "common/common_types.h" 14#include "common/common_types.h"
15#include "common/file_util.h" 15#include "common/file_util.h"
16#include "core/file_sys/vfs.h"
16#include "core/hle/kernel/kernel.h" 17#include "core/hle/kernel/kernel.h"
17 18
18namespace Kernel { 19namespace Kernel {
@@ -36,10 +37,9 @@ enum class FileType {
36/** 37/**
37 * Identifies the type of a bootable file based on the magic value in its header. 38 * Identifies the type of a bootable file based on the magic value in its header.
38 * @param file open file 39 * @param file open file
39 * @param filepath Path of the file that we are opening.
40 * @return FileType of file 40 * @return FileType of file
41 */ 41 */
42FileType IdentifyFile(FileUtil::IOFile& file, const std::string& filepath); 42FileType IdentifyFile(FileSys::VirtualFile file);
43 43
44/** 44/**
45 * Identifies the type of a bootable file based on the magic value in its header. 45 * Identifies the type of a bootable file based on the magic value in its header.
@@ -50,12 +50,12 @@ FileType IdentifyFile(FileUtil::IOFile& file, const std::string& filepath);
50FileType IdentifyFile(const std::string& file_name); 50FileType IdentifyFile(const std::string& file_name);
51 51
52/** 52/**
53 * Guess the type of a bootable file from its extension 53 * Guess the type of a bootable file from its name
54 * @param extension String extension of bootable file 54 * @param name String name of bootable file
55 * @return FileType of file. Note: this will return FileType::Unknown if it is unable to determine 55 * @return FileType of file. Note: this will return FileType::Unknown if it is unable to determine
56 * a filetype, and will never return FileType::Error. 56 * a filetype, and will never return FileType::Error.
57 */ 57 */
58FileType GuessFromExtension(const std::string& extension); 58FileType GuessFromFilename(const std::string& name);
59 59
60/** 60/**
61 * Convert a FileType into a string which can be displayed to the user. 61 * Convert a FileType into a string which can be displayed to the user.
@@ -79,7 +79,7 @@ enum class ResultStatus {
79/// Interface for loading an application 79/// Interface for loading an application
80class AppLoader : NonCopyable { 80class AppLoader : NonCopyable {
81public: 81public:
82 AppLoader(FileUtil::IOFile&& file) : file(std::move(file)) {} 82 AppLoader(FileSys::VirtualFile file) : file(std::move(file)) {}
83 virtual ~AppLoader() {} 83 virtual ~AppLoader() {}
84 84
85 /** 85 /**
@@ -154,26 +154,20 @@ public:
154 /** 154 /**
155 * Get the RomFS of the application 155 * Get the RomFS of the application
156 * Since the RomFS can be huge, we return a file reference instead of copying to a buffer 156 * Since the RomFS can be huge, we return a file reference instead of copying to a buffer
157 * @param romfs_file The file containing the RomFS 157 * @param file The file containing the RomFS
158 * @param offset The offset the romfs begins on
159 * @param size The size of the romfs
160 * @return ResultStatus result of function 158 * @return ResultStatus result of function
161 */ 159 */
162 virtual ResultStatus ReadRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset, 160 virtual ResultStatus ReadRomFS(FileSys::VirtualFile& dir) {
163 u64& size) {
164 return ResultStatus::ErrorNotImplemented; 161 return ResultStatus::ErrorNotImplemented;
165 } 162 }
166 163
167 /** 164 /**
168 * Get the update RomFS of the application 165 * Get the update RomFS of the application
169 * Since the RomFS can be huge, we return a file reference instead of copying to a buffer 166 * Since the RomFS can be huge, we return a file reference instead of copying to a buffer
170 * @param romfs_file The file containing the RomFS 167 * @param file The file containing the RomFS
171 * @param offset The offset the romfs begins on
172 * @param size The size of the romfs
173 * @return ResultStatus result of function 168 * @return ResultStatus result of function
174 */ 169 */
175 virtual ResultStatus ReadUpdateRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset, 170 virtual ResultStatus ReadUpdateRomFS(FileSys::VirtualFile& file) {
176 u64& size) {
177 return ResultStatus::ErrorNotImplemented; 171 return ResultStatus::ErrorNotImplemented;
178 } 172 }
179 173
@@ -187,7 +181,7 @@ public:
187 } 181 }
188 182
189protected: 183protected:
190 FileUtil::IOFile file; 184 FileSys::VirtualFile file;
191 bool is_loaded = false; 185 bool is_loaded = false;
192}; 186};
193 187
@@ -202,6 +196,6 @@ extern const std::initializer_list<Kernel::AddressMapping> default_address_mappi
202 * @param filename String filename of bootable file 196 * @param filename String filename of bootable file
203 * @return best loader for this file 197 * @return best loader for this file
204 */ 198 */
205std::unique_ptr<AppLoader> GetLoader(const std::string& filename); 199std::unique_ptr<AppLoader> GetLoader(FileSys::VirtualFile file);
206 200
207} // namespace Loader 201} // namespace Loader
diff --git a/src/core/loader/nca.cpp b/src/core/loader/nca.cpp
index da064f8e3..874f42b91 100644
--- a/src/core/loader/nca.cpp
+++ b/src/core/loader/nca.cpp
@@ -4,13 +4,11 @@
4 4
5#include <vector> 5#include <vector>
6 6
7#include "common/common_funcs.h"
8#include "common/file_util.h" 7#include "common/file_util.h"
9#include "common/logging/log.h" 8#include "common/logging/log.h"
10#include "common/swap.h"
11#include "core/core.h" 9#include "core/core.h"
10#include "core/file_sys/content_archive.h"
12#include "core/file_sys/program_metadata.h" 11#include "core/file_sys/program_metadata.h"
13#include "core/file_sys/romfs_factory.h"
14#include "core/hle/kernel/process.h" 12#include "core/hle/kernel/process.h"
15#include "core/hle/kernel/resource_limit.h" 13#include "core/hle/kernel/resource_limit.h"
16#include "core/hle/service/filesystem/filesystem.h" 14#include "core/hle/service/filesystem/filesystem.h"
@@ -20,208 +18,15 @@
20 18
21namespace Loader { 19namespace Loader {
22 20
23// Media offsets in headers are stored divided by 512. Mult. by this to get real offset. 21AppLoader_NCA::AppLoader_NCA(FileSys::VirtualFile file) : AppLoader(file) {}
24constexpr u64 MEDIA_OFFSET_MULTIPLIER = 0x200;
25
26constexpr u64 SECTION_HEADER_SIZE = 0x200;
27constexpr u64 SECTION_HEADER_OFFSET = 0x400;
28
29enum class NcaContentType : u8 { Program = 0, Meta = 1, Control = 2, Manual = 3, Data = 4 };
30
31enum class NcaSectionFilesystemType : u8 { PFS0 = 0x2, ROMFS = 0x3 };
32
33struct NcaSectionTableEntry {
34 u32_le media_offset;
35 u32_le media_end_offset;
36 INSERT_PADDING_BYTES(0x8);
37};
38static_assert(sizeof(NcaSectionTableEntry) == 0x10, "NcaSectionTableEntry has incorrect size.");
39
40struct NcaHeader {
41 std::array<u8, 0x100> rsa_signature_1;
42 std::array<u8, 0x100> rsa_signature_2;
43 u32_le magic;
44 u8 is_system;
45 NcaContentType content_type;
46 u8 crypto_type;
47 u8 key_index;
48 u64_le size;
49 u64_le title_id;
50 INSERT_PADDING_BYTES(0x4);
51 u32_le sdk_version;
52 u8 crypto_type_2;
53 INSERT_PADDING_BYTES(15);
54 std::array<u8, 0x10> rights_id;
55 std::array<NcaSectionTableEntry, 0x4> section_tables;
56 std::array<std::array<u8, 0x20>, 0x4> hash_tables;
57 std::array<std::array<u8, 0x10>, 0x4> key_area;
58 INSERT_PADDING_BYTES(0xC0);
59};
60static_assert(sizeof(NcaHeader) == 0x400, "NcaHeader has incorrect size.");
61
62struct NcaSectionHeaderBlock {
63 INSERT_PADDING_BYTES(3);
64 NcaSectionFilesystemType filesystem_type;
65 u8 crypto_type;
66 INSERT_PADDING_BYTES(3);
67};
68static_assert(sizeof(NcaSectionHeaderBlock) == 0x8, "NcaSectionHeaderBlock has incorrect size.");
69
70struct Pfs0Superblock {
71 NcaSectionHeaderBlock header_block;
72 std::array<u8, 0x20> hash;
73 u32_le size;
74 INSERT_PADDING_BYTES(4);
75 u64_le hash_table_offset;
76 u64_le hash_table_size;
77 u64_le pfs0_header_offset;
78 u64_le pfs0_size;
79 INSERT_PADDING_BYTES(432);
80};
81static_assert(sizeof(Pfs0Superblock) == 0x200, "Pfs0Superblock has incorrect size.");
82
83static bool IsValidNca(const NcaHeader& header) {
84 return header.magic == Common::MakeMagic('N', 'C', 'A', '2') ||
85 header.magic == Common::MakeMagic('N', 'C', 'A', '3');
86}
87
88// TODO(DarkLordZach): Add support for encrypted.
89class Nca final {
90 std::vector<FileSys::PartitionFilesystem> pfs;
91 std::vector<u64> pfs_offset;
92
93 u64 romfs_offset = 0;
94 u64 romfs_size = 0;
95
96 boost::optional<u8> exefs_id = boost::none;
97
98 FileUtil::IOFile file;
99 std::string path;
100
101 u64 GetExeFsFileOffset(const std::string& file_name) const;
102 u64 GetExeFsFileSize(const std::string& file_name) const;
103
104public:
105 ResultStatus Load(FileUtil::IOFile&& file, std::string path);
106
107 FileSys::PartitionFilesystem GetPfs(u8 id) const;
108
109 u64 GetRomFsOffset() const;
110 u64 GetRomFsSize() const;
111
112 std::vector<u8> GetExeFsFile(const std::string& file_name);
113};
114
115static bool IsPfsExeFs(const FileSys::PartitionFilesystem& pfs) {
116 // According to switchbrew, an exefs must only contain these two files:
117 return pfs.GetFileSize("main") > 0 && pfs.GetFileSize("main.npdm") > 0;
118}
119
120ResultStatus Nca::Load(FileUtil::IOFile&& in_file, std::string in_path) {
121 file = std::move(in_file);
122 path = in_path;
123 file.Seek(0, SEEK_SET);
124 std::array<u8, sizeof(NcaHeader)> header_array{};
125 if (sizeof(NcaHeader) != file.ReadBytes(header_array.data(), sizeof(NcaHeader)))
126 LOG_CRITICAL(Loader, "File reader errored out during header read.");
127
128 NcaHeader header{};
129 std::memcpy(&header, header_array.data(), sizeof(NcaHeader));
130 if (!IsValidNca(header))
131 return ResultStatus::ErrorInvalidFormat;
132
133 int number_sections =
134 std::count_if(std::begin(header.section_tables), std::end(header.section_tables),
135 [](NcaSectionTableEntry entry) { return entry.media_offset > 0; });
136
137 for (int i = 0; i < number_sections; ++i) {
138 // Seek to beginning of this section.
139 file.Seek(SECTION_HEADER_OFFSET + i * SECTION_HEADER_SIZE, SEEK_SET);
140 std::array<u8, sizeof(NcaSectionHeaderBlock)> array{};
141 if (sizeof(NcaSectionHeaderBlock) !=
142 file.ReadBytes(array.data(), sizeof(NcaSectionHeaderBlock)))
143 LOG_CRITICAL(Loader, "File reader errored out during header read.");
144
145 NcaSectionHeaderBlock block{};
146 std::memcpy(&block, array.data(), sizeof(NcaSectionHeaderBlock));
147
148 if (block.filesystem_type == NcaSectionFilesystemType::ROMFS) {
149 romfs_offset = header.section_tables[i].media_offset * MEDIA_OFFSET_MULTIPLIER;
150 romfs_size =
151 header.section_tables[i].media_end_offset * MEDIA_OFFSET_MULTIPLIER - romfs_offset;
152 } else if (block.filesystem_type == NcaSectionFilesystemType::PFS0) {
153 Pfs0Superblock sb{};
154 // Seek back to beginning of this section.
155 file.Seek(SECTION_HEADER_OFFSET + i * SECTION_HEADER_SIZE, SEEK_SET);
156 if (sizeof(Pfs0Superblock) != file.ReadBytes(&sb, sizeof(Pfs0Superblock)))
157 LOG_CRITICAL(Loader, "File reader errored out during header read.");
158
159 u64 offset = (static_cast<u64>(header.section_tables[i].media_offset) *
160 MEDIA_OFFSET_MULTIPLIER) +
161 sb.pfs0_header_offset;
162 FileSys::PartitionFilesystem npfs{};
163 ResultStatus status = npfs.Load(path, offset);
164
165 if (status == ResultStatus::Success) {
166 pfs.emplace_back(std::move(npfs));
167 pfs_offset.emplace_back(offset);
168 }
169 }
170 }
171
172 for (size_t i = 0; i < pfs.size(); ++i) {
173 if (IsPfsExeFs(pfs[i]))
174 exefs_id = i;
175 }
176
177 return ResultStatus::Success;
178}
179
180FileSys::PartitionFilesystem Nca::GetPfs(u8 id) const {
181 return pfs[id];
182}
183
184u64 Nca::GetExeFsFileOffset(const std::string& file_name) const {
185 if (exefs_id == boost::none)
186 return 0;
187 return pfs[*exefs_id].GetFileOffset(file_name) + pfs_offset[*exefs_id];
188}
189
190u64 Nca::GetExeFsFileSize(const std::string& file_name) const {
191 if (exefs_id == boost::none)
192 return 0;
193 return pfs[*exefs_id].GetFileSize(file_name);
194}
195
196u64 Nca::GetRomFsOffset() const {
197 return romfs_offset;
198}
199
200u64 Nca::GetRomFsSize() const {
201 return romfs_size;
202}
203
204std::vector<u8> Nca::GetExeFsFile(const std::string& file_name) {
205 std::vector<u8> out(GetExeFsFileSize(file_name));
206 file.Seek(GetExeFsFileOffset(file_name), SEEK_SET);
207 file.ReadBytes(out.data(), GetExeFsFileSize(file_name));
208 return out;
209}
210
211AppLoader_NCA::AppLoader_NCA(FileUtil::IOFile&& file, std::string filepath)
212 : AppLoader(std::move(file)), filepath(std::move(filepath)) {}
213
214FileType AppLoader_NCA::IdentifyType(FileUtil::IOFile& file, const std::string&) {
215 file.Seek(0, SEEK_SET);
216 std::array<u8, 0x400> header_enc_array{};
217 if (0x400 != file.ReadBytes(header_enc_array.data(), 0x400))
218 return FileType::Error;
219 22
23FileType AppLoader_NCA::IdentifyType(const FileSys::VirtualFile& file) {
220 // TODO(DarkLordZach): Assuming everything is decrypted. Add crypto support. 24 // TODO(DarkLordZach): Assuming everything is decrypted. Add crypto support.
221 NcaHeader header{}; 25 FileSys::NCAHeader header{};
222 std::memcpy(&header, header_enc_array.data(), sizeof(NcaHeader)); 26 if (sizeof(FileSys::NCAHeader) != file->ReadObject(&header))
27 return FileType::Error;
223 28
224 if (IsValidNca(header) && header.content_type == NcaContentType::Program) 29 if (IsValidNCA(header) && header.content_type == FileSys::NCAContentType::Program)
225 return FileType::NCA; 30 return FileType::NCA;
226 31
227 return FileType::Error; 32 return FileType::Error;
@@ -231,17 +36,22 @@ ResultStatus AppLoader_NCA::Load(Kernel::SharedPtr<Kernel::Process>& process) {
231 if (is_loaded) { 36 if (is_loaded) {
232 return ResultStatus::ErrorAlreadyLoaded; 37 return ResultStatus::ErrorAlreadyLoaded;
233 } 38 }
234 if (!file.IsOpen()) {
235 return ResultStatus::Error;
236 }
237 39
238 nca = std::make_unique<Nca>(); 40 nca = std::make_unique<FileSys::NCA>(file);
239 ResultStatus result = nca->Load(std::move(file), filepath); 41 ResultStatus result = nca->GetStatus();
240 if (result != ResultStatus::Success) { 42 if (result != ResultStatus::Success) {
241 return result; 43 return result;
242 } 44 }
243 45
244 result = metadata.Load(nca->GetExeFsFile("main.npdm")); 46 if (nca->GetType() != FileSys::NCAContentType::Program)
47 return ResultStatus::ErrorInvalidFormat;
48
49 auto exefs = nca->GetExeFS();
50
51 if (exefs == nullptr)
52 return ResultStatus::ErrorInvalidFormat;
53
54 result = metadata.Load(exefs->GetFile("main.npdm"));
245 if (result != ResultStatus::Success) { 55 if (result != ResultStatus::Success) {
246 return result; 56 return result;
247 } 57 }
@@ -256,7 +66,8 @@ ResultStatus AppLoader_NCA::Load(Kernel::SharedPtr<Kernel::Process>& process) {
256 for (const auto& module : {"rtld", "main", "subsdk0", "subsdk1", "subsdk2", "subsdk3", 66 for (const auto& module : {"rtld", "main", "subsdk0", "subsdk1", "subsdk2", "subsdk3",
257 "subsdk4", "subsdk5", "subsdk6", "subsdk7", "sdk"}) { 67 "subsdk4", "subsdk5", "subsdk6", "subsdk7", "sdk"}) {
258 const VAddr load_addr = next_load_addr; 68 const VAddr load_addr = next_load_addr;
259 next_load_addr = AppLoader_NSO::LoadModule(module, nca->GetExeFsFile(module), load_addr); 69
70 next_load_addr = AppLoader_NSO::LoadModule(exefs->GetFile(module), load_addr);
260 if (next_load_addr) { 71 if (next_load_addr) {
261 LOG_DEBUG(Loader, "loaded module {} @ 0x{:X}", module, load_addr); 72 LOG_DEBUG(Loader, "loaded module {} @ 0x{:X}", module, load_addr);
262 } else { 73 } else {
@@ -272,28 +83,11 @@ ResultStatus AppLoader_NCA::Load(Kernel::SharedPtr<Kernel::Process>& process) {
272 process->Run(Memory::PROCESS_IMAGE_VADDR, metadata.GetMainThreadPriority(), 83 process->Run(Memory::PROCESS_IMAGE_VADDR, metadata.GetMainThreadPriority(),
273 metadata.GetMainThreadStackSize()); 84 metadata.GetMainThreadStackSize());
274 85
275 if (nca->GetRomFsSize() > 0)
276 Service::FileSystem::RegisterFileSystem(std::make_unique<FileSys::RomFS_Factory>(*this),
277 Service::FileSystem::Type::RomFS);
278
279 is_loaded = true; 86 is_loaded = true;
280 return ResultStatus::Success;
281}
282
283ResultStatus AppLoader_NCA::ReadRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset,
284 u64& size) {
285 if (nca->GetRomFsSize() == 0) {
286 LOG_DEBUG(Loader, "No RomFS available");
287 return ResultStatus::ErrorNotUsed;
288 }
289
290 romfs_file = std::make_shared<FileUtil::IOFile>(filepath, "rb");
291
292 offset = nca->GetRomFsOffset();
293 size = nca->GetRomFsSize();
294 87
295 LOG_DEBUG(Loader, "RomFS offset: 0x{:016X}", offset); 88 const auto romfs = nca->GetRomFS();
296 LOG_DEBUG(Loader, "RomFS size: 0x{:016X}", size); 89 if (romfs != nullptr)
90 Service::FileSystem::RegisterRomFS(romfs);
297 91
298 return ResultStatus::Success; 92 return ResultStatus::Success;
299} 93}
diff --git a/src/core/loader/nca.h b/src/core/loader/nca.h
index 3b6c451d0..a5639f149 100644
--- a/src/core/loader/nca.h
+++ b/src/core/loader/nca.h
@@ -6,44 +6,37 @@
6 6
7#include <string> 7#include <string>
8#include "common/common_types.h" 8#include "common/common_types.h"
9#include "core/file_sys/partition_filesystem.h" 9#include "core/file_sys/content_archive.h"
10#include "core/file_sys/program_metadata.h" 10#include "core/file_sys/program_metadata.h"
11#include "core/hle/kernel/kernel.h" 11#include "core/hle/kernel/kernel.h"
12#include "core/loader/loader.h" 12#include "core/loader/loader.h"
13 13
14namespace Loader { 14namespace Loader {
15 15
16class Nca;
17
18/// Loads an NCA file 16/// Loads an NCA file
19class AppLoader_NCA final : public AppLoader { 17class AppLoader_NCA final : public AppLoader {
20public: 18public:
21 AppLoader_NCA(FileUtil::IOFile&& file, std::string filepath); 19 explicit AppLoader_NCA(FileSys::VirtualFile file);
22 20
23 /** 21 /**
24 * Returns the type of the file 22 * Returns the type of the file
25 * @param file FileUtil::IOFile open file 23 * @param file std::shared_ptr<VfsFile> open file
26 * @param filepath Path of the file that we are opening.
27 * @return FileType found, or FileType::Error if this loader doesn't know it 24 * @return FileType found, or FileType::Error if this loader doesn't know it
28 */ 25 */
29 static FileType IdentifyType(FileUtil::IOFile& file, const std::string& filepath); 26 static FileType IdentifyType(const FileSys::VirtualFile& file);
30 27
31 FileType GetFileType() override { 28 FileType GetFileType() override {
32 return IdentifyType(file, filepath); 29 return IdentifyType(file);
33 } 30 }
34 31
35 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override; 32 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override;
36 33
37 ResultStatus ReadRomFS(std::shared_ptr<FileUtil::IOFile>& romfs_file, u64& offset,
38 u64& size) override;
39
40 ~AppLoader_NCA(); 34 ~AppLoader_NCA();
41 35
42private: 36private:
43 std::string filepath;
44 FileSys::ProgramMetadata metadata; 37 FileSys::ProgramMetadata metadata;
45 38
46 std::unique_ptr<Nca> nca; 39 std::unique_ptr<FileSys::NCA> nca;
47}; 40};
48 41
49} // namespace Loader 42} // namespace Loader
diff --git a/src/core/loader/nro.cpp b/src/core/loader/nro.cpp
index 3853cfa1a..08b7aad7a 100644
--- a/src/core/loader/nro.cpp
+++ b/src/core/loader/nro.cpp
@@ -47,14 +47,12 @@ struct ModHeader {
47}; 47};
48static_assert(sizeof(ModHeader) == 0x1c, "ModHeader has incorrect size."); 48static_assert(sizeof(ModHeader) == 0x1c, "ModHeader has incorrect size.");
49 49
50AppLoader_NRO::AppLoader_NRO(FileUtil::IOFile&& file, std::string filepath) 50AppLoader_NRO::AppLoader_NRO(FileSys::VirtualFile file) : AppLoader(file) {}
51 : AppLoader(std::move(file)), filepath(std::move(filepath)) {}
52 51
53FileType AppLoader_NRO::IdentifyType(FileUtil::IOFile& file, const std::string&) { 52FileType AppLoader_NRO::IdentifyType(const FileSys::VirtualFile& file) {
54 // Read NSO header 53 // Read NSO header
55 NroHeader nro_header{}; 54 NroHeader nro_header{};
56 file.Seek(0, SEEK_SET); 55 if (sizeof(NroHeader) != file->ReadObject(&nro_header)) {
57 if (sizeof(NroHeader) != file.ReadBytes(&nro_header, sizeof(NroHeader))) {
58 return FileType::Error; 56 return FileType::Error;
59 } 57 }
60 if (nro_header.magic == Common::MakeMagic('N', 'R', 'O', '0')) { 58 if (nro_header.magic == Common::MakeMagic('N', 'R', 'O', '0')) {
@@ -67,16 +65,10 @@ static constexpr u32 PageAlignSize(u32 size) {
67 return (size + Memory::PAGE_MASK) & ~Memory::PAGE_MASK; 65 return (size + Memory::PAGE_MASK) & ~Memory::PAGE_MASK;
68} 66}
69 67
70bool AppLoader_NRO::LoadNro(const std::string& path, VAddr load_base) { 68bool AppLoader_NRO::LoadNro(FileSys::VirtualFile file, VAddr load_base) {
71 FileUtil::IOFile file(path, "rb");
72 if (!file.IsOpen()) {
73 return {};
74 }
75
76 // Read NSO header 69 // Read NSO header
77 NroHeader nro_header{}; 70 NroHeader nro_header{};
78 file.Seek(0, SEEK_SET); 71 if (sizeof(NroHeader) != file->ReadObject(&nro_header)) {
79 if (sizeof(NroHeader) != file.ReadBytes(&nro_header, sizeof(NroHeader))) {
80 return {}; 72 return {};
81 } 73 }
82 if (nro_header.magic != Common::MakeMagic('N', 'R', 'O', '0')) { 74 if (nro_header.magic != Common::MakeMagic('N', 'R', 'O', '0')) {
@@ -85,10 +77,9 @@ bool AppLoader_NRO::LoadNro(const std::string& path, VAddr load_base) {
85 77
86 // Build program image 78 // Build program image
87 Kernel::SharedPtr<Kernel::CodeSet> codeset = Kernel::CodeSet::Create(""); 79 Kernel::SharedPtr<Kernel::CodeSet> codeset = Kernel::CodeSet::Create("");
88 std::vector<u8> program_image; 80 std::vector<u8> program_image = file->ReadBytes(PageAlignSize(nro_header.file_size));
89 program_image.resize(PageAlignSize(nro_header.file_size)); 81 if (program_image.size() != PageAlignSize(nro_header.file_size))
90 file.Seek(0, SEEK_SET); 82 return {};
91 file.ReadBytes(program_image.data(), nro_header.file_size);
92 83
93 for (int i = 0; i < nro_header.segments.size(); ++i) { 84 for (int i = 0; i < nro_header.segments.size(); ++i) {
94 codeset->segments[i].addr = nro_header.segments[i].offset; 85 codeset->segments[i].addr = nro_header.segments[i].offset;
@@ -111,7 +102,7 @@ bool AppLoader_NRO::LoadNro(const std::string& path, VAddr load_base) {
111 program_image.resize(static_cast<u32>(program_image.size()) + bss_size); 102 program_image.resize(static_cast<u32>(program_image.size()) + bss_size);
112 103
113 // Load codeset for current process 104 // Load codeset for current process
114 codeset->name = path; 105 codeset->name = file->GetName();
115 codeset->memory = std::make_shared<std::vector<u8>>(std::move(program_image)); 106 codeset->memory = std::make_shared<std::vector<u8>>(std::move(program_image));
116 Core::CurrentProcess()->LoadModule(codeset, load_base); 107 Core::CurrentProcess()->LoadModule(codeset, load_base);
117 108
@@ -122,14 +113,11 @@ ResultStatus AppLoader_NRO::Load(Kernel::SharedPtr<Kernel::Process>& process) {
122 if (is_loaded) { 113 if (is_loaded) {
123 return ResultStatus::ErrorAlreadyLoaded; 114 return ResultStatus::ErrorAlreadyLoaded;
124 } 115 }
125 if (!file.IsOpen()) {
126 return ResultStatus::Error;
127 }
128 116
129 // Load NRO 117 // Load NRO
130 static constexpr VAddr base_addr{Memory::PROCESS_IMAGE_VADDR}; 118 static constexpr VAddr base_addr{Memory::PROCESS_IMAGE_VADDR};
131 119
132 if (!LoadNro(filepath, base_addr)) { 120 if (!LoadNro(file, base_addr)) {
133 return ResultStatus::ErrorInvalidFormat; 121 return ResultStatus::ErrorInvalidFormat;
134 } 122 }
135 123
diff --git a/src/core/loader/nro.h b/src/core/loader/nro.h
index 599adb253..2c03d06bb 100644
--- a/src/core/loader/nro.h
+++ b/src/core/loader/nro.h
@@ -15,26 +15,23 @@ namespace Loader {
15/// Loads an NRO file 15/// Loads an NRO file
16class AppLoader_NRO final : public AppLoader, Linker { 16class AppLoader_NRO final : public AppLoader, Linker {
17public: 17public:
18 AppLoader_NRO(FileUtil::IOFile&& file, std::string filepath); 18 AppLoader_NRO(FileSys::VirtualFile file);
19 19
20 /** 20 /**
21 * Returns the type of the file 21 * Returns the type of the file
22 * @param file FileUtil::IOFile open file 22 * @param file std::shared_ptr<VfsFile> open file
23 * @param filepath Path of the file that we are opening.
24 * @return FileType found, or FileType::Error if this loader doesn't know it 23 * @return FileType found, or FileType::Error if this loader doesn't know it
25 */ 24 */
26 static FileType IdentifyType(FileUtil::IOFile& file, const std::string& filepath); 25 static FileType IdentifyType(const FileSys::VirtualFile& file);
27 26
28 FileType GetFileType() override { 27 FileType GetFileType() override {
29 return IdentifyType(file, filepath); 28 return IdentifyType(file);
30 } 29 }
31 30
32 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override; 31 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override;
33 32
34private: 33private:
35 bool LoadNro(const std::string& path, VAddr load_base); 34 bool LoadNro(FileSys::VirtualFile file, VAddr load_base);
36
37 std::string filepath;
38}; 35};
39 36
40} // namespace Loader 37} // namespace Loader
diff --git a/src/core/loader/nso.cpp b/src/core/loader/nso.cpp
index 7f84e4b1b..2ed12a5ab 100644
--- a/src/core/loader/nso.cpp
+++ b/src/core/loader/nso.cpp
@@ -37,6 +37,7 @@ struct NsoHeader {
37 std::array<u32_le, 3> segments_compressed_size; 37 std::array<u32_le, 3> segments_compressed_size;
38}; 38};
39static_assert(sizeof(NsoHeader) == 0x6c, "NsoHeader has incorrect size."); 39static_assert(sizeof(NsoHeader) == 0x6c, "NsoHeader has incorrect size.");
40static_assert(std::is_trivially_copyable_v<NsoHeader>, "NsoHeader isn't trivially copyable.");
40 41
41struct ModHeader { 42struct ModHeader {
42 u32_le magic; 43 u32_le magic;
@@ -49,15 +50,11 @@ struct ModHeader {
49}; 50};
50static_assert(sizeof(ModHeader) == 0x1c, "ModHeader has incorrect size."); 51static_assert(sizeof(ModHeader) == 0x1c, "ModHeader has incorrect size.");
51 52
52AppLoader_NSO::AppLoader_NSO(FileUtil::IOFile&& file, std::string filepath) 53AppLoader_NSO::AppLoader_NSO(FileSys::VirtualFile file) : AppLoader(std::move(file)) {}
53 : AppLoader(std::move(file)), filepath(std::move(filepath)) {}
54 54
55FileType AppLoader_NSO::IdentifyType(FileUtil::IOFile& file, const std::string&) { 55FileType AppLoader_NSO::IdentifyType(const FileSys::VirtualFile& file) {
56 u32 magic = 0; 56 u32 magic = 0;
57 file.Seek(0, SEEK_SET); 57 file->ReadObject(&magic);
58 if (1 != file.ReadArray<u32>(&magic, 1)) {
59 return FileType::Error;
60 }
61 58
62 if (Common::MakeMagic('N', 'S', 'O', '0') == magic) { 59 if (Common::MakeMagic('N', 'S', 'O', '0') == magic) {
63 return FileType::NSO; 60 return FileType::NSO;
@@ -98,80 +95,27 @@ static constexpr u32 PageAlignSize(u32 size) {
98 return (size + Memory::PAGE_MASK) & ~Memory::PAGE_MASK; 95 return (size + Memory::PAGE_MASK) & ~Memory::PAGE_MASK;
99} 96}
100 97
101VAddr AppLoader_NSO::LoadModule(const std::string& name, const std::vector<u8>& file_data, 98VAddr AppLoader_NSO::LoadModule(FileSys::VirtualFile file, VAddr load_base) {
102 VAddr load_base) { 99 if (file == nullptr)
103 if (file_data.size() < sizeof(NsoHeader))
104 return {}; 100 return {};
105 101
106 NsoHeader nso_header; 102 if (file->GetSize() < sizeof(NsoHeader))
107 std::memcpy(&nso_header, file_data.data(), sizeof(NsoHeader));
108
109 if (nso_header.magic != Common::MakeMagic('N', 'S', 'O', '0'))
110 return {}; 103 return {};
111 104
112 // Build program image
113 Kernel::SharedPtr<Kernel::CodeSet> codeset = Kernel::CodeSet::Create("");
114 std::vector<u8> program_image;
115 for (int i = 0; i < nso_header.segments.size(); ++i) {
116 std::vector<u8> compressed_data(nso_header.segments_compressed_size[i]);
117 for (int j = 0; j < nso_header.segments_compressed_size[i]; ++j)
118 compressed_data[j] = file_data[nso_header.segments[i].offset + j];
119 std::vector<u8> data = DecompressSegment(compressed_data, nso_header.segments[i]);
120 program_image.resize(nso_header.segments[i].location);
121 program_image.insert(program_image.end(), data.begin(), data.end());
122 codeset->segments[i].addr = nso_header.segments[i].location;
123 codeset->segments[i].offset = nso_header.segments[i].location;
124 codeset->segments[i].size = PageAlignSize(static_cast<u32>(data.size()));
125 }
126
127 // MOD header pointer is at .text offset + 4
128 u32 module_offset;
129 std::memcpy(&module_offset, program_image.data() + 4, sizeof(u32));
130
131 // Read MOD header
132 ModHeader mod_header{};
133 // Default .bss to size in segment header if MOD0 section doesn't exist
134 u32 bss_size{PageAlignSize(nso_header.segments[2].bss_size)};
135 std::memcpy(&mod_header, program_image.data() + module_offset, sizeof(ModHeader));
136 const bool has_mod_header{mod_header.magic == Common::MakeMagic('M', 'O', 'D', '0')};
137 if (has_mod_header) {
138 // Resize program image to include .bss section and page align each section
139 bss_size = PageAlignSize(mod_header.bss_end_offset - mod_header.bss_start_offset);
140 }
141 codeset->data.size += bss_size;
142 const u32 image_size{PageAlignSize(static_cast<u32>(program_image.size()) + bss_size)};
143 program_image.resize(image_size);
144
145 // Load codeset for current process
146 codeset->name = name;
147 codeset->memory = std::make_shared<std::vector<u8>>(std::move(program_image));
148 Core::CurrentProcess()->LoadModule(codeset, load_base);
149
150 return load_base + image_size;
151}
152
153VAddr AppLoader_NSO::LoadModule(const std::string& path, VAddr load_base) {
154 FileUtil::IOFile file(path, "rb");
155 if (!file.IsOpen()) {
156 return {};
157 }
158
159 // Read NSO header
160 NsoHeader nso_header{}; 105 NsoHeader nso_header{};
161 file.Seek(0, SEEK_SET); 106 if (sizeof(NsoHeader) != file->ReadObject(&nso_header))
162 if (sizeof(NsoHeader) != file.ReadBytes(&nso_header, sizeof(NsoHeader))) {
163 return {}; 107 return {};
164 } 108
165 if (nso_header.magic != Common::MakeMagic('N', 'S', 'O', '0')) { 109 if (nso_header.magic != Common::MakeMagic('N', 'S', 'O', '0'))
166 return {}; 110 return {};
167 }
168 111
169 // Build program image 112 // Build program image
170 Kernel::SharedPtr<Kernel::CodeSet> codeset = Kernel::CodeSet::Create(""); 113 Kernel::SharedPtr<Kernel::CodeSet> codeset = Kernel::CodeSet::Create("");
171 std::vector<u8> program_image; 114 std::vector<u8> program_image;
172 for (int i = 0; i < nso_header.segments.size(); ++i) { 115 for (int i = 0; i < nso_header.segments.size(); ++i) {
173 std::vector<u8> data = 116 const std::vector<u8> compressed_data =
174 ReadSegment(file, nso_header.segments[i], nso_header.segments_compressed_size[i]); 117 file->ReadBytes(nso_header.segments_compressed_size[i], nso_header.segments[i].offset);
118 std::vector<u8> data = DecompressSegment(compressed_data, nso_header.segments[i]);
175 program_image.resize(nso_header.segments[i].location); 119 program_image.resize(nso_header.segments[i].location);
176 program_image.insert(program_image.end(), data.begin(), data.end()); 120 program_image.insert(program_image.end(), data.begin(), data.end());
177 codeset->segments[i].addr = nso_header.segments[i].location; 121 codeset->segments[i].addr = nso_header.segments[i].location;
@@ -198,7 +142,7 @@ VAddr AppLoader_NSO::LoadModule(const std::string& path, VAddr load_base) {
198 program_image.resize(image_size); 142 program_image.resize(image_size);
199 143
200 // Load codeset for current process 144 // Load codeset for current process
201 codeset->name = path; 145 codeset->name = file->GetName();
202 codeset->memory = std::make_shared<std::vector<u8>>(std::move(program_image)); 146 codeset->memory = std::make_shared<std::vector<u8>>(std::move(program_image));
203 Core::CurrentProcess()->LoadModule(codeset, load_base); 147 Core::CurrentProcess()->LoadModule(codeset, load_base);
204 148
@@ -209,13 +153,10 @@ ResultStatus AppLoader_NSO::Load(Kernel::SharedPtr<Kernel::Process>& process) {
209 if (is_loaded) { 153 if (is_loaded) {
210 return ResultStatus::ErrorAlreadyLoaded; 154 return ResultStatus::ErrorAlreadyLoaded;
211 } 155 }
212 if (!file.IsOpen()) {
213 return ResultStatus::Error;
214 }
215 156
216 // Load module 157 // Load module
217 LoadModule(filepath, Memory::PROCESS_IMAGE_VADDR); 158 LoadModule(file, Memory::PROCESS_IMAGE_VADDR);
218 LOG_DEBUG(Loader, "loaded module {} @ 0x{:X}", filepath, Memory::PROCESS_IMAGE_VADDR); 159 LOG_DEBUG(Loader, "loaded module {} @ 0x{:X}", file->GetName(), Memory::PROCESS_IMAGE_VADDR);
219 160
220 process->svc_access_mask.set(); 161 process->svc_access_mask.set();
221 process->address_mappings = default_address_mappings; 162 process->address_mappings = default_address_mappings;
diff --git a/src/core/loader/nso.h b/src/core/loader/nso.h
index 386f4d39a..3f7567500 100644
--- a/src/core/loader/nso.h
+++ b/src/core/loader/nso.h
@@ -15,29 +15,22 @@ namespace Loader {
15/// Loads an NSO file 15/// Loads an NSO file
16class AppLoader_NSO final : public AppLoader, Linker { 16class AppLoader_NSO final : public AppLoader, Linker {
17public: 17public:
18 AppLoader_NSO(FileUtil::IOFile&& file, std::string filepath); 18 explicit AppLoader_NSO(FileSys::VirtualFile file);
19 19
20 /** 20 /**
21 * Returns the type of the file 21 * Returns the type of the file
22 * @param file FileUtil::IOFile open file 22 * @param file std::shared_ptr<VfsFile> open file
23 * @param filepath Path of the file that we are opening.
24 * @return FileType found, or FileType::Error if this loader doesn't know it 23 * @return FileType found, or FileType::Error if this loader doesn't know it
25 */ 24 */
26 static FileType IdentifyType(FileUtil::IOFile& file, const std::string& filepath); 25 static FileType IdentifyType(const FileSys::VirtualFile& file);
27 26
28 FileType GetFileType() override { 27 FileType GetFileType() override {
29 return IdentifyType(file, filepath); 28 return IdentifyType(file);
30 } 29 }
31 30
32 static VAddr LoadModule(const std::string& name, const std::vector<u8>& file_data, 31 static VAddr LoadModule(FileSys::VirtualFile file, VAddr load_base);
33 VAddr load_base);
34
35 static VAddr LoadModule(const std::string& path, VAddr load_base);
36 32
37 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override; 33 ResultStatus Load(Kernel::SharedPtr<Kernel::Process>& process) override;
38
39private:
40 std::string filepath;
41}; 34};
42 35
43} // namespace Loader 36} // namespace Loader