一、为什么需要智能指针
我们先来看一段代码
#include <iostream>
#include <vector>
#include <exception>
using namespace std;
void Test()
{
int* tmp = (int*)malloc(sizeof(int)* 100);
int* p = new int;
vector<int> v1(1000000000, 10);
vector<int> v2(10, 10);
//v2[10] = 20;
free(tmp);
delete p;
}
int main()
{
try
{
Test();
}
catch (const exception& e)
{
cout << e.what() << endl;
}
catch (...)
{
cout << "未知错误" << endl;
}
return 0;
}
问题分析:上面的问题分析出来我们发现有以下两个问题?
1.由于抛出异常之后的跳转, malloc和new出来的空间,没有进行释放,存在内存泄漏的问题。
2. 异常安全问题。如果在malloc和free之间如果存在抛异常,那么还是有内存泄漏。这种问题就叫异常安全,异常介绍可参见博客 C++ 异常
二、智能指针的使用及原理
2.1 RAII
RAII(Resource Acquisition Is Initialization)是一种利用对象生命周期来控制程序资源(如内存、文件句柄、网络连接、互斥量等等)的简单技术。
在对象构造时获取资源,接着控制对资源的访问使之在对象的生命周期内始终保持有效,最后在对象析构的时候释放资源。借此,我们实际上把管理一份资源的责任托管给了一个对象。这种做法有两大好处:
- 不需要显式地释放资源。
- 采用这种方式,对象所需的资源在其生命期内始终保持有效。
//设计一个SmartPtr类
template<class T>
class SmartPtr
{
public:
SmartPtr(T* ptr = nullptr)
:_ptr(ptr)
{}
~SmartPtr()
{
if (_ptr)
delete _ptr;
cout << "~SmatrPtr()" << endl;
}
private:
T* _ptr;
};
void Test()
{
int* tmp = (int*)malloc(sizeof(int)* 100);
//将tmp指针委托给sp对象,让sp对象帮助tmp指针管理空间,直到tmp指针被销毁
SmartPtr<int> sp(tmp);
//vector<int> v(100000000, 10);
string s = "程序出错";
throw s;
free(tmp);
}
int main()
{
try
{
Test();
}
catch (string& s)
{
cout << s << endl;
}
catch (...)
{
cout << "未知错误" << endl;
}
return 0;
}
- 这里我们看到虽然在抛出异常后程序跳转至catch处,在抛出异常处后面的代码并未执行,但是由于将tmp的空间交给对象sp管理,在程序执行完之后,对象会自动调用析构函数清理资源,此时指针tmp的空间得到了正确的释放,这就是智能指针的用处。
- 两种情况都会调用析构函数:
- 1、程序正常走完,出了作用域,sp对象被销毁调用析构函数清理资源
- 2、程序抛出异常,跳出作用域到catch处,此时出了作用域sp对象被销毁调用析构函数清理资源
2.2 智能指针的原理
- 上述的SmartPtr还不能将其称为智能指针,因为它还不具有指针的行为。指针可以解引用,也可以通过->去访问所指空间中的内容,因此: == AutoPtr模板类中还得需要将* 、->重载下,才可让其像指针一样去使用 ==。
template<class T>
class SmartPtr
{
public:
SmartPtr(T* ptr = nullptr)
:_ptr(ptr)
{}
~SmartPtr()
{
if (_ptr)
delete _ptr;
cout << "~SmatrPtr()" << endl;
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
private:
T* _ptr;
};
struct Date
{
int _year;
int _mouth;
int _day;
};
int main()
{
SmartPtr<int> sp1(new int);
//sp1.operator*() = 10;
*sp1 = 10;
cout << *sp1 << endl;
SmartPtr<Date> sp2(new Date);
//sp2.operator->()->_year = 2019;
sp2->_year = 2019;
sp2->_mouth = 10;
sp2->_day = 19;
return 0;
}
总结一下智能指针的原理:
- RAII特性,利用对象的生命周期来控制程序资源。
- 重载operator*和opertaor->,具有像指针一样的行为。
三、std::auto_ptr的使用和问题
- C++98版本的库中就提供了auto_ptr的智能指针
- auto_ptr 链接:http://www.cplusplus.com/reference/memory/auto_ptr/
- auto_ptr的实现原理就是:管理权转移的思想,下面我们来看一下这个智能指针的问题
- 模拟实现一份auto_ptr了解原理
举个栗子
// C++库中的智能指针都定义在memory这个头文件中
#include <memory>
class Date
{
public:
Date() { cout << "Date()" << endl; }
~Date(){ cout << "~Date()" << endl; }
int _year;
int _month;
int _day;
};
int main()
{
auto_ptr<Date> ap(new Date);
auto_ptr<Date> copy(ap);
ap->_year = 2019;
return 0;
}
模拟实现,请看代码:
// 1.C++98智能指针auto_ptr,管理权转移,设计缺陷,严禁使用
namespace CXY
{
template<class T>
class AutoPtr//模拟实现auto_ptr
{
public:
AutoPtr(T* Ptr)
:_Ptr(Ptr)
{}
~AutoPtr()
{
if (_Ptr)
{
cout << "deleteA:" << _Ptr << endl;
delete _Ptr;
}
}
// CXY::AutoPtr<int> copy(ap);拷贝构造
AutoPtr(AutoPtr<T>& ap)
{
_Ptr = ap._Ptr;
ap._Ptr = nullptr;
}
T& operator*()
{
return *_Ptr;
}
T* operator->()
{
return _Ptr;
}
private:
T* _Ptr;
};
}
四、 std::unique_ptr的使用和问题
- C++11中开始提供更靠谱的unique_ptr
- unique_ptr链接:http://www.cplusplus.com/reference/memory/shared_ptr/
- unique_ptr的实现原理:简单粗暴的防拷贝
- 下面简化模拟实现了一份UniquePtr来了解它的原理
举个栗子
#include <memory>
class Date
{
public:
Date() { cout << "Date()" << endl; }
~Date(){ cout << "~Date()" << endl; }
int _year;
int _month;
int _day;
};
int main()
{
unique_ptr<Date> up(new Date);
// unique_ptr的设计思路非常的粗暴-防拷贝,也就是不让拷贝和赋值。
unique_ptr<Date> copy(up);
return 0;
}
模拟实现,请看代码:
// 2.C++11 unique_ptr 防拷贝,简单粗暴的设计,鼓励使用
namespace CXY
{
template<class T>
class UniquePtr
{
public:
UniquePtr(T* ptr)
:_Ptr(ptr)
{}
~UniquePtr()
{
cout << "deleteU:" << _Ptr << endl;
delete _Ptr;
}
T& operator*()
{
return *_Ptr;
}
T* operator->()
{
return _Ptr;
}
private:
//C++98防拷贝方式,只声明不实现+声明为私有
//必须声明为私有,这样类外也无法实现,才能防拷贝
//只声明编译无错误,为了防止系统生成默认的拷贝构造函数
UniquePtr(UniquePtr<int> const& up);
UniquePtr<int>& operator=(UniquePtr<int> const& up);
//C++11防拷贝方式,delete
//UniquePtr(UniqurPtr<int> const&)up = delete;
//UniquePtr<int>& operator=(UniquePtr<int> const& up) = delete;
T* _Ptr;
};
}
五、 std::shared_ptr的使用与问题
- C++11中开始提供更靠谱的并且支持拷贝的shared_ptr
- shared_ptr链接: http://www.cplusplus.com/reference/memory/shared_ptr/
- shared_ptr的原理:是通过引用计数的方式来实现多个shared_ptr对象之间共享资源。
- shared_ptr在其内部,给每个资源都维护了着一份计数,用来记录该份资源被几个对象共享。
- 在对象被销毁时(也就是析构函数调用),就说明自己不使用该资源了,对象的引用计数减一。
- 如果引用计数是0,就说明自己是最后一个使用该资源的对象,必须释放该资源;
- 如果不是0,就说明除了自己还有其他对象在使用该份资源,不能释放该资源,否则其他对象就成野指针了。
模拟实现shared_ptr,请看代码:
// 3.C++11 shared_ptr 引用计数,支持拷贝,鼓励使用
namespace CXY
{
template <class T>
class SharedPtr
{
public:
SharedPtr(T* ptr)
:_Ptr(ptr)
, _pcount(new int(1))
, _pmtx(new mutex)//处理线程安全问题,保证只有一个线程对pcount进行++
{}
~SharedPtr()
{
/*if (--(*_pcount) == 0)
{
cout << "delete:" << _ptr << endl;
delete _ptr;
delete _pcount;
}*/
Release();
}
//CXY::SharedPtr<int> copy(sp);
SharedPtr(SharedPtr<int> const& sp)
:_Ptr(sp._Ptr)
, _pcount(sp._pcount)
, _pmtx(sp._pmtx)
{
if (_Ptr)//若是一个空指针则不用计数
AddRefcount();
}
//sp = sp1
SharedPtr<int>& operator=(SharedPtr<int> const& sp)
{//先检查sp是否是最后一个指向资源的对象,如果是先释放资源
if (_Ptr != sp._Ptr)//如果不是相同的对象才进行赋值
{
Release();//释放原来的空间,指向新的空间
_Ptr = sp._Ptr;
_pcount = sp._pcount;
_pmtx = sp._pmtx;
AddRefcount();
}
return *this;
}
int AddRefcount()
{
//保证是原子操作
_pmtx->lock();
++(*_pcount);
_pmtx->unlock();
return *_pcount;
}
void Release()
{
bool deleteflag = false;
//引用计数减一,如果减到0即是最后一个使用资源的对象,就释放资源
if (--(*_pcount) == 0)
{
cout << "deleteS:" << _Ptr << endl;
delete _Ptr;
delete _pcount;
deleteflag = true;
}
if (deleteflag == true)//如果是最后一个对象,也把锁资源释放
delete _pmtx;
}
T& operator*()
{
return *_Ptr;
}
T* operator->()
{
return _Ptr;
}
int use_count()
{
return *_pcount;
}
private:
T* _Ptr;
int* _pcount;
mutex* _pmtx;
};
}
std::shared_ptr的线程安全问题
需要注意的是shared_ptr的线程安全分为两方面:
- 智能指针对象中引用计数是多个智能指针对象共享的,两个线程中智能指针的引用计数同时++或–,这个操作不是原子的,引用计数原来是1,++了两次,可能还是2.这样引用计数就错乱了。会导致资源未释放或者程序崩溃的问题。所以只能指针中引用计数++、–是需要加锁的,也就是说引用计数的操作是线程安全的。
- 智能指针管理的对象存放在堆上,两个线程中同时去访问,会导致线程安全问题。
- 注意一点,不仅是双核CPU有线程安全问题,单核CPU也有线程安全问题,由于单核CPU线程在执行时会有时间片概念,一个线程在执行时若是时间片用完了会暂时休眠,让其他进程执行,等下一个时间片再执行,此时就会造成和双核CPU同样的线程安全问题。
void SharePtrFunc(SharedPtr<Date>& sp, size_t n)
{
cout << sp.Get() << endl;
for (size_t i = 0; i < n; ++i)
{
// 这里智能指针拷贝会++计数,智能指针析构会--计数,这里是线程安全的。
//SharedPtr<Date> copy(sp);
// 这里智能指针访问管理的资源,不是线程安全的。所以我们看看这些值两个线程++了2n次,但是最
//终看到的结果,并一定是加了2n
copy->_year++;
copy->_month++;
copy->_day++;
}
}
int main()
{
SharedPtr<Date> p(new Date);
cout << p.Get() << endl;
const size_t n = 100;
thread t1(SharePtrFunc, p, n);
thread t2(SharePtrFunc, p, n);
t1.join();
t2.join();
cout << p->_year << endl;
cout << p->_month << endl;
cout << p->_day << endl;
return 0;
}
智能指针的循环引用问题
我们先来看一段代码了解一下什么是循环引用问题
#include <iostream>
#include <memory>
using namespace std;
struct ListNode
{
int _data;
shared_ptr<ListNode> _prev;
shared_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;
}
循环引用的解决方案weak_ptr
- weak_ptr其实并不属于指针指针,它并不符合RAII的特性
- 只有识别到循环引用的场景
- 使用weak_ptr
// 解决方案:在引用计数的场景下,把节点中的_prev和_next改成weak_ptr就可以了
// 原理就是,node1->_next = node2;和node2->_prev = node1;时weak_ptr的_next和_prev不会增加
//node1和node2的引用计数。
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;
}