hook任意api的函数.rar

上传者: friendan | 上传时间: 2024-01-17 14:41:21 | 文件大小: 5KB | 文件类型: RAR
/************************************************************************ Inline HOOK API V1.1 thanks to 海风月影, xIkUg ,sucsor by solosky created at 2011.06.29, updated at 2011.06.30 ---------------------------------- Inline HOOK API V1.1 2011.06.30 ---------------------------------- 1. 修复没有判断申请空间是否成功的BUG; 2. 修复释放内存的参数错误的BUG; ---------------------------------- Inline HOOK API V1.0 2011.06.29 ---------------------------------- 1. HOOK任何函数,仅需知道原型和调用方式即可(可以HOOK对象方法,但需要保存和恢复this指针) 2. UnHOOK已经HOOK的函数 感谢海风月影提供的反汇编长度引擎!! ************************************************************************/ /************************************************************************/ #include "inlinehook.h" /************************************************************************/ /* InlineHOOK一个函数 /* param ppRealFun 被HOOK的函数指针,注意这里必须指向函数的指针,因为需要修改这个值,来返回新的函数地址 /* param pHookFun 自己编写的HOOK函数指针 /* return PHOOK_ENV 成功后返回HOOK之后的环境上下文,可以用这个指针取消HOOK,失败返回NULL /************************************************************************/ PHOOK_ENV __stdcall HookFun(PVOID *ppRealFun,const PVOID pHookFun) { DWORD dwRealFun = *(DWORD*)ppRealFun; DWORD dwHookFun = (DWORD) pHookFun; //首先计算需要替换原始函数的前几个字节 DWORD replaceCodeSize = FindOpcodeSize((BYTE*)dwRealFun); while (replaceCodeSize<5) //至少5个字节,因为jmp xxxx至少需要5个字节 replaceCodeSize += FindOpcodeSize((BYTE*)((DWORD)dwRealFun + (DWORD)replaceCodeSize)); if (replaceCodeSize>16) return NULL; //分配HOOK的环境变量,存放代理代码和一些数据 PHOOK_ENV pHookEnv = (PHOOK_ENV)VirtualAlloc(NULL,sizeof(HOOK_ENV),MEM_COMMIT,PAGE_EXECUTE_READWRITE); if(pHookEnv==NULL) return NULL; memset(pHookEnv, sizeof(HOOK_ENV), 0); memset(pHookEnv->mHookStub.uJumpBack, 0x90, sizeof(pHookEnv->mHookStub.uJumpBack)); //填充跳回和跳去的机器码为nop memset(pHookEnv->mHookStub.uJumpTo, 0x90, sizeof(pHookEnv->mHookStub.uJumpTo)); //在这个环境变量上保存一些信息,便于恢复hook pHookEnv->dwReplacedBytes = replaceCodeSize; //替换的字节长度 pHookEnv->dwRealFunAddr = dwRealFun; //原始函数地址 pHookEnv->dwHookFunAddr = dwHookFun; //HOOK函数地址 //填充STUB跳转到HOOK函数的代码 *(pHookEnv->mHookStub.uJumpTo) = 0xE9; //JMP指令 *(DWORD*)(pHookEnv->mHookStub.uJumpTo+1) = dwHookFun-((DWORD)pHookEnv->mHookStub.uJumpTo)-5; //A B C => B=C-A-5 //填充STUB跳转回原函数的头replaceCodeSize个字节代码 memcpy(pHookEnv->mHookStub.uJumpBack, (PVOID)dwRealFun, replaceCodeSize); *(pHookEnv->mHookStub.uJumpBack+replaceCodeSize) = 0xE9; //JMP指令,准备跳回到原始函数 *(DWORD*)(pHookEnv->mHookStub.uJumpBack+replaceCodeSize+1) = dwRealFun-((DWORD)(pHookEnv->mHookStub.uJumpTo)+replaceCodeSize)-5; //关键的时候来了,现在需要替换原始函数的前replaceCodeSize个字节为一个JMP到STUB的指令 DWORD dwOldProctect = 0; if (!VirtualProtect((PVOID)dwRealFun, replaceCodeSize,PAGE_EXECUTE_READWRITE,&dwOldProctect)) goto FAILED; *(UCHAR*)(dwRealFun) = 0xE9; *(DWORD*)(dwRealFun+1) = (DWORD)(pHookEnv->mHookStub.uJumpTo)-dwRealFun-5; if(replaceCodeSize>5) memset((PVOID)(dwRealFun+5), 0x90, replaceCodeSize-5); if (!VirtualProtect((PVOID)dwRealFun,replaceCodeSize,dwOldProctect,&dwOldProctect)) goto FAILED; //修改真实函数地址为STUB的跳回指令开始 *(ppRealFun) = pHookEnv->mHookStub.uJumpBack; return pHookEnv; FAILED: if(pHookEnv) VirtualFree(pHookEnv, 0, MEM_RELEASE); return NULL; } /************************************************************************/ /* 取消HOOK一个函数 /* param pHookEnv HOOK的环境上下文,又hookFun函数返回 /* return bool 成功返回TRUE失败返回FALSE /************************************************************************/ BOOL __stdcall UnHookFun(PHOOK_ENV pHookEnv) { //恢复真实函数修改的前几个字节即可 DWORD dwOldProctect = 0; if (!VirtualProtect((PVOID)pHookEnv->dwRealFunAddr, pHookEnv->dwReplacedBytes, PAGE_EXECUTE_READWRITE,&dwOldProctect)) goto FAILED; memcpy((PVOID)pHookEnv->dwRealFunAddr, (PVOID)pHookEnv->mHookStub.uJumpBack, pHookEnv->dwReplacedBytes); if (!VirtualProtect((PVOID)pHookEnv->dwRealFunAddr, pHookEnv->dwReplacedBytes,dwOldProctect,&dwOldProctect)) goto FAILED; if(!VirtualFree(pHookEnv, 0, MEM_RELEASE)) goto FAILED; return TRUE; FAILED: return FALSE; } /************************************************************************/ /* 查找给定地址的机器指令长度 (感谢海风月影提供的反汇编长度引擎) /* param intr0 指令地址 /* return bool 成功返回当前指令长度 /************************************************************************/ DWORD __stdcall FindOpcodeSize(BYTE* iptr0) { BYTE* iptr = iptr0; DWORD f = 0; prefix: BYTE b = *iptr++; f |= table_1[b]; if (f&C_FUCKINGTEST) if (((*iptr)&0x38)==0x00) // ttt f=C_MODRM+C_DATAW0; // TEST else f=C_MODRM; // NOT,NEG,MUL,IMUL,DIV,IDIV if (f&C_TABLE_0F) { b = *iptr++; f = table_0F[b]; } if (f==C_ERROR) { //printf("error in X\n",b); return C_ERROR; } if (f&C_PREFIX) { f&=~C_PREFIX; goto prefix; } if (f&C_DATAW0) if (b&0x01) f|=C_DATA66; else f|=C_DATA1; if (f&C_MODRM) { b = *iptr++; BYTE mod = b & 0xC0; BYTE rm = b & 0x07; if (mod!=0xC0) { if (f&C_67) // modrm16 { if ((mod==0x00)&&(rm==0x06)) f|=C_MEM2; if (mod==0x40) f|=C_MEM1; if (mod==0x80) f|=C_MEM2; } else // modrm32 { if (mod==0x40) f|=C_MEM1; if (mod==0x80) f|=C_MEM4; if (rm==0x04) rm = (*iptr++) & 0x07; // rm<-sib.base if ((rm==0x05)&&(mod==0x00)) f|=C_MEM4; } } } // C_MODRM if (f&C_MEM67) if (f&C_67) f|=C_MEM2; else f|=C_MEM4; if (f&C_DATA66) if (f&C_66) f|=C_DATA2; else f|=C_DATA4; if (f&C_MEM1) iptr++; if (f&C_MEM2) iptr+=2; if (f&C_MEM4) iptr+=4; if (f&C_DATA1) iptr++; if (f&C_DATA2) iptr+=2; if (f&C_DATA4) iptr+=4; return iptr - iptr0; } /************************************************************************ 使用方法: 原始的LoadLibraryA的声明是: HMODULE WINAPI LoadLibraryA( LPCSTR lpLibFileName ); 1. 那么首先定义一下hook的WINAPI的类型 typedef HMODULE (WINAPI* __pfnLoadLibraryA)(LPCTSTR lpFileName); 2. 定义一个全局的变量,用于保存正式的函数地址 __pfnLoadLibraryA real_loadLibrary; 3. 自定义自己的hook函数,可以在里面调用真实的被HOOK函数地址 HMODULE my_loadLibrary(LPCTSTR lpFileName) { char s[30]; sprintf(s, "loading dll: %s", lpFileName); MessageBox(NULL, s, "MSG", MB_OK); real_loadLibrary(lpFileName); } 4. HOOK的过程 real_loadLibrary = LoadLibraryA; //保存原始的LoadLibrary PHOOK_ENV env = HookFun((PVOID*)&real_loadLibrary, (PVOID)my_loadLibrary); //HOOK, 返回PHOOK_ENV, 同时把真实的函数地址保存到了real_loadLibrary变量中 LoadLibrary("test.dll"); //正常调用,已经被HOOK UnHookFun(env); //取消HOOK LoadLibrary("test.dll"); //没效果 */ ////////////////////////////////////////////////////////////////////////////////////////////////// //定义下面这行可以作为演示使用 //#define TEST_MAIN1 #define TEST_MAIN2 #ifdef TEST_MAIN1 typedef HMODULE (WINAPI* __pfnLoadLibraryA)(LPCTSTR lpFileName); __pfnLoadLibraryA real_loadLibrary; HMODULE WINAPI my_loadLibrary(LPCTSTR lpFileName) { char s[30]; sprintf(s, "loading dll: %s", lpFileName); MessageBox(NULL, s, "MSG", MB_OK); return real_loadLibrary(lpFileName); } int main() { real_loadLibrary = LoadLibraryA; PHOOK_ENV env = HookFun((PVOID*)&real_loadLibrary, (PVOID)my_loadLibrary); LoadLibrary("test.dll"); UnHookFun(env); LoadLibrary("test.dll"); return 1; } #endif #ifdef TEST_MAIN2 //首先需要定义一个被HOOK的函数原型 typedef DWORD (* pFun)(DWORD a, DWORD b); //然后需要定义一个全局的函数指针,保存真实的函数地址(也就是经过处理后的函数的新入口) pFun real_add; //这个是被HOOK的函数 DWORD add(DWORD a, DWORD b) { printf("add: %d+%d=%d\n", a, b, a+b); return a+b; } //这个是自定义的HOOK函数,可以通过全局的函数指针调用真实的函数 DWORD my_add(DWORD a, DWORD b) { printf("my_add(mins): %d-%d=%d\n",a, b, a-b); return real_add(a, b); } int main(int argc, char **argv) { DWORD a = 10; DWORD b = 6; add(a, b); real_add = add; PHOOK_ENV env = HookFun((PVOID*)&real_add, (PVOID)my_add); add(a, b); add(4, 7); UnHookFun(env); add(a, b); return 1; } /* 输出: add: 10+6=16 my_add(mins): 10-6=4 add: 10+6=16 my_add(mins): 4-7=-3 add: 4+7=11 add: 10+6=16 */ #endif /////////////////////////////////////////////////////////////////////////

文件下载

资源详情

[{"title":"( 2 个子文件 5KB ) hook任意api的函数.rar","children":[{"title":"inlinehook.cpp <span style='color:#111;'> 8.74KB </span>","children":null,"spread":false},{"title":"inlinehook.h <span style='color:#111;'> 12.25KB </span>","children":null,"spread":false}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明