如何为C++静态库定制内存分配器并精准统计其内存使用量？

仅使用全局自定义分配器或重写全局operator new/delete不足以实现目标——这类全局替换会捕获整个进程的内存分配操作，无法区分库A与其他依赖库的内存使用。要精准统计库A的内存，需要针对性地隔离库A的内存分配逻辑，以下是几种可靠方案：

1. 为库A实现专属的内存分配接口 #

在库A内部统一使用自定义的分配/释放函数，或重载命名空间级的operator new/delete，避免污染全局空间：

• 定义库A专属的分配逻辑：

namespace LibraryA {
    // 记录内存使用的计数器（仅库A内部可见）
    static std::atomic<size_t> total_memory_used = 0;
    static std::unordered_map<void*, size_t> allocated_blocks;

    void* operator new(size_t size) {
        void* ptr = std::malloc(size + sizeof(size_t));
        if (!ptr) throw std::bad_alloc();
        // 在内存块头部存储分配大小
        *static_cast<size_t*>(ptr) = size;
        total_memory_used += size;
        allocated_blocks[static_cast<char*>(ptr) + sizeof(size_t)] = size;
        return static_cast<char*>(ptr) + sizeof(size_t);
    }

    void operator delete(void* ptr) {
        if (!ptr) return;
        char* real_ptr = static_cast<char*>(ptr) - sizeof(size_t);
        size_t size = *static_cast<size_t*>(real_ptr);
        total_memory_used -= size;
        allocated_blocks.erase(ptr);
        std::free(real_ptr);
    }

    // 数组版本的重载
    void* operator new[](size_t size) {
        return LibraryA::operator new(size);
    }

    void operator delete[](void* ptr) {
        LibraryA::operator delete(ptr);
    }

    // 提供对外查询内存使用的接口
    size_t get_total_memory_used() {
        return total_memory_used.load();
    }
}

• 确保库A内所有类、对象的内存分配都默认使用该命名空间下的new/delete，比如在库A的头文件开头添加using namespace LibraryA;（仅库A内部代码使用），或显式指定LibraryA::new。

2. 为库A的STL容器配置自定义分配器 #

如果库A使用了STL容器，需要为这些容器指定专属的自定义分配器，确保容器的内存分配也被统计：

namespace LibraryA {
    template <typename T>
    struct AAllocator {
        using value_type = T;

        T* allocate(size_t n) {
            const size_t bytes = n * sizeof(T);
            total_memory_used += bytes;
            T* ptr = static_cast<T*>(std::malloc(bytes));
            if (!ptr) throw std::bad_alloc();
            return ptr;
        }

        void deallocate(T* p, size_t n) {
            const size_t bytes = n * sizeof(T);
            total_memory_used -= bytes;
            std::free(p);
        }

        // 兼容C++17及以上的分配器要求
        template <typename U>
        struct rebind { using other = AAllocator<U>; };
    };

    // 使用示例
    using AVector = std::vector<int, AAllocator<int>>;
}

• 需保证库A中所有STL容器都使用该分配器，避免遗漏容器的内存统计。

3. 利用链接器符号隔离（谨慎使用） #

针对静态库特性，可通过编译选项限制库A内的operator new/delete仅在库内部可见，避免与全局实现冲突：

• 使用GCC/Clang时，添加编译选项-fvisibility=hidden，让库A的operator new/delete符号仅在库内可见，外部代码仍使用系统默认的内存分配逻辑。不过这种方式需要严格控制符号可见性，容易出现链接冲突，仅作为备选方案。

总结 #

要精准统计库A的内存使用，核心是让库A的内存分配逻辑与全局/其他库完全隔离——全局重写new/delete或全局分配器无法满足隔离需求，必须采用命名空间专属重载、自定义容器分配器这类针对性方案，才能确保统计结果仅包含库A的内存操作。

内容的提问来源于stack exchange，提问作者heLomaN