hostage/src/de/tudarmstadt/informatik/hostage/handler/AbstractHandler.java

package de.tudarmstadt.informatik.hostage.handler;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;

import android.content.SharedPreferences;
import android.content.SharedPreferences.Editor;
import android.preference.PreferenceManager;
import de.tudarmstadt.informatik.hostage.ConnectionRegister;
import de.tudarmstadt.informatik.hostage.HoneyListener;
import de.tudarmstadt.informatik.hostage.HoneyService;
import de.tudarmstadt.informatik.hostage.commons.HelperUtils;
import de.tudarmstadt.informatik.hostage.logging.Logger;
import de.tudarmstadt.informatik.hostage.logging.Record;
import de.tudarmstadt.informatik.hostage.logging.Record.TYPE;
import de.tudarmstadt.informatik.hostage.protocol.Protocol;

public abstract class AbstractHandler implements Runnable {

	/**
	 * Time until the socket throws a time out. The time is in milliseconds.
	 */
	protected static final int TIMEOUT = 30 * 1000;

	protected Protocol protocol;
	private Socket client;
	protected Thread thread;

	private int attack_id;
	private String BSSID;
	private String SSID;

	private HoneyListener listener;
	protected Logger log;

	/**
	 * Constructor of the class.
	 * Initializes class variables for communication and logging.
	 * Then starts itself in a new Thread.
	 * @param service The background service.
	 * @param listener The Listener that called the service.
	 * @param protocol The protocol on which the handler is running.
	 * @param client A Socket for the communication with a remote client.
	 */
	public AbstractHandler(HoneyService service, HoneyListener listener,
			Protocol protocol, Socket client) {
		this.listener = listener;
		this.log = service.getLog();
		this.protocol = protocol;
		this.client = client;
		this.thread = new Thread(this);
		SharedPreferences pref = PreferenceManager
				.getDefaultSharedPreferences(service);
		Editor editor = pref.edit();
		attack_id = pref.getInt("ATTACK_ID_COUNTER", 0);
		editor.putInt("ATTACK_ID_COUNTER", attack_id + 1);
		editor.commit();
		BSSID = HelperUtils.getBSSID(service.getApplicationContext());
		SSID = HelperUtils.getSSID(service.getApplicationContext());
		setSoTimeout(client);
		thread.start();
	}

	/**
	 * Set the timeout of the socket to the hard coded time out variable.
	 * @param client The socket
	 * @see #TIMEOUT
	 */
	private void setSoTimeout(Socket client) {
		try {
			client.setSoTimeout(TIMEOUT);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * Creates InputStream and OutputStream for the socket.
	 * Starts communication with client.
	 * When the client closes the connection or a time out occurs the handler is finished.
	 */
	@Override
	public void run() {
		InputStream in;
		OutputStream out;
		try {
			in = client.getInputStream();
			out = client.getOutputStream();
			talkToClient(in, out);
		} catch (Exception e) {
			e.printStackTrace();
			ConnectionRegister.closeConnection();
		}
		kill();
	}

	/**
	 * Sets the interrupt flag of the thread and tries to close the socket.
	 */
	public void kill() {
		thread.interrupt();
		try {
			client.close();
			ConnectionRegister.closeConnection();
		} catch (Exception e) {
			e.printStackTrace();
		}
		listener.refreshHandlers();
	}

	/**
	 * Determines if the interrupt flag of the thread is set.
	 * @return True when the flag is set, else false.
	 */
	public boolean isTerminated() {
		return thread.isInterrupted();
	}

	/**
	 * Communicates with a client using the corresponding protocol implementation.
	 * @param in InputStream of the socket.
	 * @param out OutputStream of the socket.
	 * @throws IOException
	 */
	abstract protected void talkToClient(InputStream in, OutputStream out)
			throws IOException;

	/**
	 * Creates a Record for a message exchanged with a client.
	 * @param type The type of the message.
	 * @param packet The content of the message.
	 * @return The Record representing the communication message.
	 */
	protected Record createRecord(TYPE type, String packet) {
		Record record = new Record();
		record.setAttack_id(attack_id);
		record.setProtocol(protocol.toString());
		record.setType(type);
		record.setTimestamp(System.currentTimeMillis());
		record.setLocalIP(client.getLocalAddress());
		record.setLocalPort(protocol.getPort());
		record.setRemoteIP(client.getInetAddress());
		record.setRemotePort(client.getPort());
		record.setBSSID(BSSID);
		record.setSSID(SSID);
		record.setPacket(packet);
		return record;
	}

}