ports.c File Reference

#include "precomp.h"

Include dependency graph for ports.c:

Go to the source code of this file.

Functions
NTSTATUS	PortsStartup (PPORT_SET PortSet, UINT StartingPort, UINT PortsToManage)
VOID	PortsShutdown (PPORT_SET PortSet)
VOID	DeallocatePort (PPORT_SET PortSet, ULONG Port)
BOOLEAN	AllocatePort (PPORT_SET PortSet, ULONG Port)
ULONG	AllocateAnyPort (PPORT_SET PortSet)
ULONG	AllocatePortFromRange (PPORT_SET PortSet, ULONG Lowest, ULONG Highest)

Function Documentation

AllocateAnyPort()

ULONG AllocateAnyPort

(

PPORT_SET

PortSet

)

Definition at line 69 of file ports.c.

                                           {
    ULONG AllocatedPort;
    KIRQL OldIrql;
 
    KeAcquireSpinLock( &PortSet->Lock, &OldIrql );
    AllocatedPort = RtlFindClearBits( &PortSet->ProtoBitmap, 1, 0 );
    if( AllocatedPort != (ULONG)-1 ) {
    RtlSetBit( &PortSet->ProtoBitmap, AllocatedPort );
    AllocatedPort += PortSet->StartingPort;
    KeReleaseSpinLock( &PortSet->Lock, OldIrql );
    return htons(AllocatedPort);
    }
    KeReleaseSpinLock( &PortSet->Lock, OldIrql );
 
    return -1;
}

AllocatePort()

BOOLEAN AllocatePort	(	PPORT_SET	PortSet,
		ULONG	Port
	)

Definition at line 47 of file ports.c.

                                                      {
    BOOLEAN Clear;
    KIRQL OldIrql;
 
    Port = htons(Port);
 
    if ((Port < PortSet->StartingPort) ||
        (Port >= PortSet->StartingPort + PortSet->PortsToOversee))
    {
       return FALSE;
    }
 
    Port -= PortSet->StartingPort;
 
    KeAcquireSpinLock( &PortSet->Lock, &OldIrql );
    Clear = RtlAreBitsClear( &PortSet->ProtoBitmap, Port, 1 );
    if( Clear ) RtlSetBits( &PortSet->ProtoBitmap, Port, 1 );
    KeReleaseSpinLock( &PortSet->Lock, OldIrql );
 
    return Clear;
}

Referenced by TCPAllocatePort(), and UDPAllocatePort().

AllocatePortFromRange()

ULONG AllocatePortFromRange	(	PPORT_SET	PortSet,
		ULONG	Lowest,
		ULONG	Highest
	)

Definition at line 86 of file ports.c.

                                                                              {
    ULONG AllocatedPort;
    KIRQL OldIrql;
 
    if ((Lowest < PortSet->StartingPort) ||
        (Highest >= PortSet->StartingPort + PortSet->PortsToOversee))
    {
        return -1;
    }
 
    Lowest -= PortSet->StartingPort;
    Highest -= PortSet->StartingPort;
 
    KeAcquireSpinLock( &PortSet->Lock, &OldIrql );
    AllocatedPort = RtlFindClearBits( &PortSet->ProtoBitmap, 1, Lowest );
    if( AllocatedPort != (ULONG)-1 && AllocatedPort <= Highest) {
    RtlSetBit( &PortSet->ProtoBitmap, AllocatedPort );
    AllocatedPort += PortSet->StartingPort;
    KeReleaseSpinLock( &PortSet->Lock, OldIrql );
    return htons(AllocatedPort);
    }
    KeReleaseSpinLock( &PortSet->Lock, OldIrql );
 
    return -1;
}

Referenced by TCPAllocatePort(), and UDPAllocatePort().

DeallocatePort()

VOID DeallocatePort	(	PPORT_SET	PortSet,
		ULONG	Port
	)

Definition at line 35 of file ports.c.

                                                     {
    KIRQL OldIrql;
 
    Port = htons(Port);
    ASSERT(Port >= PortSet->StartingPort);
    ASSERT(Port < PortSet->StartingPort + PortSet->PortsToOversee);
 
    KeAcquireSpinLock( &PortSet->Lock, &OldIrql );
    RtlClearBits( &PortSet->ProtoBitmap, Port - PortSet->StartingPort, 1 );
    KeReleaseSpinLock( &PortSet->Lock, OldIrql );
}

Referenced by TCPFreePort(), and UDPFreePort().

PortsShutdown()

VOID PortsShutdown

(

PPORT_SET

PortSet

)

Definition at line 31 of file ports.c.

                                        {
    ExFreePoolWithTag( PortSet->ProtoBitBuffer, PORT_SET_TAG );
}

Referenced by TCPShutdown(), and UDPShutdown().

PortsStartup()

NTSTATUS PortsStartup	(	PPORT_SET	PortSet,
		UINT	StartingPort,
		UINT	PortsToManage
	)

Definition at line 13 of file ports.c.

                                {
    PortSet->StartingPort = StartingPort;
    PortSet->PortsToOversee = PortsToManage;
 
    PortSet->ProtoBitBuffer =
    ExAllocatePoolWithTag( NonPagedPool, (PortSet->PortsToOversee + 7) / 8,
                               PORT_SET_TAG );
    if(!PortSet->ProtoBitBuffer) return STATUS_INSUFFICIENT_RESOURCES;
    RtlInitializeBitMap( &PortSet->ProtoBitmap,
             PortSet->ProtoBitBuffer,
             PortSet->PortsToOversee );
    RtlClearAllBits( &PortSet->ProtoBitmap );
    KeInitializeSpinLock( &PortSet->Lock );
    return STATUS_SUCCESS;
}

Referenced by TCPStartup(), and UDPStartup().