mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-03-30 15:08:33 +00:00
9a81b1328a
* Add the Address of node to whom connected in log of send/receiveVersion * Don't call functions before LogAndMeasureExecutionTime Co-authored-by: Ori Newman <orinewman1@gmail.com>
86 lines
2.5 KiB
Go
86 lines
2.5 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/infrastructure/logger"
|
|
"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 {
|
|
onEnd := logger.LogAndMeasureExecutionTime(log, "sendVersionFlow.start")
|
|
defer onEnd()
|
|
|
|
log.Debugf("Starting sendVersionFlow with %s", flow.peer.Address())
|
|
|
|
// Version message.
|
|
localAddress := flow.AddressManager().BestLocalAddress(flow.peer.Connection().NetAddress())
|
|
subnetworkID := flow.Config().SubnetworkID
|
|
msg := appmessage.NewMsgVersion(localAddress, flow.NetAdapter().ID(),
|
|
flow.Config().ActiveNetParams.Name, 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
|
|
log.Debugf("Waiting for verack")
|
|
_, err = flow.incomingRoute.DequeueWithTimeout(common.DefaultTimeout)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Debugf("Got verack")
|
|
return nil
|
|
}
|