123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296 |
- package de.tudarmstadt.informatik.hostage;
- import java.io.IOException;
- import java.io.InputStream;
- import java.io.OutputStream;
- import java.net.Socket;
- import java.util.List;
- import android.content.Context;
- import android.content.Intent;
- import android.content.SharedPreferences;
- import android.content.SharedPreferences.Editor;
- import android.preference.PreferenceManager;
- import android.util.Log;
- import de.tudarmstadt.informatik.hostage.commons.HelperUtils;
- import de.tudarmstadt.informatik.hostage.location.MyLocationManager;
- import de.tudarmstadt.informatik.hostage.logging.AttackRecord;
- import de.tudarmstadt.informatik.hostage.logging.Logger;
- import de.tudarmstadt.informatik.hostage.logging.MessageRecord;
- import de.tudarmstadt.informatik.hostage.logging.MessageRecord.TYPE;
- import de.tudarmstadt.informatik.hostage.logging.NetworkRecord;
- import de.tudarmstadt.informatik.hostage.logging.SyncDevice;
- import de.tudarmstadt.informatik.hostage.nio.Reader;
- import de.tudarmstadt.informatik.hostage.nio.Writer;
- import de.tudarmstadt.informatik.hostage.protocol.GHOST;
- import de.tudarmstadt.informatik.hostage.protocol.Protocol;
- import de.tudarmstadt.informatik.hostage.protocol.Protocol.TALK_FIRST;
- import de.tudarmstadt.informatik.hostage.sync.tracing.TracingSyncService;
- import de.tudarmstadt.informatik.hostage.wrapper.Packet;
- /**
- * Abstract class for a connection handler using a given protocol.
- *
- * @author Mihai Plasoianu
- * @author Wulf Pfeiffer
- * @author Lars Pandikow
- */
- public class Handler implements Runnable {
- /** Time until the socket throws a time out. The time is in milliseconds. */
- private int TIMEOUT;
- private Hostage service;
- protected Protocol protocol;
- private Socket client;
- protected Thread thread;
-
- SharedPreferences pref;
- private int attack_id;
- //private int message_id = 0;
- private String externalIP;
- private String BSSID;
- private String SSID;
- private int subnetMask;
- private int internalIPAddress;
-
- private boolean logged;
- private Listener listener;
- /**
- * 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 Handler(Hostage service, Listener listener, Protocol protocol, Socket client) {
- this.service = service;
- this.listener = listener;
- this.protocol = protocol;
- if (protocol.toString().equals("GHOST")) {
- ((GHOST) protocol).setAttackerIP(client.getInetAddress());
- ((GHOST) protocol).setCurrentPort(listener.getPort());
- }
- this.client = client;
- this.thread = new Thread(this);
- pref = PreferenceManager.getDefaultSharedPreferences(service);
- TIMEOUT = pref.getInt("timeout", 30) * 1000;
- getAndIncrementAttackID(pref);
- SharedPreferences connInfo = service.getSharedPreferences(service.getString(R.string.connection_info), Context.MODE_PRIVATE);
- BSSID = connInfo.getString(service.getString(R.string.connection_info_bssid), null);
- SSID = connInfo.getString(service.getString(R.string.connection_info_ssid), null);
- externalIP = connInfo.getString(service.getString(R.string.connection_info_external_ip), null);
- // we need this info to find out whether the attack was internal
- subnetMask = connInfo.getInt(service.getString(R.string.connection_info_subnet_mask), 0);
- internalIPAddress = connInfo.getInt(service.getString(R.string.connection_info_internal_ip), 0);
- setSoTimeout(client);
- logged = false;
- thread.start();
- }
- /**
- * 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();
- }
- /**
- * Sets the interrupt flag of the thread and tries to close the socket.
- */
- public void kill() {
- service.notifyUI(this.getClass().getName(),
- new String[] { service.getString(R.string.broadcast_started), protocol.toString(), Integer.toString(listener.getPort()) });
- thread.interrupt();
- try {
- client.close();
- } catch (Exception e) {
-
- }
- boolean upload = pref.getBoolean("pref_auto_synchronize", false);
- if(upload){
- Intent intent = new Intent(service, TracingSyncService.class);
- intent.setAction(TracingSyncService.ACTION_START_SYNC);
- service.startService(intent);
- }
- //TODO kann ConcurrentModificationException auslösen, da über collection iteriert wird während elemente entfernt werden
- listener.refreshHandlers();
- }
- /**
- * 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() {
- service.notifyUI(this.getClass().getName(),
- new String[] { service.getString(R.string.broadcast_started), protocol.toString(), Integer.toString(listener.getPort()) });
- InputStream in;
- OutputStream out;
- try {
- in = client.getInputStream();
- out = client.getOutputStream();
- talkToClient(in, out);
- } catch (Exception e) {
- e.printStackTrace();
- }
- kill();
- }
- /**
- * Gets attack ID for the attack. Also increases the attack ID counter by
- * one. Method is synchronized for thread safety.
- *
- * @param pref
- * The default SharedPreference of the application
- * @return Unique integer attack ID
- */
- private synchronized void getAndIncrementAttackID(SharedPreferences pref) {
- Editor editor = pref.edit();
- attack_id = pref.getInt("ATTACK_ID_COUNTER", 0);
- editor.putInt("ATTACK_ID_COUNTER", attack_id + 1);
- editor.commit();
- }
- /**
- * 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 a MessageRecord 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 MessageRecord createMessageRecord(TYPE type, String packet) {
- MessageRecord record = new MessageRecord(true);
- //record.setId(message_id++); // autoincrement
- record.setAttack_id(attack_id);
- record.setType(type);
- record.setTimestamp(System.currentTimeMillis());
- record.setPacket(packet);
- return record;
- }
-
- /**
- * Creates a AttackRecord for a specific attack from a client.
- *
- * @return The AttackRecord representing the attack.
- */
- protected AttackRecord createAttackRecord() {
- AttackRecord record = new AttackRecord();
- record.setAttack_id(attack_id);
- record.setSync_id(attack_id);
- record.setDevice(SyncDevice.currentDevice().getDeviceID());
- record.setProtocol(protocol.toString());
- record.setExternalIP(externalIP);
- record.setLocalIP(client.getLocalAddress().getHostAddress());
- record.setLocalPort(client.getLocalPort());
- int remoteIPAddress = HelperUtils.packInetAddress(client.getInetAddress().getAddress());
- record.setWasInternalAttack((remoteIPAddress & subnetMask) == (internalIPAddress & subnetMask));
- record.setRemoteIP(client.getInetAddress().getHostAddress());
- record.setRemotePort(client.getPort());
- record.setBssid(BSSID);
- return record;
- }
-
- /**
- * Creates a NetworkRecord containing information about the current network.
- *
- * @return The NetworkRecord representing the current network.
- */
- protected NetworkRecord createNetworkRecord() {
- NetworkRecord record = new NetworkRecord();
- record.setBssid(BSSID);
- record.setSsid(SSID);
- if (MyLocationManager.getNewestLocation() != null) {
- record.setLatitude(MyLocationManager.getNewestLocation().getLatitude());
- record.setLongitude(MyLocationManager.getNewestLocation().getLongitude());
- record.setAccuracy(MyLocationManager.getNewestLocation().getAccuracy());
- record.setTimestampLocation(MyLocationManager.getNewestLocation().getTime());
- } else {
- record.setLatitude(0.0);
- record.setLongitude(0.0);
- record.setAccuracy(Float.MAX_VALUE);
- record.setTimestampLocation(0);
- }
- return record;
- }
-
- private void log(TYPE type, String packet){
- if(!logged){
- Logger.log(Hostage.getContext(), createNetworkRecord());
- Logger.log(Hostage.getContext(), createAttackRecord());
- logged = true;
- }
- Logger.log(Hostage.getContext(), createMessageRecord(type, packet));
- }
- /**
- * Communicates with a client using the corresponding protocol
- * implementation.
- *
- * @param in
- * InputStream of the socket.
- * @param out
- * OutputStream of the socket.
- * @throws IOException
- */
- protected void talkToClient(InputStream in, OutputStream out) throws IOException {
- Reader reader = new Reader(in, protocol.toString());
- Writer writer = new Writer(out);
- Packet inputLine;
- List<Packet> outputLine;
- if (protocol.whoTalksFirst() == TALK_FIRST.SERVER) {
- outputLine = protocol.processMessage(null);
- writer.write(outputLine);
- for (Packet o : outputLine) {
- log(TYPE.SEND, o.toString());
- }
- }
- while (!thread.isInterrupted() && (inputLine = reader.read()) != null) {
- outputLine = protocol.processMessage(inputLine);
- log(TYPE.RECEIVE, inputLine.toString());
- if (outputLine != null) {
- writer.write(outputLine);
- for (Packet o : outputLine) {
- log(TYPE.SEND, o.toString());
- }
- }
- if (protocol.isClosed()) {
- break;
- }
- }
- }
- }
|