ThePerfectMapper/pm.h at master · Th3Spl/ThePerfectMapper · GitHub

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
/* Note: you must use extern if you declare it in a different .h */
#pragma once


/* C++ helper ( otherwise name will be decorated ) */
#if defined(__cplusplus)
#define cpp_fixup_s extern "C" {
#define cpp_fixup_e }
#else
#define cpp_fixup_s
#define cpp_fixup_e
#endif


/* includes a simple iniitalization check */
#define pm_iat_check( check_name ) \
	cpp_fixup_s \
	__pragma( comment( linker, "/EXPORT:#PM#@check" "=" #check_name ",DATA" ) ) \
	char check_name; \
	cpp_fixup_e

/* Adds a PDB IAT in the EAT for a function ( needs an existing typedef ) */
#define pm_iat(func_name, _typedef, _src_module, _src_name) \
	cpp_fixup_s \
    __pragma(comment(linker, "/EXPORT:#PM#" #func_name "#" _src_module "#" _src_name "=" #func_name ",DATA")) \
	_typedef func_name; \
	cpp_fixup_e

/* Adds a PDB IAT in the EAT for a function ( does not need an existing typedef ) */
#define pm_iat_ex( func_name, ret, call_conv, type_name, args, _src_module, _src_name) \
	cpp_fixup_s \
	__pragma(comment(linker, "/EXPORT:#PM#" #func_name "#" _src_module "#" _src_name "=" #func_name ",DATA")) \
	typedef ret( call_conv* type_name) args; \
	type_name func_name; \
	cpp_fixup_e

/* Adds a PDB IAT in the EAT for an offset within a structure (ex: _EPROCESS->DirectoryTableBase ) */
#define pm_iat_offset( _name, _src_module, _src_struct, _src_comp_name ) \
	cpp_fixup_s \
	__pragma( comment( linker, "/EXPORT:#PM#" #_name "#@offset#" _src_module "#" _src_struct "#" _src_comp_name "=" #_name ",DATA" )) \
	unsigned long long _name; \
	cpp_fixup_e

/* Adds a PDB IAT in the EAT for an offset within a structure with a specified type for the offset */
#define pm_iat_offset2( _name, _type, _src_module, _src_struct, _src_comp_name ) \
	cpp_fixup_s \
	__pragma( comment( linker, "/EXPORT:#PM#" #_name "#@offset#" _src_module "#" _src_struct "#" _src_comp_name "=" #_name ",DATA" )) \
	_type _name; \
	cpp_fixup_e

/* Adds a PDB IAT in the EAT for the size of a given structure */
#define pm_iat_ssize( _name, _src_module, _src_struct ) \
	cpp_fixup_s \
	__pragma( comment( linker, "/EXPORT:#PM#" #_name "#@size#" _src_module "#" _src_struct "=" #_name ",DATA" ) ) \
	unsigned long long _name; \
	cpp_fixup_e