在探针注册后服务端主动下发SSH登录监控配置

This commit is contained in:
dushixiang committed 2026-01-11 10:20:38 +08:00
1 parent 602a36df01
commit 5359ee89d4
1 file changed
+19
+19
View File
@@ -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)
}