path: root/src/main/java/org/openslx/filetransfer/Downloader.java

package org.openslx.filetransfer;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.charset.StandardCharsets;
import java.security.KeyManagementException;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;
import java.security.cert.CertificateException;

import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocket;
import javax.net.ssl.SSLSocketFactory;

public class Downloader {
	// Some instance variables.
	private SSLSocketFactory sslSocketFactory;
	private SSLSocket satelliteSocket;
	private DataOutputStream dataToServer;
	private DataInputStream dataFromServer;
	private String TOKEN = null;
	private String RANGE = null;
	private String outputFilename = null;
	private String ERROR = null;
	
	/***********************************************************************//**
	 * Constructor for satellite downloader.
	 * Tries to connect to specific ip and port and sending type of action.
	 * @param ip
	 * @param port
	 * @throws IOException 
	 * @throws KeyStoreException 
	 * @throws CertificateException 
	 * @throws NoSuchAlgorithmException 
	 * @throws KeyManagementException 
	 */
	public Downloader(String ip, int port, SSLContext context) throws IOException, KeyStoreException, NoSuchAlgorithmException, CertificateException, KeyManagementException {
		// TODO: Remove old code, that's why we have git.. ;)
		/*
	    char[] passphrase = "test123".toCharArray();
	    KeyStore keystore = KeyStore.getInstance("JKS");
	    keystore.load(new FileInputStream(pathToTrustStore), passphrase);

	    TrustManagerFactory tmf = TrustManagerFactory.getInstance(TrustManagerFactory.getDefaultAlgorithm());
	    tmf.init(keystore);

	    SSLContext context = SSLContext.getInstance("SSLv3");
	    TrustManager[] trustManagers = tmf.getTrustManagers();

	    context.init(null, trustManagers, null);
		*/
		
	    // create socket.
		sslSocketFactory = context.getSocketFactory();
		
		satelliteSocket = (SSLSocket) sslSocketFactory.createSocket(ip, port);
		
		dataToServer = new DataOutputStream(satelliteSocket.getOutputStream());
		dataToServer.writeByte('D');
		dataFromServer = new DataInputStream(satelliteSocket.getInputStream());
	}

	/***********************************************************************//**
	 * Constructor for master downloader.
	 * Given parameter is the socket over which the transfer is going.
	 * @param socket
	 * @throws IOException 
	 */
	public Downloader(SSLSocket socket) throws IOException {
		satelliteSocket = socket;
		dataToServer = new DataOutputStream(satelliteSocket.getOutputStream());
		dataFromServer = new DataInputStream(satelliteSocket.getInputStream());
	}

	/***********************************************************************//**
	 * Method for setting outputFilename.
	 * @param filename
	 */
	public void setOutputFilename(String filename)
	{
		outputFilename = filename;
	}
	
	/***********************************************************************//**
	 * Method for getting outputFilename.
	 * @return Return outputFilename or null.
	 */
	public String getOutputFilename()
	{
		if (outputFilename != null)
			return outputFilename;
		return null;
	}
	
	/***********************************************************************//**
	 * Method for sending token for identification from satellite to master.
	 * @param t
	 * @throws IOException 
	 */
	public void sendToken(String token) throws IOException {
		TOKEN = token;
		String sendToken = "TOKEN=" + TOKEN;
		byte[] data = sendToken.getBytes(StandardCharsets.UTF_8);
		dataToServer.writeByte(data.length);
		dataToServer.write(data);
	}
	
	/***********************************************************************//**
	 * Method to send range of the file, which should be uploaded.
	 * Helpful for knowing how much was already uploaded if
	 * connection aborts.
	 * @param a
	 * @param b
	 * @throws IOException 
	 */
	public void sendRange(int a, int b) throws IOException {
		RANGE = a + ":" + b;
		String sendRange = "RANGE=" + RANGE;
		byte[] data = sendRange.getBytes(StandardCharsets.UTF_8);
		dataToServer.writeByte(data.length);
		dataToServer.write(data);
	}
	
	/***********************************************************************//**
	 * Method for reading incoming token for identification.
	 * @throws IOException
	 */
	public String getToken() {
		return TOKEN;
	}
	
	/***********************************************************************//**
	 * Method for reading range of file, which is downloaded.
	 * Helpful for knowing how much is already downloaded if connection aborts.
	 */
	public String getRange() {
		return RANGE;
	}
	
	/***********************************************************************//**
	 * Getter for beginning of RANGE.
	 * @return
	 */
	public int getStartOfRange() {
		if (RANGE != null) {
			String[] splitted = RANGE.split(":");
			return Integer.parseInt(splitted[0]);
		}
		return -1;
	}
	
	/***********************************************************************//**
	 * Getter for end of RANGE.
	 * @return
	 */
	public int getEndOfRange() {
		if (RANGE != null) {
			String[] splitted = RANGE.split(":");
			return Integer.parseInt(splitted[1]);
		}
		return -1;
	}
	
	/***********************************************************************//**
	 * Method for returning difference of current Range.
	 * @return
	 */
	public int getDiffOfRange() {
		int diff = Math.abs(getEndOfRange() - getStartOfRange()); 
		return diff;
	}
	
	/***********************************************************************//**
	 * Method for reading MetaData, like TOKEN and FileRange.
	 * Split incoming bytes after first '=' and store value to specific
	 * variable.
	 * @throws IOException 
	 */
	public Boolean readMetaData() throws IOException {
		try {
			while (true) {
				byte[] incoming = new byte[255];
				
				// First get length.
				dataFromServer.read(incoming, 0, 1);
				int length = incoming[0];
				System.out.println("length: " + length);
				
				if (length == 0)
					break;
				
				/**
				 *  Read the next available bytes and split by '=' for
				 *  getting TOKEN or RANGE.
				 */
				int hasRead = 0;
				while (hasRead < length) {
					int ret = dataFromServer.read(incoming, hasRead, length - hasRead);
					if (ret == -1) {
						System.out.println("Error occured while reading Metadata.");
						return false;
					}
					hasRead += ret;
				}
				
				String data = new String(incoming, 0, length, "UTF-8");
				// System.out.println(data);
				
				String[] splitted = data.split("=");
				// System.out.println("splitted[0]: " + splitted[0]);
				// System.out.println("splitted[1]: " + splitted[1]);
				if (splitted[0] != null && splitted[0].equals("TOKEN")) {
					if (splitted[1] != null)
						TOKEN = splitted[1];
					System.out.println("TOKEN: " + TOKEN);
				}
				else if (splitted[0].equals("RANGE")) {
					if (splitted[1] != null)
						RANGE = splitted[1];
					System.out.println("RANGE: '" + RANGE + "'");
				}
				else if (splitted[0].equals("ERROR")) {
					if (splitted[1] != null)
						ERROR = splitted[1];
					System.err.println("ERROR: " + ERROR);
					this.close();
					return false;
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
		return true;
	}
	
	/***********************************************************************//**
	 * Method for reading Binary. Reading the current Range of incoming binary.
	 * @throws IOException 
	 */
	public Boolean readBinary() throws IOException {
		int length = getDiffOfRange();
		byte[] incoming = new byte[4000];
		
		int hasRead = 0;
		while (hasRead < length) {
			int ret = dataFromServer.read(incoming, hasRead, length - hasRead);
			if (ret == -1) {
				System.out.println("Error occured in Downloader.readBinary(),"
						+ " while reading binary.");
				return false;
			}
			hasRead += ret;
		}
		
		RandomAccessFile file;
		try {
			file = new RandomAccessFile(new File(outputFilename), "rw");
			file.seek(getStartOfRange());
			file.write(incoming, 0, length);
			file.close();

		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
		return true;
	}
	
	/***********************************************************************//**
	 * Method for sending error Code to server. For example in case of wrong
	 * token, send code for wrong token.
	 * @throws IOException 
	 */
	public void sendErrorCode(String errString) throws IOException {
		String sendError = "ERROR=" + errString;
		byte[] data = sendError.getBytes(StandardCharsets.UTF_8);
		dataToServer.writeByte(data.length);
		dataToServer.write(data);
	}
	
	/***********************************************************************//**
	 * Method for closing connection, if download has finished.
	 * @throws IOException 
	 */
	public void close() throws IOException {
		this.satelliteSocket.close();
	}
}