[PATCH 1/1] kernel32/tests: Don't go beyond user space virtual address limit.

[Yuxuan Shui]

From: Yuxuan Shui

In get_valloc_info, if mem happens to point to the last chunk of memory in the
user space virtual address space, "p += info2.RegionSize" will cause it to go
over the limit, and subsequent VirtualQueries will fail, thus info2 won't be
changed, thus the loop exit condition is never met (well until p wraps around,
that is).
---
 dlls/kernel32/tests/heap.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/dlls/kernel32/tests/heap.c b/dlls/kernel32/tests/heap.c
index f88a140a115..ec3549ac605 100644
--- a/dlls/kernel32/tests/heap.c
+++ b/dlls/kernel32/tests/heap.c
@@ -3648,15 +3648,20 @@ static void test_GlobalMemoryStatus(void)
 static void get_valloc_info( void *mem, char **base, SIZE_T *alloc_size )
 {
     MEMORY_BASIC_INFORMATION info, info2;
+    SYSTEM_BASIC_INFORMATION si;
+    NTSTATUS status;
     SIZE_T size;
     char *p;

+    status = NtQuerySystemInformation(SystemBasicInformation, &si, sizeof(si), NULL);
+    ok( !status, "NtQuerySystemInformation returned %#lx\n", status );
+
     size = VirtualQuery( mem, &info, sizeof(info) );
     ok( size == sizeof(info), "got %Iu.\n", size );

info2 = info;
     p = info.AllocationBase;
-    while (1)
+    while (p < (char *)si.HighestUserAddress)
     {
         size = VirtualQuery( p, &info2, sizeof(info2) );
         ok( size == sizeof(info), "got %Iu.\n", size );