ldr.cpp 24 KB

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