网络编程(12)——完善粘包处理操作(id字段)
十二、day12
之前的粘包处理是基于消息头包含的消息体长度进行对应的切包操作,但并不完整。一般来说,消息头仅包含数据域的长度,但是如果要进行逻辑处理,就需要传递一个id字段表示要处理的消息id,当然可以不在包头传id字段,将id序列化到消息体也是可以的,但是我们为了便于处理也便于回调逻辑层对应的函数,最好是将id写入包头。
之前我们设计的消息结构是这样的
而本节需要加上id字段
在此之前,先完整的复习一下基于boost::asio实现的服务器逻辑层结构
1. 服务器架构设计
1)asio底层通信
前面的asio底层通信过程如下图所示
Asio底层的通信过程
1)首先,在应用层调用async_read时,相当于在io_context中注册了一个读事件,表示程序希望在指定socket上进行异步读取操作,并提供一个读回调函数以在读取完成后做相应的处理;
2)io_context用于管理所有异步操作和相应的回调函数,且当async_read被调用时,asio会将该socket、相应的读事件和回调函数注册到系统内部的模型中(数据结构);当io_context启动时,即io_context.run时,asio根据系统使用对应的模型管理这些事件(windows是iocp,linux是epoll);
3)模型进入一个死循环,会监听所有注册的socket,并监测其状态(可读?可写?),如果socket的状态发生变化(事件被触发),模型将该事件放入就绪事件队列中;
4)io_context::run 在轮询就绪事件队列时,会依次调用每个就绪事件的回调函数(已经放在就绪事件队列中),每个回调函数都包含了处理读操作的逻辑,比如读取数据、处理错误等。
2)逻辑层结构
而服务器架构除了上面的内容之外,一般还有一个逻辑层。
一般在解析完对端发送的数据之后,还要对该请求做更进一步地处理,比如根据不同的消息id执行不同的逻辑层函数或不同的操作,比如读数据库、写数据库,还比如游戏中可能需要给玩家叠加不同的buff、增加积分等等,这些都需要交给逻辑层处理,而不仅仅是把消息发给对端。
服务器架构
上图所示的是一个完成的服务器架构,一般需要将逻辑层独立出来,因为如果在解析完对端数据后需要执行一些复杂的操作,比如玩家需要叠加各自buff或者技能,此时可能会耗时1s甚至更多,如果没有独立的逻辑层进行操作,那么系统会一直停留在执行回调函数那一步,造成阻塞,直至操作结束。
而逻辑层是独立的,回调函数只需将数据投递给逻辑队列(回调函数将数据放入队列中之后系统会运行下一步,便不会被阻塞),逻辑系统会自动从队列中取数据并做相应操作,如果需要在执行完操作之后做相应回复,那么逻辑系统会调用写事件并注册写回调给asio网络层,网络层就是asio底层通信的网络层步骤。
以上操作是基于单线程,如果是在多线程的情况下,阻塞的情况会不会好一些?
asio的多线程有两种模式。
1)第一种模式是启动n个线程,每个线程负责一个io_context,每一个io_context负责一部分的socket。比如现在有两个io_context,一个是负责socket的id为奇数的io_context,一个是负责socket的id为偶数的io_context,但同样会造成阻塞的情况。因为不管是多线程还是单线程,只要在线程中有一个io_context中运行,那么它负责的那部分回调函数的处理操作如果比较复杂时,仍会造成阻塞的情况。
2)第二种模式是一个io_context跑在多个线程中,即多个线程共享一个io_context。这种模式下不会造成之前的那种阻塞情况,因为在就绪事件队列中的事件不是一个线程处理了,而是不同的线程共享一个就绪事件队列,不同线程会触发执行不同的回调函数,即使某个回调处理的比较慢,但由于其他事件被不同线程处理了,系统并不需要阻塞等待该回调处理完成之后在执行处理其他回调。
虽然模式二办法会解决系统阻塞、超时的问题,但在现实中需要有一个逻辑层独立于网络层和应用层,这样可以极大地提高网络线程的收发能力,用多线程的方式管理网络层。
2. 完善粘包处理操作
之前的消息结构并不完善,缺少一个消息id,本节进行代码的相应改进。
首先,之前的消息节点被收发共用,只不过收数据用的是第一种构造函数,发数据用的是第二种构造函数。为了减少耦合和歧义,需要重新设计消息节点。
1) 消息节点
重新构建一个MsgNode类,并派生出RecvNode 和SendNode
-
MsgNode 表示消息节点的基类,头部的消息用该结构存储
-
RecvNode 表示接收消息的节点
-
SendNode 表示发送消息的节点
#pragma once
#include <iostream>
#include <string>
#include <boost/asio.hpp>
using std::cout;
using std::cin;
using std::endl;
class MsgNode
{
public:
short _cur_len;
short _total_len;
char* _msg;
MsgNode(short max_len) :_total_len(max_len), _cur_len(0) {
_msg = new char[_total_len + 1](); // 加()会将分配内存的每个元素初始化为0,不加不会初始化
_msg[_total_len] = '\0';
}
~MsgNode() {
std::cout << "destruct MsgNode" << endl;
delete[] _msg;
}
void Clear() {
::memset(_msg, 0, _total_len);
_cur_len = 0;
}
};
// 构造收节点
class RecvNode :public MsgNode {
private:
short _msg_id;
public:
RecvNode(short max_len, short msg_id);
};
// 构造发节点
class SendNode :public MsgNode {
private:
short _msg_id;
public:
SendNode(const char* msg, short max_len, short msg_id);
};
具体实现为:
#include "MsgNode.h"
#include "Const.h"
RecvNode::RecvNode(short max_len, short msg_id) :MsgNode(max_len),
_msg_id(msg_id) {}
// 发送的数据首地址、数据长度、消息id,发送节点总长度为消息体长度+头节点长度
SendNode::SendNode(const char* msg, short max_len, short msg_id) : MsgNode(max_len + HEAD_TOTAL_LEN)
, _msg_id(msg_id) {
// 将消息id转换为网络序,并存储至至发送节点内
short msg_id_host = boost::asio::detail::socket_ops::host_to_network_short(msg_id);
memcpy(_msg, &msg_id_host, HEAD_ID_LEN);
// 将消息体长度转换为网络序,并存储至至发送节点内
short max_len_host = boost::asio::detail::socket_ops::host_to_network_short(max_len);
memcpy(_msg + HEAD_ID_LEN, &max_len_host, HEAD_DATA_LEN);
// 将消息内容存储至发送节点内
memcpy(_msg + HEAD_ID_LEN + HEAD_DATA_LEN, msg, max_len);
}
Const.h 定义为
#pragma once
const size_t MAX_LENGTH = 1024 * 2;
const short MAX_RECVQUE = 10000;
const short MAX_SENDQUE = 1000;
const size_t HEAD_TOTAL_LEN = 4;
const size_t HEAD_ID_LEN = 2;
const size_t HEAD_DATA_LEN = 2;
构建SendNode节点时,需要将消息id和消息长度转换为网络序,然后写入数据域_msg ,前2字节存储id,id后为消息长度,偏移4字节后为消息体内容。
2)Session类
Session类和前面差不多,不过需要把收发的逻辑做相应的修改
首先,队列_send_que、消息头结构、消息体结构需要重新声明,分别使用SendNode,RecvNode,MsgNode作为元素类型。
std::queue<std::shared_ptr<SendNode> > _send_que;
std::mutex _send_lock;
std::shared_ptr<RecvNode> _recv_msg_node; // 收到的消息结构
bool _b_head_parse; // 表示是否处理完头部信息
std::shared_ptr<MsgNode> _recv_head_node; // 收到的头部结构
Session的构造函数也需要做相应变化,消息头结构的大小更改为4字节,包括id和消息体长度
CSession(boost::asio::io_context& ioc, CServer* server) : _socket(ioc), _server(server), _b_close(false),
_b_head_parse(false) {
// random_generator是函数对象,加()就是函数,再加一个()就是调用该函数
boost::uuids::uuid a_uuid = boost::uuids::random_generator()();
_uuid = boost::uuids::to_string(a_uuid);
_recv_head_node = std::make_shared<MsgNode>(HEAD_TOTAL_LEN);
}
重新定义Send函数,两个Send的重载都需要重新定义
参数列表增加msgid,构造发送节点时需输入三个参数msg, max_length, msgid(发送内容,内容长度,消息id)
void CSession::Send(char* msg, int max_length, short msgid) {
bool pending = false; // 发送标志,true时有未完成的发送操作,false为空
// 使用lock_guard锁住_send_lock,确保_send_lock(发送队列)访问的线程安全的
// 锁的存在确保了多个线程不会同时修改发送队列
std::lock_guard<std::mutex> lock(_send_lock);
int send_que_size = _send_que.size();
if (send_que_size > MAX_SENDQUE) {
cout << "session: " << _uuid << " send que fulled, size is " << MAX_SENDQUE << endl;
return;
}
// 判断队列是否有未完成的发送操作
if (_send_que.size() > 0) {
pending = true;
}
_send_que.push(std::make_shared<SendNode>(msg, max_length, msgid)); // 将发送消息存储至队列
if (pending) { // 如果有未完成的发送,直接返回
return;
}
// 异步发送
auto& msgnode = _send_que.front();
boost::asio::async_write(_socket, boost::asio::buffer(msgnode->_msg, msgnode->_total_len),
std::bind(&CSession::haddle_write, this, std::placeholders::_1, shared_from_this()));
} // 当'}'结束后,_send_lock解锁,发送队列解锁
void CSession::Send(std::string msg, short msgid) {
bool pending = false; // 发送标志,true时有未完成的发送操作,false为空
// 使用lock_guard锁住_send_lock,确保_send_lock(发送队列)访问的线程安全的
// 锁的存在确保了多个线程不会同时修改发送队列
std::lock_guard<std::mutex> lock(_send_lock);
int send_que_size = _send_que.size();
if (send_que_size > MAX_SENDQUE) {
cout << "session: " << _uuid << " send que fulled, size is " << MAX_SENDQUE << endl;
return;
}
// 判断队列是否有未完成的发送操作
if (_send_que.size() > 0) {
pending = true;
}
_send_que.push(std::make_shared<SendNode>(msg.c_str(), msg.length(),msgid)); // 将发送消息存储至队列
if (pending) { // 如果有未完成的发送,直接返回
return;
}
// 异步发送
auto& msgnode = _send_que.front();
boost::asio::async_write(_socket, boost::asio::buffer(msgnode->_msg, msgnode->_total_len),
std::bind(&CSession::haddle_write, this, std::placeholders::_1, shared_from_this()));
} // 当'}'结束后,_send_lock解锁,发送队列解锁
读回调也需更改,在文章10中haddle_write函数的基础上做修改,可参考该文章
https://zhuanlan.zhihu.com/p/722233898
void CSession::HandleRead(const boost::system::error_code& error, size_t bytes_transferred,
std::shared_ptr<CSession> _self_shared) {
if (!error) {
// 打印缓存区的数据并将该线程暂停2s
//PrintRecvData(_data, bytes_transferred);
//std::chrono::milliseconds dura(2000);
//std::this_thread::sleep_for(dura);
// 每触发一次handale_read,它会返回实际读取的字节数bytes_transferred,copy_len表示已处理的长度,每处理一字节,copy_len便加一
int copy_len = 0; // 已经处理的字符数
while (bytes_transferred > 0) { // 只要读取到数据就对其处理
if (!_b_head_parse) { // 判断消息头部是否已处理,_b_head_parse默认为false
// 异步读取到的字节数 + 已接收到的头部长度 < 头部总长度
if (bytes_transferred + _recv_head_node->_cur_len < HEAD_TOTAL_LEN) { // 收到的数据长度小于头部长度,说明头部还未全部读取
// 如果未完全接收消息头,则将接收到的数据复制到头部缓冲区
// _recv_head_node->_msg,更新当前头部的接收长度,并继续异步读取剩余数据。
memcpy(_recv_head_node->_msg + _recv_head_node->_cur_len, _data + copy_len, bytes_transferred);
_recv_head_node->_cur_len += bytes_transferred;
// 缓冲区清零,无需更新copy_len追踪已处理的字符数,因为之前读取的数据已经全部写入头部节点,下一个
// 读入的消息从头开始(copy_len=0)往头节点写
::memset(_data, 0, MAX_LENGTH);
// 继续读消息
_socket.async_read_some(boost::asio::buffer(_data, MAX_LENGTH), std::bind(&CSession::headle_read, this,
std::placeholders::_1, std::placeholders::_2, _self_shared));
return;
}
// 如果接收到的数据量足够处理消息头部,则计算头部剩余的未接收字节,
// 并将其从 _data 缓冲区复制到头部消息缓冲区 _recv_head_node->_msg
int head_remain = HEAD_TOTAL_LEN - _recv_head_node->_cur_len; // 头部剩余未复制的长度
// 填充头部节点
memcpy(_recv_head_node->_msg + _recv_head_node->_cur_len, _data + copy_len, head_remain);
copy_len += head_remain; // 更新已处理的data长度
bytes_transferred -= head_remain; // 更新剩余未处理的长度
short msg_id = 0; // 获取消息id
memcpy(&msg_id, _recv_head_node->_msg, HEAD_ID_LEN);
//网络字节序转化为本地字节序
msg_id = boost::asio::detail::socket_ops::network_to_host_short(msg_id);
cout << "msg_id is " << msg_id << endl;
// 判断id是否合法
if (msg_id > MAX_LENGTH) {
std::cout << "invaild msg_id is " << msg_id << endl;
_server->ClearSession(_uuid);
return;
}
short msg_len = 0; // 获取头部数据(消息长度)
memcpy(&msg_len, _recv_head_node->_msg + HEAD_ID_LEN, HEAD_DATA_LEN);
//网络字节序转化为本地字节序
msg_len = boost::asio::detail::socket_ops::network_to_host_short(msg_len);
cout << "msg_len is " << msg_len << endl;
if (msg_len > MAX_LENGTH) { // 判断头部长度是否非法
std::cout << "invalid data length is " << msg_len << endl;
_server->ClearSession(_uuid);
return;
}
_recv_msg_node = std::make_shared<RecvNode>(msg_len, msg_id); // 已知数据长度msg_len,构建消息内容载体
//消息的长度小于头部规定的长度,说明数据未收全,则先将部分消息放到接收节点里
if (bytes_transferred < msg_len) {
memcpy(_recv_msg_node->_msg + _recv_msg_node->_cur_len, _data + copy_len, bytes_transferred);
_recv_msg_node->_cur_len += bytes_transferred;
// copy_len不用更新,缓冲区会清零,下一个读入data的数据从头开始写入,copy_len也会被初始化为0
::memset(_data, 0, MAX_LENGTH);
_socket.async_read_some(boost::asio::buffer(_data, MAX_LENGTH),
std::bind(&CSession::headle_read, this, std::placeholders::_1, std::placeholders::_2, _self_shared));
_b_head_parse = true; //头部处理完成
return;
}
// 接收的长度多于消息内容长度
memcpy(_recv_msg_node->_msg + _recv_msg_node->_cur_len, _data + copy_len, msg_len);
_recv_msg_node->_cur_len += msg_len;
copy_len += msg_len;
bytes_transferred -= msg_len;
_recv_msg_node->_msg[_recv_msg_node->_total_len] = '\0';
// cout << "receive data is " << _recv_msg_node->_msg << endl;
// protobuf序列化
//MsgData msgdata;
//std::string receive_data;
//msgdata.ParseFromString(std::string(_recv_msg_node->_msg, _recv_msg_node->_total_len));
//std::cout << "receive msg id is " << msgdata.id () << " msg data is " << msgdata.data() << endl;
//std::string return_str = "Server has received msg, msg data is " + msgdata.data();
//MsgData msgreturn;
//msgreturn.set_id(msgdata.id());
//msgreturn.set_data(return_str);
//msgreturn.SerializeToString(&return_str);
//Send(return_str);
// jsoncpp序列化
Json::Reader reader;
Json::Value root;
reader.parse(std::string(_recv_msg_node->_msg, _recv_msg_node->_total_len), root);
std::cout << "recevie msg id is " << root["id"].asInt() << " msg data is "
<< root["data"].asString() << endl;
root["data"] = "Server has received msg, msg data is " + root["data"].asString();
std::string return_str = root.toStyledString();
Send(return_str, root["id"].asInt());
//Send(_recv_msg_node->_msg, _recv_msg_node->_total_len); // 回传
// 清理已处理的头部消息并重置,准备解析下一条消息
_b_head_parse = false;
_recv_head_node->Clear();
// 如果当前数据已经全部处理完,重置缓冲区 _data,并继续异步读取新的数据
if (bytes_transferred <= 0) {
::memset(_data, 0, MAX_LENGTH);
_socket.async_read_some(boost::asio::buffer(_data, MAX_LENGTH),
std::bind(&CSession::headle_read, this, std::placeholders::_1, std::placeholders::_2, _self_shared));
return;
}
continue; // 异步读取的消息未处理完,继续填充头节点乃至新的消息节点
}
//已经处理完头部,处理上次未接受完的消息数据
int remain_msg = _recv_msg_node->_total_len - _recv_msg_node->_cur_len;
if (bytes_transferred < remain_msg) { //接收的数据仍不足剩余未处理的
memcpy(_recv_msg_node->_msg + _recv_msg_node->_cur_len, _data + copy_len, bytes_transferred);
_recv_msg_node->_cur_len += bytes_transferred;
::memset(_data, 0, MAX_LENGTH);
_socket.async_read_some(boost::asio::buffer(_data, MAX_LENGTH),
std::bind(&CSession::headle_read, this, std::placeholders::_1, std::placeholders::_2, _self_shared));
return;
}
// 接收的数据多于剩余未处理的长度
memcpy(_recv_msg_node->_msg + _recv_msg_node->_cur_len, _data + copy_len, remain_msg);
_recv_msg_node->_cur_len += remain_msg;
bytes_transferred -= remain_msg;
copy_len += remain_msg;
_recv_msg_node->_msg[_recv_msg_node->_total_len] = '\0';
//cout << "receive data is " << _recv_msg_node->_msg << endl;
// protobuf序列化
//MsgData msgdata;
//std::string receive_data;
//msgdata.ParseFromString(std::string(_recv_msg_node->_msg, _recv_msg_node->_total_len));
//std::cout << "receive msg id is " << msgdata.id() << " msg data is " << msgdata.data() << endl;
//std::string return_str = "Server has received msg, msg data is " + msgdata.data();
//MsgData msgreturn;
//msgreturn.set_id(msgdata.id());
//msgreturn.set_data(return_str);
//msgreturn.SerializeToString(&return_str);
//Send(return_str);
//jsoncpp序列化
Json::Reader reader;
Json::Value root;
reader.parse(std::string(_recv_msg_node->_msg, _recv_msg_node->_total_len), root);
std::cout << "recevie msg id is " << root["id"].asInt() << " msg data is "
<< root["data"].asString() << endl;
root["data"] = "Server has received msg, msg data is " + root["data"].asString();
std::string return_str = root.toStyledString();
Send(return_str, root["id"].asInt());
//此处可以调用Send发送测试
//Send(_recv_msg_node->_msg, _recv_msg_node->_total_len);
//继续轮询剩余未处理数据
_b_head_parse = false;
_recv_head_node->Clear();
if (bytes_transferred <= 0) {
::memset(_data, 0, MAX_LENGTH);
_socket.async_read_some(boost::asio::buffer(_data, MAX_LENGTH),
std::bind(&CSession::headle_read, this, std::placeholders::_1, std::placeholders::_2, _self_shared));
return;
}
continue;
}
}
else {
std::cout << "handle read failed, error is " << error.what() << endl;
Close();
_server->ClearSession(_uuid);
}
}
HandleRead函数中新增一段读取消息id的代码
首先,当消息头节点_recv_head_node填充完毕后,获取头节点中存储的消息id并转换为本地字节序,并判断id的合法性;然后,解析消息长度,并构建消息体节点_recv_msg_node,读取剩下的消息体内容
short msg_id = 0; // 获取消息id
memcpy(&msg_id, _recv_head_node->_msg, HEAD_ID_LEN);
//网络字节序转化为本地字节序
msg_id = boost::asio::detail::socket_ops::network_to_host_short(msg_id);
cout << "msg_id is " << msg_id << endl;
// 判断id是否合法
if (msg_id > MAX_LENGTH) {
std::cout << "invaild msg_id is " << msg_id << endl;
_server->ClearSession(_uuid);
return;
}
3)客户端
客户端也需额外收发消息id
#include <boost/asio.hpp>
#include <iostream>
#include <json/json.h>
#include <json/value.h>
#include <json/reader.h>
using namespace boost::asio::ip;
using std::cout;
using std::endl;
const int MAX_LENGTH = 1024 * 2; // 发送和接收的长度为1024 * 2字节
const int HEAD_LENGTH = 2;
const int HEAD_TOTAL = 4;
int main()
{
try {
boost::asio::io_context ioc; // 创建上下文服务
// 127.0.0.1是本机的回路地址,也就是服务器和客户端在一个机器上
tcp::endpoint remote_ep(address::from_string("127.0.0.1"), 10086); // 构造endpoint
tcp::socket sock(ioc);
boost::system::error_code error = boost::asio::error::host_not_found; // 错误:主机未找到
sock.connect(remote_ep, error);
if (error) {
cout << "connect failed, code is " << error.value() << " error msg is " << error.message();
return 0;
}
Json::Value root;
root["id"] = 1001;
root["data"] = "hello world";
std::string request = root.toStyledString();
size_t request_length = request.length();
char send_data[MAX_LENGTH] = { 0 };
int msgid = 1001;
int msgid_host = boost::asio::detail::socket_ops::host_to_network_short(msgid);
memcpy(send_data, &msgid_host, 2);
//转为网络字节序
int request_host_length = boost::asio::detail::socket_ops::host_to_network_short(request_length);
memcpy(send_data + 2, &request_host_length, 2);
memcpy(send_data + 4, request.c_str(), request_length);
boost::asio::write(sock, boost::asio::buffer(send_data, request_length + 4));
char reply_head[HEAD_TOTAL]; // 首先读取对端发送消息的总长度
size_t reply_length = boost::asio::read(sock, boost::asio::buffer(reply_head, HEAD_TOTAL));
msgid = 0;
memcpy(&msgid, reply_head, HEAD_LENGTH);
short msglen = 0; // 消息总长度
memcpy(&msglen, reply_head + 2, HEAD_LENGTH); // 将消息总长度赋值给msglen
//转为本地字节序
msglen = boost::asio::detail::socket_ops::network_to_host_short(msglen);
char msg[MAX_LENGTH] = { 0 }; // 构建消息体(不含消息总长度)
size_t msg_length = boost::asio::read(sock, boost::asio::buffer(msg, msglen));
Json::Reader reader;
reader.parse(std::string(msg, msg_length), root);
std::cout << "msg id is " << root["id"] << " msg is " << root["data"] << endl;
getchar();
}
catch (std::exception& e) {
std::cerr << "Exception: " << e.what() << endl;
}
return 0;
}