Only download as much bytes as advertised.

[xudocci.git] / src / main / java / net / pterodactylus / irc / DccReceiver.java

diff --git a/src/main/java/net/pterodactylus/irc/DccReceiver.java b/src/main/java/net/pterodactylus/irc/DccReceiver.java
index 038f66a..5c5eac4 100644 (file)
--- a/src/main/java/net/pterodactylus/irc/DccReceiver.java
+++ b/src/main/java/net/pterodactylus/irc/DccReceiver.java
@@ -150,7 +150,7 @@ public class DccReceiver extends AbstractExecutionThreadService {
                        InputStream socketInputStream = socket.getInputStream();
                        inputStream = new BandwidthCountingInputStream(socketInputStream, 5, TimeUnit.SECONDS);
                        byte[] buffer = new byte[65536];
-                       while (isRunning()) {
+                       while (isRunning() && ((size == -1) || (progress < size))) {
                                int r = inputStream.read(buffer);
                                if (r == -1) {
                                        /* yay, eof! */