我们来看这样一段代码:
#include
using namespace std;int div()
{int a, b;cin >> a >> b;if (b == 0){throw "除0错误";}return a / b;
}void func()
{int* p1 = new int;int* p2 = new int;cout << div() << endl;delete p1;delete p2;
}int main()
{try{func();}catch (exception& e){cout << e.what() << endl;}return 0;
}
在这段代码中,如果在div
函数中发生了除0
错误,我们在main函数捕获异常,那么最终异常抛出后会跳转到main
函数中的catch
处,对于p1
和p2
申请的资源就没有得到释放,就造成了内存泄露问题.
关于内存泄露介绍,在另一篇文章中有详细的介绍:
RAII
是一种利用对象生命周期来控制程序资源(例如内存、文件句柄、网络连接、互斥量等)的简单技术.
获取到资源以后去初始化一个对象,将资源交给对象管理:资源获取即初始化
在对象构造时获取资源,接着控制对资源的访问使之在对象的生命周期内始终保持有效,最后在对象析构的时候释放资源. 借此,我们实际是把管理一份资源的责任托管给了一个对象,这种做法有两大好处:
//实现一个最简易的智能指针
template
class smart_ptr
{
public:smart_ptr(T* ptr = nullptr): _ptr(ptr){}//对象析构时自动释放所管理的资源~smart_ptr(){cout << "delete " << _ptr << endl;if (_ptr){delete _ptr;_ptr = nullptr;}}
private:T* _ptr;
};
此时,我们使用智能指针来代替裸指针,并让它发生除0错误:
#include
#include "smart_ptr.h"
using namespace std;int div()
{int a, b;cin >> a >> b;if (b == 0){throw invalid_argument("除0错误");}return a / b;
}void func()
{smart_ptr p1(new int);smart_ptr p2(new int);cout << div() << endl;
}int main()
{try{func();}catch (exception& e){cout << e.what() << endl;}return 0;
}
可以看到,刚才由于抛异常未能释放的资源现在可以正常释放.
我们所写的这个简单的智能指针smart_ptr
还不能称其为智能指针,因为它还不具有指针的行为,指针可以解引用,可以通过->去访问所指向空间内的内容,所以为了让它向指针一样,我们还需重载*
,->
运算符.
template
class smart_ptr
{
public:smart_ptr(T* ptr = nullptr): _ptr(ptr){}~smart_ptr(){cout << "delete " << _ptr << endl;if (_ptr){delete _ptr;_ptr = nullptr;}}T& operator*() const{return *_ptr;}T* get() const{return _ptr;}T* operator->() const //T* = const T* _ptr{return _ptr;}
private:T* _ptr;
};
但是这样的智能指针是有问题的,试一下它的拷贝?
那么,怎么去解决这个问题呢?
所以,接下来,我们来介绍几种C++标准库里的智能指针,来探究如何解决此问题
C++98版本的库中就提供了auto_ptr
智能指针,它解决了拷贝赋值的问题,但也还有一些不足
auto_ptr
实现原理:管理权转移的思想,下面将简化的模拟实现auto_ptr
,主要体现它的思想:
namespace myPtr
{templateclass auto_ptr{public:auto_ptr(T* ptr = nullptr): _ptr(ptr){}~auto_ptr(){if (_ptr){cout << "delete " << _ptr << endl;delete _ptr;_ptr = nullptr;}}auto_ptr(auto_ptr& sp): _ptr(sp._ptr){sp._ptr = nullptr;}auto_ptr& operator=(auto_ptr& ap){if (this != &ap){//释放当前对象管理的资源if (_ptr)delete _ptr;//管理权转移_ptr = ap._ptr;ap._ptr = nullptr;}return *this;}T& operator*() const{return *_ptr;}T* operator->() const{return _ptr;}private:T* _ptr;};
}
从它的拷贝构造和赋值重载可以看出,它以资源管理权转移的方式解决拷贝的问题.
但是,这样也带来另一个问题:
int main()
{myPtr::auto_ptr p1(new int);myPtr::auto_ptr p2 = p1;//*p1 = 10; p1已经没有对资源的管理权,不能再使用!return 0;
}
由于p1
的资源管理权已经转移给了p2
,那它自己就失去对资源的管理及使用权,造成p1
悬空.
unique_ptr
是C++11才开始提供的一种智能指针,它相比于auto_ptr
更靠谱.
unique
的实现原理:简单粗暴的禁止拷贝
我们依然是模拟实现一个unique_ptr
来理解它的原理:
template
class unique_ptr
{public:unique_ptr(T* ptr = nullptr): _ptr(ptr){}~unique_ptr(){if (_ptr){cout << "delete " << _ptr << endl;delete _ptr;_ptr = nullptr;}}unique_ptr(const unique_ptr&) = delete;unique_ptr& operator=(const unique_ptr&) = delete;T& operator*() const{return *_ptr;}T* operator->() const{return _ptr;}private:T* _ptr;
};
unique_ptr
采用一种简单粗暴的方式来解决拷贝的问题:直接删除拷贝构造函数和赋值重载函数,禁止拷贝
shared_ptr
也是C++11开始提供的,它能够解决智能指针拷贝的问题,并且它不像unique_ptr
那样直接禁止拷贝,它是支持拷贝的.
shared_ptr
的原理:通过引用计数的方式来实现多个shared_ptr
对象之间共享资源
shared_ptr
内部,给它所管理的资源维护了一份引用计数,用来记录该资源被几个对象共同管理(共享)template
class shared_ptr
{public:shared_ptr(T* ptr = nullptr): _ptr(ptr){//将引用计数初始化为1}~shared_ptr(){if (_ptr){cout << "delete " << _ptr << endl;delete _ptr;_ptr = nullptr;}}T& operator*() const{return *_ptr;}T* operator->() const{return _ptr;}private:T* _ptr;//维护一个引用计数
};
那么,这个引用计数,我们应该怎样去维护呢?
使用普通变量int _pCount
显然不行,由于私有成员变量是两个对象独有的,假如我们有两个智能指针p1
和p2
,对p1
的引用计数的–不会影响p2
定义一个静态成员变量static int _pCount
看似,好像可以,但是试一下如下场景:
template
class shared_ptr
{public:shared_ptr(T* ptr = nullptr): _ptr(ptr){//在构造函数中将引用计数初始化为1_pCount = 1;}~shared_ptr(){if (--_pCount == 0 && _ptr){cout << "delete " << _ptr << endl;delete _ptr;_ptr = nullptr;}}shared_ptr(const shared_ptr& sp){_ptr = sp._ptr;++_pCount;}T& operator*() const{return *_ptr;}T* operator->() const{return _ptr;}private:T* _ptr;//维护一个引用计数static int _pCount;
};template
int shared_ptr::_pCount = 0;
int main()
{//_pCount = 1myPtr::shared_ptr p1(new int);//_pCount = 2;myPtr::shared_ptr p2(p1);//这一步又重新将_pCount置为1,导致最终只释放的一次资源myPtr::shared_ptr p3(new int);return 0;
}
定义一个指针成员变量int* _pCount
template
class shared_ptr
{public:shared_ptr(T* ptr = nullptr): _ptr(ptr), _pCount(new int(1)){}void Release(){if (--(*_pCount) == 0 && _ptr){cout << "delete " << _ptr << endl;delete _ptr;_ptr = nullptr;delete _pCount;_pCount = nullptr;}}~shared_ptr(){Release();}shared_ptr(const shared_ptr& sp){_ptr = sp._ptr;_pCount = sp._pCount;++(*_pCount);}shared_ptr& operator=(const shared_ptr& sp){//防止自赋值if (_ptr != sp._ptr){Release();_ptr = sp._ptr;_pCount = sp._pCount;++(*_pCount);}return *this;}T& operator*() const{return *_ptr;}T* operator->() const{return _ptr;}private:T* _ptr;//维护一个引用计数int* _pCount;
};
这就是我们最终模拟实现出的shared_ptr
.
shared_ptr
的循环引用问题:
struct ListNode
{ListNode(const int& val = int()): _next(nullptr), _prev(nullptr), _val(val){}~ListNode(){cout << "~ListNode()" << endl;}myPtr::shared_ptr _next;myPtr::shared_ptr _prev;int _val;
};int main()
{myPtr::shared_ptr p1(new ListNode(1));myPtr::shared_ptr p2(new ListNode(2));p1->_next = p2;p2->_prev = p1;return 0;
}
在上面的代码中,我们应该是有两份ListNode
节点需要释放,运行程序:
没有任何节点被释放.
这便是shared_ptr
的循环引用问题.
解决方案:在引用计数的场景下,把节点中的_prev
和_next
改成weak_ptr
就可以
weak_ptr
原理:
node1->_next = node2;
node2->_prev = node1;//weak_ptr的_next和_prev不会增加node1和node2的引用计数
weak_ptr
模拟实现
template
class weak_ptr
{public:weak_ptr(){}weak_ptr(const shared_ptr& sp){_ptr = sp._ptr;}weak_ptr& operator=(const shared_ptr& sp){if (_ptr != sp.get()){_ptr = sp.get();}return *this;}T* operator->(){return _ptr;}T& operator*(){return *_ptr;}private:T* _ptr;
};
在我们写的智能指针的析构函数中,统一都使用delete来释放资源,但是,如果资源不是用new申请出来的呢?比如:new T[],malloc
,所以我们就需要定制删除器来规范释放资源的方式.
我们使用shared_ptr
来做演示:
template
struct default_delete
{void operator()(T* ptr){delete ptr;}
};template>
class shared_ptr
{public:shared_ptr(T* ptr = nullptr): _ptr(ptr), _pCount(new int(1)){}void Release(){if (--(*_pCount) == 0 && _ptr){cout << "delete " << _ptr << endl;Del del;del(_ptr);_ptr = nullptr;delete _pCount;_pCount = nullptr;}}~shared_ptr(){Release();}shared_ptr(const shared_ptr& sp){_ptr = sp._ptr;_pCount = sp._pCount;++(*_pCount);}shared_ptr& operator=(const shared_ptr& sp){//防止自赋值if (_ptr != sp._ptr){Release();_ptr = sp._ptr;_pCount = sp._pCount;++(*_pCount);}return *this;}T& operator*() const{return *_ptr;}T* operator->() const{return _ptr;}T* get() const{return _ptr;}private:T* _ptr;//维护一个引用计数int* _pCount;
};
//定制new T[]类型的删除器
template
struct DeleteArray
{void operator()(T* ptr){delete[] ptr;}
};int main()
{myPtr::shared_ptr> p(new int[10]);return 0;
}
auto_ptr