[arty] 13089: Assert writable on a new socket.

Assert writable on a new socket.
Modified: trunk/reactos/drivers/net/afd/afd/listen.c
  _____

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 );