RtlReAllocateHeap.c File Reference

#include "precomp.h"

Include dependency graph for RtlReAllocateHeap.c:

Go to the source code of this file.

Functions
static BOOLEAN	CheckBuffer (PVOID Buffer, SIZE_T Size, UCHAR Value)
static BOOLEAN	ReAllocBuffer (PUCHAR *Buffer, SIZE_T Size, SIZE_T *OldSizePtr, PCSTR Action)
	START_TEST (RtlReAllocateHeap)

Function Documentation

CheckBuffer()

static BOOLEAN CheckBuffer ( PVOID  Buffer, SIZE_T  Size, UCHAR  Value  )

static

Definition at line 12 of file RtlReAllocateHeap.c.

{
    PUCHAR Array = Buffer;
    SIZE_T i;

    for (i = 0; i < Size; i++)
        if (Array[i] != Value)
        {
            trace("Expected %x, found %x at offset %lu\n", Value, Array[i], (ULONG)i);
            return FALSE;
        }
    return TRUE;
}

Referenced by ReAllocBuffer(), and START_TEST().

ReAllocBuffer()

static BOOLEAN ReAllocBuffer ( PUCHAR *  Buffer, SIZE_T  Size, SIZE_T *  OldSizePtr, PCSTR  Action  )

static

Definition at line 31 of file RtlReAllocateHeap.c.

{
    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;
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

RtlReAllocateHeap

)

Definition at line 65 of file RtlReAllocateHeap.c.

{
    PUCHAR Buffer = NULL;
    SIZE_T OldSize = 0;
    SIZE_T Size;
    BOOLEAN Continue = TRUE;
    BOOLEAN Success;
    PVOID UserValue;
    ULONG UserFlags;
    PVOID Buffer2;

    OldSize = 0x100;
    Buffer = RtlReAllocateHeap(RtlGetProcessHeap(),
                               HEAP_ZERO_MEMORY,
                               NULL,
                               OldSize);
    ok(Buffer == NULL, "RtlReAllocateHeap succeeded for NULL\n");
    if (Buffer)
        RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer);

    Buffer = RtlAllocateHeap(RtlGetProcessHeap(),
                             HEAP_ZERO_MEMORY,
                             OldSize);
    if (!Buffer)
    {
        skip("RtlAllocateHeap failed for size %lu\n", OldSize);
        return;
    }
    ok(CheckBuffer(Buffer, OldSize, 0), "HEAP_ZERO_MEMORY not respected for 0x%lx\n", OldSize);

    for (Size = 0x78000; Size < 0x90000 && Continue; Size += 0x100)
    {
        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);

    /* Now make sure user flags/values get preserved */
    OldSize = 0x100;
    Buffer = RtlAllocateHeap(RtlGetProcessHeap(),
                             HEAP_ZERO_MEMORY | HEAP_SETTABLE_USER_VALUE | HEAP_SETTABLE_USER_FLAG2,
                             OldSize);
    if (!Buffer)
    {
        skip("RtlAllocateHeap failed for size %lu\n", OldSize);
        return;
    }

    UserValue = InvalidPointer;
    UserFlags = 0x55555555;
    Success = RtlGetUserInfoHeap(RtlGetProcessHeap(),
                                 0,
                                 Buffer,
                                 &UserValue,
                                 &UserFlags);
    ok(Success == TRUE, "RtlGetUserInfoHeap returned %u\n", Success);
    ok(UserValue == NULL, "UserValue = %p\n", UserValue);
    ok(UserFlags == HEAP_SETTABLE_USER_FLAG2, "UserFlags = %lx\n", UserFlags);

    Success = RtlSetUserFlagsHeap(RtlGetProcessHeap(),
                                  0,
                                  Buffer,
                                  HEAP_SETTABLE_USER_FLAG1 | HEAP_SETTABLE_USER_FLAG2,
                                  HEAP_SETTABLE_USER_FLAG3);
    ok(Success == TRUE, "RtlSetUserFlagsHeap returned %u\n", Success);

    Success = RtlSetUserValueHeap(RtlGetProcessHeap(),
                                  0,
                                  Buffer,
                                  &UserValue);
    ok(Success == TRUE, "RtlSetUserValueHeap returned %u\n", Success);

    UserValue = InvalidPointer;
    UserFlags = 0x55555555;
    Success = RtlGetUserInfoHeap(RtlGetProcessHeap(),
                                 0,
                                 Buffer,
                                 &UserValue,
                                 &UserFlags);
    ok(Success == TRUE, "RtlGetUserInfoHeap returned %u\n", Success);
    ok(UserValue == &UserValue, "UserValue = %p, expected %p\n", UserValue, &UserValue);
    ok(UserFlags == HEAP_SETTABLE_USER_FLAG3, "UserFlags = %lx\n", UserFlags);

    /* shrink (preserves flags) */
    Buffer2 = RtlReAllocateHeap(RtlGetProcessHeap(),
                                HEAP_REALLOC_IN_PLACE_ONLY | HEAP_SETTABLE_USER_FLAG2,
                                Buffer,
                                OldSize / 2);
    ok(Buffer2 == Buffer, "New Buffer is %p, expected %p\n", Buffer2, Buffer);
    if (Buffer2) Buffer = Buffer2;
    UserValue = InvalidPointer;
    UserFlags = 0x55555555;
    Success = RtlGetUserInfoHeap(RtlGetProcessHeap(),
                                 0,
                                 Buffer,
                                 &UserValue,
                                 &UserFlags);
    ok(Success == TRUE, "RtlGetUserInfoHeap returned %u\n", Success);
    ok(UserValue == &UserValue, "UserValue = %p, expected %p\n", UserValue, &UserValue);
    ok(UserFlags == HEAP_SETTABLE_USER_FLAG3, "UserFlags = %lx\n", UserFlags);

    /* grow (overwrites flags) */
    Buffer2 = RtlReAllocateHeap(RtlGetProcessHeap(),
                                HEAP_REALLOC_IN_PLACE_ONLY | HEAP_SETTABLE_USER_FLAG1,
                                Buffer,
                                OldSize / 4 * 3);
    ok(Buffer2 == Buffer, "New Buffer is %p, expected %p\n", Buffer2, Buffer);
    if (Buffer2) Buffer = Buffer2;
    UserValue = InvalidPointer;
    UserFlags = 0x55555555;
    Success = RtlGetUserInfoHeap(RtlGetProcessHeap(),
                                 0,
                                 Buffer,
                                 &UserValue,
                                 &UserFlags);
    ok(Success == TRUE, "RtlGetUserInfoHeap returned %u\n", Success);
    ok(UserValue == &UserValue, "UserValue = %p, expected %p\n", UserValue, &UserValue);
    ok(UserFlags == HEAP_SETTABLE_USER_FLAG1, "UserFlags = %lx\n", UserFlags);

    RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer);
}