cb21c66316
* [syncmgr] - Add blockmode, normal mode, headermode - Add config file - Add test files - removed RequestBlocks and RequestHeaders from peers, as we will use the peermanager for this - OnHeaders and OnBlock in syncmgr, now return errors - refactored all tests to use a convenience method to return a syncmgr and testHelper
59 lines
1.6 KiB
Go
59 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
|
|
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)
|
|
}
|
|
|
|
// 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.cfg.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
|
|
}
|