mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
532ada48e0
Duplicate checks are only performed in debug builds, if a duplicate is encountered, a bug in code has been found and has to be solved (do not call REGISTER from headers for example).
48 lines
1.3 KiB
C++
48 lines
1.3 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)
|
|
{
|
|
#ifdef _DEBUG
|
|
IDetour* pVFTable = reinterpret_cast<IDetour**>(pDetour)[0];
|
|
auto p = sDetour.insert(pVFTable); // Only register if VFTable isn't already registered.
|
|
|
|
assert(p.second); // Code bug: duplicate registration!!! (called 'REGISTER(...)' from a header file?).
|
|
p.second ? vDetour.push_back(pDetour) : delete pDetour;
|
|
#endif // DEBUG
|
|
return vDetour.size();
|
|
}
|
|
|
|
REGISTER(VDetour);
|
|
#endif // IDETOUR_H
|