WINDOWS下HOOK函数的方法

 

前两天和dvff谈论的一些技术总结。

程序代码：

/
//                     HOOKAPI DEMO PROGRAM
// 文件名：HOOKTEST.C
//
// 描述: 演示WINNT下HOOK系统API过程及方法
//       (扩充为HOOK任意函数的方法)
//
// 作者：东海一鱼
//
// 时间: 2010.7.22       
//
// 使用编译器： VC2003
//
// 使用第三方库: NULL
//
//
// Bug、修复纪录：
//           1、2010.8.4 增加HOOK私有函数代码(被hook函数名TestFn，hook函数名MyHookFn2)
//
//             2、2010.8.6 增加HOOK类虚拟函数代码(被hook函数名TestVirtualFn，
//                hook函数名TestVirtualFn2)
//
//                HOOK类成员函数，主要难度在于成员函数地址的获得。
//              普通成员函数指针在VC6下无法进行强制类型转换，
//              解决手段见GetClassFnAddress辅助函数。
//              类虚拟成员函数HOOK需要获得类实例对象虚表中的虚拟函数指针。                   

#include <stdlib.h>
#include  <stdio.h>
#include  <windows.h>

// #pragma pointers_to_members(best_case)

#ifdef _X86_

#define FLATJMPCODE_LENGTH 5             // x86 平坦内存模式下，绝对跳转指令长度
#define FLATJMPCMD_LENGTH  1             // 机械码0xe9长度
#define FLATJMPCMD         0xe9

#else

#error WINVER ERRORSLECT  ONLY USES ON X86!!!

#endif

///
// 测试类函数HOOK
class TestHookFnClass
{
public:
     virtual  void TestVirtualFn( char* szPlay);
     virtual  void TestVirtualFn2( char* szPlay);
     void TestHookFn( char* szPlay);
     void TestHookFn2( char* szPlay);   
};

void TestHookFnClass::TestHookFn( char* szPlay)
{
    printf( " %s\n ",szPlay);
}

void TestHookFnClass::TestHookFn2( char* szPlay)
{
    printf( " %s - %s\n  ",szPlay, " 类函数已被HOOK! ");
}

void TestHookFnClass::TestVirtualFn( char* szPlay)
{
    printf( " %s\n ",szPlay);
}

void TestHookFnClass::TestVirtualFn2( char* szPlay)
{
    printf( " %s - %s\n ",szPlay, " Virtual2 HOOKS ");
}

/
#ifdef _X86_
// 利用C变参特性，进行类成员函数指针类型转换
LPVOID GetClassFnAddress(...)                     // Add  2010.8.6
{
    LPVOID FnAddress;
   
     __asm
    {
        lea eax,FnAddress
        mov edx,[ebp+ 8]
        mov [eax],edx
    }
     return FnAddress;
}
#endif

// 获得类虚拟成员函数指针
LPVOID GetClassVirtualFnAddress(LPVOID pthis, int Index)  // Add 2010.8.6
{
    LPVOID FnAddress;
   
    *( int*)&FnAddress = *( int*)pthis;                        // lpvtable
    *( int*)&FnAddress = *( int*)(( int*)FnAddress + Index);
     return FnAddress;
}
/
// 我的新API函数
int  __stdcall MyHookFn(HWND hwnd, char* sztext, char* szTitle, int stly)
{          
   
     return printf( " %s - %s\n ",sztext, " 原API函数已被HOOK! ");    //
}

// 我的新私有桩函数
void MyHookFn2( char* szTxt)
{
     char* pBuf = ( char*)malloc(strlen(szTxt) +  sizeof( char*) *  32);
   
    __try
    {
        sprintf(pBuf, " %s - %s ",szTxt, "  原函数已被HOOK! ");
        printf( " %s\n ",pBuf);
    }
    __finally
    {
        free(pBuf);
    }
}

// 私有测试函数
void TestFn( char* szTitle)
{
    printf( " %s ",szTitle);
}

/ /
// HOOK函数
BOOL HookApi(LPVOID ApiFun,LPVOID HookFun)
{
    BOOL    IsSuccess = FALSE;
    DWORD   TempProtectVar;               // 临时保护属性变量
    MEMORY_BASIC_INFORMATION MemInfo;     // 内存分页属性信息
   
    VirtualQuery(ApiFun,&MemInfo, sizeof(MEMORY_BASIC_INFORMATION)); 
   
     if(VirtualProtect(MemInfo.BaseAddress,MemInfo.RegionSize,
        PAGE_READWRITE,&MemInfo.Protect))                             // 修改页面为可写
    {
        *(BYTE*)ApiFun = FLATJMPCMD;                                     
        *(DWORD*)((BYTE*)ApiFun + FLATJMPCMD_LENGTH) = (DWORD)HookFun -
            (DWORD)ApiFun - FLATJMPCODE_LENGTH;
       
        VirtualProtect(MemInfo.BaseAddress,MemInfo.RegionSize,
            MemInfo.Protect,&TempProtectVar);                                // 改回原属性
       
        IsSuccess = TRUE;
    }
   
     return IsSuccess;
}

int main( int argc, char** argv)
{
    HMODULE hDll;
    LPVOID  OldFun;    

    TestHookFnClass* TestClass =  new TestHookFnClass();

    __try
    {
        hDll = GetModuleHandle( " User32.dll ");
        OldFun = GetProcAddress(hDll, " MessageBoxA ");   // 要HOOK的API函数
   
         if(OldFun)
        {
             if(HookApi(OldFun,MyHookFn))   // 如果HOOK成功
                MessageBoxA( 0, " call Api MessageBox ", " Is Hookd? ",MB_OK);  // 调用原API,测试HOOK

             if(HookApi(TestFn,MyHookFn2))
                TestFn( " Private Function Hook Test ");   // 调用私有函数，测试HOOK   
           
             if(HookApi(GetClassFnAddress(TestHookFnClass::TestHookFn),
                GetClassFnAddress(TestHookFnClass::TestHookFn2)))
                TestClass->TestHookFn( " Hook Class Member Function Test! "); // 调用类成员函数，测试HOOK

             if(HookApi(GetClassVirtualFnAddress(TestClass, 0),
                GetClassFnAddress(TestHookFnClass::TestHookFn2)))
                TestClass->TestVirtualFn( " Call Class Virtual Function Test! "); // 调用类虚拟函数，测试HOOK
           
             if(HookApi(GetClassVirtualFnAddress(TestClass, 0),
                GetClassVirtualFnAddress(TestClass, 1)))                    
                TestClass->TestVirtualFn( " Call Class Virtual Function Test2! "); // 同上，桩函数为另一虚函数
        }
    }
    __finally
    {
         if(hDll)
            FreeLibrary(hDll);
         if(TestClass)
             delete(TestClass);
    }
   
     return  0;
}

转载于:https://www.cnblogs.com/HeroZearin/articles/2772100.html