Auios/BlockLua
1
0
Fork 0
forked from redo/BlockLua
Code Pull Requests Activity

.clang-format updates

Browse Source
This commit is contained in:
Connor O'Connor
2025-10-06 17:00:12 -04:00
parent 5718ba8e6b
commit ed5c254480
15 changed files with 617 additions and 650 deletions
Download Patch File Download Diff File Expand all files Collapse all files

269
inc/tsfuncs/BlFuncs.cpp
View File

@@ -38,52 +38,53 @@ BlFunctionDefIntern(tsf_BlCon__getReturnBuffer);
// C->TS Args
char *tsf_GetIntArg(signed int value) {
char *ret = tsf_BlStringStack__getArgBuffer(16);
char* tsf_GetIntArg(signed int value) {
char* ret = tsf_BlStringStack__getArgBuffer(16);
snprintf(ret, 16, "%d", value);
return ret;
}
char *tsf_GetFloatArg(float value) {
char *ret = tsf_BlStringStack__getArgBuffer(32);
char* tsf_GetFloatArg(float value) {
char* ret = tsf_BlStringStack__getArgBuffer(32);
snprintf(ret, 32, "%g", value);
return ret;
}
char *tsf_GetStringArg(char *value) {
char* tsf_GetStringArg(char* value) {
int len = strlen(value) + 1;
char *ret = tsf_BlStringStack__getArgBuffer(len);
char* ret = tsf_BlStringStack__getArgBuffer(len);
memcpy(ret, value, len);
return ret;
}
char *tsf_GetThisArg(ADDR obj) {
return tsf_GetIntArg(*(signed int *)(obj + 32));
}
char* tsf_GetThisArg(ADDR obj) { return tsf_GetIntArg(*(signed int*)(obj + 32)); }
// Eval
const char *tsf_Eval(const char *code) {
const char *argv[] = {nullptr, code};
const char* tsf_Eval(const char* code) {
const char* argv[] = {nullptr, code};
return tsf_BlCon__evaluate(0, 2, argv);
}
const char *tsf_Evalf(const char *fmt, ...) {
const char* tsf_Evalf(const char* fmt, ...) {
va_list args;
char code[4096];
va_start(args, fmt);
vsnprintf(code, 4096, fmt, args);
va_end(args);
return tsf_Eval((const char *)code);
return tsf_Eval((const char*)code);
}
// Objects
ADDR tsf_FindObject(unsigned int id) {
ADDR obj = *(ADDR *)(*(ADDR *)(tsf_gIdDictionary) + 4 * (id & 0xFFF));
ADDR obj = *(ADDR*)(*(ADDR*)(tsf_gIdDictionary) + 4 * (id & 0xFFF));
if (!obj)
return 0;
while (obj && *(unsigned int *)(obj + 32) != id) {
obj = *(ADDR *)(obj + 16);
while (obj && *(unsigned int*)(obj + 32) != id) {
obj = *(ADDR*)(obj + 16);
if (!obj)
return 0;
}
@@ -91,12 +92,10 @@ ADDR tsf_FindObject(unsigned int id) {
return obj;
}
ADDR tsf_FindObject(const char *name) {
return (ADDR)tsf_BlSim__findObject_name(name);
}
ADDR tsf_FindObject(const char* name) { return (ADDR)tsf_BlSim__findObject_name(name); }
ADDR tsf_LookupNamespace(const char *ns, const char *package) {
const char *ste_package;
ADDR tsf_LookupNamespace(const char* ns, const char* package) {
const char* ste_package;
if (package) {
ste_package = tsf_BlStringTable__insert(package, 0);
} else {
@@ -104,21 +103,19 @@ ADDR tsf_LookupNamespace(const char *ns, const char *package) {
}
if (ns) {
const char *ste_namespace = tsf_BlStringTable__insert(ns, 0);
const char* ste_namespace = tsf_BlStringTable__insert(ns, 0);
return tsf_BlNamespace__find(ste_namespace, ste_package);
} else {
return tsf_BlNamespace__find(nullptr, ste_package);
}
}
ADDR tsf_LookupNamespace(const char *ns) {
return tsf_LookupNamespace(ns, nullptr);
}
ADDR tsf_LookupNamespace(const char* ns) { return tsf_LookupNamespace(ns, nullptr); }
// Object Fields
const char *tsf_GetDataField(ADDR simObject, const char *slotName,
const char *array) {
const char *ste_slotName;
const char* tsf_GetDataField(ADDR simObject, const char* slotName, const char* array) {
const char* ste_slotName;
if (slotName) {
ste_slotName = tsf_BlStringTable__insert(slotName, 0);
} else {
@@ -128,9 +125,8 @@ const char *tsf_GetDataField(ADDR simObject, const char *slotName,
return tsf_BlSimObject__getDataField(simObject, ste_slotName, array);
}
void tsf_SetDataField(ADDR simObject, const char *slotName, const char *array,
const char *value) {
const char *ste_slotName;
void tsf_SetDataField(ADDR simObject, const char* slotName, const char* array, const char* value) {
const char* ste_slotName;
if (slotName) {
ste_slotName = tsf_BlStringTable__insert(slotName, 0);
} else {
@@ -142,83 +138,103 @@ void tsf_SetDataField(ADDR simObject, const char *slotName, const char *array,
// TS Global Variables
const char *tsf_GetVar(const char *name) {
return tsf_BlCon__getVariable(name);
const char* tsf_GetVar(const char* name) { return tsf_BlCon__getVariable(name); }
void tsf_AddVarInternal(const char* name, signed int varType, void* data) {
tsf_BlDictionary__addVariable((ADDR*)tsf_gEvalState_globalVars, name, varType, data);
}
void tsf_AddVarInternal(const char *name, signed int varType, void *data) {
tsf_BlDictionary__addVariable((ADDR *)tsf_gEvalState_globalVars, name,
varType, data);
}
void tsf_AddVar(const char* name, const char** data) { tsf_AddVarInternal(name, 10, data); }
void tsf_AddVar(const char *name, const char **data) {
tsf_AddVarInternal(name, 10, data);
}
void tsf_AddVar(const char *name, signed int *data) {
tsf_AddVarInternal(name, 4, data);
}
void tsf_AddVar(const char *name, float *data) {
tsf_AddVarInternal(name, 8, data);
}
void tsf_AddVar(const char *name, bool *data) {
tsf_AddVarInternal(name, 6, data);
}
void tsf_AddVar(const char* name, signed int* data) { tsf_AddVarInternal(name, 4, data); }
void tsf_AddVar(const char* name, float* data) { tsf_AddVarInternal(name, 8, data); }
void tsf_AddVar(const char* name, bool* data) { tsf_AddVarInternal(name, 6, data); }
// TS->C Functions
ADDR tsf_AddConsoleFuncInternal(const char *pname, const char *cname,
const char *fname, signed int cbtype,
const char *usage, signed int mina,
signed int maxa) {
const char *ste_fname = tsf_BlStringTable__insert(fname, 0);
ADDR tsf_AddConsoleFuncInternal(
const char* pname,
const char* cname,
const char* fname,
signed int cbtype,
const char* usage,
signed int mina,
signed int maxa) {
const char* ste_fname = tsf_BlStringTable__insert(fname, 0);
ADDR ns = tsf_LookupNamespace(cname, pname);
ADDR ent = tsf_BlNamespace__createLocalEntry(ns, ste_fname);
*(signed int *)tsf_mCacheSequence += 1;
tsf_BlDataChunker__freeBlocks(*(ADDR *)tsf_mCacheAllocator);
*(signed int*)tsf_mCacheSequence += 1;
tsf_BlDataChunker__freeBlocks(*(ADDR*)tsf_mCacheAllocator);
*(const char **)(ent + 24) = usage;
*(signed int *)(ent + 16) = mina;
*(signed int *)(ent + 20) = maxa;
*(signed int *)(ent + 12) = cbtype;
*(const char**)(ent + 24) = usage;
*(signed int*)(ent + 16) = mina;
*(signed int*)(ent + 20) = maxa;
*(signed int*)(ent + 12) = cbtype;
return ent;
}
void tsf_AddConsoleFunc(const char *pname, const char *cname, const char *fname,
tsf_StringCallback sc, const char *usage,
signed int mina, signed int maxa) {
ADDR ent =
tsf_AddConsoleFuncInternal(pname, cname, fname, 1, usage, mina, maxa);
*(tsf_StringCallback *)(ent + 40) = sc;
void tsf_AddConsoleFunc(
const char* pname,
const char* cname,
const char* fname,
tsf_StringCallback sc,
const char* usage,
signed int mina,
signed int maxa) {
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 1, usage, mina, maxa);
*(tsf_StringCallback*)(ent + 40) = sc;
}
void tsf_AddConsoleFunc(const char *pname, const char *cname, const char *fname,
tsf_IntCallback ic, const char *usage, signed int mina,
signed int maxa) {
ADDR ent =
tsf_AddConsoleFuncInternal(pname, cname, fname, 2, usage, mina, maxa);
*(tsf_IntCallback *)(ent + 40) = ic;
void tsf_AddConsoleFunc(
const char* pname,
const char* cname,
const char* fname,
tsf_IntCallback ic,
const char* usage,
signed int mina,
signed int maxa) {
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 2, usage, mina, maxa);
*(tsf_IntCallback*)(ent + 40) = ic;
}
void tsf_AddConsoleFunc(const char *pname, const char *cname, const char *fname,
tsf_FloatCallback fc, const char *usage,
signed int mina, signed int maxa) {
ADDR ent =
tsf_AddConsoleFuncInternal(pname, cname, fname, 3, usage, mina, maxa);
*(tsf_FloatCallback *)(ent + 40) = fc;
void tsf_AddConsoleFunc(
const char* pname,
const char* cname,
const char* fname,
tsf_FloatCallback fc,
const char* usage,
signed int mina,
signed int maxa) {
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 3, usage, mina, maxa);
*(tsf_FloatCallback*)(ent + 40) = fc;
}
void tsf_AddConsoleFunc(const char *pname, const char *cname, const char *fname,
tsf_VoidCallback vc, const char *usage, signed int mina,
signed int maxa) {
ADDR ent =
tsf_AddConsoleFuncInternal(pname, cname, fname, 4, usage, mina, maxa);
*(tsf_VoidCallback *)(ent + 40) = vc;
void tsf_AddConsoleFunc(
const char* pname,
const char* cname,
const char* fname,
tsf_VoidCallback vc,
const char* usage,
signed int mina,
signed int maxa) {
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 4, usage, mina, maxa);
*(tsf_VoidCallback*)(ent + 40) = vc;
}
void tsf_AddConsoleFunc(const char *pname, const char *cname, const char *fname,
tsf_BoolCallback bc, const char *usage, signed int mina,
signed int maxa) {
ADDR ent =
tsf_AddConsoleFuncInternal(pname, cname, fname, 5, usage, mina, maxa);
*(tsf_BoolCallback *)(ent + 40) = bc;
void tsf_AddConsoleFunc(
const char* pname,
const char* cname,
const char* fname,
tsf_BoolCallback bc,
const char* usage,
signed int mina,
signed int maxa) {
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 5, usage, mina, maxa);
*(tsf_BoolCallback*)(ent + 40) = bc;
}
// Initialization
@@ -233,49 +249,54 @@ bool tsf_InitInternal() {
tsf_BlNamespace__createLocalEntry,
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 83 EC 08 53 56 57 A1 ? ? ? ? "
"33 C5 50 8D 45 F4 64 A3 ? ? ? ? 89 4D F0");
BlScanFunctionText(tsf_BlDataChunker__freeBlocks,
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 51 53 56 57 "
"A1 ? ? ? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B D9 8B 33");
BlScanFunctionText(tsf_BlCon__evaluate,
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 56 57 A1 ? ? "
"? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B 75 10");
BlScanFunctionText(tsf_BlCon__executef,
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 53 55 56 "
"8B B4 24 ? ? ? ? 33 C9");
BlScanFunctionText(tsf_BlCon__executefSimObj,
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 53 56 8B "
"B4 24 ? ? ? ? 33 C9");
BlScanFunctionText(tsf_BlCon__getVariable,
"53 56 8B F1 57 85 F6 0F 84 ? ? ? ?");
BlScanFunctionText(
tsf_BlDataChunker__freeBlocks,
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 51 53 56 57 "
"A1 ? ? ? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B D9 8B 33");
BlScanFunctionText(
tsf_BlCon__evaluate,
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 56 57 A1 ? ? "
"? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B 75 10");
BlScanFunctionText(
tsf_BlCon__executef,
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 53 55 56 "
"8B B4 24 ? ? ? ? 33 C9");
BlScanFunctionText(
tsf_BlCon__executefSimObj,
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 53 56 8B "
"B4 24 ? ? ? ? 33 C9");
BlScanFunctionText(tsf_BlCon__getVariable, "53 56 8B F1 57 85 F6 0F 84 ? ? ? ?");
BlScanFunctionText(tsf_BlDictionary__addVariable, "8B 44 24 04 56 57 8B F9");
BlScanFunctionText(tsf_BlSim__findObject_name, "57 8B F9 8A 17");
BlScanFunctionText(tsf_BlStringStack__getArgBuffer,
"55 8B EC 83 E4 F8 8B 0D ? ? ? ? A1 ? ? ? ? 56 57 8B 7D "
"08 8D 14 01 03 D7 3B 15 ? ? ? ? 72 2C 8B 0D");
BlScanFunctionText(tsf_BlSimObject__getDataField,
"51 53 8B D9 55 56 8B 74 24 14");
BlScanFunctionText(tsf_BlSimObject__setDataField,
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 8B 84 24 "
"? ? ? ? 53 8B D9 89 44 24 04");
BlScanFunctionText(
tsf_BlStringStack__getArgBuffer,
"55 8B EC 83 E4 F8 8B 0D ? ? ? ? A1 ? ? ? ? 56 57 8B 7D "
"08 8D 14 01 03 D7 3B 15 ? ? ? ? 72 2C 8B 0D");
BlScanFunctionText(tsf_BlSimObject__getDataField, "51 53 8B D9 55 56 8B 74 24 14");
BlScanFunctionText(
tsf_BlSimObject__setDataField,
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 8B 84 24 "
"? ? ? ? 53 8B D9 89 44 24 04");
BlScanFunctionText(tsf_BlCon__getReturnBuffer, "81 F9 ? ? ? ? 76 2B");
ADDR BlScanText(tsf_mCacheSequenceLoc,
"FF 05 ? ? ? ? B9 ? ? ? ? 8B F8 E8 ? ? ? ? 8B 44 24 1C 89 47 "
"18 8B 44 24 14");
ADDR BlScanText(
tsf_mCacheSequenceLoc,
"FF 05 ? ? ? ? B9 ? ? ? ? 8B F8 E8 ? ? ? ? 8B 44 24 1C 89 47 "
"18 8B 44 24 14");
ADDR BlScanText(
tsf_mCacheAllocatorLoc,
"89 35 ? ? ? ? C7 06 ? ? ? ? A1 ? ? ? ? 68 ? ? ? ? C7 40 ? ? ? ? ? E8 ? "
"? ? ? 83 C4 04 8B 4D F4 64 89 0D ? ? ? ? 59 5E 8B E5 5D C3");
ADDR BlScanText(tsf_gIdDictionaryLoc,
"89 15 ? ? ? ? E8 ? ? ? ? 8B F0 89 75 F0");
ADDR BlScanText(tsf_gEvalState_globalVarsLoc,
"B9 ? ? ? ? E8 ? ? ? ? 68 ? ? ? ? 6A 0A 68 ? ? ? ? B9 ? ? ? "
"? E8 ? ? ? ? E8 ? ? ? ?");
ADDR BlScanText(tsf_gIdDictionaryLoc, "89 15 ? ? ? ? E8 ? ? ? ? 8B F0 89 75 F0");
ADDR BlScanText(
tsf_gEvalState_globalVarsLoc,
"B9 ? ? ? ? E8 ? ? ? ? 68 ? ? ? ? 6A 0A 68 ? ? ? ? B9 ? ? ? "
"? E8 ? ? ? ? E8 ? ? ? ?");
tsf_mCacheSequence = *(ADDR *)(tsf_mCacheSequenceLoc + 2);
tsf_mCacheAllocator = *(ADDR *)(tsf_mCacheAllocatorLoc + 2);
tsf_gIdDictionary = *(ADDR *)(tsf_gIdDictionaryLoc + 2);
tsf_gEvalState_globalVars = *(ADDR *)(tsf_gEvalState_globalVarsLoc + 1);
tsf_mCacheSequence = *(ADDR*)(tsf_mCacheSequenceLoc + 2);
tsf_mCacheAllocator = *(ADDR*)(tsf_mCacheAllocatorLoc + 2);
tsf_gIdDictionary = *(ADDR*)(tsf_gIdDictionaryLoc + 2);
tsf_gEvalState_globalVars = *(ADDR*)(tsf_gEvalState_globalVarsLoc + 1);
return true;
}

125
inc/tsfuncs/BlFuncs.hpp
View File

@@ -10,53 +10,44 @@
#include "BlHooks.hpp"
#endif
typedef const char *(*tsf_StringCallback)(ADDR, signed int, const char *[]);
typedef signed int (*tsf_IntCallback)(ADDR, signed int, const char *[]);
typedef float (*tsf_FloatCallback)(ADDR, signed int, const char *[]);
typedef void (*tsf_VoidCallback)(ADDR, signed int, const char *[]);
typedef bool (*tsf_BoolCallback)(ADDR, signed int, const char *[]);
typedef const char* (*tsf_StringCallback)(ADDR, signed int, const char*[]);
typedef signed int (*tsf_IntCallback)(ADDR, signed int, const char*[]);
typedef float (*tsf_FloatCallback)(ADDR, signed int, const char*[]);
typedef void (*tsf_VoidCallback)(ADDR, signed int, const char*[]);
typedef bool (*tsf_BoolCallback)(ADDR, signed int, const char*[]);
/* These functions are used for tsf_BlCon__executefSimObj.
They refer to a special buffer for the argument stack.
For tsf_BlCon__executef, you need to use your own buffers. */
char *tsf_GetIntArg(int);
char *tsf_GetFloatArg(float);
char *tsf_ScriptThis(ADDR);
char* tsf_GetIntArg(int);
char* tsf_GetFloatArg(float);
char* tsf_ScriptThis(ADDR);
const char *tsf_Eval(const char *);
const char *tsf_Evalf(const char *, ...);
const char* tsf_Eval(const char*);
const char* tsf_Evalf(const char*, ...);
ADDR tsf_FindObject(unsigned int);
ADDR tsf_FindObject(const char *);
ADDR tsf_FindObject(const char*);
ADDR tsf_LookupNamespace(const char *, const char *);
ADDR tsf_LookupNamespace(const char*, const char*);
const char *tsf_GetDataField(ADDR, const char *, const char *);
void tsf_SetDataField(ADDR, const char *, const char *, const char *);
const char* tsf_GetDataField(ADDR, const char*, const char*);
void tsf_SetDataField(ADDR, const char*, const char*, const char*);
const char *tsf_GetVar(const char *);
const char* tsf_GetVar(const char*);
void tsf_AddVarInternal(const char *, signed int, void *);
void tsf_AddVar(const char *, const char **);
void tsf_AddVar(const char *, signed int *);
void tsf_AddVar(const char *, float *);
void tsf_AddVar(const char *, bool *);
void tsf_AddVarInternal(const char*, signed int, void*);
void tsf_AddVar(const char*, const char**);
void tsf_AddVar(const char*, signed int*);
void tsf_AddVar(const char*, float*);
void tsf_AddVar(const char*, bool*);
ADDR tsf_AddConsoleFuncInternal(const char *, const char *, const char *,
signed int, const char *, signed int,
signed int);
void tsf_AddConsoleFunc(const char *, const char *, const char *,
tsf_StringCallback, const char *, signed int,
signed int);
void tsf_AddConsoleFunc(const char *, const char *, const char *,
tsf_IntCallback, const char *, signed int, signed int);
void tsf_AddConsoleFunc(const char *, const char *, const char *,
tsf_FloatCallback, const char *, signed int,
signed int);
void tsf_AddConsoleFunc(const char *, const char *, const char *,
tsf_VoidCallback, const char *, signed int, signed int);
void tsf_AddConsoleFunc(const char *, const char *, const char *,
tsf_BoolCallback, const char *, signed int, signed int);
ADDR tsf_AddConsoleFuncInternal(const char*, const char*, const char*, signed int, const char*, signed int, signed int);
void tsf_AddConsoleFunc(const char*, const char*, const char*, tsf_StringCallback, const char*, signed int, signed int);
void tsf_AddConsoleFunc(const char*, const char*, const char*, tsf_IntCallback, const char*, signed int, signed int);
void tsf_AddConsoleFunc(const char*, const char*, const char*, tsf_FloatCallback, const char*, signed int, signed int);
void tsf_AddConsoleFunc(const char*, const char*, const char*, tsf_VoidCallback, const char*, signed int, signed int);
void tsf_AddConsoleFunc(const char*, const char*, const char*, tsf_BoolCallback, const char*, signed int, signed int);
bool tsf_InitInternal();
@@ -65,32 +56,20 @@ extern ADDR tsf_mCacheAllocator;
extern ADDR tsf_gIdDictionary;
extern ADDR tsf_gEvalState_globalVars;
BlFunctionDefExtern(const char *, __stdcall, tsf_BlStringTable__insert,
const char *, bool);
BlFunctionDefExtern(ADDR, __fastcall, tsf_BlNamespace__find, const char *,
const char *);
BlFunctionDefExtern(ADDR, __thiscall, tsf_BlNamespace__createLocalEntry, ADDR,
const char *);
BlFunctionDefExtern(const char*, __stdcall, tsf_BlStringTable__insert, const char*, bool);
BlFunctionDefExtern(ADDR, __fastcall, tsf_BlNamespace__find, const char*, const char*);
BlFunctionDefExtern(ADDR, __thiscall, tsf_BlNamespace__createLocalEntry, ADDR, const char*);
BlFunctionDefExtern(void, __thiscall, tsf_BlDataChunker__freeBlocks, ADDR);
BlFunctionDefExtern(const char *, , tsf_BlCon__evaluate, ADDR, signed int,
const char **);
BlFunctionDefExtern(const char *, , tsf_BlCon__executef, signed int, ...);
BlFunctionDefExtern(const char *, , tsf_BlCon__executefSimObj, ADDR *,
signed int, ...);
BlFunctionDefExtern(const char *, __thiscall, tsf_BlCon__getVariable,
const char *);
BlFunctionDefExtern(void, __thiscall, tsf_BlDictionary__addVariable, ADDR *,
const char *, signed int, void *);
BlFunctionDefExtern(ADDR *, __thiscall, tsf_BlSim__findObject_name,
const char *);
BlFunctionDefExtern(char *, __stdcall, tsf_BlStringStack__getArgBuffer,
unsigned int);
BlFunctionDefExtern(const char *, __thiscall, tsf_BlSimObject__getDataField,
ADDR, const char *, const char *);
BlFunctionDefExtern(void, __thiscall, tsf_BlSimObject__setDataField, ADDR,
const char *, const char *, const char *);
BlFunctionDefExtern(char *, __fastcall, tsf_BlCon__getReturnBuffer,
unsigned int);
BlFunctionDefExtern(const char*, , tsf_BlCon__evaluate, ADDR, signed int, const char**);
BlFunctionDefExtern(const char*, , tsf_BlCon__executef, signed int, ...);
BlFunctionDefExtern(const char*, , tsf_BlCon__executefSimObj, ADDR*, signed int, ...);
BlFunctionDefExtern(const char*, __thiscall, tsf_BlCon__getVariable, const char*);
BlFunctionDefExtern(void, __thiscall, tsf_BlDictionary__addVariable, ADDR*, const char*, signed int, void*);
BlFunctionDefExtern(ADDR*, __thiscall, tsf_BlSim__findObject_name, const char*);
BlFunctionDefExtern(char*, __stdcall, tsf_BlStringStack__getArgBuffer, unsigned int);
BlFunctionDefExtern(const char*, __thiscall, tsf_BlSimObject__getDataField, ADDR, const char*, const char*);
BlFunctionDefExtern(void, __thiscall, tsf_BlSimObject__setDataField, ADDR, const char*, const char*, const char*);
BlFunctionDefExtern(char*, __fastcall, tsf_BlCon__getReturnBuffer, unsigned int);
// Function short names
@@ -100,7 +79,7 @@ BlFunctionDefExtern(char *, __fastcall, tsf_BlCon__getReturnBuffer,
#define BlIntArg tsf_GetIntArg
#define BlFloatArg tsf_GetFloatArg
#define BlThisArg tsf_GetThisArg
#define BlStringArg(x) tsf_GetStringArg((char *)x)
#define BlStringArg(x) tsf_GetStringArg((char*)x)
#define BlReturnBuffer tsf_BlCon__getReturnBuffer
#define BlObject tsf_FindObject
@@ -114,23 +93,21 @@ BlFunctionDefExtern(char *, __fastcall, tsf_BlCon__getReturnBuffer,
#define BlAddFunction tsf_AddConsoleFunc
#define __22ND_ARGUMENT(a1, a2, a3, a4, a5, a6, a7, a8, a9, a10, a11, a12, \
a13, a14, a15, a16, a17, a18, a19, a20, a21, a22, ...) \
#define __22ND_ARGUMENT( \
a1, a2, a3, a4, a5, a6, a7, a8, a9, a10, a11, a12, a13, a14, a15, a16, a17, a18, a19, a20, a21, a22, ...) \
a22
#define __NUM_LIST(...) \
__22ND_ARGUMENT(dummy, ##__VA_ARGS__, 20, 19, 18, 17, 16, 15, 14, 13, 12, \
11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0)
#define __NUM_LIST(...) \
__22ND_ARGUMENT(dummy, ##__VA_ARGS__, 20, 19, 18, 17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0)
#define BlCall(...) tsf_BlCon__executef(__NUM_LIST(__VA_ARGS__), __VA_ARGS__)
#define BlCallObj(obj, ...) \
tsf_BlCon__executefSimObj((ADDR *)obj, __NUM_LIST(__VA_ARGS__), __VA_ARGS__)
#define BlCallObj(obj, ...) tsf_BlCon__executefSimObj((ADDR*)obj, __NUM_LIST(__VA_ARGS__), __VA_ARGS__)
#define BlFuncsInit() \
if (!tsf_InitInternal()) { \
return false; \
#define BlFuncsInit() \
if (!tsf_InitInternal()) { \
return false; \
}
#define BlFuncsDeinit() \
if (!tsf_DeinitInternal()) { \
return false; \
#define BlFuncsDeinit() \
if (!tsf_DeinitInternal()) { \
return false; \
}
#endif

74
inc/tsfuncs/BlHooks.cpp
View File

@@ -27,14 +27,13 @@ void tsh_i_InitScanner() {
HMODULE module = GetModuleHandle(NULL);
if (module) {
MODULEINFO info;
GetModuleInformation(GetCurrentProcess(), module, &info,
sizeof(MODULEINFO));
GetModuleInformation(GetCurrentProcess(), module, &info, sizeof(MODULEINFO));
ImageBase = (ADDR)info.lpBaseOfDll;
ImageSize = info.SizeOfImage;
}
}
bool tsh_i_CompareData(BYTE *data, BYTE *pattern, char *mask) {
bool tsh_i_CompareData(BYTE* data, BYTE* pattern, char* mask) {
for (; *mask; ++data, ++pattern, ++mask) {
if (*mask == 'x' && *data != *pattern)
return false;
@@ -42,8 +41,7 @@ bool tsh_i_CompareData(BYTE *data, BYTE *pattern, char *mask) {
return (*mask) == 0;
}
ADDR tsh_i_FindPattern(ADDR imageBase, ADDR imageSize, BYTE *pattern,
char *mask) {
ADDR tsh_i_FindPattern(ADDR imageBase, ADDR imageSize, BYTE* pattern, char* mask) {
for (ADDR i = imageBase; i < imageBase + imageSize; i++) {
if (tsh_i_CompareData((PBYTE)i, pattern, mask)) {
return i;
@@ -53,10 +51,10 @@ ADDR tsh_i_FindPattern(ADDR imageBase, ADDR imageSize, BYTE *pattern,
}
// Convert a text-style pattern into code-style
void tsh_i_PatternTextToCode(char *text, char **opatt, char **omask) {
void tsh_i_PatternTextToCode(char* text, char** opatt, char** omask) {
unsigned int len = strlen(text);
char *patt = (char *)malloc(len);
char *mask = (char *)malloc(len);
char* patt = (char*)malloc(len);
char* mask = (char*)malloc(len);
int outidx = 0;
int val = 0;
@@ -93,15 +91,14 @@ void tsh_i_PatternTextToCode(char *text, char **opatt, char **omask) {
// Public functions for sig scanning
// Scan using code-style pattern
ADDR tsh_ScanCode(char *pattern, char *mask) {
return tsh_i_FindPattern(ImageBase, ImageSize - strlen(mask), (BYTE *)pattern,
mask);
ADDR tsh_ScanCode(char* pattern, char* mask) {
return tsh_i_FindPattern(ImageBase, ImageSize - strlen(mask), (BYTE*)pattern, mask);
}
// Scan using a text-style pattern
ADDR tsh_ScanText(char *text) {
char *patt;
char *mask;
ADDR tsh_ScanText(char* text) {
char* patt;
char* mask;
tsh_i_PatternTextToCode(text, &patt, &mask);
ADDR res = tsh_ScanCode(patt, mask);
@@ -119,54 +116,46 @@ ADDR tsh_ScanText(char *text) {
void tsh_DeprotectAddress(ADDR length, ADDR location) {
DWORD oldProtection;
VirtualProtect((void *)location, length, PAGE_EXECUTE_READWRITE,
&oldProtection);
VirtualProtect((void*)location, length, PAGE_EXECUTE_READWRITE, &oldProtection);
// tsh_DeprotectedAddresses[location] = {length, oldProtection};
}
// Patch a string of bytes by deprotecting and then overwriting
void tsh_PatchBytes(ADDR length, ADDR location, BYTE *repl) {
void tsh_PatchBytes(ADDR length, ADDR location, BYTE* repl) {
tsh_DeprotectAddress(length, location);
memcpy((void *)location, (void *)repl, (size_t)length);
memcpy((void*)location, (void*)repl, (size_t)length);
}
void tsh_PatchByte(ADDR location, BYTE value) {
tsh_PatchBytes(location, 1, &value);
}
void tsh_PatchByte(ADDR location, BYTE value) { tsh_PatchBytes(location, 1, &value); }
void tsh_ReplaceInt(ADDR addr, int rval) {
tsh_PatchBytes(4, addr, (BYTE *)(&rval));
}
void tsh_ReplaceInt(ADDR addr, int rval) { tsh_PatchBytes(4, addr, (BYTE*)(&rval)); }
int tsh_i_CallOffset(ADDR instr, ADDR func) { return func - (instr + 4); }
void tsh_i_ReplaceCall(ADDR instr, ADDR target) {
tsh_ReplaceInt(instr, tsh_i_CallOffset(instr, target));
}
void tsh_i_ReplaceCall(ADDR instr, ADDR target) { tsh_ReplaceInt(instr, tsh_i_CallOffset(instr, target)); }
void tsh_i_PatchCopy(ADDR dest, ADDR src, unsigned int len) {
for (unsigned int i = 0; i < len; i++) {
tsh_PatchByte(dest + i, *((BYTE *)(src + i)));
tsh_PatchByte(dest + i, *((BYTE*)(src + i)));
}
}
void tsh_HookFunction(ADDR victim, ADDR detour, BYTE *origbytes) {
memcpy(origbytes, (BYTE *)victim, 6); // save old data
void tsh_HookFunction(ADDR victim, ADDR detour, BYTE* origbytes) {
memcpy(origbytes, (BYTE*)victim, 6); // save old data
*(BYTE *)victim = 0xE9; // jmp rel32
*(ADDR *)(victim + 1) = (detour - (victim + 5)); // jump offset
*(BYTE *)(victim + 5) = 0xC3; // retn
*(BYTE*)victim = 0xE9; // jmp rel32
*(ADDR*)(victim + 1) = (detour - (victim + 5)); // jump offset
*(BYTE*)(victim + 5) = 0xC3; // retn
}
void tsh_UnhookFunction(ADDR victim, BYTE *origbytes) {
void tsh_UnhookFunction(ADDR victim, BYTE* origbytes) {
tsh_i_PatchCopy(victim, (ADDR)origbytes, 6); // restore old data
}
int tsh_PatchAllMatchesCode(ADDR len, char *patt, char *mask, char *replace,
bool debugprint) {
int tsh_PatchAllMatchesCode(ADDR len, char* patt, char* mask, char* replace, bool debugprint) {
int numpatched = 0;
for (ADDR i = ImageBase; i < ImageBase + ImageSize - len; i++) {
if (tsh_i_CompareData((BYTE *)i, (BYTE *)patt, mask)) {
if (tsh_i_CompareData((BYTE*)i, (BYTE*)patt, mask)) {
if (debugprint)
BlPrintf("RedoBlHooks: Patching call at %08x", i);
@@ -180,10 +169,9 @@ int tsh_PatchAllMatchesCode(ADDR len, char *patt, char *mask, char *replace,
return numpatched;
}
int tsh_PatchAllMatchesHex(ADDR len, char *text, char *replace,
bool debugprint) {
char *patt;
char *mask;
int tsh_PatchAllMatchesHex(ADDR len, char* text, char* replace, bool debugprint) {
char* patt;
char* mask;
tsh_i_PatternTextToCode(text, &patt, &mask);
int res = tsh_PatchAllMatchesCode(len, patt, mask, replace, debugprint);
@@ -199,9 +187,7 @@ int tsh_PatchAllMatchesHex(ADDR len, char *text, char *replace,
bool tsh_InitInternal() {
tsh_i_InitScanner();
BlScanFunctionText(
tsh_BlPrintf,
"8D 44 24 08 33 D2 50 FF 74 24 08 33 C9 E8 ? ? ? ? 83 C4 08 C3");
BlScanFunctionText(tsh_BlPrintf, "8D 44 24 08 33 D2 50 FF 74 24 08 33 C9 E8 ? ? ? ? 83 C4 08 C3");
return true;
}

140
inc/tsfuncs/BlHooks.hpp
View File

@@ -14,13 +14,13 @@ typedef unsigned int ADDR;
bool tsh_InitInternal();
bool tsh_DeinitInternal();
ADDR tsh_ScanCode(char *, char *);
ADDR tsh_ScanText(char *);
void tsh_HookFunction(ADDR, ADDR, BYTE *);
void tsh_UnhookFunction(ADDR, BYTE *);
int tsh_PatchAllMatches(unsigned int, char *, char *, char *, bool);
ADDR tsh_ScanCode(char*, char*);
ADDR tsh_ScanText(char*);
void tsh_HookFunction(ADDR, ADDR, BYTE*);
void tsh_UnhookFunction(ADDR, BYTE*);
int tsh_PatchAllMatches(unsigned int, char*, char*, char*, bool);
void tsh_PatchByte(ADDR, BYTE);
void tsh_PatchBytes(unsigned int, ADDR, BYTE *);
void tsh_PatchBytes(unsigned int, ADDR, BYTE*);
void tsh_PatchInt(ADDR, int);
// Debug print settings
@@ -34,13 +34,13 @@ void tsh_PatchInt(ADDR, int);
// Function short names
// Use in code when the def is not shared in a header
#define BlFunctionDef(returnType, convention, name, ...) \
typedef returnType(convention *tsh_##name##FnT)(__VA_ARGS__); \
#define BlFunctionDef(returnType, convention, name, ...) \
typedef returnType(convention* tsh_##name##FnT)(__VA_ARGS__); \
tsh_##name##FnT name;
// Use in header for shared function defs when a BlFunctionDefIntern exists in
// code
#define BlFunctionDefExtern(returnType, convention, name, ...) \
typedef returnType(convention *tsh_##name##FnT)(__VA_ARGS__); \
#define BlFunctionDefExtern(returnType, convention, name, ...) \
typedef returnType(convention* tsh_##name##FnT)(__VA_ARGS__); \
extern tsh_##name##FnT name;
// Use in code for shared function defs when a BlFunctionDefExtern exists in
// header
@@ -48,94 +48,84 @@ void tsh_PatchInt(ADDR, int);
// Scan for and assign the pattern to the variable, or err and return if not
// found
#define BlScanFunctionCode(target, patt, mask) \
target = (tsh_##target##FnT)tsh_ScanCode((char *)patt, (char *)mask); \
if (!target) { \
BlPrintf("RedoBlHooks | Cannot find function " #target "!"); \
return false; \
} else { \
if (tsh_DEBUGPRINT) \
BlPrintf("RedoBlHooks | Found function " #target " at %08x", \
(int)target); \
#define BlScanFunctionCode(target, patt, mask) \
target = (tsh_##target##FnT)tsh_ScanCode((char*)patt, (char*)mask); \
if (!target) { \
BlPrintf("RedoBlHooks | Cannot find function " #target "!"); \
return false; \
} else { \
if (tsh_DEBUGPRINT) \
BlPrintf("RedoBlHooks | Found function " #target " at %08x", (int)target); \
}
#define BlScanFunctionText(target, text) \
target = (tsh_##target##FnT)tsh_ScanText((char *)text); \
if (!target) { \
BlPrintf("RedoBlHooks | Cannot find function " #target "!"); \
return false; \
} else { \
if (tsh_DEBUGPRINT) \
BlPrintf("RedoBlHooks | Found function " #target " at %08x", \
(int)target); \
#define BlScanFunctionText(target, text) \
target = (tsh_##target##FnT)tsh_ScanText((char*)text); \
if (!target) { \
BlPrintf("RedoBlHooks | Cannot find function " #target "!"); \
return false; \
} else { \
if (tsh_DEBUGPRINT) \
BlPrintf("RedoBlHooks | Found function " #target " at %08x", (int)target); \
}
#define BlScanCode(target, patt, mask) \
target = tsh_ScanCode((char *)patt, (char *)mask); \
if (!target) { \
BlPrintf("RedoBlHooks | Cannot find pattern " #target "!"); \
return false; \
} else { \
if (tsh_DEBUGPRINT) \
BlPrintf("RedoBlHooks | Found " #target " at %08x", (int)target); \
#define BlScanCode(target, patt, mask) \
target = tsh_ScanCode((char*)patt, (char*)mask); \
if (!target) { \
BlPrintf("RedoBlHooks | Cannot find pattern " #target "!"); \
return false; \
} else { \
if (tsh_DEBUGPRINT) \
BlPrintf("RedoBlHooks | Found " #target " at %08x", (int)target); \
}
#define BlScanText(target, text) \
target = tsh_ScanText((char *)text); \
if (!target) { \
BlPrintf("RedoBlHooks | Cannot find " #target "!"); \
return false; \
} else { \
if (tsh_DEBUGPRINT) \
BlPrintf("RedoBlHooks | Found " #target " at %08x", (int)target); \
#define BlScanText(target, text) \
target = tsh_ScanText((char*)text); \
if (!target) { \
BlPrintf("RedoBlHooks | Cannot find " #target "!"); \
return false; \
} else { \
if (tsh_DEBUGPRINT) \
BlPrintf("RedoBlHooks | Found " #target " at %08x", (int)target); \
}
// Use in code to define the data and functions for hooking a function
#define BlFunctionHookDef(func) \
BYTE tsh_BlFunctionHook##func##Data[6]; \
void func##HookOn() { \
tsh_HookFunction((ADDR)func, (ADDR)func##Hook, \
tsh_BlFunctionHook##func##Data); \
} \
void func##HookOff() { \
tsh_UnhookFunction((ADDR)func, tsh_BlFunctionHook##func##Data); \
}
#define BlFunctionHookDef(func) \
BYTE tsh_BlFunctionHook##func##Data[6]; \
void func##HookOn() { tsh_HookFunction((ADDR)func, (ADDR)func##Hook, tsh_BlFunctionHook##func##Data); } \
void func##HookOff() { tsh_UnhookFunction((ADDR)func, tsh_BlFunctionHook##func##Data); }
// Use in code to initialize the hook once
#define BlFunctionHookInit(func) \
tsh_DeprotectAddress(6, (ADDR)func); \
#define BlFunctionHookInit(func) \
tsh_DeprotectAddress(6, (ADDR)func); \
func##HookOn();
// Replace all matches of the pattern with the given byte string
#define BlPatchAllMatchesCode(len, patt, mask, repl) \
tsh_PatchAllMatchesCode((ADDR)len, (char *)patt, (char *)mask, (char *)repl, \
tsh_DEBUGPRINT);
#define BlPatchAllMatchesText(len, text, repl) \
tsh_PatchAllMatchesCode((ADDR)len, (char *)text, (char *)repl, \
tsh_DEBUGPRINT);
#define BlPatchAllMatchesCode(len, patt, mask, repl) \
tsh_PatchAllMatchesCode((ADDR)len, (char*)patt, (char*)mask, (char*)repl, tsh_DEBUGPRINT);
#define BlPatchAllMatchesText(len, text, repl) \
tsh_PatchAllMatchesCode((ADDR)len, (char*)text, (char*)repl, tsh_DEBUGPRINT);
// Deprotect and replace one byte
#define BlPatchByte(addr, byte) tsh_PatchByte((ADDR)addr, (BYTE)byte);
// Deprotect and replace a byte string
#define BlPatchBytes(len, addr, repl) \
tsh_PatchBytes((ADDR)len, (ADDR)addr, (BYTE *)repl);
#define BlPatchBytes(len, addr, repl) tsh_PatchBytes((ADDR)len, (ADDR)addr, (BYTE*)repl);
// BlPrintf(char* format, ...)
#define BlPrintf(...) \
if (tsh_BlPrintf) { \
tsh_BlPrintf(__VA_ARGS__); \
#define BlPrintf(...) \
if (tsh_BlPrintf) { \
tsh_BlPrintf(__VA_ARGS__); \
}
// BlHooksInit() -> bool: success
#define BlHooksInit() \
if (!tsh_InitInternal()) { \
BlPrintf("BlHooksInit failed"); \
return false; \
#define BlHooksInit() \
if (!tsh_InitInternal()) { \
BlPrintf("BlHooksInit failed"); \
return false; \
}
// BlHooksDeinit() -> bool: success
#define BlHooksDeinit() \
if (!tsh_DeinitInternal()) { \
BlPrintf("BlHooksDeinit failed"); \
return false; \
#define BlHooksDeinit() \
if (!tsh_DeinitInternal()) { \
BlPrintf("BlHooksDeinit failed"); \
return false; \
}
// Scanned structures
BlFunctionDefExtern(void, , tsh_BlPrintf, const char *, ...);
BlFunctionDefExtern(void, , tsh_BlPrintf, const char*, ...);
#endif