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;
}
