mirror of
https://github.com/kaspanet/kaspad.git
synced 2026-03-22 16:13:45 +00:00
23a3594c18
* [NOD-1233] Remove HandleNewBlockOld. * [NOD-1233] Make ErrRouteClosed not a protocol error. * [NOD-1233] Fix ambiguous comments. * [NOD-1233] Remove a no-longer-relevant comment. * [NOD-1233] Remove some of the TODOs. * [NOD-1233] Replace fakeSourceAddress with a real sourceAddress. * [NOD-1233] Remove a no-longer-relevant TODO. * [NOD-1233] Remove TODO from handleGetNetTotals. * [NOD-1233] Remove a no-longer-relevant TODO. * [NOD-1233] Disconnect if connected to wrong partial/full type. * [NOD-1233] Get rid of mempool tags. * [NOD-1233] Remove TODOs. * [NOD-1233] Simplify a test. * [NOD-1190] Remove getNetTotals.
102 lines
3.4 KiB
Go
102 lines
3.4 KiB
Go
package handshake
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/domainmessage"
|
|
"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"
|
|
)
|
|
|
|
var (
|
|
// allowSelfConnections is only used to allow the tests to bypass the self
|
|
// connection detecting and disconnect logic since they intentionally
|
|
// do so for testing purposes.
|
|
allowSelfConnections bool
|
|
|
|
// minAcceptableProtocolVersion is the lowest protocol version that a
|
|
// connected peer may support.
|
|
minAcceptableProtocolVersion = domainmessage.ProtocolVersion
|
|
)
|
|
|
|
type receiveVersionFlow struct {
|
|
HandleHandshakeContext
|
|
incomingRoute, outgoingRoute *router.Route
|
|
peer *peerpkg.Peer
|
|
}
|
|
|
|
// ReceiveVersion waits for the peer to send a version message, sends a
|
|
// verack in response, and updates its info accordingly.
|
|
func ReceiveVersion(context HandleHandshakeContext, incomingRoute *router.Route, outgoingRoute *router.Route,
|
|
peer *peerpkg.Peer) (*domainmessage.NetAddress, error) {
|
|
|
|
flow := &receiveVersionFlow{
|
|
HandleHandshakeContext: context,
|
|
incomingRoute: incomingRoute,
|
|
outgoingRoute: outgoingRoute,
|
|
peer: peer,
|
|
}
|
|
|
|
return flow.start()
|
|
}
|
|
|
|
func (flow *receiveVersionFlow) start() (*domainmessage.NetAddress, error) {
|
|
message, err := flow.incomingRoute.DequeueWithTimeout(common.DefaultTimeout)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
msgVersion, ok := message.(*domainmessage.MsgVersion)
|
|
if !ok {
|
|
return nil, protocolerrors.New(true, "a version message must precede all others")
|
|
}
|
|
|
|
if !allowSelfConnections && flow.NetAdapter().ID().IsEqual(msgVersion.ID) {
|
|
return nil, protocolerrors.New(true, "connected to self")
|
|
}
|
|
|
|
// Disconnect and ban peers from a different network
|
|
if msgVersion.Network != flow.Config().ActiveNetParams.Name {
|
|
return nil, protocolerrors.Errorf(true, "wrong network")
|
|
}
|
|
|
|
// Notify and disconnect clients that have a protocol version that is
|
|
// too old.
|
|
//
|
|
// NOTE: If minAcceptableProtocolVersion is raised to be higher than
|
|
// domainmessage.RejectVersion, this should send a reject packet before
|
|
// disconnecting.
|
|
if msgVersion.ProtocolVersion < minAcceptableProtocolVersion {
|
|
return nil, protocolerrors.Errorf(false, "protocol version must be %d or greater",
|
|
minAcceptableProtocolVersion)
|
|
}
|
|
|
|
// Disconnect from partial nodes in networks that don't allow them
|
|
if !flow.DAG().Params.EnableNonNativeSubnetworks && msgVersion.SubnetworkID != nil {
|
|
return nil, protocolerrors.New(true, "partial nodes are not allowed")
|
|
}
|
|
|
|
// Disconnect if:
|
|
// - we are a full node and the outbound connection we've initiated is a partial node
|
|
// - the remote node is partial and our subnetwork doesn't match their subnetwork
|
|
localSubnetworkID := flow.Config().SubnetworkID
|
|
isLocalNodeFull := localSubnetworkID == nil
|
|
isRemoteNodeFull := msgVersion.SubnetworkID == nil
|
|
isOutbound := flow.peer.Connection().IsOutbound()
|
|
if (isLocalNodeFull && !isRemoteNodeFull && isOutbound) ||
|
|
(!isLocalNodeFull && !isRemoteNodeFull && !msgVersion.SubnetworkID.IsEqual(localSubnetworkID)) {
|
|
|
|
return nil, protocolerrors.New(false, "incompatible subnetworks")
|
|
}
|
|
|
|
flow.peer.UpdateFieldsFromMsgVersion(msgVersion)
|
|
err = flow.outgoingRoute.Enqueue(domainmessage.NewMsgVerAck())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
flow.peer.Connection().SetID(msgVersion.ID)
|
|
|
|
return msgVersion.Address, nil
|
|
}
|