123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156 |
- 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.Context;
- import android.content.SharedPreferences;
- import android.content.SharedPreferences.Editor;
- import android.preference.PreferenceManager;
- import de.tudarmstadt.informatik.hostage.HoneyListener;
- import de.tudarmstadt.informatik.hostage.HoneyService;
- 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;
- import de.tudarmstadt.informatik.hostage.ui.MainActivity;
- /**
- * Abstract class for a connection handler using a given protocol.
- * @author Mihai Plasoianu
- */
- public abstract class AbstractHandler implements Runnable {
- /**
- * Time until the socket throws a time out. The time is in milliseconds.
- */
- private int TIMEOUT;
- protected Protocol protocol;
- private Socket client;
- protected Thread thread;
- private int attack_id;
- private String externalIP;
- 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();
- TIMEOUT = pref.getInt("timeout", 30) * 1000;
- attack_id = pref.getInt("ATTACK_ID_COUNTER", 0);
- editor.putInt("ATTACK_ID_COUNTER", attack_id + 1);
- editor.commit();
- SharedPreferences sessionPref = service.getSharedPreferences(MainActivity.SESSION_DATA, Context.MODE_PRIVATE);
- BSSID = sessionPref.getString(MainActivity.BSSID, null);
- SSID = sessionPref.getString(MainActivity.SSID, null);
- externalIP = sessionPref.getString(MainActivity.EXTERNAL_IP, null);
- 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.
- */
- public void run() {
- InputStream in;
- OutputStream out;
- try {
- in = client.getInputStream();
- out = client.getOutputStream();
- talkToClient(in, out);
- } catch (Exception e) {
- e.printStackTrace();
- }
- kill();
- }
- /**
- * Sets the interrupt flag of the thread and tries to close the socket.
- */
- public void kill() {
- thread.interrupt();
- try {
- client.close();
- } 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.setExternalIP(externalIP);
- 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;
- }
- }
|