C++智能指针auto_ptr源码完全解析---以微软auto_ptr为例来探讨auto_ptr的用法
对于C/C++程序员来说， 内存泄露是一个谈之色变的话题， 很多时候， 机器运行1天2天都是ok的， 但运行到一个星期后， 就卡得要死。 实际上， 很多时候是内存泄露造成的。 内存泄露很容易引入， 但是定位起来非常非常难， 在内存泄露初期， 通常没有异常症状， 但随着内存泄露的累积， 内存逐渐被啃光， 最终导致卡死或者死机。
申请堆内存， 又没有正确释放， 就会导致内存泄露， 听起来好可怕啊， 那有没有什么办法可以解决这一难题呢？ 有的！ 人类的智慧还是很厉害的。 我么知道， 栈对象在离开其作用域的时候， 会自动调用析构函数， 所以， 可以考虑把某一栈对象与某一堆内存绑定，且在其析构函数中释放堆内存， 那么， 在该栈对象离开作用域时， 堆内存自动释放， 这就是智能指针(本质是栈对象)的原理，简直是妙招啊。 这个栈对象装得像指针一样， 所以我们称之为智能指针， 其实， 它不过就是个普通的栈对象而已。
在本文中， 我们来介绍智能指针中的一种------auto_ptr, 并从源码的角度来看看auto_ptr使用中存在的一些常见问题：
我们先来看一段简单的程序：
#include <iostream>
#include <memory> // 有auto_ptr这个类模板
using namespace std;
int main()
{
auto_ptr<int> p(new int(100));
// p是个对象， 但重载了*运算符， 所以我说它很能装， 装得像个指针
cout << *p << endl; // 100
return 0;
}
#include <iostream>
using namespace std;
// 微软能把代码写成这样， 也是够风骚的
// TEMPLATE CLASS auto_ptr
template<class _Ty>
class auto_ptr {
public:
typedef _Ty element_type;
explicit auto_ptr(_Ty *_P = 0) _THROW0()
: _Owns(_P != 0), _Ptr(_P) {}
auto_ptr(const auto_ptr<_Ty>& _Y) _THROW0()
: _Owns(_Y._Owns), _Ptr(_Y.release()) {}
auto_ptr<_Ty>& operator=(const auto_ptr<_Ty>& _Y) _THROW0()
{if (this != &_Y)
{if (_Ptr != _Y.get())
{if (_Owns)
delete _Ptr;
_Owns = _Y._Owns; }
else if (_Y._Owns)
_Owns = true;
_Ptr = _Y.release(); }
return (*this); }
~auto_ptr()
{if (_Owns)
delete _Ptr; }
_Ty& operator*() const _THROW0()
{return (*get()); }
_Ty *operator->() const _THROW0()
{return (get()); }
_Ty *get() const _THROW0()
{return (_Ptr); }
_Ty *release() const _THROW0()
{((auto_ptr<_Ty> *)this)->_Owns = false;
return (_Ptr); }
private:
bool _Owns;
_Ty *_Ptr;
};
int main()
{
auto_ptr<int> p(new int(100));
// p是个对象， 但重载了*运算符， 所以我说它很能装， 装得像个指针
cout << *p << endl; // 100
return 0;
} 看了上面auto_ptr的源码， 正在喝水的我， 差点呛着了。 并想问： 微软， 你还能再风骚一点么
好吧， 我也懒得计较了。 于是， 对上面代码进行风格整理， 并作出详细的注释， 就算是剖析一下auto_ptr的源码吧：
#include <iostream>
using namespace std;
// 简单类
class A
{
public:
void fun()
{
}
};
template<class T>
// 类模板
class auto_ptr
{
public:
// explicit构造函数， 禁止类型转化
explicit auto_ptr(T *p = 0) throw()
: m_bIsOwner(p != 0), m_ptr(p)
{
cout << "debug1" << endl;
}
// owner转移
auto_ptr(const auto_ptr<T>& y) throw()
: m_bIsOwner(y.m_bIsOwner), m_ptr(y.release())
{
cout << "debug2" << endl;
}
// owner转移
auto_ptr<T>& operator=(const auto_ptr<T>& y) throw()
{
cout << "debug3" << endl;
if (this != &y) // 当前对象不是y对象
{
cout << "debug4" << endl;
if (m_ptr != y.get()) // 当前对象绑定的地址不是y对象绑定的地址
{
cout << "debug5" << endl;
if (m_bIsOwner) // 如果当前对象已经绑定堆， 则要先释放
{
cout << "debug6" << endl;
delete m_ptr;
}
cout << "debug7" << endl;
m_bIsOwner = y.m_bIsOwner; // 转移owner
}
else if (y.m_bIsOwner) // 当前对象与y绑定到同一块堆上， 且y是owner, 则把y的owner转移给当前对象
{
cout << "debug8" << endl;
m_bIsOwner = true;
}
cout << "debug9" << endl;
m_ptr = y.release(); // 让y不再是owner
}
cout << "debug10" << endl;
return *this; // 返回当前对象的引用
}
// 析构函数
~auto_ptr()
{
cout << "debug11" << endl;
if (m_bIsOwner) // 只有拥有owner属性才释放堆， 这样避免重复释放
{
cout << "debug12" << endl;
delete m_ptr; // 即使m_ptr是空指针也木有关系
}
}
// 重载对象的*运算符， 使得对象"看起来"像指针， 可以执行*p操作
T& operator*() const throw()
{
cout << "debug13" << endl;
return *get();
}
// 重载对象的->运算符
T *operator->() const throw()
{
cout << "debug14" << endl;
return get();
}
// 获得对象绑定的地址
T *get() const throw()
{
cout << "debug15" << endl;
return m_ptr;
}
// 去掉对象的owner属性
T *release() const throw()
{
cout << "debug16" << endl;
((auto_ptr<T> *)this)->m_bIsOwner = false;
return m_ptr;
}
private:
bool m_bIsOwner; // 对象是否拥有为owner的标志
T *m_ptr; // 对象绑定的指针
};
int main()
{
{
cout << "------------------------------" << endl;
// 用法错误， 因为构造函数中有explicit， 不允许类型转化
//auto_ptr<int> p = new int(10);
}
{
cout << "------------------------------" << endl;
// ok
auto_ptr<int> p(new int(10));
}
{
cout << "------------------------------" << endl;
// 下面代码有严重的运行期错误, 实际上是尝试delete栈上的内容
int a = 10;
//auto_ptr<int> p(&a);
}
{
cout << "------------------------------" << endl;
auto_ptr<int> p(new int(10));
// 错误， p虽然"看似像"指针， 其本质是对象， delete p;是未定义行为
//delete p;
}
{
cout << "------------------------------" << endl;
int *q = new int(10);
auto_ptr<int> p(q);
// 错误， q释放一次， p释放一次， 重复释放啊
//delete q;
}
{
cout << "------------------------------" << endl;
auto_ptr<int> p0;
// 有debug3的打印， 但没有debug4, 知道原因了吧
p0 = p0;
}
{
cout << "------------------------------" << endl;
auto_ptr<int> p0(new int(10));
// 注意， 这是初始化， 不是复制， 所以不会有debug3的打印
auto_ptr<int> p1 = p0;
}
{
cout << "------------------------------" << endl;
auto_ptr<int> p0(new int(10));
auto_ptr<int> p1;
// 注意， 这才是赋值， 所有有debug3, debug4, debug5, debug7, debug9, debug10的打印
// 为什么没有debug6呢？ 因为当前对象p1还不是owner
p1 = p0;
}
{
cout << "------------------------------" << endl;
auto_ptr<int> p0(new int(10));
auto_ptr<int> p1(new int(20));
// 有debug6的打印， 因为当先释放p1绑定的对象， 否则内存又泄露了啊
p1 = p0;
}
{
cout << "------------------------------" << endl;
auto_ptr<int> p0(new int(10));
// 把owner转给p1
auto_ptr<int> p1(p0);
// 终于见到你了， debug8
p0 = p1;
}
{
cout << "------------------------------" << endl;
auto_ptr<int> p(new int(10));
// 见到你了， debug13
cout << *p << endl;
}
{
cout << "------------------------------" << endl;
auto_ptr<A> p(new A());
// 终于见到你了， debug15
p->fun();
}
{
cout << "------------------------------" << endl;
auto_ptr<int> p0(new int(10));
auto_ptr<int> p1(p0);
auto_ptr<int> p2(p1);
// 实际上， p3才是最后的winner, 才是最后的owner, 所以释放堆的重任在p3身上
auto_ptr<int> p3(p2);
}
{
cout << "------------------------------" << endl;
// oh, my god, 内存泄露， 本来要delete [] q; 现在析构函数只执行delete q;
int *q = new int[3];
auto_ptr<int> p(q);
}
{
cout << "------------------------------" << endl;
// oh, my god, 内存泄露， 本来要delete [] q; 现在析构函数只执行delete q;
int *q = new int[3];
auto_ptr<int> p(q);
// 已经说过， 下面语句会造成内存重复释放
//delete q;
}
// 最后说明一下， auto_ptr不适合做容器的元素， 这一点我们以后会再次讨论到
return 0;
}
好了， 不多说auto_ptr了， 一切尽在代码中。