Something went wrong. Try again.
Reactos
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849/* * 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!}