Implement a small proper proc_ui loop to be able to restart the app properly. Wtihout a clean restart unexpected behaviour is possible until the app i switched

source/main.cpp

@@ -2,10 +2,19 @@
 #include <string.h>
 
 #include <coreinit/ios.h>
+#include <coreinit/time.h>
+
+#include <coreinit/systeminfo.h>
+#include <coreinit/foreground.h>
+
+#include <proc_ui/procui.h>
+#include <coreinit/thread.h>
 
 #include <whb/proc.h>
 #include <whb/log.h>
 #include <whb/log_udp.h>
+#include <sysapp/launch.h>
+#include <coreinit/exit.h>
 
 #include "gx2sploit.h"
 
@@ -15,13 +24,41 @@ extern "C" {
 
 // Similar to SYSRelaunchTitle but with less checks
 void OSRestartGame(int,char**);
+void SYSLaunchMenu();
 
 #ifdef __cplusplus
 }
 #endif
 
+
+bool CheckRunning(){
+
+    switch(ProcUIProcessMessages(true))
+    {
+    case PROCUI_STATUS_EXITING:
+    { 
+		return false;
+    }
+    case PROCUI_STATUS_RELEASE_FOREGROUND:
+    {
+		ProcUIDrawDoneRelease();        
+        break;
+    }
+    case PROCUI_STATUS_IN_FOREGROUND:
+    {
+        break;
+    }
+    case PROCUI_STATUS_IN_BACKGROUND:
+    default:
+        break;
+    }
+    return true;
+}
+
+
 int main(int argc, char **argv) {
     WHBLogUdpInit();
+    bool returnToMenu = false;
     if(CheckKernelExploit()) {
         // When the kernel exploit is set up successfully, we signal the ios to move on.
         int mcpFd = IOS_Open("/dev/mcp", (IOSOpenMode)0);
@@ -31,10 +68,27 @@ int main(int argc, char **argv) {
            
             IOS_Ioctl(mcpFd, 100, &in, sizeof(in), &out, sizeof(out));
             IOS_Close(mcpFd);
+            returnToMenu = true;
         }
     }
+     
+   
+    
     WHBLogUdpDeinit();
-    OSRestartGame(argc,argv);
+    if(returnToMenu){     
+        ProcUIInit(OSSavesDone_ReadyToRelease);    
+        SYSLaunchMenu();
+        while(CheckRunning()){
+            // wait.
+            OSSleepTicks(OSMillisecondsToTicks(100));
+        }
+        ProcUIShutdown();
+    
+    }else{
+        OSRestartGame(argc, argv);
+    }
+
+ 
 
     return 0;
 }