log into file (#357)

[bytom/vapor.git] / netsync / consensusmgr / block_fetcher.go

diff --git a/netsync/consensusmgr/block_fetcher.go b/netsync/consensusmgr/block_fetcher.go
index 6278c61..3c92b38 100644 (file)
--- a/netsync/consensusmgr/block_fetcher.go
+++ b/netsync/consensusmgr/block_fetcher.go
@@ -5,6 +5,7 @@ import (
        "gopkg.in/karalabe/cookiejar.v2/collections/prque"
 
        "github.com/vapor/netsync/peers"
+       "github.com/vapor/p2p/security"
        "github.com/vapor/protocol/bc"
 )
 
@@ -40,10 +41,9 @@ func newBlockFetcher(chain Chain, peers *peers.PeerSet) *blockFetcher {
 
 func (f *blockFetcher) blockProcessor() {
        for {
-               height := f.chain.BestBlockHeight()
                for !f.queue.Empty() {
                        msg := f.queue.PopItem().(*blockMsg)
-                       if msg.block.Height > height+1 {
+                       if msg.block.Height > f.chain.BestBlockHeight()+1 {
                                f.queue.Push(msg, -float32(msg.block.Height))
                                break
                        }
@@ -81,7 +81,7 @@ func (f *blockFetcher) insert(msg *blockMsg) {
                        return
                }
 
-               f.peers.AddBanScore(msg.peerID, 20, 0, err.Error())
+               f.peers.ProcessIllegal(msg.peerID, security.LevelMsgIllegal, err.Error())
                return
        }
 
@@ -89,8 +89,6 @@ func (f *blockFetcher) insert(msg *blockMsg) {
                return
        }
 
-       hash := msg.block.Hash()
-       f.peers.SetStatus(msg.peerID, msg.block.Height, &hash)
        proposeMsg, err := NewBlockProposeMsg(msg.block)
        if err != nil {
                logrus.WithFields(logrus.Fields{"module": logModule, "err": err}).Error("failed on create BlockProposeMsg")