diff --git a/CMakeLists.txt b/CMakeLists.txt index 7fc0e5ee..a7a52704 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -15,6 +15,7 @@ include(cmake/git.cmake) # Fetch modules message("\nFetching modules") +include(cmake/asmjit.cmake) include(cmake/minhook.cmake) include(cmake/async-logger.cmake) include(cmake/pugixml.cmake) @@ -58,10 +59,11 @@ target_include_directories(YimMenu PRIVATE "${gtav_classes_SOURCE_DIR}" "${imgui_SOURCE_DIR}" "${minhook_SOURCE_DIR}/src/hde" + "${asmjit_SOURCE_DIR}/src" ) target_precompile_headers(YimMenu PRIVATE "${SRC_DIR}/common.hpp") -target_link_libraries(YimMenu PRIVATE pugixml minhook AsyncLogger dbghelp imgui cpr lua_static) +target_link_libraries(YimMenu PRIVATE pugixml minhook AsyncLogger dbghelp imgui cpr lua_static asmjit::asmjit) # Warnings as errors set_property(TARGET YimMenu PROPERTY COMPILE_WARNING_AS_ERROR ON) diff --git a/cmake/asmjit.cmake b/cmake/asmjit.cmake new file mode 100644 index 00000000..3e658957 --- /dev/null +++ b/cmake/asmjit.cmake @@ -0,0 +1,11 @@ +include(FetchContent) + +message("asmjit") +FetchContent_Declare( + asmjit + GIT_REPOSITORY https://github.com/asmjit/asmjit.git + GIT_TAG ffac9f36fb045dd2c6a81e1b5b9ccc115e5ef924 + GIT_PROGRESS TRUE +) +FetchContent_MakeAvailable(asmjit) +set_property(TARGET asmjit PROPERTY CXX_STANDARD 23) diff --git a/docs/lua/classes/value_wrapper.md b/docs/lua/classes/value_wrapper.md new file mode 100644 index 00000000..8d382bdb --- /dev/null +++ b/docs/lua/classes/value_wrapper.md @@ -0,0 +1,31 @@ +# Class: value_wrapper + +Class for wrapping parameters and return value of functions, used mostly by the dynamic_hook system. + +## Functions (2) + +### `get()` + +Get the value currently contained by the wrapper. + +- **Returns:** + - `any`: The current value. + +**Example Usage:** +```lua +any = value_wrapper:get() +``` + +### `set(new_value)` + +Set the new value contained by the wrapper. + +- **Parameters:** + - `new_value` (any): The new value. + +**Example Usage:** +```lua +value_wrapper:set(new_value) +``` + + diff --git a/docs/lua/tables/memory.md b/docs/lua/tables/memory.md index 0a1818a0..f4fbe537 100644 --- a/docs/lua/tables/memory.md +++ b/docs/lua/tables/memory.md @@ -2,7 +2,7 @@ Table containing helper functions related to process memory. -## Functions (5) +## Functions (6) ### `scan_pattern(pattern)` @@ -68,4 +68,39 @@ pointer = memory.allocate(size) memory.free(ptr) ``` +### `dynamic_hook(hook_name, return_type, param_types, target_func_ptr, pre_callback, post_callback)` + +**Example Usage:** +```lua +local ptr = memory.scan_pattern("some ida sig") +memory.dynamic_hook("test_hook", "float", {"const char*"}, ptr, +function(ret_val, str) + + --str:set("replaced str") + ret_val:set(69.69) + log.info("pre callback from lua", ret_val:get(), str:get()) + + -- false for skipping the original function call + return false +end, +function(ret_val, str) + log.info("post callback from lua 1", ret_val:get(), str:get()) + ret_val:set(79.69) + log.info("post callback from lua 2", ret_val:get(), str:get()) +end) +``` + +- **Parameters:** + - `hook_name` (string): The name of the hook. + - `return_type` (string): Type of the return value of the detoured function. + - `param_types` (table): Types of the parameters of the detoured function. + - `target_func_ptr` (memory.pointer): The pointer to the function to detour. + - `pre_callback` (function): The function that will be called before the original function is about to be called. The callback must match the following signature: ( return_value (value_wrapper), arg1 (value_wrapper), arg2 (value_wrapper), ... ) -> Returns true or false (boolean) depending on whether you want the original function to be called. + - `post_callback` (function): The function that will be called after the original function is called (or just after the pre callback is called, if the original function was skipped). The callback must match the following signature: ( return_value (value_wrapper), arg1 (value_wrapper), arg2 (value_wrapper), ... ) -> void + +**Example Usage:** +```lua +memory.dynamic_hook(hook_name, return_type, param_types, target_func_ptr, pre_callback, post_callback) +``` + diff --git a/src/hooking/detour_hook.hpp b/src/hooking/detour_hook.hpp index 94c6c5dc..87908ff2 100644 --- a/src/hooking/detour_hook.hpp +++ b/src/hooking/detour_hook.hpp @@ -29,6 +29,11 @@ namespace big return static_cast(m_original); } + void** get_original_ptr() + { + return &m_original; + } + void fix_hook_address(); private: diff --git a/src/lua/bindings/memory.cpp b/src/lua/bindings/memory.cpp index c4e5a609..f9b7ac2d 100644 --- a/src/lua/bindings/memory.cpp +++ b/src/lua/bindings/memory.cpp @@ -1,5 +1,6 @@ #include "memory.hpp" +#include "lua/lua_manager.hpp" #include "memory/module.hpp" #include "memory/pattern.hpp" #include "pointers.hpp" @@ -130,6 +131,145 @@ namespace lua::memory }); } + value_wrapper_t::value_wrapper_t(char* val, type_info_t type) + { + m_value = val; + m_type = type; + } + + sol::object value_wrapper_t::get(sol::this_state state_) + { + if (m_type == type_info_t::boolean_) + { + return sol::make_object(state_, *(bool*)m_value); + } + else if (m_type == type_info_t::string_) + { + return sol::make_object(state_, *(const char**)m_value); + } + else if (m_type == type_info_t::integer_) + { + return sol::make_object(state_, *(int64_t*)m_value); + } + else if (m_type == type_info_t::float_) + { + return sol::make_object(state_, *(float*)m_value); + } + else if (m_type == type_info_t::double_) + { + return sol::make_object(state_, *(double*)m_value); + } + else + { + return sol::nil; + } + } + + void value_wrapper_t::set(sol::object new_val, sol::this_state state_) + { + if (m_type == type_info_t::boolean_ && new_val.is()) + { + *(bool*)m_value = new_val.as(); + } + else if (m_type == type_info_t::string_ && new_val.is()) + { + *(const char**)m_value = new_val.as(); + } + else if (m_type == type_info_t::integer_ && new_val.is()) + { + *(int64_t*)m_value = new_val.as(); + } + else if (m_type == type_info_t::float_ && new_val.is()) + { + *(float*)m_value = new_val.as(); + } + else if (m_type == type_info_t::double_ && new_val.is()) + { + *(double*)m_value = new_val.as(); + } + } + + static std::unordered_map> target_func_ptr_to_hook; + + static bool pre_callback(const runtime_func_t::parameters_t* params, const uint8_t param_count, runtime_func_t::return_value_t* return_value, const uintptr_t target_func_ptr) + { + const auto& dyn_hook = target_func_ptr_to_hook[target_func_ptr]; + return big::g_lua_manager + ->dynamic_hook_pre_callbacks(target_func_ptr, dyn_hook->m_return_type, return_value, dyn_hook->m_param_types, params, param_count); + } + + static void post_callback(const runtime_func_t::parameters_t* params, const uint8_t param_count, runtime_func_t::return_value_t* return_value, const uintptr_t target_func_ptr) + { + const auto& dyn_hook = target_func_ptr_to_hook[target_func_ptr]; + big::g_lua_manager->dynamic_hook_post_callbacks(target_func_ptr, dyn_hook->m_return_type, return_value, dyn_hook->m_param_types, params, param_count); + } + + // Lua API: Function + // Table: memory + // Name: dynamic_hook + // Param: hook_name: string: The name of the hook. + // Param: return_type: string: Type of the return value of the detoured function. + // Param: param_types: table: Types of the parameters of the detoured function. + // Param: target_func_ptr: memory.pointer: The pointer to the function to detour. + // Param: pre_callback: function: The function that will be called before the original function is about to be called. The callback must match the following signature: ( return_value (value_wrapper), arg1 (value_wrapper), arg2 (value_wrapper), ... ) -> Returns true or false (boolean) depending on whether you want the original function to be called. + // Param: post_callback: function: The function that will be called after the original function is called (or just after the pre callback is called, if the original function was skipped). The callback must match the following signature: ( return_value (value_wrapper), arg1 (value_wrapper), arg2 (value_wrapper), ... ) -> void + // **Example Usage:** + // ```lua + // local ptr = memory.scan_pattern("some ida sig") + // memory.dynamic_hook("test_hook", "float", {"const char*"}, ptr, + // function(ret_val, str) + // + // --str:set("replaced str") + // ret_val:set(69.69) + // log.info("pre callback from lua", ret_val:get(), str:get()) + // + // -- false for skipping the original function call + // return false + // end, + // function(ret_val, str) + // log.info("post callback from lua 1", ret_val:get(), str:get()) + // ret_val:set(79.69) + // log.info("post callback from lua 2", ret_val:get(), str:get()) + // end) + // ``` + static void dynamic_hook(const std::string& hook_name, const std::string& return_type, sol::table param_types_table, lua::memory::pointer& target_func_ptr_obj, sol::protected_function pre_lua_callback, sol::protected_function post_lua_callback, sol::this_state state_) + { + const auto target_func_ptr = target_func_ptr_obj.get_address(); + if (!target_func_ptr_to_hook.contains(target_func_ptr)) + { + std::vector param_types; + for (const auto& [k, v] : param_types_table) + { + if (v.is()) + { + param_types.push_back(v.as()); + } + } + + std::unique_ptr runtime_func = std::make_unique(); + const auto jitted_func = runtime_func->make_jit_func(return_type, param_types, asmjit::Arch::kHost, pre_callback, post_callback, target_func_ptr); + + target_func_ptr_to_hook.emplace(target_func_ptr, std::move(runtime_func)); + + // TODO: The detour_hook is never cleaned up on menu unload. + target_func_ptr_to_hook[target_func_ptr]->create_and_enable_hook(hook_name, target_func_ptr, jitted_func); + } + + big::lua_module* module = sol::state_view(state_)["!this"]; + if (!module) + { + return; + } + if (pre_lua_callback.valid()) + { + module->m_dynamic_hook_pre_callbacks[target_func_ptr].push_back(pre_lua_callback); + } + if (post_lua_callback.valid()) + { + module->m_dynamic_hook_post_callbacks[target_func_ptr].push_back(post_lua_callback); + } + } + void bind(sol::state& state) { auto ns = state["memory"].get_or_create(); @@ -169,5 +309,8 @@ namespace lua::memory ns["ptr_to_handle"] = ptr_to_handle; ns["allocate"] = allocate; ns["free"] = free; + + ns.new_usertype("value_wrapper", "get", &value_wrapper_t::get, "set", &value_wrapper_t::set); + ns["dynamic_hook"] = dynamic_hook; } } \ No newline at end of file diff --git a/src/lua/bindings/memory.hpp b/src/lua/bindings/memory.hpp index 5acd4f6a..f285b3d2 100644 --- a/src/lua/bindings/memory.hpp +++ b/src/lua/bindings/memory.hpp @@ -1,4 +1,5 @@ #pragma once +#include "lua/bindings/type_info_t.hpp" #include "lua/lua_module.hpp" namespace lua::memory @@ -13,7 +14,6 @@ namespace lua::memory uint64_t m_address; public: - // Lua API: Constructor // Class: pointer // Param: address: integer: Address @@ -206,5 +206,32 @@ namespace lua::memory uint64_t get_address() const; }; + // Lua API: Class + // Name: value_wrapper + // Class for wrapping parameters and return value of functions, used mostly by the dynamic_hook system. + + class value_wrapper_t + { + char* m_value = nullptr; + type_info_t m_type = type_info_t::none_; + + public: + value_wrapper_t(char* val, type_info_t type); + + // Lua API: Function + // Class: value_wrapper + // Name: get + // Returns: any: The current value. + // Get the value currently contained by the wrapper. + sol::object get(sol::this_state state_); + + // Lua API: Function + // Class: value_wrapper + // Name: set + // Param: new_value: any: The new value. + // Set the new value contained by the wrapper. + void set(sol::object new_val, sol::this_state state_); + }; + void bind(sol::state& state); } \ No newline at end of file diff --git a/src/lua/bindings/runtime_func_t.hpp b/src/lua/bindings/runtime_func_t.hpp new file mode 100644 index 00000000..5eeabfeb --- /dev/null +++ b/src/lua/bindings/runtime_func_t.hpp @@ -0,0 +1,477 @@ +#pragma once +#include "hooking/detour_hook.hpp" +#include "lua/bindings/type_info_t.hpp" + +#include + +namespace lua::memory +{ + class runtime_func_t + { + std::vector m_jit_function_buffer; + asmjit::x86::Mem m_args_stack; + + std::unique_ptr m_detour; + + public: + type_info_t m_return_type; + std::vector m_param_types; + + struct parameters_t + { + template + void set(const uint8_t idx, const T val) const + { + *(T*)get_arg_ptr(idx) = val; + } + + template + T get(const uint8_t idx) const + { + return *(T*)get_arg_ptr(idx); + } + + // asm depends on this specific type + // we the runtime_func allocates stack space that is set to point here (check asmjit::compiler.newStack calls) + volatile uintptr_t m_arguments; + + // must be char* for aliasing rules to work when reading back out + char* get_arg_ptr(const uint8_t idx) const + { + return ((char*)&m_arguments) + sizeof(uintptr_t) * idx; + } + }; + + class return_value_t + { + uintptr_t m_return_value; + + public: + unsigned char* get() const + { + return (unsigned char*)&m_return_value; + } + }; + + typedef bool (*user_pre_callback_t)(const parameters_t* params, const uint8_t parameters_count, return_value_t* return_value, const uintptr_t target_func_ptr); + typedef void (*user_post_callback_t)(const parameters_t* params, const uint8_t parameters_count, return_value_t* return_value, const uintptr_t target_func_ptr); + + runtime_func_t() + { + m_detour = std::make_unique(); + m_return_type = type_info_t::none_; + } + + ~runtime_func_t() + { + } + + // Construct a callback given the raw signature at runtime. 'Callback' param is the C stub to transfer to, + // where parameters can be modified through a structure which is written back to the parameter slots depending + // on calling convention. + uintptr_t make_jit_func(const asmjit::FuncSignature& sig, const asmjit::Arch arch, const user_pre_callback_t pre_callback, const user_post_callback_t post_callback, const uintptr_t target_func_ptr) + { + asmjit::CodeHolder code; + auto env = asmjit::Environment::host(); + env.setArch(arch); + code.init(env); + + // initialize function + asmjit::x86::Compiler cc(&code); + asmjit::FuncNode* func = cc.addFunc(sig); + + asmjit::StringLogger log; + // clang-format off + const auto format_flags = + asmjit::FormatFlags::kMachineCode | asmjit::FormatFlags::kExplainImms | asmjit::FormatFlags::kRegCasts | + asmjit::FormatFlags::kHexImms | asmjit::FormatFlags::kHexOffsets | asmjit::FormatFlags::kPositions; + // clang-format on + + log.addFlags(format_flags); + code.setLogger(&log); + + // too small to really need it + func->frame().resetPreservedFP(); + + // map argument slots to registers, following abi. + std::vector arg_registers; + for (uint8_t arg_index = 0; arg_index < sig.argCount(); arg_index++) + { + const auto arg_type = sig.args()[arg_index]; + + asmjit::x86::Reg arg; + if (is_general_register(arg_type)) + { + arg = cc.newUIntPtr(); + } + else if (is_XMM_register(arg_type)) + { + arg = cc.newXmm(); + } + else + { + LOG(FATAL) << "Parameters wider than 64bits not supported"; + return 0; + } + + func->setArg(arg_index, arg); + arg_registers.push_back(arg); + } + + // setup the stack structure to hold arguments for user callback + uint32_t stack_size = (uint32_t)(sizeof(uintptr_t) * sig.argCount()); + m_args_stack = cc.newStack(stack_size, 16); + asmjit::x86::Mem args_stack_index(m_args_stack); + + // assigns some register as index reg + asmjit::x86::Gp i = cc.newUIntPtr(); + + // stack_index <- stack[i]. + args_stack_index.setIndex(i); + + // r/w are sizeof(uintptr_t) width now + args_stack_index.setSize(sizeof(uintptr_t)); + + // set i = 0 + cc.mov(i, 0); + // mov from arguments registers into the stack structure + for (uint8_t argIdx = 0; argIdx < sig.argCount(); argIdx++) + { + const auto argType = sig.args()[argIdx]; + + // have to cast back to explicit register types to gen right mov type + if (is_general_register(argType)) + { + cc.mov(args_stack_index, arg_registers.at(argIdx).as()); + } + else if (is_XMM_register(argType)) + { + cc.movq(args_stack_index, arg_registers.at(argIdx).as()); + } + else + { + LOG(FATAL) << "Parameters wider than 64bits not supported"; + return 0; + } + + // next structure slot (+= sizeof(uintptr_t)) + cc.add(i, sizeof(uintptr_t)); + } + + // get pointer to stack structure and pass it to the user pre callback + asmjit::x86::Gp arg_struct = cc.newUIntPtr("arg_struct"); + cc.lea(arg_struct, m_args_stack); + + // fill reg to pass struct arg count to callback + asmjit::x86::Gp arg_param_count = cc.newUInt8(); + cc.mov(arg_param_count, (uint8_t)sig.argCount()); + + // create buffer for ret val + asmjit::x86::Mem return_stack = cc.newStack(sizeof(uintptr_t), 16); + asmjit::x86::Gp return_struct = cc.newUIntPtr("return_struct"); + cc.lea(return_struct, return_stack); + + // fill reg to pass target function pointer to callback + asmjit::x86::Gp target_func_ptr_reg = cc.newUIntPtr(); + cc.mov(target_func_ptr_reg, target_func_ptr); + + asmjit::Label original_invoke_label = cc.newLabel(); + asmjit::Label skip_original_invoke_label = cc.newLabel(); + + // invoke the user pre callback + asmjit::InvokeNode* pre_callback_invoke_node; + cc.invoke(&pre_callback_invoke_node, (uintptr_t)pre_callback, asmjit::FuncSignatureT()); + + // call to user provided function (use ABI of host compiler) + pre_callback_invoke_node->setArg(0, arg_struct); + pre_callback_invoke_node->setArg(1, arg_param_count); + pre_callback_invoke_node->setArg(2, return_struct); + pre_callback_invoke_node->setArg(3, target_func_ptr_reg); + + // create a register for the user pre callback's return value + // Note: the size of the register is important for the test instruction. newUInt8 since the pre callback returns a bool. + asmjit::x86::Gp pre_callback_return_val = cc.newUInt8("pre_callback_return_val"); + // store the callback return value + pre_callback_invoke_node->setRet(0, pre_callback_return_val); + + // if the callback return value is zero, skip orig. + cc.test(pre_callback_return_val, pre_callback_return_val); + cc.jz(skip_original_invoke_label); + + // label to invoke the original function + cc.bind(original_invoke_label); + + // mov from arguments stack structure into regs + cc.mov(i, 0); // reset idx + for (uint8_t arg_idx = 0; arg_idx < sig.argCount(); arg_idx++) + { + const auto argType = sig.args()[arg_idx]; + + if (is_general_register(argType)) + { + cc.mov(arg_registers.at(arg_idx).as(), args_stack_index); + } + else if (is_XMM_register(argType)) + { + cc.movq(arg_registers.at(arg_idx).as(), args_stack_index); + } + else + { + LOG(FATAL) << "Parameters wider than 64bits not supported"; + return 0; + } + + // next structure slot (+= sizeof(uint64_t)) + cc.add(i, sizeof(uint64_t)); + } + + // deref the trampoline ptr (holder must live longer, must be concrete reg since push later) + asmjit::x86::Gp original_ptr = cc.zbx(); + cc.mov(original_ptr, m_detour->get_original_ptr()); + cc.mov(original_ptr, asmjit::x86::ptr(original_ptr)); + + asmjit::InvokeNode* original_invoke_node; + cc.invoke(&original_invoke_node, original_ptr, sig); + for (uint8_t arg_index = 0; arg_index < sig.argCount(); arg_index++) + { + original_invoke_node->setArg(arg_index, arg_registers.at(arg_index)); + } + + if (sig.hasRet()) + { + if (is_general_register(sig.ret())) + { + asmjit::x86::Gp tmp = cc.newUIntPtr(); + original_invoke_node->setRet(0, tmp); + cc.mov(return_stack, tmp); + } + else + { + asmjit::x86::Xmm tmp = cc.newXmm(); + original_invoke_node->setRet(0, tmp); + cc.movq(return_stack, tmp); + } + } + + cc.bind(skip_original_invoke_label); + + asmjit::InvokeNode* post_callback_invoke_node; + cc.invoke(&post_callback_invoke_node, (uintptr_t)post_callback, asmjit::FuncSignatureT()); + + // Set arguments for the post callback + post_callback_invoke_node->setArg(0, arg_struct); + post_callback_invoke_node->setArg(1, arg_param_count); + post_callback_invoke_node->setArg(2, return_struct); + post_callback_invoke_node->setArg(3, target_func_ptr_reg); + + if (sig.hasRet()) + { + asmjit::x86::Mem return_stack_index(return_stack); + return_stack_index.setSize(sizeof(uintptr_t)); + if (is_general_register(sig.ret())) + { + asmjit::x86::Gp tmp2 = cc.newUIntPtr(); + cc.mov(tmp2, return_stack_index); + cc.ret(tmp2); + } + else + { + asmjit::x86::Xmm tmp2 = cc.newXmm(); + cc.movq(tmp2, return_stack_index); + cc.ret(tmp2); + } + } + + cc.endFunc(); + + // write to buffer + cc.finalize(); + + // worst case, overestimates for case trampolines needed + code.flatten(); + size_t size = code.codeSize(); + + // Allocate a virtual memory (executable). + m_jit_function_buffer.reserve(size); + + DWORD old_protect; + VirtualProtect(m_jit_function_buffer.data(), size, PAGE_EXECUTE_READWRITE, &old_protect); + + // if multiple sections, resolve linkage (1 atm) + if (code.hasUnresolvedLinks()) + { + code.resolveUnresolvedLinks(); + } + + // Relocate to the base-address of the allocated memory. + code.relocateToBase((uintptr_t)m_jit_function_buffer.data()); + code.copyFlattenedData(m_jit_function_buffer.data(), size); + + LOG(VERBOSE) << "JIT Stub: " << log.data(); + + return (uintptr_t)m_jit_function_buffer.data(); + } + + // Construct a callback given the typedef as a string. Types are any valid C/C++ data type (basic types), and pointers to + // anything are just a uintptr_t. Calling convention is defaulted to whatever is typical for the compiler you use, you can override with + // stdcall, fastcall, or cdecl (cdecl is default on x86). On x64 those map to the same thing. + uintptr_t make_jit_func(const std::string& return_type, const std::vector& param_types, const asmjit::Arch arch, const user_pre_callback_t pre_callback, const user_post_callback_t post_callback, const uintptr_t target_func_ptr, std::string call_convention = "") + { + auto get_type_info_from_string = [](const std::string& s) { + if ((s.contains("const") && s.contains("char") && s.contains("*")) || s.contains("string")) + { + return type_info_t::string_; + } + else if (s.contains("bool")) + { + return type_info_t::boolean_; + } + else if (s.contains("ptr") || s.contains("pointer") || s.contains("*")) + { + // passing lua::memory::pointer + return type_info_t::ptr_; + } + else if (s.contains("float")) + { + return type_info_t::float_; + } + else if (s.contains("double")) + { + return type_info_t::double_; + } + else + { + return type_info_t::integer_; + } + }; + + m_return_type = get_type_info_from_string(return_type); + + asmjit::FuncSignature sig(get_call_convention(call_convention), asmjit::FuncSignature::kNoVarArgs, get_type_id(return_type)); + + for (const std::string& s : param_types) + { + sig.addArg(get_type_id(s)); + m_param_types.push_back(get_type_info_from_string(s)); + } + + return make_jit_func(sig, arch, pre_callback, post_callback, target_func_ptr); + } + + void create_and_enable_hook(const std::string& hook_name, uintptr_t target_func_ptr, uintptr_t jitted_func_ptr) + { + m_detour->set_instance(hook_name, (void*)target_func_ptr, (void*)jitted_func_ptr); + + m_detour->enable(); + } + + private: + // does a given type fit in a general purpose register (i.e. is it integer type) + bool is_general_register(const asmjit::TypeId type_id) const + { + switch (type_id) + { + case asmjit::TypeId::kInt8: + case asmjit::TypeId::kUInt8: + case asmjit::TypeId::kInt16: + case asmjit::TypeId::kUInt16: + case asmjit::TypeId::kInt32: + case asmjit::TypeId::kUInt32: + case asmjit::TypeId::kInt64: + case asmjit::TypeId::kUInt64: + case asmjit::TypeId::kIntPtr: + case asmjit::TypeId::kUIntPtr: return true; + default: return false; + } + } + + // float, double, simd128 + bool is_XMM_register(const asmjit::TypeId type_id) const + { + switch (type_id) + { + case asmjit::TypeId::kFloat32: + case asmjit::TypeId::kFloat64: return true; + default: return false; + } + } + + asmjit::CallConvId get_call_convention(const std::string& conv) + { + if (conv == "cdecl") + { + return asmjit::CallConvId::kCDecl; + } + else if (conv == "stdcall") + { + return asmjit::CallConvId::kStdCall; + } + else if (conv == "fastcall") + { + return asmjit::CallConvId::kFastCall; + } + + return asmjit::CallConvId::kHost; + } + + asmjit::TypeId get_type_id(const std::string& type) + { + if (type.find('*') != std::string::npos) + { + return asmjit::TypeId::kUIntPtr; + } + +#define TYPEID_MATCH_STR_IF(var, T) \ + if (var == #T) \ + { \ + return asmjit::TypeId(asmjit::TypeUtils::TypeIdOfT::kTypeId); \ + } +#define TYPEID_MATCH_STR_ELSEIF(var, T) \ + else if (var == #T) \ + { \ + return asmjit::TypeId(asmjit::TypeUtils::TypeIdOfT::kTypeId); \ + } + + TYPEID_MATCH_STR_IF(type, signed char) + TYPEID_MATCH_STR_ELSEIF(type, unsigned char) + TYPEID_MATCH_STR_ELSEIF(type, short) + TYPEID_MATCH_STR_ELSEIF(type, unsigned short) + TYPEID_MATCH_STR_ELSEIF(type, int) + TYPEID_MATCH_STR_ELSEIF(type, unsigned int) + TYPEID_MATCH_STR_ELSEIF(type, long) + TYPEID_MATCH_STR_ELSEIF(type, unsigned long) +#ifdef POLYHOOK2_OS_WINDOWS + TYPEID_MATCH_STR_ELSEIF(type, __int64) + TYPEID_MATCH_STR_ELSEIF(type, unsigned __int64) +#endif + TYPEID_MATCH_STR_ELSEIF(type, long long) + TYPEID_MATCH_STR_ELSEIF(type, unsigned long long) + TYPEID_MATCH_STR_ELSEIF(type, char) + TYPEID_MATCH_STR_ELSEIF(type, char16_t) + TYPEID_MATCH_STR_ELSEIF(type, char32_t) + TYPEID_MATCH_STR_ELSEIF(type, wchar_t) + TYPEID_MATCH_STR_ELSEIF(type, uint8_t) + TYPEID_MATCH_STR_ELSEIF(type, int8_t) + TYPEID_MATCH_STR_ELSEIF(type, uint16_t) + TYPEID_MATCH_STR_ELSEIF(type, int16_t) + TYPEID_MATCH_STR_ELSEIF(type, int32_t) + TYPEID_MATCH_STR_ELSEIF(type, uint32_t) + TYPEID_MATCH_STR_ELSEIF(type, uint64_t) + TYPEID_MATCH_STR_ELSEIF(type, int64_t) + TYPEID_MATCH_STR_ELSEIF(type, float) + TYPEID_MATCH_STR_ELSEIF(type, double) + TYPEID_MATCH_STR_ELSEIF(type, bool) + TYPEID_MATCH_STR_ELSEIF(type, void) + else if (type == "intptr_t") + { + return asmjit::TypeId::kIntPtr; + } + else if (type == "uintptr_t") + { + return asmjit::TypeId::kUIntPtr; + } + + return asmjit::TypeId::kVoid; + } + }; +} \ No newline at end of file diff --git a/src/lua/bindings/type_info_t.hpp b/src/lua/bindings/type_info_t.hpp new file mode 100644 index 00000000..7f254c9f --- /dev/null +++ b/src/lua/bindings/type_info_t.hpp @@ -0,0 +1,15 @@ +#pragma once + +namespace lua::memory +{ + enum type_info_t + { + none_, + boolean_, + string_, + integer_, + ptr_, + float_, + double_ + }; +} \ No newline at end of file diff --git a/src/lua/lua_manager.cpp b/src/lua/lua_manager.cpp index 3b315fff..ecea26fe 100644 --- a/src/lua/lua_manager.cpp +++ b/src/lua/lua_manager.cpp @@ -14,13 +14,13 @@ namespace big { rename(file, new_module_path); } - catch(const std::filesystem::filesystem_error& e) + catch (const std::filesystem::filesystem_error& e) { LOG(FATAL) << "Failed to move Lua file: " << e.what(); return std::nullopt; } - return { new_module_path }; + return {new_module_path}; } lua_manager::lua_manager(folder scripts_folder, folder scripts_config_folder) : @@ -59,7 +59,8 @@ namespace big for (const auto& script_path : script_paths) { - const auto new_module_path = move_file_relative_to_folder(m_scripts_folder.get_path(), m_disabled_scripts_folder.get_path(), script_path); + const auto new_module_path = + move_file_relative_to_folder(m_scripts_folder.get_path(), m_disabled_scripts_folder.get_path(), script_path); if (new_module_path) { load_module(*new_module_path); @@ -84,7 +85,8 @@ namespace big for (const auto& script_path : script_paths) { - const auto new_module_path = move_file_relative_to_folder(m_disabled_scripts_folder.get_path(), m_scripts_folder.get_path(), script_path); + const auto new_module_path = + move_file_relative_to_folder(m_disabled_scripts_folder.get_path(), m_scripts_folder.get_path(), script_path); if (new_module_path) { load_module(*new_module_path); @@ -173,6 +175,64 @@ namespace big } } + bool lua_manager::dynamic_hook_pre_callbacks(const uintptr_t target_func_ptr, lua::memory::type_info_t return_type, lua::memory::runtime_func_t::return_value_t* return_value, std::vector param_types, const lua::memory::runtime_func_t::parameters_t* params, const uint8_t param_count) + { + std::scoped_lock guard(m_module_lock); + + bool call_orig_if_true = true; + + for (const auto& module : m_modules) + { + const auto it = module->m_dynamic_hook_pre_callbacks.find(target_func_ptr); + if (it != module->m_dynamic_hook_pre_callbacks.end()) + { + sol::object return_value_obj = module->to_lua(return_value, return_type); + std::vector args; + for (uint8_t i = 0; i < param_count; i++) + { + args.push_back(module->to_lua(params, i, param_types)); + } + + for (const auto& cb : it->second) + { + const auto new_call_orig_if_true = cb(return_value_obj, sol::as_args(args)); + + if (call_orig_if_true && new_call_orig_if_true.valid() && new_call_orig_if_true.get_type() == sol::type::boolean + && new_call_orig_if_true.get() == false) + { + call_orig_if_true = false; + } + } + } + } + + return call_orig_if_true; + } + + void lua_manager::dynamic_hook_post_callbacks(const uintptr_t target_func_ptr, lua::memory::type_info_t return_type, lua::memory::runtime_func_t::return_value_t* return_value, std::vector param_types, const lua::memory::runtime_func_t::parameters_t* params, const uint8_t param_count) + { + std::scoped_lock guard(m_module_lock); + + for (const auto& module : m_modules) + { + const auto it = module->m_dynamic_hook_post_callbacks.find(target_func_ptr); + if (it != module->m_dynamic_hook_post_callbacks.end()) + { + sol::object return_value_obj = module->to_lua(return_value, return_type); + std::vector args; + for (uint8_t i = 0; i < param_count; i++) + { + args.push_back(module->to_lua(params, i, param_types)); + } + + for (const auto& cb : it->second) + { + cb(return_value_obj, sol::as_args(args)); + } + } + } + } + std::weak_ptr lua_manager::enable_module(rage::joaat_t module_id) { if (auto module = get_disabled_module(module_id).lock()) @@ -181,13 +241,14 @@ namespace big // unload module { - std::lock_guard guard(m_disabled_module_lock); - std::erase_if(m_disabled_modules, [module_id](auto& module) { - return module_id == module->module_id(); - }); + std::lock_guard guard(m_disabled_module_lock); + std::erase_if(m_disabled_modules, [module_id](auto& module) { + return module_id == module->module_id(); + }); } - const auto new_module_path = move_file_relative_to_folder(m_disabled_scripts_folder.get_path(), m_scripts_folder.get_path(), module_path); + const auto new_module_path = + move_file_relative_to_folder(m_disabled_scripts_folder.get_path(), m_scripts_folder.get_path(), module_path); if (new_module_path) { return load_module(*new_module_path); @@ -205,13 +266,14 @@ namespace big // unload module { - std::lock_guard guard(m_disabled_module_lock); - std::erase_if(m_modules, [module_id](auto& module) { - return module_id == module->module_id(); - }); + std::lock_guard guard(m_disabled_module_lock); + std::erase_if(m_modules, [module_id](auto& module) { + return module_id == module->module_id(); + }); } - const auto new_module_path = move_file_relative_to_folder(m_scripts_folder.get_path(), m_disabled_scripts_folder.get_path(), module_path); + const auto new_module_path = + move_file_relative_to_folder(m_scripts_folder.get_path(), m_disabled_scripts_folder.get_path(), module_path); if (new_module_path) { return load_module(*new_module_path); diff --git a/src/lua/lua_manager.hpp b/src/lua/lua_manager.hpp index 2ae88aba..10a66998 100644 --- a/src/lua/lua_manager.hpp +++ b/src/lua/lua_manager.hpp @@ -1,4 +1,5 @@ #pragma once +#include "bindings/memory.hpp" #include "core/enums.hpp" #include "lua_module.hpp" @@ -51,6 +52,9 @@ namespace big void draw_independent_gui(); void draw_gui(rage::joaat_t tab_hash); + bool dynamic_hook_pre_callbacks(const uintptr_t target_func_ptr, lua::memory::type_info_t return_type, lua::memory::runtime_func_t::return_value_t* return_value, std::vector param_types, const lua::memory::runtime_func_t::parameters_t* params, const uint8_t param_count); + void dynamic_hook_post_callbacks(const uintptr_t target_func_ptr, lua::memory::type_info_t return_type, lua::memory::runtime_func_t::return_value_t* return_value, std::vector param_types, const lua::memory::runtime_func_t::parameters_t* params, const uint8_t param_count); + std::weak_ptr enable_module(rage::joaat_t module_id); std::weak_ptr disable_module(rage::joaat_t module_id); diff --git a/src/lua/lua_module.cpp b/src/lua/lua_module.cpp index 5c37e1e7..636de3b7 100644 --- a/src/lua/lua_module.cpp +++ b/src/lua/lua_module.cpp @@ -355,4 +355,40 @@ namespace big return script->is_done(); }); } + + sol::object lua_module::to_lua(const lua::memory::runtime_func_t::parameters_t* params, const uint8_t i, const std::vector& param_types) + { + if (param_types[i] == lua::memory::type_info_t::none_) + { + return sol::nil; + } + else if (param_types[i] == lua::memory::type_info_t::ptr_) + { + return sol::make_object(m_state, lua::memory::pointer(params->get(i))); + } + else + { + return sol::make_object(m_state, lua::memory::value_wrapper_t(params->get_arg_ptr(i), param_types[i])); + } + + return sol::nil; + } + + sol::object lua_module::to_lua(lua::memory::runtime_func_t::return_value_t* return_value, const lua::memory::type_info_t return_value_type) + { + if (return_value_type == lua::memory::type_info_t::none_) + { + return sol::nil; + } + else if (return_value_type == lua::memory::type_info_t::ptr_) + { + return sol::make_object(m_state, lua::memory::pointer((uintptr_t)return_value->get())); + } + else + { + return sol::make_object(m_state, lua::memory::value_wrapper_t((char*)return_value->get(), return_value_type)); + } + + return sol::nil; + } } diff --git a/src/lua/lua_module.hpp b/src/lua/lua_module.hpp index e0945772..750fd6ef 100644 --- a/src/lua/lua_module.hpp +++ b/src/lua/lua_module.hpp @@ -2,9 +2,10 @@ #include "../script.hpp" #include "bindings/gui/gui_element.hpp" #include "core/data/menu_event.hpp" +#include "lua/bindings/runtime_func_t.hpp" +#include "lua/bindings/type_info_t.hpp" #include "lua_patch.hpp" - -#include +#include "services/gui/gui_service.hpp" namespace big { @@ -37,6 +38,9 @@ namespace big std::unordered_map> m_event_callbacks; std::vector m_allocated_memory; + std::unordered_map> m_dynamic_hook_pre_callbacks; + std::unordered_map> m_dynamic_hook_post_callbacks; + lua_module(const std::filesystem::path& module_path, folder& scripts_folder, bool disabled = false); ~lua_module(); @@ -70,5 +74,8 @@ namespace big void tick_scripts(); void cleanup_done_scripts(); + + sol::object to_lua(const lua::memory::runtime_func_t::parameters_t* params, const uint8_t i, const std::vector& param_types); + sol::object to_lua(lua::memory::runtime_func_t::return_value_t* return_value, const lua::memory::type_info_t return_value_type); }; } \ No newline at end of file