Create a branch for header work.

svn path=/branches/header-work/; revision=45691
2025-08-06 13:13:00 +00:00 · 2010-02-26 22:57:55 +00:00 · 2010-02-26 22:57:55 +00:00 · 9ea495ba33
commit 9ea495ba33
parent 14fe274b1c
19538 changed files with 0 additions and 1063950 deletions
--- a/lib/drivers/ip/transport/tcp/accept.c
+++ b/lib/drivers/ip/transport/tcp/accept.c
@ -0,0 +1,169 @@
+/*
+ * 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;
+}