Ndis过滤驱动：拷贝NetBufferList数据

yexin218

浏览: 958167 次
性别:
来自: 珠海

最近访客更多访客>>

xiao0556

zhuqinghu

zhangyuexuexi

airyoureyes

博主相关

博客

微博

相册

留言

关于我

文章分类

社区版块

存档分类

博客分类：

驱动开发

数据结构 .net 百度

今天我们来看看如何拷贝NBL中的数据。有时候需要更改数据包中的某些数据，就需要用到此功能(能直接在原来的nBL上更改，然后发送吗？)。

新建一个NBL作为你自己要存放拷贝数据的结构：
```
 pCopyNBL = allocateNetBuffAndNetBufferList(pFilter, dataLength);//Defined by self
```
这个其实是一个自定义的函数，用来分配一个空的NBL.具体代码见后面~

这里考虑有很多个NBL，而且每个NB中包含多个MDL.所以拿到原始NBL的第一个NB

pCurrentNB = NET_BUFFER_LIST_FIRST_NB(pCurrentNBL);//The first NB

再得到第一个NB的第一个MDL:

pMDL = NET_BUFFER_FIRST_MDL(pCurrentNB);

最后得到MDL中有用数据：包括分析offset,以及dataLength，起始地址等：

mdlOffset = NET_BUFFER_DATA_OFFSET(pCurrentNB);//I am not sure
...
 NdisQueryMdl( pMDL,(PVOID *)&pData,&dataBufferLength,NormalPagePriority);
                if(pData == NULL)
                {
                    DEBUGP(DL_TEST,("pData is NULL ---Fail \n"));
                    break;
                }

同理我们得到我们自定义NBL中的MDL(只有一个),同时拿到起始地址：

pCopyMDL  =  NET_BUFFER_FIRST_MDL(NET_BUFFER_LIST_FIRST_NB(pCopyNBL));
....
NdisQueryMdl(pCopyMDL,(PVOID *)&pNewBuffer, &newBufferLength,NormalPagePriority);
            if(pNewBuffer == NULL)
            {
                DEBUGP(DL_TEST,("pNewBuffer is NULL ---Fail \n"));
                break;
            }

然后把pData的数据拷贝到pNewBuffer中，这里注意拷贝的长度：

bytesToCopy = dataBufferLength - mdlOffset;
                if(bytesToCopy > newBufferLength)
                {
                    bytesToCopy = newBufferLength;
                }
                NdisMoveMemory((pNewBuffer + offset),(pData + mdlOffset), bytesToCopy);//copy data
                newBufferLength -= bytesToCopy;
                offset += bytesToCopy;
                mdlOffset = 0;//CurrentMdlOffset is used only for the first Mdl processed. For the remaining Mdls, it is 0.

这里mdlOffset=0，是指以后以后连续的MDL中没有unused 数据。就是整个MDL中的数据都是我们要的。

然后循环这个NB中所有的MDL.每个NB拷贝完之后，设置NB的datalength:

        NET_BUFFER_DATA_LENGTH(NET_BUFFER_LIST_FIRST_NB(pCopyNBL)) = dataLength;
        pCopyNBL->SourceHandle = pCurrentNBL->SourceHandle;

循环所有的NBL,把每个NBL串起来：

if(pNewNetBufferLists == NULL)
        {
            pNewNetBufferLists = pCopyNBLTail = pCopyNBL;
        }
        else
        {
            NET_BUFFER_LIST_NEXT_NBL(pCopyNBLTail) = pCopyNBL;
        }

        pCurrentNBL = NET_BUFFER_LIST_NEXT_NBL(pCurrentNBL);

结束拷贝~附上NBL分配函数代码：

PNET_BUFFER_LIST allocateNetBuffAndNetBufferList(
    IN  PMS_FILTER pFilter,
    IN  ULONG BufferSize
)
{
    PMDL  pMDL =  NULL;
    PUCHAR pMDLAddress =NULL;
    PNET_BUFFER_LIST pNetBufferList = NULL;
    DEBUGP(DL_TEST,("==============>allocateNetBuffAndNetBufferList\n"));
    pMDLAddress = (PUCHAR)NdisAllocateMemoryWithTagPriority(pFilter->FilterHandle,BufferSize,FILTER_ALLOC_TAG, LowPoolPriority);
    NdisZeroMemory(pMDLAddress, BufferSize);
    do
    {
        pMDL = NdisAllocateMdl(pFilter->FilterHandle, pMDLAddress, BufferSize);
        if(pMDL == NULL)
        {
            DEBUGP(DL_TEST,("PMDL Fail....\n"));
            break;
        }

        pNetBufferList = NdisAllocateNetBufferAndNetBufferList(
                             pFilter->SendNetBufferListPool,
                             sizeof(FILTER_SEND_NETBUFLIST_RSVD), //Request control offset delta
                             0,           // back fill size
                             pMDL,
                             0,          // Data offset
                             BufferSize);
        if(pNetBufferList == NULL)
        {
            NdisFreeMdl(pMDL);
            break;
        }
    }
    while(FALSE);
    DEBUGP(DL_TEST,("<==============allocateNetBuffAndNetBufferList\n"));
    return pNetBufferList;
}

需要注意分配MDL,一定要使用NdisAllocateMemoryWithTagPriority分配一个空间地址~

最后是检查是否拷贝正确，参考了这里的代码(打印部分自行添加~)：
基于NDIS Filter 抓包：
效果图：

浏览百度的：

浏览Javaeye的：
还有一些疑问：除了程序编译的时候还有三个Warning.就是好像我只考虑了每个NBL中只有一个NB的情况~可能还需要更多的测试~【今天我把NB也循环了，就是考虑一个NBL多个NB的情况，可是测试中发现一般都自由一个NB】