2021-12-25 22:36:38 +01:00
|
|
|
#ifndef IDETOUR_H
|
|
|
|
#define IDETOUR_H
|
|
|
|
|
2022-11-27 15:01:55 +01:00
|
|
|
#define ADDDETOUR(x,y) static std::size_t dummy_reg_##y = AddDetour( new x(), #x );
|
2021-12-25 22:36:38 +01:00
|
|
|
#define XREGISTER(x,y) ADDDETOUR(x, y)
|
|
|
|
#define REGISTER(x) XREGISTER(x, __COUNTER__)
|
|
|
|
|
|
|
|
class IDetour
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
virtual ~IDetour() { ; }
|
2022-05-13 17:51:01 +02:00
|
|
|
virtual void GetAdr(void) const = 0;
|
2022-04-11 01:44:30 +02:00
|
|
|
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;
|
2021-12-25 22:36:38 +01:00
|
|
|
};
|
|
|
|
|
2022-05-13 14:53:25 +02:00
|
|
|
class VDetour : public IDetour
|
2021-12-25 22:36:38 +01:00
|
|
|
{
|
2022-04-11 01:44:30 +02:00
|
|
|
virtual void GetAdr(void) const { }
|
|
|
|
virtual void GetFun(void) const { }
|
|
|
|
virtual void GetVar(void) const { }
|
|
|
|
virtual void GetCon(void) const { }
|
2021-12-25 22:36:38 +01:00
|
|
|
|
2022-04-11 01:44:30 +02:00
|
|
|
virtual void Attach(void) const { }
|
|
|
|
virtual void Detach(void) const { }
|
|
|
|
};
|
|
|
|
|
2022-11-27 15:01:55 +01:00
|
|
|
inline std::vector<IDetour*> vDetour;
|
|
|
|
inline std::unordered_set<IDetour*> sDetour;
|
|
|
|
inline std::size_t AddDetour(IDetour* pDetour, const char* pszName)
|
2021-12-25 22:36:38 +01:00
|
|
|
{
|
2022-11-27 15:01:55 +01:00
|
|
|
IDetour* pVFTable = reinterpret_cast<IDetour**>(pDetour)[0];
|
|
|
|
auto p = sDetour.insert(pVFTable); // Only register if VFTable isn't already registered.
|
|
|
|
|
2023-01-25 19:22:05 +01:00
|
|
|
assert(p.second); // Code bug: duplicate registration!!! (called 'REGISTER(...)' from a header file?).
|
2022-11-27 15:01:55 +01:00
|
|
|
p.second ? vDetour.push_back(pDetour) : delete pDetour;
|
2023-01-26 02:59:50 +01:00
|
|
|
|
2022-11-27 15:01:55 +01:00
|
|
|
return vDetour.size();
|
2022-04-11 01:44:30 +02:00
|
|
|
}
|
2021-12-25 22:36:38 +01:00
|
|
|
|
2022-05-13 14:53:25 +02:00
|
|
|
REGISTER(VDetour);
|
2021-12-25 22:36:38 +01:00
|
|
|
#endif // IDETOUR_H
|