mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-09 23:56:42 +00:00
53cccd405f
* [NOD-1220] Add network name to the version message. * [NOD-1220] Ban peers from the wrong network. * [NOD-1220] Add the network parameter to protowire. * [NOD-1220] Add "kaspa-" to network names.
103 lines
3.5 KiB
Go
103 lines
3.5 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 {
|
|
//TODO(libp2p) create error type for disconnect but don't ban
|
|
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")
|
|
}
|
|
|
|
// TODO(libp2p)
|
|
//// 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 := config.ActiveConfig().SubnetworkID
|
|
//isLocalNodeFull := localSubnetworkID == nil
|
|
//isRemoteNodeFull := msgVersion.SubnetworkID == nil
|
|
//if (isLocalNodeFull && !isRemoteNodeFull && !connection.IsInbound()) ||
|
|
// (!isLocalNodeFull && !isRemoteNodeFull && !msgVersion.SubnetworkID.IsEqual(localSubnetworkID)) {
|
|
//
|
|
// return nil, false, errors.New("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
|
|
}
|