diff --git a/main.go b/main.go index 1d411a0..1de853f 100644 --- a/main.go +++ b/main.go @@ -34,8 +34,14 @@ func main() { sessionMgr := auth.NewSessionManager(cfg.SessionTimeout) authMw := auth.NewAuthMiddleware(cfg.Username, cfg.Password, sessionMgr, templates) + // 转换 BlockedDomains 为 map + blockedDomainsMap := make(map[string]bool) + for _, domain := range cfg.BlockedDomains { + blockedDomainsMap[domain] = true + } + validator := security.NewRequestValidator( - cfg.BlockedDomains, + blockedDomainsMap, cfg.BlockedCIDRs, cfg.AllowedSchemes, )