sync to trunk head (37853) (except rbuild changes)
[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_PTR)(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 /* Return early here because we might be called in the mean time. */
184 if( FCB->Critical ||
185 FCB->ListenIrp.InFlightRequest ||
186 FCB->ReceiveIrp.InFlightRequest ||
187 FCB->SendIrp.InFlightRequest ||
188 FCB->ConnectIrp.InFlightRequest ) {
189 AFD_DbgPrint(MIN_TRACE,("Leaving socket alive (%x %x %x %x)\n",
190 FCB->ListenIrp.InFlightRequest,
191 FCB->ReceiveIrp.InFlightRequest,
192 FCB->SendIrp.InFlightRequest,
193 FCB->ConnectIrp.InFlightRequest));
194 ReturnEarly = TRUE;
195 }
196
197 /* After PoolReeval, this FCB should not be involved in any outstanding
198 * poll requests */
199
200 /* Cancel our pending requests */
201 for( i = 0; i < IN_FLIGHT_REQUESTS; i++ ) {
202 if( InFlightRequest[i]->InFlightRequest ) {
203 AFD_DbgPrint(MID_TRACE,("Cancelling in flight irp %d (%x)\n",
204 i, InFlightRequest[i]->InFlightRequest));
205 IoCancelIrp(InFlightRequest[i]->InFlightRequest);
206 InFlightRequest[i]->InFlightRequest = NULL;
207 }
208 }
209
210 SocketStateUnlock( FCB );
211
212 if( ReturnEarly ) return;
213
214 if( FCB->Recv.Window )
215 ExFreePool( FCB->Recv.Window );
216 if( FCB->Send.Window )
217 ExFreePool( FCB->Send.Window );
218 if( FCB->AddressFrom )
219 ExFreePool( FCB->AddressFrom );
220 if( FCB->LocalAddress )
221 ExFreePool( FCB->LocalAddress );
222 if( FCB->RemoteAddress )
223 ExFreePool( FCB->RemoteAddress );
224 if( FCB->TdiDeviceName.Buffer )
225 ExFreePool(FCB->TdiDeviceName.Buffer);
226
227 ExFreePool(FCB);
228 AFD_DbgPrint(MIN_TRACE,("Deleted (%x)\n", FCB));
229
230 AFD_DbgPrint(MIN_TRACE,("Leaving\n"));
231 }
232
233 static NTSTATUS NTAPI
234 AfdCloseSocket(PDEVICE_OBJECT DeviceObject, PIRP Irp,
235 PIO_STACK_LOCATION IrpSp)
236 {
237 PFILE_OBJECT FileObject = IrpSp->FileObject;
238 PAFD_FCB FCB = FileObject->FsContext;
239
240 AFD_DbgPrint(MID_TRACE,
241 ("AfdClose(DeviceObject %p Irp %p)\n", DeviceObject, Irp));
242
243 AFD_DbgPrint(MID_TRACE,("FCB %x\n", FCB));
244
245 FCB->PollState |= AFD_EVENT_CLOSE;
246 PollReeval( FCB->DeviceExt, FileObject );
247 KillSelectsForFCB( FCB->DeviceExt, FileObject, FALSE );
248
249 if( FCB->EventSelect ) ObDereferenceObject( FCB->EventSelect );
250
251 FileObject->FsContext = NULL;
252 DestroySocket( FCB );
253
254 Irp->IoStatus.Status = STATUS_SUCCESS;
255 Irp->IoStatus.Information = 0;
256 IoCompleteRequest(Irp, IO_NO_INCREMENT);
257
258 AFD_DbgPrint(MID_TRACE, ("Returning success.\n"));
259
260 return STATUS_SUCCESS;
261 }
262
263 static NTSTATUS NTAPI
264 AfdDisconnect(PDEVICE_OBJECT DeviceObject, PIRP Irp,
265 PIO_STACK_LOCATION IrpSp) {
266 PFILE_OBJECT FileObject = IrpSp->FileObject;
267 PAFD_FCB FCB = FileObject->FsContext;
268 PAFD_DISCONNECT_INFO DisReq;
269 IO_STATUS_BLOCK Iosb;
270 PTDI_CONNECTION_INFORMATION ConnInfo;
271 NTSTATUS Status;
272 USHORT Flags = 0;
273
274 if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
275
276 if( !(DisReq = LockRequest( Irp, IrpSp )) )
277 return UnlockAndMaybeComplete( FCB, STATUS_NO_MEMORY,
278 Irp, 0, NULL );
279
280 if (NULL == FCB->RemoteAddress)
281 {
282 ConnInfo = NULL;
283 }
284 else
285 {
286 Status = TdiBuildNullConnectionInfo
287 ( &ConnInfo, FCB->RemoteAddress->Address[0].AddressType );
288
289 if( !NT_SUCCESS(Status) || !ConnInfo )
290 return UnlockAndMaybeComplete( FCB, STATUS_NO_MEMORY,
291 Irp, 0, NULL );
292 }
293
294 if( DisReq->DisconnectType & AFD_DISCONNECT_SEND )
295 Flags |= TDI_DISCONNECT_RELEASE;
296 if( DisReq->DisconnectType & AFD_DISCONNECT_RECV ||
297 DisReq->DisconnectType & AFD_DISCONNECT_ABORT )
298 Flags |= TDI_DISCONNECT_ABORT;
299
300 Status = TdiDisconnect( FCB->Connection.Object,
301 &DisReq->Timeout,
302 Flags,
303 &Iosb,
304 NULL,
305 NULL,
306 FCB->AddressFrom,
307 ConnInfo);
308
309 if (ConnInfo) ExFreePool( ConnInfo );
310
311 return UnlockAndMaybeComplete( FCB, Status, Irp, 0, NULL );
312 }
313
314 static NTSTATUS NTAPI
315 AfdDispatch(PDEVICE_OBJECT DeviceObject, PIRP Irp)
316 {
317 PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
318 NTSTATUS Status = STATUS_SUCCESS;
319 #ifdef DBG
320 PFILE_OBJECT FileObject = IrpSp->FileObject;
321 #endif
322
323 AFD_DbgPrint(MID_TRACE,("AfdDispatch: %d\n", IrpSp->MajorFunction));
324 if( IrpSp->MajorFunction != IRP_MJ_CREATE) {
325 AFD_DbgPrint(MID_TRACE,("FO %x, IrpSp->FO %x\n",
326 FileObject, IrpSp->FileObject));
327 ASSERT(FileObject == IrpSp->FileObject);
328 }
329
330 Irp->IoStatus.Information = 0;
331
332 switch(IrpSp->MajorFunction)
333 {
334 /* opening and closing handles to the device */
335 case IRP_MJ_CREATE:
336 /* Mostly borrowed from the named pipe file system */
337 return AfdCreateSocket(DeviceObject, Irp, IrpSp);
338
339 case IRP_MJ_CLOSE:
340 /* Ditto the borrowing */
341 return AfdCloseSocket(DeviceObject, Irp, IrpSp);
342
343 /* write data */
344 case IRP_MJ_WRITE:
345 return AfdConnectedSocketWriteData( DeviceObject, Irp, IrpSp, TRUE );
346
347 /* read data */
348 case IRP_MJ_READ:
349 return AfdConnectedSocketReadData( DeviceObject, Irp, IrpSp, TRUE );
350
351 case IRP_MJ_DEVICE_CONTROL:
352 {
353 switch( IrpSp->Parameters.DeviceIoControl.IoControlCode ) {
354 case IOCTL_AFD_BIND:
355 return AfdBindSocket( DeviceObject, Irp, IrpSp );
356
357 case IOCTL_AFD_CONNECT:
358 return AfdStreamSocketConnect( DeviceObject, Irp, IrpSp );
359
360 case IOCTL_AFD_START_LISTEN:
361 return AfdListenSocket( DeviceObject, Irp, IrpSp );
362
363 case IOCTL_AFD_RECV:
364 return AfdConnectedSocketReadData( DeviceObject, Irp, IrpSp,
365 FALSE );
366
367 case IOCTL_AFD_SELECT:
368 return AfdSelect( DeviceObject, Irp, IrpSp );
369
370 case IOCTL_AFD_EVENT_SELECT:
371 return AfdEventSelect( DeviceObject, Irp, IrpSp );
372
373 case IOCTL_AFD_ENUM_NETWORK_EVENTS:
374 return AfdEnumEvents( DeviceObject, Irp, IrpSp );
375
376 case IOCTL_AFD_RECV_DATAGRAM:
377 return AfdPacketSocketReadData( DeviceObject, Irp, IrpSp );
378
379 case IOCTL_AFD_SEND:
380 return AfdConnectedSocketWriteData( DeviceObject, Irp, IrpSp,
381 FALSE );
382
383 case IOCTL_AFD_SEND_DATAGRAM:
384 return AfdPacketSocketWriteData( DeviceObject, Irp, IrpSp );
385
386 case IOCTL_AFD_GET_INFO:
387 return AfdGetInfo( DeviceObject, Irp, IrpSp );
388
389 case IOCTL_AFD_GET_CONTEXT:
390 return AfdGetContext( DeviceObject, Irp, IrpSp );
391
392 case IOCTL_AFD_SET_CONTEXT:
393 return AfdSetContext( DeviceObject, Irp, IrpSp );
394
395 case IOCTL_AFD_WAIT_FOR_LISTEN:
396 return AfdWaitForListen( DeviceObject, Irp, IrpSp );
397
398 case IOCTL_AFD_ACCEPT:
399 return AfdAccept( DeviceObject, Irp, IrpSp );
400
401 case IOCTL_AFD_DISCONNECT:
402 return AfdDisconnect( DeviceObject, Irp, IrpSp );
403
404 case IOCTL_AFD_GET_SOCK_NAME:
405 return AfdGetSockOrPeerName( DeviceObject, Irp, IrpSp, TRUE );
406
407 case IOCTL_AFD_GET_PEER_NAME:
408 return AfdGetSockOrPeerName( DeviceObject, Irp, IrpSp, FALSE );
409
410 case IOCTL_AFD_GET_TDI_HANDLES:
411 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_TDI_HANDLES\n"));
412 break;
413
414 case IOCTL_AFD_SET_INFO:
415 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_INFO\n"));
416 break;
417
418 case IOCTL_AFD_SET_CONNECT_DATA:
419 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_DATA\n"));
420 break;
421
422 case IOCTL_AFD_SET_CONNECT_OPTIONS:
423 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_OPTIONS\n"));
424 break;
425
426 case IOCTL_AFD_SET_DISCONNECT_DATA:
427 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_DATA\n"));
428 break;
429
430 case IOCTL_AFD_SET_DISCONNECT_OPTIONS:
431 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_OPTIONS\n"));
432 break;
433
434 case IOCTL_AFD_GET_CONNECT_DATA:
435 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_CONNECT_DATA\n"));
436 break;
437
438 case IOCTL_AFD_GET_CONNECT_OPTIONS:
439 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_CONNECT_OPTIONS\n"));
440 break;
441
442 case IOCTL_AFD_GET_DISCONNECT_DATA:
443 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_DISCONNECT_DATA\n"));
444 break;
445
446 case IOCTL_AFD_GET_DISCONNECT_OPTIONS:
447 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_DISCONNECT_OPTIONS\n"));
448 break;
449
450 case IOCTL_AFD_SET_CONNECT_DATA_SIZE:
451 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_DATA_SIZE\n"));
452 break;
453
454 case IOCTL_AFD_SET_CONNECT_OPTIONS_SIZE:
455 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_OPTIONS_SIZE\n"));
456 break;
457
458 case IOCTL_AFD_SET_DISCONNECT_DATA_SIZE:
459 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_DATA_SIZE\n"));
460 break;
461
462 case IOCTL_AFD_SET_DISCONNECT_OPTIONS_SIZE:
463 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_OPTIONS_SIZE\n"));
464 break;
465
466 case IOCTL_AFD_DEFER_ACCEPT:
467 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_DEFER_ACCEPT\n"));
468 break;
469
470 case IOCTL_AFD_GET_PENDING_CONNECT_DATA:
471 AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_PENDING_CONNECT_DATA\n"));
472 break;
473
474 default:
475 Status = STATUS_NOT_IMPLEMENTED;
476 AFD_DbgPrint(MIN_TRACE, ("Unknown IOCTL (0x%x)\n",
477 IrpSp->Parameters.DeviceIoControl.
478 IoControlCode));
479 break;
480 }
481 break;
482 }
483
484 /* unsupported operations */
485 default:
486 {
487 Status = STATUS_NOT_IMPLEMENTED;
488 AFD_DbgPrint(MIN_TRACE,
489 ("Irp: Unknown Major code was %x\n",
490 IrpSp->MajorFunction));
491 break;
492 }
493 }
494
495 AFD_DbgPrint(MID_TRACE, ("Returning %x\n", Status));
496 Irp->IoStatus.Status = Status;
497 IoCompleteRequest(Irp, IO_NO_INCREMENT);
498
499 return (Status);
500 }
501
502 static VOID NTAPI
503 AfdUnload(PDRIVER_OBJECT DriverObject)
504 {
505 }
506
507 NTSTATUS NTAPI
508 DriverEntry(PDRIVER_OBJECT DriverObject, PUNICODE_STRING RegistryPath)
509 {
510 PDEVICE_OBJECT DeviceObject;
511 UNICODE_STRING wstrDeviceName = RTL_CONSTANT_STRING(L"\\Device\\Afd");
512 PAFD_DEVICE_EXTENSION DeviceExt;
513 NTSTATUS Status;
514
515 /* register driver routines */
516 DriverObject->MajorFunction[IRP_MJ_CLOSE] = AfdDispatch;
517 DriverObject->MajorFunction[IRP_MJ_CREATE] = AfdDispatch;
518 DriverObject->MajorFunction[IRP_MJ_WRITE] = AfdDispatch;
519 DriverObject->MajorFunction[IRP_MJ_READ] = AfdDispatch;
520 DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = AfdDispatch;
521 DriverObject->DriverUnload = AfdUnload;
522
523 Status = IoCreateDevice
524 ( DriverObject,
525 sizeof(AFD_DEVICE_EXTENSION),
526 &wstrDeviceName,
527 FILE_DEVICE_NAMED_PIPE,
528 0,
529 FALSE,
530 &DeviceObject );
531
532 /* failure */
533 if(!NT_SUCCESS(Status))
534 {
535 return (Status);
536 }
537
538 DeviceExt = DeviceObject->DeviceExtension;
539 KeInitializeSpinLock( &DeviceExt->Lock );
540 InitializeListHead( &DeviceExt->Polls );
541
542 AFD_DbgPrint(MID_TRACE,("Device created: object %x ext %x\n",
543 DeviceObject, DeviceExt));
544
545 return (Status);
546 }
547
548 /* EOF */