Added in some security

maximegmd · Dec 20, 2020 · b07c274 · Saibamen · Dec 20, 2020 · b07c274
1 parent b9170dd
commit b07c274
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/reverse/RTTI.h b/src/reverse/RTTI.h
@@ -34,7 +34,11 @@ struct CRTTIBaseType
 };
 
 struct CClassFunction;
-struct CBaseFunction;
+struct CBaseFunction
+{
+    uint8_t pad0[0x78];
+    int32_t flags;
+};
 
 struct CClass : CRTTIBaseType
 {

diff --git a/src/reverse/Scripting.cpp b/src/reverse/Scripting.cpp
@@ -83,6 +83,11 @@ bool Scripting::Execute(const std::string& aCommand, std::string& aReturnMessage
     auto* unk10 = engine->framework->unk10;
 
     auto func = CRTTISystem::Get()->GetGlobalFunction(REDString::Hash(funcName.c_str()));
+    if (!func || func->flags & 1)
+    {
+        aReturnMessage = "Function : " + funcName + " not found or is not a global.";
+        return false;
+    }
 
     const auto scriptable = unk10->GetTypeInstance(type);