From 11801e6340ab29506499bfc9afe5df2d46a8e4d9 Mon Sep 17 00:00:00 2001
From: CrazyRedMachine <crazyredmachine86@gmail.com>
Date: Thu, 25 Apr 2024 00:09:33 +0200
Subject: [PATCH] wip custom highlight

---
 popnhax/custom_categs.cc | 87 ++++++++++++++++++++++++++++++++++++++++
 popnhax/dllmain.cc       |  2 +-
 2 files changed, 88 insertions(+), 1 deletion(-)

diff --git a/popnhax/custom_categs.cc b/popnhax/custom_categs.cc
index b2dba8e..e97a2c0 100644
--- a/popnhax/custom_categs.cc
+++ b/popnhax/custom_categs.cc
@@ -379,6 +379,91 @@ void hook_categ_listing()
     real_categ_listing();
 }
 
+void (*real_song_printf)();
+void hook_song_printf()
+{
+    __asm("push eax\n");
+    __asm("push ebx\n");
+    __asm("mov eax, [esp+0x50]\n");
+    __asm("cmp eax, _g_min_id\n");
+    __asm("jb print_regular_song\n");
+    __asm("cmp dword ptr _g_max_id, 0\n");
+    __asm("je print_custom_song\n");
+    __asm("cmp eax, _g_max_id\n");
+    __asm("ja print_regular_song\n");
+
+    __asm("print_custom_song:\n");
+    
+    __asm("lea eax, [esp+0x08]\n");
+    __asm("mov ebx, _g_categformat\n");
+    __asm("mov [eax], ebx\n");
+
+    __asm("print_regular_song:\n");
+    __asm("pop ebx\n");
+    __asm("pop eax\n");
+    real_song_printf();
+}
+
+void (*real_artist_printf)();
+void hook_artist_printf()
+{
+    __asm("push eax\n");
+    __asm("push ebx\n");
+    __asm("mov eax, [esp+0x50]\n");
+    __asm("cmp eax, _g_min_id\n");
+    __asm("jb print_regular_artist\n");
+    __asm("cmp dword ptr _g_max_id, 0\n");
+    __asm("je print_custom_artist\n");
+    __asm("cmp eax, _g_max_id\n");
+    __asm("ja print_regular_artist\n");
+
+    __asm("print_custom_artist:\n");
+    
+    __asm("lea eax, [esp+0x08]\n");
+    __asm("mov ebx, _g_categformat\n");
+    __asm("mov [eax], ebx\n");
+
+    __asm("print_regular_artist:\n");
+    __asm("pop ebx\n");
+    __asm("pop eax\n");
+    real_artist_printf();
+}
+
+static bool patch_custom_highlight(const char *game_dll_fn) {
+    DWORD dllSize = 0;
+    char *data = getDllData(game_dll_fn, &dllSize);
+
+    //hook format string for song/genre name
+    {
+        int64_t pattern_offset = search(data, dllSize, "\x83\xC4\x08\x8B\x44\x24\x50\x50\x68", 9, 0);
+        if (pattern_offset == -1) {
+            LOG("popnhax: custom_highlight: cannot find song/genre print function\n");
+            return false;
+        }
+
+        uint64_t patch_addr = (int64_t)data + pattern_offset - 0x07;
+
+        MH_CreateHook((LPVOID)patch_addr, (LPVOID)hook_song_printf,
+                     (void **)&real_song_printf);
+    }
+
+    //hook format string for artist 
+    {
+        int64_t pattern_offset = search(data, dllSize, "\x83\xC4\x08\x33\xFF\x8B\x43\x0C\x8B\x70\x04\x83\xC0\x04", 14, 0);
+        if (pattern_offset == -1) {
+            LOG("popnhax: custom_highlight: cannot find artist print function\n");
+            return false;
+        }
+
+        uint64_t patch_addr = (int64_t)data + pattern_offset - 0x07;
+
+        MH_CreateHook((LPVOID)patch_addr, (LPVOID)hook_artist_printf,
+                     (void **)&real_artist_printf);
+    }
+
+return true;
+}
+
 static bool patch_custom_categ(const char *game_dll_fn) {
 
     DWORD dllSize = 0;
@@ -608,5 +693,7 @@ bool patch_custom_categs(const char *dllFilename, uint8_t mode, uint16_t min, ui
         load_databases();
     }
 
+    patch_custom_highlight(dllFilename);
+
     return patch_custom_categ(dllFilename);
 }
\ No newline at end of file
diff --git a/popnhax/dllmain.cc b/popnhax/dllmain.cc
index 553ca50..ebf556c 100644
--- a/popnhax/dllmain.cc
+++ b/popnhax/dllmain.cc
@@ -32,7 +32,7 @@
 
 #include "SearchFile.h"
 
-#define PROGRAM_VERSION "1.11.beta1"
+#define PROGRAM_VERSION "1.11.beta2"
 
 const char *g_game_dll_fn = NULL;
 const char *g_config_fn   = NULL;