asio处理粘包的简易方式 - Vafail的个人博客

简易方式#

之前我们介绍了通过async_read_some函数监听读事件，并且绑定了读事件的回调函数HandleRead

1
_socket.async_read_some(boost::asio::buffer(_data, MAX_LENGTH), std::bind(&CSession::HandleRead, this,
2
std::placeholders::_1, std::placeholders::_2, SharedSelf()));

async_read_some 这个函数的特点是只要对端发数据，服务器接收到数据，即使没有收全对端发送的数据也会触发HandleRead函数，所以我们会在HandleRead回调函数里判断接收的字节数，接收的数据可能不满足头部长度，可能大于头部长度但小于消息体的长度，可能大于消息体的长度，还可能大于多个消息体的长度，所以要切包等，这些逻辑写起来很复杂，所以我们可以通过读取指定字节数，直到读完这些字节才触发回调函数，那么可以采用async_read函数，这个函数指定读取指定字节数，只有完全读完才会触发回调函数。

获取头部数据#

我们可以读取指定的头部长度，大小为HEAD_LENGTH字节数，只有读完HEAD_LENGTH字节才触发HandleReadHead函数

1
void CSession::Start(){
2
    _recv_head_node->Clear();
3
    boost::asio::async_read(_socket, boost::asio::buffer(_recv_head_node->_data, HEAD_LENGTH), std::bind(&CSession::HandleReadHead, this,
4
        std::placeholders::_1, std::placeholders::_2, SharedSelf()));
5
}

这样我们可以直接在HandleReadHead函数内处理头部信息

1
void CSession::HandleReadHead(const boost::system::error_code& error, size_t  bytes_transferred, std::shared_ptr<CSession> shared_self) {
2
    if (!error) {
3
        if (bytes_transferred < HEAD_LENGTH) {
4
            cout << "read head lenth error";
5
            Close();
6
            _server->ClearSession(_uuid);
7
            return;
8
        }
9

10
        //头部接收完，解析头部
11
        short data_len = 0;
12
        memcpy(&data_len, _recv_head_node->_data, HEAD_LENGTH);
13
        cout << "data_len is " << data_len << endl;
14
        //此处省略字节序转换
15
        // ...
16
        //头部长度非法
17
        if (data_len > MAX_LENGTH) {
18
            std::cout << "invalid data length is " << data_len << endl;
19
            _server->ClearSession(_uuid);
20
            return;
21
        }
22

23
        _recv_msg_node= make_shared<MsgNode>(data_len);
24
        boost::asio::async_read(_socket, boost::asio::buffer(_recv_msg_node->_data, _recv_msg_node->_total_len),
25
            std::bind(&CSession::HandleReadMsg, this,
26
            std::placeholders::_1, std::placeholders::_2, SharedSelf()));
27
    }
28
    else {
29
        std::cout << "handle read failed, error is " << error.what() << endl;
30
        Close();
31
        _server->ClearSession(_uuid);
32
    }
33
}

接下来根据头部内存储的消息体长度，获取指定长度的消息体数据，所以再次调用async_read，指定读取_recv_msg_node->_total_len长度，然后触发HandleReadMsg函数

获取消息体#

HandleReadMsg函数内解析消息体，解析完成后打印收到的消息，接下来继续监听读事件，监听读取指定头部大小字节，触发HandleReadHead函数，然后再在HandleReadHead内继续监听读事件，获取消息体长度数据后触发HandleReadMsg函数，从而达到循环监听的目的。

1
void CSession::HandleReadMsg(const boost::system::error_code& error, size_t  bytes_transferred,
2
    std::shared_ptr<CSession> shared_self) {
3
    if (!error) {
4
        PrintRecvData(_data, bytes_transferred);
5
        std::chrono::milliseconds dura(2000);
6
        std::this_thread::sleep_for(dura);
7
        _recv_msg_node->_data[_recv_msg_node->_total_len] = '\0';
8
        cout << "receive data is " << _recv_msg_node->_data << endl;
9
        Send(_recv_msg_node->_data, _recv_msg_node->_total_len);
10
        //再次接收头部数据
11
        _recv_head_node->Clear();
12
        boost::asio::async_read(_socket, boost::asio::buffer(_recv_head_node->_data, HEAD_LENGTH),
13
            std::bind(&CSession::HandleReadHead, this, std::placeholders::_1, std::placeholders::_2,
14
                SharedSelf()));
15
    }
16
    else {
17
        cout << "handle read msg failed,  error is " << error.what() << endl;
18
        Close();
19
        _server->ClearSession(_uuid);
20
    }
21
}

简易方式#

获取头部数据#

获取消息体#

文章分享

评论区