驱动开发之五 --- TDI之二 【译文】

驱动开发之五 --- TDI之二 【译文】

接上：

传输设备接口

前面socket知识的了解是为了让你对TDI API做好准备。传输设备接口是一组用于驱动中, 与传输协议驱动通讯的API. 就像TCP。传输驱动实现了这组API,所以你的驱动能够与它通讯。

这比socket的使用多少有些复杂。MSDN上的文档资料会让你更加迷惑，而不是有帮助。所以我们将一步一步地建立一个客户端连接。一旦你理解了这个，你就能够使用这些API来执行其他的操作，例如创建一个服务端等等。

体系结构

下图描述了TDI/NDIS的关系。通常，TDI是标准的传输协议接口，协议驱动开发者可以在他们的驱动中实现这些。在这种情况下，开发者希望使用他们自己的协议来实现标准的接口，而不是希望每个他们支持的协议各自实现一套接口产生争执。这并不是说这些开发者只能被限定实现TDI..他们可以在他们驱动的顶层实现任何自己想要的接口。我不是一个NDIS方面的专家，所以我简单的说这么多，希望不会出现什么错误！总之，这些典型的信息了解了，是有好处的，但由于我们只是开发TDI客户端驱动，所以这些我们并不需要去理解。

NDIS协议驱动

在这个驱动的下层，协议驱动会调用NDIS接口API，协议驱动的作用也就是实现一个协议，并且与NDIS会话，驱动的上层可以是一个私有的接口或者TDI或者两者都有. 顺便说一下，这里没有“NDIS客户端”。他们不存在。如果其他站点有指出这些驱动是“NDIS客户端”的，那是完全错误的。我曾经跟一个NDIS专家谈到“NDIS客户端”，他们不知道我所说的是什么。

NDIS中间层驱动

下层是中间层驱动，这些驱动可以用作传输，包调度或者数据过滤。

NDIS小端口驱动

最后一层是NDIS小端口驱动，他们直接与NID物理设备通讯。

你可以在msdn上找到更多的关于TDI和NDIS的体系结构信息。

步骤1：打开传输地址

第一步是创建一个“传输地址”的句柄，你需要使用ZwCreateFile 创建一个“传输地址“实例句柄。这个“传输地址“是指本机的ip地址。这不是远程计算机的ip地址。背后的原因是例如，在本机安装了多个NIC, 在本机有多个IP地址的情况下，让你绑定一个指定的IP地址。

你也可以简单的指定为 "0.0.0.0" 使用任意一块NIC.

打开句柄的办法对于那些不熟悉驱动开发的人来讲有点生硬，你必须指定“EA"或者"Extedned Attributes"，通过IRP_MJ_CREATE把它传递给驱动。你还可以在打开时，传递参数。除了可以在dos设备名的后面添加外，这时，你也能指定本地端口。如果你是在创建一个服务端，就需要在这个时候指定端口。既然我们仅仅是实现一个客户端连接，因此，我们必须无需关心端口，只要设置为0就行。

下面代码举例说明了怎样打开一个传输地址。/*

typedef struct _FILE_FULL_EA_INFORMATION {

    ULONG NextEntryOffset;

    UCHAR Flags;

    UCHAR EaNameLength;

    USHORT EaValueLength;

    CHAR EaName[1];

} FILE_FULL_EA_INFORMATION, *PFILE_FULL_EA_INFORMATION;

*/

NTSTATUS TdiFuncs_OpenTransportAddress(PHANDLE pTdiHandle,

                                      PFILE_OBJECT *pFileObject)

{

    NTSTATUS NtStatus = STATUS_INSUFFICIENT_RESOURCES;

    UNICODE_STRING usTdiDriverNameString;

    OBJECT_ATTRIBUTES oaTdiDriverNameAttributes;

    IO_STATUS_BLOCK IoStatusBlock;

    char DataBlob[sizeof(FILE_FULL_EA_INFORMATION) +

                          TDI_TRANSPORT_ADDRESS_LENGTH + 300] = {0};

    PFILE_FULL_EA_INFORMATION pExtendedAttributesInformation =

         (PFILE_FULL_EA_INFORMATION)&DataBlob;

    UINT dwEASize = 0;

    PTRANSPORT_ADDRESS pTransportAddress = NULL;

    PTDI_ADDRESS_IP pTdiAddressIp = NULL;

    

    /*

     * Initialize the name of the device to be opened. ZwCreateFile takes an

     * OBJECT_ATTRIBUTES structure as the name of the device to open.  

     * This is then a two step process.

     *

     * 1 - Create a UNICODE_STRING data structure from a unicode string.

     * 2 - Create a OBJECT_ATTRIBUTES data structure from a UNICODE_STRING.

    *

     */

    RtlInitUnicodeString(&usTdiDriverNameString, L"\\Device\\Tcp");

    InitializeObjectAttributes(&oaTdiDriverNameAttributes,

          &usTdiDriverNameString,

          OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,

          NULL, NULL);

    /*

     * The second step is to initialize the Extended Attributes data structure.

     *

     * EaName        = TdiTransportAddress, 0, TRANSPORT_ADDRESS

     * EaNameLength = Length of TdiTransportAddress

     * EaValueLength = Length of TRANSPORT_ADDRESS

     */

     RtlCopyMemory(&pExtendedAttributesInformation->EaName,

             TdiTransportAddress,

             TDI_TRANSPORT_ADDRESS_LENGTH);

     pExtendedAttributesInformation->EaNameLength =

                            TDI_TRANSPORT_ADDRESS_LENGTH;

     pExtendedAttributesInformation->EaValueLength =

                            TDI_TRANSPORT_ADDRESS_LENGTH +

              sizeof(TRANSPORT_ADDRESS) +

              sizeof(TDI_ADDRESS_IP);

     pTransportAddress =  

        (PTRANSPORT_ADDRESS)(&pExtendedAttributesInformation->EaName +

        TDI_TRANSPORT_ADDRESS_LENGTH + 1);

     /*

      * The number of transport addresses

      */

     pTransportAddress->TAAddressCount = 1;

     /*

      * This next piece will essentially describe what

      *                                    the transport being opened is.

      * AddressType   = Type of transport

      * AddressLength = Length of the address

      * Address = A data structure that is essentially

      *                          related to the chosen AddressType.

      */

     pTransportAddress->Address[0].AddressType    = TDI_ADDRESS_TYPE_IP;

     pTransportAddress->Address[0].AddressLength = sizeof(TDI_ADDRESS_IP);

     pTdiAddressIp =

          (TDI_ADDRESS_IP *)&pTransportAddress->Address[0].Address;

     /*

      * The TDI_ADDRESS_IP data structure is essentially simmilar to  

      * the usermode sockets data structure.

      *    sin_port

      *    sin_zero

      *    in_addr

      *

      *NOTE: This is the _LOCAL ADDRESS OF THE CURRENT MACHINE_ Just as with  

      *     sockets, if you don't care what port you bind this connection to t  

      *     hen just use "0". If you also only have one network card interface,   

      *     there's no reason to set the IP. "0.0.0.0" will simply use the  

      *     current machine's IP. If you have multiple NIC's or a reason to

      *     specify the local IP address then you must set TDI_ADDRESS_IP

      *     to that IP. If you are creating a server side component you may  

      *     want to specify the port, however usually to connectto another  

      *     server you really don't care what port the client is opening.

      */

     RtlZeroMemory(pTdiAddressIp, sizeof(TDI_ADDRESS_IP));

     dwEASize = sizeof(DataBlob);

     NtStatus = ZwCreateFile(pTdiHandle, FILE_READ_EA | FILE_WRITE_EA,

        &oaTdiDriverNameAttributes,

        &IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, 0, FILE_OPEN_IF, 0,

        pExtendedAttributesInformation, dwEASize);

     if(NT_SUCCESS(NtStatus))

     {

          NtStatus = ObReferenceObjectByHandle(*pTdiHandle,

                        GENERIC_READ | GENERIC_WRITE,

                        NULL,

                        KernelMode,

                        (PVOID *)pFileObject, NULL);      

          if(!NT_SUCCESS(NtStatus))

          {

              ZwClose(*pTdiHandle);

          }

     }

     return NtStatus;

}