mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-29 10:16:45 +00:00
d835f72e74
* Move existing address/bannedAddress functionality to a new addressStore object. * Implement TestAddressManager. * Implement serializeAddressKey and deserializeAddressKey. * Implement serializeNetAddress and deserializeNetAddress. * Store addresses and banned addresses to disk. * Implement restoreNotBannedAddresses. * Fix bannedDatabaseKey. * Implement restoreBannedAddresses. * Implement TestRestoreAddressManager. * Defer closing the database in TestRestoreAddressManager. * Defer closing the database in TestRestoreAddressManager. * Add a log. * Return errors where appropriate. Co-authored-by: Elichai Turkel <elichai.turkel@gmail.com>
32 lines
889 B
Go
32 lines
889 B
Go
package addressmanager
|
|
|
|
import (
|
|
"net"
|
|
"strconv"
|
|
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// AddAddressByIP adds an address where we are given an ip:port and not a
|
|
// appmessage.NetAddress.
|
|
func AddAddressByIP(am *AddressManager, addressIP string, subnetworkID *externalapi.DomainSubnetworkID) error {
|
|
// Split IP and port
|
|
ipString, portString, err := net.SplitHostPort(addressIP)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Put it in appmessage.Netaddress
|
|
ip := net.ParseIP(ipString)
|
|
if ip == nil {
|
|
return errors.Errorf("invalid ip %s", ipString)
|
|
}
|
|
port, err := strconv.ParseUint(portString, 10, 0)
|
|
if err != nil {
|
|
return errors.Errorf("invalid port %s: %s", portString, err)
|
|
}
|
|
netAddress := appmessage.NewNetAddressIPPort(ip, uint16(port), 0)
|
|
return am.AddAddresses(netAddress)
|
|
}
|