如何安全的抛出指向自己的指针

如何安全的抛出指向自己的指针

需求描述:有一个类,需要提供一个接口,这个接口需要将自己以指针的方式传出去。

方法1

class Bad
{
private:
	int *pa;
public:
	
	Bad(int n)
	{
		pa = new int(n);
	}
	~Bad()
	{
		std::cout << "Bad::~Bad() called" << std::endl;
		delete pa;
		pa = NULL;
		
	}
	Bad* getptr2()
	{
		return this;
	}
	void Test()
	{
		cout << "hello world! ";
		cout << *pa << endl;
	}

};

int main()
{
	Bad* bp3 = new Bad(100);
	Bad* bp4 = bp3->getptr2();
	delete bp4;
	bp3->Test();
	getchar();
	return 0;
}  

一个裸指针传递给调用者,谁也不知道调用者会干什么?假如调用者delete了该对象,如上述代码delete bp4;但是bp3仍然还要使用,则就会造成问题,结果如下:
在这里插入图片描述

方法2

你可能会想到使用智能指针,代码如下:

std::shared_ptr<Bad> getptr1() 
{
	return std::shared_ptr<Bad>(this);	
}

调用:

int main()
{
	{
	// 错误的示例,每个shared_ptr都认为自己是对象仅有的所有者
	std::shared_ptr<Bad> bp1(new Bad(10));
	std::shared_ptr<Bad> bp2 = bp1->getptr1();
	// 打印bp1和bp2的引用计数
	std::cout << "bp1.use_count() = " << bp1.use_count() << std::endl;
	std::cout << "bp2.use_count() = " << bp2.use_count() << std::endl;	
	}
	getchar();
	return 0;
}  // Bad 对象将会被删除两次

结果:
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述

可以看到,对象只构造了一次,但却析构了两次。并且在增加一个指向的时候,shared_ptr的计数并没有增加。也就是说,这个时候,bp1和bp2都认为自己是Bad指针的唯一拥有者,这两个shared_ptr在计数为0的时候,都会调用一次Test对象的析构函数,所以会出问题。

那么为什么会这样呢?给一个shared_ptr传递一个this指针难道不能引起shared_ptr的计数吗?
答案是:对的,shared_ptr根本认不得你传进来的指针变量是不是之前已经传过。
下面给出经典解释:

void main()
{
	int* p = new int(10);
	shared_ptr<int> sp1(p);
	shared_ptr<int> sp2(p);
	cout << sp1.use_count() << endl;
	cout << sp2.use_count() << endl;
	system("pause");
}

结果:
在这里插入图片描述
想必无需我再多说了吧。
那该如何做呢?

正确做法

enable_shared_from_this 是一个以其派生类为模板类型参数的基类模板,继承它,派生类的this指针就能变成一个 shared_ptr。
完整代码:

class Bad: public enable_shared_from_this<Bad>       //改进1
{
private:
	int *pa;
public:
	std::shared_ptr<Bad> getptr()
	{
		return shared_from_this();      //改进2
	}
	Bad(int n)
	{
		pa = new int(n);
	}
	~Bad()
	{
		std::cout << "Bad::~Bad() called" << std::endl;
		delete pa;
		pa = NULL;
		
	}
	
	void Test()
	{
		cout << "hello world! ";
		cout << *pa << endl;
	}

};

int main()
{
	{
	
	std::shared_ptr<Bad> bp1(new Bad(10));
	std::shared_ptr<Bad> bp2 = bp1->getptr();
	// 打印bp1和bp2的引用计数
	std::cout << "bp1.use_count() = " << bp1.use_count() << std::endl;
	std::cout << "bp2.use_count() = " << bp2.use_count() << std::endl;
	
	}
	system("pause");
	return 0;
}  // Bad 对象将会被删除两次

结果:
在这里插入图片描述

实现原理

enable_shared_from_this的一种实现方法是,其内部有一个weak_ptr类型的成员变量_Wptr,当shared_ptr构造的时候,如果其模板类型继承了enable_shared_from_this,则对_Wptr进行初始化操作,这样将来调用shared_from_this函数的时候,就能够通过weak_ptr构造出对应的shared_ptr。有兴趣的可以去看看源码,本人大概看了下代码,总结出了一些使用的注意的地方。shared_from_this()仅在shared_ptr的构造函数被调用之后才能使用,原因是enable_shared_from_this::_Wptr并不在构造函数中设置,而是在shared_ptr的构造函数中设置。
示例:

class Test :public enable_shared_from_this<Test>
{
public:
	void func()
	{
		shared_ptr<Test> pT = shared_from_this();
	}
};
void main()
{
	Test t;
	t.func();
}

结果:
在这里插入图片描述
注意:不要试图对一个没有被shared_ptr接管的类对象调用shared_from_this(),不然会产生未定义行为的错误,结果则是奔溃。
另外对于智能指针不熟悉的同学,可以看看我的另一篇文章C++11中SmartPointer详解,这一篇就够了

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

发如雪-ty

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

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

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

打赏作者

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

抵扣说明:

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

余额充值