[ros-diffs] [arty] 37759: fix a bug reported by aicom: get peer name fails running xitami web server

[arty at svn.reactos.org]

Author: arty
Date: Sun Nov 30 02:26:52 2008
New Revision: 37759

URL: http://svn.reactos.org/svn/reactos?rev=37759&view=rev
Log:
fix a bug reported by aicom: get peer name fails running xitami web server

Modified:
    trunk/reactos/drivers/network/afd/afd.rbuild
    trunk/reactos/drivers/network/afd/afd/info.c
    trunk/reactos/drivers/network/afd/include/afd.h

Modified: trunk/reactos/drivers/network/afd/afd.rbuild
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/network/afd/afd.rbuild?rev=37759&r1=37758&r2=37759&view=diff
==============================================================================
--- trunk/reactos/drivers/network/afd/afd.rbuild [iso-8859-1] (original)
+++ trunk/reactos/drivers/network/afd/afd.rbuild [iso-8859-1] Sun Nov 30 02:26:52 2008
@@ -3,6 +3,7 @@
 <module name="afd" type="kernelmodedriver" installbase="system32/drivers" installname="afd.sys">
 	<include base="afd">include</include>
 	<include base="ReactOS">include/reactos/drivers</include>
+	<include base="ReactOS">include/ndk</include>
 	<library>pseh</library>
 	<library>ntoskrnl</library>
 	<library>hal</library>

Modified: trunk/reactos/drivers/network/afd/afd/info.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/network/afd/afd/info.c?rev=37759&r1=37758&r2=37759&view=diff
==============================================================================
--- trunk/reactos/drivers/network/afd/afd/info.c [iso-8859-1] (original)
+++ trunk/reactos/drivers/network/afd/afd/info.c [iso-8859-1] Sun Nov 30 02:26:52 2008
@@ -78,6 +78,10 @@
     PMDL Mdl = NULL, SysMdl = NULL;
     PTDI_CONNECTION_INFORMATION ConnInfo = NULL;
     PTRANSPORT_ADDRESS TransAddr = NULL;
+    HANDLE ProcHandle = NULL;
+    BOOLEAN UnlockSysMdl = FALSE;
+    PVOID UserSpace = NULL;
+    ULONG Length, InOutLength;
 
     AFD_DbgPrint(MID_TRACE,("Called on %x\n", FCB));
 
@@ -110,28 +114,67 @@
                       TDI_QUERY_ADDRESS_INFO,
                       Mdl );
             } else {
+                // what follows is fucked up shit.
+                // i'm not sure how to avoid it
+                // sorry
+                // -- arty
+
                 if( FCB->Connection.Object == NULL || (FCB->State != SOCKET_STATE_BOUND && FCB->State != SOCKET_STATE_CONNECTED) ) {
 	            return UnlockAndMaybeComplete( FCB, STATUS_INVALID_PARAMETER, Irp, 0,
 	                                           NULL );
                 }
-
+		
                 if( NT_SUCCESS
                     ( Status = TdiBuildNullConnectionInfo
                       ( &ConnInfo,
-                        FCB->LocalAddress->Address[0].AddressType ) ) ) {
-                    SysMdl = IoAllocateMdl
-                        ( ConnInfo,
-                          sizeof( TDI_CONNECTION_INFORMATION ) +
-                          TaLengthOfTransportAddress
-                          ( ConnInfo->RemoteAddress ),
-                          FALSE,
-                          FALSE,
-                          NULL );
-                }
-
+                        FCB->RemoteAddress->Address[0].AddressType ) ) ) {
+		    
+		    Length = TaLengthOfTransportAddress
+			(ConnInfo->RemoteAddress);
+		    
+		    if (NT_SUCCESS(Status))
+			Status = ObOpenObjectByPointer
+			    (PsGetCurrentProcess(),
+			     0,
+			     NULL,
+			     PROCESS_ALL_ACCESS,
+			     PsProcessType,
+			     KernelMode,
+			     &ProcHandle);
+		    
+		    if (NT_SUCCESS(Status))
+		    {
+			InOutLength = 
+			    PAGE_ROUND_UP(sizeof(TDI_CONNECTION_INFO));
+			
+			Status = NtAllocateVirtualMemory
+			    (ProcHandle,
+			     (PVOID*)&UserSpace,
+			     PAGE_SHIFT,
+			     &InOutLength, 
+			     MEM_COMMIT,
+			     PAGE_READWRITE);
+		    }
+
+		    if (NT_SUCCESS(Status))
+		    {
+			ExFreePool(ConnInfo);
+			ConnInfo = (PTDI_CONNECTION_INFORMATION)UserSpace;
+			
+			SysMdl = IoAllocateMdl
+			    ( UserSpace, Length, FALSE, FALSE, NULL );
+		    }
+		    else
+		    {
+			ExFreePool(ConnInfo);
+			ConnInfo = NULL;
+		    }
+		}
+	    
                 if( SysMdl ) {
                     _SEH_TRY {
                         MmProbeAndLockPages( SysMdl, Irp->RequestorMode, IoModifyAccess );
+			UnlockSysMdl = TRUE;
                     } _SEH_HANDLE {
 	                AFD_DbgPrint(MIN_TRACE, ("MmProbeAndLockPages() failed.\n"));
 	                Status = _SEH_GetExceptionCode();
@@ -156,8 +199,17 @@
                     else Status = STATUS_INSUFFICIENT_RESOURCES;
 		}
 
-                if( ConnInfo ) ExFreePool( ConnInfo );
+		if (UnlockSysMdl)
+		    MmUnlockPages( SysMdl );
+
                 if( SysMdl ) IoFreeMdl( SysMdl );
+                if( ConnInfo ) 
+		    NtFreeVirtualMemory
+			( ProcHandle,
+			  (PVOID)ConnInfo, 
+			  &InOutLength,
+			  MEM_RELEASE );
+		if( ProcHandle ) NtClose(ProcHandle);
                 if( TransAddr ) MmUnmapLockedPages( TransAddr, Mdl );
                 MmUnlockPages( Mdl );
                 IoFreeMdl( Mdl );

Modified: trunk/reactos/drivers/network/afd/include/afd.h
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/network/afd/include/afd.h?rev=37759&r1=37758&r2=37759&view=diff
==============================================================================
--- trunk/reactos/drivers/network/afd/include/afd.h [iso-8859-1] (original)
+++ trunk/reactos/drivers/network/afd/include/afd.h [iso-8859-1] Sun Nov 30 02:26:52 2008
@@ -13,6 +13,12 @@
 #define _AFD_H
 
 #include <ntddk.h>
+#include <winddk.h>
+#include <ntifs.h>
+#include <ndk/obtypes.h>
+#include <ndk/obfuncs.h>
+#include <ndk/mmtypes.h>
+#include <ndk/mmfuncs.h>
 #include <tdi.h>
 #include <tdikrnl.h>
 #include <tdiinfo.h>