代码参考aoe
#define
htons(x) (USHORT)((((x) << 8) & 0xff00) | (((x) >> 8) & 0xff))
#define
ntohs(x) (USHORT)((((x) << 8) & 0xff00) | (((x) >> 8) & 0xff))
#define
AOEPROTOCOLID 0x88a2
VOID ProtocolTransferDataComplete(IN NDIS_HANDLE ProtocolBindingContext, IN PNDIS_PACKET Packet, IN NDIS_STATUS Status, IN UINT BytesTransferred) { PNDIS_BUFFER Buffer; UINT HeaderSize, DataSize; PUCHAR Data
=
NULL; PEH_HEADER Header
=
NULL;
//
通过包描述符的NDIS_BUFFER链表的表头取得包的剩余部分的NDIS_BUFFER描述符, 见ProtocolReceive函数
NdisUnchainBufferAtFront(Packet,
&
Buffer);
if
(Buffer
!=
NULL) {
//
得到包的剩余部分的内存地址和大小
NdisQueryBuffer(Buffer,
&
Data,
&
DataSize); NdisFreeBuffer(Buffer); }
else
{ DBGPRINT((
"
ProtocolTransferDataComplete Data (front) Buffer == NULL\\n
"
)); }
//
通过包描述符的NDIS_BUFFER链表的表尾取得包的剩余部分的NDIS_BUFFER描述符, 见ProtocolReceive函数
NdisUnchainBufferAtBack(Packet,
&
Buffer);
if
(Buffer
!=
NULL) {
///
/得到包头的内存地址和大小
NdisQueryBuffer(Buffer,
&
Header,
&
HeaderSize); NdisFreeBuffer(Buffer); }
else
{ DBGPRINT((
"
ProtocolTransferDataComplete Header (back) Buffer == NULL\\n
"
)); }
if
(Header
!=
NULL
&&
Data
!=
NULL) { DBGPRINT((
"
完整包大小为%d
"
, HeaderSize
+
DataSize)); }
//
AoEReply(Header->SourceMac, Header->DestinationMac, Data, DataSize);
if
(Header
!=
NULL) ExFreePool(Header);
if
(Data
!=
NULL) ExFreePool(Data); NdisFreePacket(Packet); } NDIS_STATUS ProtocolReceive(IN NDIS_HANDLE ProtocolBindingContext, IN NDIS_HANDLE MacReceiveContext, IN PVOID HeaderBuffer, IN UINT HeaderBufferSize, IN PVOID LookAheadBuffer, IN UINT LookaheadBufferSize, IN UINT PacketSize) { PADAPTER Context
=
(PADAPTER)ProtocolBindingContext; NDIS_STATUS Status; PNDIS_PACKET Packet; PNDIS_BUFFER Buffer; PEH_HEADER Header; PUCHAR HeaderCopy, Data; UINT BytesTransferred; DBGPRINT((
"
ProtocolReceive\\n
"
));
if
(HeaderBufferSize
!=
sizeof
(ETH_HEADER)) { DbgPrint(
"
ProtocolReceive HeaderBufferSize %d != sizeof(HEADER) %d\\n
"
);
return
NDIS_STATUS_NOT_ACCEPTED; } Header
=
(PEH_HEADER)HeaderBuffer;
/*
if (ntohs(Header->EthType) != AOEPROTOCOLID) return NDIS_STATUS_NOT_ACCEPTED;
*/
if
(LookaheadBufferSize
==
PacketSize) {
//
AoEReply(Header->SourceMac, Header->DestinationMac, LookAheadBuffer, PacketSize);
return
NDIS_STATUS_SUCCESS; }
//
不完整的包
DBGPRINT((
"
不完整的包
"
));
//
分配内存把包头保存包头
if
((HeaderCopy
=
(PUCHAR)ExAllocatePool(NonPagedPool, HeaderBufferSize))
==
NULL) { DbgPrint(
"
ProtocolReceive ExAllocatePool HeaderCopy\\n
"
);
return
NDIS_STATUS_NOT_ACCEPTED; } RtlCopyMemory(HeaderCopy, HeaderBuffer, HeaderBufferSize);
//
分配内存保存包的剩余部分
if
((Data
=
(PUCHAR)ExAllocatePool(NonPagedPool, PacketSize))
==
NULL) { DbgPrint(
"
ProtocolReceive ExAllocatePool HeaderData\\n
"
); ExFreePool(HeaderCopy);
return
NDIS_STATUS_NOT_ACCEPTED; }
//
分配一个包描述符
NdisAllocatePacket(
&
Status,
&
Packet, Context
->
RecvPacketPoolHandle);
if
(
!
NT_SUCCESS(Status)) { DBGPRINT((
"
ProtocolReceive NdisAllocatePacket
"
, Status)); ExFreePool(Data); ExFreePool(HeaderCopy);
return
NDIS_STATUS_NOT_ACCEPTED; }
//
分配一个NDIS_BUFFER描述符，关联内存Data
NdisAllocateBuffer(
&
Status,
&
Buffer, Context
->
RecvBufferPool, Data, PacketSize);
if
(
!
NT_SUCCESS(Status)) { DBGPRINT((
"
ProtocolReceive NdisAllocateBuffer (Data)
"
, Status)); NdisFreePacket(Packet); ExFreePool(Data); ExFreePool(HeaderCopy);
return
NDIS_STATUS_NOT_ACCEPTED; }
//
即将这个NDIS_BUFFER放到包描述符的NDIS_BUFFER链表的表头
NdisChainBufferAtFront(Packet, Buffer);
//
在分配一个NDIS_BUFFER描述符，关联内存HeaderCopy
NdisAllocateBuffer(
&
Status,
&
Buffer, Context
->
RecvBufferPool, HeaderCopy, PacketSize);
if
(
!
NT_SUCCESS(Status)) { DBGPRINT((
"
ProtocolReceive NdisAllocateBuffer (HeaderCopy)
"
, Status)); NdisUnchainBufferAtFront(Packet,
&
Buffer); NdisFreeBuffer(Buffer); NdisFreePacket(Packet); ExFreePool(Data); ExFreePool(HeaderCopy);
return
NDIS_STATUS_NOT_ACCEPTED; }
//
将描述包头的NDIS_BUFFER放到包描述符的NDIS_BUFFER链表的尾节点
NdisChainBufferAtBack(Packet, Buffer); NdisTransferData(
&
Status, Context
->
BindingHandle, MacReceiveContext,
0
, PacketSize, Packet,
&
BytesTransferred);
if
(Status
!=
NDIS_STATUS_PENDING) ProtocolTransferDataComplete(ProtocolBindingContext, Packet, Status, BytesTransferred);
return
Status; }