mirror of
				https://git.zaroz.cloud/nintendo-back-up/yuzu/yuzu.git
				synced 2025-05-12 00:45:25 +00:00 
			
		
		
		
	loader: Add support for NRO, as well as various fixes and shared linker.
This commit is contained in:
		
							parent
							
								
									d454364bca
								
							
						
					
					
						commit
						33ea53094c
					
				| @ -174,8 +174,10 @@ set(SRCS | |||||||
|             hw/y2r.cpp |             hw/y2r.cpp | ||||||
|             loader/3dsx.cpp |             loader/3dsx.cpp | ||||||
|             loader/elf.cpp |             loader/elf.cpp | ||||||
|  |             loader/linker.cpp | ||||||
|             loader/loader.cpp |             loader/loader.cpp | ||||||
|             loader/ncch.cpp |             loader/ncch.cpp | ||||||
|  |             loader/nro.cpp | ||||||
|             loader/nso.cpp |             loader/nso.cpp | ||||||
|             loader/smdh.cpp |             loader/smdh.cpp | ||||||
|             tracer/recorder.cpp |             tracer/recorder.cpp | ||||||
| @ -374,8 +376,10 @@ set(HEADERS | |||||||
|             hw/y2r.h |             hw/y2r.h | ||||||
|             loader/3dsx.h |             loader/3dsx.h | ||||||
|             loader/elf.h |             loader/elf.h | ||||||
|  |             loader/linker.h | ||||||
|             loader/loader.h |             loader/loader.h | ||||||
|             loader/ncch.h |             loader/ncch.h | ||||||
|  |             loader/nro.h | ||||||
|             loader/nso.h |             loader/nso.h | ||||||
|             loader/smdh.h |             loader/smdh.h | ||||||
|             tracer/recorder.h |             tracer/recorder.h | ||||||
|  | |||||||
							
								
								
									
										151
									
								
								src/core/loader/linker.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										151
									
								
								src/core/loader/linker.cpp
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,151 @@ | |||||||
|  | // Copyright 2017 Citra Emulator Project
 | ||||||
|  | // Licensed under GPLv2 or any later version
 | ||||||
|  | // Refer to the license.txt file included.
 | ||||||
|  | 
 | ||||||
|  | #include <vector> | ||||||
|  | 
 | ||||||
|  | #include "common/common_funcs.h" | ||||||
|  | #include "common/logging/log.h" | ||||||
|  | #include "common/swap.h" | ||||||
|  | #include "core/loader/linker.h" | ||||||
|  | #include "core/memory.h" | ||||||
|  | 
 | ||||||
|  | namespace Loader { | ||||||
|  | 
 | ||||||
|  | enum class RelocationType : u32 { ABS64 = 257, GLOB_DAT = 1025, JUMP_SLOT = 1026, RELATIVE = 1027 }; | ||||||
|  | 
 | ||||||
|  | enum DynamicType : u32 { | ||||||
|  |     DT_NULL = 0, | ||||||
|  |     DT_PLTRELSZ = 2, | ||||||
|  |     DT_STRTAB = 5, | ||||||
|  |     DT_SYMTAB = 6, | ||||||
|  |     DT_RELA = 7, | ||||||
|  |     DT_RELASZ = 8, | ||||||
|  |     DT_STRSZ = 10, | ||||||
|  |     DT_JMPREL = 23, | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | struct Elf64_Rela { | ||||||
|  |     u64_le offset; | ||||||
|  |     RelocationType type; | ||||||
|  |     u32_le symbol; | ||||||
|  |     s64_le addend; | ||||||
|  | }; | ||||||
|  | static_assert(sizeof(Elf64_Rela) == 0x18, "Elf64_Rela has incorrect size."); | ||||||
|  | 
 | ||||||
|  | struct Elf64_Dyn { | ||||||
|  |     u64_le tag; | ||||||
|  |     u64_le value; | ||||||
|  | }; | ||||||
|  | static_assert(sizeof(Elf64_Dyn) == 0x10, "Elf64_Dyn has incorrect size."); | ||||||
|  | 
 | ||||||
|  | struct Elf64_Sym { | ||||||
|  |     u32_le name; | ||||||
|  |     INSERT_PADDING_BYTES(0x2); | ||||||
|  |     u16_le shndx; | ||||||
|  |     u64_le value; | ||||||
|  |     u64_le size; | ||||||
|  | }; | ||||||
|  | static_assert(sizeof(Elf64_Sym) == 0x18, "Elf64_Sym has incorrect size."); | ||||||
|  | 
 | ||||||
|  | void Linker::WriteRelocations(std::vector<u8>& program_image, | ||||||
|  |                                   const std::vector<Symbol>& symbols, u64 relocation_offset, | ||||||
|  |                                   u64 size, bool is_jump_relocation, VAddr load_base) { | ||||||
|  |     for (u64 i = 0; i < size; i += sizeof(Elf64_Rela)) { | ||||||
|  |         Elf64_Rela rela; | ||||||
|  |         std::memcpy(&rela, &program_image[relocation_offset + i], sizeof(Elf64_Rela)); | ||||||
|  | 
 | ||||||
|  |         const Symbol& symbol = symbols[rela.symbol]; | ||||||
|  |         switch (rela.type) { | ||||||
|  |         case RelocationType::RELATIVE: { | ||||||
|  |             const u64 value = load_base + rela.addend; | ||||||
|  |             if (!symbol.name.empty()) { | ||||||
|  |                 exports[symbol.name] = value; | ||||||
|  |             } | ||||||
|  |             std::memcpy(&program_image[rela.offset], &value, sizeof(u64)); | ||||||
|  |             break; | ||||||
|  |         } | ||||||
|  |         case RelocationType::JUMP_SLOT: | ||||||
|  |         case RelocationType::GLOB_DAT: | ||||||
|  |             if (!symbol.value) { | ||||||
|  |                 imports[symbol.name] = {rela.offset + load_base, 0}; | ||||||
|  |             } else { | ||||||
|  |                 exports[symbol.name] = symbol.value; | ||||||
|  |                 std::memcpy(&program_image[rela.offset], &symbol.value, sizeof(u64)); | ||||||
|  |             } | ||||||
|  |             break; | ||||||
|  |         case RelocationType::ABS64: | ||||||
|  |             if (!symbol.value) { | ||||||
|  |                 imports[symbol.name] = {rela.offset + load_base, rela.addend}; | ||||||
|  |             } else { | ||||||
|  |                 const u64 value = symbol.value + rela.addend; | ||||||
|  |                 exports[symbol.name] = value; | ||||||
|  |                 std::memcpy(&program_image[rela.offset], &value, sizeof(u64)); | ||||||
|  |             } | ||||||
|  |             break; | ||||||
|  |         default: | ||||||
|  |             LOG_CRITICAL(Loader, "Unknown relocation type: %d", rela.type); | ||||||
|  |             break; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void Linker::Relocate(std::vector<u8>& program_image, u32 dynamic_section_offset, | ||||||
|  |                           VAddr load_base) { | ||||||
|  |     std::map<u64, u64> dynamic; | ||||||
|  |     while (dynamic_section_offset < program_image.size()) { | ||||||
|  |         Elf64_Dyn dyn; | ||||||
|  |         std::memcpy(&dyn, &program_image[dynamic_section_offset], sizeof(Elf64_Dyn)); | ||||||
|  |         dynamic_section_offset += sizeof(Elf64_Dyn); | ||||||
|  | 
 | ||||||
|  |         if (dyn.tag == DT_NULL) { | ||||||
|  |             break; | ||||||
|  |         } | ||||||
|  |         dynamic[dyn.tag] = dyn.value; | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     u64 offset = dynamic[DT_SYMTAB]; | ||||||
|  |     std::vector<Symbol> symbols; | ||||||
|  |     while (offset < program_image.size()) { | ||||||
|  |         Elf64_Sym sym; | ||||||
|  |         std::memcpy(&sym, &program_image[offset], sizeof(Elf64_Sym)); | ||||||
|  |         offset += sizeof(Elf64_Sym); | ||||||
|  | 
 | ||||||
|  |         if (sym.name >= dynamic[DT_STRSZ]) { | ||||||
|  |             break; | ||||||
|  |         } | ||||||
|  | 
 | ||||||
|  |         std::string name = reinterpret_cast<char*>(&program_image[dynamic[DT_STRTAB] + sym.name]); | ||||||
|  |         if (sym.value) { | ||||||
|  |             exports[name] = load_base + sym.value; | ||||||
|  |             symbols.emplace_back(std::move(name), load_base + sym.value); | ||||||
|  |         } else { | ||||||
|  |             symbols.emplace_back(std::move(name), 0); | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     if (dynamic.find(DT_RELA) != dynamic.end()) { | ||||||
|  |         WriteRelocations(program_image, symbols, dynamic[DT_RELA], dynamic[DT_RELASZ], false, | ||||||
|  |                          load_base); | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     if (dynamic.find(DT_JMPREL) != dynamic.end()) { | ||||||
|  |         WriteRelocations(program_image, symbols, dynamic[DT_JMPREL], dynamic[DT_PLTRELSZ], true, | ||||||
|  |                          load_base); | ||||||
|  |     } | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void Linker::ResolveImports() { | ||||||
|  |     // Resolve imports
 | ||||||
|  |     for (const auto& import : imports) { | ||||||
|  |         const auto& search = exports.find(import.first); | ||||||
|  |         if (search != exports.end()) { | ||||||
|  |             Memory::Write64(import.second.ea, search->second + import.second.addend); | ||||||
|  |         } | ||||||
|  |         else { | ||||||
|  |             LOG_ERROR(Loader, "Unresolved import: %s", import.first.c_str()); | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | } // namespace Loader
 | ||||||
							
								
								
									
										37
									
								
								src/core/loader/linker.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										37
									
								
								src/core/loader/linker.h
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,37 @@ | |||||||
|  | // Copyright 2017 Citra Emulator Project
 | ||||||
|  | // Licensed under GPLv2 or any later version
 | ||||||
|  | // Refer to the license.txt file included.
 | ||||||
|  | 
 | ||||||
|  | #pragma once | ||||||
|  | 
 | ||||||
|  | #include <map> | ||||||
|  | #include <string> | ||||||
|  | #include "common/common_types.h" | ||||||
|  | 
 | ||||||
|  | namespace Loader { | ||||||
|  | 
 | ||||||
|  | class Linker { | ||||||
|  | protected: | ||||||
|  |     struct Symbol { | ||||||
|  |         Symbol(std::string&& name, u64 value) : name(std::move(name)), value(value) {} | ||||||
|  |         std::string name; | ||||||
|  |         u64 value; | ||||||
|  |     }; | ||||||
|  | 
 | ||||||
|  |     struct Import { | ||||||
|  |         VAddr ea; | ||||||
|  |         s64 addend; | ||||||
|  |     }; | ||||||
|  | 
 | ||||||
|  |     void WriteRelocations(std::vector<u8>& program_image, const std::vector<Symbol>& symbols, | ||||||
|  |                           u64 relocation_offset, u64 size, bool is_jump_relocation, | ||||||
|  |                           VAddr load_base); | ||||||
|  |     void Relocate(std::vector<u8>& program_image, u32 dynamic_section_offset, VAddr load_base); | ||||||
|  | 
 | ||||||
|  |     void ResolveImports(); | ||||||
|  | 
 | ||||||
|  |     std::map<std::string, Import> imports; | ||||||
|  |     std::map<std::string, VAddr> exports; | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | } // namespace Loader
 | ||||||
| @ -10,6 +10,7 @@ | |||||||
| #include "core/loader/3dsx.h" | #include "core/loader/3dsx.h" | ||||||
| #include "core/loader/elf.h" | #include "core/loader/elf.h" | ||||||
| #include "core/loader/ncch.h" | #include "core/loader/ncch.h" | ||||||
|  | #include "core/loader/nro.h" | ||||||
| #include "core/loader/nso.h" | #include "core/loader/nso.h" | ||||||
| 
 | 
 | ||||||
| ////////////////////////////////////////////////////////////////////////////////////////////////////
 | ////////////////////////////////////////////////////////////////////////////////////////////////////
 | ||||||
| @ -34,6 +35,7 @@ FileType IdentifyFile(FileUtil::IOFile& file) { | |||||||
|     CHECK_TYPE(ELF) |     CHECK_TYPE(ELF) | ||||||
|     CHECK_TYPE(NCCH) |     CHECK_TYPE(NCCH) | ||||||
|     CHECK_TYPE(NSO) |     CHECK_TYPE(NSO) | ||||||
|  |     CHECK_TYPE(NRO) | ||||||
| 
 | 
 | ||||||
| #undef CHECK_TYPE | #undef CHECK_TYPE | ||||||
| 
 | 
 | ||||||
| @ -121,6 +123,10 @@ static std::unique_ptr<AppLoader> GetFileLoader(FileUtil::IOFile&& file, FileTyp | |||||||
|     case FileType::NSO: |     case FileType::NSO: | ||||||
|         return std::make_unique<AppLoader_NSO>(std::move(file), filename, filepath); |         return std::make_unique<AppLoader_NSO>(std::move(file), filename, filepath); | ||||||
| 
 | 
 | ||||||
|  |     // NX NRO file format.
 | ||||||
|  |     case FileType::NRO: | ||||||
|  |         return std::make_unique<AppLoader_NRO>(std::move(file), filename, filepath); | ||||||
|  | 
 | ||||||
|     default: |     default: | ||||||
|         return nullptr; |         return nullptr; | ||||||
|     } |     } | ||||||
|  | |||||||
| @ -33,6 +33,7 @@ enum class FileType { | |||||||
|     ELF, |     ELF, | ||||||
|     THREEDSX, // 3DSX
 |     THREEDSX, // 3DSX
 | ||||||
|     NSO, |     NSO, | ||||||
|  |     NRO, | ||||||
| }; | }; | ||||||
| 
 | 
 | ||||||
| /**
 | /**
 | ||||||
|  | |||||||
							
								
								
									
										173
									
								
								src/core/loader/nro.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										173
									
								
								src/core/loader/nro.cpp
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,173 @@ | |||||||
|  | // Copyright 2017 Citra Emulator Project
 | ||||||
|  | // Licensed under GPLv2 or any later version
 | ||||||
|  | // Refer to the license.txt file included.
 | ||||||
|  | 
 | ||||||
|  | #include <vector> | ||||||
|  | 
 | ||||||
|  | #include "common/logging/log.h" | ||||||
|  | #include "common/swap.h" | ||||||
|  | #include "core/hle/kernel/process.h" | ||||||
|  | #include "core/hle/kernel/resource_limit.h" | ||||||
|  | #include "core/loader/nro.h" | ||||||
|  | #include "core/memory.h" | ||||||
|  | 
 | ||||||
|  | namespace Loader { | ||||||
|  | 
 | ||||||
|  | struct NroSegmentHeader { | ||||||
|  |     u32_le offset; | ||||||
|  |     u32_le size; | ||||||
|  | }; | ||||||
|  | static_assert(sizeof(NroSegmentHeader) == 0x8, "NroSegmentHeader has incorrect size."); | ||||||
|  | 
 | ||||||
|  | struct NroHeader { | ||||||
|  |     INSERT_PADDING_BYTES(0x4); | ||||||
|  |     u32_le module_header_offset; | ||||||
|  |     INSERT_PADDING_BYTES(0x8); | ||||||
|  |     u32_le magic; | ||||||
|  |     INSERT_PADDING_BYTES(0x4); | ||||||
|  |     u32_le file_size; | ||||||
|  |     INSERT_PADDING_BYTES(0x4); | ||||||
|  |     std::array<NroSegmentHeader, 3> segments; // Text, RoData, Data (in that order)
 | ||||||
|  |     u32_le bss_size; | ||||||
|  |     INSERT_PADDING_BYTES(0x44); | ||||||
|  | }; | ||||||
|  | static_assert(sizeof(NroHeader) == 0x80, "NroHeader has incorrect size."); | ||||||
|  | 
 | ||||||
|  | struct ModHeader { | ||||||
|  |     u32_le magic; | ||||||
|  |     u32_le dynamic_offset; | ||||||
|  |     u32_le bss_start_offset; | ||||||
|  |     u32_le bss_end_offset; | ||||||
|  |     u32_le unwind_start_offset; | ||||||
|  |     u32_le unwind_end_offset; | ||||||
|  |     u32_le module_offset; // Offset to runtime-generated module object. typically equal to .bss base
 | ||||||
|  | }; | ||||||
|  | static_assert(sizeof(ModHeader) == 0x1c, "ModHeader has incorrect size."); | ||||||
|  | 
 | ||||||
|  | FileType AppLoader_NRO::IdentifyType(FileUtil::IOFile& file) { | ||||||
|  |     // Read NSO header
 | ||||||
|  |     NroHeader nro_header{}; | ||||||
|  |     file.Seek(0, SEEK_SET); | ||||||
|  |     if (sizeof(NroHeader) != file.ReadBytes(&nro_header, sizeof(NroHeader))) { | ||||||
|  |         return FileType::Error; | ||||||
|  |     } | ||||||
|  |     if (nro_header.magic == MakeMagic('N', 'R', 'O', '0')) { | ||||||
|  |         return FileType::NRO; | ||||||
|  |     } | ||||||
|  |     return FileType::Error; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static constexpr u32 PageAlignSize(u32 size) { | ||||||
|  |     return (size + Memory::PAGE_MASK) & ~Memory::PAGE_MASK; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static std::vector<u8> ReadSegment(FileUtil::IOFile& file, const NroSegmentHeader& header) { | ||||||
|  |     std::vector<u8> data; | ||||||
|  |     data.resize(header.size); | ||||||
|  | 
 | ||||||
|  |     file.Seek(header.offset + sizeof(NroHeader), SEEK_SET); | ||||||
|  |     size_t bytes_read{file.ReadBytes(data.data(), header.size)}; | ||||||
|  |     if (header.size != PageAlignSize(static_cast<u32>(bytes_read))) { | ||||||
|  |         LOG_CRITICAL(Loader, "Failed to read NRO segment bytes", header.size); | ||||||
|  |         return {}; | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     return data; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | VAddr AppLoader_NRO::GetEntryPoint(VAddr load_base) const { | ||||||
|  |     // Find nnMain function, set entrypoint to that address
 | ||||||
|  |     const auto& search = exports.find("nnMain"); | ||||||
|  |     if (search != exports.end()) { | ||||||
|  |         return load_base + search->second; | ||||||
|  |     } | ||||||
|  |     const VAddr entry_point{load_base + sizeof(NroHeader)}; | ||||||
|  |     LOG_ERROR(Loader, "Unable to find entrypoint, defaulting to: 0x%llx", entry_point); | ||||||
|  |     return entry_point; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | bool AppLoader_NRO::LoadNro(const std::string& path, VAddr load_base) { | ||||||
|  |     FileUtil::IOFile file(path, "rb"); | ||||||
|  |     if (!file.IsOpen()) { | ||||||
|  |         return {}; | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     // Read NSO header
 | ||||||
|  |     NroHeader nro_header{}; | ||||||
|  |     file.Seek(0, SEEK_SET); | ||||||
|  |     if (sizeof(NroHeader) != file.ReadBytes(&nro_header, sizeof(NroHeader))) { | ||||||
|  |         return {}; | ||||||
|  |     } | ||||||
|  |     if (nro_header.magic != MakeMagic('N', 'R', 'O', '0')) { | ||||||
|  |         return {}; | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     // Build program image
 | ||||||
|  |     Kernel::SharedPtr<Kernel::CodeSet> codeset = Kernel::CodeSet::Create("", 0); | ||||||
|  |     std::vector<u8> program_image; | ||||||
|  |     program_image.resize(PageAlignSize(nro_header.file_size + nro_header.bss_size)); | ||||||
|  |     file.Seek(0, SEEK_SET); | ||||||
|  |     file.ReadBytes(program_image.data(), nro_header.file_size); | ||||||
|  | 
 | ||||||
|  |     for (int i = 0; i < nro_header.segments.size(); ++i) { | ||||||
|  |         codeset->segments[i].addr = nro_header.segments[i].offset; | ||||||
|  |         codeset->segments[i].offset = nro_header.segments[i].offset; | ||||||
|  |         codeset->segments[i].size = PageAlignSize(nro_header.segments[i].size); | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     // Read MOD header
 | ||||||
|  |     ModHeader mod_header{}; | ||||||
|  |     u32 bss_size{Memory::PAGE_SIZE}; // Default .bss to page size if MOD0 section doesn't exist
 | ||||||
|  |     std::memcpy(&mod_header, program_image.data() + nro_header.module_header_offset, | ||||||
|  |                 sizeof(ModHeader)); | ||||||
|  |     const bool has_mod_header{mod_header.magic == MakeMagic('M', 'O', 'D', '0')}; | ||||||
|  |     if (has_mod_header) { | ||||||
|  |         // Resize program image to include .bss section and page align each section
 | ||||||
|  |         bss_size = PageAlignSize(mod_header.bss_end_offset - mod_header.bss_start_offset); | ||||||
|  |         codeset->data.size += bss_size; | ||||||
|  |     } | ||||||
|  |     program_image.resize(PageAlignSize(static_cast<u32>(program_image.size()) + bss_size)); | ||||||
|  | 
 | ||||||
|  |     // Relocate symbols if there was a proper MOD header - This must happen after the image has been
 | ||||||
|  |     // loaded into memory
 | ||||||
|  |     if (has_mod_header) { | ||||||
|  |         Relocate(program_image, nro_header.module_header_offset + mod_header.dynamic_offset, | ||||||
|  |                  load_base); | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     // Load codeset for current process
 | ||||||
|  |     codeset->name = path; | ||||||
|  |     codeset->memory = std::make_shared<std::vector<u8>>(std::move(program_image)); | ||||||
|  |     Kernel::g_current_process->LoadModule(codeset, load_base); | ||||||
|  | 
 | ||||||
|  |     return true; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | ResultStatus AppLoader_NRO::Load() { | ||||||
|  |     if (is_loaded) { | ||||||
|  |         return ResultStatus::ErrorAlreadyLoaded; | ||||||
|  |     } | ||||||
|  |     if (!file.IsOpen()) { | ||||||
|  |         return ResultStatus::Error; | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     // Load and relocate "main" and "sdk" NSO
 | ||||||
|  |     static constexpr VAddr main_base{0x10000000}; | ||||||
|  |     Kernel::g_current_process = Kernel::Process::Create("main"); | ||||||
|  |     if (!LoadNro(filepath, main_base)) { | ||||||
|  |         return ResultStatus::ErrorInvalidFormat; | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     Kernel::g_current_process->svc_access_mask.set(); | ||||||
|  |     Kernel::g_current_process->address_mappings = default_address_mappings; | ||||||
|  |     Kernel::g_current_process->resource_limit = | ||||||
|  |         Kernel::ResourceLimit::GetForCategory(Kernel::ResourceLimitCategory::APPLICATION); | ||||||
|  |     Kernel::g_current_process->Run(GetEntryPoint(main_base), 48, Kernel::DEFAULT_STACK_SIZE); | ||||||
|  | 
 | ||||||
|  |     ResolveImports(); | ||||||
|  | 
 | ||||||
|  |     is_loaded = true; | ||||||
|  |     return ResultStatus::Success; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | } // namespace Loader
 | ||||||
							
								
								
									
										45
									
								
								src/core/loader/nro.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										45
									
								
								src/core/loader/nro.h
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,45 @@ | |||||||
|  | // Copyright 2017 Citra Emulator Project
 | ||||||
|  | // Licensed under GPLv2 or any later version
 | ||||||
|  | // Refer to the license.txt file included.
 | ||||||
|  | 
 | ||||||
|  | #pragma once | ||||||
|  | 
 | ||||||
|  | #include <map> | ||||||
|  | #include <string> | ||||||
|  | #include "common/common_types.h" | ||||||
|  | #include "common/file_util.h" | ||||||
|  | #include "core/hle/kernel/kernel.h" | ||||||
|  | #include "core/loader/linker.h" | ||||||
|  | #include "core/loader/loader.h" | ||||||
|  | 
 | ||||||
|  | namespace Loader { | ||||||
|  | 
 | ||||||
|  | /// Loads an NRO file
 | ||||||
|  | class AppLoader_NRO final : public AppLoader, Linker { | ||||||
|  | public: | ||||||
|  |     AppLoader_NRO(FileUtil::IOFile&& file, std::string filename, std::string filepath) | ||||||
|  |         : AppLoader(std::move(file)), filename(std::move(filename)), filepath(std::move(filepath)) { | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     /**
 | ||||||
|  |      * Returns the type of the file | ||||||
|  |      * @param file FileUtil::IOFile open file | ||||||
|  |      * @return FileType found, or FileType::Error if this loader doesn't know it | ||||||
|  |      */ | ||||||
|  |     static FileType IdentifyType(FileUtil::IOFile& file); | ||||||
|  | 
 | ||||||
|  |     FileType GetFileType() override { | ||||||
|  |         return IdentifyType(file); | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     ResultStatus Load() override; | ||||||
|  | 
 | ||||||
|  | private: | ||||||
|  |     VAddr GetEntryPoint(VAddr load_base) const; | ||||||
|  |     bool LoadNro(const std::string& path, VAddr load_base); | ||||||
|  | 
 | ||||||
|  |     std::string filename; | ||||||
|  |     std::string filepath; | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | } // namespace Loader
 | ||||||
| @ -14,19 +14,6 @@ | |||||||
| 
 | 
 | ||||||
| namespace Loader { | namespace Loader { | ||||||
| 
 | 
 | ||||||
| enum class RelocationType : u32 { ABS64 = 257, GLOB_DAT = 1025, JUMP_SLOT = 1026, RELATIVE = 1027 }; |  | ||||||
| 
 |  | ||||||
| enum DynamicType : u32 { |  | ||||||
|     DT_NULL = 0, |  | ||||||
|     DT_PLTRELSZ = 2, |  | ||||||
|     DT_STRTAB = 5, |  | ||||||
|     DT_SYMTAB = 6, |  | ||||||
|     DT_RELA = 7, |  | ||||||
|     DT_RELASZ = 8, |  | ||||||
|     DT_STRSZ = 10, |  | ||||||
|     DT_JMPREL = 23, |  | ||||||
| }; |  | ||||||
| 
 |  | ||||||
| struct NsoSegmentHeader { | struct NsoSegmentHeader { | ||||||
|     u32_le offset; |     u32_le offset; | ||||||
|     u32_le location; |     u32_le location; | ||||||
| @ -46,8 +33,6 @@ struct NsoHeader { | |||||||
| static_assert(sizeof(NsoHeader) == 0x6c, "NsoHeader has incorrect size."); | static_assert(sizeof(NsoHeader) == 0x6c, "NsoHeader has incorrect size."); | ||||||
| 
 | 
 | ||||||
| struct ModHeader { | struct ModHeader { | ||||||
|     INSERT_PADDING_BYTES(0x4); |  | ||||||
|     u32_le offset_to_start; // Always 8
 |  | ||||||
|     u32_le magic; |     u32_le magic; | ||||||
|     u32_le dynamic_offset; |     u32_le dynamic_offset; | ||||||
|     u32_le bss_start_offset; |     u32_le bss_start_offset; | ||||||
| @ -56,7 +41,7 @@ struct ModHeader { | |||||||
|     u32_le eh_frame_hdr_end_offset; |     u32_le eh_frame_hdr_end_offset; | ||||||
|     u32_le module_offset; // Offset to runtime-generated module object. typically equal to .bss base
 |     u32_le module_offset; // Offset to runtime-generated module object. typically equal to .bss base
 | ||||||
| }; | }; | ||||||
| static_assert(sizeof(ModHeader) == 0x24, "ModHeader has incorrect size."); | static_assert(sizeof(ModHeader) == 0x1c, "ModHeader has incorrect size."); | ||||||
| 
 | 
 | ||||||
| FileType AppLoader_NSO::IdentifyType(FileUtil::IOFile& file) { | FileType AppLoader_NSO::IdentifyType(FileUtil::IOFile& file) { | ||||||
|     u32 magic = 0; |     u32 magic = 0; | ||||||
| @ -95,101 +80,6 @@ static std::vector<u8> ReadSegment(FileUtil::IOFile& file, const NsoSegmentHeade | |||||||
|     return uncompressed_data; |     return uncompressed_data; | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| void AppLoader_NSO::WriteRelocations(const std::vector<Symbol>& symbols, VAddr load_base, |  | ||||||
|                                      u64 relocation_offset, u64 size, bool is_jump_relocation) { |  | ||||||
|     for (u64 i = 0; i < size; i += 0x18) { |  | ||||||
|         VAddr addr = load_base + relocation_offset + i; |  | ||||||
|         u64 offset = Memory::Read64(addr); |  | ||||||
|         u64 info = Memory::Read64(addr + 8); |  | ||||||
|         u64 addend_unsigned = Memory::Read64(addr + 16); |  | ||||||
|         s64 addend{}; |  | ||||||
|         std::memcpy(&addend, &addend_unsigned, sizeof(u64)); |  | ||||||
| 
 |  | ||||||
|         RelocationType rtype = static_cast<RelocationType>(info & 0xFFFFFFFF); |  | ||||||
|         u32 rsym = static_cast<u32>(info >> 32); |  | ||||||
|         VAddr ea = load_base + offset; |  | ||||||
| 
 |  | ||||||
|         const Symbol& symbol = symbols[rsym]; |  | ||||||
| 
 |  | ||||||
|         switch (rtype) { |  | ||||||
|         case RelocationType::RELATIVE: |  | ||||||
|             if (!symbol.name.empty()) { |  | ||||||
|                 exports[symbol.name] = load_base + addend; |  | ||||||
|             } |  | ||||||
|             Memory::Write64(ea, load_base + addend); |  | ||||||
|             break; |  | ||||||
|         case RelocationType::JUMP_SLOT: |  | ||||||
|         case RelocationType::GLOB_DAT: |  | ||||||
|             if (!symbol.value) { |  | ||||||
|                 imports[symbol.name] = {ea, 0}; |  | ||||||
|             } else { |  | ||||||
|                 exports[symbol.name] = symbol.value; |  | ||||||
|                 Memory::Write64(ea, symbol.value); |  | ||||||
|             } |  | ||||||
|             break; |  | ||||||
|         case RelocationType::ABS64: |  | ||||||
|             if (!symbol.value) { |  | ||||||
|                 imports[symbol.name] = {ea, addend}; |  | ||||||
|             } else { |  | ||||||
|                 exports[symbol.name] = symbol.value + addend; |  | ||||||
|                 Memory::Write64(ea, symbol.value + addend); |  | ||||||
|             } |  | ||||||
|             break; |  | ||||||
|         default: |  | ||||||
|             LOG_CRITICAL(Loader, "Unknown relocation type: %d", rtype); |  | ||||||
|             break; |  | ||||||
|         } |  | ||||||
|     } |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| void AppLoader_NSO::Relocate(VAddr load_base, VAddr dynamic_section_addr) { |  | ||||||
|     std::map<u64, u64> dynamic; |  | ||||||
|     while (1) { |  | ||||||
|         u64 tag = Memory::Read64(dynamic_section_addr); |  | ||||||
|         u64 value = Memory::Read64(dynamic_section_addr + 8); |  | ||||||
|         dynamic_section_addr += 16; |  | ||||||
| 
 |  | ||||||
|         if (tag == DT_NULL) { |  | ||||||
|             break; |  | ||||||
|         } |  | ||||||
|         dynamic[tag] = value; |  | ||||||
|     } |  | ||||||
| 
 |  | ||||||
|     u64 strtabsize = dynamic[DT_STRSZ]; |  | ||||||
|     std::vector<u8> strtab; |  | ||||||
|     strtab.resize(strtabsize); |  | ||||||
|     Memory::ReadBlock(load_base + dynamic[DT_STRTAB], strtab.data(), strtabsize); |  | ||||||
| 
 |  | ||||||
|     VAddr addr = load_base + dynamic[DT_SYMTAB]; |  | ||||||
|     std::vector<Symbol> symbols; |  | ||||||
|     while (1) { |  | ||||||
|         const u32 stname = Memory::Read32(addr); |  | ||||||
|         const u16 stshndx = Memory::Read16(addr + 6); |  | ||||||
|         const u64 stvalue = Memory::Read64(addr + 8); |  | ||||||
|         addr += 24; |  | ||||||
| 
 |  | ||||||
|         if (stname >= strtabsize) { |  | ||||||
|             break; |  | ||||||
|         } |  | ||||||
| 
 |  | ||||||
|         std::string name = reinterpret_cast<char*>(&strtab[stname]); |  | ||||||
|         if (stvalue) { |  | ||||||
|             exports[name] = load_base + stvalue; |  | ||||||
|             symbols.emplace_back(std::move(name), load_base + stvalue); |  | ||||||
|         } else { |  | ||||||
|             symbols.emplace_back(std::move(name), 0); |  | ||||||
|         } |  | ||||||
|     } |  | ||||||
| 
 |  | ||||||
|     if (dynamic.find(DT_RELA) != dynamic.end()) { |  | ||||||
|         WriteRelocations(symbols, load_base, dynamic[DT_RELA], dynamic[DT_RELASZ], false); |  | ||||||
|     } |  | ||||||
| 
 |  | ||||||
|     if (dynamic.find(DT_JMPREL) != dynamic.end()) { |  | ||||||
|         WriteRelocations(symbols, load_base, dynamic[DT_JMPREL], dynamic[DT_PLTRELSZ], true); |  | ||||||
|     } |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| VAddr AppLoader_NSO::GetEntryPoint(VAddr load_base) const { | VAddr AppLoader_NSO::GetEntryPoint(VAddr load_base) const { | ||||||
|     // Find nnMain function, set entrypoint to that address
 |     // Find nnMain function, set entrypoint to that address
 | ||||||
|     const auto& search = exports.find("nnMain"); |     const auto& search = exports.find("nnMain"); | ||||||
| @ -233,10 +123,14 @@ bool AppLoader_NSO::LoadNso(const std::string& path, VAddr load_base) { | |||||||
|         codeset->segments[i].size = PageAlignSize(static_cast<u32>(data.size())); |         codeset->segments[i].size = PageAlignSize(static_cast<u32>(data.size())); | ||||||
|     } |     } | ||||||
| 
 | 
 | ||||||
|  |     // MOD header pointer is at .text offset + 4
 | ||||||
|  |     u32 module_offset; | ||||||
|  |     std::memcpy(&module_offset, program_image.data() + 4, sizeof(u32)); | ||||||
|  | 
 | ||||||
|     // Read MOD header
 |     // Read MOD header
 | ||||||
|     ModHeader mod_header{}; |     ModHeader mod_header{}; | ||||||
|     u32 bss_size{Memory::PAGE_SIZE}; // Default .bss to page size if MOD0 section doesn't exist
 |     u32 bss_size{Memory::PAGE_SIZE}; // Default .bss to page size if MOD0 section doesn't exist
 | ||||||
|     std::memcpy(&mod_header, program_image.data(), sizeof(ModHeader)); |     std::memcpy(&mod_header, program_image.data() + module_offset, sizeof(ModHeader)); | ||||||
|     const bool has_mod_header{mod_header.magic == MakeMagic('M', 'O', 'D', '0')}; |     const bool has_mod_header{mod_header.magic == MakeMagic('M', 'O', 'D', '0')}; | ||||||
|     if (has_mod_header) { |     if (has_mod_header) { | ||||||
|         // Resize program image to include .bss section and page align each section
 |         // Resize program image to include .bss section and page align each section
 | ||||||
| @ -245,16 +139,17 @@ bool AppLoader_NSO::LoadNso(const std::string& path, VAddr load_base) { | |||||||
|     } |     } | ||||||
|     program_image.resize(PageAlignSize(static_cast<u32>(program_image.size()) + bss_size)); |     program_image.resize(PageAlignSize(static_cast<u32>(program_image.size()) + bss_size)); | ||||||
| 
 | 
 | ||||||
|  |     // Relocate symbols if there was a proper MOD header - This must happen after the image has been
 | ||||||
|  |     // loaded into memory
 | ||||||
|  |     if (has_mod_header) { | ||||||
|  |         Relocate(program_image, module_offset + mod_header.dynamic_offset, load_base); | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|     // Load codeset for current process
 |     // Load codeset for current process
 | ||||||
|     codeset->name = path; |     codeset->name = path; | ||||||
|     codeset->memory = std::make_shared<std::vector<u8>>(std::move(program_image)); |     codeset->memory = std::make_shared<std::vector<u8>>(std::move(program_image)); | ||||||
|     Kernel::g_current_process->LoadModule(codeset, load_base); |     Kernel::g_current_process->LoadModule(codeset, load_base); | ||||||
| 
 | 
 | ||||||
|     // Relocate symbols if there was a proper MOD header - This must happen after the image has been
 |  | ||||||
|     // loaded into memory
 |  | ||||||
|     if (has_mod_header) { |  | ||||||
|         Relocate(load_base, load_base + mod_header.offset_to_start + mod_header.dynamic_offset); |  | ||||||
|     } |  | ||||||
|     return true; |     return true; | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| @ -267,13 +162,13 @@ ResultStatus AppLoader_NSO::Load() { | |||||||
|     } |     } | ||||||
| 
 | 
 | ||||||
|     // Load and relocate "main" and "sdk" NSO
 |     // Load and relocate "main" and "sdk" NSO
 | ||||||
|     static constexpr VAddr main_base{0x10000000}; |     static constexpr VAddr main_base{0x710000000}; | ||||||
|     Kernel::g_current_process = Kernel::Process::Create("main"); |     Kernel::g_current_process = Kernel::Process::Create("main"); | ||||||
|     if (!LoadNso(filepath, main_base)) { |     if (!LoadNso(filepath, main_base)) { | ||||||
|         return ResultStatus::ErrorInvalidFormat; |         return ResultStatus::ErrorInvalidFormat; | ||||||
|     } |     } | ||||||
|     const std::string sdkpath = filepath.substr(0, filepath.find_last_of("/\\")) + "/sdk"; |     const std::string sdkpath = filepath.substr(0, filepath.find_last_of("/\\")) + "/sdk"; | ||||||
|     if (!LoadNso(sdkpath, 0x20000000)) { |     if (!LoadNso(sdkpath, 0x720000000)) { | ||||||
|         LOG_WARNING(Loader, "failed to find SDK NSO"); |         LOG_WARNING(Loader, "failed to find SDK NSO"); | ||||||
|     } |     } | ||||||
| 
 | 
 | ||||||
| @ -283,15 +178,7 @@ ResultStatus AppLoader_NSO::Load() { | |||||||
|         Kernel::ResourceLimit::GetForCategory(Kernel::ResourceLimitCategory::APPLICATION); |         Kernel::ResourceLimit::GetForCategory(Kernel::ResourceLimitCategory::APPLICATION); | ||||||
|     Kernel::g_current_process->Run(GetEntryPoint(main_base), 48, Kernel::DEFAULT_STACK_SIZE); |     Kernel::g_current_process->Run(GetEntryPoint(main_base), 48, Kernel::DEFAULT_STACK_SIZE); | ||||||
| 
 | 
 | ||||||
|     // Resolve imports
 |     ResolveImports(); | ||||||
|     for (const auto& import : imports) { |  | ||||||
|         const auto& search = exports.find(import.first); |  | ||||||
|         if (search != exports.end()) { |  | ||||||
|             Memory::Write64(import.second.ea, search->second + import.second.addend); |  | ||||||
|         } else { |  | ||||||
|             LOG_ERROR(Loader, "Unresolved import: %s", import.first.c_str()); |  | ||||||
|         } |  | ||||||
|     } |  | ||||||
| 
 | 
 | ||||||
|     is_loaded = true; |     is_loaded = true; | ||||||
|     return ResultStatus::Success; |     return ResultStatus::Success; | ||||||
|  | |||||||
| @ -9,12 +9,13 @@ | |||||||
| #include "common/common_types.h" | #include "common/common_types.h" | ||||||
| #include "common/file_util.h" | #include "common/file_util.h" | ||||||
| #include "core/hle/kernel/kernel.h" | #include "core/hle/kernel/kernel.h" | ||||||
|  | #include "core/loader/linker.h" | ||||||
| #include "core/loader/loader.h" | #include "core/loader/loader.h" | ||||||
| 
 | 
 | ||||||
| namespace Loader { | namespace Loader { | ||||||
| 
 | 
 | ||||||
| /// Loads an NSO file
 | /// Loads an NSO file
 | ||||||
| class AppLoader_NSO final : public AppLoader { | class AppLoader_NSO final : public AppLoader, Linker { | ||||||
| public: | public: | ||||||
|     AppLoader_NSO(FileUtil::IOFile&& file, std::string filename, std::string filepath) |     AppLoader_NSO(FileUtil::IOFile&& file, std::string filename, std::string filepath) | ||||||
|         : AppLoader(std::move(file)), filename(std::move(filename)), filepath(std::move(filepath)) { |         : AppLoader(std::move(file)), filename(std::move(filename)), filepath(std::move(filepath)) { | ||||||
| @ -34,25 +35,8 @@ public: | |||||||
|     ResultStatus Load() override; |     ResultStatus Load() override; | ||||||
| 
 | 
 | ||||||
| private: | private: | ||||||
|     struct Symbol { |  | ||||||
|         Symbol(std::string&& name, u64 value) : name(std::move(name)), value(value) {} |  | ||||||
|         std::string name; |  | ||||||
|         u64 value; |  | ||||||
|     }; |  | ||||||
| 
 |  | ||||||
|     struct Import { |  | ||||||
|         VAddr ea; |  | ||||||
|         s64 addend; |  | ||||||
|     }; |  | ||||||
| 
 |  | ||||||
|     void WriteRelocations(const std::vector<Symbol>& symbols, VAddr load_base, |  | ||||||
|                           u64 relocation_offset, u64 size, bool is_jump_relocation); |  | ||||||
|     VAddr GetEntryPoint(VAddr load_base) const; |     VAddr GetEntryPoint(VAddr load_base) const; | ||||||
|     bool LoadNso(const std::string& path, VAddr load_base); |     bool LoadNso(const std::string& path, VAddr load_base); | ||||||
|     void Relocate(VAddr load_base, VAddr dynamic_section_addr); |  | ||||||
| 
 |  | ||||||
|     std::map<std::string, Import> imports; |  | ||||||
|     std::map<std::string, VAddr> exports; |  | ||||||
| 
 | 
 | ||||||
|     std::string filename; |     std::string filename; | ||||||
|     std::string filepath; |     std::string filepath; | ||||||
|  | |||||||
		Loading…
	
		Reference in New Issue
	
	Block a user
	 bunnei
						bunnei