pqc
/
execve_hook
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: 协议消息类型改名，为以后更多服务器命令做预留

This commit is contained in:
QCQCQC@Opi5 2025-12-13 18:42:33 +08:00
parent c05c02c098
commit 735ead43b5
2 changed files with 39 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/client.c
View File

@ -532,7 +532,7 @@ static void* response_listener_thread(void* arg) {
}
/* 处理服务端响应 */
if (msg_type == MSG_TYPE_SERVER_RESPONSE && payload != NULL) {
if (msg_type == MSG_TYPE_SERVER_RENDER && payload != NULL) {
ssize_t written = write(*output_fd, payload, payload_len);
if (written < 0) {
DEBUG_LOG("写入输出失败: %s", strerror(errno));

72
src/socket_protocol.h
View File

@ -11,14 +11,14 @@
// 消息类型枚举
typedef enum {
MSG_TYPE_INIT = 1, // 初始化连接，发送命令信息
MSG_TYPE_WINDOW_SIZE_UPDATE = 2, // 终端窗口大小更新
MSG_TYPE_SERVER_RESPONSE = 3, // 服务器响应消息
MSG_TYPE_CLOSE = 4, // 关闭连接
MSG_TYPE_TERMINAL_INPUT = 5, // 终端输入数据（键盘、鼠标等）
MSG_TYPE_TERMINAL_OUTPUT = 6, // 终端输出数据
MSG_TYPE_MOUSE_EVENT = 7, // 鼠标事件
MSG_TYPE_KEY_EVENT = 8 // 键盘事件
MSG_TYPE_INIT = 1, // 初始化连接，发送命令信息
MSG_TYPE_WINDOW_SIZE_UPDATE = 2, // 终端窗口大小更新
MSG_TYPE_SERVER_RENDER = 3, // 服务器渲染消息
MSG_TYPE_CLOSE = 4, // 关闭连接
MSG_TYPE_TERMINAL_INPUT = 5, // 终端输入数据（键盘、鼠标等）
MSG_TYPE_TERMINAL_OUTPUT = 6, // 终端输出数据
MSG_TYPE_MOUSE_EVENT = 7, // 鼠标事件
MSG_TYPE_KEY_EVENT = 8 // 键盘事件
} MessageType;
// 消息头结构（固定大小）
@ -28,29 +28,30 @@ typedef enum {
// bit 2: MSG_FLAG_COMPRESS_HC - 使用高压缩比模式
// 高 16 位: 压缩前的原始大小（用于解压缓冲区分配）
typedef struct {
uint32_t magic; // 魔数，用于验证 0x42534D54 ("BSMT")
uint32_t type; // 消息类型
uint32_t payload_len; // 载荷长度（压缩后）
uint32_t reserved; // 低 16 位: 压缩标志; 高 16 位: 原始大小/256
uint32_t magic; // 魔数，用于验证 0x42534D54 ("BSMT")
uint32_t type; // 消息类型
uint32_t payload_len; // 载荷长度（压缩后）
uint32_t reserved; // 低 16 位: 压缩标志; 高 16 位: 原始大小/256
} __attribute__((packed)) MessageHeader;
// 从 reserved 字段提取压缩标志和原始大小
#define GET_COMPRESS_FLAGS(reserved) ((reserved) & 0xFFFF)
#define GET_ORIGINAL_SIZE_HINT(reserved) (((reserved) >> 16) * 256)
#define MAKE_RESERVED(flags, orig_size) (((flags) & 0xFFFF) | (((orig_size) / 256) << 16))
#define MAKE_RESERVED(flags, orig_size) \
(((flags) & 0xFFFF) | (((orig_size) / 256) << 16))
// 终端信息结构
typedef struct {
uint32_t is_tty; // 是否为TTY
uint16_t rows; // 行数
uint16_t cols; // 列数
uint16_t x_pixel; // X像素
uint16_t y_pixel; // Y像素
uint32_t has_termios; // 是否有termios属性
uint32_t input_flags; // termios输入标志
uint32_t output_flags; // termios输出标志
uint32_t control_flags; // termios控制标志
uint32_t local_flags; // termios本地标志
uint32_t is_tty; // 是否为TTY
uint16_t rows; // 行数
uint16_t cols; // 列数
uint16_t x_pixel; // X像素
uint16_t y_pixel; // Y像素
uint32_t has_termios; // 是否有termios属性
uint32_t input_flags; // termios输入标志
uint32_t output_flags; // termios输出标志
uint32_t control_flags; // termios控制标志
uint32_t local_flags; // termios本地标志
} __attribute__((packed)) TerminalInfoFixed;
// 鼠标事件类型
@ -73,9 +74,9 @@ typedef struct {
// 键盘事件结构
typedef struct {
uint32_t key_code; // 键码
uint32_t modifiers; // 修饰键
uint32_t is_press; // 1=按下，0=释放
uint32_t key_code; // 键码
uint32_t modifiers; // 修饰键
uint32_t is_press; // 1=按下，0=释放
} __attribute__((packed)) KeyEvent;
// 魔数定义
@ -84,23 +85,26 @@ typedef struct {
#ifdef HAVE_LZ4
// 协议上下文（包含压缩状态）
typedef struct {
CompressionContext compress_ctx; // 压缩上下文
int compression_enabled; // 是否启用压缩
CompressionContext compress_ctx; // 压缩上下文
int compression_enabled; // 是否启用压缩
} ProtocolContext;
// 初始化协议上下文
void protocol_init(ProtocolContext* ctx, CompressionType compress_type, int compress_level);
void protocol_init(ProtocolContext* ctx, CompressionType compress_type,
int compress_level);
// 带压缩支持的消息读写
int write_message_compressed(int sock, ProtocolContext* ctx, MessageType type,
int write_message_compressed(int sock, ProtocolContext* ctx, MessageType type,
const void* payload, uint32_t payload_len);
int read_message_decompressed(int sock, MessageType* type, void** payload,
int read_message_decompressed(int sock, MessageType* type, void** payload,
uint32_t* payload_len, uint32_t* original_len);
#endif
// 函数声明
int write_message(int sock, MessageType type, const void* payload, uint32_t payload_len);
int read_message(int sock, MessageType* type, void** payload, uint32_t* payload_len);
int write_message(int sock, MessageType type, const void* payload,
uint32_t payload_len);
int read_message(int sock, MessageType* type, void** payload,
uint32_t* payload_len);
void free_message_payload(void* payload);
@ -110,4 +114,4 @@ int restore_terminal_mode(int fd);
int enable_mouse_tracking(int fd);
int disable_mouse_tracking(int fd);
#endif // SOCKET_PROTOCOL_H
#endif // SOCKET_PROTOCOL_H