mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-21 14:26:45 +00:00
d2379608ad
* Implement upgrade mechanism for p2p * Remove dependencies from flowcontext to v3 * Add p2p v4 * Add Ready flow * Remove copy paste code of v3 * Register SendAddresses flow at the top level * Add option to set protocol version from CLI and add TestAddressExchangeV3V4 * Send ready message on minimal net adapter * Rename defaultMaxProtocolVersion->maxAcceptableProtocolVersion
57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package ready
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/protocol/common"
|
|
"sync/atomic"
|
|
|
|
peerpkg "github.com/kaspanet/kaspad/app/protocol/peer"
|
|
"github.com/kaspanet/kaspad/app/protocol/protocolerrors"
|
|
routerpkg "github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// HandleReady notify the other peer that peer is ready for messages, and wait for the other peer
|
|
// to send a ready message before start running the flows.
|
|
func HandleReady(incomingRoute *routerpkg.Route, outgoingRoute *routerpkg.Route,
|
|
peer *peerpkg.Peer,
|
|
) error {
|
|
|
|
log.Debugf("Sending ready message to %s", peer)
|
|
|
|
isStopping := uint32(0)
|
|
err := outgoingRoute.Enqueue(appmessage.NewMsgReady())
|
|
if err != nil {
|
|
return handleError(err, "HandleReady", &isStopping)
|
|
}
|
|
|
|
_, err = incomingRoute.DequeueWithTimeout(common.DefaultTimeout)
|
|
if err != nil {
|
|
return handleError(err, "HandleReady", &isStopping)
|
|
}
|
|
|
|
log.Debugf("Got ready message from %s", peer)
|
|
|
|
return nil
|
|
}
|
|
|
|
// Ready is different from other flows, since in it should forward router.ErrRouteClosed to errChan
|
|
// Therefore we implement a separate handleError for 'ready'
|
|
func handleError(err error, flowName string, isStopping *uint32) error {
|
|
if errors.Is(err, routerpkg.ErrRouteClosed) {
|
|
if atomic.AddUint32(isStopping, 1) == 1 {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if protocolErr := (protocolerrors.ProtocolError{}); errors.As(err, &protocolErr) {
|
|
log.Errorf("Ready protocol error from %s: %s", flowName, err)
|
|
if atomic.AddUint32(isStopping, 1) == 1 {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
panic(err)
|
|
}
|