From 5359ee89d4ddf9f3b7d85143e896480d30010943 Mon Sep 17 00:00:00 2001
From: dushixiang <dushixiang@typesafe.cn>
Date: Sun, 11 Jan 2026 10:20:38 +0800
Subject: [PATCH] =?UTF-8?q?=E5=9C=A8=E6=8E=A2=E9=92=88=E6=B3=A8=E5=86=8C?=
 =?UTF-8?q?=E5=90=8E=E6=9C=8D=E5=8A=A1=E7=AB=AF=E4=B8=BB=E5=8A=A8=E4=B8=8B?=
 =?UTF-8?q?=E5=8F=91SSH=E7=99=BB=E5=BD=95=E7=9B=91=E6=8E=A7=E9=85=8D?=
 =?UTF-8?q?=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/handler/agent_ws_handler.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/internal/handler/agent_ws_handler.go b/internal/handler/agent_ws_handler.go
index ad69b14..e24f726 100644
--- a/internal/handler/agent_ws_handler.go
+++ b/internal/handler/agent_ws_handler.go
@@ -52,6 +52,11 @@ func (h *AgentHandler) HandleWebSocket(c echo.Context) error {
 		h.logger.Error("failed to send tamper config", zap.Error(err))
 		// 配置下发失败不中断连接，只记录日志
 	}
+	// 下发SSH登录监控配置
+	if err := h.sendSSHLoginConfig(conn, agent.ID); err != nil {
+		h.logger.Error("failed to send ssh login config", zap.Error(err))
+		// 配置下发失败不中断连接，只记录日志
+	}
 
 	// 创建客户端并注册到管理器
 	client := h.newClient(agent.ID, conn)
@@ -259,3 +264,17 @@ func (h *AgentHandler) sendTamperConfig(conn *websocket.Conn, agentID string) er
 
 	return conn.WriteMessage(websocket.TextMessage, msgData)
 }
+
+func (h *AgentHandler) sendSSHLoginConfig(conn *websocket.Conn, agentID string) error {
+	config, err := h.sshLoginService.GetConfig(context.Background(), agentID)
+	if err != nil {
+		return err
+	}
+	msgData, err := json.Marshal(protocol.SSHLoginConfig{
+		Enabled: config.Enabled,
+	})
+	if err != nil {
+		return err
+	}
+	return conn.WriteMessage(websocket.TextMessage, msgData)
+}