Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-08-04 10:30:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

[AFD]

Browse source 
- Handle canceled reads on datagram sockets

svn path=/trunk/; revision=53145
This commit is contained in:
Cameron Gutman 2011-08-08 22:02:55 +00:00
parent 44bc45f23c
commit 51388e8603
2 changed files with 16 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
reactos/drivers/network/afd/afd/main.c
View file

@ -693,23 +693,19 @@ AfdDisconnect(PDEVICE_OBJECT DeviceObject, PIRP Irp,
/* Mark that we can't issue another receive request */
FCB->TdiReceiveClosed = TRUE;
/* These are only for connection-oriented sockets */
if (!(FCB->Flags & AFD_ENDPOINT_CONNECTIONLESS))
{
/* Try to cancel a pending TDI receive IRP if there was one in progress */
if (FCB->ReceiveIrp.InFlightRequest)
IoCancelIrp(FCB->ReceiveIrp.InFlightRequest);
/* Try to cancel a pending TDI receive IRP if there was one in progress */
if (FCB->ReceiveIrp.InFlightRequest)
IoCancelIrp(FCB->ReceiveIrp.InFlightRequest);
/* Discard any pending data */
FCB->Recv.Content = 0;
FCB->Recv.BytesUsed = 0;
/* Discard any pending data */
FCB->Recv.Content = 0;
FCB->Recv.BytesUsed = 0;
/* Mark us as overread to complete future reads with an error */
FCB->Overread = TRUE;
/* Mark us as overread to complete future reads with an error */
FCB->Overread = TRUE;
/* Set a successful close status to indicate a shutdown on overread */
FCB->PollStatus[FD_CLOSE_BIT] = STATUS_SUCCESS;
}
/* Set a successful close status to indicate a shutdown on overread */
FCB->PollStatus[FD_CLOSE_BIT] = STATUS_SUCCESS;
/* Clear the receive event */
FCB->PollState &= ~AFD_EVENT_RECEIVE;

6
reactos/drivers/network/afd/afd/read.c
View file

@ -588,6 +588,12 @@ PacketSocketRecvComplete(
return Irp->IoStatus.Status;
}
if (FCB->TdiReceiveClosed)
{
SocketStateUnlock(FCB);
return STATUS_FILE_CLOSED;
}
DatagramRecv = ExAllocatePool( NonPagedPool, DGSize );
if( DatagramRecv ) {