X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Fnet%2Fpterodactylus%2Futil%2Ffcp%2FFcpConnection.java;h=12d52af112c0cfe360984762cf256a2bc98e1655;hb=561d15670607ad2fae532e71dfc25b35848f0775;hp=97d3b6788c77c97b967655e63061ac9be57722cc;hpb=383efb2dcafa2ff49eb3192a5b85eea8263553eb;p=jSite2.git

diff --git a/src/net/pterodactylus/util/fcp/FcpConnection.java b/src/net/pterodactylus/util/fcp/FcpConnection.java
index 97d3b67..12d52af 100644
--- a/src/net/pterodactylus/util/fcp/FcpConnection.java
+++ b/src/net/pterodactylus/util/fcp/FcpConnection.java
@@ -656,18 +656,11 @@ public class FcpConnection {
 		} else if ("IdentifierCollision".equals(messageName)) {
 			fireReceivedIdentifierCollision(new IdentifierCollision(fcpMessage));
 		} else if ("AllData".equals(messageName)) {
-			long dataLength;
-			try {
-				dataLength = Long.valueOf(fcpMessage.getField("DataLength"));
-			} catch (NumberFormatException nfe1) {
-				dataLength = -1;
-			}
-			LimitedInputStream payloadInputStream = new LimitedInputStream(remoteInputStream, dataLength);
+			LimitedInputStream payloadInputStream = getInputStream(FcpUtils.safeParseLong(fcpMessage.getField("DataLength")));
 			fireReceivedAllData(new AllData(fcpMessage, payloadInputStream));
 			try {
 				payloadInputStream.consume();
 			} catch (IOException ioe1) {
-				/* FIXME - what now? */
 				/* well, ignore. when the connection handler fails, all fails. */
 			}
 		} else if ("EndListPeerNotes".equals(messageName)) {
@@ -732,4 +725,18 @@ public class FcpConnection {
 		incomingMessageStatistics.put(name, oldValue + 1);
 	}
 
+	/**
+	 * Returns a limited input stream from the node’s input stream.
+	 * 
+	 * @param dataLength
+	 *            The length of the stream
+	 * @return The limited input stream
+	 */
+	private LimitedInputStream getInputStream(long dataLength) {
+		if (dataLength <= 0) {
+			return new LimitedInputStream(null, 0);
+		}
+		return new LimitedInputStream(remoteInputStream, dataLength);
+	}
+
 }