Reset peer during download on disc

7ba9fe4d · obscuren · 25d16f35 · 7ba9fe4d
Commit 7ba9fe4d authored Dec 24, 2014 by obscuren
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

peer.go peer.go +6 -0

No files found.
--- 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: