Assert writable on a new socket.

Modified: trunk/reactos/drivers/net/afd/afd/listen.c

--- trunk/reactos/drivers/net/afd/afd/listen.c	2005-01-16 22:53:36 UTC (rev 13088)
+++ trunk/reactos/drivers/net/afd/afd/listen.c	2005-01-17 01:30:26 UTC (rev 13089)
@@ -12,7 +12,9 @@

 #include "tdiconn.h"
 #include "debug.h"
 

-VOID SatisfyAccept( PIRP Irp, PFILE_OBJECT NewFileObject,

+VOID SatisfyAccept( PAFD_DEVICE_EXTENSION DeviceExt,
+                    PIRP Irp, 
+                    PFILE_OBJECT NewFileObject,

 		    PAFD_TDI_OBJECT_QELT Qelt ) {
     PAFD_FCB FCB = NewFileObject->FsContext;
 
@@ -31,7 +33,9 @@

     IoCompleteRequest( Irp, IO_NETWORK_INCREMENT );
 
     MakeSocketIntoConnection( FCB );

-

+    FCB->PollState |= AFD_EVENT_SEND;
+    PollReeval( DeviceExt, NewFileObject );
+    

     SocketStateUnlock( FCB );
 }
 
@@ -131,8 +135,6 @@

     if( !IsListEmpty( &FCB->PendingConnections ) ) {
 	FCB->PollState |= AFD_EVENT_ACCEPT;
 	PollReeval( FCB->DeviceExt, FCB->FileObject );

-    } else {
-	FCB->PollState &= ~AFD_EVENT_ACCEPT;

     }
 
     SocketStateUnlock( FCB );
@@ -214,6 +216,9 @@

 
 	AFD_DbgPrint(MID_TRACE,("Completed a wait for accept\n"));
 

+        FCB->PollState &= ~AFD_EVENT_ACCEPT;
+        PollReeval( FCB->DeviceExt, FCB->FileObject );
+

 	SocketStateUnlock( FCB );
 	return Status;
     } else {
@@ -227,6 +232,8 @@

 		    PIO_STACK_LOCATION IrpSp ) {
     NTSTATUS Status = STATUS_SUCCESS;
     PFILE_OBJECT FileObject = IrpSp->FileObject;

+    PAFD_DEVICE_EXTENSION DeviceExt = 
+        (PAFD_DEVICE_EXTENSION)DeviceObject->DeviceExtension;

     PAFD_FCB FCB = FileObject->FsContext;
     PAFD_ACCEPT_DATA AcceptData = Irp->AssociatedIrp.SystemBuffer;
     PLIST_ENTRY PendingConn;
@@ -284,7 +291,7 @@

             ASSERT(NewFileObject->FsContext != FCB);
 
 	    /* We have a pending connection ... complete this irp right away */

-	    SatisfyAccept( Irp, NewFileObject, PendingConnObj );

+	    SatisfyAccept( DeviceExt, Irp, NewFileObject, PendingConnObj );

 	    
 	    ObDereferenceObject( NewFileObject );