/* This file is dedicated to the public domain. */ {.desc = "inline function hooking"}; #ifdef _WIN32 #include "../src/x86.c" #include "../src/hook.c" #include "../src/os.c" #include #include #include // stub void con_warn(const char *msg, ...) { va_list l; va_start(l, msg); vfprintf(stderr, msg, l); va_end(l); } typedef int (*testfunc)(int, int); __attribute__((noinline)) static int func1(int a, int b) { return a + b; } static int (*orig_func1)(int, int); static int hook1(int a, int b) { return orig_func1(a, b) + 5; } __attribute__((noinline)) static int func2(int a, int b) { return a - b; } static int (*orig_func2)(int, int); static int hook2(int a, int b) { return orig_func2(a, b) + 5; } TEST("Inline hooks should be able to wrap the original function") { if (!hook_init()) return false; orig_func1 = (testfunc)hook_inline((void *)&func1, (void *)&hook1); if (!orig_func1) return false; return func1(5, 5) == 15; } TEST("Inline hooks should be removable again") { if (!hook_init()) return false; orig_func1 = (testfunc)hook_inline((void *)&func1, (void *)&hook1); if (!orig_func1) return false; unhook_inline((void *)orig_func1); return func1(5, 5) == 10; } TEST("Multiple functions should be able to be inline-hooked at once") { if (!hook_init()) return false; orig_func1 = (testfunc)hook_inline((void *)&func1, (void *)&hook1); if (!orig_func1) return false; orig_func2 = (testfunc)hook_inline((void *)&func2, (void *)&hook2); if (!orig_func2) return false; return func2(5, 5) == 5; } #endif // vi: sw=4 ts=4 noet tw=80 cc=80