tdiconn.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Typedefs
typedef VOID *	PTDI_CONNECTION_INFO_PAIR

Functions
PTRANSPORT_ADDRESS	TaCopyTransportAddress (PTRANSPORT_ADDRESS OtherAddress)
PTRANSPORT_ADDRESS	TaBuildNullTransportAddress (UINT AddressType)
UINT	TaLengthOfAddress (PTA_ADDRESS Addr)
UINT	TaLengthOfTransportAddress (PTRANSPORT_ADDRESS Addr)
UINT	TaLengthOfTransportAddressByType (UINT AddressType)
VOID	TaCopyTransportAddressInPlace (PTRANSPORT_ADDRESS Target, PTRANSPORT_ADDRESS Source)
UINT	TdiAddressSizeFromType (UINT Type)
UINT	TdiAddressSizeFromName (PTRANSPORT_ADDRESS Name)
NTSTATUS	TdiBuildConnectionInfoInPlace (PTDI_CONNECTION_INFORMATION ConnInfo, PTRANSPORT_ADDRESS Name)
NTSTATUS	TdiBuildConnectionInfo (PTDI_CONNECTION_INFORMATION *ConnectionInfo, PTRANSPORT_ADDRESS Name)
NTSTATUS	TdiBuildNullConnectionInfoInPlace (PTDI_CONNECTION_INFORMATION ConnInfo, ULONG Type)
	Builds a NULL TDI connection information structure.
NTSTATUS	TdiBuildNullConnectionInfo (PTDI_CONNECTION_INFORMATION *ConnectionInfo, ULONG Type)
	Builds a NULL TDI connection information structure.
NTSTATUS	TdiOpenAddressFile (PUNICODE_STRING DeviceName, PTRANSPORT_ADDRESS Name, ULONG ShareType, PHANDLE AddressHandle, PFILE_OBJECT *AddressObject)
	Opens an IPv4 address file object.
NTSTATUS	TdiAssociateAddressFile (HANDLE AddressHandle, PFILE_OBJECT ConnectionObject)
	Associates a connection endpoint to an address file object.
NTSTATUS	TdiDisassociateAddressFile (PFILE_OBJECT ConnectionObject)
	Disassociates a connection endpoint from an address file object.
NTSTATUS	TdiListen (PIRP *Irp, PFILE_OBJECT ConnectionObject, PTDI_CONNECTION_INFORMATION *RequestConnectionInfo, PTDI_CONNECTION_INFORMATION *ReturnConnectionInfo, PIO_COMPLETION_ROUTINE CompletionRoutine, PVOID CompletionContext)
	Listen on a connection endpoint for a connection request from a remote peer.
NTSTATUS	TdiAccept (PIRP *Irp, PFILE_OBJECT AcceptConnectionObject, PTDI_CONNECTION_INFORMATION RequestConnectionInfo, PTDI_CONNECTION_INFORMATION ReturnConnectionInfo, PIO_COMPLETION_ROUTINE CompletionRoutine, PVOID CompletionContext)
	Listen on a connection endpoint for a connection request from a remote peer.
NTSTATUS	TdiReceive (PIRP *Irp, PFILE_OBJECT ConnectionObject, USHORT Flags, PCHAR Buffer, UINT BufferLength, PIO_COMPLETION_ROUTINE CompletionRoutine, PVOID CompletionContext)
NTSTATUS	TdiSend (PIRP *Irp, PFILE_OBJECT ConnectionObject, USHORT Flags, PCHAR Buffer, UINT BufferLength, PIO_COMPLETION_ROUTINE CompletionRoutine, PVOID CompletionContext)
NTSTATUS	TdiReceiveDatagram (PIRP *Irp, PFILE_OBJECT TransportObject, USHORT Flags, PCHAR Buffer, UINT BufferLength, PTDI_CONNECTION_INFORMATION From, PIO_COMPLETION_ROUTINE CompletionRoutine, PVOID CompletionContext)
	Receives a datagram.
NTSTATUS	TdiSendDatagram (PIRP *Irp, PFILE_OBJECT TransportObject, PCHAR Buffer, UINT BufferLength, PTDI_CONNECTION_INFORMATION To, PIO_COMPLETION_ROUTINE CompletionRoutine, PVOID CompletionContext)
	Sends a datagram.
NTSTATUS	TdiQueryMaxDatagramLength (PFILE_OBJECT FileObject, PUINT MaxDatagramLength)

Typedef Documentation

PTDI_CONNECTION_INFO_PAIR

typedef VOID* PTDI_CONNECTION_INFO_PAIR

Definition at line 3 of file tdiconn.h.

Function Documentation

TaBuildNullTransportAddress()

PTRANSPORT_ADDRESS TaBuildNullTransportAddress

(

UINT

AddressType

)

Definition at line 123 of file tdiconn.c.

{
    UINT AddrLen;
    PTRANSPORT_ADDRESS A;
 
    AddrLen = TaLengthOfTransportAddressByType(AddressType);
    if (!AddrLen)
        return NULL;
 
    A = ExAllocatePoolWithTag(NonPagedPool, AddrLen, TAG_AFD_TRANSPORT_ADDRESS);
 
    if (A)
    {
        if (TdiBuildNullTransportAddressInPlace(A, AddressType) != STATUS_SUCCESS)
        {
            ExFreePoolWithTag(A, TAG_AFD_TRANSPORT_ADDRESS);
            return NULL;
        }
    }
 
    return A;
}

Referenced by AfdPacketSocketWriteData(), and AfdStreamSocketConnect().

TaCopyTransportAddress()

PTRANSPORT_ADDRESS TaCopyTransportAddress

(

PTRANSPORT_ADDRESS

OtherAddress

)

Definition at line 90 of file tdiconn.c.

                                                                             {
    UINT AddrLen;
    PTRANSPORT_ADDRESS A;
 
    AddrLen = TaLengthOfTransportAddress( OtherAddress );
    if (!AddrLen)
        return NULL;
 
    A = ExAllocatePoolWithTag(NonPagedPool,
                              AddrLen,
                              TAG_AFD_TRANSPORT_ADDRESS);
 
    if( A )
        TaCopyTransportAddressInPlace( A, OtherAddress );
 
    return A;
}

Referenced by AfdBindSocket(), AfdStreamSocketConnect(), PacketSocketRecvComplete(), and SatisfyAccept().

TaCopyTransportAddressInPlace()

VOID TaCopyTransportAddressInPlace	(	PTRANSPORT_ADDRESS	Target,
		PTRANSPORT_ADDRESS	Source
	)

Definition at line 84 of file tdiconn.c.

                                                                {
    UINT AddrLen = TaLengthOfTransportAddress( Source );
    RtlCopyMemory( Target, Source, AddrLen );
}

Referenced by ListenComplete(), SatisfyPreAccept(), TaCopyTransportAddress(), and TdiOpenAddressFile().

TaLengthOfAddress()

UINT TaLengthOfAddress

(

PTA_ADDRESS

Addr

)

Definition at line 42 of file tdiconn.c.

{
    UINT AddrLen = Addr->AddressLength;
 
    if (!AddrLen)
        return 0;
 
    AddrLen += 2 * sizeof( USHORT );
 
    DPRINT("AddrLen %x\n", AddrLen);
 
    return AddrLen;
}

Referenced by TaLengthOfTransportAddress().

TaLengthOfTransportAddress()

UINT TaLengthOfTransportAddress

(

PTRANSPORT_ADDRESS

Addr

)

Definition at line 56 of file tdiconn.c.

{
    UINT AddrLen = TaLengthOfAddress(&Addr->Address[0]);
 
    if (!AddrLen)
        return 0;
 
    AddrLen += sizeof(ULONG);
 
    DPRINT("AddrLen %x\n", AddrLen);
 
    return AddrLen;
}

Referenced by AfdGetPeerName(), TaCopyTransportAddress(), TaCopyTransportAddressInPlace(), and TdiOpenAddressFile().

TaLengthOfTransportAddressByType()

UINT TaLengthOfTransportAddressByType

(

UINT

AddressType

)

Definition at line 70 of file tdiconn.c.

{
    UINT AddrLen = TdiAddressSizeFromType(AddressType);
 
    if (!AddrLen)
        return 0;
 
    AddrLen += sizeof(ULONG) + 2 * sizeof(USHORT);
 
    DPRINT("AddrLen %x\n", AddrLen);
 
    return AddrLen;
}

Referenced by TaBuildNullTransportAddress(), TdiBuildNullConnectionInfo(), and TdiBuildNullConnectionInfoInPlace().

TdiAccept()

NTSTATUS TdiAccept	(	PIRP *	Irp,
		PFILE_OBJECT	AcceptConnectionObject,
		PTDI_CONNECTION_INFORMATION	RequestConnectionInfo,
		PTDI_CONNECTION_INFORMATION	ReturnConnectionInfo,
		PIO_COMPLETION_ROUTINE	CompletionRoutine,
		PVOID	CompletionContext
	)

Listen on a connection endpoint for a connection request from a remote peer.

Parameters

CompletionRoutine	= Routine to be called when IRP is completed
CompletionContext	= Context for CompletionRoutine

Returns

Status of operation May return STATUS_PENDING

Definition at line 517 of file tdi.c.

{
    PDEVICE_OBJECT DeviceObject;
 
    DPRINT("Called\n");
 
    if (!AcceptConnectionObject) {
        DPRINT("Bad connection object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    DeviceObject = IoGetRelatedDeviceObject(AcceptConnectionObject);
    if (!DeviceObject) {
        DPRINT("Bad device object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    *Irp = TdiBuildInternalDeviceControlIrp(TDI_LISTEN,
                                            DeviceObject,
                                            AcceptConnectionObject,
                                            NULL,
                                            NULL);
    if (*Irp == NULL)
        return STATUS_INSUFFICIENT_RESOURCES;
 
    TdiBuildAccept(*Irp,
                   DeviceObject,
                   AcceptConnectionObject,
                   CompletionRoutine,
                   CompletionContext,
                   RequestConnectionInfo,
                   ReturnConnectionInfo);
 
    TdiCall(*Irp, DeviceObject, NULL /* Don't wait for completion */, NULL);
 
    return STATUS_PENDING;
}

TdiAddressSizeFromName()

UINT TdiAddressSizeFromName

(

PTRANSPORT_ADDRESS

Name

)

TdiAddressSizeFromType()

UINT TdiAddressSizeFromType

(

UINT

Type

)

Definition at line 23 of file tdiconn.c.

                                                {
    switch( AddressType ) {
    case TDI_ADDRESS_TYPE_IP:
        return TDI_ADDRESS_LENGTH_IP;
    case TDI_ADDRESS_TYPE_APPLETALK:
        return TDI_ADDRESS_LENGTH_APPLETALK;
    case TDI_ADDRESS_TYPE_NETBIOS:
        return TDI_ADDRESS_LENGTH_NETBIOS;
    /* case TDI_ADDRESS_TYPE_NS: */
    case TDI_ADDRESS_TYPE_IPX:
        return TDI_ADDRESS_LENGTH_IPX;
    case TDI_ADDRESS_TYPE_VNS:
        return TDI_ADDRESS_LENGTH_VNS;
    default:
        DbgPrint("TdiAddressSizeFromType - invalid type: %x\n", AddressType);
        return 0;
    }
}

Referenced by TaLengthOfTransportAddressByType(), TdiBuildNullConnectionInfo(), TdiBuildNullConnectionInfoInPlace(), and TdiBuildNullTransportAddressInPlace().

TdiAssociateAddressFile()

NTSTATUS TdiAssociateAddressFile	(	HANDLE	AddressHandle,
		PFILE_OBJECT	ConnectionObject
	)

Associates a connection endpoint to an address file object.

Parameters

AddressHandle	= Handle to address file object
ConnectionObject	= Connection endpoint file object

Returns

Status of operation

Definition at line 366 of file tdi.c.

{
    PDEVICE_OBJECT DeviceObject;
    IO_STATUS_BLOCK Iosb;
    KEVENT Event;
    PIRP Irp;
 
    DPRINT("Called. AddressHandle (%p)  ConnectionObject (%p)\n", AddressHandle, ConnectionObject);
 
    if (!ConnectionObject) {
        DPRINT("Bad connection object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    DeviceObject = IoGetRelatedDeviceObject(ConnectionObject);
    if (!DeviceObject) {
        DPRINT("Bad device object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    KeInitializeEvent(&Event, NotificationEvent, FALSE);
 
    Irp = TdiBuildInternalDeviceControlIrp(TDI_ASSOCIATE_ADDRESS,
                                           DeviceObject,
                                           ConnectionObject,
                                           &Event,
                                           &Iosb);
    if (!Irp)
        return STATUS_INSUFFICIENT_RESOURCES;
 
    TdiBuildAssociateAddress(Irp,
                             DeviceObject,
                             ConnectionObject,
                             NULL,
                             NULL,
                             AddressHandle);
 
    return TdiCall(Irp, DeviceObject, &Event, &Iosb);
}

Referenced by StartListening(), WarmSocketForConnection(), and WskConnect().

TdiBuildConnectionInfo()

NTSTATUS TdiBuildConnectionInfo	(	PTDI_CONNECTION_INFORMATION *	ConnectionInfo,
		PTRANSPORT_ADDRESS	Name
	)

Definition at line 246 of file tdiconn.c.

                               {
    NTSTATUS Status = TdiBuildNullConnectionInfo
        ( ConnectionInfo, Address->Address[0].AddressType );
 
    if( NT_SUCCESS(Status) )
        TdiBuildConnectionInfoInPlace( *ConnectionInfo, Address );
 
    return Status;
}

Referenced by AfdBindSocket(), AfdConnectedSocketWriteData(), AfdPacketSocketWriteData(), AfdStreamSocketConnect(), SatisfyAccept(), and TdiConnectionInfoFromSocketAddress().

TdiBuildConnectionInfoInPlace()

NTSTATUS TdiBuildConnectionInfoInPlace	(	PTDI_CONNECTION_INFORMATION	ConnInfo,
		PTRANSPORT_ADDRESS	Name
	)

Definition at line 228 of file tdiconn.c.

                               {
    NTSTATUS Status = STATUS_SUCCESS;
 
    _SEH2_TRY {
        RtlCopyMemory( ConnectionInfo->RemoteAddress,
                       Address,
                       ConnectionInfo->RemoteAddressLength );
    } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
        Status = _SEH2_GetExceptionCode();
    } _SEH2_END;
 
    return Status;
}

Referenced by TdiBuildConnectionInfo().

TdiBuildNullConnectionInfo()

NTSTATUS TdiBuildNullConnectionInfo	(	PTDI_CONNECTION_INFORMATION *	ConnectionInfo,
		ULONG	Type
	)

Builds a NULL TDI connection information structure.

Parameters

ConnectionInfo	= Address of buffer pointer to allocate connection information in
Type	= TDI style address type (TDI_ADDRESS_TYPE_XXX).

Returns

Status of operation

Definition at line 180 of file tdiconn.c.

{
    PTDI_CONNECTION_INFORMATION ConnInfo;
    ULONG TdiAddressSize;
    NTSTATUS Status;
 
    TdiAddressSize = TaLengthOfTransportAddressByType(Type);
    if (!TdiAddressSize) {
        DPRINT("Invalid parameter\n");
        *ConnectionInfo = NULL;
        return STATUS_INVALID_PARAMETER;
    }
 
    ConnInfo = (PTDI_CONNECTION_INFORMATION)
        ExAllocatePoolWithTag(NonPagedPool,
                              sizeof(TDI_CONNECTION_INFORMATION) + TdiAddressSize,
                              TAG_AFD_TDI_CONNECTION_INFORMATION);
    if (!ConnInfo) {
        *ConnectionInfo = NULL;
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    Status = TdiBuildNullConnectionInfoInPlace( ConnInfo, Type );
 
    if (!NT_SUCCESS(Status))
    {
        ExFreePoolWithTag(ConnInfo, TAG_AFD_TDI_CONNECTION_INFORMATION);
        ConnInfo = NULL;
    }
 
    *ConnectionInfo = ConnInfo;
 
    return Status;
}

Referenced by AfdListenSocket(), ListenComplete(), StartListening(), TdiBuildConnectionInfo(), and WskConnect().

TdiBuildNullConnectionInfoInPlace()

NTSTATUS TdiBuildNullConnectionInfoInPlace	(	PTDI_CONNECTION_INFORMATION	ConnInfo,
		ULONG	Type
	)

Builds a NULL TDI connection information structure.

Parameters

ConnectionInfo	= Address of buffer to place connection information
Type	= TDI style address type (TDI_ADDRESS_TYPE_XXX).

Returns

Status of operation

Definition at line 146 of file tdiconn.c.

{
    ULONG TdiAddressSize;
    PTRANSPORT_ADDRESS TransportAddress;
 
    TdiAddressSize = TaLengthOfTransportAddressByType(Type);
    if (!TdiAddressSize)
    {
        DPRINT("Invalid parameter\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    RtlZeroMemory(ConnInfo,
                  sizeof(TDI_CONNECTION_INFORMATION) +
                  TdiAddressSize);
 
    ConnInfo->OptionsLength = sizeof(ULONG);
    ConnInfo->RemoteAddressLength = TdiAddressSize;
    ConnInfo->RemoteAddress = TransportAddress =
        (PTRANSPORT_ADDRESS)&ConnInfo[1];
 
    return TdiBuildNullTransportAddressInPlace(TransportAddress, Type);
}

Referenced by ListenComplete(), and TdiBuildNullConnectionInfo().

TdiDisassociateAddressFile()

NTSTATUS TdiDisassociateAddressFile

(

PFILE_OBJECT

ConnectionObject

)

Disassociates a connection endpoint from an address file object.

Parameters

ConnectionObject

= Connection endpoint file object

Returns

Status of operation

Definition at line 416 of file tdi.c.

{
    PDEVICE_OBJECT DeviceObject;
    IO_STATUS_BLOCK Iosb;
    KEVENT Event;
    PIRP Irp;
 
    DPRINT("Called. ConnectionObject (%p)\n", ConnectionObject);
 
    if (!ConnectionObject) {
        DPRINT("Bad connection object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    DeviceObject = IoGetRelatedDeviceObject(ConnectionObject);
    if (!DeviceObject) {
        DPRINT("Bad device object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    KeInitializeEvent(&Event, NotificationEvent, FALSE);
 
    Irp = TdiBuildInternalDeviceControlIrp(TDI_DISASSOCIATE_ADDRESS,
                                           DeviceObject,
                                           ConnectionObject,
                                           &Event,
                                           &Iosb);
    if (!Irp)
        return STATUS_INSUFFICIENT_RESOURCES;
 
    TdiBuildDisassociateAddress(Irp,
                                DeviceObject,
                                ConnectionObject,
                                NULL,
                                NULL);
 
    return TdiCall(Irp, DeviceObject, &Event, &Iosb);
}

Referenced by AfdCloseSocket(), SocketShutdown(), StartListening(), and WskConnect().

TdiListen()

NTSTATUS TdiListen	(	PIRP *	Irp,
		PFILE_OBJECT	ConnectionObject,
		PTDI_CONNECTION_INFORMATION *	RequestConnectionInfo,
		PTDI_CONNECTION_INFORMATION *	ReturnConnectionInfo,
		PIO_COMPLETION_ROUTINE	CompletionRoutine,
		PVOID	CompletionContext
	)

Listen on a connection endpoint for a connection request from a remote peer.

Parameters

CompletionRoutine	= Routine to be called when IRP is completed
CompletionContext	= Context for CompletionRoutine

Returns

Status of operation May return STATUS_PENDING

Definition at line 463 of file tdi.c.

{
    PDEVICE_OBJECT DeviceObject;
 
    DPRINT("Called\n");
 
    if (!ConnectionObject) {
        DPRINT("Bad connection object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    DeviceObject = IoGetRelatedDeviceObject(ConnectionObject);
    if (!DeviceObject) {
        DPRINT("Bad device object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    *Irp = TdiBuildInternalDeviceControlIrp(TDI_LISTEN,
                                            DeviceObject,
                                            ConnectionObject,
                                            NULL,
                                            NULL);
    if (*Irp == NULL)
        return STATUS_INSUFFICIENT_RESOURCES;
 
    TdiBuildListen(*Irp,
                   DeviceObject,
                   ConnectionObject,
                   CompletionRoutine,
                   CompletionContext,
                   0,
                   *RequestConnectionInfo,
                   *ReturnConnectionInfo);
 
    TdiCall(*Irp, DeviceObject, NULL /* Don't wait for completion */, NULL);
 
    return STATUS_PENDING;
}

Referenced by AfdListenSocket(), ListenComplete(), and StartListening().

TdiOpenAddressFile()

NTSTATUS TdiOpenAddressFile	(	PUNICODE_STRING	DeviceName,
		PTRANSPORT_ADDRESS	Name,
		ULONG	ShareType,
		PHANDLE	AddressHandle,
		PFILE_OBJECT *	AddressObject
	)

Opens an IPv4 address file object.

Parameters

DeviceName	= Pointer to counted string with name of device
Name	= Pointer to socket name (IPv4 address family)
AddressHandle	= Address of buffer to place address file handle
AddressObject	= Address of buffer to place address file object

Returns

Status of operation

Definition at line 148 of file tdi.c.

{
    PFILE_FULL_EA_INFORMATION EaInfo;
    NTSTATUS Status;
    ULONG EaLength;
    PTRANSPORT_ADDRESS Address;
 
    DPRINT("Called. DeviceName (%wZ)  Name (%p)\n", DeviceName, Name);
 
    /* EaName must be 0-terminated, even though TDI_TRANSPORT_ADDRESS_LENGTH does *not* include the 0 */
    EaLength = sizeof(FILE_FULL_EA_INFORMATION) +
        TDI_TRANSPORT_ADDRESS_LENGTH +
        TaLengthOfTransportAddress( Name ) + 1;
    EaInfo = (PFILE_FULL_EA_INFORMATION)ExAllocatePoolWithTag(NonPagedPool,
                                                              EaLength,
                                                              TAG_AFD_EA_INFO);
    if (!EaInfo)
        return STATUS_INSUFFICIENT_RESOURCES;
 
    RtlZeroMemory(EaInfo, EaLength);
    EaInfo->EaNameLength = TDI_TRANSPORT_ADDRESS_LENGTH;
    /* Don't copy the terminating 0; we have already zeroed it */
    RtlCopyMemory(EaInfo->EaName,
                  TdiTransportAddress,
                  TDI_TRANSPORT_ADDRESS_LENGTH);
    EaInfo->EaValueLength = sizeof(TA_IP_ADDRESS);
    Address =
        (PTRANSPORT_ADDRESS)(EaInfo->EaName + TDI_TRANSPORT_ADDRESS_LENGTH + 1); /* 0-terminated */
    TaCopyTransportAddressInPlace( Address, Name );
 
    Status = TdiOpenDevice(DeviceName,
                           EaLength,
                           EaInfo,
                           ShareType,
                           AddressHandle,
                           AddressObject);
    ExFreePoolWithTag(EaInfo, TAG_AFD_EA_INFO);
    return Status;
}

Referenced by WarmSocketForBind(), and WskBind().

TdiQueryMaxDatagramLength()

NTSTATUS TdiQueryMaxDatagramLength	(	PFILE_OBJECT	FileObject,
		PUINT	MaxDatagramLength
	)

Definition at line 203 of file tdi.c.

{
    PMDL Mdl;
    PTDI_MAX_DATAGRAM_INFO Buffer;
    NTSTATUS Status = STATUS_SUCCESS;
 
    Buffer = ExAllocatePoolWithTag(NonPagedPool,
                                   sizeof(TDI_MAX_DATAGRAM_INFO),
                                   TAG_AFD_DATA_BUFFER);
 
    if (!Buffer) return STATUS_NO_MEMORY;
 
    Mdl = IoAllocateMdl(Buffer, sizeof(TDI_MAX_DATAGRAM_INFO), FALSE, FALSE, NULL);
    if (!Mdl)
    {
        ExFreePoolWithTag(Buffer, TAG_AFD_DATA_BUFFER);
        return STATUS_NO_MEMORY;
    }
 
    _SEH2_TRY
    {
         MmProbeAndLockPages(Mdl, KernelMode, IoModifyAccess);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
         Status = _SEH2_GetExceptionCode();
    }
    _SEH2_END;
 
    if (!NT_SUCCESS(Status))
    {
        DPRINT("Failed to lock pages\n");
        IoFreeMdl(Mdl);
        ExFreePoolWithTag(Buffer, TAG_AFD_DATA_BUFFER);
        return Status;
    }
 
    Status = TdiQueryInformation(FileObject,
                                 TDI_QUERY_MAX_DATAGRAM_INFO,
                                 Mdl);
    if (!NT_SUCCESS(Status))
    {
        ExFreePoolWithTag(Buffer, TAG_AFD_DATA_BUFFER);
        return Status;
    }
 
    *MaxDatagramLength = Buffer->MaxDatagramSize;
 
    ExFreePoolWithTag(Buffer, TAG_AFD_DATA_BUFFER);
 
    return STATUS_SUCCESS;
}

Referenced by MakeSocketIntoConnection(), and WarmSocketForBind().

TdiReceive()

NTSTATUS TdiReceive	(	PIRP *	Irp,
		PFILE_OBJECT	ConnectionObject,
		USHORT	Flags,
		PCHAR	Buffer,
		UINT	BufferLength,
		PIO_COMPLETION_ROUTINE	CompletionRoutine,
		PVOID	CompletionContext
	)

Definition at line 997 of file tdi.c.

{
    PDEVICE_OBJECT DeviceObject;
    PMDL Mdl;
 
    if (!TransportObject) {
        DPRINT("Bad transport object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    DeviceObject = IoGetRelatedDeviceObject(TransportObject);
    if (!DeviceObject) {
        DPRINT("Bad device object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    *Irp = TdiBuildInternalDeviceControlIrp(TDI_RECEIVE,
                                            DeviceObject,
                                            TransportObject,
                                            NULL,
                                            NULL);
 
    if (!*Irp) {
        DPRINT("Insufficient resources.\n");
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    DPRINT("Allocating irp for %p:%u\n", Buffer,BufferLength);
 
    Mdl = IoAllocateMdl(Buffer,         /* Virtual address */
                        BufferLength,   /* Length of buffer */
                        FALSE,          /* Not secondary */
                        FALSE,          /* Don't charge quota */
                        NULL);          /* Don't use IRP */
    if (!Mdl) {
        DPRINT("Insufficient resources.\n");
        IoCompleteRequest(*Irp, IO_NO_INCREMENT);
        *Irp = NULL;
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    _SEH2_TRY {
        DPRINT("probe and lock\n");
        MmProbeAndLockPages(Mdl, (*Irp)->RequestorMode, IoModifyAccess);
        DPRINT("probe and lock done\n");
    } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
        DPRINT("MmProbeAndLockPages() failed.\n");
        IoFreeMdl(Mdl);
        IoCompleteRequest(*Irp, IO_NO_INCREMENT);
        *Irp = NULL;
        _SEH2_YIELD(return STATUS_INSUFFICIENT_RESOURCES);
    } _SEH2_END;
 
    DPRINT("AFD>>> Got an MDL: %p\n", Mdl);
 
    TdiBuildReceive(*Irp,
                    DeviceObject,
                    TransportObject,
                    CompletionRoutine,
                    CompletionContext,
                    Mdl,
                    Flags,
                    BufferLength);
 
 
    TdiCall(*Irp, DeviceObject, NULL, NULL);
    /* Does not block...  The MDL is deleted in the receive completion
       routine. */
 
    return STATUS_PENDING;
}

Referenced by MakeSocketIntoConnection(), RefillSocketBuffer(), and WskStreamIo().

TdiReceiveDatagram()

NTSTATUS TdiReceiveDatagram	(	PIRP *	Irp,
		PFILE_OBJECT	TransportObject,
		USHORT	Flags,
		PCHAR	Buffer,
		UINT	BufferLength,
		PTDI_CONNECTION_INFORMATION	Addr,
		PIO_COMPLETION_ROUTINE	CompletionRoutine,
		PVOID	CompletionContext
	)

Receives a datagram.

Parameters

TransportObject	= Pointer to transport object
From	= Receive filter (NULL if none)
Address	= Address of buffer to place remote address
Buffer	= Address of buffer to place received data
BufferSize	= Address of buffer with length of Buffer (updated)

Returns

Status of operation

Definition at line 1077 of file tdi.c.

{
    PDEVICE_OBJECT DeviceObject;
    PMDL Mdl;
 
    if (!TransportObject) {
        DPRINT("Bad tranport object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    DeviceObject = IoGetRelatedDeviceObject(TransportObject);
    if (!DeviceObject) {
        DPRINT("Bad device object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    *Irp = TdiBuildInternalDeviceControlIrp(TDI_RECEIVE_DATAGRAM,
                                            DeviceObject,
                                            TransportObject,
                                            NULL,
                                            NULL);
 
    if (!*Irp) {
        DPRINT("Insufficient resources.\n");
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    DPRINT("Allocating irp for %p:%u\n", Buffer,BufferLength);
 
    Mdl = IoAllocateMdl(Buffer,         /* Virtual address */
                        BufferLength,   /* Length of buffer */
                        FALSE,          /* Not secondary */
                        FALSE,          /* Don't charge quota */
                        NULL);          /* Don't use IRP */
    if (!Mdl) {
        DPRINT("Insufficient resources.\n");
        IoCompleteRequest(*Irp, IO_NO_INCREMENT);
        *Irp = NULL;
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    _SEH2_TRY {
        MmProbeAndLockPages(Mdl, (*Irp)->RequestorMode, IoModifyAccess);
    } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
        DPRINT("MmProbeAndLockPages() failed.\n");
        IoFreeMdl(Mdl);
        IoCompleteRequest(*Irp, IO_NO_INCREMENT);
        *Irp = NULL;
        _SEH2_YIELD(return STATUS_INSUFFICIENT_RESOURCES);
    } _SEH2_END;
 
    DPRINT("AFD>>> Got an MDL: %p\n", Mdl);
 
    TdiBuildReceiveDatagram(*Irp,
                            DeviceObject,
                            TransportObject,
                            CompletionRoutine,
                            CompletionContext,
                            Mdl,
                            BufferLength,
                            Addr,
                            Addr,
                            Flags);
 
    TdiCall(*Irp, DeviceObject, NULL, NULL);
    /* Does not block...  The MDL is deleted in the receive completion
       routine. */
 
    return STATUS_PENDING;
}

Referenced by PacketSocketRecvComplete(), and WarmSocketForBind().

TdiSend()

NTSTATUS TdiSend	(	PIRP *	Irp,
		PFILE_OBJECT	ConnectionObject,
		USHORT	Flags,
		PCHAR	Buffer,
		UINT	BufferLength,
		PIO_COMPLETION_ROUTINE	CompletionRoutine,
		PVOID	CompletionContext
	)

Definition at line 923 of file tdi.c.

{
    PDEVICE_OBJECT DeviceObject;
    PMDL Mdl;
 
    if (!TransportObject) {
        DPRINT("Bad transport object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    DeviceObject = IoGetRelatedDeviceObject(TransportObject);
    if (!DeviceObject) {
        DPRINT("Bad device object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    *Irp = TdiBuildInternalDeviceControlIrp(TDI_SEND,
                                            DeviceObject,
                                            TransportObject,
                                            NULL,
                                            NULL);
 
    if (!*Irp) {
        DPRINT("Insufficient resources.\n");
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    DPRINT("Allocating irp for %p:%u\n", Buffer,BufferLength);
 
    Mdl = IoAllocateMdl(Buffer,         /* Virtual address */
                        BufferLength,   /* Length of buffer */
                        FALSE,          /* Not secondary */
                        FALSE,          /* Don't charge quota */
                        NULL);          /* Don't use IRP */
    if (!Mdl) {
        DPRINT("Insufficient resources.\n");
        IoCompleteRequest(*Irp, IO_NO_INCREMENT);
        *Irp = NULL;
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    _SEH2_TRY {
        MmProbeAndLockPages(Mdl, (*Irp)->RequestorMode, IoReadAccess);
    } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
        DPRINT("MmProbeAndLockPages() failed.\n");
        IoFreeMdl(Mdl);
        IoCompleteRequest(*Irp, IO_NO_INCREMENT);
        *Irp = NULL;
        _SEH2_YIELD(return STATUS_INSUFFICIENT_RESOURCES);
    } _SEH2_END;
 
    DPRINT("AFD>>> Got an MDL: %p\n", Mdl);
 
    TdiBuildSend(*Irp,
                 DeviceObject,
                 TransportObject,
                 CompletionRoutine,
                 CompletionContext,
                 Mdl,
                 Flags,
                 BufferLength);
 
    /* Does not block...  The MDL is deleted in the receive completion
       routine. */
    return TdiCall(*Irp, DeviceObject, NULL, NULL);
}

Referenced by AfdConnectedSocketWriteData(), SendComplete(), and WskStreamIo().

TdiSendDatagram()

NTSTATUS TdiSendDatagram	(	PIRP *	Irp,
		PFILE_OBJECT	TransportObject,
		PCHAR	Buffer,
		UINT	BufferLength,
		PTDI_CONNECTION_INFORMATION	Addr,
		PIO_COMPLETION_ROUTINE	CompletionRoutine,
		PVOID	CompletionContext
	)

Sends a datagram.

Parameters

TransportObject	= Pointer to transport object
From	= Send filter (NULL if none)
Address	= Address of buffer to place remote address
Buffer	= Address of buffer to place send data
BufferSize	= Address of buffer with length of Buffer (updated)

Returns

Status of operation

Definition at line 1168 of file tdi.c.

{
    PDEVICE_OBJECT DeviceObject;
    PMDL Mdl;
 
    if (!TransportObject) {
        DPRINT("Bad transport object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    DPRINT("Called(TransportObject %p)\n", TransportObject);
 
    DeviceObject = IoGetRelatedDeviceObject(TransportObject);
    if (!DeviceObject) {
        DPRINT("Bad device object.\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    if (BufferLength == 0)
    {
        DPRINT("Succeeding send with length 0.\n");
        return STATUS_SUCCESS;
    }
 
    *Irp = TdiBuildInternalDeviceControlIrp(TDI_SEND_DATAGRAM,
                                            DeviceObject,
                                            TransportObject,
                                            NULL,
                                            NULL);
 
    if (!*Irp) {
        DPRINT("Insufficient resources.\n");
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    DPRINT("Allocating irp for %p:%u\n", Buffer,BufferLength);
 
    Mdl = IoAllocateMdl(Buffer,         /* Virtual address */
                        BufferLength,   /* Length of buffer */
                        FALSE,          /* Not secondary */
                        FALSE,          /* Don't charge quota */
                        NULL);          /* Don't use IRP */
 
    if (!Mdl) {
        DPRINT("Insufficient resources.\n");
        IoCompleteRequest(*Irp, IO_NO_INCREMENT);
        *Irp = NULL;
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    _SEH2_TRY {
        MmProbeAndLockPages(Mdl, (*Irp)->RequestorMode, IoReadAccess);
    } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
        DPRINT("MmProbeAndLockPages() failed.\n");
        IoFreeMdl(Mdl);
        IoCompleteRequest(*Irp, IO_NO_INCREMENT);
        *Irp = NULL;
        _SEH2_YIELD(return STATUS_INSUFFICIENT_RESOURCES);
    } _SEH2_END;
 
    DPRINT("AFD>>> Got an MDL: %p\n", Mdl);
 
    TdiBuildSendDatagram(*Irp,
                         DeviceObject,
                         TransportObject,
                         CompletionRoutine,
                         CompletionContext,
                         Mdl,
                         BufferLength,
                         Addr);
 
    /* Does not block...  The MDL is deleted in the send completion
       routine. */
    return TdiCall(*Irp, DeviceObject, NULL, NULL);
}

Referenced by AfdConnectedSocketWriteData(), AfdPacketSocketWriteData(), and WskSendTo().