mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-09-14 05:20:11 +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
54 lines
1.7 KiB
Go
54 lines
1.7 KiB
Go
package addressexchange
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/addressmanager"
|
|
"github.com/kaspanet/kaspad/netadapter/router"
|
|
"github.com/kaspanet/kaspad/wire"
|
|
"math/rand"
|
|
)
|
|
|
|
// SendAddressesContext is the interface for the context needed for the SendAddresses flow.
|
|
type SendAddressesContext interface {
|
|
AddressManager() *addressmanager.AddressManager
|
|
}
|
|
|
|
// SendAddresses sends addresses to a peer that requests it.
|
|
func SendAddresses(context SendAddressesContext, incomingRoute *router.Route, outgoingRoute *router.Route) error {
|
|
|
|
message, err := incomingRoute.Dequeue()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msgGetAddresses := message.(*wire.MsgRequestAddresses)
|
|
addresses := context.AddressManager().AddressCache(msgGetAddresses.IncludeAllSubnetworks,
|
|
msgGetAddresses.SubnetworkID)
|
|
msgAddresses := wire.NewMsgAddresses(msgGetAddresses.IncludeAllSubnetworks, msgGetAddresses.SubnetworkID)
|
|
err = msgAddresses.AddAddresses(shuffleAddresses(addresses)...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return outgoingRoute.Enqueue(msgAddresses)
|
|
}
|
|
|
|
// shuffleAddresses randomizes the given addresses sent if there are more than the maximum allowed in one message.
|
|
func shuffleAddresses(addresses []*wire.NetAddress) []*wire.NetAddress {
|
|
addressCount := len(addresses)
|
|
|
|
if addressCount < wire.MaxAddressesPerMsg {
|
|
return addresses
|
|
}
|
|
|
|
shuffleAddresses := make([]*wire.NetAddress, addressCount)
|
|
copy(shuffleAddresses, addresses)
|
|
|
|
rand.Shuffle(addressCount, func(i, j int) {
|
|
shuffleAddresses[i], shuffleAddresses[j] = shuffleAddresses[j], shuffleAddresses[i]
|
|
})
|
|
|
|
// Truncate it to the maximum size.
|
|
shuffleAddresses = shuffleAddresses[:wire.MaxAddressesPerMsg]
|
|
return shuffleAddresses
|
|
}
|