diff --git a/include/REL/Relocation.h b/include/REL/Relocation.h index 7d1bd97c5..af7f4bddd 100644 --- a/include/REL/Relocation.h +++ b/include/REL/Relocation.h @@ -282,6 +282,37 @@ namespace REL return stl::unrestricted_cast(_impl); } + template + void replace_func(const std::size_t a_count, const std::uintptr_t a_dst) requires(std::same_as) + { +#pragma pack(push, 1) + struct Assembly + { + std::uint8_t jmp; + std::uint8_t modrm; + std::int32_t disp; + std::uint64_t addr; + }; + static_assert(sizeof(Assembly) == 0xE); +#pragma pack(pop) + + Assembly assembly{ + .jmp = static_cast(0xFF), + .modrm = static_cast(0x25), + .disp = static_cast(0), + .addr = static_cast(a_dst), + }; + + safe_fill(address() + O, INT3, a_count); + safe_write(address() + O, &assembly, sizeof(assembly)); + } + + template + void replace_func(const std::size_t a_count, const F a_dst) requires(std::same_as) + { + replace_func(a_count, stl::unrestricted_cast(a_dst)); + } + template void write(const U& a_data) requires(std::same_as) { @@ -323,16 +354,6 @@ namespace REL safe_fill(address(), a_value, a_count); } -#ifdef SKSE_SUPPORT_XBYAK - void write_func(const std::size_t a_count, const std::uintptr_t a_dst) requires(std::same_as); - - template - void write_func(const std::size_t a_count, const F a_dst) requires(std::same_as) - { - write_func(a_count, stl::unrestricted_cast(a_dst)); - } -#endif - template std::uintptr_t write_vfunc(const std::size_t a_idx, const std::uintptr_t a_newFunc) requires(std::same_as) { diff --git a/src/REL/Relocation.cpp b/src/REL/Relocation.cpp index b41629390..4ce831bde 100644 --- a/src/REL/Relocation.cpp +++ b/src/REL/Relocation.cpp @@ -32,33 +32,3 @@ namespace REL assert(success); } } - -#ifdef SKSE_SUPPORT_XBYAK -# include - -namespace REL -{ - struct write_func_impl : Xbyak::CodeGenerator - { - write_func_impl(std::uintptr_t a_dst) - { - Xbyak::Label dst; - jmp(ptr[rip + dst]); - L(dst); - dq(a_dst); - } - }; - - template - void Relocation::write_func(const std::size_t a_count, const std::uintptr_t a_dst) requires(std::same_as) - { - safe_fill(address(), INT3, a_count); - auto patch = write_func_impl{ a_dst }; - patch.ready(); - assert(patch.getSize() <= a_count); - safe_write(address(), std::span{ patch.getCode(), patch.getSize() }); - } - - template void Relocation::write_func(const std::size_t, const std::uintptr_t); -} -#endif