bool TestThisPtrOffs(std::string &error)
{
GET_SHPTR(g_SHPtr);
g_PLID = 1337;
Derived *pD = MyInstanceFactory();
Base1 *pB1 = pD;
Base2 *pB2 = pD;
CAutoPtrDestruction<Derived> apd(pD);
// It should be:
// pB1 = pD
// pB2 > pB1
// 1)
// Get a callclass for pD
// Verify whether the this pointers are correct
// Also call them normally to make sure that we aren't messing it up ;)
SourceHook::CallClass<Derived> *pD_CC = SH_GET_CALLCLASS(pD);
SH_CALL(pD_CC, &Derived::Func1)();
SH_CALL(pD_CC, &Derived::Func2)();
SH_CALL(pD_CC, &Derived::Func3)();
pD->Func1();
pD->Func2();
pD->Func3();
CHECK_STATES((&g_States,
new State_Func1_Called(pB1),
new State_Func2_Called(pB2),
new State_Func3_Called(pD),
new State_Func1_Called(pB1),
new State_Func2_Called(pB2),
new State_Func3_Called(pD),
NULL), "Part 1");
SH_CALL(pD_CC, &Base1::Func1)();
SH_CALL(pD_CC, &Base2::Func2)();
CHECK_STATES((&g_States,
new State_Func1_Called(pB1),
new State_Func2_Called(pB2),
NULL), "Part 1.1");
// 2)
// Get callclasses for the other ones and verify it as well
SourceHook::CallClass<Base1> *pB1_CC = SH_GET_CALLCLASS(pB1);
SourceHook::CallClass<Base2> *pB2_CC = SH_GET_CALLCLASS(pB2);
SH_CALL(pB1_CC, &Base1::Func1)();
SH_CALL(pB2_CC, &Base2::Func2)();
CHECK_STATES((&g_States,
new State_Func1_Called(pB1),
new State_Func2_Called(pB2),
NULL), "Part 2");
// 3) Add hooks on them (referring to them through pD1 / Derived)
// Check whether the hooks are called with the correct this pointer
SH_ADD_HOOK(Derived, Func1, pD, SH_STATIC(Handler_Func1), false);
SH_ADD_HOOK(Derived, Func2, pD, SH_STATIC(Handler_Func2), false);
SH_ADD_HOOK(Derived, Func3, pD, SH_STATIC(Handler_Func3), false);
pD->Func1();
pD->Func2();
pD->Func3();
pB1->Func1();
pB2->Func2();
// The handlers should always be called with the pointer to Derived
CHECK_STATES((&g_States,
new State_Func1H_Called(pD),
new State_Func1_Called(pB1),
new State_Func2H_Called(pD),
new State_Func2_Called(pB2),
new State_Func3H_Called(pD),
new State_Func3_Called(pD),
new State_Func1H_Called(pD),
new State_Func1_Called(pB1),
new State_Func2H_Called(pD),
new State_Func2_Called(pB2),
NULL), "Part 3");
SH_REMOVE_HOOK(Derived, Func1, pD, SH_STATIC(Handler_Func1), false);
SH_REMOVE_HOOK(Derived, Func2, pD, SH_STATIC(Handler_Func2), false);
SH_REMOVE_HOOK(Derived, Func3, pD, SH_STATIC(Handler_Func3), false);
// 4)
// Now add the hooks on Base1 and Base2 and check again
// Note that the new implicit_cast should convert the pD to Base1*/Base2* :)
SH_ADD_HOOK(Base1, Func1, pD, SH_STATIC(Handler_Func1), false);
SH_ADD_HOOK(Base2, Func2, pD, SH_STATIC(Handler_Func2), false);
SH_ADD_HOOK(Derived, Func3, pD, SH_STATIC(Handler_Func3), false);
pD->Func1();
pD->Func2();
pD->Func3();
pB1->Func1();
pB2->Func2();
// This time, the handlers for Func1 should be called with pB1 and the handlers
// for Func2 should be called with pB2
CHECK_STATES((&g_States,
new State_Func1H_Called(pB1),
new State_Func1_Called(pB1),
new State_Func2H_Called(pB2),
new State_Func2_Called(pB2),
new State_Func3H_Called(pD),
new State_Func3_Called(pD),
new State_Func1H_Called(pB1),
new State_Func1_Called(pB1),
new State_Func2H_Called(pB2),
new State_Func2_Called(pB2),
NULL), "Part 4");
SH_REMOVE_HOOK(Base1, Func1, pD, SH_STATIC(Handler_Func1), false);
SH_REMOVE_HOOK(Base2, Func2, pD, SH_STATIC(Handler_Func2), false);
SH_REMOVE_HOOK(Derived, Func3, pD, SH_STATIC(Handler_Func3), false);
// 5)
// Add some hooks, and use callclasses
// 5.1) First off, add all of them on pD
SH_ADD_HOOK(Derived, Func1, pD, SH_STATIC(Handler_Func1), false);
SH_ADD_HOOK(Derived, Func2, pD, SH_STATIC(Handler_Func2), false);
SH_ADD_HOOK(Derived, Func3, pD, SH_STATIC(Handler_Func3), false);
pD->Func1();
pD->Func2();
pD->Func3();
CHECK_STATES((&g_States,
new State_Func1H_Called(pD),
new State_Func1_Called(pB1),
new State_Func2H_Called(pD),
new State_Func2_Called(pB2),
new State_Func3H_Called(pD),
new State_Func3_Called(pD),
NULL), "Part 5.1");
SH_CALL(pD_CC, &Derived::Func1)();
SH_CALL(pD_CC, &Derived::Func2)();
SH_CALL(pD_CC, &Derived::Func3)();
SH_CALL(pB1_CC, &Base1::Func1)();
SH_CALL(pB2_CC, &Base2::Func2)();
CHECK_STATES((&g_States,
new State_Func1_Called(pB1),
new State_Func2_Called(pB2),
new State_Func3_Called(pD),
new State_Func1_Called(pB1),
new State_Func2_Called(pB2),
NULL), "Part 5.2");
SH_REMOVE_HOOK(Derived, Func1, pD, SH_STATIC(Handler_Func1), false);
SH_REMOVE_HOOK(Derived, Func2, pD, SH_STATIC(Handler_Func2), false);
SH_REMOVE_HOOK(Derived, Func3, pD, SH_STATIC(Handler_Func3), false);
SH_RELEASE_CALLCLASS(pB1_CC);
SH_RELEASE_CALLCLASS(pB2_CC);
SH_RELEASE_CALLCLASS(pD_CC);
return true;
}
