基于PassThru的NDIS中间层驱动程序扩展

来源:互联网 发布:linux python pyqt5 编辑:程序博客网 时间:2024/06/05 21:04

概要:开发一个NDIS驱动是一项相对复杂的工作,这一方面是由于核心驱动本身有更多的限制和要求,有更多的“游戏规则”要求开发者理解和掌握,NDIS 的复杂性把难度更是提高了,本文以PassThru为例,加上自己的理解,讲述了NDIS驱动的处理过程和在PassThru的基础上进行扩展的基本方法,本文并不是一个入门读物,所以没有提及任何核心驱动开发的相关知识,本文主要讲述的是NDIS中间层对数据包处理的流程。在阅读过程中,关于相关 API的用法,或其它信息,请参看DDK文档。

一  NDIS驱动程序分类.
    NDIS(Network Driver Interface Specification)是Windows网络驱动程序接口标准,NDIS驱动程序分为三类:
1. NDIS Miniport NIC Driver: 底层的微端口NIC驱动,这就是网络设备的物理的驱动程序了。
2. NDIS Protocol Driver: 高层的协议驱动,用来实现某个具体的协议栈,如TCP/IP协议栈,
   并向上导出TDI接口。
3. NDIS Intermediate Driver: 中间层驱动,位于Miniport Driver和Protocol Driver之间。

二  NDIS驱动结构简介.
NDIS驱动结构

其中,最上层是一个NDIS Protocol Driver,它向上提供一个Transport Driver Interface(TDI),向下通过NDIS接口与下面的NDIS中间层的上边界交互,NDIS中间层的下边界通过NDIS接口与下层的NDIS Miniport Driver交互。最后,由NDIS Miniport Driver利用NDIS接口与物理网络设备NetCard交互。NetCard是由不同的网卡设备产商提供的,而NDIS接口库是由Microsoft 开发好的,为什么NDIS Miniport Driver不是直接与物理网卡交互,而是通过NDIS接口与下物理网卡交互呢?(我想很多人都会和我当初一样,有这个疑问)。
    事实上,这是由于Windows系统为了提高可移植性,而设计出一个硬件抽象层(HAL),硬件抽象层在内部处理不同的硬件之间的差异,并且暴露出一个统一的接口给核心驱动开发者。例如:在Intel构架的系统中,内存和外部设备的端口采用分别编址,如果要从某个外部设备的端口上读写数据的话,可能要通过专用指令 IN/OUT读写,而在Alpha构架的系统上,采用的是统一编址的方式,所以对外部设备的IO端口进行读写的话还是通过访问内存的指令,HAL提供一组服务支持函数,如果要访问外部设备上的端口数据可以使用READ_PORT_UCHAR/WRITE_PORT_UCHAR等等,核心驱动开发者不用去考虑不同硬件构架的之间的差异。在NDIS Miniport Driver中,NetCard驱动的程序,正是这样通过NDIS接口提供的一组类似功能的函数,与物理的网络设备进行交互。其中,最上层是一个NDIS Protocol Driver,它向上提供一个Transport Driver Interface(TDI),向下通过NDIS接口与下面的NDIS中间层的上边界交互,NDIS中间层的下边界通过NDIS接口与下层的NDIS Miniport Driver交互。最后,由NDIS Miniport Driver利用NDIS接口与物理网络设备NetCard交互。

三 NDIS驱动程序的数据处理流程

 NDIS驱动程序的数据处理流程
3.1 三种NDIS驱动程序的关系。                   
    通常一个NDIS Protocol Driver 的上边沿导出TDI接口,并在其下边沿向NDIS注册一组Protocolxxx操作例程;一个NDIS Miniport Driver则在其下边沿通过NDIS接口操作物理网络设备,并在其上边沿向NDIS注册一组Miniportxxx操作例程。当一个中间层介入的时候,必需遵守这个规则,因此,中间层驱动对上层来说,扮演一个Miniport Driver的角色,它在上边沿向NDIS注册一组Miniportxxx函数;对于下层Miniport Driver来说,中间层驱动扮演一个Protocol Driver的角色,因此它在下边沿向NDIS注册一组Protocolxxx函数。Miniport Driver通过调用NdisMRegisterMiniport向NDIS注册一组MiniportXxx函数。其中原型如下:
    NDIS_STATUS NdisMRegisterMiniport(
        IN NDIS_HANDLE  NdisWrapperHandle,
         IN PNDIS_MINIPORT_CHARACTERISTICS  MiniportCharacteristics,
         IN UINT  CharacteristicsLength
         );
    其中,NdisWrapperHandls是之前通过调用NdisMInitializeWrapper取得的句柄 MiniportCharacteristics包含一组MiniportXxx函数指针。

    Protocol Driver 通过调用 NdisRegisterProtocol向NDIS注册一组ProtocolXxx函数。其中原型如下:
    VOID NdisRegisterProtocol(
        OUT PNDIS_STATUS  Status,
        OUT PNDIS_HANDLE  NdisProtocolHandle,
        IN PNDIS_PROTOCOL_CHARACTERISTICS  ProtocolCharacteristics,
        IN UINT  CharacteristicsLength
        );
    其中,ProtocolCharacteristics包含一组ProtocolXxx函数。由于NDIS Intermediate Driver的双重性,它需要调用 NdisIMRegisterLayeredMiniport 向NDIS注册,并向上层导出一组MiniportXxx函数,之后,调用NdisRegisterProtocol向NIDS注册,并向下层导出一组 ProtocolXxx函数。当一个NDIS中间层介入后,如图二所示。

3.2 NDIS数据发送流程:
    当上层协议驱动要发数据时,调用NdisSend/NdisSendPackets请求NDIS发送数据包,NDIS将会调用紧接其下的中间层驱动的 MiniportSend/MiniportSendPackets,中间层驱动MiniportSend/MiniportSendPackets 有机会在这里对包进行必要的操作,然后,中间层驱动再次调用NdisSend/NdisSendPackets请NDIS发送数据包,NDIS将调用其下层的Miniport Driver的MiniportSend/MiniportSendPackets,底层MiniportSend /MiniportSendPackets通过NDIS接口控制物理网络设备,将数据发送出去。在上层请求发送数据包时,上层分配了相关的资源(如内存),希望在下层完成发送动作后,能够及时的收回相关的资源,所以,当上层调用NdisSend/NdisSendPackets返回 NDIS_STATUS_PENDING以外的任何值时,上层就可以释放资源了,如果得到返回的结果是NDIS_STATUS_PENDING话,说明下层还没有完成发送请求,以后,等下层最终完成发送请求时,下层调用NdisMSendComplete请求NDIS通知上层可以释放资源了,于是NDIS 调用上层注册的ProtocolSendComplete函数,上层在这它的ProtocolSendComplete中释放了资源后,再次调用 NdisMSendComplete请求NDIS通知更上层。

3.3 NDIS数据接收流程:
    当底层网络设备有数据到来的时候,将触发中断,相应的中断处理程序接管中断后,将可能调用Miniport Driver所注册的中断处理例程(ISR),Miniport Driver通常在这里把网卡上的数据考贝到Miniport Driver缓冲区队列中去,出于效率的考虑,Miniport Driver这时可能不会立即通知上层处理新的数据,因为很可能,马上还有随后的新的数据到来,当接收到的包的数量达到一定程度的时候,Miniport Driver会调用NdisMIndicateReceivePacket指示新的NDIS新数据的到来,这时候,NdisMIndicateReceivePacket的调用将导致NDIS调用位于Miniport上层的中间层向NDIS注册的下边沿 ProtocolReceivePacket函数。中间层驱动程序的ProtocolReceivePacket可以对收到的数据包进行相应的处理,之后,可以选择再次调用NdisMIndicateReceivePacket请求NDIS通知更上层数据包的到来,这时,NDIS调用更上层注册的 ProtocolReceive函数,上层的ProtocolReceive对数据包进行必要的处理后,继续调用 NdisMIndicateReceivePacket请求NDIS,通知更上层,最终数据包传到协议驱动中,由相关的协议栈进行处理。
    有时候,在这种级连的上传过程中,并不是那么的顺利,例如,由于某种原因,如:网络设备的驱动程序的可用缓冲区数量减少到某个指定的数量时,网络设备的驱动程序调用NdisMxxxIndicateReceivePacket请求NDIS通知上层数据的到来,这时NDIS将调用上层注册的 ProtocolReceive,上层在ProtocolReceive中进行必要的处理后,进一步调用 NdisMxxxIndicateReceivePacket使得NDIS调用上层的协议驱动注册的ProtocolReceive。在一些不太理想的情况下,一次中断,从网络设备中接收到的数据对某个协议来说并不是一个完整的协议数据包。(一般情况下,其余的数据bit已经在链路途中了,并随后立即就会到达网络设备,这有可能就发生在一个CPU在处理网络设备中断的同时,网络设备的板载存储器上就已经收到了其余的数据了,甚至DMA控制器可能已经把这些数据传到了系统的主存储器上了),这时,上层的ProtocolReceive都无法进行正常的处理(一般对某个包进行处理,都要以相关的协议为依据,进行分析)。这时,在上传到某一层时,可以调用NdisTransferData请求NDIS把随后的信息传上来,这时,NDIS又将在向上传递的途中回过头来向下调用下面的MiniportTransferData,下层重复调用NdisTransferData把这个请求传送到底程的Miniport Driver。如果在上层调用NdisTransferData时不是返回NDIS_STATUS_PENDING,则上层可以继续它的处理,而如果返回 NDIS_STATUS_PENDING,底程在最终完成请求时,调用NdisMTransferDataComplete请求NDIS通知上层传送完成,这将导致上层注册的ProtocolTransferDataComplete被调用,上层调用NdisMTransferDataComplete 请求NDIS通知更上层。由于硬件技术的发展,网络设备板载存储的增加,系统主存储器的增加,以及网络传输能力的改善,NDIS那么迂回的通过 MdisMxxxIndicateReceivePacket,ProtocolReceive,MiniportTransferData这一条坎坷的路径进行数据处理的情况似乎越来越少见了。
    接收过程是由下层传递上去的,同样,底层的Miniport分配了一些资源,如用于存储这个数据包内容的内存,我们希望这些资源最终能极时的被归还,以供以后使用。一个包在从下到上的传递过程时,如果某一层的 ProtocolReceive/ProtocolReceivePacket有兴趣对这个包进行处理的话,则需要检查这个包的OOB信息段是不是携带 NDIS_STATUS_RESOURCES,如果是的话,说明其下层资源紧缺,希望上层在处理的时候,自己考贝一份副本,以供自己使用,因为下层希望自己能够尽快收回这个包的资源,在这里,上层以后可以用自己的那份拷贝指示上层数据包的到来,这样的话,以后,中间层希望上层处理完后,能够收回所有权。另一方面,底层的Miniport并不是每一次都会在OOB信息段中设置 NDIS_STATUS_RESOURCES标志的,这在很多情况下是不必要的。当上层协议驱动完成处理时,可以调用NdisReturnPackets 通知NDIS,相应包已经处理完成,可以安全的释放相关资源了,于是NDIS将调用其下层注册的MiniportReturnPackets,下层在这里释放与这个包相关的资源,并继续调用NdisReturnPackets,请求NDIS把这个通知传给更下层。
   
    这样一来,发数据由上层发起请求往下传,而接收数据从下层往上传,有没有可能一次接收的过层是由上面发起往下传的呢?这是没有必要的,为什么?我的应用程序不就主动调用WSARecv或WSARecvFrom等函数主动向下传递的一次收数据的请求的吗?事实上是这样的,你的接收请求经过 SPI,TDI,到了最终的协议驱动时,如果协议驱动中,指定的套接字上(最重要的是端口和目的IP了)有相应的数据的能满足你的这一次的读请求的话,就完成你的请求,如果不能的话,则阻塞了。当下层有数据来的时候,数据传递到协议驱动时,协议驱动会检查包头的信息并查看当前不是有应用程序打开过相应的端口,以及和对应的目标建立过连接,如果有的话,就把数据存到协议栈中的缓冲区中去,如果对应套接字上有阻塞的接收请求的话,就判断是不是能完成它的请求了,如果能了,就完成他,如果不能,继续等以后下层传上来再重复这个过程。否则就抛弃了,另外,如果,一个应用程序建立一个套接字,并与一台机器建立连接,对方发送了数据,协议栈会把数据保存起来,也许连接超时后,协议栈的缓冲区中还有应用程序没有接收的数据的话,也被抛弃了。所以,协议驱动从来都没有必要主动往下传递一个接收数据的请求。这一点,对于有一点网络程序调试经验的人来说,似乎可以直接找到一个的证据,在调试器的调用WSARecv前下一个断点,用Sniffer抓包,可能数据已经到来了,然后,再回到调试器中执行WSARecv,可以看到收到的就是先前Sniffer下来的数据了。
(注:用一个工作在NDIS协议层或是NDIS中间层的Sniffer来观查。)

3.4 NDIS 的数据包结构。
    发送NDIS Protocol Driver分配相关的包资源,请求NDIS向下传递发送动作,接收时,NDIS Miniport Driver分配相关的包资源请求NDIS向上传递接收动作,中间层要对包进行处理,首先通过包描述符查询出这个包中的所有Buffer描述符,然后,从每一个Buffer描述符中取得相应的数据。一个包描述符中包含了一个或多个Buffer描述符,每一个Buffer描述符中包含了这个Buffer中数据的缓冲区首地址及其大小等信息,另外,一个包描述符中还包括了一些保留给开发者自己使用的Reserved字段,其于的一些字段,并没有被公开。另外 NDIS提供了一些宏和一些函数对这相应的描述符进行操作。为什么要把这个结构弄得这么复杂呢?这是由于面对分层的协议处理的时候,避免过多的数据考贝,比如对于TCP/IP协议来说,上层传递下来的数据,在经过TCP,IP层时,把TCP,IP协议头部那个Buffer分别加入这个Packet来就可以了,如果,要它们在一个Buffer中,则需要在经过TCP层时,TCP层建立一个新的Buffer,把头部放到这个Buffer中来,并把数据考贝到这个Buffer中来;到了IP层还要继续这一动作......

四 基于PassThru框架的中间层驱动程序的扩展实现
    Microsoft 在DDK中附带PassThru提供了一个的中间层驱动框架,使得开发者能够相对容易的在这个基础实现自己的NDIS中间层驱动扩展。我们将在 PassThru的基础上实现一个基本的数据包操作的扩展。对于发送出去的数据包处理,只要在PassThru中的 MiniportSend和MiniportSendPackets中加入必要的操作代码,而对于接收的数据包时,则需要在 ProtocolReceive和ProtocolReceviePackets中加入必要的操作代码,在这里,我将以windows 2003 DDK的PassThru为例,进行扩展。

4.1 发送处理
Cpp代码

   1. VOID MPSendPackets( 
   2.     IN NDIS_HANDLE MiniportAdapterContext, 
   3.     IN PPNDIS_PACKET PacketArray, 
   4.     IN UINT NumberOfPackets 
   5.     ) 
   6. { 
   7.     // 省略代码若干,请参看PassThru的源代码。 
   8.  
   9.      
  10.     // 分配一个新的包描述符。 
  11.  
  12.     NdisAllocatePacket(&Status, &MyPacket, pAdapt->SendPacketPoolHandle); 
  13.     if (Status == NDIS_STATUS_SUCCESS) 
  14.     { 
  15.         PSEND_RSVD SendRsvd; 
  16.         SendRsvd = (PSEND_RSVD)(MyPacket->ProtocolReserved); 
  17.         // 把原来的包描述符保存在新分配的包描述符中的Reserved字段中,原因在后面描述。 
  18.  
  19.         SendRsvd->OriginalPkt = Packet; 
  20.     } 
  21.     // 调用BuildMyPacket对包进行自己的处理(更改包的内容,或其它动作)。 
  22.  
  23.     if (BuildMyPacket(pAdapt, Packet, MyPacket) == FALSE) 
  24.     { 
  25.         // 如果处理失败,则把原始的包信息Copy到MyPacket。 
  26.  
  27.         // 这是为了在处理失败的情况下,也让原始的信息能发出去。 
  28.  
  29.         NDIS_PACKET_FIRST_NDIS_BUFFER(MyPacket) = NDIS_PACKET_FIRST_NDIS_BUFFER(Packet); 
  30.         NDIS_PACKET_LAST_NDIS_BUFFER(MyPacket) = NDIS_PACKET_LAST_NDIS_BUFFER(Packet); 
  31.     } 
  32.     // ...... 
  33.  
  34.     // 请求NDIS向下层传递发送动作。 
  35.  
  36.     NdisSend(&Status, pAdapt->BindingHandle, MyPacket); 
  37.     // 如果NdisSend返回非NDIS_STATUS_PENDING则释放自己的资源。 
  38.  
  39.     if (Status != NDIS_STATUS_PENDING) 
  40.     { 
  41.         DestroyMyPacket(MyPacket); 
  42.         NdisFreePacket(MyPacket); 
  43.     } 
  44.        
  45.     // ....... 
  46.  
  47.     // 如果NdisSend返回非NDIS_STATUS_PENDING,说明下层已成功发送完成 
  48.  
  49.     // 调用NdisMSendComplete请求NDIS通知上层释放资源,这将导致上层注册的 
  50.  
  51.     // ProtocolSendComplete被调用,上层在ProtocolSendComplete中释放资源后 
  52.  
  53.     // 将继续调用NdisMSendComplete请求NDIS把这个通知往上传递。 
  54.  
  55.     if (Status != NDIS_STATUS_PENDING) 
  56.     { 
  57.         NdisMSendComplete(ADAPT_MINIPORT_HANDLE(pAdapt), Packet, Status); 
  58.     } 
  59.     // 如果返回了 NDIS_STATUS_PENDING,则在该层的资源还不能释放,在底层 
  60.  
  61.     // 完成发送时,底层将调用NdisMSendComplete,请求NDIS向上传递这个通知。 
  62.  
  63.     // 该层在自己的ProtocolSendComplete释放相应的资源。 
  64.  
  65. } 
  66.      
  67. VOID PtSendComplete( 
  68.     IN NDIS_HANDLE ProtocolBindingContext, 
  69.     IN PNDIS_PACKET Packet, 
  70.     IN NDIS_STATUS Status 
  71.     ) 
  72. { 
  73.     // ....... 
  74.  
  75.     // 上面发送的时候,我们分配了一个新的包描述符,并把上层的包描述符保存在 
  76.  
  77.     // ProtocolReserved中,现在,把这个上层的包描述符还原出来。 
  78.  
  79.     PSEND_RSVD SendRsvd; 
  80.     SendRsvd = (PSEND_RSVD)(Packet->ProtocolReserved); 
  81.     Pkt = SendRsvd->OriginalPkt; 
  82.      
  83.     // ....... 
  84.  
  85.     // 释放自己的资源 
  86.  
  87.     DestroyMyPacket(Packet); 
  88.     NdisDprFreePacket(Packet); 
  89.     // 用上层的包描述符请求NDIS通知上层释放资源 
  90.  
  91.     NdisMSendComplete(pAdapt->MiniportHandle, Pkt, Status); 
  92. } 
  93.  
  94.  
  95. BOOLEAN BuildMyPacket( 
  96.     IN PADAPT pAdapt, 
  97.     IN PNDIS_PACKET original_packet, 
  98.     OUT PNDIS_PACKET MyPacket 
  99.     ) 
 100. { 
 101.     PSEND_RSVD SendRsvd; 
 102.     NDIS_STATUS Status; 
 103.     NDIS_PHYSICAL_ADDRESS phyaddr = {-1}; 
 104.     PVOID pcontent = NULL; 
 105.     ULONG total_length = 0, current_length = 0; 
 106.     PNDIS_BUFFER MyBuffer; 
 107.  
 108.     // 分配新的内存 
 109.  
 110.     Status = NdisAllocateMemory((PVOID)&pcontent, 2014, 0, phyaddr); 
 111.     if (NDIS_STATUS_SUCCESS != Status) 
 112.         return FALSE; 
 113.     NdisZeroMemory(pcontent, 2014); 
 114.      
 115.     // 把包中的数据Copy到自己的Buffer中来。 
 116.  
 117.     NdisQueryPacket(packet, NULL, NULL, &ndis_buffer, &total_length); 
 118.     while (NULL != ndis_buffer) 
 119.     { 
 120.         NdisQueryBufferSafe(ndis_buffer, &address, &current_length, NormalPagePriority); 
 121.         NdisMoveMemory(pcontent, address, current_length); 
 122.         (PUCHAR)pcontent += current_length; 
 123.         NdisGetNextBuffer(ndis_buffer, &ndis_buffer); 
 124.     } 
 125.      
 126.     // 分配新的一个Buffer描述符 
 127.  
 128.     NdisAllocateBuffer(&Status, &MyBuffer, pAdapt->SendPacketPoolHandle, 
 129.         pcontent, total_length); 
 130.     if (NDIS_STATUS_SUCCESS != Status) 
 131.     { 
 132.         NdisFreeMemory(pcontent, 2014, 0); 
 133.         return FALSE; 
 134.     } 
 135.      
 136.     // 在这里对包的内容进行你自己的处理,如果修改了内容的话,由于这是在协议栈之下, 
 137.  
 138.     // 所以要重新修正CheckSum。调整MyBuffer,和MyPacket相关信息,如长度等。 
 139.  
 140.     // 注意,这里的数据已经是网络字节数据,所以在x86的处理器上要注意字节顺序的问题 
 141.  
 142.     // 对于少量内容的修改,重新扫描整个Buffer修正CheckSum是不值得的。 
 143.  
 144.     // RFC关于Nat的文档中详细描述了基于差异分析的修正CheckSum的方法,并给出了具体算法实现。 
 145.  
 146.      
 147.     // 把新的包描述符存放到新的包描述符中的MiniportReserved中去,原因在 DestroyMyPacket中解释 
 148.  
 149.     SendRsvd = (SEND_RSVD)MyPacket->MiniportReserved; 
 150.     SendRsvd->OriginalPkt = MyPacket; 
 151.     NdisChainBufferAtFront(MyPacket, MyBuffer); 
 152.     return TRUE; 
 153. } 
 154.  
 155.  
 156. VOID DestroyMyPacket(PNDIS_PACKET MyPacket) 
 157. { 
 158.     PNDIS_BUFFER MyBuffer; 
 159.     PVOID address; 
 160.     PNDIS_BUFFER tmpBuffer, MyBuffer; 
 161.     ULONG current_length; 
 162.     PSEND_RSVD SendRsvd; 
 163.  
 164.     // 由于在 BuildMyPacket,分配资源失败的情况下,仍然用原包发送出去, 
 165.  
 166.     // 在这种情况下,是不用释放相关的Buffer资源的 
 167.  
 168.     // BuildMyPacket 当分配成功时,我把新包中的 MiniportReserved 字段 
 169.  
 170.     // 指向了新的包描述符,我以这个为依据判断是否需要释放相关的Buffer资源。 
 171.  
 172.     SendRsvd = (PSEND_RSVD)MyPacket->MiniportReserved; 
 173.     if (MySendRsvd->OriginalPkt != MyPacket) 
 174.         return; 
 175.     NdisUnchainBufferAtFront(MyPacket ,&MyBuffer); 
 176.     while (NULL != MyBuffer) 
 177.     { 
 178.         NdisQueryBufferSafe(MyBuffer, &address, &current_length, NormalPagePriority); 
 179.         NdisFreeMemory(address, current_length, 0); 
 180.         tmpBuffer = MyBuffer; 
 181.         NdisGetNextBuffer(tmpBuffer, &MyBuffer); 
 182.         NdisFreeBuffer(tempBuffer); 
 183.     } 
 184. }    

4.2 接收处理
    接收的时候,由于那个TransferData的曲折过程,使得接收处理要相对复杂一点点,在 ProtocolReceive和 ProtocolReceivePacket中的处理不同。但是由于2003 DDK中的PassThru中,没有对数据进行任何处理,所以,它的ProtocolReceive的处理相对来说,简单了一些。
Cpp代码

   1. NDIS_STATUS PtReceive( 
   2.     IN NDIS_HANDLE ProtocolBindingContext, 
   3.     IN NDIS_HANDLE MacReceiveContext, 
   4.     IN PVOID HeaderBuffer, 
   5.     IN UINT HeaderBufferSize, 
   6.     IN PVOID LookAheadBuffer, 
   7.     IN UINT LookAheadBufferSize, 
   8.     IN UINT PacketSize 
   9.     ) 
  10. { 
  11.     // ....... 
  12.  
  13.  
  14.     do 
  15.     { 
  16.         // 分配一个新的包描述符 
  17.  
  18.         NdisDprAllocatePacket(&Status, &MyPacket, pAdapt->RecvPacketPoolHandle); 
  19.         if (Status == NDIS_STATUS_SUCCESS) 
  20.         { 
  21.             // PassThru维护了一个自己的接收队列,当收到下层的包时,PassThru并不是立刻 
  22.  
  23.             // 调用NdisMIndicateReceive/NdisMXxxIndicateReceive请求NDIS 通知上层新数据 
  24.  
  25.             // 的到来,而是在自己的队列中缓冲起来,当自己的队列满了以后,PassThru将一 
  26.  
  27.             // 次性调用NdisMIndicateReceive请求NDIS通知上层数据的到来。 
  28.  
  29.             // 通常最底程的Miniport Driver也有同样的处理机制。 
  30.  
  31.             PtQueueReceivedPacket(pAdapt, MyPacket, TRUE); 
  32.             // MyPacket的一个副本被Copy到PassThru中的队列去了,现在可以释放这一个Packet了。 
  33.  
  34.             NdisDprFreePacket(MyPacket); 
  35.             break; // 注意这里,跳出了整个do-while了。 
  36.  
  37.         } 
  38.        else 
  39.        { 
  40.             // 
  41.  
  42.             // The miniport below us uses the old-style (not packet) 
  43.  
  44.             // receive indication. Fall through. 
  45.  
  46.             // 
  47.  
  48.        } 
  49.  
  50.         if (Packet != NULL) 
  51.         { 
  52.             // 当执行到这里,说明前面没有跳出do-while循环,也就是说,包申请失败 
  53.  
  54.             // 进一步说明,PassThru的缓冲队列满了,于是有必要,调用 NdisMIndicateReceive 
  55.  
  56.             // 请求NDIS通知上层数据包的到来了,这将导致上层注册的 ProtocolReceivePacket 
  57.  
  58.             // 被调用. 
  59.  
  60.             PtFlushReceiveQueue(pAdapt); 
  61.         } 
  62.          
  63.         // ...... 
  64.  
  65.         // 把队列中的包通知了给上层接收,但是这一个包,由于分配新的描述符失败 
  66.  
  67.         // 所以并没有通知给上层,因此,在这里,进行单独的处理。 
  68.  
  69.          
  70.         pAdapt->IndicateRcvComplete = TRUE; 
  71.         // 以下是进行各种协议相关的处理。 
  72.  
  73.         switch (pAdapt->Medium) 
  74.         { 
  75.         case NdisMedium802_3: 
  76.         case NdisMediumWan: 
  77.             NdisMEthIndicateReceive(pAdapt->MiniportHandle, 
  78.                                 MacReceiveContext, 
  79.                                 HeaderBuffer, 
  80.                                 HeaderBufferSize, 
  81.                                 LookAheadBuffer, 
  82.                                 LookAheadBufferSize, 
  83.                                 PacketSize); 
  84.         break; 
  85.  
  86.         case NdisMedium802_5: 
  87.             NdisMTrIndicateReceive(pAdapt->MiniportHandle, 
  88.                                 MacReceiveContext, 
  89.                                 HeaderBuffer, 
  90.                                 HeaderBufferSize, 
  91.                                 LookAheadBuffer, 
  92.                                 LookAheadBufferSize, 
  93.                                 PacketSize); 
  94.         break; 
  95.     case NdisMediumFddi: 
  96.         NdisMFddiIndicateReceive(pAdapt->MiniportHandle, 
  97.                                  MacReceiveContext, 
  98.                                  HeaderBuffer, 
  99.                                  HeaderBufferSize, 
 100.                                  LookAheadBuffer, 
 101.                                  LookAheadBufferSize, 
 102.                                  PacketSize); 
 103.          break; 
 104.  
 105.     default: 
 106.          ASSERT(FALSE); 
 107.          break; 
 108.         } 
 109.     } while(FALSE); 
 110.  
 111.     return Status; 
 112. } 

NDIS_STATUS PtReceive(
    IN NDIS_HANDLE ProtocolBindingContext,
    IN NDIS_HANDLE MacReceiveContext,
    IN PVOID HeaderBuffer,
    IN UINT HeaderBufferSize,
    IN PVOID LookAheadBuffer,
    IN UINT LookAheadBufferSize,
    IN UINT PacketSize
    )
{
    // .......


    do
    {
        // 分配一个新的包描述符

        NdisDprAllocatePacket(&Status, &MyPacket, pAdapt->RecvPacketPoolHandle);
        if (Status == NDIS_STATUS_SUCCESS)
        {
            // PassThru维护了一个自己的接收队列,当收到下层的包时,PassThru并不是立刻

            // 调用NdisMIndicateReceive/NdisMXxxIndicateReceive请求NDIS通知上层新数据

            // 的到来,而是在自己的队列中缓冲起来,当自己的队列满了以后,PassThru将一

            // 次性调用NdisMIndicateReceive请求NDIS通知上层数据的到来。

            // 通常最底程的Miniport Driver也有同样的处理机制。

            PtQueueReceivedPacket(pAdapt, MyPacket, TRUE);
            // MyPacket的一个副本被Copy到PassThru中的队列去了,现在可以释放这一个Packet了。

            NdisDprFreePacket(MyPacket);
            break; // 注意这里,跳出了整个do-while了。

        }
       else
       {
            //

            // The miniport below us uses the old-style (not packet)

            // receive indication. Fall through.

            //

       }

        if (Packet != NULL)
        {
            // 当执行到这里,说明前面没有跳出do-while循环,也就是说,包申请失败

            // 进一步说明,PassThru的缓冲队列满了,于是有必要,调用 NdisMIndicateReceive

            // 请求NDIS通知上层数据包的到来了,这将导致上层注册的 ProtocolReceivePacket

            // 被调用.

            PtFlushReceiveQueue(pAdapt);
        }
       
        // ......

        // 把队列中的包通知了给上层接收,但是这一个包,由于分配新的描述符失败

        // 所以并没有通知给上层,因此,在这里,进行单独的处理。

       
        pAdapt->IndicateRcvComplete = TRUE;
        // 以下是进行各种协议相关的处理。

        switch (pAdapt->Medium)
        {
        case NdisMedium802_3:
        case NdisMediumWan:
            NdisMEthIndicateReceive(pAdapt->MiniportHandle,
                                MacReceiveContext,
                                HeaderBuffer,
                                HeaderBufferSize,
                                LookAheadBuffer,
                                LookAheadBufferSize,
                                PacketSize);
        break;

        case NdisMedium802_5:
            NdisMTrIndicateReceive(pAdapt->MiniportHandle,
                                MacReceiveContext,
                                HeaderBuffer,
                                HeaderBufferSize,
                                LookAheadBuffer,
                                LookAheadBufferSize,
                                PacketSize);
        break;
    case NdisMediumFddi:
        NdisMFddiIndicateReceive(pAdapt->MiniportHandle,
                                 MacReceiveContext,
                                 HeaderBuffer,
                                 HeaderBufferSize,
                                 LookAheadBuffer,
                                 LookAheadBufferSize,
                                 PacketSize);
         break;

    default:
         ASSERT(FALSE);
         break;
        }
    } while(FALSE);

    return Status;
}

 注意,PassThru 并没有对收到的包进行任何处理,因此,在它的ProtocolReceive中,没有调用NdisTransferData,请求NDIS传送这个包其余的数据,它直接Indicate,把这个工作交给了上层去处理,如果,你的中间层要处理的话,就得在放入队列前面,调用 NdisTransferData,如果返回成功,则在紧接其下进行处理,如果返回 NDIS_STATUS_PENDING 的话,就把处理放到ProtocolTransferDataComplete中处理。所以,你的ProtocolReceive应该看起来是这样的过程:
1.  在把包加入PassThru的队列前,调用NdisTransferData.请求NDIS通知下层传递其余的数据。
    (这里回头走了一段曲折的路了)。
2A. 如果返回成功,则进行处理,如修改数据,重新修正CheckSum,
    在X86平台上可要注意字节顺序的问题了。
2B. 如果返回 NDIS_STATUS_PENDING 就在 ProtocolTransferDataComplete进行2A的处理。
3.  处理完成后,排入PassThru的队列。
    由于,现在网络设备硬件的发展,内存容量的提高,底程的Miniport Driver通常有一个类似PassThru的缓存处理机制,走这条曲折的线路上来,似乎很少见了。

    在搞懂了ProtocolReceive后,rotocolReceivePacket就更简单了。
Cpp代码

   1. INT 
   2. PtReceivePacket( 
   3.     IN NDIS_HANDLE ProtocolBindingContext, 
   4.     IN PNDIS_PACKET Packet 
   5.     ) 
   6.  
   7. { 
   8.     // ....... 
   9.  
  10.     // 进行你自己的处理,修改包的内容,修正CheckSum等操作,参考前面接收过程 
  11.  
  12.     // 取得Packet中的内容 
  13.  
  14.     Status = NDIS_GET_PACKET_STATUS(Packet); 
  15.     if (Status == NDIS_STATUS_RESOURCES) 
  16.     { 
  17.         // 如果下层设置了NDIS_STATUS_RESOURCES,说明下层由于资源 
  18.  
  19.         // 紧缺等原因,要求上层经快处理,于是产生一个考贝的 
  20.  
  21.         // 包描述符排入队列,马上Indicate,并调用NdisReturnPacket请求 
  22.  
  23.         // NDIS归还下层的资源。这是PassThru的处理方法 
  24.  
  25.         // 事实上,如果,你自己要修改这个包,你已经有一个新的Packet和Buffer 
  26.  
  27.         // 以及相关内容了,只要把你的这个新的包排入队列,并调用 NdisReturnPacket 
  28.  
  29.         // 归还下层的资源,而不立即Indicate上层,也是可以的。 
  30.  
  31.         PtQueueReceivedPacket(pAdapt, Packet, TRUE); 
  32.     } 
  33.     else 
  34.     { 
  35.         PtQueueReceivedPacket(pAdapt, MyPacket, FALSE); 
  36.     } 
  37.      
  38.     if (Status == NDIS_STATUS_RESOURCES) 
  39.     { 
  40.         NdisDprFreePacket(MyPacket); 
  41.     } 
  42.     // ...... 
  43.  
  44. } 

INT
PtReceivePacket(
    IN NDIS_HANDLE ProtocolBindingContext,
    IN PNDIS_PACKET Packet
    )

{
    // .......

    // 进行你自己的处理,修改包的内容,修正CheckSum等操作,参考前面接收过程

    // 取得Packet中的内容

    Status = NDIS_GET_PACKET_STATUS(Packet);
    if (Status == NDIS_STATUS_RESOURCES)
    {
        // 如果下层设置了NDIS_STATUS_RESOURCES,说明下层由于资源

        // 紧缺等原因,要求上层经快处理,于是产生一个考贝的

        // 包描述符排入队列,马上Indicate,并调用NdisReturnPacket请求

        // NDIS归还下层的资源。这是PassThru的处理方法

        // 事实上,如果,你自己要修改这个包,你已经有一个新的Packet和Buffer

        // 以及相关内容了,只要把你的这个新的包排入队列,并调用 NdisReturnPacket

        // 归还下层的资源,而不立即Indicate上层,也是可以的。

        PtQueueReceivedPacket(pAdapt, Packet, TRUE);
    }
    else
    {
        PtQueueReceivedPacket(pAdapt, MyPacket, FALSE);
    }
   
    if (Status == NDIS_STATUS_RESOURCES)
    {
        NdisDprFreePacket(MyPacket);
    }
    // ......

}

 

当上层处理完后,调用NdisReturnPacket请求NDIS归还下层的资源时,NDIS调用下层注册的 MiniportReturnPacket在这里释放资源后,再调用NdisReturnPackets通知更下层。

 
Cpp代码

   1. VOID MPReturnPacket( 
   2.     IN NDIS_HANDLE MiniportAdapterContext, 
   3.     IN PNDIS_PACKET Packet 
   4.     ) 
   5.  
   6. { 
   7.     // 如果你和我一样,在接收的时候,自己处理不成功的情况下,就把原来的数据往上传 
   8.  
   9.     // 你就可以像我处理发送的方法一样,利用那几个Reserve的字段,在这里判断, 
  10.  
  11.     // 并进行必要的资源释放。 
  12.  
  13.     // ...... 
  14.  
  15.     // 通知下层释放资源 
  16.  
  17.     NdisReturnPackets(&Packet, 1); 
  18.     // ...... 
  19.  
  20. } 

VOID MPReturnPacket(
    IN NDIS_HANDLE MiniportAdapterContext,
    IN PNDIS_PACKET Packet
    )

{
    // 如果你和我一样,在接收的时候,自己处理不成功的情况下,就把原来的数据往上传

    // 你就可以像我处理发送的方法一样,利用那几个Reserve的字段,在这里判断,

    // 并进行必要的资源释放。

    // ......

    // 通知下层释放资源

    NdisReturnPackets(&Packet, 1);
    // ......

}

 

注意:对于一个像TCP这样面向连接的发送的数据包来说,你要是改变了它的长度的话,那问题就更复杂了,你在协议栈的下方,上层协议栈不知道这个修改,而你把它发到目标计算机去,目标计算机得到的长度是修改后的,那双方的SEQ,ACK就不能同步了,这样的话,你必须记录下你的改动,并对以后的通信,做相应的修正。不然,你一改的话,接下来的通信就RST了。

五 Ring 0 和 Ring 3的通信问题
    也许你并不只是想修改过往的数据,你也想把这些数据直接传到Ring3上去,或有一些其它的理由需要在你的NDIS中间层驱动和应用程序通信, 2003 DDK中的PassThru也提供了这个功能,它已经创建好了设备对像和符号链接,你只需要修改驱动对象的DispatchTable就行了。假如,你希望Ring3的程序通能Read NDIS中间层截获的数据,你需要维护两个队列,一个用于缓存NDIS收到的数据,另一个存放Ring3的IRP请求队列等等,是的,这个已经没有什么好说的了,这一切对于你来说,也经不是难题了。但是,在兴奋之时,需要注意的是,现在是位于任何协议栈的下方,如果没有任何验证机制的话,别人只要在你的局域网中向你发送链路层的广播,或目标MAC地址是你的网卡的MAC的数据包时,你的程序都要受到影响,也就是说,别人随便发个包,你的系统都要为他操劳了,从安全的角度来说,你的系统应该尽可能减小受外部影响的程度,所以,你在加上这个功能的时候,有必要考虑这一点。

后记:
    也许,和很多人一样,我曾经试图看着DDK中的文档学习NDIS驱动开发,但是总是不顺利,到头来似乎感觉NDIS很难,DDK的文档没用,但是,在动手写起程序来时,发现了自己还是离不开DDK的文档,刚开始试图从看DDK文档中的描述,进而想掌握NDIS,这似乎是不容易的,至少对我来说是这样的,等你真正的写起程序来,遇上一些具体的问题,受到一些挫折后,再带着这些问题来看DDK文档就会有收获了。本文描述了自己学习过程中的一些理解,写出来希望能与大家交流,由于自己水平有限,错误之错在所难免,欢迎诸位指正。

QQ: 22517257. Email: Addylee2004@163.com, MSN: Addylee2004@163.com

参考资料:
    DDK开发文档。
    驱动开发网NDIS板块的讨论。
    http://www.xfocus.net/articles/200307/568.html

 

原文地址 http://www.xfocus.net/articles/200605/865.html

原创粉丝点击