Inserted key handling with private key and public key for handshake and switched arguments in call getImage to the right order.

1 files changed, 5 insertions, 1 deletions

diff --git a/src/main/java/org/openslx/satellitedaemon/filetransfer/FileDownloadWorker.java b/src/main/java/org/openslx/satellitedaemon/filetransfer/FileDownloadWorker.java
index 160e2fc..e8b8b4d 100644
--- a/src/main/java/org/openslx/satellitedaemon/filetransfer/FileDownloadWorker.java
+++ b/src/main/java/org/openslx/satellitedaemon/filetransfer/FileDownloadWorker.java
@@ -11,6 +11,7 @@ import org.openslx.filetransfer.WantRangeCallback;
 import org.openslx.imagemaster.thrift.iface.DownloadData;
 import org.openslx.satellitedaemon.Globals;
 import org.openslx.satellitedaemon.db.DbImage;
+import org.openslx.satellitedaemon.db.DbImage.Status;
 
 public class FileDownloadWorker implements Runnable {
 	private static Logger log = Logger.getLogger(FileDownloadWorker.class);
@@ -50,8 +51,11 @@ public class FileDownloadWorker implements Runnable {
 					@Override
 					public FileRange get() {
 						// get start of range.
-						if (pos >= size)
+						if (pos >= size) {
+							log.debug("Download completed.");
+							image.updateStatus(Status.successfully_decentralized);
 							return null;
+						}
 						
 						long startOfRange =  pos;
 						long endOfRange = Math.min(pos + Globals.BLOCKSIZE, image.fileSize);