mirror of
https://github.com/reactos/reactos.git
synced 2024-10-30 11:35:58 +00:00
268d1c62fe
- Fix a buffer overflow in RtlSetBits/RtlClearBits svn path=/trunk/; revision=64867
647 lines
19 KiB
C
647 lines
19 KiB
C
|
|
#include <apitest.h>
|
|
|
|
#define WIN32_NO_STATUS
|
|
#include <ndk/mmfuncs.h>
|
|
#include <ndk/rtlfuncs.h>
|
|
|
|
static
|
|
PVOID
|
|
AllocateGuarded(
|
|
_In_ SIZE_T SizeRequested)
|
|
{
|
|
NTSTATUS Status;
|
|
SIZE_T Size = PAGE_ROUND_UP(SizeRequested + PAGE_SIZE);
|
|
PVOID VirtualMemory = NULL;
|
|
PCHAR StartOfBuffer;
|
|
|
|
Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_RESERVE, PAGE_NOACCESS);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
return NULL;
|
|
|
|
Size -= PAGE_SIZE;
|
|
if (Size)
|
|
{
|
|
Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_COMMIT, PAGE_READWRITE);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
Size = 0;
|
|
Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
|
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
StartOfBuffer = VirtualMemory;
|
|
StartOfBuffer += Size - SizeRequested;
|
|
|
|
return StartOfBuffer;
|
|
}
|
|
|
|
static
|
|
VOID
|
|
FreeGuarded(
|
|
_In_ PVOID Pointer)
|
|
{
|
|
NTSTATUS Status;
|
|
PVOID VirtualMemory = (PVOID)PAGE_ROUND_DOWN((SIZE_T)Pointer);
|
|
SIZE_T Size = 0;
|
|
|
|
Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
|
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
|
}
|
|
|
|
void
|
|
Test_RtlFindMostSignificantBit(void)
|
|
{
|
|
ok_int(RtlFindMostSignificantBit(0), -1);
|
|
|
|
ok_int(RtlFindMostSignificantBit(0x0000000000000001ULL), 0);
|
|
ok_int(RtlFindMostSignificantBit(0x0000000000000002ULL), 1);
|
|
ok_int(RtlFindMostSignificantBit(0x1000000000000000ULL), 60);
|
|
ok_int(RtlFindMostSignificantBit(0x8000000000000000ULL), 63);
|
|
ok_int(RtlFindMostSignificantBit(0x8000000000000001ULL), 63);
|
|
ok_int(RtlFindMostSignificantBit(0xFFFFFFFFFFFFFFFFULL), 63);
|
|
ok_int(RtlFindMostSignificantBit(0x0000000070000000ULL), 30);
|
|
}
|
|
|
|
void
|
|
Test_RtlFindLeastSignificantBit(void)
|
|
{
|
|
ok_int(RtlFindLeastSignificantBit(0), -1);
|
|
|
|
ok_int(RtlFindLeastSignificantBit(0x0000000000000001ULL), 0);
|
|
ok_int(RtlFindLeastSignificantBit(0x0000000000000002ULL), 1);
|
|
ok_int(RtlFindLeastSignificantBit(0x1000000000000000ULL), 60);
|
|
ok_int(RtlFindLeastSignificantBit(0x8000000000000000ULL), 63);
|
|
ok_int(RtlFindLeastSignificantBit(0x8000000000000000ULL), 63);
|
|
ok_int(RtlFindLeastSignificantBit(0xFFFFFFFFFFFFFFFFULL), 0);
|
|
ok_int(RtlFindLeastSignificantBit(0x0000000070000000ULL), 28);
|
|
}
|
|
|
|
void
|
|
Test_RtlInitializeBitMap(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG Buffer[2];
|
|
BOOLEAN Exception = FALSE;
|
|
|
|
_SEH2_TRY
|
|
{
|
|
RtlInitializeBitMap(NULL, NULL, 0);
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
Exception = TRUE;
|
|
}
|
|
_SEH2_END;
|
|
ok_int(Exception, 1);
|
|
|
|
memset(Buffer, 0xcc, sizeof(Buffer));
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
ok_int(BitMapHeader.SizeOfBitMap, 0);
|
|
ok_ptr(BitMapHeader.Buffer, Buffer);
|
|
ok_hex(Buffer[0], 0xcccccccc);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 8);
|
|
ok_int(BitMapHeader.SizeOfBitMap, 8);
|
|
ok_hex(Buffer[0], 0xcccccccc);
|
|
}
|
|
|
|
void
|
|
Test_RtlClearAllBits(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
ULONG BufferSize = 2 * sizeof(*Buffer);
|
|
|
|
Buffer = AllocateGuarded(BufferSize);
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 19);
|
|
memset(Buffer, 0xcc, BufferSize);
|
|
RtlClearAllBits(&BitMapHeader);
|
|
ok_hex(Buffer[0], 0x00000000);
|
|
ok_hex(Buffer[1], 0xcccccccc);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
memset(Buffer, 0xcc, BufferSize);
|
|
RtlClearAllBits(&BitMapHeader);
|
|
ok_hex(Buffer[0], 0xcccccccc);
|
|
ok_hex(Buffer[1], 0xcccccccc);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 64);
|
|
memset(Buffer, 0xcc, BufferSize);
|
|
RtlClearAllBits(&BitMapHeader);
|
|
ok_hex(Buffer[0], 0x00000000);
|
|
ok_hex(Buffer[1], 0x00000000);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlSetAllBits(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
ULONG BufferSize = 2 * sizeof(*Buffer);
|
|
|
|
Buffer = AllocateGuarded(BufferSize);
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 19);
|
|
memset(Buffer, 0xcc, BufferSize);
|
|
RtlSetAllBits(&BitMapHeader);
|
|
ok_hex(Buffer[0], 0xffffffff);
|
|
ok_hex(Buffer[1], 0xcccccccc);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
memset(Buffer, 0xcc, BufferSize);
|
|
RtlSetAllBits(&BitMapHeader);
|
|
ok_hex(Buffer[0], 0xcccccccc);
|
|
ok_hex(Buffer[1], 0xcccccccc);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 64);
|
|
memset(Buffer, 0xcc, BufferSize);
|
|
RtlSetAllBits(&BitMapHeader);
|
|
ok_hex(Buffer[0], 0xffffffff);
|
|
ok_hex(Buffer[1], 0xffffffff);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlClearBits(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
ULONG BufferSize = 2 * sizeof(*Buffer);
|
|
|
|
Buffer = AllocateGuarded(BufferSize);
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 19);
|
|
|
|
memset(Buffer, 0xff, BufferSize);
|
|
RtlClearBits(&BitMapHeader, 0, 0);
|
|
ok_hex(Buffer[0], 0xffffffff);
|
|
ok_hex(Buffer[1], 0xffffffff);
|
|
|
|
memset(Buffer, 0xff, BufferSize);
|
|
RtlClearBits(&BitMapHeader, 0, 1);
|
|
ok_hex(Buffer[0], 0xfffffffe);
|
|
ok_hex(Buffer[1], 0xffffffff);
|
|
|
|
memset(Buffer, 0xff, BufferSize);
|
|
RtlClearBits(&BitMapHeader, 21, 1);
|
|
ok_hex(Buffer[0], 0xffdfffff);
|
|
ok_hex(Buffer[1], 0xffffffff);
|
|
|
|
memset(Buffer, 0xff, BufferSize);
|
|
RtlClearBits(&BitMapHeader, 7, 9);
|
|
ok_hex(Buffer[0], 0xffff007f);
|
|
ok_hex(Buffer[1], 0xffffffff);
|
|
|
|
memset(Buffer, 0xff, BufferSize);
|
|
RtlClearBits(&BitMapHeader, 13, 22);
|
|
ok_hex(Buffer[0], 0x00001fff);
|
|
ok_hex(Buffer[1], 0xfffffff8);
|
|
|
|
memset(Buffer, 0xff, BufferSize);
|
|
RtlClearBits(&BitMapHeader, 63, 1);
|
|
ok_hex(Buffer[0], 0xffffffff);
|
|
ok_hex(Buffer[1], 0x7fffffff);
|
|
|
|
memset(Buffer, 0xcc, BufferSize);
|
|
RtlClearBits(&BitMapHeader, 3, 6);
|
|
RtlClearBits(&BitMapHeader, 11, 5);
|
|
RtlClearBits(&BitMapHeader, 21, 7);
|
|
RtlClearBits(&BitMapHeader, 37, 4);
|
|
ok_hex(Buffer[0], 0xc00c0404);
|
|
ok_hex(Buffer[1], 0xcccccc0c);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlSetBits(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
ULONG BufferSize = 2 * sizeof(*Buffer);
|
|
|
|
Buffer = AllocateGuarded(BufferSize);
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 19);
|
|
|
|
memset(Buffer, 0x00, BufferSize);
|
|
RtlSetBits(&BitMapHeader, 0, 0);
|
|
ok_hex(Buffer[0], 0x00000000);
|
|
ok_hex(Buffer[1], 0x00000000);
|
|
|
|
memset(Buffer, 0x00, BufferSize);
|
|
RtlSetBits(&BitMapHeader, 0, 1);
|
|
ok_hex(Buffer[0], 0x00000001);
|
|
ok_hex(Buffer[1], 0x00000000);
|
|
|
|
memset(Buffer, 0x00, BufferSize);
|
|
RtlSetBits(&BitMapHeader, 21, 1);
|
|
ok_hex(Buffer[0], 0x00200000);
|
|
ok_hex(Buffer[1], 0x00000000);
|
|
|
|
memset(Buffer, 0x00, BufferSize);
|
|
RtlSetBits(&BitMapHeader, 7, 9);
|
|
ok_hex(Buffer[0], 0x0000ff80);
|
|
ok_hex(Buffer[1], 0x00000000);
|
|
|
|
memset(Buffer, 0x00, BufferSize);
|
|
RtlSetBits(&BitMapHeader, 13, 22);
|
|
ok_hex(Buffer[0], 0xffffe000);
|
|
ok_hex(Buffer[1], 0x00000007);
|
|
|
|
memset(Buffer, 0x00, BufferSize);
|
|
RtlSetBits(&BitMapHeader, 63, 1);
|
|
ok_hex(Buffer[0], 0x00000000);
|
|
ok_hex(Buffer[1], 0x80000000);
|
|
|
|
memset(Buffer, 0xcc, BufferSize);
|
|
RtlSetBits(&BitMapHeader, 3, 6);
|
|
RtlSetBits(&BitMapHeader, 11, 5);
|
|
RtlSetBits(&BitMapHeader, 21, 7);
|
|
RtlSetBits(&BitMapHeader, 37, 4);
|
|
ok_hex(Buffer[0], 0xcfecfdfc);
|
|
ok_hex(Buffer[1], 0xcccccdec);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlAreBitsClear(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
|
|
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 19);
|
|
Buffer[0] = 0x00ff00ff;
|
|
Buffer[1] = 0xc0cfc0cf;
|
|
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 0, 8), FALSE);
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 8, 8), TRUE);
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 7, 8), FALSE);
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 8, 9), FALSE);
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 24, 1), FALSE);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 31);
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 24, 1), TRUE);
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 24, 7), TRUE);
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 24, 8), FALSE);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 64);
|
|
ok_hex(RtlAreBitsClear(&BitMapHeader, 60, 4), FALSE);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlAreBitsSet(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
|
|
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 19);
|
|
Buffer[0] = 0xff00ff00;
|
|
Buffer[1] = 0x3F303F30;
|
|
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 0, 8), FALSE);
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 8, 8), TRUE);
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 7, 8), FALSE);
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 8, 9), FALSE);
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 24, 1), FALSE);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 31);
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 24, 1), TRUE);
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 24, 7), TRUE);
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 24, 8), FALSE);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 64);
|
|
ok_hex(RtlAreBitsSet(&BitMapHeader, 60, 4), FALSE);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlNumberOfSetBits(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
|
|
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
|
|
Buffer[0] = 0xff00ff0f;
|
|
Buffer[1] = 0x3F303F30;
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
ok_int(RtlNumberOfSetBits(&BitMapHeader), 0);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 4);
|
|
ok_int(RtlNumberOfSetBits(&BitMapHeader), 4);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 31);
|
|
ok_int(RtlNumberOfSetBits(&BitMapHeader), 19);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 56);
|
|
ok_int(RtlNumberOfSetBits(&BitMapHeader), 30);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 64);
|
|
ok_int(RtlNumberOfSetBits(&BitMapHeader), 36);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlNumberOfClearBits(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
|
|
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
|
|
Buffer[0] = 0xff00fff0;
|
|
Buffer[1] = 0x3F303F30;
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
ok_int(RtlNumberOfClearBits(&BitMapHeader), 0);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 4);
|
|
ok_int(RtlNumberOfClearBits(&BitMapHeader), 4);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 31);
|
|
ok_int(RtlNumberOfClearBits(&BitMapHeader), 12);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 56);
|
|
ok_int(RtlNumberOfClearBits(&BitMapHeader), 26);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 64);
|
|
ok_int(RtlNumberOfClearBits(&BitMapHeader), 28);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
void
|
|
Test_RtlFindSetBits(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
|
|
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
|
|
Buffer[0] = 0xF9F078B2;
|
|
Buffer[1] = 0x3F303F30;
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 0), 0);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 3), 0);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 1, 0), -1);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 1, 1), -1);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 8);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 3), 0);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 1, 0), 1);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 1, 1), 1);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 1, 2), 4);
|
|
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 2, 0), 4);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 3, 0), -1);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 32);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 3), 0);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 21), 16);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 12), 8);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 31), 24);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 32), 0);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 0, 39), 0);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 4, 0), 11);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 5, 0), 20);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 4, 11), 11);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 4, 12), 20);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 2, 11), 11);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 1, 32), 1);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 4, 32), 11);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 5, 32), 20);
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 64);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 5, 64), 20);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 6, 57), 40);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 7, 0), -1);
|
|
ok_int(RtlFindSetBits(&BitMapHeader, 1, 62), 1);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlFindClearBitsAndSet(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
|
|
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
|
|
Buffer[0] = 0x060F874D;
|
|
Buffer[1] = 0x3F303F30;
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 0, 0), 0);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 0, 3), 0);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 1, 0), -1);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 1, 1), -1);
|
|
ok_hex(Buffer[0], 0x060F874D);
|
|
|
|
Buffer[0] = 0x060F874D;
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 8);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 1, 0), 1);
|
|
ok_hex(Buffer[0], 0x60f874f);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 1, 1), 4);
|
|
ok_hex(Buffer[0], 0x60f875f);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 1, 2), 5);
|
|
ok_hex(Buffer[0], 0x60f877f);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 2, 0), -1);
|
|
ok_hex(Buffer[0], 0x60f877f);
|
|
|
|
Buffer[0] = 0x060F874D;
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 32);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 4, 0), 11);
|
|
ok_hex(Buffer[0], 0x60fff4d);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 5, 0), 20);
|
|
ok_hex(Buffer[0], 0x7ffff4d);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 4, 11), 27);
|
|
ok_hex(Buffer[0], 0x7fffff4d);
|
|
|
|
Buffer[0] = 0x060F874D;
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 4, 12), 20);
|
|
ok_hex(Buffer[0], 0x6ff874d);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 2, 11), 11);
|
|
ok_hex(Buffer[0], 0x6ff9f4d);
|
|
ok_int(RtlFindClearBitsAndSet(&BitMapHeader, 2, 12), 13);
|
|
ok_hex(Buffer[0], 0x6ffff4d);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlFindSetBitsAndClear(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
|
|
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
|
|
Buffer[0] = 0xF9F078B2;
|
|
Buffer[1] = 0x3F303F30;
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 0, 0), 0);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 0, 3), 0);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 1, 0), -1);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 1, 1), -1);
|
|
ok_hex(Buffer[0], 0xF9F078B2);
|
|
|
|
Buffer[0] = 0xF9F078B2;
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 8);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 1, 0), 1);
|
|
ok_hex(Buffer[0], 0xf9f078b0);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 1, 1), 4);
|
|
ok_hex(Buffer[0], 0xf9f078a0);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 1, 2), 5);
|
|
ok_hex(Buffer[0], 0xf9f07880);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 2, 0), -1);
|
|
ok_hex(Buffer[0], 0xf9f07880);
|
|
|
|
Buffer[0] = 0xF9F078B2;
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 32);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 4, 0), 11);
|
|
ok_hex(Buffer[0], 0xf9f000b2);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 5, 0), 20);
|
|
ok_hex(Buffer[0], 0xf80000b2);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 4, 11), 27);
|
|
ok_hex(Buffer[0], 0x800000b2);
|
|
|
|
Buffer[0] = 0xF9F078B2;
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 4, 12), 20);
|
|
ok_hex(Buffer[0], 0xf90078b2);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 2, 11), 11);
|
|
ok_hex(Buffer[0], 0xf90060b2);
|
|
ok_int(RtlFindSetBitsAndClear(&BitMapHeader, 2, 12), 13);
|
|
ok_hex(Buffer[0], 0xf90000b2);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlFindNextForwardRunClear(void)
|
|
{
|
|
RTL_BITMAP BitMapHeader;
|
|
ULONG *Buffer;
|
|
ULONG Index;
|
|
|
|
Buffer = AllocateGuarded(2 * sizeof(*Buffer));
|
|
Buffer[0] = 0xF9F078B2;
|
|
Buffer[1] = 0x3F303F30;
|
|
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 0);
|
|
ok_int(RtlFindNextForwardRunClear(&BitMapHeader, 0, &Index), 0);
|
|
ok_int(Index, 0);
|
|
ok_int(RtlFindNextForwardRunClear(&BitMapHeader, 1, &Index), 0);
|
|
ok_int(Index, 1);
|
|
|
|
Index = -1;
|
|
RtlInitializeBitMap(&BitMapHeader, Buffer, 8);
|
|
ok_int(RtlFindNextForwardRunClear(&BitMapHeader, 0, &Index), 1);
|
|
ok_int(Index, 0);
|
|
ok_int(RtlFindNextForwardRunClear(&BitMapHeader, 1, &Index), 2);
|
|
ok_int(Index, 2);
|
|
ok_int(RtlFindNextForwardRunClear(&BitMapHeader, 7, &Index), 0);
|
|
ok_int(Index, 8);
|
|
ok_int(RtlFindNextForwardRunClear(&BitMapHeader, 17, &Index), 0);
|
|
ok_int(Index, 17);
|
|
ok_int(RtlFindNextForwardRunClear(&BitMapHeader, 39, &Index), 0);
|
|
ok_int(Index, 39);
|
|
FreeGuarded(Buffer);
|
|
}
|
|
|
|
void
|
|
Test_RtlFindFirstRunClear(void)
|
|
{
|
|
}
|
|
|
|
void
|
|
Test_RtlFindLastBackwardRunClear(void)
|
|
{
|
|
}
|
|
|
|
void
|
|
Test_RtlFindClearRuns(void)
|
|
{
|
|
}
|
|
|
|
void
|
|
Test_RtlFindLongestRunClear(void)
|
|
{
|
|
}
|
|
|
|
|
|
START_TEST(RtlBitmap)
|
|
{
|
|
Test_RtlFindMostSignificantBit();
|
|
Test_RtlFindLeastSignificantBit();
|
|
Test_RtlInitializeBitMap();
|
|
Test_RtlClearAllBits();
|
|
Test_RtlSetAllBits();
|
|
Test_RtlClearBits();
|
|
Test_RtlSetBits();
|
|
Test_RtlAreBitsClear();
|
|
Test_RtlAreBitsSet();
|
|
Test_RtlNumberOfSetBits();
|
|
Test_RtlNumberOfClearBits();
|
|
Test_RtlFindClearBits();
|
|
Test_RtlFindSetBits();
|
|
Test_RtlFindClearBitsAndSet();
|
|
Test_RtlFindSetBitsAndClear();
|
|
Test_RtlFindNextForwardRunClear();
|
|
Test_RtlFindFirstRunClear();
|
|
Test_RtlFindLastBackwardRunClear();
|
|
Test_RtlFindClearRuns();
|
|
Test_RtlFindLongestRunClear();
|
|
}
|
|
|