mirror of
https://github.com/geode-sdk/geode.git
synced 2024-11-27 01:45:35 -05:00
119 lines
3.5 KiB
C++
119 lines
3.5 KiB
C++
#pragma once
|
|
|
|
#include <mach-o/dyld.h>
|
|
|
|
#include <type_traits>
|
|
#include <typeinfo>
|
|
#include <cstring>
|
|
|
|
namespace geode {
|
|
using dylib_t = void*;
|
|
struct PlatformInfo {
|
|
dylib_t m_dylib;
|
|
};
|
|
}
|
|
|
|
namespace geode::base {
|
|
GEODE_NOINLINE inline uintptr_t get() {
|
|
static uintptr_t base = _dyld_get_image_vmaddr_slide(0)+0x100000000;
|
|
return base;
|
|
}
|
|
}
|
|
|
|
namespace geode::cast {
|
|
using uinthalf_t = uint32_t;
|
|
using inthalf_t = int32_t;
|
|
|
|
struct DummyClass {
|
|
virtual ~DummyClass() {}
|
|
};
|
|
struct DummySingleClass : DummyClass {};
|
|
struct DummyClass2 {};
|
|
struct DummyMultipleClass : DummySingleClass, DummyClass2 {};
|
|
|
|
struct ClassTypeinfoType {
|
|
void** m_typeinfoVtable;
|
|
char const* m_typeinfoName;
|
|
};
|
|
|
|
struct SingleClassTypeinfoType : ClassTypeinfoType {
|
|
ClassTypeinfoType* m_baseClassTypeinfo;
|
|
};
|
|
|
|
#pragma pack(push, 1)
|
|
struct MultipleClassSingleEntryType {
|
|
ClassTypeinfoType* m_baseClassTypeinfo;
|
|
uint8_t m_visibilityFlag;
|
|
inthalf_t m_offset;
|
|
uint8_t m_padding[sizeof(inthalf_t) - 1];
|
|
};
|
|
#pragma pack(pop)
|
|
|
|
struct MultipleClassTypeinfoType : ClassTypeinfoType {
|
|
uint32_t m_flags;
|
|
uint32_t m_numBaseClass;
|
|
MultipleClassSingleEntryType m_baseClasses[0x100];
|
|
};
|
|
|
|
struct VtableTypeinfoType {
|
|
inthalf_t m_offset;
|
|
ClassTypeinfoType* m_typeinfo;
|
|
};
|
|
|
|
struct VtableType {
|
|
void* m_vtable[0x100];
|
|
};
|
|
|
|
struct CompleteVtableType : VtableTypeinfoType, VtableType {};
|
|
|
|
inline void** typeinfoVtableOf(void* ptr) {
|
|
auto vftable = *reinterpret_cast<VtableType**>(ptr);
|
|
|
|
auto typeinfoPtr = static_cast<VtableTypeinfoType*>(static_cast<CompleteVtableType*>(vftable));
|
|
|
|
return typeinfoPtr->m_typeinfo->m_typeinfoVtable;
|
|
}
|
|
|
|
inline void* traverseTypeinfoFor(void* ptr, ClassTypeinfoType* typeinfo, char const* afterIdent) {
|
|
DummySingleClass dummySingleClass;
|
|
DummyMultipleClass dummyMultipleClass;
|
|
|
|
{
|
|
auto optionIdent = typeinfo->m_typeinfoName;
|
|
if (std::strcmp(optionIdent, afterIdent) == 0) {
|
|
return ptr;
|
|
}
|
|
}
|
|
if (typeinfo->m_typeinfoVtable == typeinfoVtableOf(&dummySingleClass)) {
|
|
auto siTypeinfo = static_cast<SingleClassTypeinfoType*>(typeinfo);
|
|
return traverseTypeinfoFor(ptr, siTypeinfo->m_baseClassTypeinfo, afterIdent);
|
|
}
|
|
else if (typeinfo->m_typeinfoVtable == typeinfoVtableOf(&dummyMultipleClass)) {
|
|
auto vmiTypeinfo = static_cast<MultipleClassTypeinfoType*>(typeinfo);
|
|
for (int i = 0; i < vmiTypeinfo->m_numBaseClass; ++i) {
|
|
auto& entry = vmiTypeinfo->m_baseClasses[i];
|
|
auto optionPtr = reinterpret_cast<std::byte*>(ptr) + entry.m_offset;
|
|
auto ret = traverseTypeinfoFor(optionPtr, entry.m_baseClassTypeinfo, afterIdent);
|
|
if (ret != nullptr) return ret;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
template <class After, class Before>
|
|
inline After typeinfo_cast(Before ptr) {
|
|
static_assert(
|
|
std::is_polymorphic_v<std::remove_pointer_t<Before>>,
|
|
"Input is not a polymorphic type"
|
|
);
|
|
auto basePtr = dynamic_cast<void*>(ptr);
|
|
auto vftable = *reinterpret_cast<VtableType**>(basePtr);
|
|
auto typeinfo = static_cast<VtableTypeinfoType*>(static_cast<CompleteVtableType*>(vftable))->m_typeinfo;
|
|
|
|
auto afterTypeinfo = reinterpret_cast<ClassTypeinfoType const*>(&typeid(std::remove_pointer_t<After>));
|
|
auto afterIdent = afterTypeinfo->m_typeinfoName;
|
|
|
|
return static_cast<After>(traverseTypeinfoFor(basePtr, typeinfo, afterIdent));
|
|
}
|
|
}
|