目录
为什么需要智能指针
抛异常引发内存泄漏
内存泄漏
什么是内存泄漏,内存泄漏的危害
内存泄漏分类
检测内存泄漏常用工具
如何避免内存泄漏
智能指针的使用及原理
RAII
智能指针的原理
各类智能指针介绍
auto_ptr
unique_ptr
weak_ptr
定制删除器
对于下面这段代码,存在什么问题?
int div()
{
int a, b;
cin >> a >> b;
if (b == 0)
throw invalid_argument("除0错误");
return a / b;
}
void Func()
{
// 1、如果p1这里new 抛异常会如何?
// 2、如果p2这里new 抛异常会如何?
// 3、如果div调用这里又会抛异常会如何?
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;
}
分析上述代码,如果在p1处抛异常,代码直接跳转到main函数中捕获异常,不会造成资源泄漏。但是如果在p2被new时,或者调用div()函数时抛异常,代码跳转到main()函数,两个delete语句会被跳过,导致内存泄漏。
内存泄漏
内存泄漏指因为疏忽或错误造成程序未能释放已经不再使用的内存的情况。内存泄漏并不是指内存在物理上的消失,而是应用程序分配某段内存后,因为设计错误,失去了对该段内存的控制,因而造成了内存的浪费。
内存泄漏的危害
长期运行的程序出现内存泄漏,影响很大,如操作系统、后台服务等等,出现内存泄漏会导致响应越来越慢,最终卡死。
常见的内存泄漏
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;
}
C/C++程序中一般我们关心两种方面的内存泄漏:
以下几种工具可以用来帮助检测是否存在内存泄漏
【总结】
内存泄漏非常常见,解决方案分为两种:1、事前预防型:如智能指针等。 2、事后查错型:“如泄漏检测工具。
RAII(Resource Acquisition Is Initialization)是一种利用对象生命周期来控制程序资源(如内
存、文件句柄、网络连接、互斥量等等)的简单技术。
在对象构造时获取资源,接着控制对资源的访问使之在对象的生命周期内始终保持有效,最后在
对象析构的时候释放资源。借此,我们实际上把管理一份资源的责任托管给了一个对象。这种做
法有两大好处:
// 使用RAII思想设计的SmartPtr类
template
class SmartPtr {
public:
SmartPtr(T* ptr = nullptr)
: _ptr(ptr)
{}
~SmartPtr()
{
if (_ptr)
delete _ptr;
}
private:
T* _ptr;
};
上述的SmartPtr还不能将其称为智能指针,上述模拟实习的智能指针只用到了RAII的思想,即构造对象时获取资源、析构对象时回收资源,但是它还不具有指针的行为。指针可以解引用,也可以通过->去访问所指空间中的内容,因此:AutoPtr模板类中还得需要将* 、->重载下,才可让其像指针一样去使用。
template
class SmartPtr {
public:
SmartPtr(T* ptr = nullptr)
: _ptr(ptr)
{}
~SmartPtr()
{
if (_ptr)
delete _ptr;
}
T& operator*() { return *_ptr; }
T* operator->() { return _ptr; }
private:
T* _ptr;
};
总结一下智能指针的原理:
早在C++98中就已经出现了智能指针的概念,同时库中也提供了auto_ptr的智能指针。当时采用的是管理权转移的思想,下面简化模拟实现了一份auto_ptr来了解它的原理。
template
class auto_ptr
{
public:
//RAII
//保存资源
auto_ptr(T* ptr)
:_ptr(ptr)
{}
//释放资源
~auto_ptr()
{
delete _ptr;
}
//auto_ptr采用资源管理权转移 -- 但是会导致对象悬空
auto_ptr(auto_ptr& sp)
:_ptr(sp._ptr)
{
sp._ptr = nullptr;
}
//智能指针要有指针的功能
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
T& operator[](size_t pos)
{
return _ptr[pos];
}
private:
T* _ptr;
};
auto_ptr采用了RAII思想,可以解决抛异常引发的内存泄漏问题,但同时又造成了新的问题。auto_ptr在实现拷贝构造函数时采用了管理权转移的方式。
相信大伙一眼就能看出这种实现思想有很坑的一点,用一个旧的指针去构造一个新的指针之后,只有新的指针指向了内容,旧的指针指向了空,但我们仍可能去解引用旧的指针啊。例如下面的例子一定会引发错误。
void test_auto()
{
auto_ptr ap1(new int(1));
auto_ptr ap2(ap1);
*ap1 = 1; // 管理权转移以后导致ap1悬空,不能访问
*ap2 = 1;
}
auto_ptr这一问题也是被骂了很长时间,C++标准委员会也是直接表明不要使用auto_ptr。
unique_ptr是C++11标准库提供的一个靠谱的智能指针,unique_ptr既体现了RAII的思想,同时也没有auto_ptr的问题,因为它直接就没有拷贝构造函数。
unique_ptr的实现原理:简单粗暴的防拷贝,下面简化模拟实现了一份unique_ptr来了解它的原理。
template
class unique_ptr
{
public:
//RAII
//保存资源
unique_ptr(T* ptr)
:_ptr(ptr)
{}
//释放资源
~auto_ptr()
{
delete _ptr;
}
//智能指针要有指针的功能
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
T& operator[](size_t pos)
{
return _ptr[pos];
}
//C++11:可以用=delete直接封死拷贝构造和赋值构造
unique_ptr(const unique_ptr& up) = delete;
unique_ptr& operator=(const unique_ptr& up) = delete;
private:
//防拷贝
// 拷贝构造和赋值是默认成员函数,我们不写会自动生成,所以我们不需写
// C++98思路:只声明不实现,但是用的人可能会在外面强行定义,所以再加一条,声明为私有
//unique_ptr(const unique_ptr& up);
//unique_ptr& operator=(const unique_ptr& up);
T* _ptr;
};
C++11中除了提供了unique_ptr还提供更靠谱的并且支持拷贝的shared_ptr。
shared_ptr的原理:是通过引用计数的方式来实现多个shared_ptr对象之间共享资源。
实现拷贝构造时采用引用计数的方式,在实现赋值重载时,要注意等号左边的指针指向的内容是否需要被释放。
情况一:
情况二:
shared_ptr具体实现代码:
template
class shared_ptr {
public:
//RAII
//保存资源
shared_ptr(T* ptr)
:_ptr(ptr)
,_pcount(new int(1))
{}
//释放资源
~shared_ptr()
{
Release();
}
shared_ptr(const shared_ptr& sp)
:_ptr(sp._ptr)
, _pcount(sp._pcount)
{
AddCount();
}
shared_ptr& operator=(const shared_ptr& sp)
{
if (sp._ptr != _ptr)
{
Release();
_ptr = sp._ptr;
_pcount = sp._pcount;
AddCount();
}
return *this;
}
void Release()
{
if (--(*_pcount) == 0)
{
delete _ptr;
delete _pcount;
}
}
void AddCount()
{
++(*_pcount);
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
private:
T* _ptr;
int* _pcount;
};
上述的代码已经可以满足一般的应用场景,但是设计到多线程操作时,上面的shared_ptr存在线程安全的问题,所以我们最好还是实现通过加锁实现一个线程安全的版本。
template
class shared_ptr {
public:
//RAII
//保存资源
shared_ptr(T* ptr)
:_ptr(ptr)
,_pcount(new int(1))
,_pmtx(new mutex)
{}
//释放资源
~shared_ptr()
{
Release();
}
shared_ptr(const shared_ptr& sp)
:_ptr(sp._ptr)
, _pcount(sp._pcount)
,_pmtx(sp._pmtx)
{
AddCount();
}
shared_ptr& operator=(const shared_ptr& sp)
{
if (sp._ptr != _ptr)
{
Release();
_ptr = sp._ptr;
_pcount = sp._pcount;
_pmtx = sp._pmtx;
AddCount();
}
return *this;
}
void Release()
{
_pmtx->lock();
if (--(*_pcount) == 0)
{
delete _ptr;
delete _pcount;
}
_pmtx->unlock();
}
void AddCount()
{
_pmtx->lock();
++(*_pcount);
_pmtx->unlock();
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
T* get()
{
return _ptr;
}
int use_count()
{
return *_pcount;
}
private:
T* _ptr;
int* _pcount;
mutex* _pmtx;
};
循环引用的情况:
比如上面这个双向链表,我们一般就是这样使用,但是因为抛异常可能导致内存泄漏的问题,于是我们改用智能指针。n1和n2不构成循环时可以正常使用智能指针。
n1和n2构成循环引用时,智能指针无法释放空间,造成内存泄漏。
原因是
解决方案:在引用计数的场景下,把节点中的_prev和_next改成weak_ptr就可以了
weak_ptr原理就是,node1->_next = node2;和node2->_prev = node1;时weak_ptr的_next和
_prev不会增加node1和node2的引用计数。
【注意】
weak_ptr不是常规的智能指针,不支持RAII ,但是它支持像指针一样
专门设计出来,辅助解决shared_ptr的循环引用问题,
weak_ptr可以指向资源,但是他不参与管理,不增加引用计数
template
class weak_ptr
{
public:
weak_ptr()
:_ptr(nullptr)
{}
weak_ptr(const shared_ptr& sp)
:_ptr(sp.get())
{}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
T* get()
{
return _ptr;
}
private:
T* _ptr;
};
shared_ptr和weak_ptr配合使用就可以解决循环引用的问题了。
如果不是new出来的对象如何通过智能指针管理呢?其实shared_ptr设计了一个删除器来解决这
个问题。
// 仿函数的删除器
template
struct FreeFunc {
void operator()(T* ptr)
{
cout << "free:" << ptr << endl;
free(ptr);
}
};
template
struct DeleteArrayFunc {
void operator()(T* ptr)
{
cout << "delete[]" << ptr << endl;
delete[] ptr;
}
};
定制删除器版本shared_ptr
template
class shared_ptr
{
public:
shared_ptr(T* ptr = nullptr)
:_ptr(ptr)
, _pcount(new int(1))
, _pmtx(new mutex)
{}
template
shared_ptr(T* ptr, D del)
: _ptr(ptr)
, _pcount(new int(1))
, _pmtx(new mutex)
, _del(del)
{}
~shared_ptr()
{
Release();
}
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();
}
shared_ptr(const shared_ptr& sp)
:_ptr(sp._ptr)
, _pcount(sp._pcount)
, _pmtx(sp._pmtx)
{
AddCount();
}
// sp1 = sp4
// sp1 = sp1;
// sp1 = sp2;
shared_ptr& operator=(const shared_ptr& sp)
{
if (_ptr != sp._ptr)
{
Release();
_ptr = sp._ptr;
_pcount = sp._pcount;
_pmtx = sp._pmtx;
AddCount();
}
return *this;
}
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 _del = [](T* ptr){
cout << "lambda delete:" << ptr << endl;
delete ptr;
};
};
运行结果: