Initial commit for GitHub migration based on spice2x-25-03-03

2026-08-01 22:30:42 -07:00 · 2025-03-10 19:16:49 -07:00
parent 94af8d32c9
commit ed8c72852f
1008 changed files with 296815 additions and 0 deletions
@@ -0,0 +1,125 @@
+#include "libraryhook.h"
+
+#include "external/robin_hood.h"
+#include "util/detour.h"
+#include "util/logging.h"
+#include "util/utils.h"
+
+static bool LHOOK_ENABLED = false;
+static robin_hood::unordered_map<std::string, HMODULE> LIBRARIES_A;
+static robin_hood::unordered_map<std::wstring, HMODULE> LIBRARIES_W;
+static robin_hood::unordered_map<std::string, FARPROC> PROCS;
+
+static decltype(LoadLibraryA) *LoadLibraryA_orig = nullptr;
+static decltype(LoadLibraryW) *LoadLibraryW_orig = nullptr;
+static decltype(GetModuleHandleA) *GetModuleHandleA_orig = nullptr;
+static decltype(GetModuleHandleW) *GetModuleHandleW_orig = nullptr;
+static decltype(GetProcAddress) *GetProcAddress_orig = nullptr;
+
+static HMODULE WINAPI LoadLibraryA_hook(LPCTSTR lpFileName) {
+
+    // check hooks
+    if (lpFileName) {
+        auto module = LIBRARIES_A.find(lpFileName);
+        if (module != LIBRARIES_A.end()) {
+            return module->second;
+        }
+    }
+
+    // fallback
+    return LoadLibraryA_orig(lpFileName);
+}
+
+static HMODULE WINAPI LoadLibraryW_hook(LPCWSTR lpFileName) {
+
+    // check hooks
+    if (lpFileName) {
+        auto module = LIBRARIES_W.find(lpFileName);
+        if (module != LIBRARIES_W.end()) {
+            return module->second;
+        }
+    }
+
+    // fallback
+    return LoadLibraryW_orig(lpFileName);
+}
+
+static HMODULE WINAPI GetModuleHandleA_hook(LPCSTR lpModuleName) {
+
+    // check hooks
+    if (lpModuleName) {
+        auto module = LIBRARIES_A.find(lpModuleName);
+        if (module != LIBRARIES_A.end()) {
+            return module->second;
+        }
+    }
+
+    // fallback
+    return GetModuleHandleA_orig(lpModuleName);
+}
+
+static HMODULE WINAPI GetModuleHandleW_hook(LPCWSTR lpModuleName) {
+
+    // check hooks
+    if (lpModuleName) {
+        auto module = LIBRARIES_W.find(lpModuleName);
+        if (module != LIBRARIES_W.end()) {
+            return module->second;
+        }
+    }
+
+    // fallback
+    return GetModuleHandleW_orig(lpModuleName);
+}
+
+static FARPROC WINAPI GetProcAddress_hook(HMODULE hModule, LPCSTR lpProcName) {
+
+    // check for ordinal
+    if (reinterpret_cast<uintptr_t>(lpProcName) <= UINT16_MAX) {
+
+        // fallback
+        return GetProcAddress_orig(hModule, lpProcName);
+    }
+
+    // check hooks
+    if (lpProcName) {
+        auto proc = PROCS.find(lpProcName);
+        if (proc != PROCS.end()) {
+            return proc->second;
+        }
+    }
+
+    // fallback
+    return GetProcAddress_orig(hModule, lpProcName);
+}
+
+void libraryhook_enable(HMODULE module) {
+    log_info("libraryhook", "LibraryHook Attach");
+
+    if (LHOOK_ENABLED) {
+        return;
+    }
+
+    // detour
+    detour::trampoline_try("kernel32.dll", "LoadLibraryA", LoadLibraryA_hook, &LoadLibraryA_orig);
+    detour::trampoline_try("kernel32.dll", "LoadLibraryW", LoadLibraryW_hook, &LoadLibraryW_orig);
+    detour::trampoline_try("kernel32.dll", "GetModuleHandleA", GetModuleHandleA_hook, &GetModuleHandleA_orig);
+    detour::trampoline_try("kernel32.dll", "GetModuleHandleW", GetModuleHandleW_hook, &GetModuleHandleW_orig);
+    detour::trampoline_try("kernel32.dll", "GetProcAddress", GetProcAddress_hook, &GetProcAddress_orig);
+
+    // set enabled
+    LHOOK_ENABLED = true;
+}
+
+void libraryhook_hook_library(std::string library_name, HMODULE library_address) {
+
+    // add library to list
+    LIBRARIES_W.insert_or_assign(s2ws(library_name), library_address);
+    LIBRARIES_A.insert_or_assign(std::move(library_name), library_address);
+}
+
+void libraryhook_hook_proc(std::string proc_name, FARPROC proc_address) {
+
+    // add proc to list
+    PROCS.insert_or_assign(std::move(proc_name), proc_address);
+}