问题背景:
C++项目中不得不提的就是指针,C++不像其他高级语言JAVA、Python等有自己的内存回收机制。C++的内存都需要开发自己来申请和释放,为了避免开发忘记释放自己申请的内存,C++引入了智能指针的概念,常见的智能指针有四种shared_ptr、weak_ptr、scoped_ptr、auto_ptr,实际项目中最常用就是shared_ptr,本篇文章中不详述这几种智能指针的关系和内部机制,后续出文章详述。本文主要讲实际项目中智能指针和裸指针混用导致的崩溃问题,主要常见就是两个指针或者多个智能指针值交换的时候用裸指针作为临时变量的话会导致程序崩溃或者异常。
问题代码:
为了还原项目中的实际问题,突出问题所在,基于实际情况做了写了下面测试代码来简化:
class CTest
{
public:
CTest(int id):m_ID(id){}
~CTest() { cout << "delete!!" <<endl;}
int getID() {return m_ID;}
private:
int m_ID;
};
int _tmain(int argc, _TCHAR* argv[])
{
shared_ptr<CTest> pTestA = make_shared<CTest>(100);
shared_ptr<CTest> pTestB = make_shared<CTest>(200);
cout << "交换前:" << endl;
cout << pTestA->getID() << endl;
cout << pTestB->getID() << endl;
CTest* pTempC = pTestB.get();
pTestB = pTestA;
pTestA.reset(pTempC);
cout << "交换后:" << endl;
cout << pTestA->getID() << endl;
cout << pTestB->getID() << endl;
return 0;
}
上述代码要实现pTestA和pTestB交换,咋一看好像代码没啥问题,为了突出问题原因在CTest的析构函数加了输出语句,运行上面的测试代码运行结果如下:
上述代码并没有实现pTestA和pTestB的交换,原因就是中间临时变量用了裸指针,因为虽然上一句CTest* pTempC = pTestB.get()被赋值给裸指针pTempC ,但pTestB 指向的内存引用计数还是为1,当执行pTestB = pTestA;这句话的时候,pTestB 的引用计数会被置为0,从而导致pTestB 以前指向的内存被释放掉(调用了析构输出了delete),此时pTempC已经是野指针了,再后面赋给pTestA时就值已经是乱码了,导致程序崩溃。
问题解决:
不要裸指针跟智能指针混用,统一用智能指针即可解决上述崩溃问题。修改代码如下及运行结果如下:
shared_ptr<CTest> pTempC = pTestB;
pTestB = pTestA;
pTestA = pTempC;