mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-29 02:06:43 +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>
40 lines
1.3 KiB
Go
40 lines
1.3 KiB
Go
package addressexchange
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/protocol/common"
|
|
peerpkg "github.com/kaspanet/kaspad/app/protocol/peer"
|
|
"github.com/kaspanet/kaspad/app/protocol/protocolerrors"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/addressmanager"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
)
|
|
|
|
// ReceiveAddressesContext is the interface for the context needed for the ReceiveAddresses flow.
|
|
type ReceiveAddressesContext interface {
|
|
AddressManager() *addressmanager.AddressManager
|
|
}
|
|
|
|
// ReceiveAddresses asks a peer for more addresses if needed.
|
|
func ReceiveAddresses(context ReceiveAddressesContext, incomingRoute *router.Route, outgoingRoute *router.Route,
|
|
peer *peerpkg.Peer) error {
|
|
|
|
subnetworkID := peer.SubnetworkID()
|
|
msgGetAddresses := appmessage.NewMsgRequestAddresses(false, subnetworkID)
|
|
err := outgoingRoute.Enqueue(msgGetAddresses)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
message, err := incomingRoute.DequeueWithTimeout(common.DefaultTimeout)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msgAddresses := message.(*appmessage.MsgAddresses)
|
|
if len(msgAddresses.AddressList) > addressmanager.GetAddressesMax {
|
|
return protocolerrors.Errorf(true, "address count exceeded %d", addressmanager.GetAddressesMax)
|
|
}
|
|
|
|
return context.AddressManager().AddAddresses(msgAddresses.AddressList...)
|
|
}
|