feat: dynamicly hooking x64 functions from lua at runtime with arbitrary signatures. (#3301)
* feat: dynamicly hooking x64 functions from lua at runtime with arbitrary signatures. * fix missing include * c/c++ include system is a good system
This commit is contained in:
Quentin
GitHub
parent
cc3bcea9f1
commit
014281d3e9
@ -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)
|
||||
|
||||
11
cmake/asmjit.cmake
Normal file
11
cmake/asmjit.cmake
Normal file
@ -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)
|
||||
31
docs/lua/classes/value_wrapper.md
Normal file
31
docs/lua/classes/value_wrapper.md
Normal file
@ -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)
|
||||
```
|
||||
|
||||
|
||||
@ -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<string>): 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)
|
||||
```
|
||||
|
||||
|
||||
|
||||
@ -29,6 +29,11 @@ namespace big
|
||||
return static_cast<T>(m_original);
|
||||
}
|
||||
|
||||
void** get_original_ptr()
|
||||
{
|
||||
return &m_original;
|
||||
}
|
||||
|
||||
void fix_hook_address();
|
||||
|
||||
private:
|
||||
|
||||
@ -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>())
|
||||
{
|
||||
*(bool*)m_value = new_val.as<bool>();
|
||||
}
|
||||
else if (m_type == type_info_t::string_ && new_val.is<const char*>())
|
||||
{
|
||||
*(const char**)m_value = new_val.as<const char*>();
|
||||
}
|
||||
else if (m_type == type_info_t::integer_ && new_val.is<int64_t>())
|
||||
{
|
||||
*(int64_t*)m_value = new_val.as<int64_t>();
|
||||
}
|
||||
else if (m_type == type_info_t::float_ && new_val.is<float>())
|
||||
{
|
||||
*(float*)m_value = new_val.as<float>();
|
||||
}
|
||||
else if (m_type == type_info_t::double_ && new_val.is<double>())
|
||||
{
|
||||
*(double*)m_value = new_val.as<double>();
|
||||
}
|
||||
}
|
||||
|
||||
static std::unordered_map<uintptr_t, std::unique_ptr<runtime_func_t>> 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<string>: 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<std::string> param_types;
|
||||
for (const auto& [k, v] : param_types_table)
|
||||
{
|
||||
if (v.is<const char*>())
|
||||
{
|
||||
param_types.push_back(v.as<const char*>());
|
||||
}
|
||||
}
|
||||
|
||||
std::unique_ptr<runtime_func_t> runtime_func = std::make_unique<runtime_func_t>();
|
||||
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<sol::table>();
|
||||
@ -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_t>("value_wrapper", "get", &value_wrapper_t::get, "set", &value_wrapper_t::set);
|
||||
ns["dynamic_hook"] = dynamic_hook;
|
||||
}
|
||||
}
|
||||
@ -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);
|
||||
}
|
||||
477
src/lua/bindings/runtime_func_t.hpp
Normal file
477
src/lua/bindings/runtime_func_t.hpp
Normal file
@ -0,0 +1,477 @@
|
||||
#pragma once
|
||||
#include "hooking/detour_hook.hpp"
|
||||
#include "lua/bindings/type_info_t.hpp"
|
||||
|
||||
#include <asmjit/asmjit.h>
|
||||
|
||||
namespace lua::memory
|
||||
{
|
||||
class runtime_func_t
|
||||
{
|
||||
std::vector<uint8_t> m_jit_function_buffer;
|
||||
asmjit::x86::Mem m_args_stack;
|
||||
|
||||
std::unique_ptr<big::detour_hook> m_detour;
|
||||
|
||||
public:
|
||||
type_info_t m_return_type;
|
||||
std::vector<type_info_t> m_param_types;
|
||||
|
||||
struct parameters_t
|
||||
{
|
||||
template<typename T>
|
||||
void set(const uint8_t idx, const T val) const
|
||||
{
|
||||
*(T*)get_arg_ptr(idx) = val;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
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<big::detour_hook>();
|
||||
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<asmjit::x86::Reg> 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<asmjit::x86::Gp>());
|
||||
}
|
||||
else if (is_XMM_register(argType))
|
||||
{
|
||||
cc.movq(args_stack_index, arg_registers.at(argIdx).as<asmjit::x86::Xmm>());
|
||||
}
|
||||
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<bool, parameters_t*, uint8_t, return_value_t*, uintptr_t>());
|
||||
|
||||
// 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<asmjit::x86::Gp>(), args_stack_index);
|
||||
}
|
||||
else if (is_XMM_register(argType))
|
||||
{
|
||||
cc.movq(arg_registers.at(arg_idx).as<asmjit::x86::Xmm>(), 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<void, parameters_t*, uint8_t, return_value_t*, uintptr_t>());
|
||||
|
||||
// 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<std::string>& 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<T>::kTypeId); \
|
||||
}
|
||||
#define TYPEID_MATCH_STR_ELSEIF(var, T) \
|
||||
else if (var == #T) \
|
||||
{ \
|
||||
return asmjit::TypeId(asmjit::TypeUtils::TypeIdOfT<T>::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;
|
||||
}
|
||||
};
|
||||
}
|
||||
15
src/lua/bindings/type_info_t.hpp
Normal file
15
src/lua/bindings/type_info_t.hpp
Normal file
@ -0,0 +1,15 @@
|
||||
#pragma once
|
||||
|
||||
namespace lua::memory
|
||||
{
|
||||
enum type_info_t
|
||||
{
|
||||
none_,
|
||||
boolean_,
|
||||
string_,
|
||||
integer_,
|
||||
ptr_,
|
||||
float_,
|
||||
double_
|
||||
};
|
||||
}
|
||||
@ -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<lua::memory::type_info_t> 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<sol::object> 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<bool>() == 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<lua::memory::type_info_t> 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<sol::object> 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_module> lua_manager::enable_module(rage::joaat_t module_id)
|
||||
{
|
||||
if (auto module = get_disabled_module(module_id).lock())
|
||||
@ -187,7 +247,8 @@ namespace big
|
||||
});
|
||||
}
|
||||
|
||||
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);
|
||||
@ -211,7 +272,8 @@ namespace big
|
||||
});
|
||||
}
|
||||
|
||||
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);
|
||||
|
||||
@ -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<lua::memory::type_info_t> 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<lua::memory::type_info_t> param_types, const lua::memory::runtime_func_t::parameters_t* params, const uint8_t param_count);
|
||||
|
||||
std::weak_ptr<lua_module> enable_module(rage::joaat_t module_id);
|
||||
std::weak_ptr<lua_module> disable_module(rage::joaat_t module_id);
|
||||
|
||||
|
||||
@ -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<lua::memory::type_info_t>& 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<uintptr_t>(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;
|
||||
}
|
||||
}
|
||||
|
||||
@ -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 <services/gui/gui_service.hpp>
|
||||
#include "services/gui/gui_service.hpp"
|
||||
|
||||
namespace big
|
||||
{
|
||||
@ -37,6 +38,9 @@ namespace big
|
||||
std::unordered_map<menu_event, std::vector<sol::protected_function>> m_event_callbacks;
|
||||
std::vector<void*> m_allocated_memory;
|
||||
|
||||
std::unordered_map<uintptr_t, std::vector<sol::protected_function>> m_dynamic_hook_pre_callbacks;
|
||||
std::unordered_map<uintptr_t, std::vector<sol::protected_function>> 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<lua::memory::type_info_t>& 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);
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user