mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
d20250e436
Only add to detour if the VFTable pointer didn't exist yet. Not declaring these in the anonymous namespace significantly reduces code size and unnecessary memory allocations. We might want to consider refactoring this, although it will probably require a big change.
45 lines
1.2 KiB
C++
45 lines
1.2 KiB
C++
#ifndef IDETOUR_H
|
|
#define IDETOUR_H
|
|
|
|
#define ADDDETOUR(x,y) static std::size_t dummy_reg_##y = AddDetour( new x(), #x );
|
|
#define XREGISTER(x,y) ADDDETOUR(x, y)
|
|
#define REGISTER(x) XREGISTER(x, __COUNTER__)
|
|
|
|
class IDetour
|
|
{
|
|
public:
|
|
virtual ~IDetour() { ; }
|
|
virtual void GetAdr(void) const = 0;
|
|
virtual void GetFun(void) const = 0;
|
|
virtual void GetVar(void) const = 0;
|
|
virtual void GetCon(void) const = 0;
|
|
|
|
virtual void Attach(void) const = 0;
|
|
virtual void Detach(void) const = 0;
|
|
};
|
|
|
|
class VDetour : public IDetour
|
|
{
|
|
virtual void GetAdr(void) const { }
|
|
virtual void GetFun(void) const { }
|
|
virtual void GetVar(void) const { }
|
|
virtual void GetCon(void) const { }
|
|
|
|
virtual void Attach(void) const { }
|
|
virtual void Detach(void) const { }
|
|
};
|
|
|
|
inline std::vector<IDetour*> vDetour;
|
|
inline std::unordered_set<IDetour*> sDetour;
|
|
inline std::size_t AddDetour(IDetour* pDetour, const char* pszName)
|
|
{
|
|
IDetour* pVFTable = reinterpret_cast<IDetour**>(pDetour)[0];
|
|
auto p = sDetour.insert(pVFTable); // Only register if VFTable isn't already registered.
|
|
|
|
p.second ? vDetour.push_back(pDetour) : delete pDetour;
|
|
return vDetour.size();
|
|
}
|
|
|
|
REGISTER(VDetour);
|
|
#endif // IDETOUR_H
|