mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-09-13 21:10:12 +00:00
42e50e6dc2
* [NOD-1191] Convert wire protocol to 100% protobuf * [NOD-1191] Simplify wire interface and remove redundant messages * [NOD-1191] Map all proto to wire conversions * [NOD-1203] Create netadapter outside of protocol manager * [NOD-1191] Fix nil errors * [NOD-1191] Fix comments * [NOD-1191] Add converter interface * [NOD-1191] Add missing GetBlockLocator message * [NOD-1191] Change message names that starts with 'get' to 'request' * [NOD-1191] Change message commands values * [NOD-1191] Remove redundant methods * [NOD-1191] Rename message constructors * [NOD-1191] Change message commands to use iota * [NOD-1191] Add missing outputs to protobuf conversion * [NOD-1191] Make block header a required field * [NOD-1191] Rename variables * [NOD-1212] Fix test names * [NOD-1191] Rename flow names * [NOD-1191] Fix infinite loop
60 lines
2.1 KiB
Go
60 lines
2.1 KiB
Go
package addressexchange
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/addressmanager"
|
|
"github.com/kaspanet/kaspad/config"
|
|
"github.com/kaspanet/kaspad/netadapter/router"
|
|
"github.com/kaspanet/kaspad/protocol/common"
|
|
peerpkg "github.com/kaspanet/kaspad/protocol/peer"
|
|
"github.com/kaspanet/kaspad/protocol/protocolerrors"
|
|
"github.com/kaspanet/kaspad/wire"
|
|
)
|
|
|
|
// ReceiveAddressesContext is the interface for the context needed for the ReceiveAddresses flow.
|
|
type ReceiveAddressesContext interface {
|
|
Config() *config.Config
|
|
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 {
|
|
|
|
if !context.AddressManager().NeedMoreAddresses() {
|
|
return nil
|
|
}
|
|
|
|
subnetworkID := peer.SubnetworkID()
|
|
msgGetAddresses := wire.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.(*wire.MsgAddresses)
|
|
if len(msgAddresses.AddrList) > addressmanager.GetAddressesMax {
|
|
return protocolerrors.Errorf(true, "address count excceeded %d", addressmanager.GetAddressesMax)
|
|
}
|
|
|
|
if msgAddresses.IncludeAllSubnetworks {
|
|
return protocolerrors.Errorf(true, "got unexpected "+
|
|
"IncludeAllSubnetworks=true in [%s] command", msgAddresses.Command())
|
|
}
|
|
if !msgAddresses.SubnetworkID.IsEqual(context.Config().SubnetworkID) && msgAddresses.SubnetworkID != nil {
|
|
return protocolerrors.Errorf(false, "only full nodes and %s subnetwork IDs "+
|
|
"are allowed in [%s] command, but got subnetwork ID %s",
|
|
context.Config().SubnetworkID, msgAddresses.Command(), msgAddresses.SubnetworkID)
|
|
}
|
|
|
|
// TODO(libp2p) Consider adding to peer known addresses set
|
|
// TODO(libp2p) Replace with real peer IP
|
|
fakeSourceAddress := new(wire.NetAddress)
|
|
context.AddressManager().AddAddresses(msgAddresses.AddrList, fakeSourceAddress, msgAddresses.SubnetworkID)
|
|
return nil
|
|
}
|