5.2 堆管理
说明
第一次参数 -a 创建一个10页大小的堆,所以在分配堆时出错直接退出,没有销毁堆?
第二次没有参数,说明创建一个可增长的堆,成功分配11页大小的堆
第三次使用参数 -s 使用默认的堆,默认堆大小是可增长的,所以也可以分配成功
同时也可以看出这个程序默认分配了2个堆
效果
源码
#include
#include
DWORD PrintHeapSize(HANDLE hHeap, LPVOID lpMem);
int main(int argc, PCHAR argv[]){
SYSTEM_INFO si;//系统信息
HANDLE hHeap;//堆句柄
LPVOID lpMem;//内存块指针
LPVOID lpReAlloc;//内存块大小调整后的指针
DWORD dwHeapSize;//堆内存大小
HANDLE hHeaps[24];//用于保存进程中所有的堆句柄
DWORD dwHeapNum;//进程中堆的数量
//获得系统信息
GetSystemInfo(&si);
//打印系统内存分页大小和内存分配粒度
printf("系统内存页大小: 0x%x\n系统内存分配粒度: 0x%x\n", si.dwPageSize, si.dwAllocationGranularity);
//分析输入参数,如果是 -a ,创建一个最大为10个分页大小的堆
if (argc == 2 && lstrcmp(argv[1], "-a") == 0){
hHeap = HeapCreate(HEAP_NO_SERIALIZE, si.dwPageSize, si.dwPageSize * 10);
printf("创建堆,初始大小为1页,最大为10页\n");
}
//如果参数是 -s ,使用进程初始化时已经存在的堆
else if(argc == 2 && lstrcmp(argv[1],"-s") == 0){
hHeap = GetProcessHeap();
printf("获得进程已经存在的堆\n");
}
//其他参数,创建一个可增长的堆
else{
hHeap = HeapCreate(HEAP_NO_SERIALIZE, 0, 0);
printf("创建堆,初始化大小为1页,大小可变\n");
}
//判断对是否创建成功
if (hHeap == NULL){
printf("创建或获取进程堆错误: %d\n", GetLastError());
return 1;
}
//获取当前进程中一共有多少个堆,有没有新建堆
dwHeapNum = GetProcessHeaps(24, hHeaps);
if (dwHeapNum == 0){
printf("获取进程堆数量出错: %d\n", GetLastError());
} else{
printf("当前进程一共有 %u 个堆\n", dwHeapNum);
}
//在堆上分配内存,3个页面大小
lpMem = HeapAlloc(hHeap, HEAP_ZERO_MEMORY, si.dwPageSize * 3);
if(lpMem == NULL){
printf("分配内存失败: %d\n", GetLastError());
return 1;
}
printf("在堆上分配内存成功,起始地址: 0x%x\n", lpMem);
//打印当前堆内存块的大小
PrintHeapSize(hHeap, lpMem);
//在分配内存,调整为内存大小为11个分页大小
//第一种方法创建堆则出错
lpReAlloc = HeapReAlloc(hHeap, HEAP_ZERO_MEMORY,lpMem, si.dwPageSize * 11);
if (lpReAlloc == NULL){
printf("再分配堆内存失败: %d\n", GetLastError());
return 1;
}
printf("再分配堆内存,地址为: 0x%x, 原始地址为: 0x%x\n", lpReAlloc, lpMem);
//打印调整大小后的堆内存大小
PrintHeapSize(hHeap, lpReAlloc);
//释放内存
if (!HeapFree(hHeap, HEAP_NO_SERIALIZE, lpReAlloc)){
printf("释放堆失败: %d\n", GetLastError());
return 1;
}
printf("释放堆成功\n");
//如果新建了堆,销毁堆
if (argc != 2 || lstrcmp(argv[1], "-s") != 0){
printf("销毁HeapCreate创建的堆\n");
if (!HeapDestroy(hHeap)){
printf("销毁堆错误: %d\n", GetLastError());
return 1;
}
printf("销毁堆成功\n");
}
return 0;
}
//获得堆的大小并打印
DWORD PrintHeapSize(HANDLE hHeap, LPVOID lpMem){
SIZE_T dwHeapSize;
dwHeapSize = HeapSize(hHeap, HEAP_NO_SERIALIZE, lpMem);
if (dwHeapSize == -1){
printf("获取堆大小失败: %d\n", GetLastError());
return 1;
}
printf("内存块大小为: 0x%x\n", dwHeapSize);
return 0;
}
