From 79090bc2d1b11f7d21cc450b5deda4d55d7e1382 Mon Sep 17 00:00:00 2001
From: Nils Schwabe
Date: Fri, 11 Jul 2014 15:49:12 +0200
Subject: Move checking of incoming block into while loop (where it belongs)

---
 .../imagemaster/serverconnection/ConnectionHandler.java   | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

(limited to 'src')

diff --git a/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java b/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java
index c273239..f89e028 100644
--- a/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java
+++ b/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java
@@ -178,16 +178,17 @@ public class ConnectionHandler implements IncomingEvent
 
 			downloader.setOutputFilename( connections.get( token ).filepath );
 			downloader.readBinary();
+			
+			// calculate and register the incoming blocks
+			if ( diffOfRange == 0 )
+				return;
+			
+			for ( int i = startOfRange / Globals.blockSize; i < endOfRange / Globals.blockSize; i += Globals.blockSize ) {
+				connections.get( token ).image.setNeedsCheck( i );
+			}
 		}
 		downloader.close();
 
-		// calculate and register the incoming blocks
-		if ( diffOfRange == 0 )
-			return;
-		
-		for ( int i = startOfRange / Globals.blockSize; i < endOfRange / Globals.blockSize; i += Globals.blockSize ) {
-			connections.get( token ).image.setNeedsCheck( i );
-		}
 		
 	}
 }
-- 
cgit v1.2.3-55-g7522