- 解决忘记释放内存导致内存泄漏的问题。
- 解决异常安全问题。
#include
using namespace std;
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
这里new 抛异常会如何?答:
p1、p2
不会开空间,内存没有释放问:如果
p2
这里new 抛异常会如何?答:
p2
不会开空间,内存没有得到释放问:如果div调用这里又会抛异常会如何?
答:内存没有被释放。
那么如何解决呢?
可以利用智能指针来解决这个问题。
RAII(Resource Acquisition Is Initialization)是一种利用对象生命周期来控制程序资源(如内 存、文件句柄、网络连接、互斥量等等)的简单技术。
在对象构造时获取资源,接着控制对资源的访问使之在对象的生命周期内始终保持有效,最后在 对象析构的时候释放资源。借此,我们实际上把管理一份资源的责任托管给了一个对象。这种做 法有两大好处:
- 不需要显示地释放资源。
- 采用这种方式,对象所需的资源在其生命期内始终保持有效。
我们用一个类来封装一下这个指针,实现如下:
#include
using namespace std;
namespace hayaizo
{
template<class T>
class smart_ptr
{
public:
smart_ptr(T* ptr = nullptr)
:_ptr(ptr)
{}
~smart_ptr()
{
if (_ptr)
{
cout << "delete: " << _ptr << endl;
delete _ptr;
}
}
private:
T* _ptr;
};
}
int div()
{
int a, b;
cin >> a >> b;
if (b == 0)
throw invalid_argument("除0错误");
return a / b;
}
void Func()
{
hayaizo::smart_ptr<int> sp1(new int);
hayaizo::smart_ptr<int> sp2(new int);
cout << div() << endl;
}
int main(void)
{
try {
Func();
}
catch (const exception& e)
{
cout << e.what() << endl;
}
return 0;
}
运行结果:
既然是指针,那么也需要支持解引用
*
和->
,重载这两个符号就好了。
namespace hayaizo
{
template<class T>
class smart_ptr
{
public:
smart_ptr(T* ptr = nullptr)
:_ptr(ptr)
{}
~smart_ptr()
{
if (_ptr)
{
cout << "delete: " << _ptr << endl;
delete _ptr;
}
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
private:
T* _ptr;
};
}
但这样地封装会有一个致命的缺点,一个地址只能被一个智能指针指向,不然会导致同一块内存释放两次的问题,我们看看官方库中的
auto_ptr
是怎么解决的。
可以看到,sp1
的地址变成了sp2
的地址了,然后sp1
的地址变成了nullptr
。
手搓一个低配版的
auto_ptr
:
namespace hayaizo
{
template<class T>
class auto_ptr
{
public:
auto_ptr(T* ptr = nullptr)
:_ptr(ptr)
{}
auto_ptr(auto_ptr<T>&ap)
:_ptr(ap._ptr)
{
ap._ptr = nullptr;
}
auto_ptr<T> operator=(auto_ptr<T>& ap)
{
if (this != &ap)
{
if (_ptr)
{
delete _ptr;
}
_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;
}
private:
T* _ptr;
};
}
但这样确实可以解决问题,但是已经失去了原生指针的功能了,原生指针是支持同一个地址被很多个指针指向的,在介绍解决方法之前得先介绍
unique_ptr
。
unique_ptr
非常粗暴,直接不让你拷贝,把拷贝构造禁掉了。
template<class T>
class unique_ptr
{
public:
unique_ptr(T* ptr = nullptr)
:_ptr(ptr)
{}
unique_ptr(unique_ptr<T>& ap)=delete
{}
unique_ptr<T> operator=(unique_ptr<T>& ap)=delete
{}
~unique_ptr()
{
if (_ptr)
{
cout << "delete: " << _ptr << endl;
delete _ptr;
}
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
private:
T* _ptr;
};
shared_ptr的原理:是通过引用计数的方式来实现多个shared_ptr对象之间共享资源。
- shared_ptr在其内部,给每个资源都维护了着一份计数,用来记录该份资源被几个对象共 享。
- 在对象被销毁时(也就是析构函数调用),就说明自己不使用该资源了,对象的引用计数减 一。
- 如果引用计数是0,就说明自己是最后一个使用该资源的对象,必须释放该资源。
- 如果不是0,就说明除了自己还有其他对象在使用该份资源,不能释放该资源,否则其他对 象就成野指针了。
比如我用三个智能指针指向地址
0x00112233
,因此,计数为3,当计数等于0的时候再进行销毁。那么,这里的计数可以单纯用一个
int _cnt
或者static int _cnt
来表示吗?答案是否定的。
如果是
int _cnt
,那么每个对象都是单独的计数。如果是``static int _cnt`,那么每个对象都是用的同一份计数。
所以这里需要用一个指针来表示计数。
template<class T>
class shared_ptr
{
public:
shared_ptr(T* ptr = nullptr)
:_ptr(ptr)
,_cnt(new int(1))
{}
shared_ptr(shared_ptr<T>& ap)
:_ptr(ap._ptr)
,_cnt(ap._cnt)
{}
void Release()
{
if (--(*_cnt) == 0)
{
cout << "delete: " << _ptr << endl;
delete _ptr;
}
}
shared_ptr<T>& operator=(const shared_ptr<T>& sp)
{
//p1=p1的情况
if (_ptr == sp._ptr)
{
return *this;
}
Release();
_ptr = sp._ptr;
_cnt = sp._cnt;
(*_cnt)++;
return *this;
}
int use_count()
{
return *_cnt;
}
T* get() const
{
return _ptr;
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
~shared_ptr()
{
Release();
}
private:
T* _ptr;
int* _cnt;
};
}
这里的
n1
和n2
的引用计数都是2,所以形成了相互制约的局面。
n1
的销毁看n2
,n2
的销毁看n1
。
weak_ptr是为配合shared_ptr而引入的一种智能指针。
weak_ptr可以从一个shared_ptr或另一个weak_ptr对象构造,它的构造和析构不会引起shared_ptr引用记数的增加或减少
template<class T>
class weak_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;
}
private:
T* _ptr;
};
struct Node
{
int _val;
hayaizo::weak_ptr<Node> _next;
hayaizo::weak_ptr<Node> _prev;
~Node()
{
cout << "~Node" << endl;
}
};
int main(void)
{
hayaizo::shared_ptr<Node> n1(new Node);
hayaizo::shared_ptr<Node> n2(new Node);
n1->_next = n2;
n2->_prev = n1;
return 0;
}
很简单,就是
n1
内部的指针不参与引用计数,用另外的类封装起来就好了,就不会动shared_ptr
里面的引用计数了。
其实就是个仿函数,可以自己传删除方案。
//默认删除器
template<class T>
struct Delete
{
void operator()(T* ptr)
{
cout << "delete: " << ptr << endl;
delete ptr;
}
};
template<class T,class D=Delete<T>>
class shared_ptr
{
public:
shared_ptr(T* ptr = nullptr)
:_ptr(ptr)
,_cnt(new int(1))
{}
shared_ptr(shared_ptr<T>& ap)
:_ptr(ap._ptr)
,_cnt(ap._cnt)
{}
void Release()
{
if (--(*_cnt) == 0)
{
cout << "delete: " << _ptr << endl;
D del;
del(_ptr);
//D()(_ptr);匿名对象调用()
}
}
shared_ptr<T>& operator=(const shared_ptr<T>& sp)
{
//p1=p1的情况
if (_ptr == sp._ptr)
{
return *this;
}
Release();
_ptr = sp._ptr;
_cnt = sp._cnt;
(*_cnt)++;
return *this;
}
int use_count()
{
return *_cnt;
}
T* get() const
{
return _ptr;
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
~shared_ptr()
{
Release();
}
private:
T* _ptr;
int* _cnt;
};
template<class T>
struct DeleteArray
{
void operator()(T* ptr)
{
cout << "delete[]" << ptr << endl;
delete[] ptr;
}
};
template<class T>
struct Free
{
void operator()(T* ptr)
{
cout << "free" << ptr << endl;
free(ptr);
}
};
//#include
//using namespace std;
//
//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;
//}
#include
using namespace std;
namespace hayaizo
{
template<class T>
class auto_ptr
{
public:
auto_ptr(T* ptr = nullptr)
:_ptr(ptr)
{}
auto_ptr(auto_ptr<T>&ap)
:_ptr(ap._ptr)
{
ap._ptr = nullptr;
}
auto_ptr<T> operator=(auto_ptr<T>& ap)
{
if (this != &ap)
{
if (_ptr)
{
delete _ptr;
}
_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;
}
private:
T* _ptr;
};
template<class T>
class unique_ptr
{
public:
unique_ptr(T* ptr = nullptr)
:_ptr(ptr)
{}
unique_ptr(unique_ptr<T>& ap)=delete
{}
unique_ptr<T> operator=(unique_ptr<T>& ap)=delete
{}
~unique_ptr()
{
if (_ptr)
{
cout << "delete: " << _ptr << endl;
delete _ptr;
}
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
private:
T* _ptr;
};
//默认删除器
template<class T>
struct Delete
{
void operator()(T* ptr)
{
cout << "delete: " << ptr << endl;
delete ptr;
}
};
template<class T,class D=Delete<T>>
class shared_ptr
{
public:
shared_ptr(T* ptr = nullptr)
:_ptr(ptr)
,_cnt(new int(1))
{}
shared_ptr(shared_ptr<T>& ap)
:_ptr(ap._ptr)
,_cnt(ap._cnt)
{}
void Release()
{
if (--(*_cnt) == 0)
{
cout << "delete: " << _ptr << endl;
D del;
del(_ptr);
//D()(_ptr);匿名对象调用()
}
}
shared_ptr<T>& operator=(const shared_ptr<T>& sp)
{
//p1=p1的情况
if (_ptr == sp._ptr)
{
return *this;
}
Release();
_ptr = sp._ptr;
_cnt = sp._cnt;
(*_cnt)++;
return *this;
}
int use_count()
{
return *_cnt;
}
T* get() const
{
return _ptr;
}
T& operator*()
{
return *_ptr;
}
T* operator->()
{
return _ptr;
}
~shared_ptr()
{
Release();
}
private:
T* _ptr;
int* _cnt;
};
template<class T>
class weak_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;
}
private:
T* _ptr;
};
}
struct Node
{
int _val;
hayaizo::weak_ptr<Node> _next;
hayaizo::weak_ptr<Node> _prev;
~Node()
{
cout << "~Node" << endl;
}
};
template<class T>
struct DeleteArray
{
void operator()(T* ptr)
{
cout << "delete[]" << ptr << endl;
delete[] ptr;
}
};
template<class T>
struct Free
{
void operator()(T* ptr)
{
cout << "free" << ptr << endl;
free(ptr);
}
};
int main(void)
{
/*hayaizo::shared_ptr n1(new Node);
hayaizo::shared_ptr n2(new Node);
n1->_next = n2;
n2->_prev = n1;*/
hayaizo::shared_ptr<Node, DeleteArray<Node>> n1(new Node[5]);
hayaizo::shared_ptr<Node> n2(new Node);
hayaizo::shared_ptr<int, DeleteArray<int>> n3(new int[5]);
hayaizo::shared_ptr<int, Free<int>> n4((int*)malloc(sizeof(int)));
return 0;
}