就像您提到的问题下的评论一样,您可以重载new
anddelete
运算符来跟踪内存分配和释放。本文全局替换部分下的示例显示了一个非常简单的示例:
void* operator new(std::size_t sz) {
std::printf("global op new called, size = %zu\n", sz);
return std::malloc(sz);
}
void operator delete(void* ptr) noexcept {
std::puts("global op delete called");
std::free(ptr);
}
此示例的唯一问题是您无法确定释放了多少内存。要解决此问题,请查看如何在 C++ 中跟踪内存分配(尤其是新建/删除)问题的已接受答案。
上述答案中的示例使用std::map
自定义分配器来存储已分配内存块的地址和大小。在delete
运算符重载中,它删除具有指定地址的元素。几乎无需修改即可满足您的要求:
#include <map>
template<typename T>
struct MemoryMapAllocator : std::allocator<T> {
typedef typename std::allocator<T>::pointer pointer;
typedef typename std::allocator<T>::size_type size_type;
template<typename U> struct rebind { typedef MemoryMapAllocator<U> other; };
MemoryMapAllocator() {}
template<typename U>
MemoryMapAllocator(const MemoryMapAllocator<U>& u) : std::allocator<T>(u) {}
pointer allocate(size_type size, std::allocator<void>::const_pointer = 0) {
void* p = std::malloc(size * sizeof(T));
if(p == 0)
throw std::bad_alloc();
return static_cast<pointer>(p);
}
void deallocate(pointer p, size_type) {
std::free(p);
}
};
typedef std::map<void*, std::size_t, std::less<void*>,
MemoryMapAllocator<std::pair<void* const, std::size_t>>> MemoryMap;
MemoryMap& getMemoryMap() {
static MemoryMap memMap;
return memMap;
}
std::size_t totalAllocatedMemory() {
std::size_t sum = 0;
for(auto& e : getMemoryMap())
sum += e.second;
return sum;
}
void* operator new(std::size_t size) {
void* mem = std::malloc(size == 0 ? 1 : size);
if(mem == 0)
throw std::bad_alloc();
getMemoryMap()[mem] = size;
return mem;
}
void operator delete(void* mem) {
getMemoryMap().erase(mem);
std::free(mem);
}
Live Demo