mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-21 14:26:45 +00:00
2ab8065142
* Improve output of non-critical protocol errors to avoid user panic * Add log messages at the end of IBD with headers proof * Found a case where this was falsely triggered due to the wrong equality test
78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package ping
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/protocol/common"
|
|
"github.com/kaspanet/kaspad/app/protocol/flowcontext"
|
|
"github.com/pkg/errors"
|
|
"time"
|
|
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
peerpkg "github.com/kaspanet/kaspad/app/protocol/peer"
|
|
"github.com/kaspanet/kaspad/app/protocol/protocolerrors"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
"github.com/kaspanet/kaspad/util/random"
|
|
)
|
|
|
|
// SendPingsContext is the interface for the context needed for the SendPings flow.
|
|
type SendPingsContext interface {
|
|
ShutdownChan() <-chan struct{}
|
|
}
|
|
|
|
type sendPingsFlow struct {
|
|
SendPingsContext
|
|
incomingRoute, outgoingRoute *router.Route
|
|
peer *peerpkg.Peer
|
|
}
|
|
|
|
// SendPings starts sending MsgPings every pingInterval seconds to the
|
|
// given peer.
|
|
// This function assumes that incomingRoute will only return MsgPong.
|
|
func SendPings(context SendPingsContext, incomingRoute *router.Route, outgoingRoute *router.Route, peer *peerpkg.Peer) error {
|
|
flow := &sendPingsFlow{
|
|
SendPingsContext: context,
|
|
incomingRoute: incomingRoute,
|
|
outgoingRoute: outgoingRoute,
|
|
peer: peer,
|
|
}
|
|
return flow.start()
|
|
}
|
|
|
|
func (flow *sendPingsFlow) start() error {
|
|
const pingInterval = 2 * time.Minute
|
|
ticker := time.NewTicker(pingInterval)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-flow.ShutdownChan():
|
|
return nil
|
|
case <-ticker.C:
|
|
}
|
|
|
|
nonce, err := random.Uint64()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
flow.peer.SetPingPending(nonce)
|
|
|
|
pingMessage := appmessage.NewMsgPing(nonce)
|
|
err = flow.outgoingRoute.Enqueue(pingMessage)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
message, err := flow.incomingRoute.DequeueWithTimeout(common.DefaultTimeout)
|
|
if err != nil {
|
|
if errors.Is(err, router.ErrTimeout) {
|
|
return errors.Wrapf(flowcontext.ErrPingTimeout, err.Error())
|
|
}
|
|
return err
|
|
}
|
|
pongMessage := message.(*appmessage.MsgPong)
|
|
if pongMessage.Nonce != pingMessage.Nonce {
|
|
return protocolerrors.New(true, "nonce mismatch between ping and pong")
|
|
}
|
|
flow.peer.SetPingIdle()
|
|
}
|
|
}
|