c++怎么实现一个简单的内存泄漏检测工具_c++自定义分配器与内存跟踪实现

#include <unordered_map>
#include <mutex>
#include <execinfo.h>
#include <cstdint>
<p>struct AllocInfo {
size_t size;
void* call_stack[10];
int stack_size;
};</p><p>std::unordered_map<void*, AllocInfo> g_allocations;
std::mutex g_alloc_mutex;</p><p>void<em> operator new(size_t size) {
void</em> ptr = std::malloc(size);</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">if (!ptr) throw std::bad_alloc();

// 获取调用栈
AllocInfo info;
info.size = size;
info.stack_size = backtrace(info.call_stack, 10);

std::lock_guard<std::mutex> lock(g_alloc_mutex);
g_allocations[ptr] = info;

return ptr;

std::lock_guard<std::mutex> lock(g_alloc_mutex);
g_allocations.erase(ptr);

std::free(ptr);

struct LeakReporter {
    ~LeakReporter() {
        if (g_allocations.empty()) {
            printf("✅ No memory leaks detected.\n");
            return;
        }
<pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">    printf("❌ Memory leaks detected:\n");
    for (auto& pair : g_allocations) {
        printf("  Address %p, Size %zu bytes\n", pair.first, pair.second.size);
        char** symbols = backtrace_symbols(pair.second.call_stack, pair.second.stack_size);
        for (int i = 0; i < pair.second.stack_size; ++i) {
            printf("    %s\n", symbols[i]);
        }
        free(symbols);
    }
}

template<typename T>
struct TrackingAllocator {
    using value_type = T;
<pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">T* allocate(size_t count) {
    void* ptr = ::operator new(count * sizeof(T));

    AllocInfo info{count * sizeof(T)};
    info.stack_size = backtrace(info.call_stack, 10);

    std::lock_guard<std::mutex> lock(g_alloc_mutex);
    g_allocations[ptr] = info;

    return static_cast<T*>(ptr);
}

void deallocate(T* ptr, size_t count) noexcept {
    std::lock_guard<std::mutex> lock(g_alloc_mutex);
    g_allocations.erase(ptr);
    ::operator delete(ptr);
}

std::vector<int, TrackingAllocator<int>> vec;
vec.push_back(42);  // 分配会被记录