Abstract p2p security module

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

diff --git a/netsync/block_fetcher.go b/netsync/block_fetcher.go
index 777d1d5..3e59634 100644 (file)
--- a/netsync/block_fetcher.go
+++ b/netsync/block_fetcher.go
@@ -4,6 +4,7 @@ import (
        log "github.com/sirupsen/logrus"
        "gopkg.in/karalabe/cookiejar.v2/collections/prque"
 
+       "github.com/bytom/p2p/security"
        "github.com/bytom/protocol/bc"
 )
 
@@ -79,7 +80,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
        }