- Fix several cancellation and socket shutdown issues:
[reactos.git] / reactos / drivers / network / afd / afd / main.c
1 /* $Id$
2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS kernel
4 * FILE: drivers/net/afd/afd/main.c
5 * PURPOSE: Ancillary functions driver
6 * PROGRAMMER: Art Yerkes (ayerkes@speakeasy.net)
7 * UPDATE HISTORY:
8 * 20040630 Created
9 *
10 * Suggestions: Uniform naming (AfdXxx)
11 */
12
13 /* INCLUDES */
14
15 #include "afd.h"
16 #include "tdi_proto.h"
17 #include "tdiconn.h"
18 #include "debug.h"
19
20 #ifdef DBG
21
22 /* See debug.h for debug/trace constants */
23 //DWORD DebugTraceLevel = DEBUG_ULTRA;
24 DWORD DebugTraceLevel = 0;
25
26 #endif /* DBG */
27
28 void OskitDumpBuffer( PCHAR Data, UINT Len ) {
29 unsigned int i;
30
31 for( i = 0; i < Len; i++ ) {
32 if( i && !(i & 0xf) ) DbgPrint( "\n" );
33 if( !(i & 0xf) ) DbgPrint( "%08x: ", (UINT)(Data + i) );
34 DbgPrint( " %02x", Data[i] & 0xff );
35 }
36 DbgPrint("\n");
37 }
38
39 /* FUNCTIONS */
40
41 NTSTATUS NTAPI
42 DriverEntry(PDRIVER_OBJECT DriverObject, PUNICODE_STRING RegistryPath);
43
44 static NTSTATUS NTAPI
45 AfdCreateSocket(PDEVICE_OBJECT DeviceObject, PIRP Irp,
46 PIO_STACK_LOCATION IrpSp) {
47 PAFD_FCB FCB;
48 PFILE_OBJECT FileObject;
49 PAFD_DEVICE_EXTENSION DeviceExt;
50 PFILE_FULL_EA_INFORMATION EaInfo;
51 PAFD_CREATE_PACKET ConnectInfo = NULL;
52 ULONG EaLength;
53 PWCHAR EaInfoValue = NULL;
54 UINT Disposition, i;
55 NTSTATUS Status = STATUS_SUCCESS;
56
57 AFD_DbgPrint(MID_TRACE,
58 ("AfdCreate(DeviceObject %p Irp %p)\n", DeviceObject, Irp));
59
60 DeviceExt = DeviceObject->DeviceExtension;
61 FileObject = IrpSp->FileObject;
62 Disposition = (IrpSp->Parameters.Create.Options >> 24) & 0xff;
63
64 Irp->IoStatus.Information = 0;
65
66 EaInfo = Irp->AssociatedIrp.SystemBuffer;
67
68 if( EaInfo ) {
69 ConnectInfo = (PAFD_CREATE_PACKET)(EaInfo->EaName + EaInfo->EaNameLength + 1);
70 EaInfoValue = (PWCHAR)(((PCHAR)ConnectInfo) + sizeof(AFD_CREATE_PACKET));
71
72 EaLength = sizeof(FILE_FULL_EA_INFORMATION) +
73 EaInfo->EaNameLength +
74 EaInfo->EaValueLength;
75
76 AFD_DbgPrint(MID_TRACE,("EaInfo: %x, EaInfoValue: %x\n",
77 EaInfo, EaInfoValue));
78 }
79
80 AFD_DbgPrint(MID_TRACE,("About to allocate the new FCB\n"));
81
82 FCB = ExAllocatePool(NonPagedPool, sizeof(AFD_FCB));
83 if( FCB == NULL ) {
84 Irp->IoStatus.Status = STATUS_NO_MEMORY;
85 IoCompleteRequest(Irp, IO_NO_INCREMENT);
86 return STATUS_NO_MEMORY;
87 }
88
89 AFD_DbgPrint(MID_TRACE,("Initializing the new FCB @ %x (FileObject %x Flags %x)\n", FCB, FileObject, ConnectInfo ? ConnectInfo->EndpointFlags : 0));
90
91 RtlZeroMemory( FCB, sizeof( *FCB ) );
92
93 FCB->Flags = ConnectInfo ? ConnectInfo->EndpointFlags : 0;
94 FCB->State = SOCKET_STATE_CREATED;
95 FCB->FileObject = FileObject;
96 FCB->DeviceExt = DeviceExt;
97 FCB->Recv.Size = DEFAULT_RECEIVE_WINDOW_SIZE;
98 FCB->Send.Size = DEFAULT_SEND_WINDOW_SIZE;
99
100 KeInitializeSpinLock( &FCB->SpinLock );
101 ExInitializeFastMutex( &FCB->Mutex );
102 KeInitializeEvent( &FCB->StateLockedEvent, NotificationEvent, FALSE );
103
104 for( i = 0; i < MAX_FUNCTIONS; i++ ) {
105 InitializeListHead( &FCB->PendingIrpList[i] );
106 }
107
108 InitializeListHead( &FCB->DatagramList );
109 InitializeListHead( &FCB->PendingConnections );
110
111 AFD_DbgPrint(MID_TRACE,("%x: Checking command channel\n", FCB));
112
113 if( ConnectInfo ) {
114 FCB->TdiDeviceName.Length = ConnectInfo->SizeOfTransportName;
115 FCB->TdiDeviceName.MaximumLength = FCB->TdiDeviceName.Length;
116 FCB->TdiDeviceName.Buffer =
117 ExAllocatePool( NonPagedPool, FCB->TdiDeviceName.Length );
118
119 if( !FCB->TdiDeviceName.Buffer ) {
120 ExFreePool(FCB);
121 AFD_DbgPrint(MID_TRACE,("Could not copy target string\n"));
122 Irp->IoStatus.Status = STATUS_NO_MEMORY;
123 IoCompleteRequest( Irp, IO_NETWORK_INCREMENT );
124 return STATUS_NO_MEMORY;
125 }
126
127 RtlCopyMemory( FCB->TdiDeviceName.Buffer,
128 ConnectInfo->TransportName,
129 FCB->TdiDeviceName.Length );
130
131 AFD_DbgPrint(MID_TRACE,("Success: %s %wZ\n",
132 EaInfo->EaName, &FCB->TdiDeviceName));
133 } else {
134 AFD_DbgPrint(MID_TRACE,("Success: Control connection\n"));
135 }
136
137 FileObject->FsContext = FCB;
138
139 /* It seems that UDP sockets are writable from inception */
140 if( FCB->Flags & SGID_CONNECTIONLESS ) {
141 AFD_DbgPrint(MID_TRACE,("Packet oriented socket\n"));
142 /* Allocate our backup buffer */
143 FCB->Recv.Window = ExAllocatePool( NonPagedPool, FCB->Recv.Size );
144 if( !FCB->Recv.Window ) Status = STATUS_NO_MEMORY;
145 FCB->Send.Window = ExAllocatePool( NonPagedPool, FCB->Send.Size );
146 if( !FCB->Send.Window ) {
147 if( FCB->Recv.Window ) ExFreePool( FCB->Recv.Window );
148 Status = STATUS_NO_MEMORY;
149 }
150 /* A datagram socket is always sendable */
151 FCB->PollState |= AFD_EVENT_SEND;
152 PollReeval( FCB->DeviceExt, FCB->FileObject );
153 }
154
155 if( !NT_SUCCESS(Status) ) {
156 if( FCB->TdiDeviceName.Buffer ) ExFreePool( FCB->TdiDeviceName.Buffer );
157 ExFreePool( FCB );
158 FileObject->FsContext = NULL;
159 }
160
161 Irp->IoStatus.Status = Status;
162 IoCompleteRequest( Irp, IO_NETWORK_INCREMENT );
163
164 return Status;
165 }
166
167 VOID DestroySocket( PAFD_FCB FCB ) {
168 UINT i;
169 BOOLEAN ReturnEarly = FALSE;
170 PAFD_IN_FLIGHT_REQUEST InFlightRequest[IN_FLIGHT_REQUESTS];
171
172 AFD_DbgPrint(MIN_TRACE,("Called (%x)\n", FCB));
173
174 if( !SocketAcquireStateLock( FCB ) ) return;
175
176 FCB->State = SOCKET_STATE_CLOSED;
177
178 InFlightRequest[0] = &FCB->ListenIrp;
179 InFlightRequest[1] = &FCB->ReceiveIrp;
180 InFlightRequest[2] = &FCB->SendIrp;
181 InFlightRequest[3] = &FCB->ConnectIrp;
182
183 /* Cancel our pending requests */
184 for( i = 0; i < IN_FLIGHT_REQUESTS; i++ ) {
185 if( InFlightRequest[i]->InFlightRequest ) {
186 AFD_DbgPrint(MID_TRACE,("Cancelling in flight irp %d (%x)\n",
187 i, InFlightRequest[i]->InFlightRequest));
188 if (!IoCancelIrp(InFlightRequest[i]->InFlightRequest))
189 ReturnEarly = TRUE;
190 }
191 }
192
193 SocketStateUnlock( FCB );
194
195 if( ReturnEarly )
196 return;
197
198 if( FCB->Recv.Window )
199 ExFreePool( FCB->Recv.Window );
200 if( FCB->Send.Window )
201 ExFreePool( FCB->Send.Window );
202 if( FCB->AddressFrom )
203 ExFreePool( FCB->AddressFrom );
204 if( FCB->LocalAddress )
205 ExFreePool( FCB->LocalAddress );
206 if( FCB->RemoteAddress )
207 ExFreePool( FCB->RemoteAddress );
208 if( FCB->TdiDeviceName.Buffer )
209 ExFreePool(FCB->TdiDeviceName.Buffer);
210
211 if (FCB->Connection.Object)
212 {
213 NtClose(FCB->Connection.Handle);
214 ObDereferenceObject(FCB->Connection.Object);
215 }
216 if (FCB->AddressFile.Object)
217 {
218 NtClose(FCB->AddressFile.Handle);
219 ObDereferenceObject(FCB->AddressFile.Object);
220 }
221
222 ExFreePool(FCB);
223 AFD_DbgPrint(MIN_TRACE,("Deleted (%x)\n", FCB));
224
225 AFD_DbgPrint(MIN_TRACE,("Leaving\n"));
226 }
227
228 static NTSTATUS NTAPI
229 AfdCloseSocket(PDEVICE_OBJECT DeviceObject, PIRP Irp,
230 PIO_STACK_LOCATION IrpSp)
231 {
232 PFILE_OBJECT FileObject = IrpSp->FileObject;
233 PAFD_FCB FCB = FileObject->FsContext;
234
235 AFD_DbgPrint(MID_TRACE,
236 ("AfdClose(DeviceObject %p Irp %p)\n", DeviceObject, Irp));
237
238 AFD_DbgPrint(MID_TRACE,("FCB %x\n", FCB));
239
240 FCB->PollState |= AFD_EVENT_CLOSE;
241 PollReeval( FCB->DeviceExt, FileObject );
242 KillSelectsForFCB( FCB->DeviceExt, FileObject, FALSE );
243
244 if( FCB->EventSelect ) ObDereferenceObject( FCB->EventSelect );
245
246 FileObject->FsContext = NULL;
247 DestroySocket( FCB );
248
249 Irp->IoStatus.Status = STATUS_SUCCESS;
250 Irp->IoStatus.Information = 0;
251 IoCompleteRequest(Irp, IO_NO_INCREMENT);
252
253 AFD_DbgPrint(MID_TRACE, ("Returning success.\n"));
254
255 return STATUS_SUCCESS;
256 }
257
258 static NTSTATUS NTAPI
259 AfdDisconnect(PDEVICE_OBJECT DeviceObject, PIRP Irp,
260 PIO_STACK_LOCATION IrpSp) {
261 PFILE_OBJECT FileObject = IrpSp->FileObject;
262 PAFD_FCB FCB = FileObject->FsContext;
263 PAFD_DISCONNECT_INFO DisReq;
264 IO_STATUS_BLOCK Iosb;
265 PTDI_CONNECTION_INFORMATION ConnInfo;
266 NTSTATUS Status;
267 USHORT Flags = 0;
268
269 if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
270
271 if( !(DisReq = LockRequest( Irp, IrpSp )) )
272 return UnlockAndMaybeComplete( FCB, STATUS_NO_MEMORY,
273 Irp, 0, NULL );
274
275 if (NULL == FCB->RemoteAddress)
276 {
277 ConnInfo = NULL;
278 }
279 else
280 {
281 Status = TdiBuildNullConnectionInfo
282 ( &ConnInfo, FCB->RemoteAddress->Address[0].AddressType );
283
284 if( !NT_SUCCESS(Status) || !ConnInfo )
285 return UnlockAndMaybeComplete( FCB, STATUS_NO_MEMORY,
286 Irp, 0, NULL );
287 }
288
289 if( DisReq->DisconnectType & AFD_DISCONNECT_SEND )
290 Flags |= TDI_DISCONNECT_RELEASE;
291 if( DisReq->DisconnectType & AFD_DISCONNECT_RECV ||
292 DisReq->DisconnectType & AFD_DISCONNECT_ABORT )
293 Flags |= TDI_DISCONNECT_ABORT;
294
295 Status = TdiDisconnect( FCB->Connection.Object,
296 &DisReq->Timeout,
297 Flags,
298 &Iosb,
299 NULL,
300 NULL,
301 FCB->AddressFrom,
302 ConnInfo);
303
304 if (ConnInfo) ExFreePool( ConnInfo );
305
306 return UnlockAndMaybeComplete( FCB, Status, Irp, 0, NULL );
307 }
308
309 static NTSTATUS NTAPI
310 AfdDispatch(PDEVICE_OBJECT DeviceObject, PIRP Irp)
311 {
312 PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
313 NTSTATUS Status = STATUS_SUCCESS;
314 #ifdef DBG
315 PFILE_OBJECT FileObject = IrpSp->FileObject;
316 #endif
317
318 AFD_DbgPrint(MID_TRACE,("AfdDispatch: %d\n", IrpSp->MajorFunction));
319 if( IrpSp->MajorFunction != IRP_MJ_CREATE) {
320 AFD_DbgPrint(MID_TRACE,("FO %x, IrpSp->FO %x\n",
321 FileObject, IrpSp->FileObject));
322 ASSERT(FileObject == IrpSp->FileObject);
323 }
324
325 Irp->IoStatus.Information = 0;
326
327 switch(IrpSp->MajorFunction)
328 {
329 /* opening and closing handles to the device */
330 case IRP_MJ_CREATE:
331 /* Mostly borrowed from the named pipe file system */
332 return AfdCreateSocket(DeviceObject, Irp, IrpSp);
333
334 case IRP_MJ_CLOSE:
335 /* Ditto the borrowing */
336 return AfdCloseSocket(DeviceObject, Irp, IrpSp);
337
338 /* write data */
339 case IRP_MJ_WRITE:
340 return AfdConnectedSocketWriteData( DeviceObject, Irp, IrpSp, TRUE );
341
342 /* read data */
343 case IRP_MJ_READ:
344 return AfdConnectedSocketReadData( DeviceObject, Irp, IrpSp, TRUE );
345
346 case IRP_MJ_DEVICE_CONTROL:
347 {
348 switch( IrpSp->Parameters.DeviceIoControl.IoControlCode ) {
349 case IOCTL_AFD_BIND:
350 return AfdBindSocket( DeviceObject, Irp, IrpSp );
351
352 case IOCTL_AFD_CONNECT:
353 return AfdStreamSocketConnect( DeviceObject, Irp, IrpSp );
354
355 case IOCTL_AFD_START_LISTEN:
356 return AfdListenSocket( DeviceObject, Irp, IrpSp );
357
358 case IOCTL_AFD_RECV:
359 return AfdConnectedSocketReadData( DeviceObject, Irp, IrpSp,
360 FALSE );
361
362 case IOCTL_AFD_SELECT:
363 return AfdSelect( DeviceObject, Irp, IrpSp );
364
365 case IOCTL_AFD_EVENT_SELECT:
366 return AfdEventSelect( DeviceObject, Irp, IrpSp );
367
368 case IOCTL_AFD_ENUM_NETWORK_EVENTS:
369 return AfdEnumEvents( DeviceObject, Irp, IrpSp );
370
371 case IOCTL_AFD_RECV_DATAGRAM:
372 return AfdPacketSocketReadData( DeviceObject, Irp, IrpSp );
373
374 case IOCTL_AFD_SEND:
375 return AfdConnectedSocketWriteData( DeviceObject, Irp, IrpSp,
376 FALSE );
377
378 case IOCTL_AFD_SEND_DATAGRAM:
379 return AfdPacketSocketWriteData( DeviceObject, Irp, IrpSp );
380
381 case IOCTL_AFD_GET_INFO:
382 return AfdGetInfo( DeviceObject, Irp, IrpSp );
383
384 case IOCTL_AFD_GET_CONTEXT:
385 return AfdGetContext( DeviceObject, Irp, IrpSp );
386
387 case IOCTL_AFD_SET_CONTEXT:
388 return AfdSetContext( DeviceObject, Irp, IrpSp );
389
390 case IOCTL_AFD_WAIT_FOR_LISTEN:
391 return AfdWaitForListen( DeviceObject, Irp, IrpSp );
392
393 case IOCTL_AFD_ACCEPT:
394 return AfdAccept( DeviceObject, Irp, IrpSp );
395
396 case IOCTL_AFD_DISCONNECT:
397 return AfdDisconnect( DeviceObject, Irp, IrpSp );
398
399 case IOCTL_AFD_GET_SOCK_NAME:
400 return AfdGetSockName( DeviceObject, Irp, IrpSp );
401
402 case IOCTL_AFD_GET_PEER_NAME:
403 return AfdGetPeerName( DeviceObject, Irp, IrpSp );
404
405 case IOCTL_AFD_GET_TDI_HANDLES:
406 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_TDI_HANDLES\n"));
407 break;
408
409 case IOCTL_AFD_SET_INFO:
410 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_INFO\n"));
411 break;
412
413 case IOCTL_AFD_SET_CONNECT_DATA:
414 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_DATA\n"));
415 break;
416
417 case IOCTL_AFD_SET_CONNECT_OPTIONS:
418 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_OPTIONS\n"));
419 break;
420
421 case IOCTL_AFD_SET_DISCONNECT_DATA:
422 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_DATA\n"));
423 break;
424
425 case IOCTL_AFD_SET_DISCONNECT_OPTIONS:
426 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_OPTIONS\n"));
427 break;
428
429 case IOCTL_AFD_GET_CONNECT_DATA:
430 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_CONNECT_DATA\n"));
431 break;
432
433 case IOCTL_AFD_GET_CONNECT_OPTIONS:
434 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_CONNECT_OPTIONS\n"));
435 break;
436
437 case IOCTL_AFD_GET_DISCONNECT_DATA:
438 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_DISCONNECT_DATA\n"));
439 break;
440
441 case IOCTL_AFD_GET_DISCONNECT_OPTIONS:
442 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_DISCONNECT_OPTIONS\n"));
443 break;
444
445 case IOCTL_AFD_SET_CONNECT_DATA_SIZE:
446 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_DATA_SIZE\n"));
447 break;
448
449 case IOCTL_AFD_SET_CONNECT_OPTIONS_SIZE:
450 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_OPTIONS_SIZE\n"));
451 break;
452
453 case IOCTL_AFD_SET_DISCONNECT_DATA_SIZE:
454 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_DATA_SIZE\n"));
455 break;
456
457 case IOCTL_AFD_SET_DISCONNECT_OPTIONS_SIZE:
458 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_OPTIONS_SIZE\n"));
459 break;
460
461 case IOCTL_AFD_DEFER_ACCEPT:
462 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_DEFER_ACCEPT\n"));
463 break;
464
465 case IOCTL_AFD_GET_PENDING_CONNECT_DATA:
466 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_PENDING_CONNECT_DATA\n"));
467 break;
468
469 default:
470 Status = STATUS_NOT_IMPLEMENTED;
471 AFD_DbgPrint(MIN_TRACE, ("Unknown IOCTL (0x%x)\n",
472 IrpSp->Parameters.DeviceIoControl.
473 IoControlCode));
474 break;
475 }
476 break;
477 }
478
479 /* unsupported operations */
480 default:
481 {
482 Status = STATUS_NOT_IMPLEMENTED;
483 AFD_DbgPrint(MIN_TRACE,
484 ("Irp: Unknown Major code was %x\n",
485 IrpSp->MajorFunction));
486 break;
487 }
488 }
489
490 AFD_DbgPrint(MID_TRACE, ("Returning %x\n", Status));
491 Irp->IoStatus.Status = Status;
492 IoCompleteRequest(Irp, IO_NO_INCREMENT);
493
494 return (Status);
495 }
496
497 static VOID NTAPI
498 AfdUnload(PDRIVER_OBJECT DriverObject)
499 {
500 }
501
502 NTSTATUS NTAPI
503 DriverEntry(PDRIVER_OBJECT DriverObject, PUNICODE_STRING RegistryPath)
504 {
505 PDEVICE_OBJECT DeviceObject;
506 UNICODE_STRING wstrDeviceName = RTL_CONSTANT_STRING(L"\\Device\\Afd");
507 PAFD_DEVICE_EXTENSION DeviceExt;
508 NTSTATUS Status;
509
510 /* register driver routines */
511 DriverObject->MajorFunction[IRP_MJ_CLOSE] = AfdDispatch;
512 DriverObject->MajorFunction[IRP_MJ_CREATE] = AfdDispatch;
513 DriverObject->MajorFunction[IRP_MJ_WRITE] = AfdDispatch;
514 DriverObject->MajorFunction[IRP_MJ_READ] = AfdDispatch;
515 DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = AfdDispatch;
516 DriverObject->DriverUnload = AfdUnload;
517
518 Status = IoCreateDevice
519 ( DriverObject,
520 sizeof(AFD_DEVICE_EXTENSION),
521 &wstrDeviceName,
522 FILE_DEVICE_NAMED_PIPE,
523 0,
524 FALSE,
525 &DeviceObject );
526
527 /* failure */
528 if(!NT_SUCCESS(Status))
529 {
530 return (Status);
531 }
532
533 DeviceExt = DeviceObject->DeviceExtension;
534 KeInitializeSpinLock( &DeviceExt->Lock );
535 InitializeListHead( &DeviceExt->Polls );
536
537 AFD_DbgPrint(MID_TRACE,("Device created: object %x ext %x\n",
538 DeviceObject, DeviceExt));
539
540 return (Status);
541 }
542
543 /* EOF */