C++智能指针
内存泄漏
C/C++程序中一般我们关心两种方面的内存泄漏:
-
堆内存泄漏(Heap leak)
堆内存指的是程序执行中依据须要分配通过malloc / calloc / realloc / new等从堆中分配的一块内存,用完后必须通过调用相应的 free或者delete 删掉。假设程序的设计错误导致这部分内存没有被释放,那么以后这部分空间将无法再被使用,就会产生Heap Leak。
-
系统资源泄漏
指程序使用系统分配的资源,比方套接字、文件描述符、管道等没有使用对应的函数释放掉,导致系统资源的浪费,严重可导致系统效能减少,系统执行不稳定。
如何避免内存泄漏
- 工程前期良好的设计规范,养成良好的编码规范,申请的内存空间记着匹配的去释放。ps:这个理想状态。但是如果碰上异常时,就算注意释放了,还是可能会出问题。需要下一条智能指针来管理才有保证。
- 采用RAII思想或者智能指针来管理资源。
- 有些公司内部规范使用内部实现的私有内存管理库。这套库自带内存泄漏检测的功能选项。
- 出问题了使用内存泄漏工具检测。ps:不过很多工具都不够靠谱,或者收费昂贵。
总结一下:
内存泄漏非常常见,解决方案分为两种:1、事前预防型。如智能指针等。2、事后查错型。如泄漏检测工具。
智能指针的使用及原理
RAll
RAII(Resource Acquisition Is Initialization)是一种利用对象生命周期来控制程序资源(如内存、文件句柄、网络连接、互斥量等等)的简单技术。
在对象构造时获取资源,接着控制对资源的访问使之在对象的生命周期内始终保持有效,最后在对象析构的时候释放资源。借此,我们实际上把管理一份资源的责任托管给了一个对象。这种做法有两大好处:
- 不需要显式地释放资源。
- 采用这种方式,对象所需的资源在其生命期内始终保持有效。
// 使用RAII思想设计的SmartPtr类
template<class T>
class SmartPtr
{
public:
SmartPtr(T* Ptr = nullptr)
:_Ptr(Ptr)
{}
~SmartPtr()
{
if(_Ptr)
delete _Ptr;
}
private:
T* _Ptr;
};
上述的SmartPtr还不能将其称为智能指针,因为它还不具有指针的行为。指针可以解引用,也可以通过->去访问所指空间中的内容,因此:AutoPtr模板类中还得需要将、->重载下,才可让其像指针一样去使用。
template<class T>
class SmartPtr
{
public:
SmartPtr(T* Ptr)
:_Ptr(Ptr)
{}
~SmartPtr()
{
cout << "delete:" << _Ptr << endl;
delete _Ptr;
}
SmartPtr(const SmartPtr<T>& SmtPtr)
{
T t = *SmtPtr;
SmartPtr<T> tmp(new T(t));
swap(_Ptr, tmp._Ptr);
}
T& operator*()
{
return *_Ptr;
}
T* operator->()
{
return _Ptr;
}
private:
T* _Ptr;
};
总结一下智能指针的原理:
- RAII特性
- 重载operator*和opertaor->,具有像指针一样的行为。
std::auto_ptr
C++98版本的库中就提供了auto_ptr的智能指针。下面演示的auto_ptr的使用及问题。auto_ptr的实现原理:管理权转移的思想,下面简化模拟实现了一份bit::auto_ptr来了解它的原理
// C++98 管理权转移 auto_ptr
template<class T>
class auto_ptr
{
public:
auto_ptr(T* Ptr)
:_Ptr(Ptr)
{}
~auto_ptr()
{
if (_Ptr)
{
cout << "delete:" << _Ptr << endl;
delete _Ptr;
}
}
//管理权转移 转移ap中资源到当前对象中
auto_ptr(auto_ptr<T>& ap)
:_Ptr(ap._Ptr)
{
ap._Ptr = nullptr;
}
T& operator*()
{
return *_Ptr;
}
T* operator->()
{
return _Ptr;
}
private:
T* _Ptr;
};
结论:auto_ptr是一个失败设计(ptr1=pt2;执行后会导致ptr2悬空),很多公司明确要求不能使用auto_ptr。
std::unique_ptr
C++11中开始提供更靠谱的unique_ptr
unique_ptr的实现原理:简单粗暴的防拷贝,下面简化模拟实现了一份UniquePtr来了解它的原理
// C++11库才更新智能指针实现
// C++11出来之前,boost搞除了更好用的scoped_ptr/shared_ptr/weak_ptr
// C++11将boost库中智能指针精华部分吸收了过来
// C++11->unique_ptr/shared_ptr/weak_ptr
// unique_ptr/scoped_ptr
// 原理:简单粗暴 -- 防拷贝
template<class T>
class unique_ptr
{
public:
unique_ptr(T* Ptr)
:_Ptr(Ptr)
{}
~unique_ptr()
{
if (_Ptr)
{
cout << "delete:" << _Ptr << endl;
delete _Ptr;
}
}
//管理权转移
unique_ptr(unique_ptr<T>& up) = delete;
unique_ptr<T>& operator=(unique_ptr<T>& up) = delete;
T& operator*()
{
return *_Ptr;
}
T* operator->()
{
return _Ptr;
}
private:
T* _Ptr;
};
std::shared_ptr
C++11中开始提供更靠谱的并且支持拷贝的shared_ptr
shared_ptr的原理:是通过引用计数的方式来实现多个shared_ptr对象之间共享资源。
- shared_ptr在其内部,给每个资源都维护了着一份计数,用来记录该份资源被几个对象共享。
- 在对象被销毁时(也就是析构函数调用),就说明自己不使用该资源了,对象的引用计数减一。
- 如果引用计数是0,就说明自己是最后一个使用该资源的对象,必须释放该资源;
- 如果不是0,就说明除了自己还有其他对象在使用该份资源,不能释放该资源,否则其他对象就成野指针了。
- 为了维护线程安全问题,需要对Release()和AddCount()加锁
下面简化模拟实现了一份shared_ptr来了解它的原理
// shared_ptr智能指针是线程安全的吗?
// 是的,引用计数的加减是加锁保护的。但是指向资源不是线程安全的
// 指向堆上资源的线程安全问题是访问的人处理的,智能指针不管,也管不了
// 引用计数的线程安全问题,是智能指针要处理的
template<class T>
class shared_ptr
{
public:
shared_ptr(T* Ptr = nullptr)
:_Ptr(Ptr)
,_pcount(new int(1))
,_pmtx(new mutex)
{}
template<class D>
shared_ptr(T* Ptr, D del)
:_Ptr(Ptr)
, _pcount(new int(1))
, _pmtx(new mutex)
,_del(del) //用function包装传入的可执行对象del
{}
~shared_ptr()
{
Release();
}
shared_ptr(const shared_ptr<T>& sp)
:_Ptr(sp._Ptr)
, _pcount(sp._pcount)
, _pmtx(sp._pmtx)
{
AddCount();
}
shared_ptr<T>& operator=(const shared_ptr<T>& sp)
{
if (_Ptr != sp._Ptr)
{
Release();
_Ptr = sp._Ptr;
_pcount = sp._pcount;
_pmtx = sp._pmtx;
AddCount();
}
return *this;
}
void Release()
{
_pmtx->lock();
bool deleteFlag = false;
if (--(*_pcount) == 0)
{
if (_Ptr)
{
//cout << "delete:" << _Ptr << endl;
//delete _Ptr;
_del(_Ptr);
}
delete _pcount;
deleteFlag = true;
}
_pmtx->unlock();
if (deleteFlag)
delete _pmtx;
}
void AddCount()
{
_pmtx->lock();
++(*_pcount);
_pmtx->unlock();
}
T& operator*()
{
return *_Ptr;
}
T* operator->()
{
return _Ptr;
}
T* get() const
{
return _Ptr;
}
int use_count()
{
return *_pcount;
}
private:
T* _Ptr;
int* _pcount;
mutex* _pmtx;
function<void(T*)> _del = [](T* ptr) { delete ptr; }; //用function包装传入的可执行对象del
};
function<void(T*)> _del 是删除器,默认删除方式为delete ptr,使用者可以自己定义其他的删除方式如:
template <class T>
struct DeleteArray
{
void operator()(T* ptr)
{
delete[] ptr; //删除数组
}
};
传参时传入就好 shared_ptr<int> sp(new int[10],DeleteArray<int>());
std::shared_ptr的循环引用:
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;
}
循环引用分析:
- node1和node2两个智能指针对象指向两个节点,引用计数变成1,我们不需要手动delete。
- node1的_next指向node2,node2的_prev指向node1,引用计数变成2。
- node1和node2析构,引用计数减到1,但是_next还指向下一个节点。但是_prev还指向上一个节点。
- 也就是说_next析构了,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的引用计数。