智能指针的原理和使用

背景

为什么需要智能指针?

下面我们先分析一下下面这段程序有没有什么内存方面的问题?

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;
}

上面的问题分析出来我们发现有什么问题?

不管是哪里抛异常都会导致内存泄漏的问题

你们是不是想用异常的重新抛出的方法来解决?

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;
    //异常的重新抛出
    try
    {
        cout << div() << endl;
    }
    catch (...)
    {
        delete p1;
        delete p2;
        throw;
    }
}
int main()
{
    try
    {
        Func();
    }
    catch (exception &e)
    {
        cout << e.what() << endl;
    }
    return 0;
}

 但是这个有点麻烦,假如有很多要重新抛出,就要写好多次try catch

内存泄漏
什么是内存泄漏,内存泄漏的危害
什么是内存泄漏:内存泄漏指因为疏忽或错误造成程序未能释放已经不再使用的内存的情况。内
存泄漏并不是指内存在物理上的消失,而是应用程序分配某段内存后,因为设计错误,失去了对
该段内存的控制,因而造成了内存的浪费。
内存泄漏的危害:长期运行的程序出现内存泄漏,影响很大,如操作系统、后台服务等等,出现
内存泄漏会导致响应越来越慢,最终卡死。

内存泄漏分类(了解)
C/C++程序中一般我们关心两种方面的内存泄漏:
堆内存泄漏(Heap leak)
堆内存指的是程序执行中依据须要分配通过malloc / calloc / realloc / new等从堆中分配的一
块内存,用完后必须通过调用相应的 free或者delete 删掉。假设程序的设计错误导致这部分
内存没有被释放,那么以后这部分空间将无法再被使用,就会产生Heap  Leak。
系统资源泄漏
指程序使用系统分配的资源,比方套接字、文件描述符、管道等没有使用对应的函数释放
掉,导致系统资源的浪费,严重可导致系统效能减少,系统执行不稳定。

如何避免内存泄漏

1. 工程前期良好的设计规范,养成良好的编码规范,申请的内存空间记着匹配的去释放。ps:
这个理想状态。但是如果碰上异常时,就算注意释放了,还是可能会出问题。需要下一条智
能指针来管理才有保证。

2. 采用RAII思想或者智能指针来管理资源

3. 有些公司内部规范使用内部实现的私有内存管理库。这套库自带内存泄漏检测的功能选项。

4. 出问题了使用内存泄漏工具检测。ps:不过很多工具都不够靠谱,或者收费昂贵。

总结一下:内存泄漏非常常见,解决方案分为两种:1、事前预防型。如智能指针等。2、事后查错型。如泄漏检测工具。

智能指针的使用及原理

RAII

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;
};
int div()
{
    int a, b;
    cin >> a >> b;
    if (b == 0)
        throw invalid_argument("除0错误");
    return a / b;
}
void Func()
{
    ShardPtr<int> sp1(new int);
    ShardPtr<int> sp2(new int);
    cout << div() << endl;
}
int main()
{
    try
    {
        Func();
    }
    catch (const exception &e)
    {
        cout << e.what() << endl;
    }
    return 0;
}

智能指针的原理

上述的SmartPtr还不能将其称为智能指针,因为它还不具有指针的行为。指针可以解引用,也可
以通过->去访问所指空间中的内容,因此:AutoPtr模板类中还得需要将* 、->重载下,才可让其
像指针一样去使用。

template <class T>
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;
};
struct Date
{
    int _year;
    int _month;
    int _day;
};
int main()
{
    SmartPtr<int> sp1(new int);
    *sp1 = 10 ;
    cout << *sp1 << endl;
    SmartPtr<int> sparray(new Date);
    // 需要注意的是这里应该是sparray.operator->()->_year = 2018;
    // 本来应该是sparray->->_year这里语法上为了可读性,省略了一个->
    sparray->_year = 2018;
    sparray->_month = 1;
    sparray->_day = 1;
}

总结一下智能指针的原理:
1. RAII特性
2. 重载operator*和opertaor->,具有像指针一样的行为。

如果智能指针进行拷贝了呢,会出现什么问题?

崩溃了,那是因为两个智能指针指向同一份资源,释放了两次导致崩溃

怎么处理智能指针的拷贝问题呢?

深拷贝行不行呢? 不行,指针拷贝要的就是浅拷贝

智能指针主要用着三种:

  • auto_ptr:管理权转移,导致拷贝对象悬空问题,一点用没有
  • unique_ptr:防止拷贝
  • shared_ptr/weak_ptr:引用计数,支持拷贝

auto_ptr

C++98版本的库中就提供了auto_ptr的智能指针。下面演示的auto_ptr的使用及问题。
auto_ptr的实现原理:管理权转移的思想,下面简化模拟实现了一份auto_ptr来了解它的原理

// C++98 管理权转移  auto_ptr
namespace qidian
{
    template <class T>
    class auto_ptr
    {
    public:
        auto_ptr(T *ptr)
            : _ptr(ptr)
        {
        }
        auto_ptr(auto_ptr<T> &sp)
            : _ptr(sp._ptr)
        {
            // 管理权转移
            sp._ptr = nullptr;
        }
        auto_ptr<T> &operator=(auto_ptr<T> &ap)
        {
            // 检测是否为自己给自己赋值
            if (this != &ap)
            {
                // 释放当前对象中资源
                if (_ptr)
                    delete _ptr;
                // 转移ap中资源到当前对象中
                _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;
    };
}


// 结论:auto_ptr是一个失败设计,很多公司明确要求不能使用auto_ptr
int main()
{
    std::auto_ptr<int> sp1(new int);
    std::auto_ptr<int> sp2(sp1); // 管理权转移

    // sp1悬空
    *sp2 = 10;
    cout << *sp2 << endl;
    cout << *sp1 << endl;
    return 0;
}

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
// 原理:简单粗暴 -- 防拷贝
namespace qidain
{
    template <class T>
    class unique_ptr
    {
    public:
        unique_ptr(T *ptr)
            : _ptr(ptr)
        {
        }
        ~unique_ptr()
        {
            if (_ptr)
            {
                cout << "delete:" << _ptr << endl;
                delete _ptr;
            }
        }
        // 像指针一样使用
        T &operator*()
        {
            return *_ptr;
        }
        T *operator->()
        {
            return _ptr;
        }
        // 防拷贝
        unique_ptr(const unique_ptr<T> &sp) = delete;
        unique_ptr<T> &operator=(const unique_ptr<T> &sp) = delete;

    private:
        T *_ptr;
    };
}
int main()
{
  qidain::unique_ptr<int> sp1(new int);
  qidian::unique_ptr<int> sp2(sp1);

  return 0;
}

shared_ptr

引用计数原理

C++11中开始提供更靠谱的并且支持拷贝的shared_ptr
shared_ptr的原理:是通过引用计数的方式来实现多个shared_ptr对象之间共享资源。
1. shared_ptr在其内部,给每个资源都维护了着一份计数,用来记录该份资源被几个对象共享。
2. 在对象被销毁时(也就是析构函数调用),就说明自己不使用该资源了,对象的引用计数减一。
3. 如果引用计数是0,就说明自己是最后一个使用该资源的对象,必须释放该资源。
4. 如果不是0,就说明除了自己还有其他对象在使用该份资源,不能释放该资源,否则其他对
象就成野指针了。

// 引用计数支持多个拷贝管理同一个资源,最后一个析构对象释放资源
namespace qidian
{
    template <class T>
    class shared_ptr
    {
    public:
        shared_ptr(T *ptr = nullptr)
            : _ptr(ptr), _pRefCount(new int(1)), _pmtx(new mutex)
        {
        }
        shared_ptr(const shared_ptr<T> &sp)
            : _ptr(sp._ptr), _pRefCount(sp._pRefCount), _pmtx(sp._pmtx)
        {
            AddRef();
        }
        void Release()
        {
            _pmtx->lock();
            bool flag = false;
            if (--(*_pRefCount) == 0 && _ptr)
            {
                cout << "delete:" << _ptr << endl;
                delete _ptr;
                delete _pRefCount;
                flag = true;
            }
            _pmtx->unlock();
            if (flag == true)
            {
                delete _pmtx;
            }
        }
        void AddRef()
        {
            _pmtx->lock();
            ++(*_pRefCount);
            _pmtx->unlock();
        }
        shared_ptr<T> &operator=(const shared_ptr<T> &sp)
        {
            // if (this != &sp)
            if (_ptr != sp._ptr)// 管理的同一份资源,不需要重新赋值
            {
                Release();
                _ptr = sp._ptr;
                _pRefCount = sp._pRefCount;
                _pmtx = sp._pmtx;
                AddRef();
            }
            return *this;
        }
        int use_count()
        {
            return *_pRefCount;
        }
        ~shared_ptr()
        {
            Release();
        }
        // 像指针一样使用
        T &operator*()
        {
            return *_ptr;
        }
        T *operator->()
        {
            return _ptr;
        }
        T *get() const
        {
            return _ptr;
        }

    private:
        T *_ptr;
        int *_pRefCount;
        mutex *_pmtx;
    };
}

线程安全问题

shared_ptr智能指针是线程安全的吗?是的,引用计数的加减是加锁保护的。但是指向资源不是线程安全的

指向堆上资源的线程安全问题是访问的人处理的,智能指针不管,也管不了

引用计数的线程安全问题,是智能指针要处理的

通过下面的程序我们来测试shared_ptr的线程安全问题。需要注意的是shared_ptr的线程安全分为两方面:

  1. 智能指针对象中引用计数是多个智能指针对象共享的,两个线程中智能指针的引用计数同时++或--,这个操作不是原子的,引用计数原来是1,++了两次,可能还是2.这样引用计数就错乱了。会导致资源未释放或者程序崩溃的问题。所以只能指针中引用计数++、--是需要加锁的,也就是说引用计数的操作是线程安全的。
  2. 智能指针管理的对象存放在堆上,两个线程中同时去访问,会导致线程安全问题。
// 1.演示引用计数线程安全问题,就把AddRef()和Release()中的锁去掉
// 2.演示可能不出现线程安全问题,因为线程安全问题是偶现性问题,main函数的n改大一些概率就变大了,就容易出现了。
// 3.下面代码我们使用自己shared_ptr演示,是为了方便演示引用计数的线程安全问题,
//将代码中的自己shared_ptr换成std::shared_ptr进行测试,可以验证库的shared_ptr,发现结论是一样的。 
struct Date
{
    int _year = 0;
    int _month = 0;
    int _day = 0;
};
void SharePtrFunc(bit::shared_ptr<Date> &sp, size_t n, mutex &mtx)
{
    cout << sp.get() << endl;
    for (size_t i = 0; i < n; ++i)
    {
        // 这里智能指针拷贝会++计数,智能指针析构会--计数,这里是线程安全的。
        bit::shared_ptr<Date> copy(sp);
        // 这里智能指针访问管理的资源,不是线程安全的。所以我们看看这些值两个线程++了2n次,
        //但是最终看到的结果,并一定是加了2n
        {
            unique_lock<mutex> lk(mtx);// 智能指针访问管理的资源需要程序员自己加锁解决
            copy->_year++;
            copy->_month++;
            copy->_day++;
        }
    }
}
int main()
{
    qidian::shared_ptr<Date> p(new Date);
    cout << p.get() << endl;
    const size_t n = 100000;
    mutex mtx;
    thread t1(SharePtrFunc, std::ref(p), n, std::ref(mtx));
    thread t2(SharePtrFunc, std::ref(p), n, std::ref(mtx));
    t1.join();
    t2.join();
    cout << p->_year << endl;
    cout << p->_month << endl;
    cout << p->_day << endl;
    cout << p.use_count() << endl;
    return 0;
}

循环引用问题

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;
}

循环引用分析:

  1. node1和node2两个智能指针对象指向两个节点,引用计数变成1,我们不需要手动delete。
  2. node1的_next指向node2,node2的_prev指向node1,引用计数变成2。
  3. node1和node2析构,引用计数减到1,但是_next还指向下一个节点。但是_prev还指向上一个节点。
  4. 也就是说_next析构了,node2就释放了。
  5. 也就是说_prev析构了,node1就释放了。
  6. 但是_next属于node的成员,node1释放了,_next才会析构,而node1由_prev管理,_prev属于node2成员,所以这就叫循环引用,谁也不会释放。

weak_ptr

  1. 它不是常规的智能指针,不支持RAII
  2. 支持像指针一样
  3. 专门设计出来,辅助解决shared_ptr的循环引用问题,支持用shared_ptr进行构造
  4. weak_ptr可以指向资源,但是他不参与管理,不增加引用计数
// 简化版本的weak_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;
};

解决方案:在引用计数的场景下,把节点中的_prev和_next改成weak_ptr就可以了
原理就是,node1->_next = node2和node2->_prev = node1时,weak_ptr的_next和_prev不会增加node1和node2的引用计数。  

struct ListNode
{
    int _data;
    weak_ptr<ListNode> _prev;
    weak_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;
}

删除器

如果不是new出来的对象如何通过智能指针管理呢?如果是new int[ ]很多对象呢?如果是用智能指针打开一个文件呢?那么前面智能指针所用的析构函数里的delete就有问题啦。shared_ptr设计了一个删除器来解决这个问题(ps:删除器这个问题我们了解一下)

std::shared_ptr中的构造函数就提供了一个删除器,也就是一个可调用对象,可以是函数指针,lambda,仿函数,需要用户自己传

下面就是简易的shared_ptr中删除器的代码

namespace qidian
{
	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)
		{}

		~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<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;
		}

		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){
			cout << "lambda delete:" << ptr << endl;
			delete ptr; 
		};
	};


// 定制删除器-- 可调用对象
//仿函数
template <class T>
struct DeleteArray
{
    void operator()(T *ptr)
    {
        cout << "void operator()(T* ptr)" << endl;
        delete[] ptr;
    }
};

void test_shared_deletor()
{
    bit::shared_ptr<Date> sp0(new Date);

    bit::shared_ptr<Date> spa1(new Date[10], DeleteArray<Date>());
    //lambda
    bit::shared_ptr<Date> spa2(new Date[10], [](Date *ptr)
    {
		cout << "lambda delete[]:" << ptr << endl;
		delete[] ptr; 
    });

    bit::shared_ptr<FILE> spF3(fopen("Test.cpp", "r"), [](FILE *ptr)
    {
		out << "lambda fclose:" << ptr << endl;
		fclose(ptr); 
    });
}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

爱敲代码的奇点

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值