mirror of
https://github.com/alliedmodders/metamod-source.git
synced 2024-11-29 11:24:19 +01:00
e19413dd5b
--HG-- rename : sourcemm/IPluginManager.h => core/IPluginManager.h rename : sourcemm/ISmmAPI.h => core/ISmmAPI.h rename : sourcemm/ISmmPlugin.h => core/ISmmPlugin.h rename : sourcemm/ISmmPluginExt.h => core/ISmmPluginExt.h rename : sourcemm/LICENSE.txt => core/LICENSE.txt rename : sourcemm/Makefile => core/Makefile rename : sourcemm/changelog.txt => core/changelog.txt rename : sourcemm/episode1/console.cpp => core/episode1/console.cpp rename : sourcemm/episode1/console.h => core/episode1/console.h rename : sourcemm/episode1/convar_smm.h => core/episode1/convar_smm.h rename : sourcemm/episode1/msvc8/sourcemm.sln => core/episode1/msvc8/sourcemm.sln rename : sourcemm/episode1/msvc8/sourcemm.vcproj => core/episode1/msvc8/sourcemm.vcproj rename : sourcemm/episode1/provider_ep1.cpp => core/episode1/provider_ep1.cpp rename : sourcemm/episode1/provider_ep1.h => core/episode1/provider_ep1.h rename : sourcemm/episode1/vsp_listener.cpp => core/episode1/vsp_listener.cpp rename : sourcemm/episode1/vsp_listener.h => core/episode1/vsp_listener.h rename : sourcemm/metamod.cpp => core/metamod.cpp rename : sourcemm/metamod.h => core/metamod.h rename : sourcemm/metamod_console.cpp => core/metamod_console.cpp rename : sourcemm/metamod_console.h => core/metamod_console.h rename : sourcemm/metamod_oslink.cpp => core/metamod_oslink.cpp rename : sourcemm/metamod_oslink.h => core/metamod_oslink.h rename : sourcemm/metamod_plugins.cpp => core/metamod_plugins.cpp rename : sourcemm/metamod_plugins.h => core/metamod_plugins.h rename : sourcemm/metamod_provider.h => core/metamod_provider.h rename : sourcemm/metamod_util.cpp => core/metamod_util.cpp rename : sourcemm/metamod_util.h => core/metamod_util.h rename : sourcemm/episode2/console.cpp => core/provider/console.cpp rename : sourcemm/episode2/console.h => core/provider/console.h rename : sourcemm/episode2/msvc8/sourcemm.sln => core/provider/msvc8/sourcemm.sln rename : sourcemm/episode2/msvc8/sourcemm.vcproj => core/provider/msvc8/sourcemm.vcproj rename : sourcemm/episode2/msvc9/sourcemm.sln => core/provider/msvc9/sourcemm.sln rename : sourcemm/episode2/msvc9/sourcemm.vcproj => core/provider/msvc9/sourcemm.vcproj rename : sourcemm/episode2/provider_ep2.cpp => core/provider/provider_ep2.cpp rename : sourcemm/episode2/provider_ep2.h => core/provider/provider_ep2.h rename : sourcemm/episode2/vsp_listener.cpp => core/provider/vsp_listener.cpp rename : sourcemm/episode2/vsp_listener.h => core/provider/vsp_listener.h rename : sourcehook/FastDelegate.h => core/sourcehook/FastDelegate.h rename : sourcehook/generate/FastDelegate.h => core/sourcehook/generate/FastDelegate.h rename : sourcehook/generate/FastDelegate.hxx => core/sourcehook/generate/FastDelegate.hxx rename : sourcehook/generate/generate => core/sourcehook/generate/generate rename : sourcehook/generate/generate.bat => core/sourcehook/generate/generate.bat rename : sourcehook/generate/sh_memfuncinfo.h => core/sourcehook/generate/sh_memfuncinfo.h rename : sourcehook/generate/sh_memfuncinfo.hxx => core/sourcehook/generate/sh_memfuncinfo.hxx rename : sourcehook/generate/shworker.bin => core/sourcehook/generate/shworker.bin rename : sourcehook/generate/shworker.exe => core/sourcehook/generate/shworker.exe rename : sourcehook/generate/shworker/Makefile => core/sourcehook/generate/shworker/Makefile rename : sourcehook/generate/shworker/fd_hopter.cpp => core/sourcehook/generate/shworker/fd_hopter.cpp rename : sourcehook/generate/shworker/msvc7/shworker.vcproj => core/sourcehook/generate/shworker/msvc7/shworker.vcproj rename : sourcehook/generate/shworker/msvc8/shworker.vcproj => core/sourcehook/generate/shworker/msvc8/shworker.vcproj rename : sourcehook/generate/shworker/shworker.cpp => core/sourcehook/generate/shworker/shworker.cpp rename : sourcehook/generate/sourcehook.h => core/sourcehook/generate/sourcehook.h rename : sourcehook/generate/sourcehook.hxx => core/sourcehook/generate/sourcehook.hxx rename : sourcehook/sh_list.h => core/sourcehook/sh_list.h rename : sourcehook/sh_memfuncinfo.h => core/sourcehook/sh_memfuncinfo.h rename : sourcehook/sh_memory.h => core/sourcehook/sh_memory.h rename : sourcehook/sh_pagealloc.h => core/sourcehook/sh_pagealloc.h rename : sourcehook/sh_stack.h => core/sourcehook/sh_stack.h rename : sourcehook/sh_string.h => core/sourcehook/sh_string.h rename : sourcehook/sh_tinyhash.h => core/sourcehook/sh_tinyhash.h rename : sourcehook/sh_vector.h => core/sourcehook/sh_vector.h rename : sourcehook/sourcehook.cpp => core/sourcehook/sourcehook.cpp rename : sourcehook/sourcehook.h => core/sourcehook/sourcehook.h rename : sourcehook/sourcehook_hookmangen.cpp => core/sourcehook/sourcehook_hookmangen.cpp rename : sourcehook/sourcehook_hookmangen.h => core/sourcehook/sourcehook_hookmangen.h rename : sourcehook/sourcehook_hookmangen_x86.h => core/sourcehook/sourcehook_hookmangen_x86.h rename : sourcehook/sourcehook_impl.h => core/sourcehook/sourcehook_impl.h rename : sourcehook/sourcehook_impl_chook.h => core/sourcehook/sourcehook_impl_chook.h rename : sourcehook/sourcehook_impl_chookidman.h => core/sourcehook/sourcehook_impl_chookidman.h rename : sourcehook/sourcehook_impl_chookmaninfo.h => core/sourcehook/sourcehook_impl_chookmaninfo.h rename : sourcehook/sourcehook_impl_ciface.h => core/sourcehook/sourcehook_impl_ciface.h rename : sourcehook/sourcehook_impl_cproto.h => core/sourcehook/sourcehook_impl_cproto.h rename : sourcehook/sourcehook_impl_cvfnptr.h => core/sourcehook/sourcehook_impl_cvfnptr.h rename : sourcehook/sourcehook_pibuilder.h => core/sourcehook/sourcehook_pibuilder.h rename : sourcehook/test/Makefile => core/sourcehook/test/Makefile rename : sourcehook/test/generate.bat => core/sourcehook/test/generate.bat rename : sourcehook/test/main.cpp => core/sourcehook/test/main.cpp rename : sourcehook/test/msvc7/test.vcproj => core/sourcehook/test/msvc7/test.vcproj rename : sourcehook/test/msvc8/test.vcproj => core/sourcehook/test/msvc8/test.vcproj rename : sourcehook/test/sourcehook_test.h => core/sourcehook/test/sourcehook_test.h rename : sourcehook/test/test1.cpp => core/sourcehook/test/test1.cpp rename : sourcehook/test/test2.cpp => core/sourcehook/test/test2.cpp rename : sourcehook/test/test3.cpp => core/sourcehook/test/test3.cpp rename : sourcehook/test/test4.cpp => core/sourcehook/test/test4.cpp rename : sourcehook/test/testbail.cpp => core/sourcehook/test/testbail.cpp rename : sourcehook/test/testbail.h => core/sourcehook/test/testbail.h rename : sourcehook/test/testbail2.cpp => core/sourcehook/test/testbail2.cpp rename : sourcehook/test/testevents.h => core/sourcehook/test/testevents.h rename : sourcehook/test/testhookmangen.cpp => core/sourcehook/test/testhookmangen.cpp rename : sourcehook/test/testhookmangen.h => core/sourcehook/test/testhookmangen.h rename : sourcehook/test/testhookmangen.hxx => core/sourcehook/test/testhookmangen.hxx rename : sourcehook/test/testlist.cpp => core/sourcehook/test/testlist.cpp rename : sourcehook/test/testmanual.cpp => core/sourcehook/test/testmanual.cpp rename : sourcehook/test/testmulti.cpp => core/sourcehook/test/testmulti.cpp rename : sourcehook/test/testrecall.cpp => core/sourcehook/test/testrecall.cpp rename : sourcehook/test/testreentr.cpp => core/sourcehook/test/testreentr.cpp rename : sourcehook/test/testref.cpp => core/sourcehook/test/testref.cpp rename : sourcehook/test/testrefret.cpp => core/sourcehook/test/testrefret.cpp rename : sourcehook/test/testvphooks.cpp => core/sourcehook/test/testvphooks.cpp rename : sourcemm/svn_version.h => core/svn_version.h rename : sourcemm/svn_version.tpl => core/svn_version.tpl rename : sourcemm/version.rc => core/version.rc
751 lines
20 KiB
C++
751 lines
20 KiB
C++
#include <string>
|
|
#include "sourcehook_test.h"
|
|
#include "testevents.h"
|
|
|
|
#include "sh_memory.h"
|
|
|
|
namespace
|
|
{
|
|
StateList g_States;
|
|
SourceHook::ISourceHook *g_SHPtr;
|
|
SourceHook::Plugin g_PLID;
|
|
|
|
// TEST1
|
|
// Basic tests
|
|
// Hooking and callclass
|
|
|
|
MAKE_STATE_1(State_ModuleInMemory, bool);
|
|
MAKE_STATE(State_F1_Called);
|
|
MAKE_STATE_1(State_F1_PreHandler_Called, void*);
|
|
MAKE_STATE_1(State_F1_PostHandler_Called, void*);
|
|
MAKE_STATE_1(State_F1_HookAdded, bool);
|
|
MAKE_STATE(State_F1_HookRemoved);
|
|
MAKE_STATE(State_F1_CallClassGenerated);
|
|
MAKE_STATE(State_F1_CallClassReleased);
|
|
MAKE_STATE_1(State_F299_Called, std::string);
|
|
MAKE_STATE_1(State_F299_PreHandlerCalled, std::string);
|
|
MAKE_STATE_1(State_F299_PostHandlerCalled, std::string);
|
|
MAKE_STATE_1(State_F299Ret, bool);
|
|
|
|
class Test
|
|
{
|
|
public:
|
|
virtual void F1()
|
|
{
|
|
ADD_STATE(State_F1_Called);
|
|
}
|
|
virtual void F2(){}
|
|
virtual void F3(){}
|
|
virtual void F4(){}
|
|
virtual void F5(){}
|
|
virtual void F6(){}
|
|
virtual void F7(){}
|
|
virtual void F8(){}
|
|
virtual void F9(){}
|
|
virtual void F10(){}
|
|
virtual void F11(){}
|
|
virtual void F12(){}
|
|
virtual void F13(){}
|
|
virtual void F14(){}
|
|
virtual void F15(){}
|
|
virtual void F16(){}
|
|
virtual void F17(){}
|
|
virtual void F18(){}
|
|
virtual void F19(){}
|
|
virtual float F20() const { return 1.0f; } // Look! F20 is const-ed and returns a float
|
|
virtual void F21(){}
|
|
virtual void F22(){}
|
|
virtual void F23(){}
|
|
virtual void F24(){}
|
|
virtual void F25(){}
|
|
virtual void F26(){}
|
|
virtual void F27(){}
|
|
virtual void F28(){}
|
|
virtual void F29(){}
|
|
virtual void F30(){}
|
|
virtual void F31(){}
|
|
virtual void F32(){}
|
|
virtual void F33(){}
|
|
virtual void F34(){}
|
|
virtual void F35(){}
|
|
virtual void F36(){}
|
|
virtual void F37(){}
|
|
virtual void F38(){}
|
|
virtual void F39(){}
|
|
virtual void F40(){}
|
|
virtual void F41(){}
|
|
virtual void F42(){}
|
|
virtual void F43(){}
|
|
virtual void F44(){}
|
|
virtual void F45(){}
|
|
virtual void F46(){}
|
|
virtual void F47(){}
|
|
virtual void F48(){}
|
|
virtual void F49(){}
|
|
virtual void F50(){}
|
|
virtual void F51(){}
|
|
virtual void F52(){}
|
|
virtual void F53(){}
|
|
virtual void F54(){}
|
|
virtual void F55(){}
|
|
virtual void F56(){}
|
|
virtual void F57(){}
|
|
virtual void F58(){}
|
|
virtual void F59(){}
|
|
virtual void F60(int &hello){}
|
|
virtual void F61(){}
|
|
virtual void F62(){}
|
|
virtual void F63(){}
|
|
virtual void F64(){}
|
|
virtual void F65(){}
|
|
virtual void F66(){}
|
|
virtual void F67(){}
|
|
virtual void F68(){}
|
|
virtual void F69(){}
|
|
virtual void F70(){}
|
|
virtual void F71(){}
|
|
virtual void F72(){}
|
|
virtual void F73(){}
|
|
virtual void F74(){}
|
|
virtual void F75(){}
|
|
virtual void F76(){}
|
|
virtual void F77(){}
|
|
virtual void F78(){}
|
|
virtual void F79(){}
|
|
virtual void F80(){}
|
|
virtual void F81(){}
|
|
virtual void F82(){}
|
|
virtual void F83(){}
|
|
virtual void F84(){}
|
|
virtual void F85(){}
|
|
virtual void F86(){}
|
|
virtual void F87(){}
|
|
virtual void F88(){}
|
|
virtual void F89(){}
|
|
virtual void F90(){}
|
|
virtual void F91(){}
|
|
virtual void F92(){}
|
|
virtual void F93(){}
|
|
virtual void F94(){}
|
|
virtual void F95(){}
|
|
virtual void F96(){}
|
|
virtual void F97(){}
|
|
virtual void F98(){}
|
|
virtual void F99(){}
|
|
virtual void F100(){}
|
|
virtual void F101(){}
|
|
virtual void F102(){}
|
|
virtual void F103(){}
|
|
virtual void F104(){}
|
|
virtual void F105(){}
|
|
virtual void F106(){}
|
|
virtual void F107(){}
|
|
virtual void F108(){}
|
|
virtual void F109(){}
|
|
virtual void F110(){}
|
|
virtual void F111(){}
|
|
virtual void F112(){}
|
|
virtual void F113(){}
|
|
virtual void F114(){}
|
|
virtual void F115(){}
|
|
virtual void F116(){}
|
|
virtual void F117(){}
|
|
virtual void F118(){}
|
|
virtual void F119(){}
|
|
virtual void F120(){}
|
|
virtual void F121(){}
|
|
virtual void F122(){}
|
|
virtual void F123(){}
|
|
virtual void F124(){}
|
|
virtual void F125(){}
|
|
virtual void F126(){}
|
|
virtual void F127(){}
|
|
virtual void F128(){}
|
|
virtual void F129(){}
|
|
virtual void F130(){}
|
|
virtual void F131(){}
|
|
virtual void F132(){}
|
|
virtual void F133(){}
|
|
virtual void F134(){}
|
|
virtual void F135(){}
|
|
virtual void F136(){}
|
|
virtual void F137(){}
|
|
virtual void F138(){}
|
|
virtual void F139(){}
|
|
virtual void F140(){}
|
|
virtual void F141(){}
|
|
virtual void F142(){}
|
|
virtual void F143(){}
|
|
virtual void F144(){}
|
|
virtual void F145(){}
|
|
virtual void F146(){}
|
|
virtual void F147(){}
|
|
virtual void F148(){}
|
|
virtual void F149(){}
|
|
virtual void F150(){}
|
|
virtual void F151(){}
|
|
virtual void F152(){}
|
|
virtual void F153(){}
|
|
virtual void F154(){}
|
|
virtual void F155(){}
|
|
virtual void F156(){}
|
|
virtual void F157(){}
|
|
virtual void F158(){}
|
|
virtual void F159(){}
|
|
virtual void F160(){}
|
|
virtual void F161(){}
|
|
virtual void F162(){}
|
|
virtual void F163(){}
|
|
virtual void F164(){}
|
|
virtual void F165(){}
|
|
virtual void F166(){}
|
|
virtual void F167(){}
|
|
virtual void F168(){}
|
|
virtual void F169(){}
|
|
virtual void F170(){}
|
|
virtual void F171(){}
|
|
virtual void F172(){}
|
|
virtual void F173(){}
|
|
virtual void F174(){}
|
|
virtual void F175(){}
|
|
virtual void F176(){}
|
|
virtual void F177(){}
|
|
virtual void F178(){}
|
|
virtual void F179(){}
|
|
virtual void F180(){}
|
|
virtual void F181(){}
|
|
virtual void F182(){}
|
|
virtual void F183(){}
|
|
virtual void F184(){}
|
|
virtual void F185(){}
|
|
virtual void F186(){}
|
|
virtual void F187(){}
|
|
virtual void F188(){}
|
|
virtual void F189(){}
|
|
virtual void F190(){}
|
|
virtual void F191(){}
|
|
virtual void F192(){}
|
|
virtual void F193(){}
|
|
virtual void F194(){}
|
|
virtual void F195(){}
|
|
virtual void F196(){}
|
|
virtual void F197(){}
|
|
virtual void F198(){}
|
|
virtual void F199(){}
|
|
virtual void F200(){}
|
|
virtual void F201(){}
|
|
virtual void F202(){}
|
|
virtual void F203(){}
|
|
virtual void F204(){}
|
|
virtual void F205(){}
|
|
virtual void F206(){}
|
|
virtual void F207(){}
|
|
virtual void F208(){}
|
|
virtual void F209(){}
|
|
virtual void F210(){}
|
|
virtual void F211(){}
|
|
virtual void F212(){}
|
|
virtual void F213(){}
|
|
virtual void F214(){}
|
|
virtual void F215(){}
|
|
virtual void F216(){}
|
|
virtual void F217(){}
|
|
virtual void F218(){}
|
|
virtual void F219(){}
|
|
virtual void F220(){}
|
|
virtual void F221(){}
|
|
virtual void F222(){}
|
|
virtual void F223(){}
|
|
virtual void F224(){}
|
|
virtual void F225(){}
|
|
virtual void F226(){}
|
|
virtual void F227(){}
|
|
virtual void F228(){}
|
|
virtual void F229(){}
|
|
virtual void F230(){}
|
|
virtual void F231(){}
|
|
virtual void F232(){}
|
|
virtual void F233(){}
|
|
virtual void F234(){}
|
|
virtual void F235(){}
|
|
virtual void F236(){}
|
|
virtual void F237(){}
|
|
virtual void F238(){}
|
|
virtual void F239(){}
|
|
virtual void F240(){}
|
|
virtual void F241(){}
|
|
virtual void F242(){}
|
|
virtual void F243(){}
|
|
virtual void F244(){}
|
|
virtual void F245(){}
|
|
virtual void F246(){}
|
|
virtual void F247(){}
|
|
virtual void F248(){}
|
|
virtual void F249(){}
|
|
virtual void F250(){}
|
|
virtual void F251(){}
|
|
virtual void F252(){}
|
|
virtual void F253(){}
|
|
virtual void F254(){}
|
|
virtual void F255(){}
|
|
virtual void F256(){}
|
|
virtual void F257(){}
|
|
virtual void F258(){}
|
|
virtual void F259(){}
|
|
virtual void F260(){}
|
|
virtual void F261(){}
|
|
virtual void F262(){}
|
|
virtual void F263(){}
|
|
virtual void F264(){}
|
|
virtual void F265(){}
|
|
virtual void F266(){}
|
|
virtual void F267(){}
|
|
virtual void F268(){}
|
|
virtual void F269(){}
|
|
virtual void F270(){}
|
|
virtual void F271(){}
|
|
virtual void F272(){}
|
|
virtual void F273(){}
|
|
virtual void F274(){}
|
|
virtual void F275(){}
|
|
virtual void F276(){}
|
|
virtual void F277(){}
|
|
virtual void F278(){}
|
|
virtual void F279(){}
|
|
virtual void F280(){}
|
|
virtual void F281(){}
|
|
virtual void F282(){}
|
|
virtual void F283(){}
|
|
virtual void F284(){}
|
|
virtual void F285(){}
|
|
virtual void F286(){}
|
|
virtual void F287(){}
|
|
virtual void F288(){}
|
|
virtual void F289(){}
|
|
virtual void F290(){}
|
|
virtual void F291(){}
|
|
virtual void F292(){}
|
|
virtual void F293(){}
|
|
virtual void F294(){}
|
|
virtual void F295(){}
|
|
virtual void F296(){}
|
|
virtual void F297(){}
|
|
virtual void F298(){}
|
|
virtual bool F299(const char *mwah)
|
|
{
|
|
ADD_STATE(State_F299_Called(mwah));
|
|
return true;
|
|
}
|
|
};
|
|
|
|
SH_DECL_HOOK1(Test, F299, SH_NOATTRIB, 0, bool, const char *);
|
|
SH_DECL_HOOK0_void(Test, F1, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F2, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F3, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F4, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F5, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F6, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F7, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F8, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F9, SH_NOATTRIB, 0);
|
|
SH_DECL_HOOK0_void(Test, F10, SH_NOATTRIB, 0);
|
|
|
|
SH_DECL_HOOK0(Test, F20, const, 0, float);
|
|
SH_DECL_HOOK1_void(Test, F60, SH_NOATTRIB, 0, int&);
|
|
|
|
SH_DECL_EXTERN1(Test, F299, SH_NOATTRIB, 0, bool, const char *);
|
|
SH_DECL_EXTERN0_void(Test, F1, SH_NOATTRIB, 0);
|
|
|
|
META_RES g_F1Pre_WhatToDo;
|
|
META_RES g_F1Post_WhatToDo;
|
|
|
|
struct HandlersF1
|
|
{
|
|
void Pre()
|
|
{
|
|
ADD_STATE(State_F1_PreHandler_Called(reinterpret_cast<void*>(this)));
|
|
RETURN_META(g_F1Pre_WhatToDo);
|
|
}
|
|
|
|
void Post()
|
|
{
|
|
ADD_STATE(State_F1_PostHandler_Called(reinterpret_cast<void*>(this)));
|
|
RETURN_META(g_F1Post_WhatToDo);
|
|
}
|
|
};
|
|
|
|
META_RES g_F299Pre_WhatToDo;
|
|
bool g_F299Pre_WhatToRet;
|
|
|
|
bool F299_Pre(const char *mwah)
|
|
{
|
|
ADD_STATE(State_F299_PreHandlerCalled(mwah));
|
|
RETURN_META_VALUE(g_F299Pre_WhatToDo, g_F299Pre_WhatToRet);
|
|
}
|
|
|
|
bool F299_Post(const char *mwah)
|
|
{
|
|
ADD_STATE(State_F299_PostHandlerCalled(mwah));
|
|
RETURN_META_VALUE(MRES_OVERRIDE, META_RESULT_STATUS >= MRES_OVERRIDE ? !META_RESULT_OVERRIDE_RET(bool) :
|
|
!META_RESULT_ORIG_RET(bool));
|
|
}
|
|
|
|
void F60_Pre(int &hello)
|
|
{
|
|
hello = 10;
|
|
}
|
|
|
|
Test *MyTestFactory()
|
|
{
|
|
return new Test;
|
|
}
|
|
}
|
|
|
|
template <class T> T func(T a)
|
|
{
|
|
return a;
|
|
}
|
|
|
|
bool TestBasic(std::string &error)
|
|
{
|
|
// Simple test for ModuleInMemory
|
|
// 1) &error should on the stack
|
|
// 2) 0 should not be mapped
|
|
ADD_STATE(State_ModuleInMemory(SourceHook::ModuleInMemory(reinterpret_cast<char*>(&error), sizeof(error))));
|
|
ADD_STATE(State_ModuleInMemory(SourceHook::ModuleInMemory(0, 1)));
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_ModuleInMemory(true),
|
|
new State_ModuleInMemory(false),
|
|
NULL), "ModuleInMemory");
|
|
|
|
GET_SHPTR(g_SHPtr);
|
|
g_PLID = 1337;
|
|
|
|
HandlersF1 f1_handlers;
|
|
Test *pTest = MyTestFactory();
|
|
CAutoPtrDestruction<Test> apd(pTest);
|
|
|
|
void *pOrigVfnPtrF1 = (*reinterpret_cast<void***>(pTest))[0];
|
|
|
|
// 1) Get a call class and call the member through it and normally
|
|
SourceHook::CallClass<Test> *cc = SH_GET_CALLCLASS(pTest);
|
|
|
|
ADD_STATE(State_F1_CallClassGenerated);
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_CallClassGenerated,
|
|
new State_F1_Called,
|
|
new State_F1_Called,
|
|
NULL), "Part 1");
|
|
|
|
CHECK_COND(SH_GET_ORIG_VFNPTR_ENTRY(pTest, &Test::F1) == pOrigVfnPtrF1, "Part S1");
|
|
|
|
// 2) Request a call class again
|
|
SourceHook::CallClass<Test> *cc2 = SH_GET_CALLCLASS(pTest);
|
|
ADD_STATE(State_F1_CallClassGenerated);
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
SH_CALL(cc2, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
SH_RELEASE_CALLCLASS(cc2);
|
|
ADD_STATE(State_F1_CallClassReleased);
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_CallClassGenerated,
|
|
new State_F1_Called,
|
|
new State_F1_Called,
|
|
new State_F1_Called,
|
|
new State_F1_CallClassReleased,
|
|
new State_F1_Called,
|
|
new State_F1_Called,
|
|
NULL), "Part 2");
|
|
|
|
// 3) Add a pre hook
|
|
// (one add memfunc in old format)
|
|
g_F1Pre_WhatToDo = MRES_SUPERCEDE;
|
|
ADD_STATE(State_F1_HookAdded(SH_ADD_HOOK_MEMFUNC(Test, F1, pTest, &f1_handlers, &HandlersF1::Pre, false) ? true : false));
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_HookAdded(true),
|
|
new State_F1_Called,
|
|
new State_F1_PreHandler_Called(&f1_handlers),
|
|
NULL), "Part 3");
|
|
|
|
CHECK_COND(SH_GET_ORIG_VFNPTR_ENTRY(pTest, &Test::F1) == pOrigVfnPtrF1, "Part S3");
|
|
|
|
// 4) Rerequest the callclass
|
|
SH_RELEASE_CALLCLASS(cc);
|
|
|
|
ADD_STATE(State_F1_CallClassReleased);
|
|
cc2 = SH_GET_CALLCLASS(pTest);
|
|
ADD_STATE(State_F1_CallClassGenerated);
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_CallClassReleased,
|
|
new State_F1_CallClassGenerated,
|
|
new State_F1_Called,
|
|
new State_F1_PreHandler_Called(&f1_handlers),
|
|
NULL), "Part 4");
|
|
|
|
// 5) Check ignore / supercede
|
|
g_F1Pre_WhatToDo = MRES_SUPERCEDE;
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
g_F1Pre_WhatToDo = MRES_IGNORED;
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_Called,
|
|
new State_F1_PreHandler_Called(&f1_handlers),
|
|
new State_F1_Called,
|
|
new State_F1_PreHandler_Called(&f1_handlers),
|
|
new State_F1_Called,
|
|
NULL), "Part 5");
|
|
|
|
// 6) remove the hook again
|
|
// (one remove memfunc in old format)
|
|
SH_REMOVE_HOOK_MEMFUNC(Test, F1, pTest, &f1_handlers, &HandlersF1::Pre, false);
|
|
ADD_STATE(State_F1_HookRemoved);
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_HookRemoved,
|
|
new State_F1_Called,
|
|
new State_F1_Called,
|
|
NULL), "Part 6");
|
|
|
|
// 7) add a post hook now
|
|
g_F1Post_WhatToDo = MRES_IGNORED;
|
|
ADD_STATE(State_F1_HookAdded(SH_ADD_HOOK(Test, F1, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Post), true) ? true : false));
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_HookAdded(true),
|
|
new State_F1_Called,
|
|
new State_F1_Called,
|
|
new State_F1_PostHandler_Called(&f1_handlers),
|
|
NULL), "Part 7");
|
|
|
|
// 8) And a pre hook again
|
|
g_F1Pre_WhatToDo = MRES_IGNORED;
|
|
ADD_STATE(State_F1_HookAdded(SH_ADD_HOOK(Test, F1, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false) ? true : false));
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
g_F1Pre_WhatToDo = MRES_SUPERCEDE;
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_HookAdded(true),
|
|
new State_F1_Called,
|
|
new State_F1_PreHandler_Called(&f1_handlers),
|
|
new State_F1_Called,
|
|
new State_F1_PostHandler_Called(&f1_handlers),
|
|
new State_F1_Called,
|
|
new State_F1_PreHandler_Called(&f1_handlers),
|
|
new State_F1_PostHandler_Called(&f1_handlers),
|
|
NULL), "Part 8");
|
|
|
|
// 9) Remove all hooks
|
|
SH_REMOVE_HOOK(Test, F1, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
ADD_STATE(State_F1_HookRemoved);
|
|
SH_REMOVE_HOOK(Test, F1, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Post), true);
|
|
ADD_STATE(State_F1_HookRemoved);
|
|
|
|
SH_CALL(cc, &Test::F1)();
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_HookRemoved,
|
|
new State_F1_HookRemoved,
|
|
new State_F1_Called,
|
|
new State_F1_Called,
|
|
NULL), "Part 9");
|
|
|
|
// 10) Some checks on F299
|
|
g_F299Pre_WhatToDo = MRES_IGNORED;
|
|
g_F299Pre_WhatToRet = false;
|
|
|
|
ADD_STATE(State_F299Ret(pTest->F299("hi")));
|
|
ADD_STATE(State_F299Ret(SH_CALL(cc, &Test::F299)("hi")));
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
NULL), "Part 10.1");
|
|
|
|
// (one add staticfunc in old format)
|
|
SH_ADD_HOOK_STATICFUNC(Test, F299, pTest, F299_Pre, false);
|
|
ADD_STATE(State_F299Ret(pTest->F299("hi")));
|
|
ADD_STATE(State_F299Ret(SH_CALL(cc, &Test::F299)("hi")));
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F299_PreHandlerCalled("hi"),
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
NULL), "Part 10.2");
|
|
|
|
SH_ADD_HOOK(Test, F299, pTest, SH_STATIC(F299_Post), true);
|
|
ADD_STATE(State_F299Ret(pTest->F299("hi")));
|
|
ADD_STATE(State_F299Ret(SH_CALL(cc, &Test::F299)("hi")));
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F299_PreHandlerCalled("hi"),
|
|
new State_F299_Called("hi"),
|
|
new State_F299_PostHandlerCalled("hi"),
|
|
new State_F299Ret(false),
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
NULL), "Part 10.3");
|
|
|
|
g_F299Pre_WhatToDo = MRES_OVERRIDE;
|
|
ADD_STATE(State_F299Ret(pTest->F299("hi")));
|
|
ADD_STATE(State_F299Ret(SH_CALL(cc, &Test::F299)("hi")));
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F299_PreHandlerCalled("hi"),
|
|
new State_F299_Called("hi"),
|
|
new State_F299_PostHandlerCalled("hi"),
|
|
new State_F299Ret(true),
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
NULL), "Part 10.4");
|
|
|
|
g_F299Pre_WhatToDo = MRES_SUPERCEDE;
|
|
ADD_STATE(State_F299Ret(pTest->F299("hi")));
|
|
ADD_STATE(State_F299Ret(SH_CALL(cc, &Test::F299)("hi")));
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F299_PreHandlerCalled("hi"),
|
|
new State_F299_PostHandlerCalled("hi"),
|
|
new State_F299Ret(true),
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
NULL), "Part 10.5");
|
|
|
|
// (one remove staticfunc in old format)
|
|
SH_REMOVE_HOOK_STATICFUNC(Test, F299, pTest, F299_Pre, false);
|
|
ADD_STATE(State_F299Ret(pTest->F299("hi")));
|
|
ADD_STATE(State_F299Ret(SH_CALL(cc, &Test::F299)("hi")));
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F299_Called("hi"),
|
|
new State_F299_PostHandlerCalled("hi"),
|
|
new State_F299Ret(false),
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
NULL), "Part 10.6");
|
|
|
|
SH_REMOVE_HOOK(Test, F299, pTest, SH_STATIC(F299_Post), true);
|
|
ADD_STATE(State_F299Ret(pTest->F299("hi")));
|
|
ADD_STATE(State_F299Ret(SH_CALL(cc, &Test::F299)("hi")));
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
new State_F299_Called("hi"),
|
|
new State_F299Ret(true),
|
|
NULL), "Part 10.7");
|
|
|
|
// 11) Release callclass
|
|
SH_RELEASE_CALLCLASS(cc);
|
|
ADD_STATE(State_F1_CallClassReleased);
|
|
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_CallClassReleased,
|
|
NULL), "Part 11");
|
|
|
|
// 11 1/2) Test removing hook by id
|
|
|
|
g_F1Pre_WhatToDo = MRES_IGNORED;
|
|
|
|
pTest->F1();
|
|
int hookPre = SH_ADD_HOOK(Test, F1, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
int hookPost = SH_ADD_HOOK(Test, F1, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Post), true);
|
|
|
|
pTest->F1();
|
|
SH_REMOVE_HOOK_ID(hookPost);
|
|
pTest->F1();
|
|
SH_REMOVE_HOOK_ID(hookPre);
|
|
pTest->F1();
|
|
|
|
CHECK_STATES((&g_States,
|
|
new State_F1_Called,
|
|
|
|
new State_F1_PreHandler_Called(&f1_handlers),
|
|
new State_F1_Called,
|
|
new State_F1_PostHandler_Called(&f1_handlers),
|
|
|
|
new State_F1_PreHandler_Called(&f1_handlers),
|
|
new State_F1_Called,
|
|
|
|
new State_F1_Called,
|
|
|
|
NULL), "Part 11 1/2");
|
|
|
|
// 12) Try out one ref param
|
|
SH_ADD_HOOK(Test, F60, pTest, SH_STATIC(F60_Pre), false);
|
|
int a = 0;
|
|
pTest->F60(a);
|
|
CHECK_COND(a == 10, "Part12");
|
|
SH_REMOVE_HOOK(Test, F60, pTest, SH_STATIC(F60_Pre), false);
|
|
|
|
// 13) Remove hooks after the instance has been removed
|
|
SH_ADD_HOOK(Test, F1, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), true);
|
|
SH_ADD_HOOK(Test, F2, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), true);
|
|
SH_ADD_HOOK(Test, F3, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_ADD_HOOK(Test, F4, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), true);
|
|
SH_ADD_HOOK(Test, F5, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_ADD_HOOK(Test, F6, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), true);
|
|
SH_ADD_HOOK(Test, F7, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_ADD_HOOK(Test, F8, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_ADD_HOOK(Test, F9, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_ADD_HOOK(Test, F10, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
|
|
delete pTest;
|
|
apd.clear();
|
|
|
|
SH_REMOVE_HOOK(Test, F1, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), true);
|
|
SH_REMOVE_HOOK(Test, F2, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), true);
|
|
SH_REMOVE_HOOK(Test, F3, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_REMOVE_HOOK(Test, F4, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), true);
|
|
SH_REMOVE_HOOK(Test, F5, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_REMOVE_HOOK(Test, F6, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), true);
|
|
SH_REMOVE_HOOK(Test, F7, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_REMOVE_HOOK(Test, F8, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_REMOVE_HOOK(Test, F9, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
SH_REMOVE_HOOK(Test, F10, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false);
|
|
CHECK_COND(SH_REMOVE_HOOK(Test, F10, pTest, SH_MEMBER(&f1_handlers, &HandlersF1::Pre), false) == false, "Part 12.1");
|
|
|
|
Test_CompleteShutdown(g_SHPtr);
|
|
|
|
return true;
|
|
}
|