From 640bd8346526d1a5fb6f0230281d4026ebd10a8b Mon Sep 17 00:00:00 2001 From: Yahtoo Ma Date: Mon, 5 Aug 2019 10:10:24 +0800 Subject: [PATCH] Del expired banned peers when add peer --- p2p/security/blacklist.go | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/p2p/security/blacklist.go b/p2p/security/blacklist.go index 8f5682a8..951d659b 100644 --- a/p2p/security/blacklist.go +++ b/p2p/security/blacklist.go @@ -38,6 +38,13 @@ func (bl *Blacklist) AddPeer(ip string) error { bl.mtx.Lock() defer bl.mtx.Unlock() + // delete expired banned peers + for peer, banEnd := range bl.peers { + if time.Now().Before(banEnd) { + delete(bl.peers, peer) + } + } + // add banned peer bl.peers[ip] = time.Now().Add(defaultBanDuration) dataJSON, err := json.Marshal(bl.peers) if err != nil { -- 2.11.0