From c4c14496afc12b3f4f5ae6e9c335e379a212d1c2 Mon Sep 17 00:00:00 2001
From: Maschell <Maschell@gmx.de>
Date: Mon, 30 Nov 2020 17:45:51 +0100
Subject: [PATCH] Add support for special return values

---
 source/main.cpp | 49 +++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 39 insertions(+), 10 deletions(-)

diff --git a/source/main.cpp b/source/main.cpp
index e78cede..8ba46ec 100644
--- a/source/main.cpp
+++ b/source/main.cpp
@@ -25,6 +25,8 @@
 
 void SplashScreen(int32_t durationInMs);
 
+void forceDefaultTitleIDToWiiUMenu();
+
 bool CheckRunning() {
 
     switch (ProcUIProcessMessages(true)) {
@@ -46,6 +48,7 @@ bool CheckRunning() {
 }
 
 extern "C" void _SYSLaunchMenuWithCheckingAccount(nn::act::SlotNo slot);
+extern "C" void SYSLaunchMiiStudio(void * SysMiiStudioArgs);
 
 int main(int argc, char **argv) {
     WHBLogUdpInit();
@@ -87,10 +90,35 @@ int main(int argc, char **argv) {
         if (entryPoint != 0) {
             DEBUG_FUNCTION_LINE("New entrypoint at %08X", entryPoint);
             int res = ((int (*)(int, char **)) entryPoint)(argc, argv);
-            if (res > 0) {
+            WHBLogUdpInit();
+            if (res >= 0) {                
                 DEBUG_FUNCTION_LINE("Returning result of payload");
+                if(res & 1){
+                    DEBUG_FUNCTION_LINE("Force default title id");
+                    forceDefaultTitleIDToWiiUMenu();
+                }
+                if(res & 4){
+                    DEBUG_FUNCTION_LINE("do produi");
+                    ProcUIInit(OSSavesDone_ReadyToRelease);
+                }
+                if(res & 2){
+                    DEBUG_FUNCTION_LINE("Launch menu");
+                    SYSLaunchMenu();
+                }else if(res & 8){
+                    DEBUG_FUNCTION_LINE("Launch mii maker");
+                    SYSLaunchMiiStudio(0);
+                }
+                if(res & 4){                   
+                    DEBUG_FUNCTION_LINE("ProcUIInit done");
+                     while (CheckRunning()) {
+                        // wait.
+                        OSSleepTicks(OSMillisecondsToTicks(100));
+                    }
+                    ProcUIShutdown();
+                }
+                DEBUG_FUNCTION_LINE("Exiting.");
                 WHBLogUdpDeinit();
-                return 0;
+                return res;
             } else {
                 loadWithoutHacks = true;
             }
@@ -104,14 +132,7 @@ int main(int argc, char **argv) {
 
     if (loadWithoutHacks) {
         DEBUG_FUNCTION_LINE("Load Wii U Menu");
-        // Restore the default title id to the normal Wii U Menu.
-        unsigned long long sysmenuIdUll = _SYSGetSystemApplicationTitleId(SYSTEM_APP_ID_HOME_MENU);
-        memcpy((void *) 0xF417FFF0, &sysmenuIdUll, 8);
-        DCStoreRange((void *) 0xF417FFF0, 0x8);
-
-        DEBUG_FUNCTION_LINE("Forcing start of title: %016llX", sysmenuIdUll);
-
-        ExecuteIOSExploit();
+        forceDefaultTitleIDToWiiUMenu();
 
         nn::act::Initialize();
         nn::act::SlotNo slot = nn::act::GetSlotNo();
@@ -139,6 +160,14 @@ int main(int argc, char **argv) {
     return 0;
 }
 
+void forceDefaultTitleIDToWiiUMenu() {// Restore the default title id to the normal Wii U Menu.
+    unsigned long long sysmenuIdUll = _SYSGetSystemApplicationTitleId(SYSTEM_APP_ID_HOME_MENU);
+    memcpy((void *) 0xF417FFF0, &sysmenuIdUll, 8);
+    DCStoreRange((void *) 0xF417FFF0, 0x8);
+    DEBUG_FUNCTION_LINE("Forcing start of title: %016llX", sysmenuIdUll);
+    ExecuteIOSExploit();
+}
+
 
 void SplashScreen(int32_t durationInMs) {
     int32_t screen_buf0_size = 0;