C/C++ Inline-hook(内联钩子)函数的实现,Linux/Windows AMD64 IL平台支持!

钩子汇编指令:JMP RVA

Linux 平台(GCC ##AMD64)

#include <unistd.h>
#include <sys/mman.h>
#include <cstdio>
#include <string>
#include <cstdlib>
#include <cstring>
#include <assert.h>
#include <stdexcept>

#define PAGE_START(x, pagesize)	((x) & ~((pagesize) - 1))

static void AdjustProtectMemoryPermissions(void* func, int permissions)
{
    int pagesize = getpagesize();
    int64_t start = PAGE_START((int64_t)func, pagesize);

    if (mprotect((void*)start, (size_t)pagesize, permissions) < 0) 
    {
        throw std::runtime_error("mprotect error");
    }
}

static void InstallHooker(char* func, char* jump)
{
    int RVA = ((int64_t)func) + 5;
    RVA = (int64_t)(jump) - RVA;

    AdjustProtectMemoryPermissions(func, PROT_READ | PROT_WRITE | PROT_EXEC);
    func[0] = 0xE9;
    *(int*)(func + 1) = RVA;
    AdjustProtectMemoryPermissions(func, PROT_READ | PROT_EXEC);
}

static void test(int x)
{
    printf("%s: %d\n", __FUNCTION__, x);
}

static void test_dohooker(int x)
{
    printf("%s: %d\n", __FUNCTION__, x);
}

int main(int argc, const char* argv[]) noexcept
{
    InstallHooker((char*)&test, (char*)&test_dohooker);
    test(100);
    return 0;
}

钩子汇编指令:MOV RAX; JMP RAX

Windows 平台(VC++ ##AMD64)

#include <cstdio>
#include <Windows.h>
#include <stdexcept>

static void AdjustProtectMemoryPermissions(void* func, int permissions)
{
    DWORD permissions_ = 0;
    if (!VirtualProtect(func, 12, permissions, &permissions_))
    {
        throw std::runtime_error("VirtualProtect error");
    }
}

static void InstallHooker(char* func, char* jump)
{
    AdjustProtectMemoryPermissions(func, PAGE_EXECUTE_READWRITE);
    func[0] = 0x48;
    func[1] = 0xB8;
    *(char**)(func + 2) = jump;
    func[10] = 0xFF;
    func[11] = 0xE0;
    AdjustProtectMemoryPermissions(func, PAGE_EXECUTE_READ);
}

static void test(int x)
{
    printf("%s: %d\n", __FUNCTION__, x);
}

static void test_dohooker(int x)
{
    printf("%s: %d\n", __FUNCTION__, x);
}

int main(int argc, const char* argv[]) noexcept
{
    InstallHooker((char*)&test, (char*)&test_dohooker);
    test(100);
    return 0;
}

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值