rtti::is_vtable

cursey · Sep 25, 2024 · 153b177 · 153b177
1 parent f21fff2
commit 153b177
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/include/utility/RTTI.hpp b/include/utility/RTTI.hpp
@@ -11,6 +11,7 @@ struct _s_RTTICompleteObjectLocator;
 
 namespace utility {
 namespace rtti {
+    bool is_vtable(const void* vtable);
     _s_RTTICompleteObjectLocator* get_locator(const void* obj);
     std::type_info* get_type_info(const void* obj);
     std::type_info* get_type_info(HMODULE m, std::string_view type_name);

diff --git a/src/RTTI.cpp b/src/RTTI.cpp
@@ -144,6 +144,28 @@ std::optional<HMODULE> cached_get_module_within(uintptr_t addr) {
 }
 }
 
+bool is_vtable(const void* vtable) {
+    if (vtable == nullptr) {
+        return false;
+    }
+
+    const auto module_within = detail::cached_get_module_within((uintptr_t)vtable);
+
+    if (!module_within) {
+        return false;
+    }
+
+    bool result = false;
+
+    detail::for_each(*module_within, [&](const detail::Vtable& entry) {
+        if (entry.vtable == (uintptr_t)vtable) {
+            result = true;
+        }
+    });
+
+    return result;
+}
+
 _s_RTTICompleteObjectLocator* get_locator(const void* obj) {
     if (obj == nullptr || *(void**)obj == nullptr) {
         return nullptr;