mirror of
https://github.com/reactos/reactos.git
synced 2024-09-12 13:54:41 +00:00
79 lines
2.2 KiB
C
79 lines
2.2 KiB
C
![]() |
/*
|
||
|
* PROJECT: ReactOS api tests
|
||
|
* LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
|
||
|
* PURPOSE: Test for RtlReAllocateHeap
|
||
|
* PROGRAMMERS: Thomas Faber <thomas.faber@reactos.org>
|
||
|
*/
|
||
|
|
||
|
#include <apitest.h>
|
||
|
|
||
|
#define WIN32_NO_STATUS
|
||
|
#include <ndk/rtlfuncs.h>
|
||
|
|
||
|
static
|
||
|
BOOLEAN
|
||
|
CheckBuffer(
|
||
|
PVOID Buffer,
|
||
|
SIZE_T Size,
|
||
|
UCHAR Value)
|
||
|
{
|
||
|
PUCHAR Array = Buffer;
|
||
|
SIZE_T i;
|
||
|
|
||
|
for (i = 0; i < Size; i++)
|
||
|
if (Array[i] != Value)
|
||
|
{
|
||
|
trace("Expected %x, found %x at offset %lu\n", Value, Array[i], (ULONG)i);
|
||
|
return FALSE;
|
||
|
}
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
START_TEST(RtlReAllocateHeap)
|
||
|
{
|
||
|
PUCHAR Buffer = NULL;
|
||
|
PUCHAR NewBuffer;
|
||
|
SIZE_T OldSize = 0;
|
||
|
SIZE_T Size;
|
||
|
|
||
|
OldSize = 0x100;
|
||
|
Buffer = RtlReAllocateHeap(RtlGetProcessHeap(),
|
||
|
HEAP_ZERO_MEMORY,
|
||
|
NULL,
|
||
|
OldSize);
|
||
|
ok(Buffer == NULL, "RtlReAllocateHeap succeeded for NULL\n");
|
||
|
if (Buffer)
|
||
|
RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer);
|
||
|
|
||
|
Buffer = RtlAllocateHeap(RtlGetProcessHeap(),
|
||
|
HEAP_ZERO_MEMORY,
|
||
|
OldSize);
|
||
|
if (!Buffer)
|
||
|
{
|
||
|
skip("RtlAllocateHeap failed for size %lu\n", OldSize);
|
||
|
return;
|
||
|
}
|
||
|
ok(CheckBuffer(Buffer, OldSize, 0), "HEAP_ZERO_MEMORY not respected for 0x%lx\n", OldSize);
|
||
|
|
||
|
for (Size = 0x78000; Size < 0x90000; Size += 0x100)
|
||
|
{
|
||
|
RtlFillMemory(Buffer, OldSize, 0x7a);
|
||
|
NewBuffer = RtlReAllocateHeap(RtlGetProcessHeap(),
|
||
|
HEAP_ZERO_MEMORY,
|
||
|
Buffer,
|
||
|
Size);
|
||
|
if (!NewBuffer)
|
||
|
{
|
||
|
skip("RtlReAllocateHeap failed for size %lu\n", Size);
|
||
|
break;
|
||
|
}
|
||
|
Buffer = NewBuffer;
|
||
|
ok_hex(RtlSizeHeap(RtlGetProcessHeap(), 0, Buffer), Size);
|
||
|
ok(CheckBuffer(Buffer, OldSize, 0x7a), "CheckBuffer failed at size 0x%lx -> 0x%lx\n", OldSize, Size);
|
||
|
ok(CheckBuffer(Buffer + OldSize, Size - OldSize, 0), "HEAP_ZERO_MEMORY not respected for 0x%lx -> 0x%lx\n", OldSize, Size);
|
||
|
OldSize = Size;
|
||
|
}
|
||
|
RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer);
|
||
|
}
|
||
|
|