SDK: Add UObject::find_property and get_property

praydog · Jul 9, 2023 · dd89684 · dd89684
1 parent 059a13d
commit dd89684
Show file tree

Hide file tree

Showing 6 changed files with 86 additions and 2 deletions.
diff --git a/shared/sdk/FProperty.hpp b/shared/sdk/FProperty.hpp
@@ -2,8 +2,8 @@
 
 #include <cstdint>
 
-#include "UClass.hpp"
 #include "FField.hpp"
+#include "UClass.hpp"
 
 namespace sdk {
 class UProperty;

diff --git a/shared/sdk/UClass.cpp b/shared/sdk/UClass.cpp
@@ -10,6 +10,7 @@
 #include "UProperty.hpp"
 #include "FProperty.hpp"
 #include "FField.hpp"
+#include "FProperty.hpp"
 
 #include "UClass.hpp"
 
@@ -556,4 +557,32 @@ void UScriptStruct::update_offsets() {
         continue;
     }
 }
+
+FProperty* UStruct::find_property(std::wstring_view name) const {
+    for (auto super = this; super != nullptr; super = (UClass*)super->get_super_struct()) {
+        for (auto child = super->get_child_properties(); child != nullptr; child = child->get_next()) {
+            SPDLOG_INFO("[UStruct] Checking child property {}", utility::narrow(child->get_field_name().to_string()));
+
+            if (child->get_field_name().to_string() == name) {
+                return (FProperty*)child;
+            }
+        }
+    }
+
+    return nullptr;
+}
+
+UProperty* UStruct::find_uproperty(std::wstring_view name) const {
+    for (auto super = this; super != nullptr; super = (UClass*)super->get_super_struct()) {
+        for (auto child = super->get_children(); child != nullptr; child = child->get_next()) {
+            SPDLOG_INFO("[UStruct] Checking child UProperty {}", utility::narrow(child->get_fname().to_string()));
+
+            if (child->get_fname().to_string() == name) {
+                return (UProperty*)child;
+            }
+        }
+    }
+
+    return nullptr;
+}
 }
diff --git a/shared/sdk/UClass.hpp b/shared/sdk/UClass.hpp
@@ -6,6 +6,8 @@ namespace sdk {
 class UClass;
 class UStruct;
 class FField;
+class FProperty;
+class UProperty;
 
 class UField : public UObject {
 public:
@@ -50,6 +52,9 @@ class UStruct : public UField {
         return false;
     }
 
+    FProperty* find_property(std::wstring_view name) const;
+    UProperty* find_uproperty(std::wstring_view name) const;
+
 protected:
     static void resolve_field_offsets(uint32_t child_search_start);
     static void resolve_function_offsets(uint32_t child_search_start);

diff --git a/shared/sdk/UObject.cpp b/shared/sdk/UObject.cpp
@@ -1,6 +1,8 @@
 #include "UObjectArray.hpp"
 
 #include "UClass.hpp"
+#include "FProperty.hpp"
+#include "UProperty.hpp"
 #include "UObject.hpp"
 
 namespace sdk {
@@ -11,4 +13,40 @@ UClass* UObject::static_class() {
 bool UObject::is_a(UClass* other) const {
     return get_class()->is_a((UStruct*)other);
 }
+
+void* UObject::get_property_data(std::wstring_view name) const {
+    const auto klass = get_class();
+
+    if (klass == nullptr) {
+        return nullptr;
+    }
+
+    const auto prop = klass->find_property(name);
+
+    if (prop == nullptr) {
+        return nullptr;
+    }
+
+    const auto offset = prop->get_offset();
+
+    return (void*)((uintptr_t)this + offset);
+}
+
+void* UObject::get_uproperty_data(std::wstring_view name) const {
+    const auto klass = get_class();
+
+    if (klass == nullptr) {
+        return nullptr;
+    }
+
+    const auto prop = klass->find_uproperty(name);
+
+    if (prop == nullptr) {
+        return nullptr;
+    }
+
+    const auto offset = prop->get_offset();
+
+    return (void*)((uintptr_t)this + offset);
+}
 }
diff --git a/shared/sdk/UObject.hpp b/shared/sdk/UObject.hpp
@@ -9,6 +9,18 @@ class UObject : public UObjectBase {
 public:
     static UClass* static_class();
 
+    void* get_property_data(std::wstring_view name) const;
+    template <typename T>
+    T& get_property(std::wstring_view name) const {
+        return *(T*)get_property_data(name);
+    }
+
+    void* get_uproperty_data(std::wstring_view name) const;
+    template <typename T>
+    T& get_uproperty(std::wstring_view name) const {
+        return *(T*)get_uproperty_data(name);
+    }
+
     template <typename T>
     bool is_a() const {
         return is_a(T::static_class());

diff --git a/shared/sdk/UObjectBase.cpp b/shared/sdk/UObjectBase.cpp
@@ -113,7 +113,7 @@ void UObjectBase::update_process_event_index() try {
                     return utility::ExhaustionResult::CONTINUE;
                 }
 
-                if (*displacement == 0x0 || IsBadReadPtr((void*)*displacement, sizeof(void*) * 4)) {
+                if (*displacement == 0x0 || IsBadReadPtr((void*)*displacement, sizeof(void*) * 10)) {
                     return utility::ExhaustionResult::CONTINUE;
                 }