Abort message queueing if peer was disconnected

yuhuashangedward · Jul 5, 2018 · d0365cb · d0365cb
1 parent 7503803
commit d0365cb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ethereumj-core/src/main/java/org/ethereum/net/MessageQueue.java b/ethereumj-core/src/main/java/org/ethereum/net/MessageQueue.java
@@ -97,6 +97,11 @@ public void setChannel(Channel channel) {
     }
 
     public void sendMessage(Message msg) {
+        if (channel.isDisconnected()) {
+            logger.warn("{}: attempt to send [{}] message after disconnect", channel, msg.getCommand().name());
+            return;
+        }
+
         if (msg instanceof PingMessage) {
             if (hasPing) return;
             hasPing = true;