BLShadowLib/blv20intrn.h at master · vitawrap/BLShadowLib · 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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#pragma once

#ifndef _BLV20INTRN_H
#define _BLV20INTRN_H

#include "framework.h"

// BL v20 DLL utility toolkit by vitawrap

namespace BL {
	//extern DWORD imageBase;

	//VOID init(HMODULE);

	template <typename T>
	SIZE_T write(DWORD, T const*);

	//LPVOID alloc(SIZE_T);
	//bool free(LPVOID);
}

template<typename T> SIZE_T BL::write(DWORD address, T const* data) {

	DWORD oldProtection;
	VirtualProtect((LPVOID)address, sizeof(T), PAGE_EXECUTE_READWRITE, &oldProtection);

	//SIZE_T bwritten = 0;
	//WriteProcessMemory(GetCurrentProcess(), (LPVOID)address, (LPCVOID)data, sizeof(T), &bwritten);
	CopyMemory((LPVOID)address, (LPCVOID)data, sizeof(T));

	VirtualProtect((LPVOID)address, sizeof(T), oldProtection, &oldProtection);

	return 0; //bwritten;
}

// Macros
#define BL_DECLARE_NAMESPACE(name) namespace name // Used to declare namespaced functions
#define BL_DECLARE_FUNCTION(ret, name, args) typedef ret (__cdecl *name##Proc) args; extern name##Proc name;
#define BL_DECLARE_METHOD(ret, name, args) typedef ret (__thiscall *name##Proc) args; extern name##Proc name;
#define BL_DEFINE_FUNCTION(name, addr) name##Proc name = (name##Proc)(/*BL::imageBase +*/ (DWORD)(addr))
#define BL_DEFINE_NS_FUNCTION(ns, name, addr) ns::name##Proc ns::name = (ns::name##Proc)(/*BL::imageBase +*/ (DWORD)(addr))

#define TypeS32			4
#define TypeBool		6
#define TypeF32			8
#define TypeString		10
#define TypeCaseString	11

// Dummy object typedefs
typedef LPVOID LPSimObject;
typedef LPVOID LPConsoleConstructor;

// Console constructor callback typedefs
typedef LPCSTR(*StringCallback)(LPSimObject obj, INT32 argc, LPCSTR argv[]);
typedef INT32(*IntCallback)(LPSimObject obj, INT32 argc, LPCSTR argv[]);
typedef FLOAT(*FloatCallback)(LPSimObject obj, INT32 argc, LPCSTR argv[]);
typedef VOID(*VoidCallback)(LPSimObject obj, INT32 argc, LPCSTR argv[]);
typedef bool    (*BoolCallback)(LPSimObject obj, INT32 argc, LPCSTR argv[]);

// Known BL functions
BL_DECLARE_NAMESPACE(Con) {
	BL_DECLARE_FUNCTION(VOID, setVariable, (LPCSTR name, LPCSTR value));
	BL_DECLARE_FUNCTION(BOOL, addVariable, (LPCSTR name, INT32 t, LPVOID dp));
	BL_DECLARE_FUNCTION(LPCSTR, evaluate, (LPCSTR string, BOOL echo, LPCSTR fileName));
	BL_DECLARE_FUNCTION(VOID, printf, (LPCSTR fmt, ...));
	BL_DECLARE_FUNCTION(VOID, errorf, (LPCSTR fmt, ...));

	BL_DECLARE_FUNCTION(VOID, addCommandS, (LPCSTR className, LPCSTR funcName, StringCallback	sfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs));
	BL_DECLARE_FUNCTION(VOID, addCommandI, (LPCSTR className, LPCSTR funcName, IntCallback		ifunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs));
	BL_DECLARE_FUNCTION(VOID, addCommandF, (LPCSTR className, LPCSTR funcName, FloatCallback	ffunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs));
	BL_DECLARE_FUNCTION(VOID, addCommandV, (LPCSTR className, LPCSTR funcName, VoidCallback		vfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs));
	BL_DECLARE_FUNCTION(VOID, addCommandB, (LPCSTR className, LPCSTR funcName, BoolCallback		bfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs));

	VOID addCommand(LPCSTR className, LPCSTR funcName, StringCallback	sfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
	VOID addCommand(LPCSTR className, LPCSTR funcName, IntCallback		ifunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
	VOID addCommand(LPCSTR className, LPCSTR funcName, FloatCallback	ffunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
	VOID addCommand(LPCSTR className, LPCSTR funcName, VoidCallback	    vfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
	VOID addCommand(LPCSTR className, LPCSTR funcName, BoolCallback	    bfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
}

// Console constructor size
#define SIZEOF_CCONSTRUCTOR 0x30

// Dummy class to replicate static function construction behavior
// and allow for TGE-like console function macros.
class ConsoleConstructor {
	char dummy[0x30];
public:
	ConsoleConstructor(LPCSTR className, LPCSTR funcName, StringCallback	sfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
	ConsoleConstructor(LPCSTR className, LPCSTR funcName, IntCallback		ifunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
	ConsoleConstructor(LPCSTR className, LPCSTR funcName, FloatCallback		ffunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
	ConsoleConstructor(LPCSTR className, LPCSTR funcName, VoidCallback	    vfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
	ConsoleConstructor(LPCSTR className, LPCSTR funcName, BoolCallback	    bfunc, LPCSTR usage, INT32 minArgs, INT32 maxArgs);
};

BL_DECLARE_NAMESPACE(Sim) {
	BL_DECLARE_FUNCTION(LPSimObject, findObject, (LPCSTR name));
}

// Yes, I really went through all this just to provide you
// with the authentic TGE ConsoleFunction macro experience
#define ConsoleFunction(name, returnType, minArgs, maxArgs, usage1)							\
	static returnType c##name(LPSimObject, INT32, LPCSTR*argv);								\
	static ConsoleConstructor g##name##obj(NULL,#name,c##name,usage1,minArgs,maxArgs);	\
	static returnType c##name(LPSimObject, INT32 argc, LPCSTR*argv)

#endif //_BLV20INTRN_H