C++封装IATHOOK类实例

时间:2014-10-18 03:18来源:网络整理作者:网络点击: 次

分享到：

这篇文章主要介绍了C++封装IATHOOK类的实现方法,对IAT的HOOK实例进行了封装,非常具有实用价值,需要的朋友可以参考下

本文实例讲述了C++封装IATHOOK类的实现方法。分享给大家供大家参考。具体方法如下：

1. 定义成类的静态成员，从而实现自动调用

static CAPIHOOK sm_LoadLibraryA;  

static CAPIHOOK sm_LoadLibraryW;  

static CAPIHOOK sm_LoadLibraryExA;  

static CAPIHOOK sm_LoadLibraryExW;  

static CAPIHOOK sm_GetProcAddress;

2. ReplaceIATEntryInAllMods中遍历模块的框架

复制代码代码如下:

void CAPIHOOK::ReplaceIATEntryInAllMods(LPCTSTR pszExportMod, PROC pfnCurrent, PROC pfnNewFunc, BOOL bExcludeAPIHookMod)  

{  

    //取得当前模块句柄  

    HMODULE hModThis = NULL;  

    if (bExcludeAPIHookMod)  

    {  

        MEMORY_BASIC_INFORMATION mbi;  

        if (0 != ::VirtualQuery(ReplaceIATEntryInAllMods, &mbi, sizeof(MEMORY_BASIC_INFORMATION))) //ReplaceIATEntryInAllMods必须为类的static函数  

        {  

            hModThis = (HMODULE)mbi.AllocationBase;  

        }  

    }  

    //取得本进程的模块列表  

    HANDLE hModuleSnap = INVALID_HANDLE_VALUE;   

    MODULEENTRY32 me32;  

    hModuleSnap = ::CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, GetCurrentProcessId());  

    if (INVALID_HANDLE_VALUE == hModuleSnap)  

    {  

        return;  

    }  

    me32.dwSize = sizeof( MODULEENTRY32 );   

    if( !Module32First( hModuleSnap, &me32 ) )   

    {   

        return;  

    }  

    do   

    { //对每一个模块  

        if (me32.hModule != hModThis)  

        {  

            ReplaceIATEntryInOneMod(pszExportMod, pfnCurrent, pfnNewFunc, me32.hModule);  

        }  

    } while( Module32Next( hModuleSnap, &me32 ) );   

    ::CloseHandle(hModuleSnap); //配对写  

}

3. 遍历链表摘除自己的框架

复制代码代码如下:

CAPIHOOK::~CAPIHOOK(void)  

{  

    //取消对函数的HOOK  

    ReplaceIATEntryInAllMods(m_pszModName, m_pfnHook, m_pfnOrig, TRUE);  

    //把自己从链表中删除  

    CAPIHOOK* p = sm_pHeader;  

    if (p == this)  

    {  

        sm_pHeader = this->m_pNext;  

    }  

    else  

    {  

        while(p != NULL)  

        {  

            if (p->m_pNext == this)  

            {  

                p->m_pNext = this->m_pNext;  

                break;  

            }  

            p = p->m_pNext;  

        }  

    }  

}