From fe2f227dfc3818f17364d89876c3b434e116441b Mon Sep 17 00:00:00 2001
From: Michael Scire <SciresM@gmail.com>
Date: Thu, 26 Apr 2018 16:48:57 -0600
Subject: [PATCH] Loader: Implement ldr:ro->Initialize()

---
 stratosphere/loader/source/ldr_ro_service.cpp | 23 +++++++++++++------
 stratosphere/loader/source/ldr_ro_service.hpp | 10 ++++----
 2 files changed, 21 insertions(+), 12 deletions(-)

diff --git a/stratosphere/loader/source/ldr_ro_service.cpp b/stratosphere/loader/source/ldr_ro_service.cpp
index f5f7f5984..2375a0b0b 100644
--- a/stratosphere/loader/source/ldr_ro_service.cpp
+++ b/stratosphere/loader/source/ldr_ro_service.cpp
@@ -31,27 +31,36 @@ Result RelocatableObjectsService::dispatch(IpcParsedCommand &r, IpcCommand &out_
 }
 
 
-std::tuple<Result, u64> load_nro(PidDescriptor pid, u64 nro_address, u64 nro_size, u64 bss_address, u64 bss_size) {
+std::tuple<Result, u64> RelocatableObjectsService::load_nro(PidDescriptor pid_desc, u64 nro_address, u64 nro_size, u64 bss_address, u64 bss_size) {
     /* TODO */
     return std::make_tuple(0xF601, 0);
 }
 
-std::tuple<Result> unload_nro(PidDescriptor pid, u64 nro_address) {
+std::tuple<Result> RelocatableObjectsService::unload_nro(PidDescriptor pid_desc, u64 nro_address) {
     /* TODO */
     return std::make_tuple(0xF601);
 }
 
-std::tuple<Result> load_nrr(PidDescriptor pid, u64 nrr_address, u64 nrr_size) {
+std::tuple<Result> RelocatableObjectsService::load_nrr(PidDescriptor pid_desc, u64 nrr_address, u64 nrr_size) {
     /* TODO */
     return std::make_tuple(0xF601);
 }
 
-std::tuple<Result> unload_nrr(PidDescriptor pid, u64 nrr_address) {
+std::tuple<Result> RelocatableObjectsService::unload_nrr(PidDescriptor pid_desc, u64 nrr_address) {
     /* TODO */
     return std::make_tuple(0xF601);
 }
 
-std::tuple<Result> initialize(PidDescriptor pid, CopiedHandle process_h) {
-    /* TODO */
-    return std::make_tuple(0xF601);
+std::tuple<Result> RelocatableObjectsService::initialize(PidDescriptor pid_desc, CopiedHandle process_h) {
+    u64 handle_pid;
+    Result rc = 0xAE09;
+    if (R_SUCCEEDED(svcGetProcessId(&handle_pid, process_h.handle)) && handle_pid == pid_desc.pid) {
+        if (this->has_initialized) {
+            svcCloseHandle(this->process_handle);
+        }
+        this->process_handle = process_h.handle;
+        this->has_initialized = true;
+        rc = 0;
+    }
+    return std::make_tuple(rc);
 }
diff --git a/stratosphere/loader/source/ldr_ro_service.hpp b/stratosphere/loader/source/ldr_ro_service.hpp
index 29d057ba0..f2a02f252 100644
--- a/stratosphere/loader/source/ldr_ro_service.hpp
+++ b/stratosphere/loader/source/ldr_ro_service.hpp
@@ -26,9 +26,9 @@ class RelocatableObjectsService : IServiceObject {
         
     private:
         /* Actual commands. */
-        std::tuple<Result, u64> load_nro(PidDescriptor pid, u64 nro_address, u64 nro_size, u64 bss_address, u64 bss_size);
-        std::tuple<Result> unload_nro(PidDescriptor pid, u64 nro_address);
-        std::tuple<Result> load_nrr(PidDescriptor pid, u64 nrr_address, u64 nrr_size);
-        std::tuple<Result> unload_nrr(PidDescriptor pid, u64 nrr_address);
-        std::tuple<Result> initialize(PidDescriptor pid, CopiedHandle process_h);
+        std::tuple<Result, u64> load_nro(PidDescriptor pid_desc, u64 nro_address, u64 nro_size, u64 bss_address, u64 bss_size);
+        std::tuple<Result> unload_nro(PidDescriptor pid_desc, u64 nro_address);
+        std::tuple<Result> load_nrr(PidDescriptor pid_desc, u64 nrr_address, u64 nrr_size);
+        std::tuple<Result> unload_nrr(PidDescriptor pid_desc, u64 nrr_address);
+        std::tuple<Result> initialize(PidDescriptor pid_desc, CopiedHandle process_h);
 };
\ No newline at end of file