summaryrefslogtreecommitdiffhomepage
path: root/test/hook.test.c
blob: 7cbfd15d935de4d48c3f333b007651c1657bda40 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
/* This file is dedicated to the public domain. */

{.desc = "inline function hooking"};

#ifdef _WIN32

#include "../src/x86.c"
#include "../src/hook.c"

#include <stdarg.h>
#include <stdio.h>
#include <string.h>

// stubs
void con_warn(const char *msg, ...) {
	va_list l;
	va_start(l, msg);
	vfprintf(stderr, msg, l);
	va_end(l);
}

__attribute__((noinline))
static int some_function(int a, int b) { return a + b; }
static int (*orig_some_function)(int, int);
static int some_hook(int a, int b) {
	return orig_some_function(a, b) + 5;
}
__attribute__((noinline))
static int other_function(int a, int b) { return a - b; }
static int (*orig_other_function)(int, int);
static int other_hook(int a, int b) {
	return orig_other_function(a, b) + 5;
}

TEST("Inline hooks should be able to wrap the original function") {
	if (!hook_init()) return false;
	orig_some_function = hook_inline(&some_function, &some_hook);
	if (!orig_some_function) return false;
	return some_function(5, 5) == 15;
}

TEST("Inline hooks should be removable again") {
	if (!hook_init()) return false;
	orig_some_function = hook_inline(&some_function, &some_hook);
	if (!orig_some_function) return false;
	unhook_inline(orig_some_function);
	return some_function(5, 5) == 10;
}

TEST("Multiple functions should be able to be inline hooked at once") {
	if (!hook_init()) return false;
	orig_some_function = hook_inline(&some_function, &some_hook);
	if (!orig_some_function) return false;

	orig_other_function = hook_inline(&other_function, &other_hook);
	if (!orig_other_function) return false;

	return other_function(5, 5) == 5;
}

#endif

// vi: sw=4 ts=4 noet tw=80 cc=80