[tfaber] 66960: [NTDLL_APITEST] - Also test decreasing the size in RtlReAllocateHeap. Patch by Mark Jansen. CORE-9441

Author: tfaber
Date: Sun Mar 29 14:47:21 2015
New Revision: 66960

URL: http://svn.reactos.org/svn/reactos?rev=66960&view=rev
Log:
[NTDLL_APITEST]
- Also test decreasing the size in RtlReAllocateHeap. Patch by Mark Jansen.
CORE-9441

Modified:
    trunk/rostests/apitests/ntdll/RtlReAllocateHeap.c

Modified: trunk/rostests/apitests/ntdll/RtlReAllocateHeap.c
URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/RtlReAlloca...
==============================================================================
--- trunk/rostests/apitests/ntdll/RtlReAllocateHeap.c	[iso-8859-1] (original)
+++ trunk/rostests/apitests/ntdll/RtlReAllocateHeap.c	[iso-8859-1] Sun Mar 29 14:47:21 2015
@@ -6,8 +6,6 @@
  */

#include <apitest.h>
-
-#define WIN32_NO_STATUS
 #include <ndk/rtlfuncs.h>

static
@@ -29,12 +27,48 @@
     return TRUE;
 }

+static
+BOOLEAN
+ReAllocBuffer(
+    PUCHAR *Buffer,
+    SIZE_T Size,
+    SIZE_T *OldSizePtr,
+    PCSTR Action)
+{
+    PUCHAR NewBuffer;
+    SIZE_T OldSize = *OldSizePtr;
+
+    RtlFillMemory(*Buffer, OldSize, 0x7a);
+    NewBuffer = RtlReAllocateHeap(RtlGetProcessHeap(),
+                                  HEAP_ZERO_MEMORY,
+                                  *Buffer,
+                                  Size);
+    if (!NewBuffer)
+    {
+        skip("RtlReAllocateHeap failed for size %lu (%s)\n", Size, Action);
+        return FALSE;
+    }
+    *Buffer = NewBuffer;
+    ok_hex(RtlSizeHeap(RtlGetProcessHeap(), 0, NewBuffer), Size);
+    if (OldSize < Size)
+    {
+        ok(CheckBuffer(NewBuffer, OldSize, 0x7a), "CheckBuffer failed at size 0x%lx -> 0x%lx\n", OldSize, Size);
+        ok(CheckBuffer(NewBuffer + OldSize, Size - OldSize, 0), "HEAP_ZERO_MEMORY not respected for 0x%lx -> 0x%lx\n", OldSize, Size);
+    }
+    else
+    {
+        ok(CheckBuffer(NewBuffer, Size, 0x7a), "CheckBuffer failed at size 0x%lx -> 0x%lx\n", OldSize, Size);
+    }
+    *OldSizePtr = Size;
+    return TRUE;
+}
+
 START_TEST(RtlReAllocateHeap)
 {
     PUCHAR Buffer = NULL;
-    PUCHAR NewBuffer;
     SIZE_T OldSize = 0;
     SIZE_T Size;
+    BOOLEAN Continue = TRUE;

OldSize = 0x100;
     Buffer = RtlReAllocateHeap(RtlGetProcessHeap(),
@@ -55,24 +89,15 @@
     }
     ok(CheckBuffer(Buffer, OldSize, 0), "HEAP_ZERO_MEMORY not respected for 0x%lx\n", OldSize);

-    for (Size = 0x78000; Size < 0x90000; Size += 0x100)
+    for (Size = 0x78000; Size < 0x90000 && Continue; 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;
+        Continue = ReAllocBuffer(&Buffer, Size, &OldSize, "growing");
+    }
+
+    /* and back again */
+    for (Size -= 0x100; Size >= 0x78000 && Continue; Size -= 0x100)
+    {
+        Continue = ReAllocBuffer(&Buffer, Size, &OldSize, "shrinking");
     }
     RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer);
 }
-