From 17c1d9b828eb9546243c7f152ed53b4108767f5e Mon Sep 17 00:00:00 2001 From: Cameron Gutman Date: Sat, 30 Aug 2008 02:04:30 +0000 Subject: [PATCH] - Check that we successfully allocated memory - Return TDI_NO_RESOURCES instead of STATUS_NO_MEMORY svn path=/branches/aicom-network-fixes/; revision=35783 --- drivers/network/tcpip/tcpip/ninfo.c | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/drivers/network/tcpip/tcpip/ninfo.c b/drivers/network/tcpip/tcpip/ninfo.c index c32822447fb..f39b35b5929 100644 --- a/drivers/network/tcpip/tcpip/ninfo.c +++ b/drivers/network/tcpip/tcpip/ninfo.c @@ -27,6 +27,9 @@ TDI_STATUS InfoTdiQueryGetAddrTable( PNDIS_BUFFER Buffer, TI_DbgPrint(DEBUG_INFO, ("Called.\n")); + if (!IpAddress) + return TDI_NO_RESOURCES; + TcpipAcquireSpinLock(&InterfaceListLock, &OldIrql); ForEachInterface(CurrentIF) { @@ -80,7 +83,7 @@ TDI_STATUS InfoTdiQueryGetRouteTable( PNDIS_BUFFER Buffer, PUINT BufferSize ) { if( !RCache || !RouteEntries ) { if( RCache ) ExFreePool( RCache ); if( RouteEntries ) ExFreePool( RouteEntries ); - return STATUS_NO_MEMORY; + return TDI_NO_RESOURCES; } RtlZeroMemory( RouteEntries, Size );