c++ 共享指针shared_ptr不恰当的使用造成循环引用进而内存泄漏

记录下遇到的共享指针shared_ptr不恰当的使用,导致指针一直不被释放,进而内存泄漏问题

原因:bind时其中一个函数入参绑定了一个共享指针,而这个绑定后的函数对象又保存在共享指针所指向的对象里(在本例中是间接存在,即共享指针所指向的对象有个成员变量,然后这里变量也是个类对象且其中保存了bind后的函数对象)

场景是:

  • 用boost网络库写服务端代码,用bind绑定了session会话对象然后生成了一个回调函数对象,用于后续请求处理时调用这个回调函数进行请求应答,将应答数据返回给客户端。
  • 问题出现:当客户端关闭连接后,正常session也会释放,因为boost的底层监听列表已经移除了这个session指针,但是实际是没有释放,因为session->m_reqclient.m_funcResponse这个函数对象bind了session而且还是以共享指针的形式绑定的,所以产生了循环引用导致一直计数一直不为0

解决方法:
使用弱指针weak_ptr,即response这个函数入参由
(std::shared_ptr session, const std::string& data)
改为
(std::weak_ptr session, const std::string& data)

示例代码如下:
精简了很多代码,方便看。

#pragma once
#include <iostream>
#include <functional>
#include <memory>


class Session;
class ReqInterface;

static int dorequst(std::shared_ptr<Session> session);

class ReqInterface
{
public:
    void Dorequest(std::string data) {//请求业务处理
        data = data + "toresponse";
        if (m_funcResponse)
        {
            m_funcResponse(data);//请求应答
        }
    }

public:
    std::function<int(std::string)> m_funcResponse;
};

class Session : public std::enable_shared_from_this<Session>
{
public:
    ReqInterface* m_reqclient;

public:
	Session(){}
	~Session(){}

    void run() {//开始运行 不断监听客户端请求 有请求收到时调用on_recv

    }
private:
    void on_recv() {
        dorequst(shared_from_this());
    }
public:
    std::string get_reqdata() {
        return std::string();//从sokect读取请求数据
    }
    int do_write(const std::string& data) {
        //将数据写入sokcet
        return 0;
    }
};

static int response(std::shared_ptr<Session> session, const std::string& data) {
    session->do_write(data);
    return 0;
}

static int dorequst(std::shared_ptr<Session> session)
{//请求分发
    if (session->m_reqclient == nullptr) {
        auto client = new ReqInterface;
        client->m_funcResponse = std::bind(&response, session, std::placeholders::_1);
        session->m_reqclient = client;
    }
    session->m_reqclient->Dorequest(session->get_reqdata());
    return 0;
}

修复后的代码,改为了weak_ptr

#pragma once
#include <iostream>
#include <functional>
#include <memory>


class Session;
class ReqInterface;

static int dorequst(std::shared_ptr<Session> session);

class ReqInterface
{
public:
    void Dorequest(std::string data) {//请求业务处理
        data = data + "toresponse";
        if (m_funcResponse)
        {
            m_funcResponse(data);//请求应答
        }
    }

public:
    std::function<int(std::string)> m_funcResponse;
};

class Session : public std::enable_shared_from_this<Session>
{
public:
    ReqInterface* m_reqclient;

public:
	Session(){}
	~Session(){}

    void run() {//开始运行 不断监听客户端请求 有请求收到时调用on_recv

    }
private:
    void on_recv() {
        dorequst(shared_from_this());
    }
public:
    std::string get_reqdata() {
        return std::string();//从sokect读取请求数据
    }
    int do_write(const std::string& data) {
        //将数据写入sokcet
        return 0;
    }
};

//改用弱指针 防止循环引用不能释放问题
static int response(std::weak_ptr<Session> session, const std::string& data) {
    if (!session.expired())
    {
        session.lock()->do_write(data);
    }
    return 0;
}

static int dorequst(std::shared_ptr<Session> session)
{
    if (session->m_reqclient == nullptr) {
        auto client = new ReqInterface;
        client->m_funcResponse = std::bind(&response, std::weak_ptr<Session>(session), std::placeholders::_1);
        session->m_reqclient = client;
    }
    session->m_reqclient->Dorequest(session->get_reqdata());
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值