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 #if 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->GroupID = ConnectInfo ? ConnectInfo->GroupID : 0;
  95     FCB->GroupType = 0; /* FIXME */
  96     FCB->State = SOCKET_STATE_CREATED;
  97     FCB->FileObject = FileObject;
  98     FCB->DeviceExt = DeviceExt;
  99     FCB->Recv.Size = DEFAULT_RECEIVE_WINDOW_SIZE;
 100     FCB->Send.Size = DEFAULT_SEND_WINDOW_SIZE;
 101     FCB->AddressFile.Handle = INVALID_HANDLE_VALUE;
 102     FCB->Connection.Handle = INVALID_HANDLE_VALUE;
 103
 104     KeInitializeSpinLock( &FCB->SpinLock );
 105     ExInitializeFastMutex( &FCB->Mutex );
 106     KeInitializeEvent( &FCB->StateLockedEvent, NotificationEvent, FALSE );
 107
 108     for( i = 0; i < MAX_FUNCTIONS; i++ ) {
 109         InitializeListHead( &FCB->PendingIrpList[i] );
 110     }
 111
 112     InitializeListHead( &FCB->DatagramList );
 113     InitializeListHead( &FCB->PendingConnections );
 114
 115     AFD_DbgPrint(MID_TRACE,("%x: Checking command channel\n", FCB));
 116
 117     if( ConnectInfo ) {
 118         FCB->TdiDeviceName.Length = ConnectInfo->SizeOfTransportName;
 119         FCB->TdiDeviceName.MaximumLength = FCB->TdiDeviceName.Length;
 120         FCB->TdiDeviceName.Buffer =
 121             ExAllocatePool( NonPagedPool, FCB->TdiDeviceName.Length );
 122
 123         if( !FCB->TdiDeviceName.Buffer ) {
 124             ExFreePool(FCB);
 125             AFD_DbgPrint(MID_TRACE,("Could not copy target string\n"));
 126             Irp->IoStatus.Status = STATUS_NO_MEMORY;
 127             IoCompleteRequest( Irp, IO_NETWORK_INCREMENT );
 128             return STATUS_NO_MEMORY;
 129         }
 130
 131         RtlCopyMemory( FCB->TdiDeviceName.Buffer,
 132                        ConnectInfo->TransportName,
 133                        FCB->TdiDeviceName.Length );
 134
 135         AFD_DbgPrint(MID_TRACE,("Success: %s %wZ\n",
 136                                 EaInfo->EaName, &FCB->TdiDeviceName));
 137     } else {
 138         AFD_DbgPrint(MID_TRACE,("Success: Control connection\n"));
 139     }
 140
 141     FileObject->FsContext = FCB;
 142
 143     /* It seems that UDP sockets are writable from inception */
 144     if( FCB->Flags & SGID_CONNECTIONLESS ) {
 145         AFD_DbgPrint(MID_TRACE,("Packet oriented socket\n"));
 146         /* Allocate our backup buffer */
 147         FCB->Recv.Window = ExAllocatePool( NonPagedPool, FCB->Recv.Size );
 148         if( !FCB->Recv.Window ) Status = STATUS_NO_MEMORY;
 149         if( NT_SUCCESS(Status) )
 150         {
 151             FCB->Send.Window = ExAllocatePool( NonPagedPool, FCB->Send.Size );
 152             if( !FCB->Send.Window ) {
 153                  if( FCB->Recv.Window ) ExFreePool( FCB->Recv.Window );
 154                  Status = STATUS_NO_MEMORY;
 155             }
 156              }
 157         /* A datagram socket is always sendable */
 158         FCB->PollState |= AFD_EVENT_SEND;
 159         PollReeval( FCB->DeviceExt, FCB->FileObject );
 160     }
 161
 162     if( !NT_SUCCESS(Status) ) {
 163         if( FCB->TdiDeviceName.Buffer ) ExFreePool( FCB->TdiDeviceName.Buffer );
 164         ExFreePool( FCB );
 165         FileObject->FsContext = NULL;
 166     }
 167
 168     Irp->IoStatus.Status = Status;
 169     IoCompleteRequest( Irp, IO_NETWORK_INCREMENT );
 170
 171     return Status;
 172 }
 173
 174 static NTSTATUS NTAPI
 175 AfdCleanupSocket(PDEVICE_OBJECT DeviceObject, PIRP Irp,
 176                  PIO_STACK_LOCATION IrpSp)
 177 {
 178     PFILE_OBJECT FileObject = IrpSp->FileObject;
 179     PAFD_FCB FCB = FileObject->FsContext;
 180     PLIST_ENTRY CurrentEntry, NextEntry;
 181     UINT Function;
 182     PIRP CurrentIrp;
 183
 184     if( !SocketAcquireStateLock( FCB ) ) return LostSocket(Irp);
 185
 186     for (Function = 0; Function < MAX_FUNCTIONS; Function++)
 187     {
 188         CurrentEntry = FCB->PendingIrpList[Function].Flink;
 189         while (CurrentEntry != &FCB->PendingIrpList[Function])
 190         {
 191            NextEntry = CurrentEntry->Flink;
 192            CurrentIrp = CONTAINING_RECORD(CurrentEntry, IRP, Tail.Overlay.ListEntry);
 193
 194            /* The cancel routine will remove the IRP from the list */
 195            IoCancelIrp(CurrentIrp);
 196
 197            CurrentEntry = NextEntry;
 198         }
 199     }
 200
 201     KillSelectsForFCB( FCB->DeviceExt, FileObject, FALSE );
 202
 203     return UnlockAndMaybeComplete(FCB, STATUS_SUCCESS, Irp, 0);
 204 }
 205
 206 static NTSTATUS NTAPI
 207 AfdCloseSocket(PDEVICE_OBJECT DeviceObject, PIRP Irp,
 208                PIO_STACK_LOCATION IrpSp)
 209 {
 210     PFILE_OBJECT FileObject = IrpSp->FileObject;
 211     PAFD_FCB FCB = FileObject->FsContext;
 212     UINT i;
 213     PAFD_IN_FLIGHT_REQUEST InFlightRequest[IN_FLIGHT_REQUESTS];
 214
 215     AFD_DbgPrint(MID_TRACE,
 216                  ("AfdClose(DeviceObject %p Irp %p)\n", DeviceObject, Irp));
 217
 218     if( !SocketAcquireStateLock( FCB ) ) return STATUS_FILE_CLOSED;
 219
 220     FCB->State = SOCKET_STATE_CLOSED;
 221     FCB->PollState = AFD_EVENT_CLOSE;
 222     PollReeval( FCB->DeviceExt, FCB->FileObject );
 223
 224     InFlightRequest[0] = &FCB->ListenIrp;
 225     InFlightRequest[1] = &FCB->ReceiveIrp;
 226     InFlightRequest[2] = &FCB->SendIrp;
 227     InFlightRequest[3] = &FCB->ConnectIrp;
 228
 229     /* Cancel our pending requests */
 230     for( i = 0; i < IN_FLIGHT_REQUESTS; i++ ) {
 231         if( InFlightRequest[i]->InFlightRequest ) {
 232             AFD_DbgPrint(MID_TRACE,("Cancelling in flight irp %d (%x)\n",
 233                                     i, InFlightRequest[i]->InFlightRequest));
 234             IoCancelIrp(InFlightRequest[i]->InFlightRequest);
 235         }
 236     }
 237
 238     KillSelectsForFCB( FCB->DeviceExt, FileObject, FALSE );
 239
 240     SocketStateUnlock( FCB );
 241
 242     if( FCB->EventSelect )
 243         ObDereferenceObject( FCB->EventSelect );
 244
 245     if( FCB->Context )
 246         ExFreePool( FCB->Context );
 247
 248     if( FCB->Recv.Window )
 249         ExFreePool( FCB->Recv.Window );
 250
 251     if( FCB->Send.Window )
 252         ExFreePool( FCB->Send.Window );
 253
 254     if( FCB->AddressFrom )
 255         ExFreePool( FCB->AddressFrom );
 256
 257     if( FCB->LocalAddress )
 258         ExFreePool( FCB->LocalAddress );
 259
 260     if( FCB->RemoteAddress )
 261         ExFreePool( FCB->RemoteAddress );
 262
 263     if( FCB->Connection.Object )
 264         ObDereferenceObject(FCB->Connection.Object);
 265
 266     if( FCB->AddressFile.Object )
 267         ObDereferenceObject(FCB->AddressFile.Object);
 268
 269     if( FCB->AddressFile.Handle != INVALID_HANDLE_VALUE )
 270     {
 271         if (ZwClose(FCB->AddressFile.Handle) == STATUS_INVALID_HANDLE)
 272         {
 273             DbgPrint("INVALID ADDRESS FILE HANDLE VALUE: %x %x\n", FCB->AddressFile.Handle, FCB->AddressFile.Object);
 274         }
 275     }
 276
 277     if( FCB->Connection.Handle != INVALID_HANDLE_VALUE )
 278     {
 279         if (ZwClose(FCB->Connection.Handle) == STATUS_INVALID_HANDLE)
 280         {
 281             DbgPrint("INVALID CONNECTION HANDLE VALUE: %x %x\n", FCB->Connection.Handle, FCB->Connection.Object);
 282         }
 283     }
 284
 285     if( FCB->TdiDeviceName.Buffer )
 286         ExFreePool(FCB->TdiDeviceName.Buffer);
 287
 288     ExFreePool(FCB);
 289
 290     Irp->IoStatus.Status = STATUS_SUCCESS;
 291     Irp->IoStatus.Information = 0;
 292     IoCompleteRequest(Irp, IO_NETWORK_INCREMENT);
 293
 294     AFD_DbgPrint(MID_TRACE, ("Returning success.\n"));
 295
 296     return STATUS_SUCCESS;
 297 }
 298
 299 static NTSTATUS NTAPI
 300 AfdDisconnect(PDEVICE_OBJECT DeviceObject, PIRP Irp,
 301               PIO_STACK_LOCATION IrpSp) {
 302     PFILE_OBJECT FileObject = IrpSp->FileObject;
 303     PAFD_FCB FCB = FileObject->FsContext;
 304     PAFD_DISCONNECT_INFO DisReq;
 305     IO_STATUS_BLOCK Iosb;
 306     PTDI_CONNECTION_INFORMATION ConnInfo;
 307     NTSTATUS Status;
 308     USHORT Flags = 0;
 309
 310     if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
 311
 312     if( !(DisReq = LockRequest( Irp, IrpSp )) )
 313         return UnlockAndMaybeComplete( FCB, STATUS_NO_MEMORY,
 314                                        Irp, 0 );
 315
 316     if (NULL == FCB->RemoteAddress)
 317       {
 318         ConnInfo = NULL;
 319       }
 320     else
 321       {
 322         Status = TdiBuildNullConnectionInfo
 323             ( &ConnInfo, FCB->RemoteAddress->Address[0].AddressType );
 324
 325         if( !NT_SUCCESS(Status) || !ConnInfo )
 326             return UnlockAndMaybeComplete( FCB, STATUS_NO_MEMORY,
 327                                            Irp, 0 );
 328       }
 329
 330     if( DisReq->DisconnectType & AFD_DISCONNECT_SEND )
 331         Flags |= TDI_DISCONNECT_RELEASE;
 332     if( DisReq->DisconnectType & AFD_DISCONNECT_RECV ||
 333         DisReq->DisconnectType & AFD_DISCONNECT_ABORT )
 334         Flags |= TDI_DISCONNECT_ABORT;
 335
 336     Status = TdiDisconnect( FCB->Connection.Object,
 337                             &DisReq->Timeout,
 338                             Flags,
 339                             &Iosb,
 340                             NULL,
 341                             NULL,
 342                             FCB->AddressFrom,
 343                             ConnInfo);
 344
 345     if (ConnInfo) ExFreePool( ConnInfo );
 346
 347     FCB->PollState |= AFD_EVENT_DISCONNECT;
 348     PollReeval( FCB->DeviceExt, FCB->FileObject );
 349
 350     return UnlockAndMaybeComplete( FCB, Status, Irp, 0 );
 351 }
 352
 353 static NTSTATUS NTAPI
 354 AfdDispatch(PDEVICE_OBJECT DeviceObject, PIRP Irp)
 355 {
 356     PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
 357     NTSTATUS Status = STATUS_SUCCESS;
 358 #if DBG
 359     PFILE_OBJECT FileObject = IrpSp->FileObject;
 360 #endif
 361
 362     AFD_DbgPrint(MID_TRACE,("AfdDispatch: %d\n", IrpSp->MajorFunction));
 363     if( IrpSp->MajorFunction != IRP_MJ_CREATE) {
 364         AFD_DbgPrint(MID_TRACE,("FO %x, IrpSp->FO %x\n",
 365                                 FileObject, IrpSp->FileObject));
 366         ASSERT(FileObject == IrpSp->FileObject);
 367     }
 368
 369     Irp->IoStatus.Information = 0;
 370
 371     switch(IrpSp->MajorFunction)
 372     {
 373         /* opening and closing handles to the device */
 374     case IRP_MJ_CREATE:
 375         /* Mostly borrowed from the named pipe file system */
 376         return AfdCreateSocket(DeviceObject, Irp, IrpSp);
 377
 378     case IRP_MJ_CLOSE:
 379         /* Ditto the borrowing */
 380         return AfdCloseSocket(DeviceObject, Irp, IrpSp);
 381
 382     case IRP_MJ_CLEANUP:
 383         return AfdCleanupSocket(DeviceObject, Irp, IrpSp);
 384
 385     /* write data */
 386     case IRP_MJ_WRITE:
 387         return AfdConnectedSocketWriteData( DeviceObject, Irp, IrpSp, TRUE );
 388
 389     /* read data */
 390     case IRP_MJ_READ:
 391         return AfdConnectedSocketReadData( DeviceObject, Irp, IrpSp, TRUE );
 392
 393     case IRP_MJ_DEVICE_CONTROL:
 394     {
 395         switch( IrpSp->Parameters.DeviceIoControl.IoControlCode ) {
 396         case IOCTL_AFD_BIND:
 397             return AfdBindSocket( DeviceObject, Irp, IrpSp );
 398
 399         case IOCTL_AFD_CONNECT:
 400             return AfdStreamSocketConnect( DeviceObject, Irp, IrpSp );
 401
 402         case IOCTL_AFD_START_LISTEN:
 403             return AfdListenSocket( DeviceObject, Irp, IrpSp );
 404
 405         case IOCTL_AFD_RECV:
 406             return AfdConnectedSocketReadData( DeviceObject, Irp, IrpSp,
 407                                                FALSE );
 408
 409         case IOCTL_AFD_SELECT:
 410             return AfdSelect( DeviceObject, Irp, IrpSp );
 411
 412         case IOCTL_AFD_EVENT_SELECT:
 413             return AfdEventSelect( DeviceObject, Irp, IrpSp );
 414
 415         case IOCTL_AFD_ENUM_NETWORK_EVENTS:
 416             return AfdEnumEvents( DeviceObject, Irp, IrpSp );
 417
 418         case IOCTL_AFD_RECV_DATAGRAM:
 419             return AfdPacketSocketReadData( DeviceObject, Irp, IrpSp );
 420
 421         case IOCTL_AFD_SEND:
 422             return AfdConnectedSocketWriteData( DeviceObject, Irp, IrpSp,
 423                                                 FALSE );
 424
 425         case IOCTL_AFD_SEND_DATAGRAM:
 426             return AfdPacketSocketWriteData( DeviceObject, Irp, IrpSp );
 427
 428         case IOCTL_AFD_GET_INFO:
 429             return AfdGetInfo( DeviceObject, Irp, IrpSp );
 430
 431         case IOCTL_AFD_SET_INFO:
 432             return AfdSetInfo( DeviceObject, Irp, IrpSp );
 433
 434         case IOCTL_AFD_GET_CONTEXT:
 435             return AfdGetContext( DeviceObject, Irp, IrpSp );
 436
 437         case IOCTL_AFD_SET_CONTEXT:
 438             return AfdSetContext( DeviceObject, Irp, IrpSp );
 439
 440         case IOCTL_AFD_WAIT_FOR_LISTEN:
 441             return AfdWaitForListen( DeviceObject, Irp, IrpSp );
 442
 443         case IOCTL_AFD_ACCEPT:
 444             return AfdAccept( DeviceObject, Irp, IrpSp );
 445
 446         case IOCTL_AFD_DISCONNECT:
 447             return AfdDisconnect( DeviceObject, Irp, IrpSp );
 448
 449         case IOCTL_AFD_GET_SOCK_NAME:
 450             return AfdGetSockName( DeviceObject, Irp, IrpSp );
 451
 452         case IOCTL_AFD_GET_PEER_NAME:
 453             return AfdGetPeerName( DeviceObject, Irp, IrpSp );
 454
 455         case IOCTL_AFD_GET_TDI_HANDLES:
 456             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_TDI_HANDLES\n"));
 457             break;
 458
 459         case IOCTL_AFD_SET_CONNECT_DATA:
 460             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_DATA\n"));
 461             break;
 462
 463         case IOCTL_AFD_SET_CONNECT_OPTIONS:
 464             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_OPTIONS\n"));
 465             break;
 466
 467         case IOCTL_AFD_SET_DISCONNECT_DATA:
 468             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_DATA\n"));
 469             break;
 470
 471         case IOCTL_AFD_SET_DISCONNECT_OPTIONS:
 472             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_OPTIONS\n"));
 473             break;
 474
 475         case IOCTL_AFD_GET_CONNECT_DATA:
 476             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_CONNECT_DATA\n"));
 477             break;
 478
 479         case IOCTL_AFD_GET_CONNECT_OPTIONS:
 480             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_CONNECT_OPTIONS\n"));
 481             break;
 482
 483         case IOCTL_AFD_GET_DISCONNECT_DATA:
 484             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_DISCONNECT_DATA\n"));
 485             break;
 486
 487         case IOCTL_AFD_GET_DISCONNECT_OPTIONS:
 488             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_DISCONNECT_OPTIONS\n"));
 489             break;
 490
 491         case IOCTL_AFD_SET_CONNECT_DATA_SIZE:
 492             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_DATA_SIZE\n"));
 493             break;
 494
 495         case IOCTL_AFD_SET_CONNECT_OPTIONS_SIZE:
 496             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_CONNECT_OPTIONS_SIZE\n"));
 497             break;
 498
 499         case IOCTL_AFD_SET_DISCONNECT_DATA_SIZE:
 500             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_DATA_SIZE\n"));
 501             break;
 502
 503         case IOCTL_AFD_SET_DISCONNECT_OPTIONS_SIZE:
 504             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_SET_DISCONNECT_OPTIONS_SIZE\n"));
 505             break;
 506
 507         case IOCTL_AFD_DEFER_ACCEPT:
 508             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_DEFER_ACCEPT\n"));
 509             break;
 510
 511         case IOCTL_AFD_GET_PENDING_CONNECT_DATA:
 512             AFD_DbgPrint(MIN_TRACE, ("IOCTL_AFD_GET_PENDING_CONNECT_DATA\n"));
 513             break;
 514
 515         default:
 516             Status = STATUS_NOT_IMPLEMENTED;
 517             AFD_DbgPrint(MIN_TRACE, ("Unknown IOCTL (0x%x)\n",
 518                                      IrpSp->Parameters.DeviceIoControl.
 519                                      IoControlCode));
 520             break;
 521         }
 522         break;
 523     }
 524
 525 /* unsupported operations */
 526     default:
 527     {
 528         Status = STATUS_NOT_IMPLEMENTED;
 529         AFD_DbgPrint(MIN_TRACE,
 530                      ("Irp: Unknown Major code was %x\n",
 531                       IrpSp->MajorFunction));
 532         break;
 533     }
 534     }
 535
 536     AFD_DbgPrint(MID_TRACE, ("Returning %x\n", Status));
 537     Irp->IoStatus.Status = Status;
 538     IoCompleteRequest(Irp, IO_NO_INCREMENT);
 539
 540     return (Status);
 541 }
 542
 543 VOID NTAPI
 544 AfdCancelHandler(PDEVICE_OBJECT DeviceObject,
 545                  PIRP Irp)
 546 {
 547     PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
 548     PFILE_OBJECT FileObject = IrpSp->FileObject;
 549     PAFD_FCB FCB = FileObject->FsContext;
 550     UINT Function;
 551     PAFD_RECV_INFO RecvReq;
 552     PAFD_SEND_INFO SendReq;
 553     PLIST_ENTRY CurrentEntry;
 554     PIRP CurrentIrp;
 555     PAFD_DEVICE_EXTENSION DeviceExt = DeviceObject->DeviceExtension;
 556     KIRQL OldIrql;
 557     PAFD_ACTIVE_POLL Poll;
 558     PAFD_POLL_INFO PollReq;
 559
 560     IoReleaseCancelSpinLock(Irp->CancelIrql);
 561
 562     if (!SocketAcquireStateLock(FCB))
 563         return;
 564
 565     ASSERT(IrpSp->MajorFunction == IRP_MJ_DEVICE_CONTROL);
 566
 567     switch (IrpSp->Parameters.DeviceIoControl.IoControlCode)
 568     {
 569         case IOCTL_AFD_RECV:
 570         RecvReq = IrpSp->Parameters.DeviceIoControl.Type3InputBuffer;
 571         UnlockBuffers(RecvReq->BufferArray, RecvReq->BufferCount, FALSE);
 572         /* Fall through */
 573
 574         case IOCTL_AFD_RECV_DATAGRAM:
 575         Function = FUNCTION_RECV;
 576         break;
 577
 578         case IOCTL_AFD_SEND:
 579         SendReq = IrpSp->Parameters.DeviceIoControl.Type3InputBuffer;
 580         UnlockBuffers(SendReq->BufferArray, SendReq->BufferCount, FALSE);
 581         /* Fall through */
 582
 583         case IOCTL_AFD_SEND_DATAGRAM:
 584         Function = FUNCTION_SEND;
 585         break;
 586
 587         case IOCTL_AFD_CONNECT:
 588         Function = FUNCTION_CONNECT;
 589         break;
 590
 591         case IOCTL_AFD_WAIT_FOR_LISTEN:
 592         Function = FUNCTION_PREACCEPT;
 593         break;
 594
 595         case IOCTL_AFD_SELECT:
 596         KeAcquireSpinLock(&DeviceExt->Lock, &OldIrql);
 597
 598         CurrentEntry = DeviceExt->Polls.Flink;
 599         while (CurrentEntry != &DeviceExt->Polls)
 600         {
 601             Poll = CONTAINING_RECORD(CurrentEntry, AFD_ACTIVE_POLL, ListEntry);
 602             CurrentIrp = Poll->Irp;
 603             PollReq = CurrentIrp->AssociatedIrp.SystemBuffer;
 604
 605             if (CurrentIrp == Irp)
 606             {
 607                 ZeroEvents(PollReq->Handles, PollReq->HandleCount);
 608                 SignalSocket(Poll, NULL, PollReq, STATUS_CANCELLED);
 609                 break;
 610             }
 611             else
 612             {
 613                 CurrentEntry = CurrentEntry->Flink;
 614             }
 615         }
 616
 617         KeReleaseSpinLock(&DeviceExt->Lock, OldIrql);
 618
 619         /* IRP already completed by SignalSocket */
 620         SocketStateUnlock(FCB);
 621         return;
 622
 623         default:
 624         ASSERT(FALSE);
 625         UnlockAndMaybeComplete(FCB, STATUS_CANCELLED, Irp, 0);
 626         return;
 627     }
 628
 629     CurrentEntry = FCB->PendingIrpList[Function].Flink;
 630     while (CurrentEntry != &FCB->PendingIrpList[Function])
 631     {
 632         CurrentIrp = CONTAINING_RECORD(CurrentEntry, IRP, Tail.Overlay.ListEntry);
 633
 634         if (CurrentIrp == Irp)
 635         {
 636             RemoveEntryList(CurrentEntry);
 637             break;
 638         }
 639         else
 640         {
 641             CurrentEntry = CurrentEntry->Flink;
 642         }
 643     }
 644
 645     UnlockAndMaybeComplete(FCB, STATUS_CANCELLED, Irp, 0);
 646 }
 647
 648 static VOID NTAPI
 649 AfdUnload(PDRIVER_OBJECT DriverObject)
 650 {
 651 }
 652
 653 NTSTATUS NTAPI
 654 DriverEntry(PDRIVER_OBJECT DriverObject, PUNICODE_STRING RegistryPath)
 655 {
 656     PDEVICE_OBJECT DeviceObject;
 657     UNICODE_STRING wstrDeviceName = RTL_CONSTANT_STRING(L"\\Device\\Afd");
 658     PAFD_DEVICE_EXTENSION DeviceExt;
 659     NTSTATUS Status;
 660
 661     /* register driver routines */
 662     DriverObject->MajorFunction[IRP_MJ_CLOSE] = AfdDispatch;
 663     DriverObject->MajorFunction[IRP_MJ_CREATE] = AfdDispatch;
 664     DriverObject->MajorFunction[IRP_MJ_CLEANUP] = AfdDispatch;
 665     DriverObject->MajorFunction[IRP_MJ_WRITE] = AfdDispatch;
 666     DriverObject->MajorFunction[IRP_MJ_READ] = AfdDispatch;
 667     DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = AfdDispatch;
 668     DriverObject->DriverUnload = AfdUnload;
 669
 670     Status = IoCreateDevice
 671         ( DriverObject,
 672           sizeof(AFD_DEVICE_EXTENSION),
 673           &wstrDeviceName,
 674           FILE_DEVICE_NAMED_PIPE,
 675           0,
 676           FALSE,
 677           &DeviceObject );
 678
 679     /* failure */
 680     if(!NT_SUCCESS(Status))
 681     {
 682         return (Status);
 683     }
 684
 685     DeviceExt = DeviceObject->DeviceExtension;
 686     KeInitializeSpinLock( &DeviceExt->Lock );
 687     InitializeListHead( &DeviceExt->Polls );
 688
 689     AFD_DbgPrint(MID_TRACE,("Device created: object %x ext %x\n",
 690                             DeviceObject, DeviceExt));
 691
 692     return (Status);
 693 }
 694
 695 /* EOF */