Reset peer during download on disc

1 files changed, 6 insertions, 0 deletions

diff --git a/peer.go b/peer.go
index 13f0239d4..ceb49a5af 100644
--- a/peer.go
+++ b/peer.go
@@ -412,6 +412,12 @@ func (p *Peer) HandleInbound() {
 				//}
 
 			case wire.MsgDiscTy:
+				blockPool := p.ethereum.blockPool
+				if blockPool.peer == p {
+					blockPool.peer = nil
+					blockPool.td = ethutil.Big0
+				}
+
 				p.Stop()
 				peerlogger.Infoln("Disconnect peer: ", DiscReason(msg.Data.Get(0).Uint()))
 			case wire.MsgPingTy: