内存管理器

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

 

#if defined(USE_MEM_MANAGER)#define NEW new((__FILE__), (__LINE__))#define DELETE deletevoid* 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_tsize;std::stringfile;size_tline;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