mirror of
https://github.com/reactos/reactos.git
synced 2024-07-02 18:54:25 +00:00
49 lines
1.7 KiB
C
49 lines
1.7 KiB
C
/*
|
|
* PROJECT: ReactOS api tests
|
|
* LICENSE: LGPL-2.1-or-later (https://spdx.org/licenses/LGPL-2.1-or-later)
|
|
* PURPOSE: Test for RtlGetProcessHeaps
|
|
* COPYRIGHT: Copyright 2023 Mark Jansen <mark.jansen@reactos.org>
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
START_TEST(RtlGetProcessHeaps)
|
|
{
|
|
HANDLE HeapArray[40] = {0};
|
|
ULONG MaxHeapArraySize = _countof(HeapArray);
|
|
|
|
// Can call it with NULL
|
|
ULONG InitialNumHeaps = RtlGetProcessHeaps(0, NULL);
|
|
ok(InitialNumHeaps > 0, "Expected at least one heap\n");
|
|
ok(InitialNumHeaps < MaxHeapArraySize, "Expected less heaps, got %lu\n", InitialNumHeaps);
|
|
|
|
// Grab all heaps
|
|
ULONG NumHeaps = RtlGetProcessHeaps(MaxHeapArraySize, HeapArray);
|
|
ok_eq_ulong(NumHeaps, InitialNumHeaps);
|
|
for (ULONG n = 0; n < min(NumHeaps, MaxHeapArraySize); ++n)
|
|
{
|
|
ok(HeapArray[n] != NULL, "Heap[%lu] == NULL\n", n);
|
|
}
|
|
|
|
PVOID HeapPtr = RtlCreateHeap(HEAP_GROWABLE, NULL, 0, 0x10000, NULL, NULL);
|
|
|
|
memset(HeapArray, 0, sizeof(HeapArray));
|
|
NumHeaps = RtlGetProcessHeaps(MaxHeapArraySize, HeapArray);
|
|
ok_eq_ulong(InitialNumHeaps + 1, NumHeaps);
|
|
if (NumHeaps > 0 && NumHeaps <= MaxHeapArraySize)
|
|
{
|
|
// A new heap is added at the end
|
|
ok_ptr(HeapArray[NumHeaps - 1], HeapPtr);
|
|
}
|
|
|
|
RtlDestroyHeap(HeapPtr);
|
|
|
|
// Just ask for one heap, the number of heaps available is returned, but only one heap ptr filled!
|
|
memset(HeapArray, 0xff, sizeof(HeapArray));
|
|
NumHeaps = RtlGetProcessHeaps(1, HeapArray);
|
|
ok_eq_ulong(NumHeaps, InitialNumHeaps);
|
|
|
|
ok_ptr(HeapArray[0], RtlGetProcessHeap()); // First item is the process heap
|
|
ok_ptr(HeapArray[1], INVALID_HANDLE_VALUE); // Next item is not touched!
|
|
}
|