更新 main.go
This commit is contained in:
8
main.go
8
main.go
@@ -34,8 +34,14 @@ func main() {
|
|||||||
sessionMgr := auth.NewSessionManager(cfg.SessionTimeout)
|
sessionMgr := auth.NewSessionManager(cfg.SessionTimeout)
|
||||||
authMw := auth.NewAuthMiddleware(cfg.Username, cfg.Password, sessionMgr, templates)
|
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(
|
validator := security.NewRequestValidator(
|
||||||
cfg.BlockedDomains,
|
blockedDomainsMap,
|
||||||
cfg.BlockedCIDRs,
|
cfg.BlockedCIDRs,
|
||||||
cfg.AllowedSchemes,
|
cfg.AllowedSchemes,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user