1
0
mirror of https://github.com/alliedmodders/metamod-source.git synced 2024-12-02 14:24:16 +01:00
HLMetaModOfficial/sourcehook/test/testreentr.cpp

601 lines
13 KiB
C++
Raw Normal View History

// Reentrancy test
// = calling hooks from hook handlers, etc
#include <string>
#include "sourcehook_test.h"
#include "testevents.h"
namespace
{
StateList g_States;
SourceHook::ISourceHook *g_SHPtr;
SourceHook::Plugin g_PLID;
struct C1;
struct C2;
struct C3;
struct C4;
struct C5;
struct C6;
struct C7;
struct C8;
C1 *g_pC1;
C2 *g_pC2;
C3 *g_pC3;
C4 *g_pC4;
C5 *g_pC5;
C6 *g_pC6;
C7 *g_pC7;
C8 *g_pC8;
// param1: This pointer
MAKE_STATE_1(State_C1_F, C1*);
MAKE_STATE_1(State_C2_F, C2*);
MAKE_STATE_1(State_C3_F, C3*);
MAKE_STATE_1(State_C4_F, C4*);
MAKE_STATE_1(State_C5_F, C5*);
MAKE_STATE_1(State_C6_F, C6*);
MAKE_STATE_1(State_C7_F, C7*);
MAKE_STATE_1(State_C8_F, C8*);
MAKE_STATE_1(State_C1_G, C1*);
MAKE_STATE_1(State_C2_G, C2*);
MAKE_STATE_1(State_C3_G, C3*);
MAKE_STATE_1(State_C4_G, C4*);
MAKE_STATE_1(State_C5_G, C5*);
MAKE_STATE_1(State_C6_G, C6*);
MAKE_STATE_1(State_C7_G, C7*);
MAKE_STATE_1(State_C8_G, C8*);
MAKE_STATE_1(State_H_C1_F, C1*);
MAKE_STATE_1(State_H_C2_F, C2*);
MAKE_STATE_1(State_H_C3_F, C3*);
MAKE_STATE_1(State_H_C4_F, C4*);
MAKE_STATE_1(State_H_C5_F, C5*);
MAKE_STATE_1(State_H_C6_F, C6*);
MAKE_STATE_1(State_H_C7_F, C7*);
MAKE_STATE_1(State_H_C8_F, C8*);
MAKE_STATE_1(State_H_C1_G, C1*);
MAKE_STATE_1(State_H_C2_G, C2*);
MAKE_STATE_1(State_H_C3_G, C3*);
MAKE_STATE_1(State_H_C4_G, C4*);
MAKE_STATE_1(State_H_C5_G, C5*);
MAKE_STATE_1(State_H_C6_G, C6*);
MAKE_STATE_1(State_H_C7_G, C7*);
MAKE_STATE_1(State_H_C8_G, C8*);
MAKE_STATE_1(State_H2_C4_G, C4*);
int g_TestID;
struct C1
{
virtual void F()
{
ADD_STATE(State_C1_F(this));
}
virtual int G()
{
ADD_STATE(State_C1_G(this));
return 1;
}
};
struct C2
{
virtual void F()
{
ADD_STATE(State_C2_F(this));
}
virtual int G()
{
ADD_STATE(State_C2_G(this));
return 2;
}
};
struct C3
{
virtual void F()
{
ADD_STATE(State_C3_F(this));
}
virtual int G()
{
ADD_STATE(State_C3_G(this));
return 3;
}
};
struct C4
{
virtual void F()
{
ADD_STATE(State_C4_F(this));
}
virtual int G()
{
ADD_STATE(State_C4_G(this));
return 4;
}
};
struct C5
{
virtual void F()
{
ADD_STATE(State_C5_F(this));
}
virtual int G()
{
ADD_STATE(State_C5_G(this));
return 5;
}
};
struct C6
{
virtual void F()
{
ADD_STATE(State_C6_F(this));
}
virtual int G()
{
ADD_STATE(State_C6_G(this));
return 6;
}
};
struct C7
{
virtual void F()
{
ADD_STATE(State_C7_F(this));
}
virtual int G()
{
ADD_STATE(State_C7_G(this));
return 7;
}
};
struct C8
{
virtual void F()
{
ADD_STATE(State_C8_F(this));
}
virtual int G()
{
ADD_STATE(State_C8_G(this));
return 8;
}
};
SH_DECL_HOOK0_void(C1, F, SH_NOATTRIB, 0);
SH_DECL_HOOK0(C1, G, SH_NOATTRIB, 0, int);
SH_DECL_HOOK0_void(C2, F, SH_NOATTRIB, 0);
SH_DECL_HOOK0(C2, G, SH_NOATTRIB, 0, int);
SH_DECL_HOOK0_void(C3, F, SH_NOATTRIB, 0);
SH_DECL_HOOK0(C3, G, SH_NOATTRIB, 0, int);
SH_DECL_HOOK0_void(C4, F, SH_NOATTRIB, 0);
SH_DECL_HOOK0(C4, G, SH_NOATTRIB, 0, int);
SH_DECL_HOOK0_void(C5, F, SH_NOATTRIB, 0);
SH_DECL_HOOK0(C5, G, SH_NOATTRIB, 0, int);
SH_DECL_HOOK0_void(C6, F, SH_NOATTRIB, 0);
SH_DECL_HOOK0(C6, G, SH_NOATTRIB, 0, int);
SH_DECL_HOOK0_void(C7, F, SH_NOATTRIB, 0);
SH_DECL_HOOK0(C7, G, SH_NOATTRIB, 0, int);
SH_DECL_HOOK0_void(C8, F, SH_NOATTRIB, 0);
SH_DECL_HOOK0(C8, G, SH_NOATTRIB, 0, int);
void Handler_C1_F()
{
ADD_STATE(State_H_C1_F(META_IFACEPTR(C1)));
g_pC1->G();
}
int Handler_C1_G()
{
ADD_STATE(State_H_C1_G(META_IFACEPTR(C1)));
g_pC2->F();
return 1;
}
void Handler_C2_F()
{
ADD_STATE(State_H_C2_F(META_IFACEPTR(C2)));
g_pC2->G();
}
int Handler_C2_G()
{
ADD_STATE(State_H_C2_G(META_IFACEPTR(C2)));
g_pC3->F();
return 2;
}
void Handler_C3_F()
{
ADD_STATE(State_H_C3_F(META_IFACEPTR(C3)));
g_pC3->G();
}
int Handler_C3_G()
{
ADD_STATE(State_H_C3_G(META_IFACEPTR(C3)));
g_pC4->F();
return 3;
}
void Handler_C4_F()
{
ADD_STATE(State_H_C4_F(META_IFACEPTR(C4)));
g_pC4->G();
}
int Handler_C4_G()
{
ADD_STATE(State_H_C4_G(META_IFACEPTR(C4)));
g_pC5->F();
return 4;
}
int Handler2_C4_G()
{
ADD_STATE(State_H2_C4_G(META_IFACEPTR(C4)));
return 4;
}
void Handler_C5_F()
{
ADD_STATE(State_H_C5_F(META_IFACEPTR(C5)));
g_pC5->G();
}
int Handler_C5_G()
{
ADD_STATE(State_H_C5_G(META_IFACEPTR(C5)));
g_pC6->F();
return 5;
}
void Handler_C6_F()
{
ADD_STATE(State_H_C6_F(META_IFACEPTR(C6)));
g_pC6->G();
}
int Handler_C6_G()
{
ADD_STATE(State_H_C6_G(META_IFACEPTR(C6)));
g_pC7->F();
return 6;
}
void Handler_C7_F()
{
if (g_TestID == 1 || g_TestID == 2)
SH_REMOVE_HOOK_STATICFUNC(C4, G, g_pC4, Handler2_C4_G, false);
if (g_TestID == 2)
SH_REMOVE_HOOK_STATICFUNC(C4, G, g_pC4, Handler_C4_G, false);
ADD_STATE(State_H_C7_F(META_IFACEPTR(C7)));
g_pC7->G();
}
int Handler_C7_G()
{
ADD_STATE(State_H_C7_G(META_IFACEPTR(C7)));
g_pC8->F();
return 7;
}
void Handler_C8_F()
{
ADD_STATE(State_H_C8_F(META_IFACEPTR(C8)));
g_pC8->G();
}
int Handler_C8_G()
{
ADD_STATE(State_H_C8_G(META_IFACEPTR(C8)));
return 8;
}
C1 g_C1;
C2 g_C2;
C3 g_C3;
C4 g_C4;
C5 g_C5;
C6 g_C6;
C7 g_C7;
C8 g_C8;
}
bool TestReentr(std::string &error)
{
GET_SHPTR(g_SHPtr);
g_PLID = 1337;
g_pC1 = &g_C1;
g_pC2 = &g_C2;
g_pC3 = &g_C3;
g_pC4 = &g_C4;
g_pC5 = &g_C5;
g_pC6 = &g_C6;
g_pC7 = &g_C7;
g_pC8 = &g_C8;
SH_ADD_HOOK_STATICFUNC(C1, F, g_pC1, Handler_C1_F, false);
SH_ADD_HOOK_STATICFUNC(C1, G, g_pC1, Handler_C1_G, false);
SH_ADD_HOOK_STATICFUNC(C2, F, g_pC2, Handler_C2_F, false);
SH_ADD_HOOK_STATICFUNC(C2, G, g_pC2, Handler_C2_G, false);
SH_ADD_HOOK_STATICFUNC(C3, F, g_pC3, Handler_C3_F, false);
SH_ADD_HOOK_STATICFUNC(C3, G, g_pC3, Handler_C3_G, false);
SH_ADD_HOOK_STATICFUNC(C4, F, g_pC4, Handler_C4_F, false);
SH_ADD_HOOK_STATICFUNC(C4, G, g_pC4, Handler_C4_G, false);
SH_ADD_HOOK_STATICFUNC(C5, F, g_pC5, Handler_C5_F, false);
SH_ADD_HOOK_STATICFUNC(C5, G, g_pC5, Handler_C5_G, false);
SH_ADD_HOOK_STATICFUNC(C6, F, g_pC6, Handler_C6_F, false);
SH_ADD_HOOK_STATICFUNC(C6, G, g_pC6, Handler_C6_G, false);
SH_ADD_HOOK_STATICFUNC(C7, F, g_pC7, Handler_C7_F, false);
SH_ADD_HOOK_STATICFUNC(C7, G, g_pC7, Handler_C7_G, false);
SH_ADD_HOOK_STATICFUNC(C8, F, g_pC8, Handler_C8_F, false);
SH_ADD_HOOK_STATICFUNC(C8, G, g_pC8, Handler_C8_G, false);
g_pC1->F();
CHECK_STATES((&g_States,
new State_H_C1_F(g_pC1),
new State_H_C1_G(g_pC1),
new State_H_C2_F(g_pC2),
new State_H_C2_G(g_pC2),
new State_H_C3_F(g_pC3),
new State_H_C3_G(g_pC3),
new State_H_C4_F(g_pC4),
new State_H_C4_G(g_pC4),
new State_H_C5_F(g_pC5),
new State_H_C5_G(g_pC5),
new State_H_C6_F(g_pC6),
new State_H_C6_G(g_pC6),
new State_H_C7_F(g_pC7),
new State_H_C7_G(g_pC7),
new State_H_C8_F(g_pC8),
new State_H_C8_G(g_pC8),
new State_C8_G(g_pC8),
new State_C8_F(g_pC8),
new State_C7_G(g_pC7),
new State_C7_F(g_pC7),
new State_C6_G(g_pC6),
new State_C6_F(g_pC6),
new State_C5_G(g_pC5),
new State_C5_F(g_pC5),
new State_C4_G(g_pC4),
new State_C4_F(g_pC4),
new State_C3_G(g_pC3),
new State_C3_F(g_pC3),
new State_C2_G(g_pC2),
new State_C2_F(g_pC2),
new State_C1_G(g_pC1),
new State_C1_F(g_pC1),
NULL), "1");
SH_ADD_HOOK_STATICFUNC(C4, G, g_pC4, Handler2_C4_G, false);
g_pC1->F();
CHECK_STATES((&g_States,
new State_H_C1_F(g_pC1),
new State_H_C1_G(g_pC1),
new State_H_C2_F(g_pC2),
new State_H_C2_G(g_pC2),
new State_H_C3_F(g_pC3),
new State_H_C3_G(g_pC3),
new State_H_C4_F(g_pC4),
new State_H_C4_G(g_pC4),
new State_H_C5_F(g_pC5),
new State_H_C5_G(g_pC5),
new State_H_C6_F(g_pC6),
new State_H_C6_G(g_pC6),
new State_H_C7_F(g_pC7),
new State_H_C7_G(g_pC7),
new State_H_C8_F(g_pC8),
new State_H_C8_G(g_pC8),
new State_C8_G(g_pC8),
new State_C8_F(g_pC8),
new State_C7_G(g_pC7),
new State_C7_F(g_pC7),
new State_C6_G(g_pC6),
new State_C6_F(g_pC6),
new State_C5_G(g_pC5),
new State_C5_F(g_pC5),
new State_H2_C4_G(g_pC4),
new State_C4_G(g_pC4),
new State_C4_F(g_pC4),
new State_C3_G(g_pC3),
new State_C3_F(g_pC3),
new State_C2_G(g_pC2),
new State_C2_F(g_pC2),
new State_C1_G(g_pC1),
new State_C1_F(g_pC1),
NULL), "2");
g_pC1->F();
CHECK_STATES((&g_States,
new State_H_C1_F(g_pC1),
new State_H_C1_G(g_pC1),
new State_H_C2_F(g_pC2),
new State_H_C2_G(g_pC2),
new State_H_C3_F(g_pC3),
new State_H_C3_G(g_pC3),
new State_H_C4_F(g_pC4),
new State_H_C4_G(g_pC4),
new State_H_C5_F(g_pC5),
new State_H_C5_G(g_pC5),
new State_H_C6_F(g_pC6),
new State_H_C6_G(g_pC6),
new State_H_C7_F(g_pC7),
new State_H_C7_G(g_pC7),
new State_H_C8_F(g_pC8),
new State_H_C8_G(g_pC8),
new State_C8_G(g_pC8),
new State_C8_F(g_pC8),
new State_C7_G(g_pC7),
new State_C7_F(g_pC7),
new State_C6_G(g_pC6),
new State_C6_F(g_pC6),
new State_C5_G(g_pC5),
new State_C5_F(g_pC5),
new State_H2_C4_G(g_pC4),
new State_C4_G(g_pC4),
new State_C4_F(g_pC4),
new State_C3_G(g_pC3),
new State_C3_F(g_pC3),
new State_C2_G(g_pC2),
new State_C2_F(g_pC2),
new State_C1_G(g_pC1),
new State_C1_F(g_pC1),
NULL), "3");
g_TestID = 1;
g_pC1->F();
CHECK_STATES((&g_States,
new State_H_C1_F(g_pC1),
new State_H_C1_G(g_pC1),
new State_H_C2_F(g_pC2),
new State_H_C2_G(g_pC2),
new State_H_C3_F(g_pC3),
new State_H_C3_G(g_pC3),
new State_H_C4_F(g_pC4),
new State_H_C4_G(g_pC4),
new State_H_C5_F(g_pC5),
new State_H_C5_G(g_pC5),
new State_H_C6_F(g_pC6),
new State_H_C6_G(g_pC6),
new State_H_C7_F(g_pC7),
new State_H_C7_G(g_pC7),
new State_H_C8_F(g_pC8),
new State_H_C8_G(g_pC8),
new State_C8_G(g_pC8),
new State_C8_F(g_pC8),
new State_C7_G(g_pC7),
new State_C7_F(g_pC7),
new State_C6_G(g_pC6),
new State_C6_F(g_pC6),
new State_C5_G(g_pC5),
new State_C5_F(g_pC5),
new State_C4_G(g_pC4),
new State_C4_F(g_pC4),
new State_C3_G(g_pC3),
new State_C3_F(g_pC3),
new State_C2_G(g_pC2),
new State_C2_F(g_pC2),
new State_C1_G(g_pC1),
new State_C1_F(g_pC1),
NULL), "4");
g_pC1->F();
CHECK_STATES((&g_States,
new State_H_C1_F(g_pC1),
new State_H_C1_G(g_pC1),
new State_H_C2_F(g_pC2),
new State_H_C2_G(g_pC2),
new State_H_C3_F(g_pC3),
new State_H_C3_G(g_pC3),
new State_H_C4_F(g_pC4),
new State_H_C4_G(g_pC4),
new State_H_C5_F(g_pC5),
new State_H_C5_G(g_pC5),
new State_H_C6_F(g_pC6),
new State_H_C6_G(g_pC6),
new State_H_C7_F(g_pC7),
new State_H_C7_G(g_pC7),
new State_H_C8_F(g_pC8),
new State_H_C8_G(g_pC8),
new State_C8_G(g_pC8),
new State_C8_F(g_pC8),
new State_C7_G(g_pC7),
new State_C7_F(g_pC7),
new State_C6_G(g_pC6),
new State_C6_F(g_pC6),
new State_C5_G(g_pC5),
new State_C5_F(g_pC5),
new State_C4_G(g_pC4),
new State_C4_F(g_pC4),
new State_C3_G(g_pC3),
new State_C3_F(g_pC3),
new State_C2_G(g_pC2),
new State_C2_F(g_pC2),
new State_C1_G(g_pC1),
new State_C1_F(g_pC1),
NULL), "5");
SH_ADD_HOOK_STATICFUNC(C4, G, g_pC4, Handler2_C4_G, false);
g_TestID = 2;
g_pC1->F();
CHECK_STATES((&g_States,
new State_H_C1_F(g_pC1),
new State_H_C1_G(g_pC1),
new State_H_C2_F(g_pC2),
new State_H_C2_G(g_pC2),
new State_H_C3_F(g_pC3),
new State_H_C3_G(g_pC3),
new State_H_C4_F(g_pC4),
new State_H_C4_G(g_pC4),
new State_H_C5_F(g_pC5),
new State_H_C5_G(g_pC5),
new State_H_C6_F(g_pC6),
new State_H_C6_G(g_pC6),
new State_H_C7_F(g_pC7),
new State_H_C7_G(g_pC7),
new State_H_C8_F(g_pC8),
new State_H_C8_G(g_pC8),
new State_C8_G(g_pC8),
new State_C8_F(g_pC8),
new State_C7_G(g_pC7),
new State_C7_F(g_pC7),
new State_C6_G(g_pC6),
new State_C6_F(g_pC6),
new State_C5_G(g_pC5),
new State_C5_F(g_pC5),
new State_C4_G(g_pC4),
new State_C4_F(g_pC4),
new State_C3_G(g_pC3),
new State_C3_F(g_pC3),
new State_C2_G(g_pC2),
new State_C2_F(g_pC2),
new State_C1_G(g_pC1),
new State_C1_F(g_pC1),
NULL), "6");
g_pC1->F();
CHECK_STATES((&g_States,
new State_H_C1_F(g_pC1),
new State_H_C1_G(g_pC1),
new State_H_C2_F(g_pC2),
new State_H_C2_G(g_pC2),
new State_H_C3_F(g_pC3),
new State_H_C3_G(g_pC3),
new State_H_C4_F(g_pC4),
new State_C4_G(g_pC4),
new State_C4_F(g_pC4),
new State_C3_G(g_pC3),
new State_C3_F(g_pC3),
new State_C2_G(g_pC2),
new State_C2_F(g_pC2),
new State_C1_G(g_pC1),
new State_C1_F(g_pC1),
NULL), "7");
SH_REMOVE_HOOK_STATICFUNC(C1, F, g_pC1, Handler_C1_F, false);
SH_REMOVE_HOOK_STATICFUNC(C1, G, g_pC1, Handler_C1_G, false);
SH_REMOVE_HOOK_STATICFUNC(C2, F, g_pC2, Handler_C2_F, false);
SH_REMOVE_HOOK_STATICFUNC(C2, G, g_pC2, Handler_C2_G, false);
SH_REMOVE_HOOK_STATICFUNC(C3, F, g_pC3, Handler_C3_F, false);
SH_REMOVE_HOOK_STATICFUNC(C3, G, g_pC3, Handler_C3_G, false);
SH_REMOVE_HOOK_STATICFUNC(C4, F, g_pC4, Handler_C4_F, false);
SH_REMOVE_HOOK_STATICFUNC(C4, G, g_pC4, Handler_C4_G, false);
SH_REMOVE_HOOK_STATICFUNC(C5, F, g_pC5, Handler_C5_F, false);
SH_REMOVE_HOOK_STATICFUNC(C5, G, g_pC5, Handler_C5_G, false);
SH_REMOVE_HOOK_STATICFUNC(C6, F, g_pC6, Handler_C6_F, false);
SH_REMOVE_HOOK_STATICFUNC(C6, G, g_pC6, Handler_C6_G, false);
SH_REMOVE_HOOK_STATICFUNC(C7, F, g_pC7, Handler_C7_F, false);
SH_REMOVE_HOOK_STATICFUNC(C7, G, g_pC7, Handler_C7_G, false);
SH_REMOVE_HOOK_STATICFUNC(C8, F, g_pC8, Handler_C8_F, false);
SH_REMOVE_HOOK_STATICFUNC(C8, G, g_pC8, Handler_C8_G, false);
return true;
}