[转]C/C++ HOOK API（原理深入剖析之-LoadLibraryA）

9月都快结束了，之前一直忙到写自己的东西加上上班。基本没有时间研究下汇编和C C++方面的感兴趣的东西。再怎么说嘛，9月还是得写一篇撒，以后每月至少一篇吧。给自己定了，希望大家监督。嘿嘿！

这篇文章就来谈谈平常很常见的HOOK技术，这里呢。写得比较简单，方法很多。只讲原理！希望大鸟们别吐我口水哈 - -。好！切入正题。

首先是概念吧。什么是钩子（HOOK）？

钩子(Hook)，是Windows消息处理机制的一个平台,应用程序可以在上面设置子程以监视指定窗口的某种消息，而且所监视的窗口可以是其他进程所创建的。当消息到达后，在目标窗口处理函数之前处理它。钩子机制允许应用程序截获处理window消息或特定事件。
钩子实际上是一个处理消息的程序段，通过系统调用，把它挂入系统。每当特定的消息发出，在没有到达目的窗口前，钩子程序就先捕获该消息，亦即钩子函数先得到控制权。这时钩子函数即可以加工处理（改变）该消息，也可以不作处理而继续传递该消息，还可以强制结束消息的传递。
这上面只是一个概念，对它有所了解而已。上面主要应用在Windows消息处理机制里面的一个解释。这里我只是单纯的谈谈拦截我们常用的LoadLibraryA加载这个函数。让我们的程序或者目标程序在调用这个函数加载链接库的时候，先执行我们自己写的函数，然后在进行正常加载。通俗的说就是a----->b.  我们在中间加上一个c。 a-------->c----->b让他先执行c然后再执行b。这里的c就是我们自己的函数了。
呵呵，概念说得差不多了，开始行动写代码撒：

#include <iostream> #include <Windows.h> using namespace std; #pragma warning( disable: 4309 ) #pragma warning( disable: 4311 ) typedef HMODULE ( WINAPI *HOOKAPI )( IN LPCSTR ); #define MYHOOKMETHOD ( __fun ) HMODULE WINAPI __fun #define DECLARE_REGISTER ( __0bj, __lawfunc, __newfunc ) Inline_Hook< HOOKAPI, 1 > __Obj( __lawfunc, __newfunc ) struct __InlineHOOK_Base { DWORD _argsBytes; void* _lawFunc; void* _newFunc; char _lawByteCode[16]; char _newByteCode[16]; bool unhook ( void ) { // It's hooked. if ( memcmp( _newByteCode, _lawFunc, 16 ) == 0 ) { DWORD dwOldFlag; VirtualProtect( _lawFunc, 8, PAGE_EXECUTE_READWRITE, &dwOldFlag ); memcpy( _lawFunc, _lawByteCode, 16 ); VirtualProtect( _lawFunc, 8, dwOldFlag, &dwOldFlag ); return true; } return false; } bool hook ( void ) { // It's saved. if ( memcmp( _lawByteCode, _lawFunc, 16 ) == 0 ) { DWORD dwOldFlag; VirtualProtect( _lawFunc, 8, PAGE_EXECUTE_READWRITE, &dwOldFlag ); memcpy( _lawFunc, _newByteCode, 16 ); VirtualProtect( _lawFunc, 8, dwOldFlag, &dwOldFlag ); return true; } return false; } __InlineHOOK_Base( void* lawfun, void* newfun, DWORD args ); }; void __declspec( naked ) __Inline_Hook_Func ( void ) { __asm { push ebp // save maybe usefull register. push ebx push esi push ecx call __InlineHOOK_Base::unhook // first, remove the hook in order to call the normal function. test eax, eax // check the remove was successful jz __return __getargnum: mov eax, dword ptr[esp] // esp just is ecx, also is __InlineHOOK_Base's this pointer. mov ecx, dword ptr[eax] // get first 4 bytes, that is params total size. shr ecx, 2 // get params num, equal with __InlineHOOK_Base::_argsBytes / sizeof( DWORD ) test ecx, ecx // check whether there are params. jz __callfunc // no param __pushargs: mov edx, esp // __InlineHOOK_Base's this pointer. add edx, 14h // navigate to first call ret addr. add edx, dword ptr[eax]; // add params size. push dword ptr[edx]; // push the dll file name pointer. loop __pushargs __callfunc: call [eax+8] // call my function . mov ecx, dword ptr[esp] // get __InlineHOOK_Base's this pointer. push edx // save my function return value. push eax call __InlineHOOK_Base::hook // rehook. pop eax // get saved return value, provided to my superiors to use pop edx __return: pop ecx pop esi pop ebx pop ebp ret } } __InlineHOOK_Base ::__ InlineHOOK_Base ( void* lawfun, void* newfun, DWORD args ) : _lawFunc( lawfun ), _newFunc( newfun ), _argsBytes( args * 4