码迷，mamicode.com
首页 > Windows程序 > 详细

实现拦截API的钩子(Hook)

时间：2015-11-28 00:56:26      阅读：241      评论：0      收藏：0      [点我收藏+]

标签：

道理不多讲，简单说就是将系统API的跳转地址，替换为我们自己写的API的地址，所以要求我们自定义的API函数要和被拦截的API有相同的参数。在用完后，记得恢复。

 

因为要挂全局的钩子，所以Hook的部分，做成DLL。   源码下载

 

Hook.DLL主工程文件代码

 

[delphi] view plaincopy
 
  1. library Hook;  
  2.   
  3. uses  
  4.   SysUtils,  
  5.   Windows,  
  6.   Classes,  
  7.   ApiDefine in ‘ApiDefine.pas‘,  
  8.   APIHook in ‘APIHook.pas‘;  
  9.   
  10. {$R *.res}  
  11. var  
  12.   HookHandle: HHook;  
  13.   
  14. function HookProc(code:Integer;wparam:WPARAM;lparam:LPARAM):LRESULT;stdcall;  
  15. begin  
  16.   Result := CallNextHookEx(HookHandle,code,wparam,lparam);  
  17. end;  
  18.   
  19. procedure SetHook;stdcall;  
  20. begin  
  21.   HookHandle := SetWindowsHookEx(WH_GETMESSAGE,@HookProc,HInstance,0);  
  22. end;  
  23.   
  24. procedure StopHook;stdcall;  
  25. begin  
  26.   UnhookWindowsHookEx(HookHandle);  
  27. end;  
  28.   
  29. exports  
  30.   SetHook name ‘SetHook‘,  
  31.   StopHook name ‘StopHook‘;  
  32.   
  33. {已启动就挂上，修改API函数指向}  
  34. begin  
  35.   API_Hook;  
  36. end.  

 

 

APIHook单元，这个单元实现对API地址的替换

 

[c-sharp] view plaincopy
 
  1. unit APIHook;  
  2.   
  3. interface  
  4.   
  5. uses  
  6.   Windows, SysUtils, Classes;  
  7.   
  8. type  
  9.   //引入表入口数据结构  
  10.   Image_Import_Entry = packed record  
  11.     OriginalFirstThunk:DWORD;  
  12.     TimeDateStamp:DWORD;  
  13.     ForwarderChain:DWORD;  
  14.     Name:DWORD;  
  15.     FirstThunk:DWORD;  
  16.   end;  
  17.   PImage_Import_Entry = ^Image_Import_Entry;  
  18.   TImportCode = packed record  
  19.     JmpCode: Word;  
  20.     AddressOfPFun: PPointer;  
  21.   end;  
  22.   PImportCode = ^TImportCode;  
  23.   
  24.   function GetFunTrueAddress(Code:Pointer):Pointer;  
  25.   function ReplaceFunAddress(oldfun:Pointer;newfun:Pointer):Integer;  
  26.   
  27. implementation  
  28.   
  29. //获得实际地址  
  30. function GetFunTrueAddress(Code: Pointer): Pointer;  
  31. var  
  32.    func: PImportCode;  
  33. begin  
  34.    Result := Code;  
  35.    if Code = nil then exit;  
  36.    try  
  37.       func := code;  
  38.       if (func.JmpCode = $25FF) then  
  39.       begin  
  40.          Result := func.AddressOfPFun^;  
  41.       end;  
  42.    except  
  43.       Result := nil;  
  44.    end;  
  45. end;  
  46.   
  47. //替换地址  
  48. function ReplaceFunAddress(oldfun:Pointer;newfun:Pointer): Integer;  
  49. var  
  50.    IsDone: TList;  
  51.    function ReplaceAddressInModule(hModule: THandle; OldFunc, NewFunc: Pointer): Integer;  
  52.    var  
  53.       DosHeader: PImageDosHeader;  
  54.       NTHeader: PImageNTHeaders;  
  55.       ImportDesc: PImage_Import_Entry;  
  56.       RVA: DWORD;  
  57.       Func: ^Pointer;  
  58.       DLL: string;  
  59.       f: Pointer;  
  60.       written: DWORD;  
  61.    begin  
  62.       Result := 0;  
  63.       DosHeader := Pointer(hModule);  
  64.       //已经找过，则退出  
  65.       if IsDone.IndexOf(DosHeader) >= 0 then exit;  
  66.       IsDone.Add(DosHeader);  
  67.   
  68.       oldfun := GetFunTrueAddress(OldFunc);  
  69.   
  70.       if IsBadReadPtr(DosHeader, SizeOf(TImageDosHeader)) then exit;  
  71.       if DosHeader.e_magic <> IMAGE_DOS_SIGNATURE then exit;  
  72.       NTHeader := Pointer(Integer(DosHeader) + DosHeader._lfanew);  
  73.       //引入表的虚拟地址  
  74.       RVA := NTHeader^.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress;  
  75.   
  76.       if RVA = 0 then exit;  
  77.       ImportDesc := pointer(integer(DosHeader) + RVA);  
  78.       while (ImportDesc^.Name <> 0) do  
  79.       begin  
  80.         //引入文件名  
  81.          DLL := PChar(Integer(DosHeader) + ImportDesc^.Name);  
  82.          //获得该DLL的句柄，然后递归查找  
  83.          ReplaceAddressInModule(GetModuleHandle(PChar(DLL)), oldfun, newfun);  
  84.          //引入函数入口  
  85.          Func := Pointer(Integer(DOSHeader) + ImportDesc.FirstThunk);  
  86.          //如果函数指针不为空  
  87.          while Func^ <> nil do  
  88.          begin  
  89.            //取得真是地址  
  90.             f := GetFunTrueAddress(Func^);  
  91.             //如果和我们要拦截的Api函数地址一样  
  92.             if f = oldfun then  
  93.             begin  
  94.               //替换成我们自己的Api地址  
  95.                WriteProcessMemory(GetCurrentProcess, Func, @NewFunc, 4, written);  
  96.                if Written > 0 then Inc(Result);  
  97.             end;  
  98.             //继续找  
  99.             Inc(Func);  
  100.          end;  
  101.          Inc(ImportDesc);  
  102.       end;  
  103.    end;  
  104.   
  105. begin  
  106.    IsDone := TList.Create;  
  107.    try  
  108.      //GetModuleHandle,参数nil，为获取自身的模块句柄  
  109.       Result := ReplaceAddressInModule(GetModuleHandle(nil), oldfun, newfun);  
  110.    finally  
  111.       IsDone.Free;  
  112.    end;  
  113. end;  
  114. end.  

 

 

ApiDefine单元，这里实现我们自定义的API

 

[delphi] view plaincopy
 
  1. unit ApiDefine;  
  2.   
  3. interface  
  4.   
  5. uses  
  6.   Windows, SysUtils, Classes,Messages,APIHook,ShellAPI;  
  7.   
  8.   procedure API_Hook;  
  9.   procedure API_UnHook;  
  10.   
  11. implementation  
  12.   
  13. //自定义Api的类型  
  14. type  
  15.   TMsgA = function(hwn: hwnd; lptext: pchar; lpcapion: pchar; utype: cardinal):integer; stdcall;  
  16.   TShellExc = function(hwn: HWND;lpoperate: PChar;lpfilename: PChar; lpparam: PChar; lpdir:PChar;cmd:Integer):Integer;stdcall;  
  17.   TTextOut = function(DC:HDC;X:Integer;Y:Integer;options:Integer;rect:PRect;str:PAnsiChar;count:Integer;dx:PInteger):Boolean;stdcall;  
  18. var  
  19.   oldMsgA : TMsgA;  
  20.   oldShellExc : TShellExc;  
  21.   oldTextOut : TTextOut;  
  22.   
  23. //自定义Api的实现  
  24. function NewMsgA(hwn: hwnd; lptext: pchar; lpcaption: pchar; utype: cardinal):integer; stdcall;  
  25. begin  
  26.   Result := oldMsgA(hwn,‘成功拦截MessageBoxA‘,‘哈哈‘,utype);  
  27. end;    
  28.   
  29. function NewShellExc(hwn: HWND;lpoperate: PChar;lpfilename: PChar; lpparam: PChar; lpdir:PChar;cmd:Integer):Integer;stdcall;  
  30. begin  
  31.   Result := oldShellExc(hwn,lpoperate,‘c:/2.txt‘,lpfilename,lpdir,cmd);  
  32. end;  
  33.   
  34. {TextOut调用的是ExtTextOut}  
  35. function NewTextOut(DC:HDC;X:Integer;Y:Integer;options:Integer;rect:PRect;str:PAnsiChar;count:Integer;dx:PInteger):Boolean;stdcall;  
  36. begin  
  37.   {这个rect也是可以修改的，以便容纳更多的字符显示}  
  38.   Result := oldTextOut(DC,50,50,options,rect,‘中国‘,count,dx);  
  39. end;  
  40.   
  41. procedure API_Hook;  
  42. begin  
  43.   if @oldMsgA = nil then  
  44.     @oldMsgA := GetFunTrueAddress(@MessageBoxA);  
  45.   if @oldShellExc = nil then  
  46.     @oldShellExc := GetFunTrueAddress(@ShellExecute);  
  47.   if @oldTextOut = nil then  
  48.     @oldTextOut := GetFunTrueAddress(@ExtTextOut);  
  49.   //替换    
  50.   ReplaceFunAddress(@oldMsgA,@NewMsgA);  
  51.   ReplaceFunAddress(@oldShellExc,@NewShellExc);  
  52.   ReplaceFunAddress(@oldTextOut,@NewTextOut);  
  53. end;  
  54.   
  55. procedure API_UnHook;  
  56. begin  
  57.   if @oldMsgA <> nil then  
  58.     ReplaceFunAddress(@NewMsgA,@oldMsgA);  
  59.   if @oldShellExc <> nil then  
  60.     ReplaceFunAddress(@NewShellExc,@oldShellExc);  
  61.   if @oldTextOut <> nil then  
  62.     ReplaceFunAddress(@NewTextOut,@oldTextOut);  
  63. end;  
  64.   
  65. initialization  
  66. //结束时恢复原Api地址  
  67. finalization  
  68.   API_UnHook;  
  69.   
  70. end.  

 

 

主程序代码，大家可以把，消息、打开文件、画文字的代码写到另外的程序，本程序只负责挂钩和摘钩，那样可以看到系统钩子的效果。

 

[delphi] view plaincopy
 
  1. unit TestMain;  
  2.   
  3. interface  
  4.   
  5. uses  
  6.   Windows, Messages, SysUtils, Variants, Classes, Graphics, Controls, Forms,  
  7.   Dialogs, StdCtrls,ShellAPI;  
  8.   
  9. type  
  10.   TForm1 = class(TForm)  
  11.     btn_Hook: TButton;  
  12.     btn_Msg: TButton;  
  13.     btn_UnHook: TButton;  
  14.     btn_OpenFiel: TButton;  
  15.     btn_TextOut: TButton;  
  16.     procedure btn_HookClick(Sender: TObject);  
  17.     procedure btn_MsgClick(Sender: TObject);  
  18.     procedure btn_UnHookClick(Sender: TObject);  
  19.     procedure btn_OpenFielClick(Sender: TObject);  
  20.     procedure btn_TextOutClick(Sender: TObject);  
  21.   private  
  22.     { Private declarations }  
  23.   public  
  24.     { Public declarations }  
  25.   end;  
  26.   
  27. var  
  28.   Form1: TForm1;  
  29.   
  30. implementation  
  31.   
  32. procedure SetHook;stdcall;external ‘Hook.dll‘;  
  33. procedure StopHook;stdcall;external ‘Hook.dll‘;  
  34.   
  35. {$R *.dfm}  
  36.   
  37. procedure TForm1.btn_HookClick(Sender: TObject);  
  38. begin  
  39.   SetHook;  
  40. end;  
  41. procedure TForm1.btn_UnHookClick(Sender: TObject);  
  42. begin  
  43.   StopHook;  
  44. end;  
  45.   
  46. {被拦截后，执行我们自己的NewMsgA方法}  
  47. procedure TForm1.btn_MsgClick(Sender: TObject);  
  48. begin  
  49.   MessageBoxA(Handle,‘能拦住我吗‘,‘询问‘,MB_OK);  
  50. end;  
  51.   
  52. {本想打开c:/1.txt，被拦截后，打开c:/2.txt}  
  53. procedure TForm1.btn_OpenFielClick(Sender: TObject);  
  54. begin  
  55.   ShellExecute(Handle,‘open‘,‘c:/1.txt‘,nil,nil,SW_NORMAL);  
  56. end;  
  57. {本想在0,0出画出‘Hello‘，被拦截后，在50,50的位置画出‘中国‘}  
  58. procedure TForm1.btn_TextOutClick(Sender: TObject);  
  59. begin   
  60.   Self.Canvas.TextOut(0,0,‘Hello‘);  
  61. end;  
  62.   
  63. end.  

 

 

下图是执行画文字代码后的效果，本想在[0,0]坐标画出‘Hello‘，被拦截后，在[50,50]的位置画出‘中国‘

 

技术分享

 

http://blog.csdn.net/bdmh/article/details/6104475

实现拦截API的钩子(Hook)

标签：

原文地址：http://www.cnblogs.com/findumars/p/5001977.html
踩
(0)
赞
(0)
   
举报
评论 一句话评论（0）
登录后才能评论！
分享档案
更多>
2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)
周排行
mamicode.com排行更多图片
更多
友情链接
兰亭集智   国之画   百度统计   站长统计   阿里云   chrome插件   新版天听网
关于我们 - 联系我们 - 留言反馈
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码！