//
//
//  tdiaddr.c
//
//  This file contains code relating to manipulation of address objects
//  that is specific to the VXD environment.  It creates address endpoints
//  with the transport provider.

#include <nbtprocs.h>
#include <tdistat.h>    // TDI error codes

//----------------------------------------------------------------------------
NTSTATUS
NbtTdiOpenAddress (
    OUT PHANDLE             pFileHandle,
    OUT PDEVICE_OBJECT      *ppDeviceObject,
    OUT PFILE_OBJECT        *ppFileObject,
    IN  tDEVICECONTEXT      *pDeviceContext,
    IN  USHORT               PortNumber,
    IN  ULONG               IpAddress,
    IN  ULONG               Flags
    )
/*++

Routine Description:

    Note: This synchronous call may take a number of seconds. It runs in
    the context of the caller.  The code Opens an Address object with the
    transport provider and then sets up event handlers for Receive,
    Disconnect, Datagrams and Errors.

    The address data structures are found in tdi.h , but they are rather
    confusing since the definitions have been spread across several data types.
    This section shows the complete data type for Ip address:

    typedef struct
    {
        int     TA_AddressCount;
        struct _TA_ADDRESS
        {
            USHORT  AddressType;
            USHORT  AddressLength;
            struct _TDI_ADDRESS_IP
            {
                USHORT  sin_port;
                USHORT  in_addr;
                UCHAR   sin_zero[8];
            } TDI_ADDRESS_IP

        } TA_ADDRESS[AddressCount];

    } TRANSPORT_ADDRESS

    An EA buffer is allocated (for the IRP), with an EA name of "TransportAddress"
    and value is a structure of type TRANSPORT_ADDRESS.

Arguments:

    bTCP    - a boolean to say if we are openning a TCP port or a UDP port

Return Value:

    The function value is the status of the operation.

--*/
{
    NTSTATUS                    status = STATUS_SUCCESS ;
    TA_IP_ADDRESS               taip ;   // This is really a TRANSPORT_ADDRESS
    TDI_REQUEST                 tdirequest ;

    DbgPrint("TdiOpenAddress called\n\r");


    taip.TAAddressCount = 1 ;
    taip.Address[0].AddressLength       = sizeof( TDI_ADDRESS_IP ) ;
    taip.Address[0].AddressType         = TDI_ADDRESS_TYPE_IP ;
    taip.Address[0].Address[0].sin_port = htons(PortNumber);    // put in network order
    taip.Address[0].Address[0].in_addr  = htonl(IpAddress);
    CTEZeroMemory( taip.Address[0].Address[0].sin_zero,
                   sizeof( taip.Address[0].Address[0].sin_zero ) ) ;

    #define TCP_PORT    6
    #define UDP_PORT   17
    status = TdiVxdOpenAddress( &tdirequest,
                                (PTRANSPORT_ADDRESS) &taip,
                                Flags & SESSION_FLAG ? TCP_PORT : UDP_PORT,
                                NULL ) ;

    if ( status == TDI_SUCCESS )
    {
        HANDLE hAddress = tdirequest.Handle.AddressHandle ;

        //
        //  As a VXD, the p*FileObject in the DeviceContext structure will
        //  contain the TDI Address.  For compatibility with NT (may want
        //  to change this with an environment specific address).
        //
        *ppFileObject = (PFILE_OBJECT) hAddress ;

        if (Flags & TCP_FLAG)
        {
            // TCP port needs several event handlers for connection
            // management
            status = TdiVxdSetEventHandler(
                            hAddress,
                            TDI_EVENT_RECEIVE,
                            (PVOID)TdiReceiveHandler,
                            (PVOID)pDeviceContext);

            ASSERTMSG( "Failed to set Receive event handler",
                       status == TDI_SUCCESS );

            status = TdiVxdSetEventHandler(
                            hAddress,
                            TDI_EVENT_DISCONNECT,
                            (PVOID)TdiDisconnectHandler,
                            (PVOID)pDeviceContext);
            ASSERTMSG( "Failed to set Disconnect event handler",
                       status == TDI_SUCCESS);

            // only set an connect handler if the session flag is set.
            // In this case the address being opened is the Netbios session
            // port 139
            if (Flags & SESSION_FLAG)
            {
                status = TdiVxdSetEventHandler(
                                hAddress,
                                TDI_EVENT_CONNECT,
                                (PVOID)TdiConnectHandler,
                                (PVOID)pDeviceContext);

                ASSERTMSG((PUCHAR)"Failed to set Receive event handler",
                          status == TDI_SUCCESS );
            }
        }
        else
        {
            // Datagram ports only need this event handler
            if (PortNumber == NBT_DATAGRAM_UDP_PORT)
            {
                // Datagram Udp Handler
                status = TdiVxdSetEventHandler(
                                hAddress,
                                TDI_EVENT_RECEIVE_DATAGRAM,
                                (PVOID)TdiRcvDatagramHandler,
                                (PVOID)pDeviceContext);
                ASSERTMSG("Failed to set Receive Datagram event handler",
                          status == TDI_SUCCESS );
            }
            else
            {
                // Name Service Udp handler
                status = TdiVxdSetEventHandler(
                                hAddress,
                                TDI_EVENT_RECEIVE_DATAGRAM,
                                (PVOID)TdiRcvNameSrvHandler,
                                (PVOID)pDeviceContext);
                ASSERTMSG( "Failed to set Receive Datagram event handler",
                           status == TDI_SUCCESS );
            }
        }

        status = TdiVxdSetEventHandler(
                        hAddress,
                        TDI_EVENT_ERROR,
                        (PVOID)TdiErrorHandler,
                        (PVOID)pDeviceContext);
        ASSERTMSG("Failed to set Error event handler",
                  status == TDI_SUCCESS );


    }

#ifdef DEBUG
    if ( status != STATUS_SUCCESS )
    {
        DbgPrint("NbtTdiOpenAddress: status == ") ;
        DbgPrintNum( status ) ; DbgPrint("\n\r") ;
    }
#endif
    return(status);
}