From 52b664b0aec24bffc040da06bbe8be1d3e7e4ab8 Mon Sep 17 00:00:00 2001
From: Maran <maran.hidskes@gmail.com>
Date: Mon, 12 May 2014 16:30:21 +0200
Subject: [PATCH] Removed peers from peerlist as soon as they disconnect. Might
 fix #13

We used to wait for the reaping timer to clean up the peerlist, not any longer
---
 peer.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/peer.go b/peer.go
index 359e83566..932dbdc18 100644
--- a/peer.go
+++ b/peer.go
@@ -2,6 +2,7 @@ package eth
 
 import (
 	"bytes"
+	"container/list"
 	"fmt"
 	"github.com/ethereum/eth-go/ethchain"
 	"github.com/ethereum/eth-go/ethutil"
@@ -515,6 +516,15 @@ func (p *Peer) Stop() {
 		p.writeMessage(ethwire.NewMessage(ethwire.MsgDiscTy, ""))
 		p.conn.Close()
 	}
+
+	// Pre-emptively remove the peer; don't wait for reaping. We already know it's dead if we are here
+	p.ethereum.peerMut.Lock()
+	defer p.ethereum.peerMut.Unlock()
+	eachPeer(p.ethereum.peers, func(peer *Peer, e *list.Element) {
+		if peer == p {
+			p.ethereum.peers.Remove(e)
+		}
+	})
 }
 
 func (p *Peer) pushHandshake() error {