Skip to content

Commit e15241b

Browse files
mixern6mixern6
committed
Add new proto info
1 parent 28cb3df commit e15241b

File tree

5 files changed

+205
-36
lines changed

5 files changed

+205
-36
lines changed

‎core/sourcehook/sourcehook.h‎

Lines changed: 143 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,20 @@ namespace SourceHook
184184
*/
185185
struct PassInfo
186186
{
187+
PassInfo()
188+
: size(0)
189+
, type(0)
190+
, flags(0)
191+
{
192+
}
193+
194+
PassInfo(size_t typeSize, int typeId, unsigned int typeFlags)
195+
: size(typeSize)
196+
, type(typeId)
197+
, flags(typeFlags)
198+
{
199+
}
200+
187201
enum PassType
188202
{
189203
PassType_Unknown=0, /**< Unknown -- no extra info available */
@@ -217,6 +231,22 @@ namespace SourceHook
217231

218232
struct V2Info
219233
{
234+
V2Info()
235+
: pNormalCtor(nullptr)
236+
, pCopyCtor(nullptr)
237+
, pDtor(nullptr)
238+
, pAssignOperator(nullptr)
239+
{
240+
}
241+
242+
V2Info(void *normalCtor, void *copyCtor, void *dtor, void *assignOperator)
243+
: pNormalCtor(normalCtor)
244+
, pCopyCtor(copyCtor)
245+
, pDtor(dtor)
246+
, pAssignOperator(assignOperator)
247+
{
248+
}
249+
220250
void *pNormalCtor;
221251
void *pCopyCtor;
222252
void *pDtor;
@@ -252,6 +282,19 @@ namespace SourceHook
252282
const PassInfo::V2Info *paramsPassInfo2;
253283
};
254284

285+
class IProtoInfo
286+
{
287+
public:
288+
virtual ~IProtoInfo() = default;
289+
virtual int GetNumOfParams() const = 0;
290+
virtual const PassInfo &GetRetPassInfo() const = 0;
291+
virtual const PassInfo *GetParamsPassInfo() const = 0;
292+
virtual int GetConvention() const = 0;
293+
virtual int GetVersion() const = 0;
294+
virtual const PassInfo::V2Info &GetRetPassInfo2() const = 0;
295+
virtual const PassInfo::V2Info *GetParamsPassInfo2() const = 0;
296+
};
297+
255298
struct IHookManagerInfo;
256299

257300
/**
@@ -409,6 +452,9 @@ namespace SourceHook
409452
{
410453
virtual void SetInfo(int hookman_version, int vtbloffs, int vtblidx,
411454
ProtoInfo *proto, void *hookfunc_vfnptr) = 0;
455+
456+
virtual void SetInfo(int hookman_version, int vtbloffs, int vtblidx,
457+
IProtoInfo *proto, void *hookfunc_vfnptr) = 0;
412458
};
413459

414460
// I'm adding support for functions which return references.
@@ -5130,12 +5176,17 @@ class PassInfoInitializer
51305176
InitializePassInfo<sizeof...(ArgsType), 0, ArgsType...>();
51315177
}
51325178

5133-
const PassInfo *ParamsPassInfo() const
5179+
const PassInfo *GetParamsPassInfo() const
51345180
{
51355181
return params_;
51365182
}
51375183

5138-
const size_t ParamsPassInfoSize() const
5184+
const PassInfo::V2Info *GetParamsPassInfoV2() const
5185+
{
5186+
return paramsV2_;
5187+
}
5188+
5189+
const size_t GetParamsPassInfoSize() const
51395190
{
51405191
return sizeof...(ArgsType);
51415192
}
@@ -5155,6 +5206,32 @@ class PassInfoInitializer
51555206
}
51565207

51575208
PassInfo params_[sizeof...(ArgsType)];
5209+
PassInfo::V2Info paramsV2_[sizeof...(ArgsType)];
5210+
};
5211+
5212+
// For zero arguments
5213+
template<>
5214+
class PassInfoInitializer<>
5215+
{
5216+
public:
5217+
constexpr PassInfoInitializer()
5218+
{
5219+
}
5220+
5221+
const PassInfo *GetParamsPassInfo() const
5222+
{
5223+
return nullptr;
5224+
}
5225+
5226+
const PassInfo::V2Info *GetParamsPassInfoV2() const
5227+
{
5228+
return nullptr;
5229+
}
5230+
5231+
const size_t GetParamsPassInfoSize() const
5232+
{
5233+
return 0;
5234+
}
51585235
};
51595236

51605237
template <typename T>
@@ -5181,7 +5258,7 @@ struct ReturnTypeInfo<void>
51815258
{
51825259
static const size_t size()
51835260
{
5184-
return 0; // why isn't it sizeof(void) like in TypeInfo<T>?
5261+
return 0;
51855262
}
51865263

51875264
static const int type()
@@ -5222,6 +5299,65 @@ class CHookManagerMemberFuncHandler : public IHookManagerMemberFunc
52225299
HookManagerMemberFunc func_;
52235300
};
52245301

5302+
template<typename ReturnType, class ... Params>
5303+
class CProtoInfo : public IProtoInfo
5304+
{
5305+
public:
5306+
constexpr CProtoInfo()
5307+
: retPassInfo(ReturnTypeInfo<ReturnType>::size(),
5308+
ReturnTypeInfo<ReturnType>::type(),
5309+
ReturnTypeInfo<ReturnType>::flags())
5310+
{
5311+
}
5312+
5313+
virtual ~CProtoInfo() override
5314+
{
5315+
}
5316+
5317+
virtual int GetNumOfParams() const override
5318+
{
5319+
return paramsPassInfo.GetParamsPassInfoSize();
5320+
}
5321+
5322+
virtual const PassInfo &GetRetPassInfo() const override
5323+
{
5324+
return retPassInfo;
5325+
}
5326+
5327+
virtual const PassInfo *GetParamsPassInfo() const override
5328+
{
5329+
return paramsPassInfo.GetParamsPassInfo();
5330+
}
5331+
5332+
virtual int GetConvention() const override
5333+
{
5334+
return 0;
5335+
}
5336+
5337+
// version of the ProtoInfo structure.
5338+
virtual int GetVersion() const override
5339+
{
5340+
// 1 for Version2
5341+
return 1;
5342+
}
5343+
5344+
virtual const PassInfo::V2Info &GetRetPassInfo2() const override
5345+
{
5346+
return retPassInfo2;
5347+
}
5348+
5349+
virtual const PassInfo::V2Info *GetParamsPassInfo2() const override
5350+
{
5351+
return paramsPassInfo.GetParamsPassInfoV2();
5352+
}
5353+
5354+
private:
5355+
int numOfParams; //!< number of parameters
5356+
PassInfo retPassInfo; //!< PassInfo for the return value. size=0 -> no retval
5357+
PassInfo::V2Info retPassInfo2; //!< Version2 only
5358+
PassInfoInitializer<Params...> paramsPassInfo; //!< PassInfos for the parameters
5359+
};
5360+
52255361
template<typename ReturnType, class ... Params>
52265362
class ManualHookHandler : public CHookManagerMemberFuncHandler<ManualHookHandler<ReturnType, Params...>>
52275363
{
@@ -5237,40 +5373,18 @@ class ManualHookHandler : public CHookManagerMemberFuncHandler<ManualHookHandler
52375373
typedef ReturnType (T::*HookFunc)(Params...);
52385374
};
52395375

5240-
ManualHookHandler()
5376+
constexpr ManualHookHandler()
52415377
: CHookManagerMemberFuncHandler<ThisType>(this, &ThisType::HookManPubFunc)
5242-
, thisPointerOffset_(0)
5243-
, vTableIndex_(0)
5244-
, vtableOffset_(0)
52455378
, msMFI_{false, 0, 0, 0}
52465379
, msHI_(nullptr)
5247-
, msProto_{sizeof...(Params),
5248-
{ReturnTypeInfo<ReturnType>::size(), ReturnTypeInfo<ReturnType>::type(), ReturnTypeInfo<ReturnType>::flags()},
5249-
paramInfosM_.ParamsPassInfo(),
5250-
0,
5251-
__SH_EPI,
5252-
paramInfos2M_}
52535380
{
5254-
for(PassInfo::V2Info& paramInfo : paramInfos2M_)
5255-
{
5256-
paramInfo = __SH_EPI;
5257-
}
52585381
}
52595382

52605383
virtual ~ManualHookHandler()
52615384
{
52625385
//TODO apply RAII, cleanup all related hooks here
52635386
}
52645387

5265-
// TODO probably not needed for manual hooks
5266-
void Reconfigure()
5267-
{
5268-
msMFI_.isVirtual = true;
5269-
msMFI_.thisptroffs = thisPointerOffset_;
5270-
msMFI_.vtblindex = vTableIndex_;
5271-
msMFI_.vtbloffs = vtableOffset_;
5272-
}
5273-
52745388
void Reconfigure(int vtblindex, int vtbloffs = 0, int thisptroffs = 0)
52755389
{
52765390
g_SHPtr->RemoveHookManager(g_PLID, this);
@@ -5285,7 +5399,7 @@ class ManualHookHandler : public CHookManagerMemberFuncHandler<ManualHookHandler
52855399
typename ManualHookHandler::FD handler(callbackInstPtr, callbackFuncPtr);
52865400
typename ThisType::CMyDelegateImpl* tmp = new typename ThisType::CMyDelegateImpl(handler); // TODO use unique_ptr here
52875401

5288-
return g_SHPtr->AddHook(g_PLID, mode, iface, thisPointerOffset_, this, tmp, post);
5402+
return g_SHPtr->AddHook(g_PLID, mode, iface, 0, this, tmp, post);
52895403
}
52905404

52915405
template<typename T>
@@ -5294,7 +5408,7 @@ class ManualHookHandler : public CHookManagerMemberFuncHandler<ManualHookHandler
52945408
typename ManualHookHandler::FD handler(callbackInstPtr, callbackFuncPtr);
52955409
typename ThisType::CMyDelegateImpl tmp(handler);
52965410

5297-
return g_SHPtr->RemoveHook(g_PLID, iface, thisPointerOffset_, this, &tmp, post);
5411+
return g_SHPtr->RemoveHook(g_PLID, iface, 0, this, &tmp, post);
52985412
}
52995413

53005414
// For void return type only
@@ -5557,16 +5671,10 @@ class ManualHookHandler : public CHookManagerMemberFuncHandler<ManualHookHandler
55575671
}
55585672

55595673
private:
5560-
int thisPointerOffset_; // thisptroffs
5561-
int vTableIndex_; // vtblindex
5562-
int vtableOffset_; // vtbloffs
5563-
55645674
MemFuncInfo msMFI_; // ms_MFI
55655675
IHookManagerInfo *msHI_; // ms_HI
55665676

5567-
PassInfoInitializer<void, Params...> paramInfosM_; // ParamInfosM
5568-
PassInfo::V2Info paramInfos2M_[sizeof...(Params) + 1]; // ParamInfos2M
5569-
ProtoInfo msProto_; // ms_Proto
5677+
CProtoInfo<ReturnType, Params...> msProto_; // ms_Proto
55705678
};
55715679

55725680
} // SourceHook

‎core/sourcehook/sourcehook_impl_chookmaninfo.cpp‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,16 @@ namespace SourceHook
3737
m_HookfuncVfnptr = hookfunc_vfnptr;
3838
}
3939

40+
void CHookManager::SetInfo(int hookman_version, int vtbloffs, int vtblidx,
41+
IProtoInfo *proto, void *hookfunc_vfnptr)
42+
{
43+
m_Version = hookman_version;
44+
m_VtblOffs = vtbloffs;
45+
m_VtblIdx = vtblidx;
46+
m_Proto = proto;
47+
m_HookfuncVfnptr = hookfunc_vfnptr;
48+
}
49+
4050
void CHookManager::Register()
4151
{
4252
m_PubFunc(true, this);

‎core/sourcehook/sourcehook_impl_chookmaninfo.h‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,9 @@ namespace SourceHook
7474
// *** IHookManagerInfo interface ***
7575
void SetInfo(int hookman_version, int vtbloffs, int vtblidx,
7676
ProtoInfo *proto, void *hookfunc_vfnptr);
77+
78+
void SetInfo(int hookman_version, int vtbloffs, int vtblidx,
79+
IProtoInfo *proto, void *hookfunc_vfnptr);
7780
};
7881

7982
class CHookManList : public List<CHookManager>

‎core/sourcehook/sourcehook_impl_cproto.cpp‎

Lines changed: 42 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,6 @@ namespace SourceHook
3737
m_RetPassInfo.pDtor = NULL;
3838
m_RetPassInfo.pAssignOperator = NULL;
3939

40-
4140
m_ParamsPassInfo.resize(pProto->numOfParams);
4241

4342
for (int i = 1; i <= pProto->numOfParams; ++i)
@@ -89,6 +88,48 @@ namespace SourceHook
8988
}
9089
}
9190

91+
void CProto::Fill(const IProtoInfo *pProto)
92+
{
93+
if (pProto == nullptr)
94+
m_Version = -1;
95+
96+
m_ParamsPassInfo.clear();
97+
98+
// Version 1 if 0
99+
// Version 2 if 1
100+
m_Version = pProto->GetVersion();
101+
m_Convention = pProto->GetConvention();
102+
m_NumOfParams = pProto->GetNumOfParams();
103+
104+
const PassInfo& retPassInfo = pProto->GetRetPassInfo();
105+
106+
m_RetPassInfo.size = retPassInfo.size;
107+
m_RetPassInfo.type = retPassInfo.type;
108+
m_RetPassInfo.flags = retPassInfo.flags;
109+
110+
const PassInfo::V2Info& retPassInfo2 = pProto->GetRetPassInfo2();
111+
m_RetPassInfo.pNormalCtor = retPassInfo2.pNormalCtor; // should be nullptr for Version 1
112+
m_RetPassInfo.pCopyCtor = retPassInfo2.pCopyCtor; // should be nullptr for Version 1
113+
m_RetPassInfo.pDtor = retPassInfo2.pDtor; // should be nullptr for Version 1
114+
m_RetPassInfo.pAssignOperator = retPassInfo2.pAssignOperator; // should be nullptr for Version 1
115+
116+
m_ParamsPassInfo.resize(pProto->GetNumOfParams());
117+
118+
const PassInfo* paramsPassInfo = pProto->GetParamsPassInfo();
119+
const PassInfo::V2Info* paramsPassInfo2 = pProto->GetParamsPassInfo2();
120+
for (size_t i = 0; i != pProto->GetNumOfParams(); ++i)
121+
{
122+
m_ParamsPassInfo[i].size = paramsPassInfo[i].size;
123+
m_ParamsPassInfo[i].type = paramsPassInfo[i].type;
124+
m_ParamsPassInfo[i].flags = paramsPassInfo[i].flags;
125+
126+
m_ParamsPassInfo[i].pNormalCtor = paramsPassInfo2[i].pNormalCtor; // should be nullptr for Version 1
127+
m_ParamsPassInfo[i].pCopyCtor = paramsPassInfo2[i].pCopyCtor; // should be nullptr for Version 1
128+
m_ParamsPassInfo[i].pDtor = paramsPassInfo2[i].pDtor; // should be nullptr for Version 1
129+
m_ParamsPassInfo[i].pAssignOperator = paramsPassInfo2[i].pAssignOperator; // should be nullptr for Version 1
130+
}
131+
}
132+
92133
// Basic compat test
93134
// Other than this, we assume that the plugins know what they're doing
94135
bool CProto::operator == (const CProto &other) const

‎core/sourcehook/sourcehook_impl_cproto.h‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,8 @@ namespace SourceHook
5353

5454
void Fill(const ProtoInfo *pProto);
5555

56+
void Fill(const IProtoInfo *pProto);
57+
5658
// For old sourcehook.h: flags 0 -> assume ByVal
5759
static unsigned int GetRealFlags(const PassInfo &info)
5860
{
@@ -69,6 +71,11 @@ namespace SourceHook
6971
Fill(pProto);
7072
}
7173

74+
CProto(const IProtoInfo *pProto)
75+
{
76+
Fill(pProto);
77+
}
78+
7279
CProto(const CProto &other) : m_Version(other.m_Version), m_NumOfParams(other.m_NumOfParams),
7380
m_RetPassInfo(other.m_RetPassInfo), m_ParamsPassInfo(other.m_ParamsPassInfo),
7481
m_Convention(other.m_Convention)

0 commit comments

Comments
 (0)