mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-22 06:46:46 +00:00
e5598c15a7
* add p2p v5 which is currently identical to v4 * set all internal imports to v5 * wip * set default version to 5 * protobuf gen for new ibd chain locator * wire for new ibd chain locator types * new ibd shared block algo -- only basic test passing * address the case where pruning points disagree, now both IBD tests pass * protobuf gen for new past diff request message * wire for new request past diff message * handle and flow for new request past diff message - logic unimplemented yet * implement ibd sync past diff of relay and selected tip * go fmt * remove unused methods * missed one err check * addressing simple comments * apply the traversal limit logic and sort headers * rename pastdiff -> anticone * apply Don't relay blocks in virtual anticone #1970 to v5 * go fmt * Fixed minor comments * Limit the number of chain negotiation restarts
53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
package addressexchange
|
|
|
|
import (
|
|
"math/rand"
|
|
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/addressmanager"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
)
|
|
|
|
// 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 {
|
|
for {
|
|
_, err := incomingRoute.Dequeue()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
addresses := context.AddressManager().Addresses()
|
|
msgAddresses := appmessage.NewMsgAddresses(shuffleAddresses(addresses))
|
|
|
|
err = outgoingRoute.Enqueue(msgAddresses)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
// shuffleAddresses randomizes the given addresses sent if there are more than the maximum allowed in one message.
|
|
func shuffleAddresses(addresses []*appmessage.NetAddress) []*appmessage.NetAddress {
|
|
addressCount := len(addresses)
|
|
|
|
if addressCount < appmessage.MaxAddressesPerMsg {
|
|
return addresses
|
|
}
|
|
|
|
shuffleAddresses := make([]*appmessage.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[:appmessage.MaxAddressesPerMsg]
|
|
return shuffleAddresses
|
|
}
|