mirrors/tendermint
1
0
Fork 0
mirror of https://github.com/tendermint/tendermint.git synced 2026-01-08 22:23:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

ban function requires a ban duration as an argument

Browse Source
This commit is contained in:
Callum Michael Waters
2020-03-11 15:06:42 +01:00
parent 429febde80
commit 08ccbdcb43
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
p2p/pex/addrbook.go
View File

@@ -339,7 +339,7 @@ func (a *addrBook) ReinstateBadPeers() {
a.mtx.Lock()
defer a.mtx.Unlock()
for _, ka := range a.badPeers {
if !ka.isBanned(defaultBanTime) {
if !ka.isBanned() {
bucket := a.calcNewBucket(ka.Addr, ka.Src)
a.addToNewBucket(ka, bucket)
delete(a.badPeers, ka.ID())
@@ -761,7 +761,7 @@ func (a *addrBook) addBadPeer(addr *p2p.NetAddress) bool {
if ka != nil {
if _, alreadyBadPeer := a.badPeers[addr.ID]; !alreadyBadPeer {
// add to bad peer list
ka.ban()
ka.ban(defaultBanTime)
a.badPeers[addr.ID] = ka
}
return true

8
p2p/pex/known_address.go
View File

@@ -55,12 +55,12 @@ func (ka *knownAddress) markGood() {
ka.LastSuccess = now
}
func (ka *knownAddress) ban() {
ka.LastBanTime = time.Now()
func (ka *knownAddress) ban(banTime time.Duration) {
ka.LastBanTime = time.Now().Add(banTime)
}
func (ka *knownAddress) isBanned(banTime time.Duration) bool {
return ka.LastBanTime.Add(banTime).Before(time.Now())
func (ka *knownAddress) isBanned() bool {
return ka.LastBanTime.Before(time.Now())
}
func (ka *knownAddress) addBucketRef(bucketIdx int) int {