【C++】智能指针
一、为什么要智能指针
下面我们先分析下面这段程序有没有什么内存方面的问题？
int Div(int a, int b)
{
if (b == 0)
throw invalid_argument("除0错误");
else
return a / b;
}
void Func()
{
// 1、如果 p1 这里 new 抛异常会如何？
// 2、如果 p2 这里 new 抛异常会如何？
// 3、如果 Div 这里又会抛异常会如何？
int* p1 = new int[10];// 如果这里抛异常没有内存泄漏
int* p2 = new int[10];// p1内存泄漏
int a, b;
cin >> a >> b;
cout << Div(a, b) << endl;// p1、p2内存泄漏
delete[] p1;
delete[] p2;
}
int main()
{
try
{
Func();
}
catch (exception& e)
{
cout << e.what() << endl;
}
return 0;
}
二、内存泄漏
内存泄漏及其危害
什么是内存泄漏：内存泄漏指因为疏忽或错误造成程序未能释放已经不再使用的内存的情况。
内存泄漏并不是指内存在物理上的消失，而是应用程序分配某段内存后，因为设计错误，失去了对该段内存的控制，因而造成了内存的浪费。
内存泄漏的危害：长期运行的程序出现内存泄漏，影响很大，如操作系统、后台服务等等，出现内存泄漏会导致响应越来越慢，最终卡死。
void MemoryLeaks()
{
// 1. 内存申请了忘记释放
int* p1 = (int*)malloc(sizeof(int));
int* p2 = new int;
// 2. 异常安全问题
int* p3 = new int[10];
Func(); // 这里Func函数抛异常导致 delete[] p3未执行，p3没被释放.
delete[] p3;
}
三、智能指针的使用及原理
1. RAII
RAII（Resource Acquisition Is Initialization）是一种利用对象生命周期来控制程序资源（如内存、文件句柄、网络连接、互斥量等等）的简单技术。
在对象构造时获取资源，接着控制对资源的访问使之在对象的生命周期内始终保持有效，最后在对象析构时释放资源。我们实际上把管理一份资源的责任托管给了一个对象。这种做法有两大好处：
- 不需要显式地释放资源。
- 采用这种方式，对象所需的资源在其生命期内始终保持有效。
// 使用RAII思想设计的SmartPtr类
template<class T>
class SmartPtr
{
T* _ptr;
public:
SmartPtr(T* ptr = nullptr)
{
_ptr = ptr;
}
~SmartPtr()
{
if (_ptr)
{
cout << _ptr << endl;
// 至于这里如何决定是使用delete还是delete[] -- 定制删除器
delete _ptr;
}
}
};
int Div(int a, int b)
{
if (b == 0)
throw invalid_argument("除0错误");
else
return a / b;
}
void Func()
{
int* p1 = new int[10];
SmartPtr<int> sp1(p1);// 出了作用域sp1调用析构释放资源
int* p2 = new int[10];
SmartPtr<int> sp2(p2);
int a, b;
cin >> a >> b;
cout << Div(a, b) << endl;
}
int main()
{
try
{
Func();
}
catch (exception& e)
{
cout << e.what() << endl;
}
return 0;
}
2. 智能指针的原理
上述的SmartPtr还不能将其称为智能指针，因为它还不具有指针的行为。指针可以解引用，也可 以通过->去访问所指空间中的内容，因此：auto_ptr模板类中还得需要重载* 、->，才可让其像指针一样去使用
template<class T>
class SmartPtr
{
private:
T* _ptr;
public:
SmartPtr(T* ptr = nullptr)
{
_ptr = ptr;
}
~SmartPtr()
{
if (_ptr)
{
//cout << _ptr << endl;
delete _ptr;
}
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
};
struct Date
{
int _year;
int _month;
int _day;
};
int main()
{
SmartPtr<int> sp1(new int);
*sp1 = 10;
cout << *sp1 << endl;
SmartPtr<Date> sp2(new Date);
// 需要注意的是这里应该是sp2.operator->()->_year = 2023;
// 本来应该是sp2->->_year这里语法上为了可读性，省略了一个->
sp2->_year = 2023;
sp2->_month = 1;
sp2->_day = 1;
cout << sp2->_year << endl;
}
总结一下智能指针的原理：
RAII特性- 重载
operator*和opertaor->，具有像指针一样的行为。
3. std::auto_ptr
std::auto_ptr，C++98 版本的库中就提供了auto_ptr的智能指针。
auto_ptr 的实现原理：管理权转移的思想，下面简化模拟实现一份 auto_ptr 来了解它的原理
namespace nb
{
template<class T>
class auto_ptr
{
private:
T* _ptr;
public:
auto_ptr(T* ptr)
:_ptr(ptr)
{
ptr = nullptr;
}
auto_ptr(auto_ptr<T>& sp)
:_ptr(sp._ptr)
{
// 管理权转移
sp._ptr = nullptr;// 如果不置空会发生重复析构
}
auto_ptr<T>& operator=(auto_ptr<T>& ap)
{
// 检测是否为自己给自己赋值
if (this != &ap)
{
// 释放当前对象中资源
if (_ptr)
delete _ptr;
// 转移 ap 中资源到当前对象中
_ptr = ap._ptr;
ap._ptr = nullptr;
}
return *this;
}
~auto_ptr()
{
if (_ptr)
{
cout << "delete: " << _ptr << endl;
delete _ptr;
}
}
// 像指针一样使用
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
};
}
int main()
{
std::auto_ptr<int> sp1(new int);
std::auto_ptr<int> sp2(sp1); // 管理权转移
*sp2 = 10;
cout << *sp2 << endl;
// sp1悬空
//cout << *sp1 << endl;// 空指针无法解引用
return 0;
}
结论：auto_ptr 是一个失败设计，很多公司明确要求不能使用 auto_ptr
4. std::unique_ptr
C++11 中开始提供更靠谱的 unique_ptr
unique_ptr 的实现原理：简单粗暴的防拷贝，下面简化模拟实现一份 unique_ptr 来了解它的原理。
namespace nb
{
template<class T>
class unique_ptr
{
private:
T* _ptr;
public:
unique_ptr(T* ptr)
:_ptr(ptr)
{
ptr = nullptr;
}
~unique_ptr()
{
if (_ptr)
{
delete _ptr;
}
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
// 防止拷贝
unique_ptr(const unique_ptr<T>& uptr) = delete;
unique_ptr<T>& operator=(const unique_ptr<T>& sp) = delete;
};
}
int main()
{
nb::unique_ptr<int> sp1(new int);
//nb::unique_ptr sp2(sp1); //
*sp1 = 10;
cout << *sp1 << endl;
return 0;
}
5. std::shared_ptr
C++11 中开始提供更靠谱的并且支持拷贝的 shared_ptr
shared_ptr 的原理：是通过引用计数的方式来实现多个 shared_ptr 对象之间共享资源。
例如： 寝室最后走的关空调
- shared_ptr 在其内部，给每个资源都维护着一份计数，用来记录该份资源被几个对象共享。
- 在对象被销毁时(也就是析构函数调用)，就说明自己不使用该资源了，对象的引用计数减一。
- 如果引用计数是0，就说明自己是最后一个使用该资源的对象，必须释放该资源；
- 如果不是0，就说明除了自己还有其他对象在使用该份资源，不能释放该资源，否则其他对象就成野指针了
template<class T>
class shared_ptr
{
private:
T* _ptr;
// 注意这里不能给int、static int
int* _pRefCount;// 指向引用计数的指针
mutex* _pmtx;// 互斥量
public:
shared_ptr(T* ptr = nullptr)
:_ptr(ptr),
_pRefCount(new int(1)),// 初始化给1
_pmtx(new mutex)
{}
shared_ptr(const shared_ptr<T>& sp)
:_ptr(sp._ptr),
_pRefCount(sp._pRefCount),
_pmtx(sp._pmtx)
{
AddRef();// 增加资源的引用计数
}
// sp1 = sp2
shared_ptr<T>& operator=(const shared_ptr<T>& sp)
{
// 下面这种写法可以防止sp1 = sp1这种情况
// 但不能防止这种：sp2(sp1); sp1 = sp2;
// 所以要判断资源的地址是否一样
// if (this != &sp)
if (_ptr != sp._ptr)
{
Release();// 检查sp1原来的资源是否要释放
// 管理新资源
_ptr = sp._ptr;
_pRefCount = sp._pRefCount;
_pmtx = sp._pmtx;
AddRef();// 增加新资源的引用计数
}
return *this;
}
void Release()
{
_pmtx->lock();
bool flag = false;
if (--(*_pRefCount) == 0 && _ptr)
{
cout << "delete:" << _ptr << endl;
delete _ptr;
delete _pRefCount;
flag = true;
}
_pmtx->unlock();// 注意锁的释放
if (flag)
{
delete _pmtx;
}
}
void AddRef()
{
_pmtx->lock();
++(*_pRefCount);// 如果是int，++后不会改变另一个对象的计数
_pmtx->unlock();
}
int use_count()
{
return *_pRefCount;
}
T* get() const
{
return _ptr;
}
~shared_ptr()
{
Release();
}
// 像指针一样使用
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
};
std::shared_ptr 的线程安全问题：
通过下面的程序我们来测试 shared_ptr 的线程安全问题。需要注意的是shared_ptr 的线程安全分为两方面：
- 智能指针对象中引用计数是多个智能指针对象共享的，两个线程中智能指针的引用计数同时 ++ 或 - -，这个操作不是原子的，引用计数原来是1，++ 了两次，可能还是2.这样引用计数就错乱了。会导致资源未释放或者程序崩溃的问题。所以智能指针中引用计数 ++、- - 是需要加锁的，也就是说引用计数的操作是线程安全的。
- 智能指针管理的资源对象存放在堆上，两个线程同时去访问，会导致线程安全问题。
// shared_ptr智能指针是线程安全的吗？
// 是的，引用计数的加减是加锁保护的。但是指向资源不是线程安全的
// 指向堆上资源的线程安全问题是访问的人处理的，智能指针不管，也管不了
// 引用计数的线程安全问题，是智能指针要处理的
int main()
{
nb::shared_ptr<int> sp1(new int);
nb::shared_ptr<int> sp2(sp1);
nb::shared_ptr<int> sp3(sp1);
nb::shared_ptr<int> sp4(new int);
nb::shared_ptr<int> sp5(sp4);
sp1 = sp1;
sp1 = sp2;
sp1 = sp4;
sp2 = sp4;
sp3 = sp4;
*sp1 = 2;
*sp2 = 3;
return 0;
}
上图拷贝后资源、引用计数和锁的地址都是一样的。
线程安全问题演示：
如果想演示引用计数线程安全问题，就把 AddRef 和 Release 中的锁去掉。
下面演示的是智能指针管理的资源的线程安全问题。演示可能不出现线程安全问题，因为线程安全问题是偶现性问题，main 函数的 n 改大一些概率就变大了，就更容易出现了。
struct Date
{
int _year = 0;
int _month = 0;
int _day = 0;
};
void SharePtrFunc(nb::shared_ptr<Date>& sp, size_t n, mutex& mtx)
{
cout << sp.get() << endl;
for (size_t i = 0; i < n; ++i)
{
// 这里智能指针拷贝会++计数，智能指针析构会--计数，这里是线程安全的。
nb::shared_ptr<Date> copy(sp);
// 这里智能指针访问管理的资源，不是线程安全的。
// 理论上这些值两个线程++了2n次，但是最终看到的结果，并一定是加了2n
// 加个{}是为了让锁出了作用域自动销毁
{
// 不加锁不能保证智能指针管理的资源是线程安全的
unique_lock<mutex> lk(mtx);
copy->_year++;
copy->_month++;
copy->_day++;
}
}
}
int main()
{
nb::shared_ptr<Date> p(new Date);
cout << p.get() << endl;
const size_t n = 100000;
mutex mtx;
thread t1(SharePtrFunc, std::ref(p), n, std::ref(mtx));
thread t2(SharePtrFunc, std::ref(p), n, std::ref(mtx));
t1.join();
t2.join();
cout << p->_year << " " << p->_month << " " << p->_day << endl;
cout << p.use_count() << endl;
return 0;
}
6. std::shared_ptr 的循环引用
struct ListNode
{
int _data;
shared_ptr<ListNode> _prev;// 智能指针对象管理资源，会自动释放资源
shared_ptr<ListNode> _next;
~ListNode()// 写析构只是为了看有没有自动调析构
{
cout << "~ListNode()" << endl;
}
};
int main()
{
std::shared_ptr<ListNode> node1(new ListNode);
std::shared_ptr<ListNode> node2(new ListNode);
cout << node1.use_count() << endl;
cout << node2.use_count() << endl;
node1->_next = node2;
node2->_prev = node1;
cout << node1.use_count() << endl;
cout << node2.use_count() << endl;
return 0;
}
输出：
1
1
2
2
main 函数结束后要释放两个节点的空间，但是从输出可以知道智能指针对象的引用计数不会减到0，所以没调析构释放资源。
循环引用分析：
- node1 和 node2 两个智能指针对象指向两个节点，引用计数变成1，我们不需要手动 delete。
- node1 的 _next 指向 node2，node2 的 _prev 指向 node1，引用计数变成2。
- node1 和 node2 析构，引用计数减到1，但是 node1 的 _next 还指向下一个节点，node2 的 _prev 还指向上 一个节点。
- 如果 node1 的 _next 析构了，node2 就释放了。
- 如果 node2 的 _prev 析构了，node1 就释放了。
- 但是 _next 属于 node 的成员，node1 释放了，它的成员 _next 才会析构（对象销毁，成员才会销毁），而 node1 由_prev 管理，_prev 属于 node2 成员，所以这就叫循环引用，谁也不会释放，进而导致内存泄漏。
解决方案：
在引用计数的场景下，把节点中的 _prev 和 _next 改成 weak_ptr 就可以了。
原理就是，node1->_next = node2;和 node2->_prev = node1;时 weak_ptr 的 _next 和_prev 不会增加 node1 和 node2 的引用计数（weak_ptr 只指向资源，可以访问资源但不管理资源）。
// 简化版本的 weak_ptr 实现
template<class T>
class weak_ptr
{
private:
T* _ptr;
public:
// 没有接收指针的构造函数
weak_ptr()
:_ptr(nullptr)
{}
weak_ptr(const shared_ptr<T>& sp)
:_ptr(sp.get())
{}
weak_ptr<T>& operator=(const shared_ptr<T>& sp)
{
_ptr = sp.get();
return *this;
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
};
struct ListNode
{
int _data;
weak_ptr<ListNode> _prev;
weak_ptr<ListNode> _next;
~ListNode()
{
cout << "~ListNode()" << endl;
}
};
int main()
{
shared_ptr<ListNode> node1(new ListNode);
shared_ptr<ListNode> node2(new ListNode);
cout << node1.use_count() << endl;
cout << node2.use_count() << endl;
node1->_next = node2;
node2->_prev = node1;
cout << node1.use_count() << endl;
cout << node2.use_count() << endl;
return 0;
}
7. 定制删除器
如果不是 new 出来的对象如何通过智能指针管理呢？其实 shared_ptr 设计了一个删除器来解决这个问题
// 仿函数的删除器
template<class T>
struct FreeFunc {
void operator()(T* ptr)
{
cout << "free:" << ptr << endl;
free(ptr);
}
};
template<class T>
struct DeleteArrayFunc {
void operator()(T* ptr)
{
cout << "delete[]" << ptr << endl;
delete[] ptr;
}
};
class A {};
int main()
{
FreeFunc<int> freeFunc;
// 构造函数传删除器对象
std::shared_ptr<int> sp1((int*)malloc(4), freeFunc);
DeleteArrayFunc<int> deleteArrayFunc;
std::shared_ptr<int> sp2((int*)malloc(4), deleteArrayFunc);
std::shared_ptr<A> sp4(new A[10], [](A* p) {delete[] p; });
std::shared_ptr<FILE> sp5(fopen("test.txt", "w"), [](FILE* p) {fclose(p); });
return 0;
}