refactor: 按 WCH 官方 TCPServer 例程重写 TCP JSON server

核心变更:去掉 g_json_socket_client,listen socket 直接承载收发数据。

参考 EVT/EXAM/ETH/TCPServer 例程:
- 创建 PROTO_TYPE_TCP socket → WCHNET_SocketListen
- 同一 socket 处理 CONNECT + RECV + DISCONNECT + TIMEOUT
- 不需要 'accepted socket' 检测

移除的复杂逻辑:
- g_json_socket_client 变量及所有 'newly accepted' 检测代码
- WCHNET_HandleSockInt 第二路由条件(socketid!=listen,!=TCP,!=UDP 那串)
- tcp_json_handle_sock_int 中 listen/client 分离处理
- listen==client overlap 的迂回保护

修改文件:
- tcp_json_srv.h: 移除 g_json_socket_client extern
- tcp_json_srv.c: 移除 g_json_socket_client,handler 简化为 4 个 if
- net_srv.c: 路由简化为仅 socketid==g_json_socket_listen
This commit is contained in:
wangfq
2026-07-01 11:18:43 +08:00
parent 7804d97a45
commit ba35ea8ae3
4 changed files with 34 additions and 71 deletions

4
.gitignore vendored
View File

@@ -67,7 +67,6 @@ mdk_v5/
*.url *.url
*.txt *.txt
at32_ide_proj/ at32_ide_proj/
*.uvguix* *.uvguix*
*.cproject *.cproject
@@ -83,3 +82,6 @@ at32_ide_proj/
*.pdf *.pdf
*.PDF *.PDF
wchreference/

View File

@@ -49,8 +49,7 @@ typedef enum {
*===========================================================================*/ *===========================================================================*/
extern uint8_t SocketId_TCP; extern uint8_t SocketId_TCP;
extern uint8_t SocketId_UDP; extern uint8_t SocketId_UDP;
extern uint8_t g_json_socket_listen; // listen socket ID extern uint8_t g_json_socket_listen; // TCP socket ID (listen + data)
extern uint8_t g_json_socket_client; // accepted client socket ID (0xFF = none)
extern TcpJsonAuthState g_json_auth_state; extern TcpJsonAuthState g_json_auth_state;
extern uint32_t g_json_auth_timer; // ms timer for auth timeout extern uint32_t g_json_auth_timer; // ms timer for auth timeout
extern uint8_t g_json_pwd_retry; // password retry counter extern uint8_t g_json_pwd_retry; // password retry counter

View File

@@ -481,15 +481,8 @@ void WCHNET_HandleSockInt(uint8_t socketid, uint8_t intstat)
{ {
uint8_t i; uint8_t i;
// Route JSON protocol socket events (listen socket, client socket) // Route JSON protocol socket events (same socket handles listen + data)
if (socketid == g_json_socket_listen || socketid == g_json_socket_client) { if (socketid == g_json_socket_listen) {
tcp_json_handle_sock_int(socketid, intstat);
return;
}
// Also catch newly accepted TCP connections that might be JSON clients
// Exclude: listen socket, SSC TCP, UDP
if ((intstat & SINT_STAT_CONNECT) && socketid != g_json_socket_listen
&& socketid != SocketId_TCP && socketid != SocketId_UDP) {
tcp_json_handle_sock_int(socketid, intstat); tcp_json_handle_sock_int(socketid, intstat);
return; return;
} }

View File

@@ -24,8 +24,7 @@
/*=========================================================================== /*===========================================================================
* Global State * Global State
*===========================================================================*/ *===========================================================================*/
uint8_t g_json_socket_listen = 0xFF; // listen socket ID uint8_t g_json_socket_listen = 0xFF; // TCP socket ID — handles both listen + client data
uint8_t g_json_socket_client = 0xFF; // accepted client socket ID
TcpJsonAuthState g_json_auth_state = JSON_STATE_WAIT_AUTH; TcpJsonAuthState g_json_auth_state = JSON_STATE_WAIT_AUTH;
uint32_t g_json_auth_timer = 0; uint32_t g_json_auth_timer = 0;
uint8_t g_json_pwd_retry = 0; uint8_t g_json_pwd_retry = 0;
@@ -624,8 +623,22 @@ void tcp_json_srv_init(void) {
} }
void tcp_json_handle_sock_int(uint8_t socketid, uint8_t intstat) { void tcp_json_handle_sock_int(uint8_t socketid, uint8_t intstat) {
// === RECV: Always process for the client socket (handles listen==client overlap) === // Only process events on our TCP listen/data socket
if (socketid == g_json_socket_client && (intstat & SINT_STAT_RECV)) { if (socketid != g_json_socket_listen) return;
// === CONNECT: client connected — configure recv buffer, init auth state ===
if (intstat & SINT_STAT_CONNECT) {
WCHNET_ModifyRecvBuf(socketid, (uint32_t)g_json_wchnet_buf, RECE_BUF_LEN);
g_json_auth_state = JSON_STATE_WAIT_AUTH;
g_json_pwd_retry = 0;
g_json_auth_timer = mstick();
g_json_recv_len = 0;
memset(g_json_recv_buf, 0, sizeof(g_json_recv_buf));
PRINT("JSON: Client connected on socket %d, recv buf configured\n", socketid);
}
// === RECV: read and process incoming data ===
if (intstat & SINT_STAT_RECV) {
uint32_t recv_len = WCHNET_SocketRecvLen(socketid, NULL); uint32_t recv_len = WCHNET_SocketRecvLen(socketid, NULL);
if (recv_len > 0) { if (recv_len > 0) {
uint16_t space = TCP_JSON_RECV_BUF_LEN - g_json_recv_len; uint16_t space = TCP_JSON_RECV_BUF_LEN - g_json_recv_len;
@@ -652,76 +665,32 @@ void tcp_json_handle_sock_int(uint8_t socketid, uint8_t intstat) {
} }
} }
// === Listen socket events (log only, no action needed) === // === DISCONNECT: client disconnected, reset state ===
if (socketid == g_json_socket_listen && socketid != g_json_socket_client) { if (intstat & SINT_STAT_DISCONNECT) {
if (intstat & SINT_STAT_CONNECT) { PRINT("JSON: Client disconnected (id=%d)\n", socketid);
PRINT("JSON: Listen socket got CONNECT\n");
}
if (intstat & SINT_STAT_DISCONNECT) {
PRINT("JSON: Listen socket disconnect (unexpected)\n");
}
}
// === Newly accepted client (CONNECT on unknown socket, but NOT the listen socket) ===
if ((intstat & SINT_STAT_CONNECT) && socketid != g_json_socket_client
&& socketid != g_json_socket_listen
&& socketid != SocketId_TCP && socketid != SocketId_UDP) {
g_json_socket_client = socketid;
g_json_auth_state = JSON_STATE_WAIT_AUTH; g_json_auth_state = JSON_STATE_WAIT_AUTH;
g_json_pwd_retry = 0; g_json_pwd_retry = 0;
g_json_auth_timer = mstick();
g_json_recv_len = 0; g_json_recv_len = 0;
memset(g_json_recv_buf, 0, sizeof(g_json_recv_buf));
WCHNET_ModifyRecvBuf(socketid, (uint32_t)g_json_wchnet_buf, RECE_BUF_LEN);
PRINT("JSON: Client accepted on socket %d, recv buf configured\n", socketid);
return;
} }
// === Client socket events (non-RECV: CONNECT, DISCONNECT, TIMEOUT) === // === TIMEOUT: connection timed out ===
if (socketid == g_json_socket_client) { if (intstat & SINT_STAT_TIM_OUT) {
if (intstat & SINT_STAT_CONNECT) { PRINT("JSON: Client timeout (id=%d)\n", socketid);
PRINT("JSON: Client socket connected (id=%d)\n", socketid);
g_json_auth_state = JSON_STATE_WAIT_AUTH;
g_json_pwd_retry = 0;
g_json_auth_timer = mstick();
g_json_recv_len = 0;
}
if (intstat & SINT_STAT_DISCONNECT) {
PRINT("JSON: Client disconnected (id=%d)\n", socketid);
g_json_socket_client = 0xFF;
g_json_auth_state = JSON_STATE_WAIT_AUTH;
g_json_pwd_retry = 0;
g_json_recv_len = 0;
}
if (intstat & SINT_STAT_TIM_OUT) {
PRINT("JSON: Client timeout (id=%d)\n", socketid);
g_json_socket_client = 0xFF;
g_json_auth_state = JSON_STATE_WAIT_AUTH;
g_json_pwd_retry = 0;
g_json_recv_len = 0;
}
return;
}
// === Fallback: connect on a socket we haven't tracked yet ===
if (intstat & SINT_STAT_CONNECT && g_json_socket_client == 0xFF && socketid != g_json_socket_listen) {
g_json_socket_client = socketid;
g_json_auth_state = JSON_STATE_WAIT_AUTH; g_json_auth_state = JSON_STATE_WAIT_AUTH;
g_json_pwd_retry = 0; g_json_pwd_retry = 0;
g_json_auth_timer = mstick();
g_json_recv_len = 0; g_json_recv_len = 0;
memset(g_json_recv_buf, 0, sizeof(g_json_recv_buf));
PRINT("JSON: Client accepted on socket %d (fallback)\n", socketid);
} }
} }
void tcp_json_poll(void) { void tcp_json_poll(void) {
// Only poll if we have an active connection (socket configured)
if (g_json_socket_listen == 0xFF) return;
// Auth timeout check // Auth timeout check
if (g_json_auth_state == JSON_STATE_WAIT_AUTH && g_json_socket_client != 0xFF) { if (g_json_auth_state == JSON_STATE_WAIT_AUTH) {
if (mstick() - g_json_auth_timer > TCP_JSON_AUTH_TIMEOUT_MS) { if (mstick() - g_json_auth_timer > TCP_JSON_AUTH_TIMEOUT_MS) {
PRINT("JSON: Auth timeout, closing connection\n"); PRINT("JSON: Auth timeout, closing connection\n");
WCHNET_SocketClose(g_json_socket_client, TCP_CLOSE_NORMAL); WCHNET_SocketClose(g_json_socket_listen, TCP_CLOSE_NORMAL);
g_json_socket_client = 0xFF;
g_json_auth_state = JSON_STATE_WAIT_AUTH; g_json_auth_state = JSON_STATE_WAIT_AUTH;
g_json_pwd_retry = 0; g_json_pwd_retry = 0;
g_json_recv_len = 0; g_json_recv_len = 0;