Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-03-30 15:08:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Unban address after one day (#1479)

Browse Source 
* Unban address after one day

* Unban addresses one by one

Co-authored-by: stasatdaglabs <39559713+stasatdaglabs@users.noreply.github.com>
This commit is contained in:
Ori Newman 2021-02-01 18:52:49 +02:00 committed by GitHub
parent b636ae234e
commit aeb8e9d2cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
infrastructure/network/addressmanager/addressmanager.go
View File

@ -5,8 +5,10 @@
package addressmanager
import (
"github.com/kaspanet/kaspad/util/mstime"
"net"
"sync"
"time"
"github.com/kaspanet/kaspad/app/appmessage"
"github.com/pkg/errors"
@ -205,14 +207,13 @@ func (am *AddressManager) Unban(address *appmessage.NetAddress) error {
defer am.mutex.Unlock()
key := netAddressKey(address)
bannedAddress, ok := am.bannedAddresses[key.address]
_, ok := am.bannedAddresses[key.address]
if !ok {
return errors.Wrapf(ErrAddressNotFound, "address %s "+
"is not registered with the address manager as banned", address.TCPAddress())
}
delete(am.bannedAddresses, key.address)
am.addresses[key] = bannedAddress
return nil
}
@ -222,6 +223,7 @@ func (am *AddressManager) IsBanned(address *appmessage.NetAddress) (bool, error)
defer am.mutex.Unlock()
key := netAddressKey(address)
am.unbanIfOldEnough(key.address)
if _, ok := am.bannedAddresses[key.address]; !ok {
if _, ok = am.addresses[key]; !ok {
return false, errors.Wrapf(ErrAddressNotFound, "address %s "+
@ -233,3 +235,15 @@ func (am *AddressManager) IsBanned(address *appmessage.NetAddress) (bool, error)
return true, nil
}
func (am *AddressManager) unbanIfOldEnough(ipv6Address ipv6) {
address, ok := am.bannedAddresses[ipv6Address]
if !ok {
return
}
const maxBanTime = 24 * time.Hour
if mstime.Since(address.Timestamp) > maxBanTime {
delete(am.bannedAddresses, ipv6Address)
}
}