mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-22 14:56:44 +00:00
e5598c15a7
* add p2p v5 which is currently identical to v4 * set all internal imports to v5 * wip * set default version to 5 * protobuf gen for new ibd chain locator * wire for new ibd chain locator types * new ibd shared block algo -- only basic test passing * address the case where pruning points disagree, now both IBD tests pass * protobuf gen for new past diff request message * wire for new request past diff message * handle and flow for new request past diff message - logic unimplemented yet * implement ibd sync past diff of relay and selected tip * go fmt * remove unused methods * missed one err check * addressing simple comments * apply the traversal limit logic and sort headers * rename pastdiff -> anticone * apply Don't relay blocks in virtual anticone #1970 to v5 * go fmt * Fixed minor comments * Limit the number of chain negotiation restarts
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package ping
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/protocol/common"
|
|
"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 {
|
|
return err
|
|
}
|
|
pongMessage := message.(*appmessage.MsgPong)
|
|
if pongMessage.Nonce != pingMessage.Nonce {
|
|
return protocolerrors.New(true, "nonce mismatch between ping and pong")
|
|
}
|
|
flow.peer.SetPingIdle()
|
|
}
|
|
}
|