diff --git a/firewall/firewall.go b/firewall/firewall.go index 2c08cf8..0f1aaa2 100644 --- a/firewall/firewall.go +++ b/firewall/firewall.go @@ -41,10 +41,11 @@ func ReloadBlacklist() { var bl blacklist err = json.Unmarshal(f, &bl) if err != nil { + Logger.Errorf("failed to unmarshal blacklist: %v", err) return } blacklistedAsn.Clear() - bannedIPs = bart.Table[int]{} + bannedIPs = &bart.Table[int]{} for _, v := range bl.BlacklistedAsn { blacklistedAsn.Store(v, true) } @@ -60,14 +61,14 @@ func ReloadBlacklist() { var WindowSize = 120 * time.Second -const MaxStringsPerIp = 3 +const MaxStringsPerIp = 4 var resourcesForIPCache = gcache.New(1000).Simple().Build() var whitelist = map[string]bool{ "51.210.0.171": true, } -var bannedIPs = bart.Table[int]{} +var bannedIPs = &bart.Table[int]{} var blacklistedAsn = xsync.NewMapOf[int, bool]() var Logger = logger.GetLogger() diff --git a/firewall/firewall_test.go b/firewall/firewall_test.go index b332b5e..f5563c8 100644 --- a/firewall/firewall_test.go +++ b/firewall/firewall_test.go @@ -57,4 +57,5 @@ func Test_checkBannedIp(t *testing.T) { } ip = "1.1.1.1" assert.False(t, CheckBans(ip)) + }