Reset peer during download on disc

10 years ago · 7ba9fe4d5d
parent 25d16f358a
commit 7ba9fe4d5d
1 changed files with 6 additions and 0 deletions
--- 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: