mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-10-14 00:59:33 +00:00
05db135d23
* [NOD-1124] Move Router to the router package. * [NOD-1124] Implement SetOnRouteCapacityReachedHandler. * [NOD-1124] Use Routes instead of bare channels. * [NOD-1124] Fix merge errors. * [NOD-1124] Connect the Router to the Connection. * [NOD-1124] Fix merge errors. * [NOD-1124] Move some variables around. * [NOD-1124] Fix unreachable code. * [NOD-1124] Fix a variable name. * [NOD-1124] Rename AddRoute to AddIncomingRoute. * [NOD-1124] Rename SetRouter to Start. * [NOD-1124] Make AddIncomingRoute create a Route by itself. * [NOD-1124] Replace IncomingRoute with EnqueueIncomingMessage. * [NOD-1124] Make Enqueue and Dequeue return isOpen instead of err. * [NOD-1124] Remove writeDuringDisconnectLock. * [NOD-1124] In sendLoop, move outgoingRoute to outside the loop. * [NOD-1124] Start the connection loops only when Start is called. * [NOD-1124] Replace OnIDReceivedHandler with AssociateRouterID. * [NOD-1124] Add isOpen to Enqueue and Dequeue. * [NOD-1124] Protect errChan from writing during disconnect.
53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
package handlerelayblockrequests
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/blockdag"
|
|
"github.com/kaspanet/kaspad/netadapter/router"
|
|
peerpkg "github.com/kaspanet/kaspad/protocol/peer"
|
|
"github.com/kaspanet/kaspad/wire"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// HandleRelayBlockRequests listens to wire.MsgGetRelayBlocks messages and sends
|
|
// their corresponding blocks to the requesting peer.
|
|
func HandleRelayBlockRequests(incomingRoute *router.Route, outgoingRoute *router.Route,
|
|
peer *peerpkg.Peer, dag *blockdag.BlockDAG) error {
|
|
|
|
for {
|
|
message, isOpen := incomingRoute.Dequeue()
|
|
if !isOpen {
|
|
return nil
|
|
}
|
|
getRelayBlocksMessage := message.(*wire.MsgGetRelayBlocks)
|
|
for _, hash := range getRelayBlocksMessage.Hashes {
|
|
// Fetch the block from the database.
|
|
block, err := dag.BlockByHash(hash)
|
|
if blockdag.IsNotInDAGErr(err) {
|
|
return errors.Errorf("block %s not found", hash)
|
|
} else if err != nil {
|
|
panic(errors.Wrapf(err, "unable to fetch requested block hash %s", hash))
|
|
}
|
|
msgBlock := block.MsgBlock()
|
|
|
|
// If we are a full node and the peer is a partial node, we must convert
|
|
// the block to a partial block.
|
|
nodeSubnetworkID := dag.SubnetworkID()
|
|
peerSubnetworkID, err := peer.SubnetworkID()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
isNodeFull := nodeSubnetworkID == nil
|
|
isPeerFull := peerSubnetworkID == nil
|
|
if isNodeFull && !isPeerFull {
|
|
msgBlock.ConvertToPartial(peerSubnetworkID)
|
|
}
|
|
|
|
isOpen = outgoingRoute.Enqueue(msgBlock)
|
|
if !isOpen {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
}
|