在C++程序开发过程中，动态内存管理如果操作不当很容易引发内存泄漏，标准库提供的智能指针虽然能自动管理内存生命周期，但在需要定制化泄漏检测的场景下，开发者往往需要自行实现智能指针并加入检测逻辑，同时配合开发工具提升排查效率。
自定义智能指针的核心设计思路
自定义智能指针本质是封装原生指针的RAII类，核心要解决的问题是跟踪指针指向内存的引用情况，在合适的时候释放内存，同时记录内存的分配和释放信息用于泄漏检测。核心组成部分包括原生指针成员、引用计数指针、内存分配信息记录结构。
引用计数的实现
引用计数用于记录当前有多少个智能指针对象共享同一块内存，当引用计数降为0时，自动释放对应的内存。为了避免多个智能指针对象拷贝时引用计数不同步，引用计数本身需要存储在堆上，所有共享同一块内存的智能指针都指向同一个引用计数变量。
我们可以定义一个辅助结构来存储引用计数和内存分配的相关信息：
// 内存分配信息结构
struct MemoryInfo {
int ref_count; // 引用计数
const char* file; // 分配内存的文件名
int line; // 分配内存的行号
void* ptr; // 指向的实际内存地址
};
// 自定义智能指针模板类
template <typename T>
class CustomSmartPtr {
private:
T* raw_ptr; // 原生指针
MemoryInfo* mem_info; // 内存信息指针
public:
// 构造函数，分配内存时记录信息
CustomSmartPtr(T* ptr, const char* file, int line) : raw_ptr(ptr) {
mem_info = new MemoryInfo();
mem_info->ref_count = 1;
mem_info->file = file;
mem_info->line = line;
mem_info->ptr = ptr;
// 这里可以将mem_info加入全局的待检测内存列表
}
// 拷贝构造函数，引用计数加1
CustomSmartPtr(const CustomSmartPtr<T>& other) {
raw_ptr = other.raw_ptr;
mem_info = other.mem_info;
mem_info->ref_count++;
}
// 赋值运算符重载
CustomSmartPtr<T>& operator=(const CustomSmartPtr<T>& other) {
if (this != &other) {
// 先处理当前对象的旧引用计数
if (mem_info != nullptr) {
mem_info->ref_count--;
if (mem_info->ref_count == 0) {
delete raw_ptr;
// 从全局待检测列表移除mem_info
delete mem_info;
}
}
// 赋值新对象的信息
raw_ptr = other.raw_ptr;
mem_info = other.mem_info;
mem_info->ref_count++;
}
return *this;
}
// 解引用运算符重载
T& operator*() const {
return *raw_ptr;
}
// 箭头运算符重载
T* operator->() const {
return raw_ptr;
}
// 析构函数，引用计数减1，为0时释放内存
~CustomSmartPtr() {
if (mem_info != nullptr) {
mem_info->ref_count--;
if (mem_info->ref_count == 0) {
delete raw_ptr;
// 从全局待检测列表移除mem_info
delete mem_info;
}
}
}
};
全局内存跟踪列表
为了检测内存泄漏，我们需要一个全局的容器来存储所有未释放的内存信息，程序退出时检查这个容器中是否还有残留的MemoryInfo对象，如果有就说明发生了内存泄漏，输出对应的文件名和行号即可。
可以使用一个全局的哈希表或者链表来存储这些信息，注意要处理多线程场景下的同步问题，这里以单线程场景为例：
#include <map>
#include <iostream>
// 全局内存跟踪表，key为内存地址，value为MemoryInfo指针
std::map<void*, MemoryInfo*> g_memory_tracker;
// 添加内存信息到跟踪表
void add_memory_record(MemoryInfo* info) {
g_memory_tracker[info->ptr] = info;
}
// 移除内存信息从跟踪表
void remove_memory_record(void* ptr) {
g_memory_tracker.erase(ptr);
}
// 检测内存泄漏的函数，程序退出时调用
void check_memory_leak() {
if (g_memory_tracker.empty()) {
std::cout << "No memory leak detected." << std::endl;
return;
}
std::cout << "Memory leak detected:" << std::endl;
for (auto& item : g_memory_tracker) {
MemoryInfo* info = item.second;
std::cout << "Leaked memory at " << info->ptr
<< ", allocated in file: " << info->file
<< ", line: " << info->line << std::endl;
}
}
修改之前的CustomSmartPtr构造函数和析构函数，加入跟踪表的添加和移除逻辑即可。
开发工具辅助检测
自定义智能指针虽然能实现基础的泄漏检测，但在复杂场景下可能存在覆盖不全的问题，配合开发工具可以提升检测的全面性和效率。
使用Valgrind辅助检测
Valgrind是常用的内存检测工具，在Linux环境下可以直接检测程序的内存泄漏、非法内存访问等问题。即使使用了自定义智能指针，也可以用Valgrind验证是否有遗漏的泄漏点。编译程序时加上-g选项保留调试信息，运行命令valgrind --leak-check=full ./your_program即可输出详细的泄漏信息。
使用Visual Studio诊断工具
在Windows平台使用Visual Studio开发时，自带的诊断工具可以检测内存泄漏。可以在程序入口和出口加入_CrtSetDbgFlag相关配置，开启调试堆的内存跟踪，程序退出时会自动输出内存泄漏的详细信息，包括泄漏内存的分配位置。
IDE插件的配合使用
很多C++ IDE都支持内存检测插件，比如CLion自带的内存分析工具，可以在运行时实时监控内存分配和释放情况，配合自定义智能指针的检测逻辑，能更快速地定位泄漏的具体代码位置，减少排查时间。
注意事项
- 自定义智能指针的引用计数操作要注意线程安全，多线程场景下需要加锁保护引用计数的增减。
- 全局内存跟踪表需要在程序退出时确保所有智能指针都已经析构，否则可能出现误报。
- 不要对同一个原生指针创建多个独立的
CustomSmartPtr对象，否则会导致重复释放的问题，和标准的shared_ptr一样要避免循环引用的问题。 - 开发工具的检测结果和自定义智能指针的检测结果可以互相印证，提升检测的准确性。