751d2711d4
- refactored OnBlockBlockMode to use blockPool - syncmgr now checks for future blocks, instead of looking for a FutureBlockErr from the Chain. This makes it so that syncmgr does not depend on the Chain package. - removed GetBestBlockHash function from config(unused) - refactored all tests in the syncmgr package
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package syncmgr
|
|
|
|
import (
|
|
"github.com/CityOfZion/neo-go/pkg/wire/payload"
|
|
)
|
|
|
|
func (s *Syncmgr) normalModeOnHeaders(peer SyncPeer, hdrs []*payload.BlockBase) error {
|
|
// If in normal mode, first process the headers
|
|
err := s.cfg.ProcessHeaders(hdrs)
|
|
if err != nil {
|
|
// If something went wrong with processing the headers
|
|
// Ask another peer for the headers.
|
|
//XXX: Increment banscore for this peer
|
|
return s.cfg.FetchHeadersAgain(hdrs[0].Hash)
|
|
}
|
|
|
|
lenHeaders := len(hdrs)
|
|
firstHash := hdrs[0].Hash
|
|
firstHdrIndex := hdrs[0].Index
|
|
lastHash := hdrs[lenHeaders-1].Hash
|
|
|
|
// Update syncmgr latest header
|
|
s.headerHash = lastHash
|
|
|
|
// If there are 2k headers, then ask for more headers and switch back to headers mode.
|
|
if lenHeaders == 2000 {
|
|
s.syncmode = headersMode
|
|
return s.cfg.RequestHeaders(lastHash)
|
|
}
|
|
|
|
// Ask for the corresponding block iff there is < 2k headers
|
|
// then switch to blocksMode
|
|
// Bounds state that len > 1 && len!= 2000 & maxHeadersInMessage == 2000
|
|
// This means that we have less than 2k headers
|
|
s.syncmode = blockMode
|
|
return s.cfg.RequestBlock(firstHash, firstHdrIndex)
|
|
}
|
|
|
|
// normalModeOnBlock is called when the sync manager is normal mode
|
|
// and receives a block.
|
|
func (s *Syncmgr) normalModeOnBlock(peer SyncPeer, block payload.Block) error {
|
|
// stop the timer that periodically asks for blocks
|
|
s.timer.Stop()
|
|
|
|
// process block
|
|
err := s.processBlock(block)
|
|
if err != nil {
|
|
s.timer.Reset(blockTimer)
|
|
return s.cfg.FetchBlockAgain(block.Hash)
|
|
}
|
|
|
|
diff := peer.Height() - block.Index
|
|
if diff > trailingHeight {
|
|
s.syncmode = headersMode
|
|
return s.cfg.RequestHeaders(block.Hash)
|
|
}
|
|
|
|
s.timer.Reset(blockTimer)
|
|
return nil
|
|
}
|