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

package org.openslx.filetransfer;

import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLServerSocket;
import javax.net.ssl.SSLServerSocketFactory;
import javax.net.ssl.SSLSocket;

import org.apache.log4j.BasicConfigurator;
import org.apache.log4j.Logger;
import org.slf4j.LoggerFactory;

// TODO: (all files) apply formatting using strg+shift+f *after* importing scheme from ./extras/

public class Listener extends Thread
{
	private IncomingEvent incomingEvent;
	/*
	private static String pathToKeyStore =
			"/home/bjoern/javadev/DataTransfer/mySrvKeyStore.jks";
			*/
	private SSLContext context;
	private int port;
	final private int U = 85; // hex - code 'U' = 85.
	final private int D = 68; // hex - code 'D' = 68.
	
	private static Logger log = Logger.getLogger( Listener.class );
	
	static {
		// This is a temporary workaround for this annoying log4j error msg.
		// Initializing the logger before anything else is done.
		BasicConfigurator.configure();
		LoggerFactory.getLogger( "ROOT" );
	}

	/***********************************************************************/
	/**
	 * Constructor for class Listener, which gets an instance of IncomingEvent.
	 * 
	 * @param e
	 */
	public Listener( IncomingEvent e, SSLContext context, int port )
	{
		this.incomingEvent = e;
		this.context = context;
		this.port = port;
	}

	/***********************************************************************/
	/**
	 * Method listen, should run from Master Server. Listen for incoming
	 * connection, and start Downloader or Uploader.
	 * 
	 */
	private void listen()
	{ 
		try {
   		SSLServerSocketFactory sslServerSocketFactory = context.getServerSocketFactory();
   		SSLServerSocket welcomeSocket =
   				(SSLServerSocket)sslServerSocketFactory.createServerSocket( this.port );
   
   		while ( !isInterrupted() ) {
   			SSLSocket connectionSocket = (SSLSocket)welcomeSocket.accept();
   			connectionSocket.setSoTimeout( 2000 ); // 2 second timeout enough? Maybe even use a small thread pool for handling accepted connections
   			// TODO: Handle SocketTimeoutException for all reads and writes in Downloader and Uploader
   
   			byte[] b = new byte[ 1 ];
   			int length = connectionSocket.getInputStream().read( b );
   
   			System.out.println( length );
   
   			if ( b[0] == U ) {
   				log.info( "recognized U --> starting Downloader" ); // TODO: Use Logger (see masterserver code for example)
   				// --> start Downloader(socket).
   				Downloader d = new Downloader( connectionSocket );
   				incomingEvent.incomingDownloader( d );
   			}
   			else if ( b[0] == D ) {
   				log.info( "recognized D --> starting Uploader" );
   				// --> start Uploader(socket).
   				Uploader u = new Uploader( connectionSocket );
   				incomingEvent.incomingUploader( u );
   			}
   			else {
   				log.info( "Got invalid option ... close connection" );
   				connectionSocket.close();
   			}
   		}
		} catch (Exception e) {
			e.printStackTrace();	// same as writing to System.err.println(e.toString).
		}
	}

	public int getPort()
	{
		return this.port;
	}

	@Override
	public void run()
	{
		try {
			this.listen();
		} catch ( Exception e ) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
}