ULONG AllocatePortFromRange( PPORT_SET PortSet, ULONG Lowest, ULONG Highest ) {
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;
}
ULONG AllocateAnyPort( PPORT_SET PortSet ) {
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;
}
void
Test_RtlFindClearBits(void)
{
RTL_BITMAP BitMapHeader;
ULONG *Buffer;
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
Buffer[0] = 0x060F874D;
Buffer[1] = 0x3F303F30;
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 0), 0);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 3), 0);
ok_int(RtlFindClearBits(&BitMapHeader, 1, 0), -1);
ok_int(RtlFindClearBits(&BitMapHeader, 1, 1), -1);
RtlInitializeBitMap(&BitMapHeader, Buffer, 8);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 3), 0);
ok_int(RtlFindClearBits(&BitMapHeader, 1, 0), 1);
ok_int(RtlFindClearBits(&BitMapHeader, 1, 1), 1);
ok_int(RtlFindClearBits(&BitMapHeader, 1, 2), 4);
ok_int(RtlFindClearBits(&BitMapHeader, 2, 0), 4);
ok_int(RtlFindClearBits(&BitMapHeader, 3, 0), -1);
RtlInitializeBitMap(&BitMapHeader, Buffer, 32);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 3), 0);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 21), 16);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 12), 8);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 31), 24);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 32), 0);
ok_int(RtlFindClearBits(&BitMapHeader, 0, 39), 0);
ok_int(RtlFindClearBits(&BitMapHeader, 4, 0), 11);
ok_int(RtlFindClearBits(&BitMapHeader, 5, 0), 20);
ok_int(RtlFindClearBits(&BitMapHeader, 4, 11), 11);
ok_int(RtlFindClearBits(&BitMapHeader, 4, 12), 20);
ok_int(RtlFindClearBits(&BitMapHeader, 2, 11), 11);
ok_int(RtlFindClearBits(&BitMapHeader, 2, 12), 12);
ok_int(RtlFindClearBits(&BitMapHeader, 1, 32), 1);
ok_int(RtlFindClearBits(&BitMapHeader, 4, 32), 11);
ok_int(RtlFindClearBits(&BitMapHeader, 5, 32), 20);
RtlInitializeBitMap(&BitMapHeader, Buffer, 64);
ok_int(RtlFindClearBits(&BitMapHeader, 5, 64), 20);
ok_int(RtlFindClearBits(&BitMapHeader, 9, 28), 27);
ok_int(RtlFindClearBits(&BitMapHeader, 10, 0), -1);
Buffer[1] = 0xFF303F30;
ok_int(RtlFindClearBits(&BitMapHeader, 1, 56), 1);
FreeGuarded(Buffer);
}
ULONG NtdllBitmap::FindClearBits( ULONG number_to_find, ULONG hint_index )
{
assert(RtlFindClearBits != NULL);
return RtlFindClearBits(this, number_to_find, hint_index);
}
