在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