C++实现修改函数代码HOOK的封装方法

更新时间：2014年10月20日 10:08:11 投稿：shichen2014

这篇文章主要介绍了C++实现修改函数代码HOOK的封装方法,有助于深入了解C++的HOOK原理,需要的朋友可以参考下

本文实例讲述了C++实现修改函数代码HOOK的封装方法，分享给大家供大家参考。具体实现方法如下：

一、对外的接口如下：

1. 类初始化时对函数HOOK
2. 取消挂钩：
void UnHook();
3. 重新挂钩：
void ReHook();

在初始化时HOOK的代码：

复制代码代码如下:

*(DWORD*)(m_btNewBytes+1) = (DWORD)pfnHook;

8个字节的代码地址 0xB8, 0x00, 0x00,0x40,0x00,0xFF,0xE0,0x00 只要把第二位和第三位的数据改成函数的地址，调用原先的函数时就会调到自定义的函数执行.

二、实现方法：

.h头文件如下：

复制代码代码如下:

#ifndef  _ULHOOK_H__  

#define _ULHOOK_H__  

#include <Windows.h>  

#pragma once  

class CULHook  

{  

public:  

    CULHook(LPSTR lpszModName, LPSTR lpszFuncNme, PROC pfnHook);  

    ~CULHook(void);  

    //取消挂钩  

    void UnHook();  

    //重新挂钩  

    void ReHook();  

protected:  

    PROC m_pfnOrig;  

    BYTE m_btNewBytes[8];  

    BYTE m_btOldBytes[8];  

    HMODULE m_hModule;  

};  

#endif

.cpp源文件如下：

复制代码代码如下:

#include "ULHook.h"  

CULHook::CULHook(LPSTR lpszModName, LPSTR lpszFuncNme, PROC pfnHook)  

{  

    BYTE btNewBytes[] = {0xB8, 0x00, 0x00,0x40,0x00,0xFF,0xE0,0x00};  

    memcpy(m_btNewBytes, btNewBytes, 8);  

    *(DWORD*)(m_btNewBytes+1) = (DWORD)pfnHook;  

    m_hModule = ::LoadLibraryA(lpszModName);  

    if (NULL == m_hModule)  

    {  

        m_pfnOrig = NULL;  

        return;  

    }  

    m_pfnOrig = (PROC)::GetProcAddress(m_hModule, lpszFuncNme);  

    if (NULL != m_pfnOrig)  

    {  

        MEMORY_BASIC_INFORMATION mbi = {0};  

        DWORD dwOldProtect;  

        ::VirtualQuery(m_pfnOrig, &mbi, sizeof(mbi));  

        ::VirtualProtect(m_pfnOrig, 8, PAGE_READWRITE, &dwOldProtect);  

        memcpy(m_btOldBytes, m_pfnOrig, 8);  

        ::WriteProcessMemory(GetCurrentProcess(), (VOID*)m_pfnOrig, m_btNewBytes, 8, NULL);  

        ::VirtualProtect(m_pfnOrig, 8, dwOldProtect, NULL);  

    }  

}  

CULHook::~CULHook(void)  

{  

    UnHook();  

    if (m_hModule!=NULL)  

    {  

        ::FreeLibrary(m_hModule);  

    }  

}  

void CULHook::UnHook()  

{  

    if (m_pfnOrig != NULL)  

    {  

        MEMORY_BASIC_INFORMATION mbi = {0};  

        DWORD dwOldProtect;  

        ::VirtualQuery(m_pfnOrig, &mbi, sizeof(mbi));  

        ::VirtualProtect(m_pfnOrig, 8, PAGE_READWRITE, &dwOldProtect);  

        ::WriteProcessMemory(GetCurrentProcess(), (VOID*)m_pfnOrig, m_btOldBytes, 8, NULL);  

        ::VirtualProtect(m_pfnOrig, 8, dwOldProtect, NULL);  

    }  

}  

void CULHook::ReHook()  

{  

    if (m_pfnOrig != NULL)  

    {  

        MEMORY_BASIC_INFORMATION mbi = {0};  

        DWORD dwOldProtect;  

        ::VirtualQuery(m_pfnOrig, &mbi, sizeof(mbi));  

        ::VirtualProtect(m_pfnOrig, 8, PAGE_READWRITE, &dwOldProtect);  

        ::WriteProcessMemory(GetCurrentProcess(), (VOID*)m_pfnOrig, m_btNewBytes, 8, NULL);  

        ::VirtualProtect(m_pfnOrig, 8, dwOldProtect, NULL);  

    }  

}

希望本文所述对大家的C++程序设计有所帮助。

您可能感兴趣的文章:

typedef和#define的用法以及区别
以下是对C/C++语言中，typedef和#define的用法以及区别进行了详细的介绍，需要的朋友可以过来参考下
2013-10-10
C++ STL关联式容器自定义排序规则的2种方法
这篇文章主要介绍了C++ STL关联式容器自定义排序规则的2种方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-03-03
C/C++中组合详解及其作用介绍
这篇文章主要介绍了C/C++中组合的详解及其作用介绍，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-09-09
C++ RTTI与4种类型转换的深入理解
这篇文章主要给大家介绍了关于C++ RTTI与4种类型转换的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-03-03
快速模式匹配算法(KMP)的深入理解
本篇文章是对快速模式匹配算法(KMP)进行了详细的分析介绍，需要的朋友参考下
2013-05-05
C语言实例实现二叉搜索树详解
二叉搜索树是以一棵二叉树来组织的。每个节点是一个对象，包含的属性有left，right，p和key，其中，left指向该节点的左孩子，right指向该节点的右孩子，p指向该节点的父节点，key是它的值
2022-05-05
C++实现并优化异常系统
异常处理是C++的一项语言机制,用于在程序中处理异常事件,下面这篇文章主要给大家介绍了关于C++中异常的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
2022-08-08
C/C++中运算符的优先级、运算符的结合性详解
这篇文章主要介绍了C/C++中运算符的优先级、运算符的结合性详解的相关资料,需要的朋友可以参考下
2017-02-02
C语言实现二叉树层次遍历介绍
大家好，本篇文章主要讲的是C语言实现二叉树层次遍历介绍，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下
2022-01-01
解析使用C++编写无错代码的方法技巧
本篇文章是对使用C++编写无错代码的方法进行了详细的分析介绍，需要的朋友参考下
2013-05-05