Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-09-03 09:09:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

- Make PortsStartup return NTSTATUS so we can make sure memory allocated succeeded

Browse source 
- Change the callers to check the return status

svn path=/branches/aicom-network-fixes/; revision=36006
This commit is contained in:
Cameron Gutman 2008-09-06 21:30:09 +00:00
parent 4c942b62d6
commit 2044568591
3 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/drivers/ip/network/ports.c
View file

@ -10,7 +10,7 @@
#include "precomp.h"
VOID PortsStartup( PPORT_SET PortSet,
NTSTATUS PortsStartup( PPORT_SET PortSet,
UINT StartingPort,
UINT PortsToManage ) {
PortSet->StartingPort = StartingPort;
@ -19,11 +19,13 @@ VOID PortsStartup( PPORT_SET PortSet,
PortSet->PortsToOversee - 1;
PortSet->ProtoBitBuffer =
PoolAllocateBuffer( (PortSet->PortsToOversee + 7) / 8 );
if(!PortSet->ProtoBitBuffer) return STATUS_INSUFFICIENT_RESOURCES;
RtlInitializeBitMap( &PortSet->ProtoBitmap,
PortSet->ProtoBitBuffer,
PortSet->PortsToOversee );
RtlClearAllBits( &PortSet->ProtoBitmap );
ExInitializeFastMutex( &PortSet->Mutex );
return STATUS_SUCCESS;
}
VOID PortsShutdown( PPORT_SET PortSet ) {

6
lib/drivers/ip/transport/tcp/tcp.c
View file

@ -455,7 +455,11 @@ NTSTATUS TCPStartup(VOID)
return Status;
}
PortsStartup( &TCPPorts, 1, 0xfffe );
Status = PortsStartup( &TCPPorts, 1, 0xfffe );
if( !NT_SUCCESS(Status) ) {
TCPMemShutdown();
return Status;
}
RegisterOskitTCPEventHandlers( &EventHandlers );
InitOskitTCP();

6
lib/drivers/ip/transport/udp/udp.c
View file

@ -302,8 +302,12 @@ NTSTATUS UDPStartup(
#ifdef __NTDRIVER__
RtlZeroMemory(&UDPStats, sizeof(UDP_STATISTICS));
#endif
NTSTATUS Status;
PortsStartup( &UDPPorts, 1, 0xfffe );
Status = PortsStartup( &UDPPorts, 1, 0xfffe );
if( !NT_SUCCESS(Status) ) return Status;
/* Register this protocol with IP layer */
IPRegisterProtocol(IPPROTO_UDP, UDPReceive);