mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-16 19:16:39 +00:00
23c1ea6c31
* [NOD-1525] Implement headers first IBD * [NOD-1525] Fix proto translators * [NOD-1525] Register missing flows * [NOD-1525] Rename SyncStateNormal->SyncStateRelay, simplifiy IBD peer selection code and get rid of panic in FinishIBD * [NOD-1525] Remove redundant methods from interface
50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
package ibd
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/protocol/protocolerrors"
|
|
"github.com/kaspanet/kaspad/domain"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// HandleIBDBlockRequestsContext is the interface for the context needed for the HandleIBDBlockRequests flow.
|
|
type HandleIBDBlockRequestsContext interface {
|
|
Domain() domain.Domain
|
|
}
|
|
|
|
// HandleIBDBlockRequests listens to appmessage.MsgRequestRelayBlocks messages and sends
|
|
// their corresponding blocks to the requesting peer.
|
|
func HandleIBDBlockRequests(context HandleIBDBlockRequestsContext, incomingRoute *router.Route,
|
|
outgoingRoute *router.Route) error {
|
|
|
|
for {
|
|
message, err := incomingRoute.Dequeue()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
msgRequestIBDBlocks := message.(*appmessage.MsgRequestIBDBlocks)
|
|
for _, hash := range msgRequestIBDBlocks.Hashes {
|
|
// Fetch the block from the database.
|
|
blockInfo, err := context.Domain().Consensus().GetBlockInfo(hash)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !blockInfo.Exists {
|
|
return protocolerrors.Errorf(true, "block %s not found", hash)
|
|
}
|
|
block, err := context.Domain().Consensus().GetBlock(hash)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "unable to fetch requested block hash %s", hash)
|
|
}
|
|
|
|
// TODO (Partial nodes): Convert block to partial block if needed
|
|
|
|
err = outgoingRoute.Enqueue(appmessage.DomainBlockToMsgBlock(block))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|