内存管理器

最新推荐文章于 2019-04-08 08:23:38 发布

ba_jie

最新推荐文章于 2019-04-08 08:23:38 发布

阅读量755

点赞数

CC 4.0 BY-SA版权

分类专栏： c/c++ 文章标签： delete file manager iterator null struct

本文链接：https://blog.youkuaiyun.com/ba_jie/article/details/6361417

c/c++ 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了如何在C++中创建一个自定义内存管理器，通过宏定义NEW和DELETE，结合operator new和operator delete，记录每次内存分配和释放的位置信息。MemoryManager结构体用于存储内存块信息，并使用迭代器进行查找和删除。CustomAllocMem和CustomFreeMem函数实现了自定义的内存分配和释放，同时提供CustomCheckMemory函数检查未释放的内存。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在stdafx.h里放入如下声明：

#if defined(USE_MEM_MANAGER) # define NEW new((__FILE__), (__LINE__)) # define DELETE delete void* operator new(size_t size, const char* file, int line); void* operator new[](size_t size, const char* file, int line); void operator delete(void* p); void operator delete [](void* p); #else # define NEW new # define DELETE delete #endif

建立一个新文件：MemoryManager.cpp

#if defined(USE_MEM_MANAGER) #include <stdlib.h> #include <vector> #include <string> #include <algorithm> #include <functional> // 每次内存分配都记录到该类里 struct MemBlock { void* ptr; size_t size; std::string file; size_t line; MemBlock(void* _ptr, size_t _size, const char* _file, size_t _line) { ptr = _ptr; size = _size; file = _file; line = _line; } ~MemBlock() { // 可以在里面打log } }; typedef std::vector<MemBlock *> MemoryManager; static MemoryManager memmgr; struct IsMemEqual { bool operator()(const MemBlock* block) { return (block->ptr == ptr); } IsMemEqual(void * p) : ptr(p) { } void * ptr; }; void* CustomAllocMem(size_t size, const char* file, int line) { void * ptr = malloc(size); if (ptr != NULL) { // operator new (size_t size) 没有被重新定义，所以MemBlock不会被递归的加入到内存管理器 MemBlock* mem = new MemBlock(ptr, size, file, line); memmgr.push_back(mem); } return ptr; } void CustomFreeMem(void * p) { MemoryManager::iterator itr = std::find_if(memmgr.begin(), memmgr.end(), IsMemEqual(p)); if (itr != memmgr.end()) { delete *itr; memmgr.erase(itr); } free(p); } void CustomCheckMemory() { // 检查memmgr里还剩下多少内存未被释放 } void* operator new(size_t size, const char* file, int line) { if (size == 0) return NULL; return CustomAllocMem(size, file, line); } void* operator new[](size_t size, const char* file, int line) { if (size == 0) return NULL; return CustomAllocMem(size, file, line); } void operator delete(void* p) { if (p == NULL) return; CustomFreeMem(p); } void operator delete [](void* p) { if (p == NULL) return; CustomFreeMem(p); } #endif // USE_MEM_MANAGER