77 lines
3.0 KiB
Plaintext
77 lines
3.0 KiB
Plaintext
#ifndef TCP_SERVER_H
|
||
#define TCP_SERVER_H
|
||
|
||
#include <boost/asio.hpp>
|
||
#include <boost/crc.hpp>
|
||
#include <vector>
|
||
#include <string>
|
||
#include <memory> // For std::enable_shared_from_this
|
||
#include <cstdint> // For uint8_t, uint16_t, uint32_t
|
||
// 使用 Boost.Asio 的命名空间
|
||
using boost::asio::ip::tcp;
|
||
// ---------------- CRC32 Declaration (使用 Boost.CRC) ----------------
|
||
inline uint32_t calculate_crc32(const uint8_t* data, size_t length);
|
||
// ---------------- Protocol Definition ----------------
|
||
namespace proto {
|
||
struct ProtocolConfig {
|
||
static constexpr std::size_t kMagicLen = 4; // 魔数 4 字节
|
||
static constexpr std::size_t kMsgTypeLen = 2; // 消息类型 2 字节
|
||
static constexpr std::size_t kPayloadLenFieldLen = 4; // 载荷长度 4 字节
|
||
static constexpr std::size_t kCrcLen = 4; // CRC 校验位 4 字节
|
||
// 完整的头部长度 (Magic + MsgType + PayloadLenField)
|
||
static constexpr std::size_t kHeaderLen = kMagicLen + kMsgTypeLen + kPayloadLenFieldLen; // 4 + 2 + 4 = 10 字节
|
||
// 魔数常量
|
||
static constexpr uint8_t kMagic[kMagicLen] = {0x62, 0x6e, 0x73, 0x00};
|
||
};
|
||
// 字节序转换函数声明
|
||
inline uint16_t be16(const uint8_t* p);
|
||
inline uint32_t be32(const uint8_t* p);
|
||
inline void write_be16(uint8_t* p, uint16_t val);
|
||
inline void write_be32(uint8_t* p, uint32_t val);
|
||
|
||
struct Message {
|
||
uint16_t msg_type = 0; // 消息类型 2 字节
|
||
std::vector<uint8_t> payload;
|
||
uint32_t received_crc = 0; // 新增字段，用于存储接收到的CRC，方便调试
|
||
};
|
||
inline bool match_magic(const uint8_t* p);
|
||
}
|
||
// ---------------- Session Class (处理单个客户端连接) ----------------
|
||
class session : public std::enable_shared_from_this<session> {
|
||
public:
|
||
explicit session(tcp::socket socket);
|
||
void start();
|
||
private:
|
||
// 自定义缓冲区结构体
|
||
struct ReceiveBuffer {
|
||
static constexpr std::size_t capacity_ = 8192;
|
||
uint8_t data_[capacity_];
|
||
std::size_t read_idx = 0; // 已读到缓冲区的数据量 (读指针)
|
||
std::size_t write_idx = 0; // 已处理/消耗的字节数 (写指针，指向下一个待处理字节)
|
||
void compact();
|
||
std::size_t data_size() const;
|
||
std::size_t free_space() const;
|
||
const uint8_t* current_data() const;
|
||
uint8_t* write_ptr();
|
||
} receive_buffer_;
|
||
void do_read_some();
|
||
void parse_messages();
|
||
std::string payload_to_ascii_string(const std::vector<uint8_t>& payload);
|
||
std::string bytes_to_hex_string(const uint8_t* data, std::size_t len);
|
||
void on_message(proto::Message msg);
|
||
// 修改原有的 send_response 为更通用的 send_message
|
||
void send_message(uint16_t msg_type, const std::vector<uint8_t>& payload);
|
||
tcp::socket socket_;
|
||
};
|
||
// ---------------- TCPServer Class ----------------
|
||
// 这个类负责监听端口并接受新的客户端连接
|
||
class TCPServer {
|
||
public:
|
||
TCPServer(boost::asio::io_context& io_context, uint16_t port);
|
||
private:
|
||
void do_accept();
|
||
tcp::acceptor acceptor_;
|
||
};
|
||
|
||
#endif // SERVER_H
|