Bemanitools v5.26 release

src/main/hooklib/app.c

@@ -0,0 +1,79 @@
+#include <windows.h>
+
+#include "hook/table.h"
+
+#include "hooklib/app.h"
+
+#include "imports/avs.h"
+#include "imports/eapki.h"
+
+#include "util/log.h"
+#include "util/str.h"
+
+static dll_entry_init_t hook_dll_entry_init;
+static dll_entry_main_t hook_dll_entry_main;
+static dll_entry_init_t next_dll_entry_init;
+static dll_entry_main_t next_dll_entry_main;
+static void *(STDCALL *next_GetProcAddress)(HMODULE mod, const char *sym);
+
+static void * STDCALL my_GetProcAddress(HMODULE mod, const char *sym);
+
+static const struct hook_symbol mod_hooks[] = {
+    {
+        .name   = "GetProcAddress",
+        .patch  = my_GetProcAddress,
+        .link   = (void **) &next_GetProcAddress,
+    },
+};
+
+static void * STDCALL my_GetProcAddress(HMODULE mod, const char *sym)
+{
+    void *next;
+
+    next = next_GetProcAddress(mod, sym);
+
+    if (next == NULL || ((intptr_t) sym) <= UINT16_MAX) {
+        return next;
+    }
+
+    if (str_eq(sym, "dll_entry_init")) {
+        next_dll_entry_init = next;
+
+        if (hook_dll_entry_init != NULL) {
+            return hook_dll_entry_init;
+        } else {
+            return next;
+        }
+    } else if (str_eq(sym, "dll_entry_main")) {
+        next_dll_entry_main = next;
+
+        if (hook_dll_entry_main != NULL) {
+            return hook_dll_entry_main;
+        } else {
+            return next;
+        }
+    } else {
+        return next;
+    }
+}
+
+void app_hook_init(dll_entry_init_t init, dll_entry_main_t main_)
+{
+    hook_dll_entry_init = init;
+    hook_dll_entry_main = main_;
+    hook_table_apply(NULL, "kernel32.dll", mod_hooks, lengthof(mod_hooks));
+}
+
+bool app_hook_invoke_init(char *sidcode, struct property_node *config)
+{
+    log_assert(sidcode != NULL);
+    log_assert(config != NULL);
+
+    return next_dll_entry_init(sidcode, config);
+}
+
+bool app_hook_invoke_main(void)
+{
+    return next_dll_entry_main();
+}
+