Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-20 03:38:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NTDLL_APITEST] Add test for RtlGetProcessHeaps

Browse source
This commit is contained in:
Mark Jansen 2023-12-08 00:18:57 +01:00
parent b8cdd1a879
commit e8f9564c20
No known key found for this signature in database
GPG key ID: B39240EE84BEAE8B
3 changed files with 51 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
modules/rostests/apitests/ntdll/CMakeLists.txt
View file

@ -79,6 +79,7 @@ list(APPEND SOURCE
RtlGetLengthWithoutTrailingPathSeperators.c
RtlGetLongestNtPathLength.c
RtlGetNtProductType.c
RtlGetProcessHeaps.c
RtlGetUnloadEventTrace.c
RtlHandle.c
RtlImageDirectoryEntryToData.c

48
modules/rostests/apitests/ntdll/RtlGetProcessHeaps.c Normal file
View file

@ -0,0 +1,48 @@
/*
* 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!
}

2
modules/rostests/apitests/ntdll/testlist.c
View file

@ -76,6 +76,7 @@ extern void func_RtlGetLengthWithoutLastFullDosOrNtPathElement(void);
extern void func_RtlGetLengthWithoutTrailingPathSeperators(void);
extern void func_RtlGetLongestNtPathLength(void);
extern void func_RtlGetNtProductType(void);
extern void func_RtlGetProcessHeaps(void);
extern void func_RtlGetUnloadEventTrace(void);
extern void func_RtlHandle(void);
extern void func_RtlImageDirectoryEntryToData(void);
@ -175,6 +176,7 @@ const struct test winetest_testlist[] =
{ "RtlGetLengthWithoutTrailingPathSeperators", func_RtlGetLengthWithoutTrailingPathSeperators },
{ "RtlGetLongestNtPathLength", func_RtlGetLongestNtPathLength },
{ "RtlGetNtProductType", func_RtlGetNtProductType },
{ "RtlGetProcessHeaps", func_RtlGetProcessHeaps },
{ "RtlGetUnloadEventTrace", func_RtlGetUnloadEventTrace },
{ "RtlHandle", func_RtlHandle },
{ "RtlImageDirectoryEntryToData", func_RtlImageDirectoryEntryToData },