Fixed a BufferUnderflowException in message parsing when PeerExchange…

… is closing
mpetazzoni · Jul 1, 2016 · fce1fc7 · fce1fc7
1 parent 7acb2d0
commit fce1fc7
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/core/src/main/java/com/turn/ttorrent/client/peer/PeerExchange.java b/core/src/main/java/com/turn/ttorrent/client/peer/PeerExchange.java
@@ -350,7 +350,14 @@ public void run() {
 					}
 
 					buffer.rewind();
-
+
+					if (stop) {
+						// The buffer may contain the type from the last message
+						// if we were stopped before reading the payload and cause
+						// BufferUnderflowException in parsing.
+						break;
+					}
+
 					try {
 						PeerMessage message = PeerMessage.parse(buffer, torrent);
 						logger.trace("Received {} from {}", message, peer);