SSH 登录事件增加 IP 白名单

This commit is contained in:
dushixiang committed 2026-01-11 13:56:21 +08:00
1 parent 7965cd329a
commit f0cff6a878
4 files changed
+17 -20

No files matched your search

+3 -5
View File
@@ -4,6 +4,7 @@ import (
"context"
"net/http"
"github.com/dushixiang/pika/internal/models"
"github.com/dushixiang/pika/internal/service"
"github.com/go-orz/orz"
"github.com/labstack/echo/v4"
@@ -50,15 +51,12 @@ func (h *SSHLoginHandler) GetConfig(c echo.Context) error {
func (h *SSHLoginHandler) UpdateConfig(c echo.Context) error {
agentID := c.Param("id")
var req struct {
Enabled bool `json:"enabled"`
}
var req models.SSHLoginConfigData
if err := c.Bind(&req); err != nil {
return err
}
err := h.service.UpdateConfig(c.Request().Context(), agentID, req.Enabled)
err := h.service.UpdateConfig(c.Request().Context(), agentID, &req)
if err != nil {
h.logger.Error("更新SSH登录监控配置失败", zap.Error(err))
return c.JSON(http.StatusInternalServerError, map[string]string{
+3 -5
View File
@@ -4,6 +4,7 @@ import (
"context"
"net/http"
"github.com/dushixiang/pika/internal/models"
"github.com/dushixiang/pika/internal/service"
"github.com/go-orz/orz"
"github.com/labstack/echo/v4"
@@ -27,16 +28,13 @@ func NewTamperHandler(logger *zap.Logger, tamperService *service.TamperService)
func (h *TamperHandler) UpdateConfig(c echo.Context) error {
agentID := c.Param("id")
var req struct {
Enabled bool `json:"enabled"`
Paths []string `json:"paths"`
}
var req models.TamperProtectConfigData
if err := c.Bind(&req); err != nil {
return err
}
err := h.tamperService.UpdateConfig(c.Request().Context(), agentID, req.Enabled, req.Paths)
err := h.tamperService.UpdateConfig(c.Request().Context(), agentID, &req)
if err != nil {
h.logger.Error("更新防篡改配置失败", zap.Error(err), zap.String("agentId", agentID))
return c.JSON(http.StatusInternalServerError, map[string]interface{}{