[arty] 14360: Make sure that the socket structure list does't accumulate duplicates.

Make sure that the socket structure list does't accumulate duplicates.
Modified: trunk/reactos/lib/msafd/include/msafd.h
Modified: trunk/reactos/lib/msafd/misc/dllmain.c
  _____

Modified: trunk/reactos/lib/msafd/include/msafd.h
--- trunk/reactos/lib/msafd/include/msafd.h	2005-03-27 21:34:41 UTC
(rev 14359)
+++ trunk/reactos/lib/msafd/include/msafd.h	2005-03-28 02:50:23 UTC
(rev 14360)
@@ -412,6 +412,8 @@

SOCKET Handle
 );

+VOID DeleteSocketStructure( SOCKET Handle );
+
 int GetSocketInformation(
    PSOCKET_INFORMATION Socket,
    ULONG				AfdInformationClass,
  _____

Modified: trunk/reactos/lib/msafd/misc/dllmain.c
--- trunk/reactos/lib/msafd/misc/dllmain.c	2005-03-27 21:34:41 UTC
(rev 14359)
+++ trunk/reactos/lib/msafd/misc/dllmain.c	2005-03-28 02:50:23 UTC
(rev 14360)
@@ -62,7 +62,7 @@

ULONG						SizeOfEA;
    PAFD_CREATE_PACKET			AfdPacket;
    HANDLE						Sock;
-	PSOCKET_INFORMATION			Socket = NULL;
+	PSOCKET_INFORMATION			Socket = NULL,
PrevSocket = NULL;
     PFILE_FULL_EA_INFORMATION	EABuffer = NULL;
    PHELPER_DATA				HelperData;
    PVOID
HelperDLLContext;
@@ -225,6 +225,16 @@
    /* Save Handle */
    Socket->Handle = (SOCKET)Sock;

+        /* XXX See if there's a structure we can reuse -- We need to do
this
+         * more properly. */
+        PrevSocket = GetSocketStructure( (SOCKET)Sock );
+
+        if( PrevSocket ) {
+            RtlCopyMemory( PrevSocket, Socket, sizeof(*Socket) );
+            RtlFreeHeap( GlobalHeap, 0, Socket );
+            Socket = PrevSocket;
+        }
+
    /* Save Group Info */
    if (g != 0) {
    	GetSocketInformation(Socket, AFD_INFO_GROUP_ID_TYPE, 0,
&GroupData);
@@ -235,13 +245,13 @@

/* Get Window Sizes and Save them */
    GetSocketInformation (Socket,
-
AFD_INFO_SEND_WINDOW_SIZE, 
-
&Socket->SharedData.SizeOfSendBuffer, 
-							NULL);
+                              AFD_INFO_SEND_WINDOW_SIZE, 
+                              &Socket->SharedData.SizeOfSendBuffer, 
+                              NULL);
    GetSocketInformation (Socket, 
-
AFD_INFO_RECEIVE_WINDOW_SIZE, 
-
&Socket->SharedData.SizeOfRecvBuffer, 
-							NULL);
+                              AFD_INFO_RECEIVE_WINDOW_SIZE, 
+                              &Socket->SharedData.SizeOfRecvBuffer, 
+                              NULL);

/* Save in Process Sockets List */
    Sockets[SocketCount] = Socket;
@@ -424,6 +434,7 @@

/* Close the handle */
     NtClose((HANDLE)Handle);
+
     return NO_ERROR;
 }

@@ -591,6 +602,9 @@
    ( readfds ? readfds->fd_count : 0 ) + 
    ( writefds ? writefds->fd_count : 0 ) + 
    ( exceptfds ? exceptfds->fd_count : 0 );
+
+    if( HandleCount < 0 || nfds != 0 ) HandleCount = nfds * 3;
+
     PollBufferSize = sizeof(*PollInfo) + 
    (HandleCount * sizeof(AFD_HANDLE));

@@ -987,7 +1001,7 @@
             RtlCopyMemory (SocketAddress,

&ListenReceiveData->Address.Address[0].AddressType, 
                            sizeof(*RemoteAddress));
-            if( *SocketAddressLength )
+            if( SocketAddressLength )
                 *SocketAddressLength =

ListenReceiveData->Address.Address[0].AddressLength;
         }
@@ -1287,6 +1301,12 @@

SocketAddress->Address[0].Address,

SocketAddress->Address[0].AddressLength);
    		*NameLength = 2 +
SocketAddress->Address[0].AddressLength;
+                        AFD_DbgPrint
+                            (MID_TRACE,
+                             ("NameLength %d Address: %x Port %x\n",
+                              *NameLength, 
+                              ((struct sockaddr_in
*)Name)->sin_addr.s_addr,
+                              ((struct sockaddr_in *)Name)->sin_port));
    		HeapFree(GlobalHeap, 0, TdiAddress);
    		return 0;
    	} else {
@@ -1365,6 +1385,13 @@
    		RtlCopyMemory (Name->sa_data,

SocketAddress->Address[0].Address,

SocketAddress->Address[0].AddressLength);
+			*NameLength = 2 +
SocketAddress->Address[0].AddressLength;
+                        AFD_DbgPrint
+                            (MID_TRACE,
+                             ("NameLength %d Address: %s Port %x\n",
+                              *NameLength, 
+                              ((struct sockaddr_in
*)Name)->sin_addr.s_addr,
+                              ((struct sockaddr_in *)Name)->sin_port));
    		HeapFree(GlobalHeap, 0, TdiAddress);
    		return 0;
    	} else {
@@ -1684,6 +1711,9 @@
    	InfoData.Information.LargeInteger = *LargeInteger;
    }

+        AFD_DbgPrint(MID_TRACE,("XXX Info %x (Data %x)\n",
+                                AfdInformationClass, *Ulong));
+
    /* Send IOCTL */
    Status = NtDeviceIoControlFile( (HANDLE)Socket->Handle,
    				SockEvent,