在探针注册后服务端主动下发SSH登录监控配置
This commit is contained in:
1 parent
602a36df01
commit
5359ee89d4
1 file changed
+19
@@ -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)
|
||||
}
|
||||
Reference in new issue
Block a user