/*
 * COPYRIGHT:   See COPYING in the top level directory
 * PROJECT:     ReactOS TCP/IP protocol driver
 * FILE:        transport/tcp/accept.c
 * PURPOSE:     Transmission Control Protocol Listen/Accept code
 * PROGRAMMERS: Art Yerkes (arty@users.sf.net)
 * REVISIONS:
 *   arty 12/21/2004 Created
 */

#include "precomp.h"

/* Listener->Lock MUST be acquired */
NTSTATUS TCPServiceListeningSocket( PCONNECTION_ENDPOINT Listener,
                    PCONNECTION_ENDPOINT Connection,
                    PTDI_REQUEST_KERNEL Request ) {
    NTSTATUS Status;
    SOCKADDR_IN OutAddr;
    OSK_UINT OutAddrLen;
    PTA_IP_ADDRESS RequestAddressReturn;
    PTDI_CONNECTION_INFORMATION WhoIsConnecting;

    /* Unpack TDI info -- We need the return connection information
     * struct to return the address so it can be filtered if needed
     * by WSAAccept -- The returned address will be passed on to
     * userland after we complete this irp */
    WhoIsConnecting = (PTDI_CONNECTION_INFORMATION)
    Request->ReturnConnectionInformation;

    Status = TCPTranslateError
    ( OskitTCPAccept( Listener->SocketContext,
              &Connection->SocketContext,
              Connection,
              &OutAddr,
              sizeof(OutAddr),
              &OutAddrLen,
              Request->RequestFlags & TDI_QUERY_ACCEPT ? 0 : 1 ) );

    TI_DbgPrint(DEBUG_TCP,("Status %x\n", Status));

    if( NT_SUCCESS(Status) && Status != STATUS_PENDING ) {
    RequestAddressReturn = WhoIsConnecting->RemoteAddress;

    TI_DbgPrint(DEBUG_TCP,("Copying address to %x (Who %x)\n",
                   RequestAddressReturn, WhoIsConnecting));

        RequestAddressReturn->TAAddressCount = 1;
    RequestAddressReturn->Address[0].AddressLength = OutAddrLen;

        /* BSD uses the first byte of the sockaddr struct as a length.
         * Since windows doesn't do that we strip it */
    RequestAddressReturn->Address[0].AddressType =
        (OutAddr.sin_family >> 8) & 0xff;

    RtlCopyMemory( &RequestAddressReturn->Address[0].Address,
               ((PCHAR)&OutAddr) + sizeof(USHORT),
               sizeof(RequestAddressReturn->Address[0].Address[0]) );

    TI_DbgPrint(DEBUG_TCP,("Done copying\n"));
    }

    TI_DbgPrint(DEBUG_TCP,("Status %x\n", Status));

    return Status;
}

/* This listen is on a socket we keep as internal.  That socket has the same
 * lifetime as the address file */
NTSTATUS TCPListen( PCONNECTION_ENDPOINT Connection, UINT Backlog ) {
    NTSTATUS Status = STATUS_SUCCESS;
    SOCKADDR_IN AddressToBind;
    KIRQL OldIrql;

    ASSERT(Connection);
    ASSERT_KM_POINTER(Connection->AddressFile);

    LockObject(Connection, &OldIrql);

    TI_DbgPrint(DEBUG_TCP,("TCPListen started\n"));

    TI_DbgPrint(DEBUG_TCP,("Connection->SocketContext %x\n",
    Connection->SocketContext));

    AddressToBind.sin_family = AF_INET;
    memcpy( &AddressToBind.sin_addr,
        &Connection->AddressFile->Address.Address.IPv4Address,
        sizeof(AddressToBind.sin_addr) );
    AddressToBind.sin_port = Connection->AddressFile->Port;

    TI_DbgPrint(DEBUG_TCP,("AddressToBind - %x:%x\n", AddressToBind.sin_addr, AddressToBind.sin_port));

    Status = TCPTranslateError( OskitTCPBind( Connection->SocketContext,
                        &AddressToBind,
                        sizeof(AddressToBind) ) );

    if (NT_SUCCESS(Status))
        Status = TCPTranslateError( OskitTCPListen( Connection->SocketContext, Backlog ) );

    UnlockObject(Connection, OldIrql);

    TI_DbgPrint(DEBUG_TCP,("TCPListen finished %x\n", Status));

    return Status;
}

BOOLEAN TCPAbortListenForSocket( PCONNECTION_ENDPOINT Listener,
                  PCONNECTION_ENDPOINT Connection ) {
    PLIST_ENTRY ListEntry;
    PTDI_BUCKET Bucket;
    KIRQL OldIrql;
    BOOLEAN Found = FALSE;

    LockObject(Listener, &OldIrql);

    ListEntry = Listener->ListenRequest.Flink;
    while ( ListEntry != &Listener->ListenRequest ) {
    Bucket = CONTAINING_RECORD(ListEntry, TDI_BUCKET, Entry);

    if( Bucket->AssociatedEndpoint == Connection ) {
        RemoveEntryList( &Bucket->Entry );
        ExFreePoolWithTag( Bucket, TDI_BUCKET_TAG );
        Found = TRUE;
        break;
    }

    ListEntry = ListEntry->Flink;
    }

    UnlockObject(Listener, OldIrql);

    return Found;
}

NTSTATUS TCPAccept ( PTDI_REQUEST Request,
                     PCONNECTION_ENDPOINT Listener,
                     PCONNECTION_ENDPOINT Connection,
                     PTCP_COMPLETION_ROUTINE Complete,
                     PVOID Context )
{
    NTSTATUS Status;
    PTDI_BUCKET Bucket;
    KIRQL OldIrql;

    TI_DbgPrint(DEBUG_TCP,("TCPAccept started\n"));

    LockObject(Listener, &OldIrql);

    Status = TCPServiceListeningSocket( Listener, Connection,
                       (PTDI_REQUEST_KERNEL)Request );

    if( Status == STATUS_PENDING ) {
        Bucket = ExAllocatePoolWithTag( NonPagedPool, sizeof(*Bucket),
                                        TDI_BUCKET_TAG );

        if( Bucket ) {
            ReferenceObject(Connection);
            Bucket->AssociatedEndpoint = Connection;
            Bucket->Request.RequestNotifyObject = Complete;
            Bucket->Request.RequestContext = Context;
            InsertTailList( &Listener->ListenRequest, &Bucket->Entry );
        } else
            Status = STATUS_NO_MEMORY;
    }

    UnlockObject(Listener, OldIrql);

    TI_DbgPrint(DEBUG_TCP,("TCPAccept finished %x\n", Status));
    return Status;
}