4.2 接收处理
接收的时候，由于那个TransferData的曲折过程，使得接收处理要相对复杂一点点，在ProtocolReceive和 ProtocolReceivePacket中的处理不同。但是由于2003 DDK中的PassThru中，没有对数据进行任何处理，所以，它的ProtocolReceive的处理相对来说，简单了一些。
|
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就更简单了。
|
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通知更下层。
|
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
基于PassThru的NDIS中间层驱动程序扩展（前部分）
阅读(3410) | 评论(0) | 转发(0) |
