ldr.cpp 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692
  1. // SPDX-FileCopyrightText: Copyright 2018 yuzu Emulator Project
  2. // SPDX-License-Identifier: GPL-2.0-or-later
  3. #include <memory>
  4. #include <fmt/format.h>
  5. #include <mbedtls/sha256.h>
  6. #include "common/alignment.h"
  7. #include "common/hex_util.h"
  8. #include "common/scope_exit.h"
  9. #include "core/core.h"
  10. #include "core/hle/ipc_helpers.h"
  11. #include "core/hle/kernel/k_page_table.h"
  12. #include "core/hle/kernel/svc_results.h"
  13. #include "core/hle/kernel/svc_types.h"
  14. #include "core/hle/service/ldr/ldr.h"
  15. #include "core/hle/service/service.h"
  16. #include "core/loader/nro.h"
  17. #include "core/memory.h"
  18. namespace Service::LDR {
  19. constexpr Result ERROR_INSUFFICIENT_ADDRESS_SPACE{ErrorModule::RO, 2};
  20. [[maybe_unused]] constexpr Result ERROR_INVALID_MEMORY_STATE{ErrorModule::Loader, 51};
  21. constexpr Result ERROR_INVALID_NRO{ErrorModule::Loader, 52};
  22. constexpr Result ERROR_INVALID_NRR{ErrorModule::Loader, 53};
  23. constexpr Result ERROR_MISSING_NRR_HASH{ErrorModule::Loader, 54};
  24. constexpr Result ERROR_MAXIMUM_NRO{ErrorModule::Loader, 55};
  25. constexpr Result ERROR_MAXIMUM_NRR{ErrorModule::Loader, 56};
  26. constexpr Result ERROR_ALREADY_LOADED{ErrorModule::Loader, 57};
  27. constexpr Result ERROR_INVALID_ALIGNMENT{ErrorModule::Loader, 81};
  28. constexpr Result ERROR_INVALID_SIZE{ErrorModule::Loader, 82};
  29. constexpr Result ERROR_INVALID_NRO_ADDRESS{ErrorModule::Loader, 84};
  30. [[maybe_unused]] constexpr Result ERROR_INVALID_NRR_ADDRESS{ErrorModule::Loader, 85};
  31. constexpr Result ERROR_NOT_INITIALIZED{ErrorModule::Loader, 87};
  32. constexpr std::size_t MAXIMUM_LOADED_RO{0x40};
  33. constexpr std::size_t MAXIMUM_MAP_RETRIES{0x200};
  34. constexpr std::size_t TEXT_INDEX{0};
  35. constexpr std::size_t RO_INDEX{1};
  36. constexpr std::size_t DATA_INDEX{2};
  37. struct NRRCertification {
  38. u64_le application_id_mask;
  39. u64_le application_id_pattern;
  40. INSERT_PADDING_BYTES(0x10);
  41. std::array<u8, 0x100> public_key; // Also known as modulus
  42. std::array<u8, 0x100> signature;
  43. };
  44. static_assert(sizeof(NRRCertification) == 0x220, "NRRCertification has invalid size.");
  45. struct NRRHeader {
  46. u32_le magic;
  47. u32_le certification_signature_key_generation; // 9.0.0+
  48. INSERT_PADDING_WORDS(2);
  49. NRRCertification certification;
  50. std::array<u8, 0x100> signature;
  51. u64_le application_id;
  52. u32_le size;
  53. u8 nrr_kind; // 7.0.0+
  54. INSERT_PADDING_BYTES(3);
  55. u32_le hash_offset;
  56. u32_le hash_count;
  57. INSERT_PADDING_WORDS(2);
  58. };
  59. static_assert(sizeof(NRRHeader) == 0x350, "NRRHeader has invalid size.");
  60. struct SegmentHeader {
  61. u32_le memory_offset;
  62. u32_le memory_size;
  63. };
  64. static_assert(sizeof(SegmentHeader) == 0x8, "SegmentHeader has invalid size.");
  65. struct NROHeader {
  66. // Switchbrew calls this "Start" (0x10)
  67. INSERT_PADDING_WORDS(1);
  68. u32_le mod_offset;
  69. INSERT_PADDING_WORDS(2);
  70. // Switchbrew calls this "Header" (0x70)
  71. u32_le magic;
  72. u32_le version;
  73. u32_le nro_size;
  74. u32_le flags;
  75. // .text, .ro, .data
  76. std::array<SegmentHeader, 3> segment_headers;
  77. u32_le bss_size;
  78. INSERT_PADDING_WORDS(1);
  79. std::array<u8, 0x20> build_id;
  80. u32_le dso_handle_offset;
  81. INSERT_PADDING_WORDS(1);
  82. // .apiInfo, .dynstr, .dynsym
  83. std::array<SegmentHeader, 3> segment_headers_2;
  84. };
  85. static_assert(sizeof(NROHeader) == 0x80, "NROHeader has invalid size.");
  86. using SHA256Hash = std::array<u8, 0x20>;
  87. struct NROInfo {
  88. SHA256Hash hash{};
  89. VAddr nro_address{};
  90. std::size_t nro_size{};
  91. VAddr bss_address{};
  92. std::size_t bss_size{};
  93. std::size_t text_size{};
  94. std::size_t ro_size{};
  95. std::size_t data_size{};
  96. VAddr src_addr{};
  97. };
  98. static_assert(sizeof(NROInfo) == 0x60, "NROInfo has invalid size.");
  99. class DebugMonitor final : public ServiceFramework<DebugMonitor> {
  100. public:
  101. explicit DebugMonitor(Core::System& system_) : ServiceFramework{system_, "ldr:dmnt"} {
  102. // clang-format off
  103. static const FunctionInfo functions[] = {
  104. {0, nullptr, "SetProgramArgument"},
  105. {1, nullptr, "FlushArguments"},
  106. {2, nullptr, "GetProcessModuleInfo"},
  107. };
  108. // clang-format on
  109. RegisterHandlers(functions);
  110. }
  111. };
  112. class ProcessManager final : public ServiceFramework<ProcessManager> {
  113. public:
  114. explicit ProcessManager(Core::System& system_) : ServiceFramework{system_, "ldr:pm"} {
  115. // clang-format off
  116. static const FunctionInfo functions[] = {
  117. {0, nullptr, "CreateProcess"},
  118. {1, nullptr, "GetProgramInfo"},
  119. {2, nullptr, "PinProgram"},
  120. {3, nullptr, "UnpinProgram"},
  121. {4, nullptr, "SetEnabledProgramVerification"},
  122. };
  123. // clang-format on
  124. RegisterHandlers(functions);
  125. }
  126. };
  127. class Shell final : public ServiceFramework<Shell> {
  128. public:
  129. explicit Shell(Core::System& system_) : ServiceFramework{system_, "ldr:shel"} {
  130. // clang-format off
  131. static const FunctionInfo functions[] = {
  132. {0, nullptr, "SetProgramArgument"},
  133. {1, nullptr, "FlushArguments"},
  134. };
  135. // clang-format on
  136. RegisterHandlers(functions);
  137. }
  138. };
  139. class RelocatableObject final : public ServiceFramework<RelocatableObject> {
  140. public:
  141. explicit RelocatableObject(Core::System& system_)
  142. : ServiceFramework{system_, "ldr:ro", ServiceThreadType::CreateNew} {
  143. // clang-format off
  144. static const FunctionInfo functions[] = {
  145. {0, &RelocatableObject::LoadModule, "LoadModule"},
  146. {1, &RelocatableObject::UnloadModule, "UnloadModule"},
  147. {2, &RelocatableObject::RegisterModuleInfo, "RegisterModuleInfo"},
  148. {3, &RelocatableObject::UnregisterModuleInfo, "UnregisterModuleInfo"},
  149. {4, &RelocatableObject::Initialize, "Initialize"},
  150. {10, nullptr, "RegisterModuleInfo2"},
  151. };
  152. // clang-format on
  153. RegisterHandlers(functions);
  154. }
  155. void RegisterModuleInfo(Kernel::HLERequestContext& ctx) {
  156. struct Parameters {
  157. u64_le process_id;
  158. u64_le nrr_address;
  159. u64_le nrr_size;
  160. };
  161. IPC::RequestParser rp{ctx};
  162. const auto [process_id, nrr_address, nrr_size] = rp.PopRaw<Parameters>();
  163. LOG_DEBUG(Service_LDR,
  164. "called with process_id={:016X}, nrr_address={:016X}, nrr_size={:016X}",
  165. process_id, nrr_address, nrr_size);
  166. if (!initialized) {
  167. LOG_ERROR(Service_LDR, "LDR:RO not initialized before use!");
  168. IPC::ResponseBuilder rb{ctx, 2};
  169. rb.Push(ERROR_NOT_INITIALIZED);
  170. return;
  171. }
  172. if (nrr.size() >= MAXIMUM_LOADED_RO) {
  173. LOG_ERROR(Service_LDR, "Loading new NRR would exceed the maximum number of loaded NRRs "
  174. "(0x40)! Failing...");
  175. IPC::ResponseBuilder rb{ctx, 2};
  176. rb.Push(ERROR_MAXIMUM_NRR);
  177. return;
  178. }
  179. // NRR Address does not fall on 0x1000 byte boundary
  180. if (!Common::Is4KBAligned(nrr_address)) {
  181. LOG_ERROR(Service_LDR, "NRR Address has invalid alignment (actual {:016X})!",
  182. nrr_address);
  183. IPC::ResponseBuilder rb{ctx, 2};
  184. rb.Push(ERROR_INVALID_ALIGNMENT);
  185. return;
  186. }
  187. // NRR Size is zero or causes overflow
  188. if (nrr_address + nrr_size <= nrr_address || nrr_size == 0 ||
  189. !Common::Is4KBAligned(nrr_size)) {
  190. LOG_ERROR(Service_LDR, "NRR Size is invalid! (nrr_address={:016X}, nrr_size={:016X})",
  191. nrr_address, nrr_size);
  192. IPC::ResponseBuilder rb{ctx, 2};
  193. rb.Push(ERROR_INVALID_SIZE);
  194. return;
  195. }
  196. // Read NRR data from memory
  197. std::vector<u8> nrr_data(nrr_size);
  198. system.Memory().ReadBlock(nrr_address, nrr_data.data(), nrr_size);
  199. NRRHeader header;
  200. std::memcpy(&header, nrr_data.data(), sizeof(NRRHeader));
  201. if (header.magic != Common::MakeMagic('N', 'R', 'R', '0')) {
  202. LOG_ERROR(Service_LDR, "NRR did not have magic 'NRR0' (actual {:08X})!", header.magic);
  203. IPC::ResponseBuilder rb{ctx, 2};
  204. rb.Push(ERROR_INVALID_NRR);
  205. return;
  206. }
  207. if (header.size != nrr_size) {
  208. LOG_ERROR(Service_LDR,
  209. "NRR header reported size did not match LoadNrr parameter size! "
  210. "(header_size={:016X}, loadnrr_size={:016X})",
  211. header.size, nrr_size);
  212. IPC::ResponseBuilder rb{ctx, 2};
  213. rb.Push(ERROR_INVALID_SIZE);
  214. return;
  215. }
  216. if (system.GetApplicationProcessProgramID() != header.application_id) {
  217. LOG_ERROR(Service_LDR,
  218. "Attempting to load NRR with title ID other than current process. (actual "
  219. "{:016X})!",
  220. header.application_id);
  221. IPC::ResponseBuilder rb{ctx, 2};
  222. rb.Push(ERROR_INVALID_NRR);
  223. return;
  224. }
  225. std::vector<SHA256Hash> hashes;
  226. // Copy all hashes in the NRR (specified by hash count/hash offset) into vector.
  227. for (std::size_t i = header.hash_offset;
  228. i < (header.hash_offset + (header.hash_count * sizeof(SHA256Hash))); i += 8) {
  229. SHA256Hash hash;
  230. std::memcpy(hash.data(), nrr_data.data() + i, sizeof(SHA256Hash));
  231. hashes.emplace_back(hash);
  232. }
  233. nrr.insert_or_assign(nrr_address, std::move(hashes));
  234. IPC::ResponseBuilder rb{ctx, 2};
  235. rb.Push(ResultSuccess);
  236. }
  237. void UnregisterModuleInfo(Kernel::HLERequestContext& ctx) {
  238. IPC::RequestParser rp{ctx};
  239. const auto pid = rp.Pop<u64>();
  240. const auto nrr_address = rp.Pop<VAddr>();
  241. LOG_DEBUG(Service_LDR, "called with pid={}, nrr_address={:016X}", pid, nrr_address);
  242. nrr.erase(nrr_address);
  243. IPC::ResponseBuilder rb{ctx, 2};
  244. rb.Push(ResultSuccess);
  245. }
  246. bool ValidateRegionForMap(Kernel::KPageTable& page_table, VAddr start, std::size_t size) const {
  247. const std::size_t padding_size{page_table.GetNumGuardPages() * Kernel::PageSize};
  248. const auto start_info{page_table.QueryInfo(start - 1)};
  249. if (start_info.GetState() != Kernel::KMemoryState::Free) {
  250. return {};
  251. }
  252. if (start_info.GetAddress() > (start - padding_size)) {
  253. return {};
  254. }
  255. const auto end_info{page_table.QueryInfo(start + size)};
  256. if (end_info.GetState() != Kernel::KMemoryState::Free) {
  257. return {};
  258. }
  259. return (start + size + padding_size) <= (end_info.GetAddress() + end_info.GetSize());
  260. }
  261. Result GetAvailableMapRegion(Kernel::KPageTable& page_table, u64 size, VAddr& out_addr) {
  262. size = Common::AlignUp(size, Kernel::PageSize);
  263. size += page_table.GetNumGuardPages() * Kernel::PageSize * 4;
  264. const auto is_region_available = [&](VAddr addr) {
  265. const auto end_addr = addr + size;
  266. while (addr < end_addr) {
  267. if (system.Memory().IsValidVirtualAddress(addr)) {
  268. return false;
  269. }
  270. if (!page_table.IsInsideAddressSpace(out_addr, size)) {
  271. return false;
  272. }
  273. if (page_table.IsInsideHeapRegion(out_addr, size)) {
  274. return false;
  275. }
  276. if (page_table.IsInsideAliasRegion(out_addr, size)) {
  277. return false;
  278. }
  279. addr += Kernel::PageSize;
  280. }
  281. return true;
  282. };
  283. bool succeeded = false;
  284. const auto map_region_end =
  285. page_table.GetAliasCodeRegionStart() + page_table.GetAliasCodeRegionSize();
  286. while (current_map_addr < map_region_end) {
  287. if (is_region_available(current_map_addr)) {
  288. succeeded = true;
  289. break;
  290. }
  291. current_map_addr += 0x100000;
  292. }
  293. if (!succeeded) {
  294. ASSERT_MSG(false, "Out of address space!");
  295. return Kernel::ResultOutOfMemory;
  296. }
  297. out_addr = current_map_addr;
  298. current_map_addr += size;
  299. return ResultSuccess;
  300. }
  301. ResultVal<VAddr> MapProcessCodeMemory(Kernel::KProcess* process, VAddr base_addr, u64 size) {
  302. auto& page_table{process->PageTable()};
  303. VAddr addr{};
  304. for (std::size_t retry = 0; retry < MAXIMUM_MAP_RETRIES; retry++) {
  305. R_TRY(GetAvailableMapRegion(page_table, size, addr));
  306. const Result result{page_table.MapCodeMemory(addr, base_addr, size)};
  307. if (result == Kernel::ResultInvalidCurrentMemory) {
  308. continue;
  309. }
  310. R_TRY(result);
  311. if (ValidateRegionForMap(page_table, addr, size)) {
  312. return addr;
  313. }
  314. }
  315. return ERROR_INSUFFICIENT_ADDRESS_SPACE;
  316. }
  317. ResultVal<VAddr> MapNro(Kernel::KProcess* process, VAddr nro_addr, std::size_t nro_size,
  318. VAddr bss_addr, std::size_t bss_size, std::size_t size) {
  319. for (std::size_t retry = 0; retry < MAXIMUM_MAP_RETRIES; retry++) {
  320. auto& page_table{process->PageTable()};
  321. VAddr addr{};
  322. CASCADE_RESULT(addr, MapProcessCodeMemory(process, nro_addr, nro_size));
  323. if (bss_size) {
  324. auto block_guard = detail::ScopeExit([&] {
  325. page_table.UnmapCodeMemory(
  326. addr + nro_size, bss_addr, bss_size,
  327. Kernel::KPageTable::ICacheInvalidationStrategy::InvalidateRange);
  328. page_table.UnmapCodeMemory(
  329. addr, nro_addr, nro_size,
  330. Kernel::KPageTable::ICacheInvalidationStrategy::InvalidateRange);
  331. });
  332. const Result result{page_table.MapCodeMemory(addr + nro_size, bss_addr, bss_size)};
  333. if (result == Kernel::ResultInvalidCurrentMemory) {
  334. continue;
  335. }
  336. if (result.IsError()) {
  337. return result;
  338. }
  339. block_guard.Cancel();
  340. }
  341. if (ValidateRegionForMap(page_table, addr, size)) {
  342. return addr;
  343. }
  344. }
  345. return ERROR_INSUFFICIENT_ADDRESS_SPACE;
  346. }
  347. Result LoadNro(Kernel::KProcess* process, const NROHeader& nro_header, VAddr nro_addr,
  348. VAddr start) const {
  349. const VAddr text_start{start + nro_header.segment_headers[TEXT_INDEX].memory_offset};
  350. const VAddr ro_start{start + nro_header.segment_headers[RO_INDEX].memory_offset};
  351. const VAddr data_start{start + nro_header.segment_headers[DATA_INDEX].memory_offset};
  352. const VAddr bss_start{data_start + nro_header.segment_headers[DATA_INDEX].memory_size};
  353. const VAddr bss_end_addr{
  354. Common::AlignUp(bss_start + nro_header.bss_size, Kernel::PageSize)};
  355. const auto CopyCode = [this, process](VAddr src_addr, VAddr dst_addr, u64 size) {
  356. system.Memory().CopyBlock(*process, dst_addr, src_addr, size);
  357. };
  358. CopyCode(nro_addr + nro_header.segment_headers[TEXT_INDEX].memory_offset, text_start,
  359. nro_header.segment_headers[TEXT_INDEX].memory_size);
  360. CopyCode(nro_addr + nro_header.segment_headers[RO_INDEX].memory_offset, ro_start,
  361. nro_header.segment_headers[RO_INDEX].memory_size);
  362. CopyCode(nro_addr + nro_header.segment_headers[DATA_INDEX].memory_offset, data_start,
  363. nro_header.segment_headers[DATA_INDEX].memory_size);
  364. CASCADE_CODE(process->PageTable().SetProcessMemoryPermission(
  365. text_start, ro_start - text_start, Kernel::Svc::MemoryPermission::ReadExecute));
  366. CASCADE_CODE(process->PageTable().SetProcessMemoryPermission(
  367. ro_start, data_start - ro_start, Kernel::Svc::MemoryPermission::Read));
  368. return process->PageTable().SetProcessMemoryPermission(
  369. data_start, bss_end_addr - data_start, Kernel::Svc::MemoryPermission::ReadWrite);
  370. }
  371. void LoadModule(Kernel::HLERequestContext& ctx) {
  372. struct Parameters {
  373. u64_le process_id;
  374. u64_le image_address;
  375. u64_le image_size;
  376. u64_le bss_address;
  377. u64_le bss_size;
  378. };
  379. IPC::RequestParser rp{ctx};
  380. const auto [process_id, nro_address, nro_size, bss_address, bss_size] =
  381. rp.PopRaw<Parameters>();
  382. LOG_DEBUG(Service_LDR,
  383. "called with pid={:016X}, nro_addr={:016X}, nro_size={:016X}, bss_addr={:016X}, "
  384. "bss_size={:016X}",
  385. process_id, nro_address, nro_size, bss_address, bss_size);
  386. if (!initialized) {
  387. LOG_ERROR(Service_LDR, "LDR:RO not initialized before use!");
  388. IPC::ResponseBuilder rb{ctx, 2};
  389. rb.Push(ERROR_NOT_INITIALIZED);
  390. return;
  391. }
  392. if (nro.size() >= MAXIMUM_LOADED_RO) {
  393. LOG_ERROR(Service_LDR, "Loading new NRO would exceed the maximum number of loaded NROs "
  394. "(0x40)! Failing...");
  395. IPC::ResponseBuilder rb{ctx, 2};
  396. rb.Push(ERROR_MAXIMUM_NRO);
  397. return;
  398. }
  399. // NRO Address does not fall on 0x1000 byte boundary
  400. if (!Common::Is4KBAligned(nro_address)) {
  401. LOG_ERROR(Service_LDR, "NRO Address has invalid alignment (actual {:016X})!",
  402. nro_address);
  403. IPC::ResponseBuilder rb{ctx, 2};
  404. rb.Push(ERROR_INVALID_ALIGNMENT);
  405. return;
  406. }
  407. // NRO Size or BSS Size is zero or causes overflow
  408. const auto nro_size_valid =
  409. nro_size != 0 && nro_address + nro_size > nro_address && Common::Is4KBAligned(nro_size);
  410. const auto bss_size_valid = nro_size + bss_size >= nro_size &&
  411. (bss_size == 0 || bss_address + bss_size > bss_address);
  412. if (!nro_size_valid || !bss_size_valid) {
  413. LOG_ERROR(Service_LDR,
  414. "NRO Size or BSS Size is invalid! (nro_address={:016X}, nro_size={:016X}, "
  415. "bss_address={:016X}, bss_size={:016X})",
  416. nro_address, nro_size, bss_address, bss_size);
  417. IPC::ResponseBuilder rb{ctx, 2};
  418. rb.Push(ERROR_INVALID_SIZE);
  419. return;
  420. }
  421. // Read NRO data from memory
  422. std::vector<u8> nro_data(nro_size);
  423. system.Memory().ReadBlock(nro_address, nro_data.data(), nro_size);
  424. SHA256Hash hash{};
  425. mbedtls_sha256_ret(nro_data.data(), nro_data.size(), hash.data(), 0);
  426. // NRO Hash is already loaded
  427. if (std::any_of(nro.begin(), nro.end(), [&hash](const std::pair<VAddr, NROInfo>& info) {
  428. return info.second.hash == hash;
  429. })) {
  430. LOG_ERROR(Service_LDR, "NRO is already loaded!");
  431. IPC::ResponseBuilder rb{ctx, 2};
  432. rb.Push(ERROR_ALREADY_LOADED);
  433. return;
  434. }
  435. // NRO Hash is not in any loaded NRR
  436. if (!IsValidNROHash(hash)) {
  437. LOG_ERROR(Service_LDR,
  438. "NRO hash is not present in any currently loaded NRRs (hash={})!",
  439. Common::HexToString(hash));
  440. IPC::ResponseBuilder rb{ctx, 2};
  441. rb.Push(ERROR_MISSING_NRR_HASH);
  442. return;
  443. }
  444. // Load and validate the NRO header
  445. NROHeader header{};
  446. std::memcpy(&header, nro_data.data(), sizeof(NROHeader));
  447. if (!IsValidNRO(header, nro_size, bss_size)) {
  448. LOG_ERROR(Service_LDR, "NRO was invalid!");
  449. IPC::ResponseBuilder rb{ctx, 2};
  450. rb.Push(ERROR_INVALID_NRO);
  451. return;
  452. }
  453. // Map memory for the NRO
  454. const auto map_result{MapNro(system.ApplicationProcess(), nro_address, nro_size,
  455. bss_address, bss_size, nro_size + bss_size)};
  456. if (map_result.Failed()) {
  457. IPC::ResponseBuilder rb{ctx, 2};
  458. rb.Push(map_result.Code());
  459. }
  460. // Load the NRO into the mapped memory
  461. if (const auto result{
  462. LoadNro(system.ApplicationProcess(), header, nro_address, *map_result)};
  463. result.IsError()) {
  464. IPC::ResponseBuilder rb{ctx, 2};
  465. rb.Push(map_result.Code());
  466. }
  467. // Track the loaded NRO
  468. nro.insert_or_assign(*map_result,
  469. NROInfo{hash, *map_result, nro_size, bss_address, bss_size,
  470. header.segment_headers[TEXT_INDEX].memory_size,
  471. header.segment_headers[RO_INDEX].memory_size,
  472. header.segment_headers[DATA_INDEX].memory_size, nro_address});
  473. IPC::ResponseBuilder rb{ctx, 4};
  474. rb.Push(ResultSuccess);
  475. rb.Push(*map_result);
  476. }
  477. Result UnmapNro(const NROInfo& info) {
  478. // Each region must be unmapped separately to validate memory state
  479. auto& page_table{system.ApplicationProcess()->PageTable()};
  480. if (info.bss_size != 0) {
  481. CASCADE_CODE(page_table.UnmapCodeMemory(
  482. info.nro_address + info.text_size + info.ro_size + info.data_size, info.bss_address,
  483. info.bss_size, Kernel::KPageTable::ICacheInvalidationStrategy::InvalidateRange));
  484. }
  485. CASCADE_CODE(page_table.UnmapCodeMemory(
  486. info.nro_address + info.text_size + info.ro_size,
  487. info.src_addr + info.text_size + info.ro_size, info.data_size,
  488. Kernel::KPageTable::ICacheInvalidationStrategy::InvalidateRange));
  489. CASCADE_CODE(page_table.UnmapCodeMemory(
  490. info.nro_address + info.text_size, info.src_addr + info.text_size, info.ro_size,
  491. Kernel::KPageTable::ICacheInvalidationStrategy::InvalidateRange));
  492. CASCADE_CODE(page_table.UnmapCodeMemory(
  493. info.nro_address, info.src_addr, info.text_size,
  494. Kernel::KPageTable::ICacheInvalidationStrategy::InvalidateRange));
  495. return ResultSuccess;
  496. }
  497. void UnloadModule(Kernel::HLERequestContext& ctx) {
  498. if (!initialized) {
  499. LOG_ERROR(Service_LDR, "LDR:RO not initialized before use!");
  500. IPC::ResponseBuilder rb{ctx, 2};
  501. rb.Push(ERROR_NOT_INITIALIZED);
  502. return;
  503. }
  504. struct Parameters {
  505. u64_le process_id;
  506. u64_le nro_address;
  507. };
  508. IPC::RequestParser rp{ctx};
  509. const auto [process_id, nro_address] = rp.PopRaw<Parameters>();
  510. LOG_DEBUG(Service_LDR, "called with process_id={:016X}, nro_address=0x{:016X}", process_id,
  511. nro_address);
  512. if (!Common::Is4KBAligned(nro_address)) {
  513. LOG_ERROR(Service_LDR, "NRO address has invalid alignment (nro_address=0x{:016X})",
  514. nro_address);
  515. IPC::ResponseBuilder rb{ctx, 2};
  516. rb.Push(ERROR_INVALID_ALIGNMENT);
  517. return;
  518. }
  519. const auto iter = nro.find(nro_address);
  520. if (iter == nro.end()) {
  521. LOG_ERROR(Service_LDR,
  522. "The NRO attempting to be unmapped was not mapped or has an invalid address "
  523. "(nro_address=0x{:016X})!",
  524. nro_address);
  525. IPC::ResponseBuilder rb{ctx, 2};
  526. rb.Push(ERROR_INVALID_NRO_ADDRESS);
  527. return;
  528. }
  529. const auto result{UnmapNro(iter->second)};
  530. nro.erase(iter);
  531. IPC::ResponseBuilder rb{ctx, 2};
  532. rb.Push(result);
  533. }
  534. void Initialize(Kernel::HLERequestContext& ctx) {
  535. LOG_WARNING(Service_LDR, "(STUBBED) called");
  536. initialized = true;
  537. current_map_addr = system.ApplicationProcess()->PageTable().GetAliasCodeRegionStart();
  538. IPC::ResponseBuilder rb{ctx, 2};
  539. rb.Push(ResultSuccess);
  540. }
  541. private:
  542. bool initialized{};
  543. std::map<VAddr, NROInfo> nro;
  544. std::map<VAddr, std::vector<SHA256Hash>> nrr;
  545. VAddr current_map_addr{};
  546. bool IsValidNROHash(const SHA256Hash& hash) const {
  547. return std::any_of(nrr.begin(), nrr.end(), [&hash](const auto& p) {
  548. return std::find(p.second.begin(), p.second.end(), hash) != p.second.end();
  549. });
  550. }
  551. static bool IsValidNRO(const NROHeader& header, u64 nro_size, u64 bss_size) {
  552. return header.magic == Common::MakeMagic('N', 'R', 'O', '0') &&
  553. header.nro_size == nro_size && header.bss_size == bss_size &&
  554. header.segment_headers[RO_INDEX].memory_offset ==
  555. header.segment_headers[TEXT_INDEX].memory_offset +
  556. header.segment_headers[TEXT_INDEX].memory_size &&
  557. header.segment_headers[DATA_INDEX].memory_offset ==
  558. header.segment_headers[RO_INDEX].memory_offset +
  559. header.segment_headers[RO_INDEX].memory_size &&
  560. nro_size == header.segment_headers[DATA_INDEX].memory_offset +
  561. header.segment_headers[DATA_INDEX].memory_size &&
  562. Common::Is4KBAligned(header.segment_headers[TEXT_INDEX].memory_size) &&
  563. Common::Is4KBAligned(header.segment_headers[RO_INDEX].memory_size) &&
  564. Common::Is4KBAligned(header.segment_headers[DATA_INDEX].memory_size);
  565. }
  566. };
  567. void InstallInterfaces(SM::ServiceManager& sm, Core::System& system) {
  568. std::make_shared<DebugMonitor>(system)->InstallAsService(sm);
  569. std::make_shared<ProcessManager>(system)->InstallAsService(sm);
  570. std::make_shared<Shell>(system)->InstallAsService(sm);
  571. std::make_shared<RelocatableObject>(system)->InstallAsService(sm);
  572. }
  573. } // namespace Service::LDR