HeapReAlloc() -- 重新分配内存

beyes

HeapReAlloc() 原型如下：

[C++] 纯文本查看 复制代码

LPVOID WINAPI HeapReAlloc(
  __in  HANDLE hHeap,
  __in  DWORD dwFlags,
  __in  LPVOID lpMem,
  __in  SIZE_T dwBytes
);

该函数用来重新分配内存，改变已经分配好的堆内存块的大小。

第 1 个参数 hHeap 是输入参数，表明从此参数所指定的对上进行内存再分配。

第 2 个参数 dwFlags 是输入参数，是内存分配标志。它可以是下面的几个值：

HEAP_GENERATE_EXCEPTIONS ： 如果分配失败则抛出异常，而不是返回 NULL 。异常值可能是 STATUS_NO_MEMORY ，此时表示可获得的内存容量不足；也可能是 STATUS_ACCESS_VIOLATION ，此时表示非法存取。

HEAP_NO_SERIALIZE ： 不连续存取。

HEAP_ZERO_MEMORY ： 将分配的内存块中的内容全部清零。

HEAP_REALLOC_IN_PLACE_ONLY ： 该选项表示在重新分配内存时原先那块内存不能被移动。如果不指定该值，那么函数可能会将原来的内存块移动到别的地方。指定了该值，如果遇到不移动位置就不能重新分配的情况那么函数执行失败，而原来的内存块不会被改变。

第 3 个参数 lpMem 是输入参数，是原来内存块的起始地址。

第 4 个参数 dwBytes 是输入参数，表示内存块调整大小后的值，可以比原来的大，也可以比原来的小。

测试代码：

[C++] 纯文本查看 复制代码

// Heap.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <locale.h>
#include <Windows.h>

int _tmain(int argc, _TCHAR* argv[])
{
    setlocale (LC_ALL, "CHS");

    SYSTEM_INFO SystemInfo;
    HANDLE hHeap;


    GetSystemInfo (&SystemInfo);

    _tprintf (TEXT("系统内存分页大小：0x%x\n"), SystemInfo.dwPageSize);
    _tprintf (TEXT("系统内存分配粒度：0x%x\n"), SystemInfo.dwAllocationGranularity);

    /*
    //创建一个最大为 5 个分页大小的堆
    hHeap = HeapCreate (HEAP_NO_SERIALIZE, SystemInfo.dwPageSize, SystemInfo.dwPageSize * 5);
    if (hHeap == NULL) {
        _tprintf (TEXT("创建堆错误 ： %d\n"), GetLastError());
        return (-1);
    }

    //SIZE_T    dwHeapSize = HeapSize (hHeap, HEAP_NO_SERIALIZE, 
    */
    HANDLE hHeaps[24];
    DWORD    dwHeapNum = GetProcessHeaps(24, hHeaps);

    _tprintf (TEXT("当前进程默认总共有 %d 个堆\n"), dwHeapNum);

    //创建一个最大为 10 个分页大小的堆
    hHeap = HeapCreate (HEAP_NO_SERIALIZE, SystemInfo.dwPageSize, SystemInfo.dwPageSize * 30);
    if (hHeap == NULL) {
        _tprintf (TEXT("创建堆错误 ： %d\n"), GetLastError());
        return (-1);
    }
    _tprintf (TEXT("创建了一个初始化大小为1页,最大10页的堆.\n"));

    dwHeapNum = GetProcessHeaps(24, hHeaps);
    _tprintf (TEXT("进程现在总共有 %d 个堆\n"), dwHeapNum);


    LPVOID    lpMem;    //内存块指针

    //在堆上分配 3 个页面大小的内存
    lpMem = HeapAlloc (hHeap, HEAP_ZERO_MEMORY, SystemInfo.dwPageSize * 3);
    if (lpMem == NULL) {
        _tprintf (TEXT("HeapAlloc error : %d\n"), GetLastError());
        return (-2);
    }
    _tprintf (TEXT("在堆上分配内存成功，起始地址为：0x%x\n"), lpMem);


    SIZE_T dwHeapSize = HeapSize(hHeap, HEAP_NO_SERIALIZE, lpMem);
    if (dwHeapSize == -1) {
        _tprintf (TEXT("获取堆内存块大小错误 ： %d\n"), GetLastError());
        return (-3);
    }
    _tprintf (TEXT("内存块的大小为 : 0x%x\n"), dwHeapSize);

    LPVOID lpReAlloc;        //重分配后的内存块起始指针

    //重新分配内存
    lpReAlloc = HeapReAlloc (hHeap, HEAP_ZERO_MEMORY, lpMem, SystemInfo.dwPageSize * 8);
    if (lpReAlloc == NULL) {
        _tprintf (TEXT("HeapReAlloc error : %d\n"), GetLastError());
        return (-4);
    }
    dwHeapSize = HeapSize(hHeap, HEAP_NO_SERIALIZE, lpReAlloc);
    _tprintf (TEXT("重新分配后内存块的大小为 : 0x%x\n"), dwHeapSize);


    return 0;
}

运行输出：

D:\WinAPI\Heap\Debug>Heap.exe
系统内存分页大小：0x1000
系统内存分配粒度：0x10000
当前进程默认总共有 2 个堆
创建了一个初始化大小为1页,最大10页的堆.
进程现在总共有 3 个堆
在堆上分配内存成功，起始地址为：0x8c0578
内存块的大小为 : 0x3000
重新分配后内存块的大小为 : 0x8000

注意，程序中一开始使用 HeapCreate() 函数创建了一个最多 30 个页的堆，最终实际分配的可能没有 30 个，而是少于 30 个，这样一来，若是在下面的 HeapReAlloc() 函数里要求调整内存块的大小为 29 个页则可能遭遇调整失败。