Inline Hook 之(监视任意函数)

前面已经写过两次inline hook的博文了,第一篇为:《C/C++ HOOK API(原理深入剖析之-LoadLibraryA)》,这篇博文的方法是通过修改任意函数的前面N个字节,实现跳转并进入到我们自定义的hook函数里,执行完毕我们的hook函数之后,再直接调用被hook的函数。第一篇的方法没有考虑多线程的情况,所以在多线程环境下会有问题。第二篇为:《 Inline HOOK API 改进版(hot-patching)》,这篇的初衷是为了解决多线程的问题,因为这种方式是一直hook的,直到程序结束。因此在多线程情况下,并不会在hook与unhook期间执行被hook的函数。不过,第二种方式有局限性,它主要针对windows的很多API而设计,在函数头部必须有空闲的两个字节和函数首地址之前的5个空闲字节(一般为5个nop指令),这样能够实现一个short jmp和一个long jmp。从而实现了hook,在此就不再累述了。

 

好了,本文已经是第三次写hook相关的东西了。前两种hook有个共同点就是他们都是直接hook掉某个函数,也就是说,当执行完我们自定义的hook函数之后,又从被hook函数的首部开始执行,被hook函数一进入就被跳转了。而本文,则要实现在某个函数体内部任意地方进行hook并跳转,执行完我们的函数之后,再回到原来的位置继续向下执行完未执行的逻辑。那么,初看这种方式似乎与前面写的两篇hook没有什么差别,都是hook,都是跳转然后回到被hook的函数。但仔细一想,你会发现本文要实现的方式要比前面两种hook复杂,因为hook的地方是函数体内任意地方,那么回来的时候就不是直接调用被hook的函数了,而是要回到之前hook的地方去。这期间就涉及到hook函数的返回地址问题和被hook函数的返回地址问题。

 

说了这么多,可能还是有点晕,先不管为什么要这么做,也不管这种hook方式能有什么用途(在本文最后会说明用途),下面我们先写一些代码,在实践中来想一想这种方式有什么用途,并且与之前的两篇hook进行比较。

 

首先,我们需要一个自定义的hook函数,这个函数也就是被hook函数被hook后跳转到的地方,这个hook函数负责hook与unhook,还可以监视寄存器,监视内存,也可以管理hook的次数,以供我们灵活的hook需求。直接贴代码吧:

#include #include #pragma warning( disable : 4311 ) #pragma warning( disable : 4312 ) #define HOOK_BYTES 5 typedef unsigned int uint; uint hookAddr = 0; char old_code[ HOOK_BYTES ]; char new_code[ HOOK_BYTES ]; void printRegisters( void ); bool hook( void ) { DWORD dwFlag; if ( VirtualProtect( ( void* )hookAddr, HOOK_BYTES, PAGE_EXECUTE_READWRITE, &dwFlag ) ) { memcpy( old_code, ( void* )hookAddr, HOOK_BYTES ); memcpy( ( void* )hookAddr, new_code, HOOK_BYTES ); VirtualProtect( ( void* )hookAddr, HOOK_BYTES, dwFlag, &dwFlag ); return true; } return false; } void unhook( void ) { DWORD dwFlag; if ( VirtualProtect( ( void* )hookAddr, HOOK_BYTES, PAGE_EXECUTE_READWRITE, &dwFlag ) ) { memcpy( ( void* )hookAddr, old_code, HOOK_BYTES ); VirtualProtect( ( void* )hookAddr, HOOK_BYTES, dwFlag, &dwFlag ); } } namespace global { uint gEAX = 0; uint gEBX = 0; uint gECX = 0; uint gEDX = 0; uint gESP = 0; uint gEBP = 0; uint gESI = 0; uint gEDI = 0; uint gRet = 0; // 临时的返回地址 uint gTmp = 0; // 一些临时的值保存 uint gPar = 0; // 被hook函数的正常返回地址 uint gCnt = 1; // 当前hook的次数 uint gMax = 0; // 最大hook次数,为0表示一直hook bool bEnt = 0; // 是否为第一次进入hook函数 } void __declspec( naked ) hook_jmp( void ) { __asm { __entry: pushad { cmp global::bEnt, 0 // 如果没有进入则表示需要unhook je __first cmp global::gMax, 0 // 如果为0,则一直启用hook逻辑 je __second mov eax, global::gCnt cmp eax, global::gMax // 如果当前hook次数没有达到最大次数,则继续 jl __second mov global::gCnt, 1 // reset state mov global::bEnt, 0 // reset state mov global::gMax, 0 // reset state mov eax, global::gPar // 被hook函数的正常返回地址 mov global::gRet, eax // 准备跳转到被hook函数的上层调用,结束hook popad jmp __ret } __first: // 保存相关重要寄存器值 { popad mov global::gEAX, eax mov global::gEBX, ebx mov global::gECX, ecx mov global::gEDX, edx mov global::gESP, esp mov global::gEBP, ebp mov global::gESI, esi mov global::gEDI, edi } // 第一次进入,unhook并监视相关状态 pushad { mov global::bEnt, 1 // 记录状态 mov edi, global::gEBP // 被hook函数的ebp mov eax, [ edi + 4 ] // 被hook函数的返回地址(其上层调用地址) mov global::gPar, eax // 保存返回地址 mov esi, __entry // 将被hook函数的返回地址修改为 mov [ edi + 4 ], esi // 本函数的首地址,以便执行完被hook函数的 // 剩余逻辑之后能够返回到本函数,决定是否 // 还需要hook。 call printRegisters // 打印寄存器值[测试],或者其他 call unhook // unhook mov eax, hookAddr // 获得被hook的内存地址 mov global::gRet, eax } popad pop global::gTmp // 移除本函数的返回地址,并将hook的地址设置 jmp __ret // 为本函数的返回地址,从而实现跳转 __second: // 第二次进入, 继续hook, 这次进入是被hook函数ret返回的,没有新的ret地址被压栈 { mov global::bEnt, 0 // 设置状态 add global::gCnt, 1 // 增加hook计数 call hook // hook mov eax, global::gPar // 将被hook函数的返回地址设置为本函数的 mov global::gRet, eax // 返回地址,从而实现正常的函数流程 } popad __ret: push global::gRet // 修改本函数的返回地址 ret } } void setHookBytes( uint addr ) { hookAddr = addr; new_code[ 0 ] = ( char )0xe8; // call 指令机器码 ( uint& )new_code[ 1 ] = ( uint )hook_jmp - addr - 5; // 计算跳转偏移 } void printRegisters( void ) { printf( "EAX = 0x%08x/n", global::gEAX ); printf( "EBX = 0x%08x/n", global::gEBX ); printf( "ECX = 0x%08x/n", global::gECX ); printf( "EDX = 0x%08x/n", global::gEDX ); printf( "ESP = 0x%08x/n", global::gESP ); printf( "EBP = 0x%08x/n", global::gEBP ); printf( "ESI = 0x%08x/n", global::gESI ); printf( "EDI = 0x%08x/n", global::gEDI ); }

如上,hook_jmp函数即为我们自定义的hook函数,当被hook函数被hook之后,就会跳转到这个函数里,执行相关逻辑,上面我加了很详细的注释。应该很容易看懂。还是先看怎么使用这套方法,再来细说,代码如下:

void testHook( void ) { printf( "This is a hook test 1./n" ); printf( "This is a hook test 2./n" ); printf( "This is a hook test 3./n" ); printf( "This is a hook test 4./n" ); printf( "______________________/n" ); } int main( void ) { uint hook_addr = 0x0042ec7b; setHookBytes( hook_addr ); global::gMax = 2; if ( hook() ) { testHook(); testHook(); testHook(); } system( "pause" ); return 0; }

如上,testHook函数即为被hook的函数,在main函数中,0x0042ec7b则为testHook函数里的第二个printf调用的地址,在你的机器上可能不一样。这里只是测试之用。testHook函数具体反汇编代码如下:void testHook( void ) { 0042EC50 push ebp 0042EC51 mov ebp,esp 0042EC53 sub esp,0C0h 0042EC59 push ebx 0042EC5A push esi 0042EC5B push edi 0042EC5C lea edi,[ebp-0C0h] 0042EC62 mov ecx,30h 0042EC67 mov eax,0CCCCCCCCh 0042EC6C rep stos dword ptr es:[edi] printf( "This is a hook test 1./n" ); 0042EC6E push offset string "This is a hook test 1./n" (487E24h) 0042EC73 call @ILT+4550(_printf) (42D1CBh) 0042EC78 add esp,4 printf( "This is a hook test 2./n" ); 0042EC7B push offset string "This is a hook test 2./n" (487E08h) 0042EC80 call @ILT+4550(_printf) (42D1CBh) 0042EC85 add esp,4 printf( "This is a hook test 3./n" ); 0042EC88 push offset string "This is a hook test 3./n" (487DECh) 0042EC8D call @ILT+4550(_printf) (42D1CBh) 0042EC92 add esp,4 printf( "This is a hook test 4./n" ); 0042EC95 push offset string "This is a hook test 4./n" (487DD0h) 0042EC9A call @ILT+4550(_printf) (42D1CBh) 0042EC9F add esp,4 printf( "______________________/n" ); 0042ECA2 push offset string "_____________________./n" (487DB4h) 0042ECA7 call @ILT+4550(_printf) (42D1CBh) 0042ECAC add esp,4 } 0042ECAF pop edi 0042ECB0 pop esi 0042ECB1 pop ebx 0042ECB2 add esp,0C0h 0042ECB8 cmp ebp,esp 0042ECBA call @ILT+3570(__RTC_CheckEsp) (42CDF7h) 0042ECBF mov esp,ebp 0042ECC1 pop ebp 0042ECC2 ret

我们hook的就是第18行(0042EC7B)那句代码,setHookBytes构建了一个5字节的call语句,0xe8为CALL指令的机器码,后面4个字节是CALL的偏移量(目标地址 - 当前地址 - CALL指令占用的5个字节)。

 

在main函数中,构建了hook的5个字节之后,设置了hook次数,如main函数那段代码的第15行:global::gMax = 2,则会hook两次。然后是main函数那段代码的第16行,调用hook函数,将5个字节的call指令写入0042EC7B中,并保存了0042EC7B中原来的代码到old_code中。之后,我们便可以调用testHook函数进行测试hook的流程了。最终输出结果为:

This is a hook test 1.
EAX = 0x00000017
EBX = 0x7ffdc000
ECX = 0x8df97741
EDX = 0x00499148
ESP = 0x0012fd84
EBP = 0x0012fe54
ESI = 0x00000000
EDI = 0x0012fe54

This is a hook test 2.
This is a hook test 3.
This is a hook test 4.

______________________
This is a hook test 1.
EAX = 0x00000017
EBX = 0x7ffdc000
ECX = 0x8df97741
EDX = 0x00499148
ESP = 0x0012fd84
EBP = 0x0012fe54
ESI = 0x00000000
EDI = 0x0012fe54

This is a hook test 2.
This is a hook test 3.
This is a hook test 4.

______________________
This is a hook test 1.
This is a hook test 2.
This is a hook test 3.
This is a hook test 4.

______________________

 

可以看出,前面两次调用testHook函数时,都执行了hook_jmp函数,并调用了printRegisters函数将寄存器打印了出来,之后又回到testHook中,继续输出后面的3句字符串。当两次hook之后,第三次调用testHook时,就不会再输出寄存器了,也没有被hook了。

 

我们来看几个比较hook_jmp中比较关键的几个地方:

第104到110行:这段汇编代码,主要用于保存testHook函数(被hook函数)的正常的返回地址(main函数里调用testHook的下一句指令的地址)到global::gPar变量中,并将hook_jmp的首地址(也就是__entry标签指示的地址)写入testHook函数的返回地址所在的内存里。这样当unhook并执行完testHook之后又能回到hook_jmp中,进一步判断是否需要下一次hook。如果不需要再hook(已经达到最大hook次数)时,则会执行第79到80行的两句汇编代码,这两句汇编代码的作用是将hook_jmp函数的返回地址设置为testHook函数正常的返回地址,也就是main函数里调用testHook函数的下一句汇编代码的地址(ret指令的原理如果不清楚,请看前两篇hook文章或查阅相关资料)。这样一来,当不再需要hook时,就能顺利的从hook_jmp函数返回后直接跳转到main函数的作用域里。这样整个调用流程就符合原本的调用流程了。

 

第113行:这句代码是在调用了printRegister函数之后进行unhook操作,将原本的5个字节的代码重新拷贝到testHook函数的相应代码地址的内存里,本例中为testHook函数中第二句printf函数调用的地址。unhook之后,第115到116行的两句代码与第79到80行的两句汇编代码类似,只不过这时是将被hook的内存地址设置为hook_jmp的返回地址,这样就能在第一次进入testHook函数并执行完毕返回时,能够跳转到被hook的地址(hookTest函数里第2句printf调用的地址)继续向下执行剩余的逻辑。

 

第131到132行:这两句汇编代码与第79到80行的两句汇编代码一致,都是将main函数里相应的代码地址设置为hook_jmp函数的ret返回地址,这样就能直接从hook_jmp跳转到main函数里继续向下执行,这样也就代表testHook被顺利的调用完成。

 

所以,总结下来,hook_jmp函数会进入两次,第一次用于监视一些数据,本例只监视了相关寄存器,还可以增加监视指定内存地址等等。第一次进入时,会保存被hook函数(testHook函数)的返回地址,并将其修改为hook_jmp函数的首地址,这样做是为了执行完testHook函数之后能够第二次进入hook_jmp函数。那么,第二次进入后,首先是判断是否还需要hook,不需要则直接返回到main函数里,如果还要继续hook,则再次调用hook函数,然后跳转到main函数里。这样就构成了一个严密的调用流程,一切都看起来很和谐的调用,有点类似缓冲区溢出攻击的原理。

 

hook_jmp函数中需要注意寄存器的保存,否则输出的寄存器值并不是testHook函数执行到hook位置时的寄存器状态,这样就丧失了监视的意义。

 

原理上其实比较简单,构建稍微细致了一些,与前两篇hook最大的不同就是需要手动修改ret的返回地址,从而达到hook的目的,不像之前的两篇hook,在进入hook函数之后,要回到被hook的函数时,只需要直接call就可以了,并不需要维护ret指令的返回地址。另外,由于本文的hook方式与第一篇的hook方式类似,所以本文的方式并没有支持多线程环境。

 

再者,本文的方式还有另外一些局限性:

1. 不能hook函数大括号以外的代码,也就是反汇编代码中大括号之前和之后的代码,因为本文的hook方式是需要通过被hook函数的ebp获取ret地址的,如果设置ebp的相关代码被hook,就得不到正确的ret返回地址等。而且hook大括号之外的代码意义并不大。

2. 被hook的函数的栈帧不能被编译器优化掉,否则得不到ret地址和ebp。

3. hook的代码地址必须为某条汇编指令机器码序列的首地址,本文并没有兼容任意代码地址hook,当然任意代码地址是可以实现的,不过在实际中要监视一些状态,这样做并没有太大的意义。所以你需要明确的知道你hook的代码地址是合法的,并且不会破坏原有的代码逻辑。

 

好了,整个hook的逻辑和原理就介绍得差不多了,再附上一张流程图:

Inline Hook 之(监视任意函数)_第1张图片

如图所示,testHook函数如果被hook了,它是不会直接返回到main函数的,而是要先进入hook_jmp函数,再由hook_jmp函数返回到main函数,知道不再hook的时候,testHook才有“权利”直接返回到main函数里。

 

好了差不多就介绍完了,最后说说这套hook的用途,其实最初我是想用于反外挂检测上,于是写了这么一套hook逻辑,方便在不用调试客户端程序的情况下,监视在运行有外挂的客户端的一些关键的反外挂检测点,例如某个反外挂检测逻辑是否被执行,某些检测结果值的监视等。外挂一般加了比较强悍的保护壳,这些壳具有反调试等功能。典型的就是VMP壳和UPX壳。当然在这之前尝试了脱壳等方式调试跟踪有外挂的客户端,但是脱壳和绕过反调试机制的工作量都相对较大一些,我们的一贯思想又是以最少的工作量去做到好的反外挂效果。于是我便写了这套hook规则,用于检测当前反外挂系统是否被破解,是否被绕开。通过这套规则,就可以很方便的设立观测点,监视相关逻辑和数据,当然还可以有更多用途,此处不再一一说明,另外为此还写了一个可视化工具,方便工作中使用,并成功检测了多个外挂的破解机制,如图:

Inline Hook 之(监视任意函数)_第2张图片

 

 

左边为hook逻辑的界面,右边tool页面为调用任意函数的功能,同时支持查看模块和查看并修改内存。基本上已经够用了。- -

好了,本文到此结束,由于水平有限,可能存在bug,还望指教,衷心感谢!

 

 

 

 

 

 

 

 

 

你可能感兴趣的:(汇编语言,C/C++,inline汇编语言,masefee透视C++)