From 97d1ca3cc974f35046b8676fb1547772ba79f312 Mon Sep 17 00:00:00 2001
From: hexkyz <mike.hexkyz@gmail.com>
Date: Fri, 31 Aug 2018 20:10:06 +0000
Subject: [PATCH] fusee: Disable built-in PM system module.

---
 fusee/fusee-secondary/src/stratosphere.c | 65 ++++++++++++++++++------
 1 file changed, 49 insertions(+), 16 deletions(-)

diff --git a/fusee/fusee-secondary/src/stratosphere.c b/fusee/fusee-secondary/src/stratosphere.c
index a0d1b5db0..4664da5d0 100644
--- a/fusee/fusee-secondary/src/stratosphere.c
+++ b/fusee/fusee-secondary/src/stratosphere.c
@@ -19,6 +19,11 @@
 
 static ini1_header_t *g_stratosphere_ini1 = NULL;
 
+static bool g_stratosphere_loader_enabled = true;
+static bool g_stratosphere_sm_enabled = true;
+static bool g_stratosphere_pm_enabled = false;
+static bool g_stratosphere_boot_enabled = false;
+
 extern const uint8_t boot_100_kip[], boot_200_kip[];
 extern const uint8_t loader_kip[], pm_kip[], sm_kip[];
 extern const uint32_t boot_100_kip_size, boot_200_kip_size;
@@ -27,8 +32,9 @@ extern const uint32_t loader_kip_size, pm_kip_size, sm_kip_size;
 /* GCC doesn't consider the size as const... we have to write it ourselves. */
 
 ini1_header_t *stratosphere_get_ini1(uint32_t target_firmware) {
-    /* const uint8_t *boot_kip = NULL; */
+    const uint8_t *boot_kip = NULL;
     uint32_t boot_kip_size = 0;
+    uint32_t num_processes = 0;
     uint8_t *data;
 
     if (g_stratosphere_ini1 != NULL) {
@@ -36,14 +42,36 @@ ini1_header_t *stratosphere_get_ini1(uint32_t target_firmware) {
     }
 
     if (target_firmware <= EXOSPHERE_TARGET_FIRMWARE_100) {
-        /* boot_kip = boot_100_kip; */
+        boot_kip = boot_100_kip;
         boot_kip_size = boot_100_kip_size;
     } else {
-        /* boot_kip = boot_200_kip; */
+        boot_kip = boot_200_kip;
         boot_kip_size = boot_200_kip_size;
     }
 
-    size_t size = sizeof(ini1_header_t) + loader_kip_size + pm_kip_size + sm_kip_size + boot_kip_size;
+    size_t size = sizeof(ini1_header_t);
+    
+    /* Calculate our processes' sizes. */
+    if (g_stratosphere_loader_enabled) {
+        size += loader_kip_size;
+        num_processes++;
+    }
+    
+    if (g_stratosphere_pm_enabled) {
+        size += pm_kip_size;
+        num_processes++;
+    }
+    
+    if (g_stratosphere_sm_enabled) {
+        size += sm_kip_size;
+        num_processes++;
+    }
+    
+    if (g_stratosphere_boot_enabled) {
+        size += boot_kip_size;
+        num_processes++;
+    }
+    
     g_stratosphere_ini1 = (ini1_header_t *)malloc(size);
 
     if (g_stratosphere_ini1 == NULL) {
@@ -52,26 +80,31 @@ ini1_header_t *stratosphere_get_ini1(uint32_t target_firmware) {
 
     g_stratosphere_ini1->magic = MAGIC_INI1;
     g_stratosphere_ini1->size = size;
-    g_stratosphere_ini1->num_processes = 3; /*TODO: Change to 4 when boot is supported. */
+    g_stratosphere_ini1->num_processes = num_processes;
     g_stratosphere_ini1->_0xC = 0;
 
     data = g_stratosphere_ini1->kip_data;
 
     /* Copy our processes. */
-    memcpy(data, loader_kip, loader_kip_size);
-    data += loader_kip_size;
+    if (g_stratosphere_loader_enabled) {
+        memcpy(data, loader_kip, loader_kip_size);
+        data += loader_kip_size;
+    }
 
-    memcpy(data, pm_kip, pm_kip_size);
-    data += pm_kip_size;
+    if (g_stratosphere_pm_enabled) {
+        memcpy(data, pm_kip, pm_kip_size);
+        data += pm_kip_size;
+    }
 
-    memcpy(data, sm_kip, sm_kip_size);
-    data += sm_kip_size;
+    if (g_stratosphere_sm_enabled) {
+        memcpy(data, sm_kip, sm_kip_size);
+        data += sm_kip_size;
+    }
 
-    /* TODO: Uncomment when boot is supported. */
-    /*
-    memcpy(data, boot_kip, boot_kip_size);
-    data += boot_kip_size;
-    */
+    if (g_stratosphere_boot_enabled) {
+        memcpy(data, boot_kip, boot_kip_size);
+        data += boot_kip_size;
+    }
     
     return g_stratosphere_ini1;
 }