Do not queue messages if the peer isn't connected (e.g. timing out)

2 files changed, 4 insertions, 1 deletions

diff --git a/ethchain/state_manager.go b/ethchain/state_manager.go
index 8f1eb1ce5..f830f2022 100644
--- a/ethchain/state_manager.go
+++ b/ethchain/state_manager.go
@@ -199,7 +199,6 @@ func (sm *StateManager) ProcessBlock(block *Block, dontReact bool) error {
 		}
 
 		sm.Ethereum.Broadcast(ethwire.MsgBlockTy, []interface{}{block.Value().Val})
-
 	} else {
 		fmt.Println("total diff failed")
 	}
diff --git a/peer.go b/peer.go
index 932dbdc18..70759f246 100644
--- a/peer.go
+++ b/peer.go
@@ -187,6 +187,10 @@ func NewOutboundPeer(addr string, ethereum *Ethereum, caps Caps) *Peer {
 
 // Outputs any RLP encoded data to the peer
 func (p *Peer) QueueMessage(msg *ethwire.Msg) {
+	if atomic.LoadInt32(&p.connected) != 1 {
+		return
+	}
+
 	p.outputQueue <- msg
 }