chore: update auth security

2026-04-27 14:14:36 +08:00
parent cf9acf3ea0
commit 5a8baf7a0b
4 changed files with 22 additions and 15 deletions
@@ -404,6 +404,7 @@ func (c *AgentController) WSConnect(ctx *gin.Context) {
 		logger.Errorf("[AgentWS] 升级连接失败: %v, Agent #%s, IP=%s", err, agent.ID, ip)
 		return
 	}
+	conn.SetReadLimit(constant.MaxMessageSize)

 	// 连接成功，重置失败计数
 	c.wsManager.RecordConnectSuccess(ip)