mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-23 22:42:32 +00:00
7a7821e1c8
* [NOD-1313] Refactor AddressManager (#918) * [NOD-1313] Refactor AddressManager. * [NOD-1313]Remove old tests.Minor improvements,fixes. * [NOD-1313] After merge fixes. Fix import cycle. * [NOD-1313] Integration tests fixes. * [NOD-1313] Allocate new slice for the returned key. * [NOD-1313] AddressManager improvements and fixes. * Move local and banned addresses to separate lists. * Move AddressManager config to the separate file. * Add LocalAddressManager. * Remove redundant KnownAddress structure. * Restore local addresses functionality. * Call initListeners from the LocalAddressManager. * AddressManager minor improvements and fixes. * [NOD-1313] Minor fixes. * [NOD-1313] Implement HandleGetPeerAddresses. Refactoring. * [NOD-1313] After-merge fixes. * [NOD-1313] Minor improvements. * AddressManager: added BannedAddresses() method. * AddressManager: HandleGetPeerAddresses() add banned addresses separately. * AddressManager: remove addressEntry redundant struct. * ConnectionManager: checkOutgoingConnections() minor improvements and fixes. * Minor refactoring. * Minor fixes. * [NOD-1313] GetPeerAddresses RPC message update * GetPeerAddresses RPC: add BannedAddresses in the separate field. * Update protobuf. * [NOD-1534] Update messages.pb.go Co-authored-by: Kirill <gammerxpower@gmail.com>
85 lines
2.6 KiB
Go
85 lines
2.6 KiB
Go
package handshake
|
|
|
|
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/domain/consensus/utils/consensusserialization"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
"github.com/kaspanet/kaspad/version"
|
|
)
|
|
|
|
var (
|
|
// userAgentName is the user agent name and is used to help identify
|
|
// ourselves to other kaspa peers.
|
|
userAgentName = "kaspad"
|
|
|
|
// userAgentVersion is the user agent version and is used to help
|
|
// identify ourselves to other kaspa peers.
|
|
userAgentVersion = version.Version()
|
|
|
|
// defaultServices describes the default services that are supported by
|
|
// the server.
|
|
defaultServices = appmessage.DefaultServices
|
|
|
|
// defaultRequiredServices describes the default services that are
|
|
// required to be supported by outbound peers.
|
|
defaultRequiredServices = appmessage.SFNodeNetwork
|
|
)
|
|
|
|
type sendVersionFlow struct {
|
|
HandleHandshakeContext
|
|
|
|
incomingRoute, outgoingRoute *router.Route
|
|
peer *peerpkg.Peer
|
|
}
|
|
|
|
// SendVersion sends a version to a peer and waits for verack.
|
|
func SendVersion(context HandleHandshakeContext, incomingRoute *router.Route,
|
|
outgoingRoute *router.Route, peer *peerpkg.Peer) error {
|
|
|
|
flow := &sendVersionFlow{
|
|
HandleHandshakeContext: context,
|
|
incomingRoute: incomingRoute,
|
|
outgoingRoute: outgoingRoute,
|
|
peer: peer,
|
|
}
|
|
return flow.start()
|
|
}
|
|
|
|
func (flow *sendVersionFlow) start() error {
|
|
virtualSelectedParent, err := flow.Domain().Consensus().GetVirtualSelectedParent()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
selectedTipHash := consensusserialization.BlockHash(virtualSelectedParent)
|
|
subnetworkID := flow.Config().SubnetworkID
|
|
|
|
// Version message.
|
|
localAddress := flow.AddressManager().BestLocalAddress(flow.peer.Connection().NetAddress())
|
|
msg := appmessage.NewMsgVersion(localAddress, flow.NetAdapter().ID(),
|
|
flow.Config().ActiveNetParams.Name, selectedTipHash, subnetworkID)
|
|
msg.AddUserAgent(userAgentName, userAgentVersion, flow.Config().UserAgentComments...)
|
|
|
|
// Advertise the services flag
|
|
msg.Services = defaultServices
|
|
|
|
// Advertise our max supported protocol version.
|
|
msg.ProtocolVersion = appmessage.ProtocolVersion
|
|
|
|
// Advertise if inv messages for transactions are desired.
|
|
msg.DisableRelayTx = flow.Config().BlocksOnly
|
|
|
|
err = flow.outgoingRoute.Enqueue(msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Wait for verack
|
|
_, err = flow.incomingRoute.DequeueWithTimeout(common.DefaultTimeout)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|