cheatlib中函数钩子模块的原理

/* 说明:  将pOrigAddr处的函数直接替换为pHookAddr处的函数执行
* 注意:  pOrigAddr和pHookAddr处的函数定义必须一致
*        此函数一般写在dll中,注入到程序中将程序中的函数替换为dll中的
* 参数:  pOrigAddr - 源函数地址
*        pHookAddr - hook函数地址
* 返回值:PFuncHookInfo */
PFuncHookInfo FuncHook(LPVOID pOrigAddr, LPVOID pHookAddr)
{
        DWORD oldProtect;
        VirtualProtect(pOrigAddr, 5, PAGE_EXECUTE_READWRITE, &oldProtect);
        PFuncHookInfo ptInfo = (PFuncHookInfo)malloc(sizeof(FuncHookInfo));
        if(ptInfo == NULL) return NULL;
        ptInfo->pOrigFuncAddr = pOrigAddr;
        ptInfo->pHookFuncAddr = pHookAddr;
        ptInfo->last_return_value = 0;
        ptInfo->pbOpCode = (BYTE*)malloc(sizeof(BYTE)*5);
        if(ptInfo->pbOpCode != NULL) memcpy(ptInfo->pbOpCode, pOrigAddr, 5);
        JmpBuilder((BYTE*)pOrigAddr, (DWORD)pHookAddr, (DWORD)pOrigAddr);
        VirtualProtect(pOrigAddr, 5, PAGE_EXECUTE, &oldProtect);
        return ptInfo;
}

typedef struct _FuncHookInfo{
        LPVOID pOrigFuncAddr;       // 代码源地址
        LPVOID pHookFuncAddr;       // Hook代码源地址
        BYTE *pbOpCode;             // 机器码用于恢复现场
        int last_return_value;      // CallOrigFunc源函数返回值(eax)
        int last_return_2nd_value;  // 在返回值是有两个整型值的结构体时这里保存第二个元素(edx)
} FuncHookInfo, *PFuncHookInfo;

void IntToByte(int i, BYTE *bytes)
{
        assert(bytes != NULL);
        bytes[0] = (byte) (0xff & i);
        bytes[1] = (byte) ((0xff00 & i) >> 8);
        bytes[2] = (byte) ((0xff0000 & i) >> 16);
        bytes[3] = (byte) ((0xff000000 & i) >> 24);
}

void JmpBuilder(BYTE *pCmdOutput, DWORD dwTargetAddr, DWORD dwCurrentAddr)
{
        assert(pCmdOutput != NULL);
        pCmdOutput[0] = 0xE9;
        DWORD jmpOffset = dwTargetAddr - dwCurrentAddr - 5;
        IntToByte(jmpOffset, pCmdOutput+1);
}

/* 说明:    撤销函数钩子
* 参数:    ptInfo  - FuncHook函数返回值
* 返回值:  void */
void FuncUnhook(PFuncHookInfo ptInfo)
{
        assert(ptInfo != NULL && ptInfo->pbOpCode != NULL);
        DWORD oldProtect;
        VirtualProtect(ptInfo->pOrigFuncAddr, 5, PAGE_EXECUTE_READWRITE, &oldProtect);
        memcpy(ptInfo->pOrigFuncAddr, ptInfo->pbOpCode, 5);
        VirtualProtect(ptInfo->pOrigFuncAddr, 5, PAGE_EXECUTE, &oldProtect);
        free(ptInfo->pbOpCode);
        free(ptInfo);
}

#include <stdio.h>

typedef struct _st{
        int a;
        int b;
} st, *pst;

st test()
{
        return (st){1, 2};
}

int main()
{
        printf("%d\n", test().a);
        return 0;
}

#include <stdio.h>

typedef struct _st{
        int a;
        int b;
        int c;
} st, *pst;

st test()
{
        return (st){1, 2, 3};
}

int main()
{
        printf("%d\n", test().a);
        return 0;
}

/* 说明:  在Hook函数里调用源函数
* 注意:  函数参数必须一致,否则会出现栈损
*        不支持返回结构体的函数,否则可能会覆盖栈内的合法数据
* 参数:  PFuncHookInfo ptInfo  - FuncHook函数的返回值
*        ...                   - 函数参数 */
#define CallOrigFunc(ptInfo, ...) do{ \
        DWORD oldProtect; \
        VirtualProtect(ptInfo->pOrigFuncAddr, 5, PAGE_EXECUTE_READWRITE, &oldProtect); \
        memcpy(ptInfo->pOrigFuncAddr, ptInfo->pbOpCode, 5); \
        cheatlib_func_caller(ptInfo->pOrigFuncAddr, __VA_ARGS__); \
        __asm__ __volatile__( \
                        "movl %%eax, %0;" \
                        "movl %%edx, %1;":: \
                        "m"(ptInfo->last_return_value), \
                        "m"(ptInfo->last_return_2nd_value): \
                        "eax", "edx"); \
        JmpBuilder((BYTE*)ptInfo->pOrigFuncAddr, (DWORD)ptInfo->pHookFuncAddr, (DWORD)ptInfo->pOrigFuncAddr); \
        VirtualProtect(ptInfo->pOrigFuncAddr, 5, PAGE_EXECUTE, &oldProtect); \
} while(0)

void __attribute__((naked)) cheatlib_func_caller(LPVOID pOrigFuncAddr, ...)
{
        __asm__ __volatile__(
                        "popl %%eax;"
                        "popl %%ebx;"
                        "pushl %%eax;"
                        "jmp *%%ebx;"
                        :);
}

        __asm__ __volatile__( \
                        "movl %%eax, %0;" \
                        "movl %%edx, %1;":: \
                        "m"(ptInfo->last_return_value), \
                        "m"(ptInfo->last_return_2nd_value): \
                        "eax", "edx"); \

/* 说明:  在Hook函数里调用源函数
* 注意:  函数参数必须一致,否则会出现栈损
*        只支持返回结构体的函数,否则会出现栈损
*        如果结构体内的元素都是整型且数量小于或等于二的话
*        那么元素将分别保存在eax和edx里
*        这个情况下不适合使用此宏,而是使用CallOrigFunc宏
* 参数:  PFuncHookInfo ptInfo  - FuncHook函数的返回值
*        void *pSaveStructAddr - 函数返回的结构体保存位置
*        ...                 - 函数参数 */
#define CallOrigFunc_RetStruct(ptInfo, pSaveStructAddr, ...) do{ \
        DWORD oldProtect; \
        VirtualProtect(ptInfo->pOrigFuncAddr, 5, PAGE_EXECUTE_READWRITE, &oldProtect); \
        memcpy(ptInfo->pOrigFuncAddr, ptInfo->pbOpCode, 5); \
        cheatlib_ret_struct_func_caller(pSaveStructAddr, ptInfo->pOrigFuncAddr, __VA_ARGS__); \
        JmpBuilder((BYTE*)ptInfo->pOrigFuncAddr, (DWORD)ptInfo->pHookFuncAddr, (DWORD)ptInfo->pOrigFuncAddr); \
        VirtualProtect(ptInfo->pOrigFuncAddr, 5, PAGE_EXECUTE, &oldProtect); \
} while(0)

void __attribute__((naked)) cheatlib_ret_struct_func_caller(LPVOID pStructAddr, LPVOID pOrigFuncAddr, ...)
{
        __asm__ __volatile__(
                        "popl %%eax;"
                        "popl %%ebx;"
                        "popl %%ecx;"
                        "pushl %%ebx;"
                        "pushl %%eax;"
                        "jmp *%%ecx;"
                        :);
}

#include "cheatlib_funchook.h"
#include <stdio.h>
#include <windows.h>

PFuncHookInfo ptInfo;

int WINAPI hmsgbox(HWND hWnd,LPCTSTR lpText,LPCTSTR lpCaption,UINT uType)
{
        CallOrigFunc(ptInfo, hWnd, "Your MessageBoxA has been hooked!", lpCaption, uType);
        return 0;
}

int hprintf(const char* str, ...){
        MessageBox(NULL, "hooked printf", str, MB_OK);
        return 0;
}

int main()
{
        ptInfo = FuncHook((LPVOID)&MessageBoxA, (LPVOID)&hmsgbox);
        FuncHook((LPVOID)&printf, (LPVOID)&hprintf);
        printf("main: printf()");
        MessageBoxA(NULL, "main: MessageBoxA()", "Info", MB_OK);
        FuncUnhook(ptInfo);
        MessageBoxA(NULL, "main: MessageBoxA()", "Info", MB_OK);
        return 0;
}