[TCPIP, IP]
[reactos.git] / lib / drivers / ip / transport / tcp / accept.c
1 /*
2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS TCP/IP protocol driver
4 * FILE: transport/tcp/accept.c
5 * PURPOSE: Transmission Control Protocol Listen/Accept code
6 * PROGRAMMERS: Art Yerkes (arty@users.sf.net)
7 * REVISIONS:
8 * arty 12/21/2004 Created
9 */
10
11 #include "precomp.h"
12
13 /* Listener->Lock MUST be acquired */
14 NTSTATUS TCPServiceListeningSocket( PCONNECTION_ENDPOINT Listener,
15 PCONNECTION_ENDPOINT Connection,
16 PTDI_REQUEST_KERNEL Request ) {
17 NTSTATUS Status;
18 SOCKADDR_IN OutAddr;
19 OSK_UINT OutAddrLen;
20 PTA_IP_ADDRESS RequestAddressReturn;
21 PTDI_CONNECTION_INFORMATION WhoIsConnecting;
22
23 /* Unpack TDI info -- We need the return connection information
24 * struct to return the address so it can be filtered if needed
25 * by WSAAccept -- The returned address will be passed on to
26 * userland after we complete this irp */
27 WhoIsConnecting = (PTDI_CONNECTION_INFORMATION)
28 Request->ReturnConnectionInformation;
29
30 Status = TCPTranslateError
31 ( OskitTCPAccept( Listener->SocketContext,
32 &Connection->SocketContext,
33 Connection,
34 &OutAddr,
35 sizeof(OutAddr),
36 &OutAddrLen,
37 Request->RequestFlags & TDI_QUERY_ACCEPT ? 0 : 1 ) );
38
39 TI_DbgPrint(DEBUG_TCP,("Status %x\n", Status));
40
41 if( NT_SUCCESS(Status) && Status != STATUS_PENDING ) {
42 RequestAddressReturn = WhoIsConnecting->RemoteAddress;
43
44 TI_DbgPrint(DEBUG_TCP,("Copying address to %x (Who %x)\n",
45 RequestAddressReturn, WhoIsConnecting));
46
47 RequestAddressReturn->TAAddressCount = 1;
48 RequestAddressReturn->Address[0].AddressLength = OutAddrLen;
49
50 /* BSD uses the first byte of the sockaddr struct as a length.
51 * Since windows doesn't do that we strip it */
52 RequestAddressReturn->Address[0].AddressType =
53 (OutAddr.sin_family >> 8) & 0xff;
54
55 RtlCopyMemory( &RequestAddressReturn->Address[0].Address,
56 ((PCHAR)&OutAddr) + sizeof(USHORT),
57 sizeof(RequestAddressReturn->Address[0].Address[0]) );
58
59 TI_DbgPrint(DEBUG_TCP,("Done copying\n"));
60 }
61
62 TI_DbgPrint(DEBUG_TCP,("Status %x\n", Status));
63
64 return Status;
65 }
66
67 /* This listen is on a socket we keep as internal. That socket has the same
68 * lifetime as the address file */
69 NTSTATUS TCPListen( PCONNECTION_ENDPOINT Connection, UINT Backlog ) {
70 NTSTATUS Status = STATUS_SUCCESS;
71 SOCKADDR_IN AddressToBind;
72 KIRQL OldIrql;
73
74 ASSERT(Connection);
75 ASSERT_KM_POINTER(Connection->AddressFile);
76
77 LockObject(Connection, &OldIrql);
78
79 TI_DbgPrint(DEBUG_TCP,("TCPListen started\n"));
80
81 TI_DbgPrint(DEBUG_TCP,("Connection->SocketContext %x\n",
82 Connection->SocketContext));
83
84 AddressToBind.sin_family = AF_INET;
85 memcpy( &AddressToBind.sin_addr,
86 &Connection->AddressFile->Address.Address.IPv4Address,
87 sizeof(AddressToBind.sin_addr) );
88 AddressToBind.sin_port = Connection->AddressFile->Port;
89
90 TI_DbgPrint(DEBUG_TCP,("AddressToBind - %x:%x\n", AddressToBind.sin_addr, AddressToBind.sin_port));
91
92 Status = TCPTranslateError( OskitTCPBind( Connection->SocketContext,
93 &AddressToBind,
94 sizeof(AddressToBind) ) );
95
96 if (NT_SUCCESS(Status))
97 Status = TCPTranslateError( OskitTCPListen( Connection->SocketContext, Backlog ) );
98
99 UnlockObject(Connection, OldIrql);
100
101 TI_DbgPrint(DEBUG_TCP,("TCPListen finished %x\n", Status));
102
103 return Status;
104 }
105
106 BOOLEAN TCPAbortListenForSocket( PCONNECTION_ENDPOINT Listener,
107 PCONNECTION_ENDPOINT Connection ) {
108 PLIST_ENTRY ListEntry;
109 PTDI_BUCKET Bucket;
110 KIRQL OldIrql;
111 BOOLEAN Found = FALSE;
112
113 LockObject(Listener, &OldIrql);
114
115 ListEntry = Listener->ListenRequest.Flink;
116 while ( ListEntry != &Listener->ListenRequest ) {
117 Bucket = CONTAINING_RECORD(ListEntry, TDI_BUCKET, Entry);
118
119 if( Bucket->AssociatedEndpoint == Connection ) {
120 RemoveEntryList( &Bucket->Entry );
121 ExFreePoolWithTag( Bucket, TDI_BUCKET_TAG );
122 Found = TRUE;
123 break;
124 }
125
126 ListEntry = ListEntry->Flink;
127 }
128
129 UnlockObject(Listener, OldIrql);
130
131 return Found;
132 }
133
134 NTSTATUS TCPAccept ( PTDI_REQUEST Request,
135 PCONNECTION_ENDPOINT Listener,
136 PCONNECTION_ENDPOINT Connection,
137 PTCP_COMPLETION_ROUTINE Complete,
138 PVOID Context )
139 {
140 NTSTATUS Status;
141 PTDI_BUCKET Bucket;
142 KIRQL OldIrql;
143
144 TI_DbgPrint(DEBUG_TCP,("TCPAccept started\n"));
145
146 LockObject(Listener, &OldIrql);
147
148 Status = TCPServiceListeningSocket( Listener, Connection,
149 (PTDI_REQUEST_KERNEL)Request );
150
151 if( Status == STATUS_PENDING ) {
152 Bucket = ExAllocatePoolWithTag( NonPagedPool, sizeof(*Bucket),
153 TDI_BUCKET_TAG );
154
155 if( Bucket ) {
156 ReferenceObject(Connection);
157 Bucket->AssociatedEndpoint = Connection;
158 Bucket->Request.RequestNotifyObject = Complete;
159 Bucket->Request.RequestContext = Context;
160 InsertTailList( &Listener->ListenRequest, &Bucket->Entry );
161 } else
162 Status = STATUS_NO_MEMORY;
163 }
164
165 UnlockObject(Listener, OldIrql);
166
167 TI_DbgPrint(DEBUG_TCP,("TCPAccept finished %x\n", Status));
168 return Status;
169 }