智能指针源码解析

从源码理解智能指针（一）——auto_ptr、unique_ptr

从源码理解智能指针（二）—— shared_ptr、weak_ptr

 

std::shared_ptr<void>的工作原理，这篇是为数不多的讲的抓住要领的一篇：

/** C1. shared_ptr 构造函数 **/

template<class _Tp>
template<class _Yp>
shared_ptr<_Tp>::shared_ptr(_Yp* __p,
                            typename enable_if<is_convertible<_Yp*, element_type*>::value, __nat>::type)
    : __ptr_(__p)
{
    unique_ptr<_Yp> __hold(__p);
    typedef typename __shared_ptr_default_allocator<_Yp>::type _AllocT;
    typedef __shared_ptr_pointer<_Yp*, default_delete<_Yp>, _AllocT > _CntrlBlk;
    __cntrl_ = new _CntrlBlk(__p, default_delete<_Yp>(), _AllocT());
    __hold.release();
    __enable_weak_this(__p, __p);
}

/** C2. shared_ptr 拷贝构造 **/
template<class _Tp>
inline
shared_ptr<_Tp>::shared_ptr(const shared_ptr& __r) _NOEXCEPT
    : __ptr_(__r.__ptr_),
      __cntrl_(__r.__cntrl_)
{
    if (__cntrl_)
        __cntrl_->__add_shared();
}

/** C3. shared_ptr::__cntrl_ 类型 **/
__shared_weak_count* __cntrl_;

/** C4. shared_ptr 析构函数 **/

template<class _Tp>
shared_ptr<_Tp>::~shared_ptr()
{
    if (__cntrl_)
        __cntrl_->__release_shared();
}

/** C5. __shared_weak_count::__release_shared **/

bool __release_shared() _NOEXCEPT {
    if (__libcpp_atomic_refcount_decrement(__shared_owners_) == -1) {
    __on_zero_shared();
    return true;
    }
    return false;
}

/** C6. __shared_ptr_pointer::__release_shared **/
template <class _Tp, class _Dp, class _Alloc>
/* 重点 !!! virtual !!! */
void
__shared_ptr_pointer<_Tp, _Dp, _Alloc>::__on_zero_shared() _NOEXCEPT
{
    // __data__ 是 内部工具类__compressed_pair
    // __data_.first().second()是 deleter
    // __data_.first().first() 是 shared_ptr<T> 中T类型的指针
    __data_.first().second()(__data_.first().first());
    __data_.first().second().~_Dp();
}


/** C7. default deleter **/

template <class _Tp>
struct _LIBCPP_TEMPLATE_VIS default_delete {
    // ... 此处省略若干行
  void operator()(_Tp* __ptr) const _NOEXCEPT {
    delete __ptr;
  }
};

•  每一个shared_ptr 内部有一个control block，里面会存放一个要维护的指针，一个计数，一个删除器(deleter)，一个分配(allocator)。这里我们要关心的是删除器。顾名思义，它是用来删除指针的。
• shared_ptr中的 有一个__cntrl_即control block。字段的类型是__shared_weak_count指针，这个类是一个非模板类。shared_ptr<T>创建的control block的类型是一个类模板template <class _Tp, class _Dp, class _Alloc> __shared_ptr_pointer继承自__shared_weak_count。 见代码C3。
• 当创建一个新的shared_ptr的时候，构建一个新的control block。见代码C1。
• 当一个shared_ptr A赋值或者拷贝构造给另一个shared_ptrB的时候（当然是在类型能够转换的前提下），B会把A的__cntrl_拷一份，同时将其引用加一。注意，这个时候拷贝的是指针，__cntrl_还是指向最初创建的那个对象。见代码C2。
• 当shared_ptr被析构或者重置的时候会调用__cntrl_->__release_shared()。见代码C4。
•  __cntrl_->__release_shared()如果发现当前的计数为-1的时候，调用__on_zero_shared()。见代码C5。
• __on_zero_shared是一个虚方法，那么它就会调用到最初创建的__shared_ptr_pointer的实现。
• __shared_ptr_pointer的实现中是有完整的类型和删除器的信息的。见代码C6。
• 默认的删除器很简单的执行了delete __ptr操作，因为类型是已知的所以能够正确的调用到析构函数。见代码C7。