1
0
mirror of https://github.com/alliedmodders/metamod-source.git synced 2024-12-13 01:08:51 +01:00
HLMetaModOfficial/core-legacy/sourcehook/sourcehook.cpp

1464 lines
42 KiB
C++
Raw Normal View History

/* ======== SourceHook ========
* Copyright (C) 2004-2007 Metamod:Source Development Team
* No warranties of any kind
*
* License: zlib/libpng
*
* Author(s): Pavol "PM OnoTo" Marko
* Contributors: Scott "Damaged Soul" Ehlert
* ============================
*/
/**
* @file sourcehook.cpp
* @brief Contains the implementation of the SourceHook API
*/
#if defined __GNUC__
#include <stdint.h>
#endif
#include "sourcehook_impl.h"
#include "sh_memory.h"
namespace SourceHook
{
template<>
int HashFunction<int>(const int & k)
{
return k;
}
template<>
int Compare<int>(const int & k1, const int & k2)
{
if (k1 == k2)
return 0;
if (k1 > k2)
return 1;
if (k1 < k2)
return -1;
return 0;
}
CSourceHookImpl::CSourceHookImpl() : m_OneIgnore(NULL), m_IgnoreActive(false)
{
}
CSourceHookImpl::~CSourceHookImpl()
{
}
int CSourceHookImpl::GetIfaceVersion()
{
return SH_IFACE_VERSION;
}
int CSourceHookImpl::GetImplVersion()
{
return SH_IMPL_VERSION;
}
bool CSourceHookImpl::IsPluginInUse(Plugin plug)
{
// Iterate through all hook managers which are in this plugin
// Iterate through their vfnptrs, ifaces, hooks
// If a hook from an other plugin is found, return true
// Return false otherwise
#define TMP_CHECK_LIST(name) \
for (hook_iter = iface_iter->name.m_List.begin(); hook_iter != iface_iter->name.m_List.end(); ++hook_iter) \
if (hook_iter->plug == plug) \
return true;
for (HookManContList::iterator hmcl_iter = m_HookMans.begin();
hmcl_iter != m_HookMans.end(); ++hmcl_iter)
{
for (CHookManagerContainer::iterator hmil_iter = hmcl_iter->begin();
hmil_iter != hmcl_iter->end(); ++hmil_iter)
{
if (hmil_iter->m_Plug != plug)
continue;
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = hmil_iter->m_VfnPtrs.begin();
vfnptr_iter != hmil_iter->m_VfnPtrs.end(); ++vfnptr_iter)
{
for (CVfnPtr::IfaceListIter iface_iter = vfnptr_iter->m_Ifaces.begin();
iface_iter != vfnptr_iter->m_Ifaces.end(); ++iface_iter)
{
List<HookInfo>::iterator hook_iter;
TMP_CHECK_LIST(m_PreHooks);
TMP_CHECK_LIST(m_PostHooks);
}
}
}
}
#undef TMP_CHECK_LIST
return false;
}
void CSourceHookImpl::UnloadPlugin(Plugin plug)
{
// 1) Manually remove all hooks by this plugin
// 2) Manually remove all hook mans by this plugin
List<RemoveHookInfo> hookstoremove;
List<RemoveHookManInfo> hookmanstoremove;
HookManInfoList::iterator hmil_iter;
#define TMP_CHECK_LIST(name, ispost) \
for (hook_iter = iface_iter->name.m_List.begin(); hook_iter != iface_iter->name.m_List.end(); ++hook_iter) \
if (hook_iter->plug == plug) \
hookstoremove.push_back(RemoveHookInfo(hook_iter->plug, iface_iter->m_Ptr, \
hook_iter->thisptr_offs, hmil_iter->m_Func, hook_iter->handler, ispost)); \
for (HookManContList::iterator hmcl_iter = m_HookMans.begin();
hmcl_iter != m_HookMans.end(); ++hmcl_iter)
{
for (CHookManagerContainer::iterator hmil_iter = hmcl_iter->begin();
hmil_iter != hmcl_iter->end(); ++hmil_iter)
{
if (hmil_iter->m_Plug == plug)
hookmanstoremove.push_back(RemoveHookManInfo(plug, hmil_iter->m_Func));
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = hmil_iter->m_VfnPtrs.begin();
vfnptr_iter != hmil_iter->m_VfnPtrs.end(); ++vfnptr_iter)
{
for (CVfnPtr::IfaceListIter iface_iter = vfnptr_iter->m_Ifaces.begin();
iface_iter != vfnptr_iter->m_Ifaces.end(); ++iface_iter)
{
List<HookInfo>::iterator hook_iter;
TMP_CHECK_LIST(m_PreHooks, false);
TMP_CHECK_LIST(m_PostHooks, true);
}
}
}
}
#undef TMP_CHECK_LIST
for (List<RemoveHookInfo>::iterator rmiter = hookstoremove.begin(); rmiter != hookstoremove.end(); ++rmiter)
RemoveHook(*rmiter);
for (List<RemoveHookManInfo>::iterator rmiter = hookmanstoremove.begin(); rmiter != hookmanstoremove.end(); ++rmiter)
RemoveHookManager(*rmiter);
}
void CSourceHookImpl::RemoveHookManager(Plugin plug, HookManagerPubFunc pubFunc)
{
// Moo!
CHookManagerInfo tmp;
if (pubFunc(HA_GetInfo, &tmp) != 0)
return;
HookManContList::iterator hmcl_iter = m_HookMans.find(
CHookManagerContainer::HMCI(tmp.m_Proto, tmp.m_VtblOffs, tmp.m_VtblIdx));
if (hmcl_iter == m_HookMans.end())
{
// Moo?
return;
}
CHookManagerContainer::iterator hmil_iter = hmcl_iter->find(
CHookManagerInfo::Descriptor(plug, pubFunc));
if (hmil_iter == hmcl_iter->end())
{
// Moo?
return;
}
bool stillInUse = false;
List<RemoveHookInfo> hookstoremove;
// Iterate through all of the hook manager's hooks. Remove all hooks from the hookman's plugin.
#define TMP_CHECK_LIST(name, ispost) \
for (hook_iter = iface_iter->name.m_List.begin(); hook_iter != iface_iter->name.m_List.end(); ++hook_iter) \
if (hook_iter->plug == plug) \
hookstoremove.push_back(RemoveHookInfo(hook_iter->plug, iface_iter->m_Ptr, \
hook_iter->thisptr_offs, hmil_iter->m_Func, hook_iter->handler, ispost)); \
else \
stillInUse = true; \
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = hmil_iter->m_VfnPtrs.begin();
vfnptr_iter != hmil_iter->m_VfnPtrs.end(); ++vfnptr_iter)
{
for (CVfnPtr::IfaceListIter iface_iter = vfnptr_iter->m_Ifaces.begin();
iface_iter != vfnptr_iter->m_Ifaces.end(); ++iface_iter)
{
List<HookInfo>::iterator hook_iter;
TMP_CHECK_LIST(m_PreHooks, false);
TMP_CHECK_LIST(m_PostHooks, true);
}
}
#undef TMP_CHECK_LIST
for (List<RemoveHookInfo>::iterator rmiter = hookstoremove.begin(); rmiter != hookstoremove.end(); ++rmiter)
RemoveHook(*rmiter);
CHookManagerInfo info = *hmil_iter;
// Unlink the hook manager from the list
hmcl_iter->erase(hmil_iter);
// If there were any hooks from other plugins, find a new suitable hook manager.
if (stillInUse)
{
// Find a suitable hook manager in an other plugin
hmil_iter = hmcl_iter->begin();
// This should _never_ happen.
// If there is a hook from an other plugin, the plugin must have provided a hook manager as well.
SH_ASSERT(hmil_iter != hmcl_iter->end(),
("Could not find a suitable hook manager in an other plugin!"));
// AddHook should make sure that every plugin only has _one_ hook manager for _one_ proto/vi/vo
SH_ASSERT(hmil_iter->m_Plug != plug, ("New hook manager from same plugin!"));
// The first hook manager should be always used - so the new hook manager has to be empty
SH_ASSERT(hmil_iter->m_VfnPtrs.empty(), ("New hook manager not empty!"));
// Move the vfnptrs from the old hook manager to the new one
hmil_iter->m_VfnPtrs = info.m_VfnPtrs;
// Unregister the old one, register the new one
info.m_Func(HA_Unregister, NULL);
hmil_iter->m_Func(HA_Register, &(*hmil_iter));
// Go through all vfnptrs in this hookman and patch them to point to the new manager's handler!
// or whatever
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = hmil_iter->m_VfnPtrs.begin();
vfnptr_iter != hmil_iter->m_VfnPtrs.end(); ++vfnptr_iter)
{
// And DEREFERENCE newHookMan->m_HookfuncVfnptr!
// otherwise it will be executing the vtable... had to find out the hard way
*reinterpret_cast<void**>(vfnptr_iter->m_Ptr) = *reinterpret_cast<void**>(hmil_iter->m_HookfuncVfnptr);
}
}
}
void CSourceHookImpl::RemoveHookManager(RemoveHookManInfo info)
{
RemoveHookManager(info.plug, info.hookman);
}
void CSourceHookImpl::CompleteShutdown()
{
// Remove all hooks
List<RemoveHookInfo> hookstoremove;
#define TMP_CHECK_LIST(name, ispost) \
for (hook_iter = iface_iter->name.m_List.begin(); hook_iter != iface_iter->name.m_List.end(); ++hook_iter) \
hookstoremove.push_back(RemoveHookInfo(hook_iter->plug, iface_iter->m_Ptr, \
hook_iter->thisptr_offs, hmil_iter->m_Func, hook_iter->handler, ispost))
for (HookManContList::iterator hmcl_iter = m_HookMans.begin();
hmcl_iter != m_HookMans.end(); ++hmcl_iter)
{
for (CHookManagerContainer::iterator hmil_iter = hmcl_iter->begin();
hmil_iter != hmcl_iter->end(); ++hmil_iter)
{
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = hmil_iter->m_VfnPtrs.begin();
vfnptr_iter != hmil_iter->m_VfnPtrs.end(); ++vfnptr_iter)
{
for (CVfnPtr::IfaceListIter iface_iter = vfnptr_iter->m_Ifaces.begin();
iface_iter != vfnptr_iter->m_Ifaces.end(); ++iface_iter)
{
List<HookInfo>::iterator hook_iter;
TMP_CHECK_LIST(m_PreHooks, false);
TMP_CHECK_LIST(m_PostHooks, true);
}
}
}
}
#undef TMP_CHECK_LIST
for (List<RemoveHookInfo>::iterator rmiter = hookstoremove.begin(); rmiter != hookstoremove.end(); ++rmiter)
RemoveHook(*rmiter);
m_HookMans.clear();
}
bool CSourceHookImpl::AddHook(Plugin plug, void *iface, int thisptr_offs, HookManagerPubFunc myHookMan, ISHDelegate *handler, bool post)
{
return AddHookNew(plug, Hook_Normal, iface, thisptr_offs, myHookMan, handler, post) != 0 ? true : false;
}
int CSourceHookImpl::AddHookNew(Plugin plug, AddHookMode mode, void *iface, int thisptr_offs, HookManagerPubFunc myHookMan,
ISHDelegate *handler, bool post)
{
void *adjustediface = NULL;
void **cur_vtptr = NULL;
void *cur_vfnptr = NULL;
if (mode != Hook_Normal && mode != Hook_VP && mode != Hook_DVP)
return 0;
// 1) Get info about the hook manager
CHookManagerInfo tmp;
if (myHookMan(HA_GetInfo, &tmp) != 0)
return 0;
tmp.m_Func = myHookMan;
tmp.m_Plug = plug;
CHookManagerContainer::HMCI hmci(tmp.m_Proto, tmp.m_VtblOffs, tmp.m_VtblIdx);
// Add the container if not already added
HookManContList::iterator hmcl_iter = m_HookMans.find(hmci);
if (hmcl_iter == m_HookMans.end())
{
m_HookMans.push_back(CHookManagerContainer(hmci));
hmcl_iter = m_HookMans.end();
--hmcl_iter;
}
hmcl_iter->AddHookManager(plug, tmp);
CHookManagerContainer::iterator hookman = hmcl_iter->begin();
SH_ASSERT(hookman != hmcl_iter->end(), ("No hookman found - but we've just added one!"));
// Check whether any other container holds a hook manager which holds this vfnptr
// If yes, it means that the two are incompatible, so return false.
for (HookManContList::iterator hmcl_iter2 = m_HookMans.begin();
hmcl_iter2 != m_HookMans.end(); ++hmcl_iter2)
{
if (hmcl_iter2 != hmcl_iter)
{
for (CHookManagerContainer::iterator hmil_iter = hmcl_iter2->begin();
hmil_iter != hmcl_iter2->end(); ++hmil_iter)
{
if (hmil_iter->m_VfnPtrs.find(cur_vfnptr) != hmil_iter->m_VfnPtrs.end())
return 0;
}
}
}
// Tell it to store the pointer if it's not already active
if (hookman->m_VfnPtrs.empty())
hookman->m_Func(HA_Register, &(*hookman));
// find vfnptr
if (mode == Hook_Normal || mode == Hook_VP)
{
adjustediface = reinterpret_cast<void*>(reinterpret_cast<char*>(iface) + thisptr_offs);
cur_vtptr = *reinterpret_cast<void***>(
reinterpret_cast<char*>(adjustediface) + tmp.m_VtblOffs);
cur_vfnptr = reinterpret_cast<void*>(cur_vtptr + tmp.m_VtblIdx);
// For Hook_VP, adjustediface will be set to NULL later
// because we first have to patch callclasses (callclasses are deprecated but left in for backwards compat)
}
else if (mode == Hook_DVP)
{
adjustediface = NULL;
cur_vtptr = reinterpret_cast<void**>(iface);
cur_vfnptr = cur_vtptr + tmp.m_VtblIdx;
}
CHookManagerInfo::VfnPtrListIter vfnptr_iter = hookman->m_VfnPtrs.find(cur_vfnptr);
if (vfnptr_iter == hookman->m_VfnPtrs.end())
{
// Add a new one
CVfnPtr vfp(cur_vfnptr, &m_OneIgnore);
// Alter vtable entry
if (!MakePageWritable(cur_vfnptr))
return 0;
*reinterpret_cast<void**>(cur_vfnptr) = *reinterpret_cast<void**>(hookman->m_HookfuncVfnptr);
hookman->m_VfnPtrs.push_back(vfp);
// Make vfnptr_iter point to the new element
vfnptr_iter = hookman->m_VfnPtrs.end();
--vfnptr_iter;
// Now that it is done, check whether we have to update any callclasses
ApplyCallClassPatches(adjustediface, tmp.m_VtblOffs, tmp.m_VtblIdx, vfp.m_OrigEntry);
}
// If it's a VP hook, set adjustediface to NULL now ( see the comments at the beginning of sourcehook_impl.h )
if (mode == Hook_VP)
adjustediface = NULL;
CVfnPtr::IfaceListIter iface_iter = vfnptr_iter->m_Ifaces.find(adjustediface);
if (iface_iter == vfnptr_iter->m_Ifaces.end())
{
// Add a new one
vfnptr_iter->m_Ifaces.push_back(CIface(adjustediface));
// Make iface_iter point to the new element
iface_iter = vfnptr_iter->m_Ifaces.end();
--iface_iter;
// If this is a VP-Hook-NULL interface, go through all other interfaces of this vfnptr and tell them!
if (adjustediface == NULL)
{
for (CVfnPtr::IfaceListIter iface_iter2 = vfnptr_iter->m_Ifaces.begin();
iface_iter2 != vfnptr_iter->m_Ifaces.end(); ++iface_iter2)
{
if (*iface_iter2 != NULL)
{
iface_iter2->m_PreHooks.SetVPList(&iface_iter->m_PreHooks.m_List);
iface_iter2->m_PostHooks.SetVPList(&iface_iter->m_PostHooks.m_List);
}
}
}
else
{
// If this is a new non-VP-Hook-NULL interface, look for a non-VP-Hook-NULL interface!
for (CVfnPtr::IfaceListIter iface_iter2 = vfnptr_iter->m_Ifaces.begin();
iface_iter2 != vfnptr_iter->m_Ifaces.end(); ++iface_iter2)
{
if (*iface_iter2 == NULL)
{
iface_iter->m_PreHooks.SetVPList(&iface_iter2->m_PreHooks.m_List);
iface_iter->m_PostHooks.SetVPList(&iface_iter2->m_PostHooks.m_List);
break; // There can only be one!
}
}
}
}
// Add the hook
HookInfo hookinfo;
hookinfo.handler = handler;
hookinfo.plug = plug;
hookinfo.paused = false;
hookinfo.thisptr_offs = thisptr_offs;
hookinfo.hookid = m_HookIDMan.New(tmp.m_Proto, tmp.m_VtblOffs, tmp.m_VtblIdx, cur_vfnptr,
adjustediface, plug, thisptr_offs, handler, post);
if (post)
iface_iter->m_PostHooks.m_List.push_back(hookinfo);
else
iface_iter->m_PreHooks.m_List.push_back(hookinfo);
return hookinfo.hookid;
}
bool CSourceHookImpl::RemoveHookByID(Plugin plug, int hookid)
{
const CHookIDManager::Entry *hentry;
hentry = m_HookIDMan.QueryHook(hookid);
if (!hentry)
{
// hookid doesn't exist !
return false;
}
HookManContList::iterator hmcl_iter = m_HookMans.find(
CHookManagerContainer::HMCI(hentry->proto.GetProto(), hentry->vtbl_offs, hentry->vtbl_idx));
if (hmcl_iter == m_HookMans.end() || hmcl_iter->empty())
return false;
CHookManagerContainer::iterator hookman = hmcl_iter->begin();
// Find vfnptr
CHookManagerInfo::VfnPtrListIter vfnptr_iter = hookman->m_VfnPtrs.find(hentry->vfnptr);
if (vfnptr_iter == hookman->m_VfnPtrs.end())
return false;
// Find iface
CVfnPtr::IfaceListIter iface_iter = vfnptr_iter->m_Ifaces.find(hentry->adjustediface);
if (iface_iter == vfnptr_iter->m_Ifaces.end())
return false;
// Find hook
List<HookInfo> &hooks = hentry->post ? iface_iter->m_PostHooks.m_List : iface_iter->m_PreHooks.m_List;
List<HookInfo>::iterator hookiter = hooks.find(hookid);
if (hookiter == hooks.end())
return false;
hookiter->handler->DeleteThis();
// Move all iterators pointing at this
List<HookInfo>::iterator oldhookiter = hookiter;
hookiter = hooks.erase(hookiter);
List<HookInfo>::iterator newhookiter = hookiter;
--newhookiter; // The hook loop will ++ it then
CHookList::CIter *pItIter;
for (pItIter = iface_iter->m_PreHooks.m_UsedIters; pItIter; pItIter = pItIter->m_pNext)
if (pItIter->m_Iter == oldhookiter)
pItIter->m_Iter = newhookiter;
// If this is VP-Hook-NULL interface, also check all other interfaces of this vfnptr
if (*iface_iter == NULL)
{
for (CVfnPtr::IfaceListIter iface_iter2 = vfnptr_iter->m_Ifaces.begin();
iface_iter2 != vfnptr_iter->m_Ifaces.end(); ++iface_iter2)
{
if (*iface_iter2 != NULL)
{
for (pItIter = iface_iter2->m_PreHooks.m_UsedIters; pItIter; pItIter = pItIter->m_pNext)
if (pItIter->m_Iter == oldhookiter)
pItIter->m_Iter = newhookiter;
}
}
}
if (iface_iter->m_PostHooks.m_List.empty() && iface_iter->m_PreHooks.m_List.empty())
{
// If this is a VP-Hook-NULL interface, go through all other interfaces of this vfnptr and tell them!
if (*iface_iter == NULL)
{
for (CVfnPtr::IfaceListIter iface_iter2 = vfnptr_iter->m_Ifaces.begin();
iface_iter2 != vfnptr_iter->m_Ifaces.end(); ++iface_iter2)
{
if (iface_iter2->m_Ptr != NULL)
{
iface_iter2->m_PreHooks.ClearVPList();
iface_iter2->m_PostHooks.ClearVPList();
}
}
}
// There are no hooks on this iface anymore...
for (HookLoopInfoStack::iterator hli_iter = m_HLIStack.begin();
hli_iter != m_HLIStack.end(); ++hli_iter)
{
if (hli_iter->pCurIface == static_cast<IIface*>(&(*iface_iter)))
hli_iter->shouldContinue = false;
}
iface_iter = vfnptr_iter->m_Ifaces.erase(iface_iter);
if (vfnptr_iter->m_Ifaces.empty())
{
// No ifaces at all -> Deactivate the hook
// Only patch the vfnptr back if the module is still in memory
// If it's not, do not remove stuff like we did before
// First off we did it wrong (shutdown the whole hookman, uh..) and secondly applications may be
// confused by RemoveHook returning false then (yeah, I know, I made this one up, no one checks for RemoveHook error)
if (ModuleInMemory(reinterpret_cast<char*>(vfnptr_iter->m_Ptr), SH_PTRSIZE))
{
*reinterpret_cast<void**>(vfnptr_iter->m_Ptr) = vfnptr_iter->m_OrigEntry;
}
hookman->m_VfnPtrs.erase(vfnptr_iter);
// Remove callclass patch
for (Impl_CallClassList::iterator cciter = m_CallClasses.begin(); cciter != m_CallClasses.end(); ++cciter)
if (cciter->m_Ptr == hentry->adjustediface)
cciter->RemoveCallClassPatch(hentry->vtbl_offs, hentry->vtbl_idx);
if (hookman->m_VfnPtrs.empty())
{
// Unregister the hook manager
hookman->m_Func(HA_Unregister, NULL);
}
}
}
// Don't forget to remove the hookid from m_HookIDMan
m_HookIDMan.Remove(hookid);
return true;
}
bool CSourceHookImpl::RemoveHook(RemoveHookInfo info)
{
return RemoveHook(info.plug, info.iface, info.thisptr_offs, info.hookman, info.handler, info.post);
}
bool CSourceHookImpl::RemoveHook(Plugin plug, void *iface, int thisptr_offs, HookManagerPubFunc myHookMan, ISHDelegate *handler, bool post)
{
// Get info about hook manager and compute adjustediface
CHookManagerInfo tmp;
if (myHookMan(HA_GetInfo, &tmp) != 0)
return false;
void *adjustediface = reinterpret_cast<void*>(reinterpret_cast<char*>(iface)+thisptr_offs);
// Loop through all hooks and remove those which match:
// hookman, vfnptr, iface, plug, adjusted iface, this ptr offs, handler, post
CVector<int> removehooks;
m_HookIDMan.FindAllHooks(removehooks, tmp.m_Proto, tmp.m_VtblOffs, tmp.m_VtblIdx, adjustediface, plug, thisptr_offs, handler, post);
if (removehooks.empty())
return false;
bool status = false;
for (CVector<int>::iterator iter = removehooks.begin(); iter != removehooks.end(); ++iter)
{
if (RemoveHookByID(plug, *iter))
status = true;
}
return status;
}
DeprecatedCallClass<void> *CSourceHookImpl::GetCallClass(void *iface, size_t size)
{
for (Impl_CallClassList::iterator cciter = m_CallClasses.begin(); cciter != m_CallClasses.end(); ++cciter)
{
if (cciter->m_Ptr == iface && cciter->m_ObjSize == size)
{
++cciter->m_RefCounter;
return &(*cciter);
}
}
// Make a new one
CCallClassImpl tmp(iface, size);
ApplyCallClassPatches(tmp);
m_CallClasses.push_back(tmp);
return &m_CallClasses.back();
}
void CSourceHookImpl::ReleaseCallClass(DeprecatedCallClass<void> *ptr)
{
Impl_CallClassList::iterator iter = m_CallClasses.find(ptr);
if (iter == m_CallClasses.end())
return;
--iter->m_RefCounter;
if (iter->m_RefCounter < 1)
m_CallClasses.erase(iter);
}
void CSourceHookImpl::ApplyCallClassPatches(CCallClassImpl &cc)
{
for (HookManContList::iterator hmcl_iter = m_HookMans.begin();
hmcl_iter != m_HookMans.end(); ++hmcl_iter)
{
for (CHookManagerContainer::iterator hookman = hmcl_iter->begin();
hookman != hmcl_iter->end(); ++hookman)
{
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = hookman->m_VfnPtrs.begin();
vfnptr_iter != hookman->m_VfnPtrs.end(); ++vfnptr_iter)
{
for (CVfnPtr::IfaceListIter iface_iter = vfnptr_iter->m_Ifaces.begin();
iface_iter != vfnptr_iter->m_Ifaces.end(); ++iface_iter)
{
if (iface_iter->m_Ptr >= cc.m_Ptr &&
iface_iter->m_Ptr < (reinterpret_cast<char*>(cc.m_Ptr) + cc.m_ObjSize))
{
cc.ApplyCallClassPatch(static_cast<int>(reinterpret_cast<char*>(iface_iter->m_Ptr) -
reinterpret_cast<char*>(cc.m_Ptr)) + hookman->m_VtblOffs,
hookman->m_VtblIdx, vfnptr_iter->m_OrigEntry);
}
}
}
}
}
}
void CSourceHookImpl::ApplyCallClassPatches(void *ifaceptr, int vtbl_offs, int vtbl_idx, void *orig_entry)
{
for (Impl_CallClassList::iterator cc_iter = m_CallClasses.begin(); cc_iter != m_CallClasses.end();
++cc_iter)
{
if (ifaceptr >= cc_iter->m_Ptr &&
ifaceptr < (reinterpret_cast<char*>(cc_iter->m_Ptr) + cc_iter->m_ObjSize))
{
cc_iter->ApplyCallClassPatch(static_cast<int>(reinterpret_cast<char*>(ifaceptr) -
reinterpret_cast<char*>(cc_iter->m_Ptr)) + vtbl_offs, vtbl_idx, orig_entry);
}
}
}
void CSourceHookImpl::RemoveCallClassPatches(void *ifaceptr, int vtbl_offs, int vtbl_idx)
{
for (Impl_CallClassList::iterator cc_iter = m_CallClasses.begin(); cc_iter != m_CallClasses.end();
++cc_iter)
{
if (ifaceptr >= cc_iter->m_Ptr &&
ifaceptr < (reinterpret_cast<char*>(cc_iter->m_Ptr) + cc_iter->m_ObjSize))
{
cc_iter->RemoveCallClassPatch(vtbl_offs, vtbl_idx);
}
}
}
void CSourceHookImpl::SetPluginPaused(Plugin plug, bool paused)
{
// Go through all hook managers, all interfaces, and set the status of all hooks of this plugin to paused
for (HookManContList::iterator hmcl_iter = m_HookMans.begin();
hmcl_iter != m_HookMans.end(); ++hmcl_iter)
{
for (CHookManagerContainer::iterator hookmaniter = hmcl_iter->begin();
hookmaniter != hmcl_iter->end(); ++hookmaniter)
{
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = hookmaniter->m_VfnPtrs.begin();
vfnptr_iter != hookmaniter->m_VfnPtrs.end(); ++vfnptr_iter)
{
for (CVfnPtr::IfaceListIter ifaceiter = vfnptr_iter->m_Ifaces.begin();
ifaceiter != vfnptr_iter->m_Ifaces.end(); ++ifaceiter)
{
for (List<HookInfo>::iterator hookiter = ifaceiter->m_PreHooks.m_List.begin();
hookiter != ifaceiter->m_PreHooks.m_List.end(); ++hookiter)
if (plug == hookiter->plug)
hookiter->paused = paused;
for (List<HookInfo>::iterator hookiter = ifaceiter->m_PostHooks.m_List.begin();
hookiter != ifaceiter->m_PostHooks.m_List.end(); ++hookiter)
if (plug == hookiter->plug)
hookiter->paused = paused;
}
}
}
}
}
void CSourceHookImpl::PausePlugin(Plugin plug)
{
SetPluginPaused(plug, true);
}
void CSourceHookImpl::UnpausePlugin(Plugin plug)
{
SetPluginPaused(plug, false);
}
void CSourceHookImpl::HookLoopBegin(IIface *pIface)
{
HookLoopInfo hli = {0};
hli.pCurIface = pIface;
hli.shouldContinue = true;
hli.recall = HookLoopInfo::Recall_No;
static_cast<CIface*>(pIface)->m_PreHooks.RQFlagReset();
static_cast<CIface*>(pIface)->m_PostHooks.RQFlagReset();
m_HLIStack.push(hli);
}
void CSourceHookImpl::HookLoopEnd()
{
// When in a post recall... make sure status is high enough so that it returns the orig ret.
if (m_HLIStack.size() > 1 && m_HLIStack.second().recall == HookLoopInfo::Recall_Post2)
*m_HLIStack.front().pStatus = MRES_SUPERCEDE; // THAT'LL TEACH THEM!
m_HLIStack.pop();
}
void CSourceHookImpl::SetRes(META_RES res)
{
*m_HLIStack.front().pCurRes = res;
}
META_RES CSourceHookImpl::GetPrevRes()
{
return *m_HLIStack.front().pPrevRes;
}
META_RES CSourceHookImpl::GetStatus()
{
return *m_HLIStack.front().pStatus;
}
const void *CSourceHookImpl::GetOrigRet()
{
// When in a post recall... return the orig ret of the previous hookloop.
if (m_HLIStack.size() > 1 && m_HLIStack.second().recall == HookLoopInfo::Recall_Post2)
return m_HLIStack.second().pOrigRet;
return m_HLIStack.front().pOrigRet;
}
const void *CSourceHookImpl::GetOverrideRet()
{
// pOverrideRet is always set since recalls were introduced
// GetOverrideRetPtr was added; a function which always returns pOverrideRet
// so that RETURN_META_VALUE_NEWPARAMS can provide an override result
// This means that we have to filter GetOverrideRet:
// If the status variable is < MRES_OVERRIDE, return NULL.
return (*m_HLIStack.front().pStatus < MRES_OVERRIDE) ?
NULL : m_HLIStack.front().pOverrideRet;
}
void *CSourceHookImpl::GetOverrideRetPtr()
{
// As described in the comment above: always return pOverrideRet
return m_HLIStack.front().pOverrideRet;
}
void *CSourceHookImpl::GetIfacePtr()
{
return *m_HLIStack.front().pIfacePtrPtr;
}
void CSourceHookImpl::SetCurResPtr(META_RES *mres)
{
m_HLIStack.front().pCurRes = mres;
}
void CSourceHookImpl::SetPrevResPtr(META_RES *mres)
{
m_HLIStack.front().pPrevRes = mres;
// If we're recalling, drag the previous mres value to the new hookfunc
if (m_HLIStack.size() > 1 && m_HLIStack.second().recall)
*mres = *m_HLIStack.second().pPrevRes;
}
void CSourceHookImpl::SetStatusPtr(META_RES *mres)
{
m_HLIStack.front().pStatus = mres;
// If we're recalling, drag the previous mres value to the new hookfunc
if (m_HLIStack.size() > 1 && m_HLIStack.second().recall)
*mres = *m_HLIStack.second().pStatus;
}
void CSourceHookImpl::SetIfacePtrPtr(void **pp)
{
m_HLIStack.front().pIfacePtrPtr = pp;
}
void CSourceHookImpl::SetOrigRetPtr(const void *ptr)
{
m_HLIStack.front().pOrigRet = ptr;
}
void CSourceHookImpl::SetOverrideRetPtr(void *ptr)
{
m_HLIStack.front().pOverrideRet = ptr;
}
// New function which does all of the above + more :)
void *CSourceHookImpl::SetupHookLoop(META_RES *statusPtr, META_RES *prevResPtr, META_RES *curResPtr,
void **ifacePtrPtr, const void *origRetPtr, void *overrideRetPtr)
{
HookLoopInfo &hli = m_HLIStack.front();
hli.pStatus = statusPtr;
hli.pPrevRes = prevResPtr;
hli.pCurRes = curResPtr;
hli.pIfacePtrPtr = ifacePtrPtr;
hli.pOrigRet = origRetPtr;
// Handle some recall stuff
if (m_HLIStack.size() > 1 && m_HLIStack.second().recall)
{
HookLoopInfo &other = m_HLIStack.second();
*statusPtr = *other.pStatus;
*prevResPtr = *other.pStatus;
// When the status is low so there's no override return value and we're in a post recall,
// give it the orig return value as override return value.
if (*statusPtr < MRES_OVERRIDE && other.recall == HookLoopInfo::Recall_Post1)
hli.pOverrideRet = const_cast<void*>(other.pOrigRet);
else // Otherwise, transfer override ret normally
hli.pOverrideRet = other.pOverrideRet;
}
else
hli.pOverrideRet = overrideRetPtr;
// Tell the hook func which override ret ptr to use
return hli.pOverrideRet;
}
bool CSourceHookImpl::ShouldContinue()
{
// If recall is true, we shall not continue either.
// This is because, if it's true and ShouldContinue is called, it suggests that the
// actual recall is done and that we are back in the original handler which shall return
// immediately.
// Post-recalls:
// The second element on the stack has recall set to Recall_Post1.
// This means that we want to skip this part and the original function calling thing, so
// we store the status mres value, set status to MRES_SUPERCEDE, and return false.
// We also set the recall flag to Recall_Post2, so the next time the thing calls us, we
// can return true (so that the thing goes into post hooks).
if (m_HLIStack.size() > 1)
{
if (m_HLIStack.second().recall == HookLoopInfo::Recall_Post1)
{
m_HLIStack.front().temporaryStatus = *m_HLIStack.front().pStatus;
*m_HLIStack.front().pStatus = MRES_SUPERCEDE;
m_HLIStack.second().recall = HookLoopInfo::Recall_Post2;
return false;
}
else if (m_HLIStack.second().recall == HookLoopInfo::Recall_Post2)
{
*m_HLIStack.front().pStatus = m_HLIStack.front().temporaryStatus;
return m_HLIStack.front().shouldContinue;
}
}
return m_HLIStack.front().shouldContinue && !m_HLIStack.front().recall;
}
void CSourceHookImpl::DoRecall()
{
if (!m_HLIStack.empty())
{
// Also watch out for post recalls! Described at the beginning of sourcehook_impl.h
m_HLIStack.front().recall =
static_cast<CIface*>(m_HLIStack.front().pCurIface)->m_PostHooks.RQFlagGet() ?
HookLoopInfo::Recall_Post1 : HookLoopInfo::Recall_Pre;
CHookList *mlist = static_cast<CHookList*>(m_HLIStack.front().recall == HookLoopInfo::Recall_Pre ?
m_HLIStack.front().pCurIface->GetPreHooks() : m_HLIStack.front().pCurIface->GetPostHooks());
mlist->m_Recall = true;
// The hookfunc usually does this, but it won't have a chance to see it,
// so for recalls, we update status here if it's required
if (*m_HLIStack.front().pCurRes > *m_HLIStack.front().pStatus)
*m_HLIStack.front().pStatus = *m_HLIStack.front().pCurRes;
}
}
void CSourceHookImpl::SetIgnoreHooks(Plugin plug, void *vfnptr)
{
m_OneIgnore = vfnptr;
}
void CSourceHookImpl::ResetIgnoreHooks(Plugin plug, void *vfnptr)
{
m_OneIgnore = NULL;
}
void *CSourceHookImpl::GetOrigVfnPtrEntry(void *vfnptr)
{
for (HookManContList::iterator hmcl_iter = m_HookMans.begin();
hmcl_iter != m_HookMans.end(); ++hmcl_iter)
{
for (CHookManagerContainer::iterator hookmaniter = hmcl_iter->begin();
hookmaniter != hmcl_iter->end(); ++hookmaniter)
{
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = hookmaniter->m_VfnPtrs.begin();
vfnptr_iter != hookmaniter->m_VfnPtrs.end(); ++vfnptr_iter)
{
if (vfnptr_iter->m_Ptr == vfnptr)
return vfnptr_iter->m_OrigEntry;
}
}
}
return NULL;
}
////////////////////////////
// CCallClassImpl
////////////////////////////
CSourceHookImpl::CCallClassImpl::CCallClassImpl(void *ptr, size_t size)
: m_Ptr(ptr), m_ObjSize(size), m_RefCounter(1)
{
}
CSourceHookImpl::CCallClassImpl::~CCallClassImpl()
{
}
void *CSourceHookImpl::CCallClassImpl::GetThisPtr()
{
return m_Ptr;
}
void *CSourceHookImpl::CCallClassImpl::GetOrigFunc(int vtbloffs, int vtblidx)
{
OrigVTables::iterator iter = m_VT.find(vtbloffs);
if (iter != m_VT.end() && iter->val.size() > (size_t)vtblidx)
{
return iter->val[vtblidx];
}
return NULL;
}
void CSourceHookImpl::CCallClassImpl::ApplyCallClassPatch(int vtbl_offs, int vtbl_idx, void *orig_entry)
{
OrigFuncs &tmpvec = m_VT[vtbl_offs];
if (tmpvec.size() <= (size_t)vtbl_idx)
tmpvec.resize(vtbl_idx+1, NULL);
tmpvec[vtbl_idx] = orig_entry;
}
void CSourceHookImpl::CCallClassImpl::RemoveCallClassPatch(int vtbl_offs, int vtbl_idx)
{
OrigVTables::iterator iter = m_VT.find(vtbl_offs);
if (iter != m_VT.end())
{
if (iter->val.size() > (size_t)vtbl_idx)
{
iter->val[vtbl_idx] = 0;
OrigFuncs &of = iter->val;
OrigFuncs::iterator lastused = of.end();
for (OrigFuncs::iterator viter = of.begin(); viter != of.end(); ++viter)
{
if (*viter)
lastused = viter;
}
if (lastused == of.end())
{
// No used element => Remove the whole vector
m_VT.erase(iter);
return;
}
of.resize(lastused - of.begin() + 1);
}
}
}
////////////////////////////
// CHookManagerInfo
////////////////////////////
CSourceHookImpl::CHookManagerInfo::CHookManagerInfo() : m_HookManVersion(0)
{
}
CSourceHookImpl::CHookManagerInfo::~CHookManagerInfo()
{
}
IVfnPtr *CSourceHookImpl::CHookManagerInfo::FindVfnPtr(void *vfnptr)
{
VfnPtrListIter iter = m_VfnPtrs.find(vfnptr);
return iter == m_VfnPtrs.end() ? NULL : &(*iter);
}
void CSourceHookImpl::CHookManagerInfo::SetInfo(int vtbl_offs, int vtbl_idx, const char *proto)
{
m_VtblOffs = vtbl_offs;
m_VtblIdx = vtbl_idx;
m_Proto = proto;
}
void CSourceHookImpl::CHookManagerInfo::SetHookfuncVfnptr(void *hookfunc_vfnptr)
{
m_HookfuncVfnptr = hookfunc_vfnptr;
}
void CSourceHookImpl::CHookManagerInfo::SetVersion(int version)
{
m_HookManVersion = version;
}
////////////////////////////
// CVfnPtr
////////////////////////////
// If you get a crash here, the ptr passed is invalid
// This usually means a SH_DECL_MANUALHOOK* with wrong thisptroffs/vtbloffs/vtblidx
CSourceHookImpl::CVfnPtr::CVfnPtr(void *ptr, void **pOneIgnore) : m_Ptr(ptr),
m_OrigEntry(*reinterpret_cast<void**>(ptr)), m_pOneIgnore(pOneIgnore)
{
}
CSourceHookImpl::CVfnPtr::~CVfnPtr()
{
}
void *CSourceHookImpl::CVfnPtr::GetVfnPtr()
{
return m_Ptr;
}
void *CSourceHookImpl::CVfnPtr::GetOrigEntry()
{
return m_OrigEntry;
}
IIface *CSourceHookImpl::CVfnPtr::FindIface(void *ptr)
{
if (m_Ptr == *m_pOneIgnore)
{
*m_pOneIgnore = NULL; // Only once!
return NULL;
}
IfaceListIter iter = m_Ifaces.find(ptr);
// If nothing is found, check for a NULL-interface (VP hooks only)
if (iter == m_Ifaces.end())
{
iter = m_Ifaces.find((void*)0);
return iter == m_Ifaces.end() ? NULL : &(*iter);
}
else
{
return &(*iter);
}
}
////////////////////////////
// CIface
////////////////////////////
CSourceHookImpl::CIface::CIface(void *ptr) : m_Ptr(ptr)
{
}
CSourceHookImpl::CIface::~CIface()
{
}
void *CSourceHookImpl::CIface::GetPtr()
{
return m_Ptr;
}
IHookList *CSourceHookImpl::CIface::GetPreHooks()
{
return &m_PreHooks;
}
IHookList *CSourceHookImpl::CIface::GetPostHooks()
{
return &m_PostHooks;
}
////////////////////////////
// CHookList
////////////////////////////
CSourceHookImpl::CHookList::CHookList() : m_VPList(NULL), m_FreeIters(NULL), m_UsedIters(NULL),
m_Recall(false)
{
}
CSourceHookImpl::CHookList::CHookList(const CHookList &other) : m_VPList(other.m_VPList), m_List(other.m_List),
m_FreeIters(NULL), m_UsedIters(NULL), m_Recall(false)
{
}
CSourceHookImpl::CHookList::~CHookList()
{
while (m_FreeIters)
{
CIter *iter = m_FreeIters->m_pNext;
delete m_FreeIters;
m_FreeIters = iter;
}
while (m_UsedIters)
{
CIter *iter = m_UsedIters->m_pNext;
delete m_UsedIters;
m_UsedIters = iter;
}
}
IHookList::IIter *CSourceHookImpl::CHookList::GetIter()
{
m_RQFlag = true;
CIter *ret;
if (m_FreeIters)
{
ret = m_FreeIters;
m_FreeIters = ret->m_pNext;
ret->GoToBegin();
}
else
{
ret = new CIter(this);
}
// Muuuh, if we're recalling, it shall be a copy of the last iterator, incremented by one
if (m_Recall && m_UsedIters)
{
ret->Set(m_UsedIters); // m_UsedIters is the last returned and not released iterator
ret->Next(); // Use next instead of directly incrementing its m_Iter:
// skips paused plugins
}
ret->m_pNext = m_UsedIters;
ret->m_pPrev = NULL;
if (m_UsedIters)
m_UsedIters->m_pPrev = ret;
m_UsedIters = ret;
m_Recall = false;
return ret;
}
void CSourceHookImpl::CHookList::ReleaseIter(IIter *pIter)
{
CIter *pIter2 = static_cast<CIter*>(pIter);
// Unlink from m_UsedIters
if (pIter2->m_pNext)
pIter2->m_pNext->m_pPrev = pIter2->m_pPrev;
if (pIter2->m_pPrev)
pIter2->m_pPrev->m_pNext = pIter2->m_pNext;
if (pIter2 == m_UsedIters)
m_UsedIters = NULL;
// Link to m_FreeIters
pIter2->m_pNext = m_FreeIters;
m_FreeIters = pIter2;
// Reset recall state.
m_Recall = false;
}
void CSourceHookImpl::CHookList::SetVPList(List<HookInfo> *newList)
{
m_VPList = newList;
// Update cached CIter objects
CIter *pTmp;
pTmp = m_FreeIters;
while (pTmp)
{
pTmp->m_Iter.SetListLeft(m_VPList);
pTmp = pTmp->m_pNext;
}
pTmp = m_UsedIters;
while (pTmp)
{
pTmp->m_Iter.SetListLeft(m_VPList);
pTmp = pTmp->m_pNext;
}
}
void CSourceHookImpl::CHookList::ClearVPList()
{
SetVPList(NULL);
}
CSourceHookImpl::CHookList::CIter::CIter(CHookList *pList) : m_pList(pList),
m_Iter(pList->m_VPList, &m_pList->m_List), m_pNext(NULL)
{
}
CSourceHookImpl::CHookList::CIter::~CIter()
{
}
void CSourceHookImpl::CHookList::CIter::Set(CIter *pOther)
{
m_Iter = pOther->m_Iter;
}
void CSourceHookImpl::CHookList::CIter::GoToBegin()
{
m_Iter.GoToBegin();
SkipPaused();
}
bool CSourceHookImpl::CHookList::CIter::End()
{
if (!m_pList)
return false;
return m_Iter.End();;
}
void CSourceHookImpl::CHookList::CIter::Next()
{
if (!m_pList)
return;
++m_Iter;
SkipPaused();
}
void CSourceHookImpl::CHookList::CIter::Clear()
{
m_pList = NULL;
}
void CSourceHookImpl::CHookList::CIter::SkipPaused()
{
while (!m_Iter.End() && m_Iter->paused)
++m_Iter;
}
ISHDelegate *CSourceHookImpl::CHookList::CIter::Handler()
{
return m_Iter->handler;
}
int CSourceHookImpl::CHookList::CIter::ThisPtrOffs()
{
return m_Iter->thisptr_offs;
}
//////////////////////////////////////////////////////////////////////////
// CProto
//////////////////////////////////////////////////////////////////////////
char *CSourceHookImpl::CProto::DupProto(const char *p)
{
if (!p)
return NULL;
char *newproto;
if (*p)
{
size_t len = strlen(p);
newproto = new char[len+1];
memcpy(newproto, p, len+1);
} else {
const ProtoInfo *pi1 = reinterpret_cast<const ProtoInfo *>(p);
int *ar_copy = new int[pi1->numOfParams + 1];
for (int i = 0; i <= pi1->numOfParams; i++)
ar_copy[i] = pi1->params[i];
ProtoInfo *copy = new ProtoInfo(pi1->retTypeSize, pi1->numOfParams, ar_copy);
newproto = reinterpret_cast<char *>(copy);
}
return newproto;
}
void CSourceHookImpl::CProto::FreeProto(char *prot)
{
if (!prot)
return;
if (*prot)
{
delete [] prot;
} else {
ProtoInfo *pi = reinterpret_cast<ProtoInfo *>(prot);
delete [] pi->params;
delete pi;
}
}
bool CSourceHookImpl::CProto::Equal(const char *p1, const char *p2)
{
if (!p1 || !p2)
return false;
if (*p1 && *p2) // Case1: Both old
{
// As in old versions
return strcmp(p1, p2) == 0;
}
else if (!*p1 && !*p2) // Case2: Both new
{
const ProtoInfo *pi1 = reinterpret_cast<const ProtoInfo*>(p1);
const ProtoInfo *pi2 = reinterpret_cast<const ProtoInfo*>(p2);
if (pi1->retTypeSize == pi2->retTypeSize &&
pi1->numOfParams == pi2->numOfParams)
{
// params[0] is 0 for "normal" functions and -1 for vararg functions
// params[1] is size of first parameter
// params[2] is size of second parameter ...
for (int i = 0; i <= pi1->numOfParams; ++i)
{
if (pi1->params[i] != pi2->params[i])
return false;
}
return true;
}
else
{
return false;
}
}
else // Case3: Mixed old/new
{
// Trust the user
return true;
}
}
//////////////////////////////////////////////////////////////////////////
// CHookManagerContainer
//////////////////////////////////////////////////////////////////////////
void CSourceHookImpl::CHookManagerContainer::AddHookManager(Plugin plug, const CHookManagerInfo &hookman)
{
iterator iter;
// Check whether such a hook manager already exists; if yes, ignore.
for (iter = begin(); iter != end(); ++iter)
{
if (iter->m_Plug == plug && iter->m_Func == hookman.m_Func)
return;
}
// It doesn't -> add it. Add it to the end of its version group.
for (iter = begin(); iter != end(); ++iter)
{
if (iter->m_HookManVersion < hookman.m_HookManVersion)
break;
}
bool isBeginning = iter == begin();
insert(iter, hookman);
// If we inserted it at the beginning and if the dam breaks open many years too soon
// and if there is more than one hookman and if the second one isn't empty, transfer
// hooks from second to first
if (isBeginning && size() > 1)
{
iter = begin();
iterator second = iter;
++second;
if (!second->m_VfnPtrs.empty())
{
// Move the vfnptrs from the old hook manager to the new one
iter->m_VfnPtrs = second->m_VfnPtrs;
second->m_VfnPtrs.clear();
// Unregister the old one, register the new one
second->m_Func(HA_Unregister, NULL);
iter->m_Func(HA_Register, &(*iter));
// Go through all vfnptrs in this hookman and patch them to point to the new manager's handler!
// or whatever
for (CHookManagerInfo::VfnPtrListIter vfnptr_iter = iter->m_VfnPtrs.begin();
vfnptr_iter != iter->m_VfnPtrs.end(); ++vfnptr_iter)
{
// And DEREFERENCE newHookMan->m_HookfuncVfnptr!
// otherwise it will be executing the vtable... had to find out the hard way
*reinterpret_cast<void**>(vfnptr_iter->m_Ptr) = *reinterpret_cast<void**>(iter->m_HookfuncVfnptr);
}
}
}
}
//////////////////////////////////////////////////////////////////////////
// CHookIDManager
//////////////////////////////////////////////////////////////////////////
CSourceHookImpl::CHookIDManager::CHookIDManager()
{
}
int CSourceHookImpl::CHookIDManager::New(const CProto &proto, int vtbl_offs, int vtbl_idx, void *vfnptr,
void *adjustediface, Plugin plug, int thisptr_offs, ISHDelegate *handler, bool post)
{
Entry tmp(proto, vtbl_offs, vtbl_idx, vfnptr, adjustediface, plug, thisptr_offs, handler, post);
size_t cursize = m_Entries.size();
for (size_t i = 0; i < cursize; ++i)
{
if (m_Entries[i].isfree)
{
m_Entries[i] = tmp;
return static_cast<int>(i) + 1;
}
}
m_Entries.push_back(tmp);
return static_cast<int>(m_Entries.size()); // return size() because hookid = id+1 anyway
}
bool CSourceHookImpl::CHookIDManager::Remove(int hookid)
{
int realid = hookid - 1;
if (realid < 0 || realid >= static_cast<int>(m_Entries.size()) || m_Entries[realid].isfree)
return false;
m_Entries[realid].isfree = true;
// :TODO: remove free ids from back sometimes ??
return true;
}
const CSourceHookImpl::CHookIDManager::Entry * CSourceHookImpl::CHookIDManager::QueryHook(int hookid)
{
int realid = hookid - 1;
if (realid < 0 || realid >= static_cast<int>(m_Entries.size()) || m_Entries[realid].isfree)
return NULL;
return &m_Entries[realid];
}
void CSourceHookImpl::CHookIDManager::FindAllHooks(CVector<int> &output, const CProto &proto, int vtbl_offs,
int vtbl_idx, void *adjustediface, Plugin plug, int thisptr_offs, ISHDelegate *handler, bool post)
{
// oh my god, a lot of parameters...
size_t cursize = m_Entries.size();
for (size_t i = 0; i < cursize; ++i)
{
if (!m_Entries[i].isfree && m_Entries[i].proto == proto && m_Entries[i].vtbl_offs == vtbl_offs &&
m_Entries[i].vtbl_idx == vtbl_idx && m_Entries[i].adjustediface == adjustediface && m_Entries[i].plug == plug &&
m_Entries[i].thisptr_offs == thisptr_offs && m_Entries[i].handler->IsEqual(handler) && m_Entries[i].post == post)
{
output.push_back(static_cast<int>(i) + 1);
}
}
}
}