NDIS网络数据监控程序NDISMonitor(1)-----驱动程序(编译过程与源码讲解)

转载请标明是引用于 http://blog.csdn.net/chenyujing1234 

欢迎大家拍砖!

 

一、编译过程

弄了半天终于编译通过了,看来驱动的开发确实没有应用来得简单啊。

由于DDK提供的ndis.h有4个,什么时候要调用哪个我确实不清楚:

(1)我先用#include <d:\winddk\7600.16385.1\inc\ddk\ndis.h>

结果报错了:

[plain] view plain copy
  1. 1>1>errors in directory e:\g2\fft\usbdri~1\ndismo~1\ndismo~4\driver3  
  2. 1>1>e:\g2\fft\usbdriver\ndismonitor\ndismonitor_v1-00_kernel_vs\driver3\ndishook.c(380) : error C2037: left of 'ProtocolNextOpen' specifies undefined struct/union '_NDIS_OPEN_BLOCK'  
  3. 1>1>e:\g2\fft\usbdriver\ndismonitor\ndismonitor_v1-00_kernel_vs\driver3\ndishook.c(393) : error C2037: left of 'AdapterHandle' specifies undefined struct/union '_NDIS_OPEN_BLOCK'  
  4. 1>1>e:\g2\fft\usbdriver\ndismonitor\ndismonitor_v1-00_kernel_vs\driver3\ndishook.c(734) : error C2037: left of 'ReceiveHandler' specifies undefined struct/union '_NDIS_OPEN_BLOCK'  


定位代码是在:

  1. NDIS_OPEN_BLOCK* GetNextOpen( IN NDIS_OPEN_BLOCK* pnobOpenBlock )  
  2. {  
  3.     // Return the information required.  
  4.   
  5.     return pnobOpenBlock->ProtocolNextOpen;  
  6. }  


我想是因为找不到结构体NDIS_OPEN_BLOCK的定义的原因。关于此结构体请参考文章<<NDIS的NDIS_PROTOCOL_BLOCK和NDIS_OPEN_BLOCK的介绍>>。

于是在DDK包中执行ProtocolNextOpen这个变量,总共出现在三个地方:

怪不得会编译出错,原来在

d:\winddk\7600.16385.1\inc\ddk\ndis.h中根据就没有此成员变量。

看一看以上三个含有ProtocolNextOpen的地方的定义吧:

(1、1)D:\WINDDK\3790.1830\inc\ddk\w2k\ndis.h

(1、2) D:\WINDDK\3790.1830\inc\ddk\wnet\ndis.h

(1、3)D:\WINDDK\3790.1830\inc\ddk\wxp\ndis.h

综上看来我们有两种可选:

D:\WINDDK\3790.1830\inc\ddk\w2k\ndis.h

D:\WINDDK\3790.1830\inc\ddk\wxp\ndis.h

(2)试看看D:\WINDDK\3790.1830\inc\ddk\wxp\ndis.h吧

报以下错误:

[plain] view plain copy
  1. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(7156) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  
  2. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(10163) : error C2061: syntax error : identifier 'W_SEND_HANDLER'  
  3. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(10164) : error C2061: syntax error : identifier 'WTransferDataHandler'  
  4. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(10164) : error C2059: syntax error : ';'  
  5. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(10169) : error C2061: syntax error : identifier 'WSendPacketsHandler'  
  6. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(10169) : error C2059: syntax error : ';'  
  7. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(10171) : error C2061: syntax error : identifier 'CancelSendPacketsHandler'  
  8. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(10171) : error C2059: syntax error : ';'  
  9. 1>1>d:\winddk\3790.1830\inc\ddk\wxp\ndis.h(10182) : error C2061: syntax error : identifier 'QC'  


这么多错误,我想放弃引方案。

(3)试看看D:\WINDDK\3790.1830\inc\ddk\w2k\ndis.h

只有两个报错:

[plain] view plain copy
  1. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(5567) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  
  2. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(5567) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  
  3. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(5567) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  
  4. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(5567) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  
  5. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(5567) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  
  6. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(5567) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  
  7. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(5567) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  
  8. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(5567) : error C2011: '_MAP_REGISTER_ENTRY' : 'struct' type redefinition  

(3、1)既然是重定义,那么我把它的定义屏蔽掉吧:

(3、2)再试看看,这个错误消除了,现在还剩下一个错:

[plain] view plain copy
  1. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(9297) : error C1083: Cannot open include file: 'afilter.h': No such file or directory  
  2. 1>Compiling - ndishook.c  
  3. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(9297) : error C1083: Cannot open include file: 'afilter.h': No such file or directory  
  4. 1>Compiling - vpcknt.c  
  5. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(9297) : error C1083: Cannot open include file: 'afilter.h': No such file or directory  
  6. 1>Compiling - ndishook.c  
  7. 1>1>d:\winddk\3790.1830\inc\ddk\w2k\ndis.h(9297) : error C1083: Cannot open include file: 'afilter.h': No such file or directory  


这个好解决,给它指定绝对路径嘛:

改进:(经网友xkjcf要求,把绝对路径去掉)

(3、2、1)在sources文件中添加INCLUDES路径;

路径总线:

A、我测试发现“工具->选项->项目解决方案->C++目录->包含里设置的INCLUDE路径”与sources路径下的INCLUDE路径有优先级,

其中sources中的路径有更高的优先级。

B、在sources文件中若是有多个路径,且都含有同一个.h文件,那么此时包哪一个是没有顺序关系的。

eg:

在此sources下有:

两个路径下都有usbdlib.h 文件,可是我在引用它时:

却引用到了后者的路径,以致报错:

 

(3、2、2)

再试看看,哈哈^-^ ,另人兴奋的.sys文件产生了。

 

二、源码分析

 

 1、DriverEntry

与我数的Windows驱动的入口开发一样,这里的DriverEntry没有什么特别的地方:

1、1 创建设备和设备SymbolicLink名字。

  1. ntStatus = IoCreateDevice( DriverObject,  
  2.         sizeof( DEVICE_EXTENSION ),  
  3.         & uszDriverString,  
  4.         FILE_DEVICE_UNKNOWN,  
  5.         0,  
  6.         FALSE,  
  7.         & pDeviceObject );  


 

  1. // Point uszDeviceString at the device name.  
  2.   
  3.     RtlInitUnicodeString( & uszDeviceString, L"\\DosDevices\\" SYSDRIVER_NAME_WIDE );  
  4.   
  5.     // Create symbolic link to the user-visible name.  
  6.   
  7.     ntStatus = IoCreateSymbolicLink( & uszDeviceString, & uszDriverString );  


创建SymbolicLink时名字为:\\DosDevices\\VPCKnt。

这里说明驱动中的SymbolicLink命名的一个特点:DosDevices的符号链接名就是??, 所以"\\DosDevices\\XXXX"其实就是"\\??\\XXXX"

  1. TEXT(<a>\\\\.\VPCKnt</a>),   

1、2   分配派遣函数

  1. DriverObject->DriverUnload                           = VPCKernelNtDriverUnload;  
  2. DriverObject->MajorFunction[ IRP_MJ_CREATE ]         = VPCKernelNtDispatchCreate;  
  3. DriverObject->MajorFunction[ IRP_MJ_CLOSE ]          = VPCKernelNtDispatchClose;  
  4. DriverObject->MajorFunction[ IRP_MJ_DEVICE_CONTROL ] = VPCKernelNtDispatchIoctl;  


以上函数是包含完成NDIS抓包的所有逻辑,我们重点分析。

1、3   PDEVICE_EXTENSION扩展附加及初始化

PDEVICE_EXTENSION为我们息定义的结构体,把它附加到设备象中,这样以后在派遣函数中我们就可以从扩展数据中获得我们自定义的数据

  1. typedef struct _DEVICE_EXTENSION   
  2. {   
  3.     // Boolean表明是否用户态的service/application已经创建文件对象  
  4.     ULONG           WasCreated;  
  5.     FAST_MUTEX      WasCreatedFastMutex;  
  6.   
  7.     // InterceptNDIS表明是否我们不得不去中断NDIS活动  
  8.     ULONG           InterceptNDIS;  
  9.   
  10.     // 关于NDIS Hook的状态信息  
  11.     PNT_PROTOCOL_LIST           NdisProtocolList;  
  12.     PNDISHOOK_HANDLER_STUB      NdisStubsList;  
  13.     DWORD                       NdisStubsNum;  
  14.   
  15.     // 客户端想要表现的Open Adapter的列表  
  16.     OALIST_ITEM*                OaList;  
  17.     DWORD                       OaListItems;  
  18.     NDIS_SPIN_LOCK              OaListSpinLock;  
  19.   
  20.     // 等待被接收的包列表  
  21.     NEXT_PACKET*                PacketsBuff;  
  22.     DWORD                       PacketsBuffMaxItems;  
  23.     DWORD                       PacketsBuffStart /* first allocated; -1 => empty */, PacketsBuffEnd /* first free */;  
  24.     DWORD                       PacketsLost;  
  25.     PKEVENT                     PacketsReadyEvent;  
  26.     NDIS_SPIN_LOCK              PacketsBuffSpinLock;  
  27.   
  28.     // Loopback trick stuff.  
  29.     DWORD       dwLtLoopbackAdapterIpAddress;   // tipically: 0xA9FE1981; zero if lt is disabled.  
  30.     DWORD       dwLtTrickNatIpAddress;          // tipically: 0xA9FE1980; zero if lt is disabled.  
  31.   
  32.     // Other.  
  33.     NDIS_SPIN_LOCK              ReceiveWorkItemSpinLock;  
  34.     DWORD                       ReceiveWorkItemIdCounter;  
  35.   
  36.     NDIS_SPIN_LOCK              PacketSerialCounterSpinLock;  
  37.     DWORD                       PacketSerialCounter;  
  38.   
  39. } DEVICE_EXTENSION, *PDEVICE_EXTENSION;  


 

  1. extension = pDeviceObject->DeviceExtension;  


 

初始化同步操作互斥体及其他逻辑变量:

  1. // Initialize WasCreated and its mutex.  
  2.     extension->WasCreated = 0;  
  3.     ExInitializeFastMutex( & extension->WasCreatedFastMutex );  
  4.   
  5.     // Initialize InterceptNDIS.  
  6.     extension->InterceptNDIS = 0;  
  7.   
  8.     // 初始化NDIS状态变量  
  9.     extension->NdisProtocolList = NULL;  
  10.     extension->NdisStubsList = NULL;  
  11.     extension->NdisStubsNum = 0;  
  12.   
  13.     // 初始化OaList事件  
  14.     extension->OaList = NULL;  
  15.     extension->OaListItems = 0;  
  16.     NdisAllocateSpinLock( & extension->OaListSpinLock );  
  17.   
  18.     // 初始化包Buffer  
  19.     extension->PacketsBuff = NULL;  
  20.     extension->PacketsBuffMaxItems = 0;  
  21.     extension->PacketsBuffStart = -1;  
  22.     extension->PacketsBuffEnd = 0;  
  23.     extension->PacketsLost = 0;  
  24.     extension->PacketsReadyEvent = NULL;  
  25.   
  26.     NdisAllocateSpinLock( & extension->PacketsBuffSpinLock );  
  27.   
  28.     // 初始化其他res....  
  29.     NdisAllocateSpinLock( & extension->ReceiveWorkItemSpinLock );  
  30.     extension->ReceiveWorkItemIdCounter = 0;  
  31.   
  32.     NdisAllocateSpinLock( & extension->PacketSerialCounterSpinLock );  
  33.     extension->PacketSerialCounter = 0;  
  34.   
  35.     // Initialize the lt stuff.  
  36.     extension->dwLtLoopbackAdapterIpAddress = 0;  
  37.     extension->dwLtTrickNatIpAddress = 0;  

 

2、IRP_MJ_CREATE派遣函数VPCKernelNtDispatchCreate

此函数相对简单,只是根据DeviceExtension中自定义的数据来做“仅允许一个文件对象的创建”的处理而已。

3、 IRP_MJ_CLOSE派遣函数VPCKernelNtDispatchClose

标识客户端关闭了后去做清理工作:

(1)擦除已经分配的打开的adapters;

  1. ReleaseOaList ();  

(2)停止中断;

  1. extension->InterceptNDIS = 0;  

(3)Unhook NDIS Handler functions;

Unhook NDIS其实就是重新装载原始的句柄指针,在Unhook NDIS 的前后都调用KeDelayExecutionThread让CPU停下来半秒。

关于KeDelayExecutionThread请参考我的文章: <<DDK下的Sleep函数KeDelayExecutionThread>>

  1. KeDelayExecutionThread( UserMode, FALSE, & liWaitElapse );  
  2.   
  3.         // Unhook NDIS Handler functions.  
  4.         if ( extension->NdisProtocolList )  
  5.             UnhookInstalledProtocols(  
  6.                 (PNT_PROTOCOL) ( (BYTE*) extension->NdisProtocolList + sizeof( NT_PROTOCOL_LIST ) ),  
  7.                 extension->NdisProtocolList->dwProtocolsNum,  
  8.                 (PNT_OPEN_ADAPTER) ( (BYTE*) extension->NdisProtocolList + sizeof( NT_PROTOCOL_LIST ) + extension->NdisProtocolList->dwProtocolsNum * sizeof( NT_PROTOCOL ) ),  
  9.                 extension->NdisProtocolList->dwOpenAdaptersNum );  
  10.   
  11.         KeDelayExecutionThread( UserMode, FALSE, & liWaitElapse );  

以上UnhookIntalledProtocols函数的参数大家看得有点晕了,其实这是作者对extension->NdisProtocoList的一个数据结构定义罢了:

  1. typedef struct _NT_PROTOCOL_LIST  
  2. {  
  3.     // ...Info...  
  4.     DWORD               dwProtocolsNum;  
  5.     DWORD               dwOpenAdaptersNum;  
  6.   
  7.     // ...Data...  
  8.     //  
  9.     // x List of NT_PROTOCOL(s) follows.  
  10.     // x List of NT_OPEN_ADAPTER(s) follows.  
  11.     //  
  12. } NT_PROTOCOL_LIST, *PNT_PROTOCOL_LIST;  

它的最前面是存着协议和Adapter数据的个数信息,后面分别是存着协议数组和Adapter数据。

 

  1. NTSTATUS UnhookInstalledProtocols( PNT_PROTOCOL pntpProtocols, DWORD dwProtocolsSize, PNT_OPEN_ADAPTER pntoaAdapters, DWORD dwAdaptersSize )  
  2. {  
  3.     DWORD               i;  
  4.     PNT_OPEN_ADAPTER    pntoaThisAdapter;  
  5.   
  6.     // 重载原始的句柄指针.  
  7.     for ( i=0; i<dwAdaptersSize; i++ )  
  8.     {  
  9.         pntoaThisAdapter = & pntoaAdapters[ i ];  
  10.   
  11.         // Restore the Original Pointers.  
  12.         if ( pntoaThisAdapter->Original_SendHandler )  
  13.             RestoreHandlerPointerSecure( & pntoaThisAdapter->pobBlockPtr->SendHandler, pntoaThisAdapter->Original_SendHandler, pntoaThisAdapter->Stub_SendHandler );  
  14.         if ( pntoaThisAdapter->Original_ReceiveHandler )  
  15.             RestoreHandlerPointerSecure( & pntoaThisAdapter->pobBlockPtr->ReceiveHandler, pntoaThisAdapter->Original_ReceiveHandler, pntoaThisAdapter->Stub_ReceiveHandler );  
  16.         if ( pntoaThisAdapter->Original_PostNt31ReceiveHandler )  
  17.             RestoreHandlerPointerSecure( & pntoaThisAdapter->pobBlockPtr->PostNt31ReceiveHandler, pntoaThisAdapter->Original_PostNt31ReceiveHandler, pntoaThisAdapter->Stub_PostNt31ReceiveHandler );  
  18.         if ( pntoaThisAdapter->Original_SendPacketsHandler )  
  19.             RestoreHandlerPointerSecure( & pntoaThisAdapter->pobBlockPtr->SendPacketsHandler, pntoaThisAdapter->Original_SendPacketsHandler, pntoaThisAdapter->Stub_SendPacketsHandler );  
  20.         if ( pntoaThisAdapter->Original_TransferDataHandler )  
  21.             RestoreHandlerPointerSecure( & pntoaThisAdapter->pobBlockPtr->TransferDataHandler, pntoaThisAdapter->Original_TransferDataHandler, pntoaThisAdapter->Stub_TransferDataHandler );  
  22.         if ( pntoaThisAdapter->Original_ReceivePacketHandler )  
  23.             RestoreHandlerPointerSecure( & pntoaThisAdapter->pobBlockPtr->ReceivePacketHandler, pntoaThisAdapter->Original_ReceivePacketHandler, pntoaThisAdapter->Stub_ReceivePacketHandler );  
  24.         if ( pntoaThisAdapter->Original_TDCompleteHandler && pntoaThisAdapter->pmbMiniportPtr )  
  25.             RestoreHandlerPointerSecure( & pntoaThisAdapter->pmbMiniportPtr->TDCompleteHandler, pntoaThisAdapter->Original_TDCompleteHandler, pntoaThisAdapter->Stub_TDCompleteHandler );  
  26.     }  
  27.   
  28.     // Return to the caller.  
  29.     return STATUS_SUCCESS;  
  30. }  

 

(4)释放内存;

(4、1)释放协议数组中每个成员中的pbWorkItemHeader

  1. static VOID ReleaseReceiveWorkItems ()  
  2. {  
  3.     PDEVICE_EXTENSION   extension = g_pDeviceObject->DeviceExtension;  
  4.     int                 i, c;  
  5.     NT_PROTOCOL*        pBase = NULL;  
  6.     NT_PROTOCOL*        pThis = NULL;  
  7.   
  8.     if ( extension->NdisProtocolList == NULL ||  
  9.         extension->NdisProtocolList->dwProtocolsNum == 0 )  
  10.     {  
  11.         return;  
  12.     }  
  13.   
  14.     pBase = (NT_PROTOCOL*) ( (BYTE*) extension->NdisProtocolList + sizeof( NT_PROTOCOL_LIST ) );  
  15.     NdisAcquireSpinLock( & extension->ReceiveWorkItemSpinLock );  
  16.         c = extension->NdisProtocolList->dwProtocolsNum;  
  17.         for ( i=0; i<c; i ++ )  
  18.         {  
  19.             pThis = & pBase[ i ];  
  20.   
  21.             if ( pThis->pbWorkItemHeader )  
  22.             {  
  23.                 // free...  
  24.                 ExFreePool( pThis->pbWorkItemHeader );  
  25.                 pThis->pbWorkItemHeader = NULL;  
  26.                 pThis->uiWorkItemHeaderSize = 0;  
  27.             }  
  28.         }  
  29.     NdisReleaseSpinLock( & extension->ReceiveWorkItemSpinLock );  
  30. }  


(4、2)释放包Buffer

  1. static VOID ReleasePackets ()  


 

5、IRP_MJ_DEVICE_CONTROL派遣函数VPCKernelNtDispatchIoctl

此函数主要主是一个switch ..... case 语句,case了应用程序对函数的的四种IOCTROL码。

(5、1)IOCTL_VPCKNT_GET_VERSION

因为Irp->AssociatedIrp.SystemBuffer指向的Buffer就是应用层得到的Out Buffer,所以把版本号写到这块内存就对了。

(5、2)IOCTL_VPCKNT_INITIALIZE_HOOK

(5、2、1)分配包Buffer

分配的包的最大个数,由上层决定,默认是32。

  1. extension->PacketsBuff = (NEXT_PACKET*) ExAllocatePool( NonPagedPool,  
  2.                         ihiHookInput.dwPacketsBuffMaxItems * sizeof( NEXT_PACKET ) );  
  3.   
  4. if ( extension->PacketsBuff )  
  5. {  
  6.     extension->PacketsBuffMaxItems = ihiHookInput.dwPacketsBuffMaxItems;  
  7.     memset( extension->PacketsBuff, 0, ihiHookInput.dwPacketsBuffMaxItems * sizeof( NEXT_PACKET ) );  
  8. }  

(5、2、2)获得包准备好事件

调用ObReferenceObjectByHandle,由应用层的hNotificationEvent得到对应的PKEVENT。(原理参考:<<Window XP驱动开发(二十二) 驱动程序的同步处理>>)

  1. if ( ihiHookInput.hNotificationEvent )  
  2.                 {  
  3.                     NTSTATUS        ntEvRes = ObReferenceObjectByHandle( ihiHookInput.hNotificationEvent,  
  4.                         0, (POBJECT_TYPE) NULL, UserMode, (PVOID) & extension->PacketsReadyEvent, (POBJECT_HANDLE_INFORMATION) NULL);  
  5.   
  6.                     if ( ntEvRes != STATUS_SUCCESS )  
  7.                         extension->PacketsReadyEvent = NULL;  
  8.                 }  

(5、2、3)注册协议并安装协议

 关于这部分的内容特别重要,我们将在“6、核心源码”中分析。

  1. __try  
  2.                     {  
  3.                         nsRegRes = RegisterFakeProtocol( & hFakeProtocolHandle, SYSDRIVER_NAME_ANSI );  
  4.   
  5.                         if ( nsRegRes == STATUS_SUCCESS )  
  6.                         {  
  7.                             nsHookRes = HookInstalledProtocols(  
  8.                                 & pplProtocolList, & pnhhsStubsList, & dwStubsNum,  
  9.                                 hFakeProtocolHandle );  
  10.                         }  
  11.                     }  
  12.                     __except ( EXCEPTION_EXECUTE_HANDLER )  
  13.                     {  
  14.                     }  

安装后得到Ndis Hook的结果,即下图的数据结构:

(5、2、4)把Ndis Hook的结果返回到应用程序的Out Buffer中;

  1. pbOutputBufferPayload = (BYTE*) Irp->AssociatedIrp.SystemBuffer + sizeof( ihoHookOutput );  
  2.                     // Inform about the Ndis Hook results.  
  3.                     if ( nsHookRes == STATUS_SUCCESS ) // NDIS Hook Ok.  
  4.                     {  
  5.                         // Info.  
  6.                         ihoHookOutput.bNdisHookSucceeded = 1;  
  7.                         ihoHookOutput.dwProtocolListBufferSize = sizeof( NT_PROTOCOL_LIST ) +  
  8.                             pplProtocolList->dwProtocolsNum * sizeof( NT_PROTOCOL ) +  
  9.                             pplProtocolList->dwOpenAdaptersNum * sizeof( NT_OPEN_ADAPTER );  
  10.   
  11.                         // Data.  
  12.   
  13.                         memcpy( pbOutputBufferPayload, pplProtocolList, ihoHookOutput.dwProtocolListBufferSize );  
  14.                         pbOutputBufferPayload += ihoHookOutput.dwProtocolListBufferSize;  
  15.                     }  
  16.                     else // NDIS Hook Failed.  
  17.                     {  
  18.                         // Info.  
  19.   
  20.                         ihoHookOutput.bNdisHookSucceeded = 0;  
  21.                         ihoHookOutput.dwProtocolListBufferSize = 0;  
  22.                     }  
  23.   
  24.                     // Copy the Header structure.  
  25.                     * (INITIALIZE_HOOK_OUTPUT*) Irp->AssociatedIrp.SystemBuffer = ihoHookOutput;  

 

(5、3)IOCTL_VPCKNT_SUBMIT_OALIST

(5、4)IOCTL_VPCKNT_GET_NEXT_PACKET

 

 6、核心源码分析

(6、1) 注册协议

注册的PROTOCOL_CHARACTERISTICS为NDIS 4.0版本的。

  1. NDIS40_PROTOCOL_CHARACTERISTICS     ndis40pcFakeProtCharacts;  

在初始化ndis40pcFakeProtCharacts结构体时给它预留了15个自定义的函数,根据实现需要可完成这15个函数的定义。

  1. ndis40pcFakeProtCharacts.OpenAdapterCompleteHandler = & FakeProtocol_OpenAdapterComplete;  
  2.     ndis40pcFakeProtCharacts.CloseAdapterCompleteHandler = & FakeProtocol_CloseAdapterComplete;  
  3.     ndis40pcFakeProtCharacts.SendCompleteHandler = & FakeProtocol_SendComplete;  
  4.     ndis40pcFakeProtCharacts.TransferDataCompleteHandler = & FakeProtocol_TransferDataComplete;  
  5.     ndis40pcFakeProtCharacts.ResetCompleteHandler = & FakeProtocol_ResetComplete;  
  6.     ndis40pcFakeProtCharacts.RequestCompleteHandler = & FakeProtocol_RequestComplete;  
  7.     ndis40pcFakeProtCharacts.ReceiveHandler = & FakeProtocol_Receive;  
  8.     ndis40pcFakeProtCharacts.ReceiveCompleteHandler = & FakeProtocol_ReceiveComplete;  
  9.     ndis40pcFakeProtCharacts.StatusHandler = & FakeProtocol_Status;  
  10.     ndis40pcFakeProtCharacts.StatusCompleteHandler = & FakeProtocol_StatusComplete;  
  11.   
  12.     NdisInitializeString( & ndis40pcFakeProtCharacts.Name, pszProtocolName );  
  13.   
  14.     ndis40pcFakeProtCharacts.ReceivePacketHandler = & FakeProtocol_ReceivePacket;  
  15.     ndis40pcFakeProtCharacts.BindAdapterHandler = & FakeProtocol_BindAdapter;  
  16.     ndis40pcFakeProtCharacts.UnbindAdapterHandler = & FakeProtocol_UnbindAdapter;  
  17.     ndis40pcFakeProtCharacts.PnPEventHandler = & FakeProtocol_PnpEvent;  
  18.     ndis40pcFakeProtCharacts.UnloadHandler = & FakeProtocol_UnloadProtocol;  

然后调用NdisRegisterProtocol注册。

(6、2)安装协议

(6、2、1)分配要求的内存

包括PNT_PROTOCOL(1M,作为协议数组)、PNT_OPEN_ADAPTER(1M,作为协议Adpater 的数组)、PNDISHOOK_HANDLER_STUB(128K,用于Hook句柄的存根)

 (6、2、2)循环访问列表协议,获得已安装的列表协议句柄;

循环获得协议的方法是调用GetNextProtocol函数,此函数根据NDIS50_PROTOCOL_BLOCK结构体中的NextProtocol 来访问下一个协议,能这样做的原因是因为协议之间是一条单链表。

这里把NDIS_HANDLE 强转为NDIS50_PROTOCOL_BLOCK类型,再次说明了两者其实是同一结构的。

  1. NDIS_HANDLE GetNextProtocol( IN NDIS_HANDLE hProtocolHandle )  
  2. {  
  3.     DWORD                       dwNdisVersion;  
  4.     NDIS50_PROTOCOL_BLOCK*      pn50pbProtBlockPtr;  
  5.   
  6.     // Return the Next Protocol in the Linked List.  
  7.     dwNdisVersion = GetNdisVersion ();  
  8.   
  9.     if ( dwNdisVersion == 0xFFFFFFFF )  
  10.         return NULL;  
  11.   
  12.     switch( dwNdisVersion )  
  13.     {  
  14.     case 0x00050000:  
  15.     case 0x00050001:  
  16.   
  17.         // Windows 2000 / Windows XP NDIS Version.  
  18.   
  19.         pn50pbProtBlockPtr = (NDIS50_PROTOCOL_BLOCK*) hProtocolHandle;  
  20.   
  21.         return (NDIS_HANDLE) pn50pbProtBlockPtr->NextProtocol;  
  22.   
  23.     default:  
  24.   
  25.         // Unrecognized NDIS Version. Exit.  
  26.   
  27.         return NULL;  
  28.     }  
  29. }  

(6、2、3)根据列表协议句柄,获得对应的Characteristics数据结构。

  1. NDIS50_PROTOCOL_CHARACTERISTICS* GetProtocolCharacteristics( IN NDIS_HANDLE hProtocolHandle )  
  2. {  
  3.     DWORD                       dwNdisVersion;  
  4.     NDIS50_PROTOCOL_BLOCK*      pn50pbProtBlockPtr;  
  5.   
  6.     // Return the Protocol Characteristics information for this protocol.  
  7.     dwNdisVersion = GetNdisVersion ();  
  8.     if ( dwNdisVersion == 0xFFFFFFFF )  
  9.         return NULL;  
  10.   
  11.     switch( dwNdisVersion )  
  12.     {  
  13.     case 0x00050000:  
  14.     case 0x00050001:  
  15.         // Windows 2000 / Windows XP NDIS Version.  
  16.         pn50pbProtBlockPtr = (NDIS50_PROTOCOL_BLOCK*) hProtocolHandle;  
  17.         return & pn50pbProtBlockPtr->ProtocolCharacteristics;  
  18.     default:  
  19.         // Unrecognized NDIS Version. Exit.  
  20.   
  21.         return NULL;  
  22.     }  
  23. }  

从以上两步的结果中提取信息进行编号保存在协议数组中

(6、2、4)从协议句柄中获得协议中打开的Adapter。

从结果中提取信息进行编号保存在协议Adapter数组中

综合(6、2、3)(6、2、4)我们得到了协议、Characteristics及协议Adapter的关系图。

 (6、2、5)Hook处理各种Handle

(6、2、5、1)Hook 发送句柄

首先判断原有的SendHandler是否可以Hook:

其实原理很简单,因为Hook的句柄我们会存在一个区域内,通过判断是否在这个区域内能判断是不是Hook过。

  1. static BOOLEAN CanHook( IN PVOID fnptr )  
  2. {  
  3.     if ( fnptr == NULL || g_pbCanHookRefMemStart == NULL || g_pbCanHookRefMemEnd == NULL ) // # pointer problem...  
  4.         return FALSE;  
  5.     else if ( ((BYTE*)fnptr) >= g_pbCanHookRefMemStart && ((BYTE*)fnptr) <= g_pbCanHookRefMemEnd ) // # already hooked...  
  6.         return FALSE;  
  7.     else // # ok, can hook...  
  8.         return TRUE;  
  9. }  

然后填充存根代码:

bPushImm32Opcode、dwOperationID、bJmpRel32Opcode分别填充固定的码值,这没什么可讲;

但dwJmpDisplacement值的得到有点费解(若有知道的网友可否告诉我一下)

  1. static void FillStubCode( PNDISHOOK_HANDLER_STUB pnhhsThisStub, DWORD dwThisAdapterOrd, DWORD dwAddInfo )  
  2. {  
  3.     // Fill the Stub structure.  
  4.   
  5.     pnhhsThisStub->bPushImm32Opcode = 0x68;  
  6.     pnhhsThisStub->dwOperationID = ( dwThisAdapterOrd << 0x10 ) | dwAddInfo;  
  7.     pnhhsThisStub->bJmpRel32Opcode = 0xE9;  
  8.     pnhhsThisStub->dwJmpDisplacement =  
  9.         (DWORD) & HandlerGeneralDispatcher -  
  10.         ( (DWORD) pnhhsThisStub + FIELD_OFFSET( NDISHOOK_HANDLER_STUB, bJmpRel32Opcode ) ) -  
  11.         5;  
  12. }  

!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!(重点中的重点,改日讲解)

 

  1. static PVOID                g_vpvHandlersVector[] =  
  2.                                 {  
  3.                                     & New_SendHandler,              // SENDHANDLER_FNID  
  4.                                     & New_ReceiveHandler,           // RECEIVEHANDLER_FNID  
  5.                                     & New_PostNt31ReceiveHandler,   // POSTNT31RECEIVEHANDLER_FNID  
  6.                                     & New_SendPacketsHandler,       // SENDPACKETSHANDLER_FNID  
  7.                                     & New_TransferDataHandler,      // TRANSFERDATAHANDLER_FNID  
  8.                                     & New_ReceivePacketHandler,     // RECEIVEPACKETHANDLER_FNID  
  9.                                     & New_TDCompleteHandler         // TDCOMPLETEHANDLER_FNID  
  10.                                 };  


  1. static void __declspec(naked) HandlerGeneralDispatcher ( void )  
  2. {  
  3.     __asm  
  4.     {  
  5.         // ### Load in EAX and ECX the Code found in the Stack. ###  
  6.   
  7.         mov     eax, dword ptr[ esp ]   // eax = Code: 0xYYYYZZZZ. (y = Adapter Ordinal, z = Handler Ordinal.)  
  8.         mov     ecx, eax                // ecx = Code: 0xYYYYZZZZ.  
  9.   
  10.         shr     eax, 0x10               // eax = 0x0000YYYY  
  11.         and     ecx, 0xFFFF             // ecx = 0x0000ZZZZ  
  12.   
  13.         mov     dword ptr[ esp ], eax   // Updates Code in Stack: 0x0000YYYY  
  14.   
  15.         // ### Call the appropriate Handler based on the Handler Ordinal. ###  
  16.   
  17.         lea     eax, [ offset g_vpvHandlersVector + ecx * 4 ];  
  18.         call    dword ptr[ eax ]  
  19.   
  20.         // ### Pop in ECX the "HookPrivateStorage" parameter. ###  
  21.   
  22.         pop     ecx  
  23.   
  24.         // ### Check whether we have to call the Original Handler or we have to return control to the Caller. ###  
  25.   
  26.         cmp     ecx, 0x1000  
  27.         ja      callOriginalHandler  
  28.   
  29.         // ### Return the Control to the Caller. ###  
  30.   
  31.         pop     edx  
  32.         add     esp, ecx  
  33.         jmp     edx  
  34.   
  35. callOriginalHandler:  
  36.   
  37.         // ### Call the Original Handler. ###  
  38.   
  39.         jmp     ecx  
  40.     }  
  41. }  

 

最后将记录原有的发送句柄后(目的是在UnhookInstalledProtocols时能重载原始句柄指指),将自己的发送Handle附加到原有的发送句柄上。

  1. pntoaThisAdapter->Original_SendHandler = pobAdapter->SendHandler;  
  2.                             pntoaThisAdapter->Stub_SendHandler = pnhhsThisStub;  
  3.                             pobAdapter->SendHandler = (SEND_HANDLER) pnhhsThisStub;  

这样pobAdapter->SendHandler  就会调用我们自定义的New_SendHandler函数了。关于它的处理我们在“7、   自定义处理函数讲解”中进行讲解。

 

(6、2、5、2)Hook 接收句柄

与Hook 发送句柄类似。

(6、2、5、3)Hook postnt31接收句柄

与Hook 发送句柄类似。

(6、2、5、4)Hook 发送包句柄

与Hook 发送句柄类似。

(6、2、5、5)Hook 传送数据句柄

与Hook 发送句柄类似。

(6、2、5、6)Hook 接收包句柄

与Hook 发送句柄类似。

(6、2、5、6)Hook 传送数据完成句柄

与Hook 发送句柄类似。

 

 7、自定义处理函数的讲解

(7、1)New_SendHandler

 以入参HookPrivateStorage来标识哪一协议,哪一Adapter

如果判断如果需要中断NDIS请求,那么就调用Intercept_SendHandler将此数据包进行截获;

截获完成后,根据返回结果判断是否要将原始的处句柄返回回去(默认是需要将原始句柄返回),这样我们截获了包后此包能再次发送出去

  1. NDIS_STATUS __cdecl New_SendHandler( IN OUT DWORD HookPrivateStorage, IN OUT DWORD CallingFnRetAddress, IN OUT NDIS_HANDLE MacBindingHandle, IN OUT PNDIS_PACKET Packet )  
  2. {  
  3.     NDIS_STATUS             nsRetStatus = NDIS_STATUS_SUCCESS;  
  4.     SEND_HANDLER            pfnOriginal = NULL;  
  5.     DWORD                   OpenAdapterId = HookPrivateStorage;  
  6.     PDEVICE_EXTENSION       pdeExtension = g_pDeviceObject->DeviceExtension;  
  7.     BOOLEAN                 bExecuteOriginalHandler = TRUE;  
  8.   
  9.     // Get a pointer to the Original Handler.  
  10.     pfnOriginal = g_pntoaAdapters[ OpenAdapterId ].Original_SendHandler;  
  11.   
  12.     // Check if we have to intercept the NDIS Request.  
  13.     if ( pdeExtension->InterceptNDIS )  
  14.         bExecuteOriginalHandler = Intercept_SendHandler(  
  15.             & g_pntpProtocols[ g_pntoaAdapters[ OpenAdapterId ].dwProtocolOrd ], & g_pntoaAdapters[ OpenAdapterId ],  
  16.             & nsRetStatus, & MacBindingHandle, & Packet );  
  17.   
  18.     // Make call the Original Handler and return.  
  19.   
  20.     if ( bExecuteOriginalHandler )  
  21.         HookPrivateStorage = (DWORD) pfnOriginal; // Original Handler.  
  22.     else  
  23.         HookPrivateStorage = 2 * 0x4; // Passed Parameters Size.  
  24.   
  25.     return nsRetStatus;  
  26. }  


(7、1、1)在Intercept_SendHandler里先判断此Adapter是不是要被处理:

  1. // process.  
  2.     if ( HaveToProcess( pnoaAdapter ) )  
  3.     {  


判断原理是判断它是不是在OaList列表中。

(7、1、2)如果判断此Adapter是需要处理的那么就调用NdisPacket2MemoryRegion为此包分配分内存。

首先调用系统API NdisQueryPacket查询此PNDIS_PACKET包的PNDIS_BUFFER和总共的内存大小

  1. NdisQueryPacket( Packet,  
  2.         NULL, NULL,  
  3.         & pnbCurrent,  
  4.         & uiTotalSize );  

然后根据PNDIS_PACKET包的内存大小及我们要添加的头的大小算出自定义包所需要的内存大小:

  1. ulMemSize = ulHeaderBufferSize + uiTotalSize;  
  2.     if ( ulMemSize == 0 )  
  3.         return NULL;  
  4.   
  5.     pbMem = (BYTE*) ExAllocatePool( NonPagedPool, ulMemSize );  
  6.     if ( pbMem == NULL )  
  7.         return NULL;  

最后把输入的包的BUFFER复制到我们自定义的内存中。

  1. while( pnbCurrent )  
  2.     {  
  3.         NdisQueryBuffer( pnbCurrent, & pvPtr, & uiPtrDim );  
  4.   
  5.         if ( pvPtr == NULL || uiCount + uiPtrDim > uiTotalSize )  
  6.         {  
  7.             ExFreePool( pbMem );  
  8.             return NULL;  
  9.         }  
  10.         else  
  11.         {  
  12.             memcpy( pbMem + uiCount, pvPtr, uiPtrDim );  
  13.             uiCount += uiPtrDim;  
  14.         }  
  15.   
  16.         NdisGetNextBuffer( pnbCurrent, & pnbCurrent );  
  17.     }  


(7、1、3)把从(7、1、2)中得到的自定义数据内存保存到extern的包队列中,并通过事件告知应用程序可以取包了。

  1. VOID QueuePacket( IN PNT_OPEN_ADAPTER pnoaAdapter, IN BYTE* pbData, IN DWORD dwDataSize, BYTE bDirection, DWORD dwSerial )  
  2. {  
  3.     PDEVICE_EXTENSION       extension = g_pDeviceObject->DeviceExtension;  
  4.   
  5.     // add.  
  6.   
  7.     NdisAcquireSpinLock( & extension->PacketsBuffSpinLock );  
  8.   
  9.         if ( extension->PacketsBuff && extension->PacketsBuffMaxItems )  
  10.         {  
  11.             NEXT_PACKET*        pThis = & extension->PacketsBuff[ extension->PacketsBuffEnd ];  
  12.   
  13.             //  
  14.             // manage the circular buffer.  
  15.             //  
  16.   
  17.             // compare end and start.  
  18.   
  19.             if ( extension->PacketsBuffEnd == extension->PacketsBuffStart )  
  20.             {  
  21.                 // ### free ###  
  22.                 if ( pThis->pbData )  
  23.                     ExFreePool( pThis->pbData );  
  24.   
  25.                 // ### inc counter ###  
  26.                 extension->PacketsLost ++;  
  27.   
  28.                 // ### inc start pos ###  
  29.                 extension->PacketsBuffStart ++;  
  30.                 if ( extension->PacketsBuffStart == extension->PacketsBuffMaxItems )  
  31.                     extension->PacketsBuffStart = 0;  
  32.             }  
  33.   
  34.             // inc end pos.  
  35.   
  36.             extension->PacketsBuffEnd ++;  
  37.             if ( extension->PacketsBuffEnd == extension->PacketsBuffMaxItems )  
  38.                 extension->PacketsBuffEnd = 0;  
  39.   
  40.             if ( extension->PacketsBuffStart == -1 )  
  41.                 extension->PacketsBuffStart = 0;  
  42.   
  43.             //  
  44.             // fill this one.  
  45.             //  
  46.   
  47.             pThis->dwOpenAdapterOrdinal = pnoaAdapter->dwOrdinal;  
  48.   
  49.             pThis->dwPacketSerial = dwSerial;  
  50.   
  51.             pThis->pbData = pbData;  
  52.             pThis->dwDataLength = dwDataSize;  
  53.   
  54.             pThis->bDirection = bDirection;  
  55.   
  56.             //  
  57.             // set the user event.  
  58.             //  
  59.   
  60.             if ( extension->PacketsReadyEvent )  
  61.                 KeSetEvent( extension->PacketsReadyEvent, 0, FALSE );  
  62.         }  
  63.   
  64.     NdisReleaseSpinLock( & extension->PacketsBuffSpinLock );  
  65.   
  66.     // return.  
  67.   
  68.     return;  
  69. }  


(7、1、4)如果是ARP或IP协议那么发送包回到协议

  1. VOID SendPacketHook( IN PNT_PROTOCOL pnpProtocol, IN PNT_OPEN_ADAPTER pnoaAdapter, IN BYTE* pbData, IN DWORD dwDataSize )  


 


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值