更新 main.go

2025-12-15 02:58:02 +08:00
parent 3dd381b60f
commit abe2586163
1 changed files with 7 additions and 1 deletions
--- 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,
    )