[ros-diffs] [cgutman] 53028: [AFD] - Handle TDI receive completion better

Author: cgutman
Date: Tue Aug  2 13:46:16 2011
New Revision: 53028

URL: http://svn.reactos.org/svn/reactos?rev=53028&view=rev
Log:
[AFD]
- Handle TDI receive completion better

Modified:
    trunk/reactos/drivers/network/afd/afd/read.c

Modified: trunk/reactos/drivers/network/afd/afd/read.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/network/afd/afd/re…
==============================================================================
--- trunk/reactos/drivers/network/afd/afd/read.c [iso-8859-1] (original)
+++ trunk/reactos/drivers/network/afd/afd/read.c [iso-8859-1] Tue Aug  2 13:46:16 2011
@@ -23,22 +23,38 @@
  */
 #include "afd.h"
 
-static VOID HandleEOFOnIrp( PAFD_FCB FCB, NTSTATUS Status, ULONG_PTR Information )
+static VOID HandleReceiveComplete( PAFD_FCB FCB, NTSTATUS Status, ULONG_PTR Information
)
 {
-	if (Status != STATUS_SUCCESS)
-	{
+    FCB->Recv.BytesUsed = 0;
+
+    /* Receive successful with new data */
+    if (Status == STATUS_SUCCESS && Information)
+    {
+        FCB->Recv.Content = Information;
+    }
+    /* Receive successful with no data (graceful closure) */
+    else if (Status == STATUS_SUCCESS)
+    {
+        FCB->Recv.Content = 0;
         FCB->TdiReceiveClosed = TRUE;
-
+            
+        /* Signal graceful receive shutdown */
+        FCB->PollState |= AFD_EVENT_DISCONNECT;
+        FCB->PollStatus[FD_CLOSE_BIT] = Status;
+            
+        PollReeval( FCB->DeviceExt, FCB->FileObject );
+    }
+    /* Receive failed with no data (unexpected closure) */
+    else
+    {
+        FCB->Recv.Content = 0;
+        FCB->TdiReceiveClosed = TRUE;
+        
         /* Signal complete connection failure immediately */
 		FCB->PollState |= AFD_EVENT_CLOSE;
 		FCB->PollStatus[FD_CLOSE_BIT] = Status;
 
 		PollReeval( FCB->DeviceExt, FCB->FileObject );
-	}
-    else if (Status == STATUS_SUCCESS && !Information)
-    {
-        /* Wait to signal graceful close until all data is read */
-        FCB->TdiReceiveClosed = TRUE;
     }
 }
 
@@ -228,17 +244,6 @@
         FCB->PollStatus[FD_READ_BIT] = STATUS_SUCCESS;
         PollReeval( FCB->DeviceExt, FCB->FileObject );
     }
-    else if (CantReadMore(FCB) &&
-             !(FCB->PollState & (AFD_EVENT_CLOSE | AFD_EVENT_ABORT)))
-    {
-        FCB->PollState &= ~AFD_EVENT_RECEIVE;
-
-        /* Signal graceful receive shutdown */
-        FCB->PollState |= AFD_EVENT_DISCONNECT;
-		FCB->PollStatus[FD_CLOSE_BIT] = STATUS_SUCCESS;
-		
-		PollReeval( FCB->DeviceExt, FCB->FileObject );
-    }
     else
     {
         FCB->PollState &= ~AFD_EVENT_RECEIVE;
@@ -272,9 +277,6 @@
 
     ASSERT(FCB->ReceiveIrp.InFlightRequest == Irp);
     FCB->ReceiveIrp.InFlightRequest = NULL;
-
-    FCB->Recv.Content = Irp->IoStatus.Information;
-    FCB->Recv.BytesUsed = 0;
 
     if( FCB->State == SOCKET_STATE_CLOSED ) {
         /* Cleanup our IRP queue because the FCB is being destroyed */
@@ -298,7 +300,7 @@
         return STATUS_INVALID_PARAMETER;
     }
 	
-	HandleEOFOnIrp( FCB, Irp->IoStatus.Status, Irp->IoStatus.Information );
+	HandleReceiveComplete( FCB, Irp->IoStatus.Status, Irp->IoStatus.Information );
 
 	ReceiveActivity( FCB, NULL );