From abe2586163d12bd2707033fb847e5059cef5e1b7 Mon Sep 17 00:00:00 2001
From: XOF <xofine@noreply.localhost>
Date: Mon, 15 Dec 2025 02:58:02 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20main.go?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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,
     )