From ee9bfd514cc92a1865c985437d1bf7bd1e979ac1 Mon Sep 17 00:00:00 2001
From: Michael Scire <SciresM@gmail.com>
Date: Tue, 24 Apr 2018 06:14:04 -0600
Subject: [PATCH] Loader: Implement argument passing.

---
 stratosphere/loader/source/ldr_nso.cpp | 26 +++++++++++++++++++++++++-
 stratosphere/loader/source/ldr_nso.hpp |  7 +++++++
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/stratosphere/loader/source/ldr_nso.cpp b/stratosphere/loader/source/ldr_nso.cpp
index 6df1ee8c9..2761dedf6 100644
--- a/stratosphere/loader/source/ldr_nso.cpp
+++ b/stratosphere/loader/source/ldr_nso.cpp
@@ -234,6 +234,30 @@ Result NsoUtils::LoadNsosIntoProcessMemory(Handle process_h, u64 title_id, NsoLo
         }
     }
     
-    /* TODO: Map in arguments, here. */
+    if (args != NULL && args_size) {
+        u64 arg_map_addr = 0;
+        if (R_FAILED((rc = MapUtils::LocateSpaceForMap(&arg_map_addr, extents->args_size)))) {
+            return rc;
+        }
+        
+        NsoArgument *arg_map_base = (NsoArgument *)arg_map_addr;
+        
+        if (R_FAILED((rc = svcMapProcessMemory(arg_map_base, process_h, extents->args_address, extents->args_size)))) {
+            return rc;
+        }
+        
+        arg_map_base->allocated_space = extents->args_size;
+        arg_map_base->args_size = args_size;
+        std::fill(arg_map_base->_0x8, arg_map_base->_0x8 + sizeof(arg_map_base->_0x8), 0);
+        std::copy(args, args + args_size, arg_map_base->arguments);
+        
+        if (R_FAILED((rc = svcUnmapProcessMemory(arg_map_base, process_h, extents->args_address, extents->args_size)))) {
+            return rc;
+        }
+        
+        if (R_FAILED((rc = svcSetProcessMemoryPermission(process_h, extents->args_address, extents->args_size, 3)))) {
+            return rc;
+        }
+    }
     return rc;
 }
diff --git a/stratosphere/loader/source/ldr_nso.hpp b/stratosphere/loader/source/ldr_nso.hpp
index f26998496..8424f7d25 100644
--- a/stratosphere/loader/source/ldr_nso.hpp
+++ b/stratosphere/loader/source/ldr_nso.hpp
@@ -37,6 +37,13 @@ class NsoUtils {
             u64 nso_sizes[NSO_NUM_MAX];
         };
         
+        struct NsoArgument {
+            u32 allocated_space;
+            u32 args_size;
+            u8  _0x8[0x18];
+            u8  arguments[];
+        };
+        
         
         static_assert(sizeof(NsoHeader) == 0x100, "Incorrectly defined NsoHeader!");