add: migrate source code of luahook

blcrack/cracker/utils.cpp

@@ -0,0 +1,158 @@
+#include "utils.hpp"
+
+#include <link.h>
+#include <spdlog/spdlog.h>
+#include <chrono>
+#include <thread>
+#include <sol/state_view.hpp>
+
+#include "il2cpp.hpp"
+
+struct enum_so_info {
+    const char* name;
+    void* base_addr;
+};
+
+static int get_so_base_callback(struct dl_phdr_info *info, size_t size, void *data) {
+    auto* enum_info = (enum_so_info *)data;
+    if (info->dlpi_name && strstr(info->dlpi_name, enum_info->name)) {
+        enum_info->base_addr = (void*)info->dlpi_addr;
+    }
+    return 0;
+}
+
+void* Utils::get_so_base_address(const std::string& name) {
+    enum_so_info so_info = {name.c_str(), nullptr};
+    dl_iterate_phdr(get_so_base_callback, &so_info);
+    if (so_info.base_addr) {
+        SPDLOG_INFO("Found {} at {}", name.c_str(), so_info.base_addr);
+        return so_info.base_addr;
+    }
+    // TODO: throw exception instead of returning nullptr
+    return nullptr;
+}
+
+static LuaScriptMgr_get_Inst get_lua_script_mgr_get_inst() {
+    Il2CppDomain* domain = il2cpp_domain_get();
+    size_t assembly_count = 0;
+    Il2CppAssembly** assemblies = il2cpp_domain_get_assemblies(domain, &assembly_count);
+
+    Il2CppClass* cls = nullptr;
+    for (size_t i = 0; i < assembly_count; ++i) {
+        Il2CppImage* image = il2cpp_assembly_get_image(assemblies[i]);
+        Il2CppClass* klass = il2cpp_class_from_name(image, "", "LuaScriptMgr");
+        if (klass != nullptr) {
+            cls = klass;
+            break;
+        }
+    }
+    if (cls != nullptr) {
+        MethodInfo* method_info = il2cpp_class_get_method_from_name(cls, "get_Inst", 0);
+        if (method_info != nullptr) {
+            auto method = reinterpret_cast<LuaScriptMgr_get_Inst>(method_info->methodPointer);
+            SPDLOG_INFO("Found LuaScriptMgr.get_Inst at {}", (void*)method_info->methodPointer);
+            return method;
+        }
+    }
+    SPDLOG_ERROR("Failed to get LuaScriptMgr.get_Inst");
+    throw std::runtime_error("Failed to get LuaScriptMgr.get_Inst");
+}
+
+lua_State* Utils::get_lua_state() {
+    LuaScriptMgr_get_Inst get_instance = get_lua_script_mgr_get_inst();
+    struct LuaScriptMgr_o* lua_script_mgr = get_instance();
+
+    LuaInterface_LuaState_o* luaState = nullptr;
+    while (luaState == nullptr) {
+        luaState = lua_script_mgr->fields.luaState;
+        std::this_thread::sleep_for(std::chrono::seconds(1));
+    }
+    SPDLOG_INFO("lua_State: {}", (void*)luaState->fields.L);
+
+    // std::this_thread::sleep_for(std::chrono::seconds(10));
+
+    return (lua_State*)luaState->fields.L;
+}
+
+std::string Utils::Lua::get_type_name(sol::type t) {
+    switch (t) {
+        case sol::type::nil: return "nil";
+        case sol::type::boolean: return "boolean";
+        case sol::type::lightuserdata: return "lightuserdata";
+        case sol::type::number: return "number";
+        case sol::type::string: return "string";
+        case sol::type::table: return "table";
+        case sol::type::function: return "function";
+        case sol::type::userdata: return "userdata";
+        case sol::type::thread: return "thread";
+        case sol::type::poly: return "poly";
+        default: return "unknown";
+    }
+}
+
+std::string Utils::Lua::get_type_name(sol::object& o) {
+    return get_type_name(o.get_type());
+}
+
+void Utils::Lua::print_table_fields_type(sol::table& t) {
+    for (auto& p : t) {
+        if (p.first.get_type() == sol::type::string) {
+            if (p.second.get_type() == sol::type::number) {
+                SPDLOG_INFO("[{}] = {}", p.first.as<std::string>(), p.second.as<double>());
+            } else if (p.second.get_type() == sol::type::boolean) {
+                SPDLOG_INFO("[{}] = {}", p.first.as<std::string>(), p.second.as<bool>());
+            } else if (p.second.get_type() == sol::type::string) {
+                SPDLOG_INFO("[{}] = {}", p.first.as<std::string>(), p.second.as<std::string>());
+            } else {
+                SPDLOG_INFO("[{}] -> {}", p.first.as<std::string>(), Utils::Lua::get_type_name(p.second));
+            }
+        } else {
+            if (p.second.get_type() == sol::type::number) {
+                SPDLOG_INFO("[{}] = {}", p.first.as<double>(), p.second.as<double>());
+            } else if (p.second.get_type() == sol::type::boolean) {
+                SPDLOG_INFO("[{}] = {}", p.first.as<double>(), p.second.as<bool>());
+            } else if (p.second.get_type() == sol::type::string) {
+                SPDLOG_INFO("[{}] = {}", p.first.as<double>(), p.second.as<std::string>());
+            } else {
+                SPDLOG_INFO("[{}] -> {}", p.first.as<double>(), Utils::Lua::get_type_name(p.second));
+            }
+        }
+    }
+}
+
+void Utils::Lua::print_traceback(sol::this_state& L) {
+    sol::state_view state(L);
+    sol::function trace = state["debug"]["traceback"];
+    std::string trace_str = trace();
+    SPDLOG_INFO("trace: {}", trace_str);
+}
+
+void Utils::Lua::print_object(sol::object& obj) {
+    auto type = obj.get_type();
+    if (type == sol::type::nil) {
+        SPDLOG_INFO("[nil]");
+    } else if (type == sol::type::boolean) {
+        bool b = obj.as<bool>();
+        SPDLOG_INFO("[boolean] = {}", b);
+    } else if (type == sol::type::number){
+        double number = obj.as<double>();
+        SPDLOG_INFO("[number] = {}", number);
+    } else if (type == sol::type::string) {
+        std::string str = obj.as<std::string>();
+        SPDLOG_INFO("[string] = {}", str);
+    } else if (type == sol::type::table) {
+        SPDLOG_INFO("[table]");
+        sol::table table = obj;
+        print_table_fields_type(table);
+    } else if (type == sol::type::function) {
+        SPDLOG_INFO("[function]");
+    } else if (type == sol::type::userdata) {
+        SPDLOG_INFO("[userdata]");
+    } else if (type == sol::type::thread) {
+        SPDLOG_INFO("[thread]");
+    } else if (type == sol::type::poly) {
+        SPDLOG_INFO("[poly]");
+    } else {
+        SPDLOG_INFO("Unknown type: {}", static_cast<int>(type));
+    }
+}