Move checking of incoming block into while loop (where it belongs)

1 files changed, 8 insertions, 7 deletions

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 );
-		}
 		
 	}
 }