mirror of
https://github.com/alliedmodders/metamod-source.git
synced 2024-12-12 00:08:50 +01:00
50fe3e72d9
--HG-- extra : convert_revision : svn%3Ac2935e3e-5518-0410-8daf-afa5dab7d4e3/trunk%40182
1213 lines
35 KiB
C++
1213 lines
35 KiB
C++
/* ======== SourceHook ========
|
|
* Copyright (C) 2004-2005 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 SourceHook::HashFunction<int>(const int & k)
|
|
{
|
|
return k;
|
|
}
|
|
template<>
|
|
int SourceHook::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()
|
|
{
|
|
}
|
|
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)
|
|
{
|
|
void *adjustediface = reinterpret_cast<void*>(reinterpret_cast<char*>(iface) + thisptr_offs);
|
|
|
|
// 1) Get info about the hook manager
|
|
CHookManagerInfo tmp;
|
|
if (myHookMan(HA_GetInfo, &tmp) != 0)
|
|
return false;
|
|
|
|
tmp.m_Func = myHookMan;
|
|
tmp.m_Plug = plug;
|
|
|
|
CHookManagerContainer::HMCI hmci(tmp.m_Proto, tmp.m_VtblOffs, tmp.m_VtblIdx);
|
|
|
|
void **cur_vtptr = *reinterpret_cast<void***>(
|
|
reinterpret_cast<char*>(adjustediface) + tmp.m_VtblOffs);
|
|
void *cur_vfnptr = reinterpret_cast<void*>(cur_vtptr + 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 false;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Tell it to store the pointer if it's not already active
|
|
if (hookman->m_VfnPtrs.empty())
|
|
hookman->m_Func(HA_Register, &(*hookman));
|
|
|
|
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);
|
|
|
|
// Alter vtable entry
|
|
if (!SetMemAccess(cur_vtptr, sizeof(void*) * (tmp.m_VtblIdx + 1), SH_MEM_READ | SH_MEM_WRITE))
|
|
return false;
|
|
|
|
*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);
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
// Add the hook
|
|
HookInfo hookinfo;
|
|
hookinfo.handler = handler;
|
|
hookinfo.plug = plug;
|
|
hookinfo.paused = false;
|
|
hookinfo.thisptr_offs = thisptr_offs;
|
|
if (post)
|
|
iface_iter->m_PostHooks.m_List.push_back(hookinfo);
|
|
else
|
|
iface_iter->m_PreHooks.m_List.push_back(hookinfo);
|
|
|
|
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)
|
|
{
|
|
void *adjustediface = reinterpret_cast<void*>(reinterpret_cast<char*>(iface)+thisptr_offs);
|
|
CHookManagerInfo tmp;
|
|
if (myHookMan(HA_GetInfo, &tmp) != 0)
|
|
return false;
|
|
|
|
// Find the hook manager and the hook
|
|
|
|
HookManContList::iterator hmcl_iter = m_HookMans.find(
|
|
CHookManagerContainer::HMCI(tmp.m_Proto, tmp.m_VtblOffs, tmp.m_VtblIdx));
|
|
if (hmcl_iter == m_HookMans.end() || hmcl_iter->empty())
|
|
return false;
|
|
|
|
CHookManagerContainer::iterator hookman = hmcl_iter->begin();
|
|
|
|
if (!ModuleInMemory(reinterpret_cast<char*>(adjustediface) + tmp.m_VtblOffs,
|
|
sizeof(void*) * (tmp.m_VtblIdx + 1)))
|
|
{
|
|
// The module the vtable was in is already unloaded.
|
|
hookman->m_VfnPtrs.clear();
|
|
hookman->m_Func(HA_Unregister, NULL);
|
|
return true;
|
|
}
|
|
|
|
void **cur_vtptr = *reinterpret_cast<void***>(
|
|
reinterpret_cast<char*>(adjustediface) + tmp.m_VtblOffs);
|
|
void *cur_vfnptr = reinterpret_cast<void*>(cur_vtptr + tmp.m_VtblIdx);
|
|
|
|
CHookManagerInfo::VfnPtrListIter vfnptr_iter = hookman->m_VfnPtrs.find(cur_vfnptr);
|
|
|
|
if (vfnptr_iter == hookman->m_VfnPtrs.end())
|
|
return false;
|
|
|
|
for (CVfnPtr::IfaceListIter iface_iter = vfnptr_iter->m_Ifaces.begin();
|
|
iface_iter != vfnptr_iter->m_Ifaces.end();)
|
|
{
|
|
if (iface_iter->m_Ptr != adjustediface)
|
|
{
|
|
iface_iter++;
|
|
continue;
|
|
}
|
|
List<HookInfo> &hooks =
|
|
post ? iface_iter->m_PostHooks.m_List : iface_iter->m_PreHooks.m_List;
|
|
|
|
bool erase;
|
|
for (List<HookInfo>::iterator hookiter = hooks.begin();
|
|
hookiter != hooks.end(); )
|
|
{
|
|
erase = hookiter->plug == plug && hookiter->handler->IsEqual(handler) &&
|
|
hookiter->thisptr_offs == thisptr_offs;
|
|
if (erase)
|
|
{
|
|
hookiter->handler->DeleteThis(); // Make the _plugin_ delete the handler object
|
|
|
|
// 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;
|
|
}
|
|
else
|
|
++hookiter;
|
|
}
|
|
if (iface_iter->m_PostHooks.m_List.empty() && iface_iter->m_PreHooks.m_List.empty())
|
|
{
|
|
// 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
|
|
*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 == adjustediface)
|
|
cciter->RemoveCallClassPatch(tmp.m_VtblOffs, tmp.m_VtblIdx);
|
|
|
|
if (hookman->m_VfnPtrs.empty())
|
|
{
|
|
// Unregister the hook manager
|
|
hookman->m_Func(HA_Unregister, NULL);
|
|
}
|
|
|
|
// Don't try to continue looping through ifaces
|
|
// - the list is already invalid
|
|
return true;
|
|
}
|
|
}
|
|
else
|
|
++iface_iter;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
GenericCallClass *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(GenericCallClass *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;
|
|
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;
|
|
|
|
}
|
|
}
|
|
|
|
////////////////////////////
|
|
// 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);
|
|
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) : m_Ptr(ptr), m_OrigEntry(*reinterpret_cast<void**>(ptr))
|
|
{
|
|
}
|
|
CSourceHookImpl::CVfnPtr::~CVfnPtr()
|
|
{
|
|
}
|
|
|
|
void *CSourceHookImpl::CVfnPtr::GetVfnPtr()
|
|
{
|
|
return m_Ptr;
|
|
}
|
|
|
|
void *CSourceHookImpl::CVfnPtr::GetOrigEntry()
|
|
{
|
|
return m_OrigEntry;
|
|
}
|
|
|
|
IIface *CSourceHookImpl::CVfnPtr::FindIface(void *ptr)
|
|
{
|
|
IfaceListIter iter = m_Ifaces.find(ptr);
|
|
return iter == m_Ifaces.end() ? NULL : &(*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_FreeIters(NULL), m_UsedIters(NULL),
|
|
m_Recall(false)
|
|
{
|
|
}
|
|
CSourceHookImpl::CHookList::CHookList(const CHookList &other) : 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;
|
|
}
|
|
|
|
CSourceHookImpl::CHookList::CIter::CIter(CHookList *pList) : m_pList(pList), m_pNext(NULL)
|
|
{
|
|
GoToBegin();
|
|
}
|
|
CSourceHookImpl::CHookList::CIter::~CIter()
|
|
{
|
|
}
|
|
|
|
void CSourceHookImpl::CHookList::CIter::Set(CIter *pOther)
|
|
{
|
|
m_Iter = pOther->m_Iter;
|
|
}
|
|
|
|
void CSourceHookImpl::CHookList::CIter::GoToBegin()
|
|
{
|
|
m_Iter = m_pList->m_List.begin();
|
|
SkipPaused();
|
|
}
|
|
|
|
bool CSourceHookImpl::CHookList::CIter::End()
|
|
{
|
|
if (!m_pList)
|
|
return false;
|
|
return m_Iter == m_pList->m_List.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 != m_pList->m_List.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)
|
|
{
|
|
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)
|
|
{
|
|
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) // 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);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|