From adc18fcc35354f9cbd2893f28531d74d49e7cef3 Mon Sep 17 00:00:00 2001
From: CrazyRedMachine <crazyredmachine86@gmail.com>
Date: Sun, 20 Aug 2023 12:19:57 +0200
Subject: [PATCH] back_to_song_select: no need to press red button

---
 popnhax/dllmain.cc | 66 ++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 64 insertions(+), 2 deletions(-)

diff --git a/popnhax/dllmain.cc b/popnhax/dllmain.cc
index 907e463..edc27db 100644
--- a/popnhax/dllmain.cc
+++ b/popnhax/dllmain.cc
@@ -275,6 +275,7 @@ bool g_pfree_mode = false;
 bool g_end_session = false;
 bool g_return_to_options = false;
 bool g_return_to_song_select = false;
+bool g_return_to_song_select_num9 = false;
 void (*real_screen_transition)();
 void quickexit_screen_transition()
 {
@@ -381,12 +382,24 @@ void backtosongselect_option_screen()
     __asm("cmp bl, 8\n");
     __asm("jne exit_back_select\n");
     g_return_to_song_select = true;
+    g_return_to_song_select_num9 = true;
 
     __asm("exit_back_select:\n");
 
     real_option_screen_later();
 }
 
+void (*real_backtosongselect_option_screen_auto_leave)();
+void backtosongselect_option_screen_auto_leave()
+{
+    if ( g_return_to_song_select_num9 )
+    {
+        g_return_to_song_select_num9 = false;
+        __asm("mov al, 1\n");
+    }
+    real_backtosongselect_option_screen_auto_leave();
+}
+
 void (*real_option_screen_yellow)();
 void backtosongselect_option_yellow()
 {
@@ -403,11 +416,25 @@ void backtosongselect_option_yellow()
     __asm("jne exit_back_select_yellow\n");
 
     g_return_to_song_select = true;
+    g_return_to_song_select_num9 = true;
 
     __asm("exit_back_select_yellow:\n");
     real_option_screen_yellow();
 }
 
+uint32_t g_option_yellow_leave_addr = 0;
+void (*real_backtosongselect_option_screen_yellow_auto_leave)();
+void backtosongselect_option_screen_yellow_auto_leave()
+{
+    if ( g_return_to_song_select_num9 )
+    {
+        g_return_to_song_select_num9 = false;
+        __asm("push %0\n": :"m"(g_option_yellow_leave_addr));
+        __asm("ret\n");
+    }
+    real_backtosongselect_option_screen_yellow_auto_leave();
+}
+
 static bool r_ran;
 static bool regul_flg;
 //static bool dj_auto;
@@ -2057,7 +2084,7 @@ static bool patch_quick_retire(bool pfree)
             int64_t pattern_offset = search(data, dllSize, "\x8B\x85\x0C\x0A\x00\x00\x83\x78\x34\x00\x75", 11, 0);
 
             if (pattern_offset == -1) {
-                LOG("popnhax: quick retry: cannot retrieve option screen loop function\n");
+                LOG("popnhax: back to song select: cannot retrieve option screen loop function\n");
                 return false;
             }
 
@@ -2065,6 +2092,19 @@ static bool patch_quick_retire(bool pfree)
             MH_CreateHook((LPVOID)patch_addr, (LPVOID)backtosongselect_option_screen,
                           (void **)&real_option_screen_later);
         }
+        /* automatically leave option screen after numpad 9 press */
+        {
+            int64_t pattern_offset = search(data, dllSize, "\x84\xC0\x75\x63\x8B\x85\x10\x0A\x00\x00\x83\xC0\x04\xBF\x0C\x00\x00\x00", 18, 0);
+
+            if (pattern_offset == -1) {
+                LOG("popnhax: back to song select: cannot retrieve option screen loop function\n");
+                return false;
+            }
+
+            uint64_t patch_addr = (int64_t)data + pattern_offset;
+            MH_CreateHook((LPVOID)patch_addr, (LPVOID)backtosongselect_option_screen_auto_leave,
+                          (void **)&real_backtosongselect_option_screen_auto_leave);
+        }
 
         /* go back to song select with numpad 9 on song option screen (after pressing yellow) */
         {
@@ -2079,7 +2119,29 @@ static bool patch_quick_retire(bool pfree)
             MH_CreateHook((LPVOID)patch_addr, (LPVOID)backtosongselect_option_yellow,
                           (void **)&real_option_screen_yellow);
         }
-        LOG("popnhax: quick retry: return to song select enabled\n");
+        /* automatically leave after numpad 9 press */
+        {
+            int64_t pattern_offset = search(data, dllSize, "\x8B\x55\x00\x8B\x82\x9C\x00\x00\x00\x6A\x01\x8B\xCD\xFF\xD0\x80\xBD", 17, 0);
+
+            if (pattern_offset == -1) {
+                LOG("popnhax: back to song select: cannot retrieve option screen yellow leave addr\n");
+                return false;
+            }
+
+            g_option_yellow_leave_addr = (int32_t)data + pattern_offset - 0x05;
+            pattern_offset = search(data, dllSize, "\x84\xC0\x0F\x84\xF1\x00\x00\x00\x8B\xC5", 10, 0);
+
+            if (pattern_offset == -1) {
+                LOG("popnhax: back to song select: cannot retrieve option screen yellow button check function\n");
+                return false;
+            }
+
+            uint64_t patch_addr = (int64_t)data + pattern_offset;
+            MH_CreateHook((LPVOID)patch_addr, (LPVOID)backtosongselect_option_screen_yellow_auto_leave,
+                          (void **)&real_backtosongselect_option_screen_yellow_auto_leave);
+        }
+
+        LOG("popnhax: quick retire: return to song select enabled\n");
     }
 
     if (pfree)