diff options
author | Nils Schwabe | 2014-07-04 10:20:04 +0200 |
---|---|---|
committer | Nils Schwabe | 2014-07-04 10:20:04 +0200 |
commit | 63af3084df7d6ea17defe104844a7298cda7b459 (patch) | |
tree | a27364921ff16c4fe2dc7ea9a9f7d424a10c1a64 /src/main/java/org/openslx/imagemaster/serverconnection | |
parent | Add implementation for the new up- and download protocoll (diff) | |
download | masterserver-63af3084df7d6ea17defe104844a7298cda7b459.tar.gz masterserver-63af3084df7d6ea17defe104844a7298cda7b459.tar.xz masterserver-63af3084df7d6ea17defe104844a7298cda7b459.zip |
Make use of the filetransfer classes
Diffstat (limited to 'src/main/java/org/openslx/imagemaster/serverconnection')
-rw-r--r-- | src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java | 135 | ||||
-rw-r--r-- | src/main/java/org/openslx/imagemaster/serverconnection/ImageProcessor.java | 68 |
2 files changed, 174 insertions, 29 deletions
diff --git a/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java b/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java new file mode 100644 index 0000000..498e058 --- /dev/null +++ b/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java @@ -0,0 +1,135 @@ +package org.openslx.imagemaster.serverconnection; + +import java.io.FileInputStream; +import java.io.FileNotFoundException; +import java.io.IOException; +import java.security.KeyManagementException; +import java.security.KeyStore; +import java.security.KeyStoreException; +import java.security.NoSuchAlgorithmException; +import java.security.UnrecoverableKeyException; +import java.security.cert.CertificateException; +import java.util.HashMap; +import java.util.LinkedList; +import java.util.List; +import java.util.Map; + +import javax.net.ssl.KeyManager; +import javax.net.ssl.KeyManagerFactory; +import javax.net.ssl.SSLContext; + +import org.apache.log4j.Logger; +import org.openslx.filetransfer.Downloader; +import org.openslx.filetransfer.IncomingEvent; +import org.openslx.filetransfer.Listener; +import org.openslx.filetransfer.Uploader; +import org.openslx.imagemaster.Globals; +import org.openslx.imagemaster.db.DbImage; +import org.openslx.imagemaster.util.Tuple; + +/** + * Class to handle all incoming and outgoing connections. + */ +public class ConnectionHandler implements IncomingEvent +{ + private static Logger log = Logger.getLogger( ConnectionHandler.class ); + private static SSLContext sslContext; + /** + * Key: token, + * Value: Tuple of the listener and the filepath. + */ + private static Map<String, Tuple<Thread, String>> activeListeners = new HashMap<>(); + private static List<Integer> possiblePorts = new LinkedList<>(); + private static IncomingEvent eventHandler = new ConnectionHandler(); + + static { + possiblePorts.add( 1234 ); + possiblePorts.add( 1235 ); + possiblePorts.add( 1236 ); + possiblePorts.add( 1237 ); + possiblePorts.add( 1238 ); + possiblePorts.add( 1239 ); + possiblePorts.add( 1240 ); + possiblePorts.add( 1241 ); + possiblePorts.add( 1242 ); + possiblePorts.add( 1243 ); + possiblePorts.add( 1244 ); + + try { + String pathToKeyStore = Globals.getSslKeystoreFile(); + char[] passphrase = Globals.getSslKeystorePassword().toCharArray(); + KeyStore keystore = KeyStore.getInstance("JKS"); + keystore.load(new FileInputStream(pathToKeyStore), passphrase); + KeyManagerFactory kmf = KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm()); + kmf.init(keystore, passphrase); + sslContext = SSLContext.getInstance("SSLv3"); + KeyManager[] keyManagers = kmf.getKeyManagers(); + + sslContext.init(keyManagers, null, null); + } catch (FileNotFoundException e) { + log.error( "Could not find keystore." ); + } catch ( KeyStoreException e ) { + log.error( "KeyStore implemenation not supported." ); + } catch ( NoSuchAlgorithmException e ) { + log.error( "Could not find such Algorithm" ); + } catch ( CertificateException e ) { + log.error( "Certificate unvalid." ); + } catch ( IOException e ) { + log.error( "Could not read keyfile" ); + } catch ( UnrecoverableKeyException e ) { + log.error( "Key in keystore is not valid" ); + } catch ( KeyManagementException e ) { + log.error( "Context initialization failed." ); + } + } + + public static void addConnection(String token, String filepath) + { + int port = possiblePorts.remove( 0 ); + Listener listener = new Listener( eventHandler, sslContext, port ); + + listener.start(); + + activeListeners.put( token, new Tuple<Thread, String>(listener, filepath) ); + } + + public static boolean hasConnection( String token ) + { + return activeListeners.containsKey( token ); + } + + public static void removeConnection( String token ) + { + Listener l = (Listener)activeListeners.remove( token ).x; + l.interrupt(); + possiblePorts.add(l.getPort()); // add port back to possible's list + } + + /** + * Server is uploading - client is downloading! + */ + @Override + public void incomingUploader( Uploader uploader ) throws IOException + { + // TODO: Handle incoming uploads (client download requests) + } + + /** + * Server is downloading - client is uploading! + */ + @Override + public void incomingDownloader( Downloader downloader ) throws IOException + { + // try to read meta data + while (downloader.readMetaData()) { + // check token to identify the client + String token = downloader.getToken(); + if (!activeListeners.containsKey( token )) { + return; + } + downloader.setOutputFilename( activeListeners.get( token ).y ); + downloader.readBinary(); + downloader.close(); + } + } +} diff --git a/src/main/java/org/openslx/imagemaster/serverconnection/ImageProcessor.java b/src/main/java/org/openslx/imagemaster/serverconnection/ImageProcessor.java index d4ae717..c7ee973 100644 --- a/src/main/java/org/openslx/imagemaster/serverconnection/ImageProcessor.java +++ b/src/main/java/org/openslx/imagemaster/serverconnection/ImageProcessor.java @@ -14,6 +14,7 @@ import org.openslx.imagemaster.util.RandomString; public class ImageProcessor { + private static Logger log = Logger.getLogger( ImageProcessor.class ); /** @@ -22,14 +23,15 @@ public class ImageProcessor private static final int AMOUNT = 20; /** * The uploading images. - * Key: imageUUID, - * Value: imageInfos + * Key: imageUUID, + * Value: imageInfos */ private static HashMap<String, ImageInfos> uploadingImages = new HashMap<>(); /** * Checks if this image is already uploading and returns a new list with missing blocks if so. * Puts the new image into processing list else. + * * @param serverSessionId The uploading server * @param imageData The data of the image * @return @@ -38,9 +40,9 @@ public class ImageProcessor { // check image data // TODO: do security checks - + String uuid = imageData.uuid; - + // check if image is already uploading if ( uploadingImages.containsKey( uuid ) ) { List<Integer> missing = getMissingBlocks( uuid, AMOUNT ); @@ -52,19 +54,20 @@ public class ImageProcessor uploadingImages.get( uuid ).setLastSentBlocks( missing ); return new UploadInfos( uploadingImages.get( uuid ).getToken(), missing ); } - + // insert new image and generate token - synchronized (uploadingImages) { - int nBlocks = (int)Math.ceil(imageData.fileSize / Globals.blockSize); + synchronized ( uploadingImages ) { + int nBlocks = (int)Math.ceil( imageData.fileSize / Globals.blockSize ); List<Integer> allBlocks = new LinkedList<>(); - for (int i = 0; i < nBlocks; i++) { // fill empty list with all block numbers + for ( int i = 0; i < nBlocks; i++ ) { // fill empty list with all block numbers allBlocks.add( i ); } String token = RandomString.generate( 100, false ); - // TODO: init updownloader class - uploadingImages.put( uuid, new ImageInfos(token, allBlocks, serverSessionId, new Timestamp(System.currentTimeMillis())) ); - DbImage.insert( imageData, System.currentTimeMillis(), token, allBlocks, serverSessionId ); - + String filepath = Globals.getImageDir() + "/" + uuid + ".vmdk"; + ConnectionHandler.addConnection( token, filepath ); + uploadingImages.put( uuid, new ImageInfos( token, allBlocks, serverSessionId, new Timestamp( System.currentTimeMillis() ) ) ); + DbImage.insert( imageData, System.currentTimeMillis(), token, allBlocks, serverSessionId, filepath ); + List<Integer> missing = getMissingBlocks( uuid, AMOUNT ); if ( missing.isEmpty() ) { uploadDone( uuid ); @@ -73,51 +76,58 @@ public class ImageProcessor return new UploadInfos( token, missing ); } } - + /** * Returns a specified number of missing blocks. + * * @param imageUUID The image of which you want to get the missing blocks from * @param amount The amount of blocks that you want to get - * @return The missing blocksht + * @return The missing blocks */ private static List<Integer> getMissingBlocks( String imageUUID, int amount ) { List<Integer> list = uploadingImages.get( imageUUID ).getMissingBlocks(); List<Integer> result = new LinkedList<>(); - + if ( amount > list.size() ) amount = list.size(); - + for ( int i = 0; i < amount; i++ ) { result.add( list.get( i ) ); } return result; } - + /** * Is triggered when an upload of an image is done. * Removes image from process list, updates db entry and moves file on hard drive. + * * @param uuid */ - private static void uploadDone( String uuid ) { - synchronized(uploadingImages) { - uploadingImages.remove( uuid ); - DbImage.updateMissingBlocks( uuid, null ); + private static void uploadDone( String uuid ) + { + String token; + synchronized ( uploadingImages ) { + token = uploadingImages.remove( uuid ).getToken(); } + DbImage.updateMissingBlocks( uuid, null ); // file was already downloaded in the right location by the updownloader class. + // remove the connection so that it can be used by a new client + ConnectionHandler.removeConnection( token ); } - + /** * Checks pending uploads in database and adds them to process list again. */ - public static void checkUploading() { + public static void checkUploading() + { List<DbImage> list = DbImage.getUploadingImages(); - for (DbImage image : list) { - // TODO: init updownloader class - String token = RandomString.generate( 100, false ); - ImageInfos infos = new ImageInfos(token, image.missingBlocks, image.serverSessionId, image.timestamp); - uploadingImages.put( image.UUID, infos); + for ( DbImage image : list ) { + String token = image.token; + ConnectionHandler.addConnection( token, image.imagePath ); + ImageInfos infos = new ImageInfos( token, image.missingBlocks, image.serverSessionId, image.timestamp ); + uploadingImages.put( image.UUID, infos ); } - log.info("Added " + list.size() + " pending upload(s) to process list again."); + log.info( "Added " + list.size() + " pending upload(s) to process list again." ); } } |