aboutsummaryrefslogtreecommitdiff
path: root/runtime/hook_manager.h
blob: dff44bf40fb7d399e03faafa949c5e7e9b5b38df (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
#ifndef HOOK_MANAGER_H
#define HOOK_MANAGER_H

class HookedAPI
{
public:
	HookedAPI();
	bool Hook(void *api, void *handler, void **result);
	void Unhook();
	void *old_handler() const { return old_handler_; }
private:
	static size_t page_size_;
	size_t size_;
	void *api_;
	void *old_handler_;
	uint32_t crc_;
};

class HookManager
{
public:
	HookManager();
	void Begin();
	void End();
	void *HookAPI(HMODULE dll, const char *api_name, void *handler, bool show_error = true, void **result = NULL);
	bool UnhookAPI(void * handler);
private:
	HookManager(const HookManager &src);
	HookManager & operator = (const HookManager &);// block the assignment operator
	void GetThreads();
	void SuspendThreads();
	void ResumeThreads();
	void FreeThreads();

	vector<HookedAPI *> api_list_;
	vector<HANDLE> thread_list_;
	size_t update_count_;
};

#endif