#include #include #include #include #include #include "sdk/amxxmodule.h" #include "CVector.h" #include "CString.h" #include "sh_stack.h" #include "hook.h" #include "forward.h" #include "ham_const.h" #include "ham_utils.h" #include "DataHandler.h" extern bool gDoForwards; // Return value pushes #define PUSH_VOID() ReturnStack.push(new Data(RET_VOID, NULL)); OrigReturnStack.push(new Data(RET_VOID, NULL)); #define PUSH_INT() ReturnStack.push(new Data(RET_INTEGER, (void *)&ret)); OrigReturnStack.push(new Data(RET_INTEGER, (void *)&origret)); #define PUSH_FLOAT() ReturnStack.push(new Data(RET_FLOAT, (void *)&ret)); OrigReturnStack.push(new Data(RET_FLOAT, (void *)&origret)); #define PUSH_VECTOR() ReturnStack.push(new Data(RET_VECTOR, (void *)&ret)); OrigReturnStack.push(new Data(RET_VECTOR, (void *)&origret)); #define PUSH_CBASE() ReturnStack.push(new Data(RET_CBASE, (void *)&ret)); OrigReturnStack.push(new Data(RET_CBASE, (void *)&origret)); #define PUSH_STRING() ReturnStack.push(new Data(RET_STRING, (void *)&ret)); OrigReturnStack.push(new Data(RET_STRING, (void *)&origret)); // Pop off return values #define POP() delete ReturnStack.front(); ReturnStack.pop(); delete OrigReturnStack.front(); OrigReturnStack.pop(); // Parameter value pushes #define MAKE_VECTOR() \ int iThis=PrivateToIndex(pthis); \ CVector *__vec=new CVector; \ ParamStack.push(__vec); \ P_CBASE(pthis, iThis) #define P_INT(___PARAM) __vec->push_back(new Data(RET_INTEGER, (void *) & (___PARAM))); #define P_FLOAT(___PARAM) __vec->push_back(new Data(RET_FLOAT, (void *) & (___PARAM))); #define P_VECTOR(___PARAM) __vec->push_back(new Data(RET_VECTOR, (void *) & (___PARAM))); #define P_STR(___PARAM) __vec->push_back(new Data(RET_STRING, (void *) & (___PARAM))); #define P_CBASE(__PARAM, __INDEX) __vec->push_back(new Data(RET_CBASE, (void *) & (__PARAM), reinterpret_cast(& (__INDEX)))); #define P_ENTVAR(__PARAM, __INDEX) __vec->push_back(new Data(RET_ENTVAR, (void *) & (__PARAM), reinterpret_cast(& (__INDEX)))); #define P_TRACE(__PARAM) __vec->push_back(new Data(RET_TRACE, (void *) (__PARAM))); #define P_PTRVECTOR(__PARAM) __vec->push_back(new Data(RET_VECTOR, (void *) (__PARAM))); #define P_PTRFLOAT(__PARAM) __vec->push_back(new Data(RET_FLOAT, (void *) (__PARAM))); #define P_ITEMINFO(__PARAM) __vec->push_back(new Data(RET_ITEMINFO, (void *) & (__PARAM))); #define KILL_VECTOR() \ CVector::iterator end=__vec->end(); \ for (CVector::iterator i=__vec->begin(); i!=end; ++i) \ { \ delete (*i); \ } \ delete __vec; \ ParamStack.pop(); #define PRE_START() \ bool DoForwards=gDoForwards; \ gDoForwards=true; \ int result=HAM_UNSET; \ ReturnStatus.push(&result); \ int thisresult=HAM_UNSET; \ if (DoForwards) \ { \ CVector::iterator end=hook->pre.end(); \ for (CVector::iterator i=hook->pre.begin(); i!=end; i++) \ { \ if ((*i)->state == FSTATE_OK) \ { \ thisresult=MF_ExecuteForward((*i)->id,iThis #define PRE_END() \ ); \ } \ if (thisresult > result) \ { \ result=thisresult; \ } \ } \ } \ if (result < HAM_SUPERCEDE) \ { #define POST_START() \ } \ if (DoForwards) \ { \ CVector::iterator end=hook->post.end(); \ for (CVector::iterator i=hook->post.begin(); i!=end; i++) \ { \ if ((*i)->state == FSTATE_OK) \ { \ MF_ExecuteForward((*i)->id,iThis #define POST_END() \ ); \ } \ } \ } \ ReturnStatus.pop(); #define CHECK_RETURN() \ if (thisresult < HAM_OVERRIDE) \ { \ return origret; \ } #define CHECK_RETURN_STR() \ if (thisresult < HAM_OVERRIDE) \ { \ return origret.c_str(); \ } #define CHECK_RETURN_VEC() \ if (thisresult < HAM_OVERRIDE) \ { \ memcpy(out, &origret, sizeof(Vector)); \ return; \ } void Hook_Void_Void(Hook *hook, void *pthis) { PUSH_VOID() MAKE_VECTOR() PRE_START() PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis,0); #elif defined __linux__ reinterpret_cast(hook->func)(pthis); #endif POST_START() POST_END() KILL_VECTOR() POP() } int Hook_Int_Void(Hook *hook, void *pthis) { int ret=0; int origret=0; PUSH_INT() MAKE_VECTOR() PRE_START() PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis,0); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis); #endif POST_START() POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } void Hook_Void_Entvar(Hook *hook, void *pthis, entvars_t *entvar) { PUSH_VOID() int iOther=EntvarToIndex(entvar); MAKE_VECTOR() P_ENTVAR(entvar, iOther) PRE_START() , iOther PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, entvar); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, entvar); #endif POST_START() , iOther POST_END() KILL_VECTOR() POP() } void Hook_Void_Cbase(Hook *hook, void *pthis, void *other) { PUSH_VOID() int iOther=PrivateToIndex(other); MAKE_VECTOR() P_CBASE(other, iOther) PRE_START() , iOther PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, other); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, other); #endif POST_START() , iOther POST_END() KILL_VECTOR() POP() } int Hook_Int_Float_Int(Hook *hook, void *pthis, float f1, int i1) { int ret=0; int origret=0; PUSH_INT() MAKE_VECTOR() P_FLOAT(f1) P_INT(i1) PRE_START() , f1, i1 PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, f1, i1); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, f1, i1); #endif POST_START() , f1, i1 POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } void Hook_Void_Entvar_Int(Hook *hook, void *pthis, entvars_t *ev1, int i1) { PUSH_VOID() int iOther=EntvarToIndex(ev1); MAKE_VECTOR() P_ENTVAR(ev1, iOther) P_INT(i1) PRE_START() , iOther, i1 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, ev1, i1); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, ev1, i1); #endif POST_START() , iOther, i1 POST_END() KILL_VECTOR() POP() } int Hook_Int_Cbase(Hook *hook, void *pthis, void *cb1) { int ret=0; int origret=0; PUSH_INT() int iOther=PrivateToIndex(cb1); MAKE_VECTOR() P_CBASE(cb1, iOther) PRE_START() , iOther PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, cb1); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, cb1); #endif POST_START() , iOther POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } void Hook_Void_Int_Int(Hook *hook, void *pthis, int i1, int i2) { PUSH_VOID() MAKE_VECTOR() P_INT(i1) P_INT(i2) PRE_START() ,i1, i2 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, i1, i2); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, i1, i2); #endif POST_START() ,i1, i2 POST_END() KILL_VECTOR() POP() } int Hook_Int_Int_Str_Int(Hook *hook, void *pthis, int i1, const char *sz1, int i2) { int ret=0; int origret=0; PUSH_INT() String a=sz1; MAKE_VECTOR() P_INT(i1) P_STR(a) P_INT(i2) PRE_START() ,i1, a.c_str(), i2 PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, i1, a.c_str(), i2); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, i1, a.c_str(), i2); #endif POST_START() ,i1, a.c_str(), i2 POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } int Hook_Int_Int(Hook *hook, void *pthis, int i1) { int ret=0; int origret=0; PUSH_INT() MAKE_VECTOR() P_INT(i1) PRE_START() ,i1 PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, i1); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, i1); #endif POST_START() ,i1 POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } int Hook_Int_Entvar(Hook *hook, void *pthis, entvars_t *ev1) { int ret=0; int origret=0; PUSH_INT() int iOther=EntvarToIndex(ev1); MAKE_VECTOR() P_ENTVAR(ev1, iOther) PRE_START() ,iOther PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, ev1); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, ev1); #endif POST_START() , iOther POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } int Hook_Int_Entvar_Entvar_Float_Int(Hook *hook, void *pthis, entvars_t *inflictor, entvars_t *attacker, float damage, int damagebits) { int ret=0; int origret=0; PUSH_INT() int iInflictor=EntvarToIndex(inflictor); int iAttacker=EntvarToIndex(attacker); MAKE_VECTOR() P_ENTVAR(inflictor, iInflictor) P_ENTVAR(attacker, iAttacker) P_FLOAT(damage) P_INT(damagebits) PRE_START() ,iInflictor, iAttacker, damage, damagebits PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, inflictor, attacker, damage, damagebits); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, inflictor, attacker, damage, damagebits); #endif POST_START() ,iInflictor, iAttacker, damage, damagebits POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } int Hook_Int_Entvar_Entvar_Float_Float_Int(Hook *hook, void *pthis, entvars_t *inflictor, entvars_t *attacker, float damage, float unknown, int damagebits) { int ret=0; int origret=0; PUSH_INT() int iInflictor=EntvarToIndex(inflictor); int iAttacker=EntvarToIndex(attacker); MAKE_VECTOR() P_ENTVAR(inflictor, iInflictor) P_ENTVAR(attacker, iAttacker) P_FLOAT(damage) P_FLOAT(unknown) P_INT(damagebits) PRE_START() ,iInflictor, iAttacker, damage, unknown, damagebits PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, inflictor, attacker, damage, unknown, damagebits); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, inflictor, attacker, damage, unknown, damagebits); #endif POST_START() ,iInflictor, iAttacker, damage, unknown, damagebits POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } void Hook_Void_Int(Hook *hook, void *pthis, int i1) { PUSH_VOID() MAKE_VECTOR() P_INT(i1) PRE_START() , i1 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, i1); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, i1); #endif POST_START() ,i1 POST_END() KILL_VECTOR() POP() } void Hook_Void_Cbase_Cbase_Int_Float(Hook *hook, void *pthis, void *cb1, void *cb2, int i1, float f1) { PUSH_VOID() int iCaller=PrivateToIndex(cb1); int iActivator=PrivateToIndex(cb2); MAKE_VECTOR() P_CBASE(cb1, iCaller) P_CBASE(cb2, iActivator) P_INT(i1) P_FLOAT(f1) PRE_START() ,iCaller, iActivator, i1, f1 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, cb1, cb2, i1, f1); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, cb1, cb2, i1, f1); #endif POST_START() ,iCaller, iActivator, i1, f1 POST_END() KILL_VECTOR() POP() } void Hook_Void_Entvar_Float_Vector_Trace_Int(Hook *hook, void *pthis, entvars_t *ev1, float f1, Vector v1, TraceResult *tr1, int i1) { PUSH_VOID() int iev1=EntvarToIndex(ev1); MAKE_VECTOR() P_ENTVAR(ev1, iev1) P_FLOAT(f1) P_VECTOR(v1) P_TRACE(tr1) P_INT(i1) PRE_START() ,iev1, f1, MF_PrepareCellArrayA(reinterpret_cast(&v1), 3, false), tr1, i1 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, ev1, f1, v1, tr1, i1); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, ev1, f1, v1, tr1, i1); #endif POST_START() , iev1, f1, MF_PrepareCellArrayA(reinterpret_cast(&v1), 3, false), tr1, i1 POST_END() KILL_VECTOR() POP() } void Hook_Void_Float_Vector_Trace_Int(Hook *hook, void *pthis, float f1, Vector v1, TraceResult *tr1, int i1) { PUSH_VOID() MAKE_VECTOR() P_FLOAT(f1) P_VECTOR(v1) P_TRACE(tr1) P_INT(i1) PRE_START() , f1, MF_PrepareCellArrayA(reinterpret_cast(&v1), 3, false), tr1, i1 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, f1, v1, tr1, i1); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, f1, v1, tr1, i1); #endif POST_START() , f1, MF_PrepareCellArrayA(reinterpret_cast(&v1), 3, false), tr1, i1 POST_END() KILL_VECTOR() POP() } const char *Hook_Str_Void(Hook *hook, void *pthis) { String ret; String origret; MAKE_VECTOR() PUSH_STRING() PRE_START() PRE_END() #if defined _WIN32 origret.assign(reinterpret_cast(hook->func)(pthis, 0)); #elif defined __linux__ origret.assign(reinterpret_cast(hook->func)(pthis)); #endif POST_START() POST_END() KILL_VECTOR() POP() CHECK_RETURN_STR(); return ret.c_str(); } void *Hook_Cbase_Void(Hook *hook, void *pthis) { void *ret=NULL; void *origret=NULL; PUSH_CBASE() MAKE_VECTOR() PRE_START() PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis); #endif POST_START() POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } #ifdef _WIN32 void Hook_Vector_Void(Hook *hook, void *pthis, Vector *out) #elif defined __linux__ void Hook_Vector_Void(Hook *hook, Vector *out, void *pthis) #endif { Vector ret; Vector origret; PUSH_VECTOR() MAKE_VECTOR() memset(&ret, 0x0, sizeof(Vector)); memset(&origret, 0x0, sizeof(Vector)); PRE_START() PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, &origret); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis); #endif POST_START() POST_END() KILL_VECTOR() POP() CHECK_RETURN_VEC() memcpy(out, &ret, sizeof(Vector)); } #ifdef _WIN32 void Hook_Vector_pVector(Hook *hook, void *pthis, Vector *out, Vector *v1) #elif defined __linux__ void Hook_Vector_pVector(Hook *hook, Vector *out, void *pthis, Vector *v1) #endif { Vector ret; Vector origret; PUSH_VECTOR() MAKE_VECTOR() P_PTRVECTOR(v1) memset(&ret, 0x0, sizeof(Vector)); memset(&origret, 0x0, sizeof(Vector)); PRE_START() , MF_PrepareCellArrayA(reinterpret_cast(v1), 3, false) PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, &origret, v1); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, v1); #endif POST_START() , MF_PrepareCellArrayA(reinterpret_cast(v1), 3, false) POST_END() KILL_VECTOR() POP() CHECK_RETURN_VEC() memcpy(out, &ret, sizeof(Vector)); } int Hook_Int_pVector(Hook *hook, void *pthis, Vector *v1) { int ret=0; int origret=0; PUSH_INT() MAKE_VECTOR() P_PTRVECTOR(v1) PRE_START() , MF_PrepareCellArrayA(reinterpret_cast(v1), 3, false) PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, v1); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, v1); #endif POST_START() , MF_PrepareCellArrayA(reinterpret_cast(v1), 3, false) POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } void Hook_Void_Entvar_Float_Float(Hook *hook, void *pthis, entvars_t *ev1, float f1, float f2) { PUSH_VOID() cell cev1=EntvarToIndex(ev1); MAKE_VECTOR() P_ENTVAR(ev1, cev1) P_FLOAT(f1) P_FLOAT(f2) PRE_START() , cev1, f1, f2 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, ev1, f1, f2); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, ev1, f1, f2); #endif POST_START() , cev1, f1, f2 POST_END() KILL_VECTOR() POP() } int Hook_Int_pFloat_pFloat(Hook *hook, void *pthis, float *f1, float *f2) { int ret=0; int origret=0; PUSH_INT() MAKE_VECTOR() P_PTRFLOAT(f1) P_PTRFLOAT(f2) PRE_START() , f1 != NULL ? *f1 : 0, f2 != NULL ? *f2 : 0 PRE_END() #if defined _WIN32 origret=reinterpret_cast(hook->func)(pthis, 0, f1, f2); #elif defined __linux__ origret=reinterpret_cast(hook->func)(pthis, f1, f2); #endif POST_START() , f1 != NULL ? *f1 : 0, f2 != NULL ? *f2 : 0 POST_END() KILL_VECTOR() POP() CHECK_RETURN() return ret; } void Hook_Void_Entvar_Float(Hook *hook, void *pthis, entvars_t *ev1, float f1) { PUSH_VOID() cell cev1=EntvarToIndex(ev1); MAKE_VECTOR() P_ENTVAR(ev1, cev1) P_FLOAT(f1) PRE_START() , cev1, f1 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, ev1, f1); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, ev1, f1); #endif POST_START() , cev1, f1 POST_END() KILL_VECTOR() POP() } void Hook_Void_Int_Int_Int(Hook *hook, void *pthis, int i1, int i2, int i3) { PUSH_VOID() MAKE_VECTOR() P_INT(i1) P_INT(i2) P_INT(i3) PRE_START() ,i1, i2, i3 PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, i1, i2, i3); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, i1, i2, i3); #endif POST_START() ,i1, i2, i3 POST_END() KILL_VECTOR() POP() } void Hook_Void_ItemInfo(Hook *hook, void *pthis, void *iteminfo) { PUSH_VOID() MAKE_VECTOR() P_ITEMINFO(iteminfo) PRE_START() ,iteminfo PRE_END() #if defined _WIN32 reinterpret_cast(hook->func)(pthis, 0, iteminfo); #elif defined __linux__ reinterpret_cast(hook->func)(pthis, iteminfo); #endif POST_START() ,iteminfo POST_END() KILL_VECTOR() POP() }