hostage/src/de/tudarmstadt/informatik/hostage/protocol/GhostProtocol.java

package de.tudarmstadt.informatik.hostage.protocol;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.IOException;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;

import de.tudarmstadt.informatik.hostage.wrapper.Packet;

/**
 * Ghost Protocol. This protocol mirrors an incoming connection back to the
 * attacker on the same port, that it is running on. It will send all incoming
 * requests back to the attacker on the mirrored connection and will relpy with
 * the responses it get's from this mirrored connection.
 * 
 * @author Wulf Pfeiffer
 */
public class GhostProtocol implements Protocol {

	private boolean isClosed = false;

	@Override
	public int getDefaultPort() {
		return 5050; // TODO dynamic port / whats the default!? (1433)
	}

	@Override
	public TALK_FIRST whoTalksFirst() {
		return TALK_FIRST.CLIENT;
	}

	@Override
	public List<Packet> processMessage(Packet requestPacket) {
		List<Packet> responsePackets = new ArrayList<Packet>();
		try {
			if (mirroredConnection == null) {
				mirroredConnection = new Socket("192.168.178.86", 5050); // TODO
				mirrorInputStream = new BufferedInputStream(
						mirroredConnection.getInputStream());
				mirrorOutputStream = new BufferedOutputStream(
						mirroredConnection.getOutputStream());
			}
			if (mirroredConnection.isInputShutdown()
					|| mirroredConnection.isOutputShutdown()) {
				mirrorInputStream.close();
				mirrorOutputStream.close();
				mirroredConnection.close();
				isClosed = true;
			}

			mirrorOutputStream.write(requestPacket.getMessage());
			mirrorOutputStream.flush();

			int availableBytes;
			while ((availableBytes = mirrorInputStream.available()) <= 0) {
				try {
					Thread.sleep(1);
				} catch (InterruptedException e) {
					e.printStackTrace();
				}
			}
			byte[] mirrorResponse = new byte[availableBytes];
			mirrorInputStream.read(mirrorResponse);
			responsePackets.add(new Packet(mirrorResponse));
		} catch (IOException e) {
			e.printStackTrace();
		}
		return responsePackets;
	}

	@Override
	public boolean isClosed() {
		return isClosed;
	}

	@Override
	public boolean isSecure() {
		return false;
	}

	@Override
	public Class<byte[]> getType() {
		return byte[].class;
	}

	@Override
	public String toString() {
		return "GhostProtocol";
	}

	private Socket mirroredConnection;
	private BufferedInputStream mirrorInputStream;
	private BufferedOutputStream mirrorOutputStream;
}