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

package de.tudarmstadt.informatik.hostage.handler;

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

import de.tudarmstadt.informatik.hostage.HoneyListener;
import de.tudarmstadt.informatik.hostage.HoneyService;
import de.tudarmstadt.informatik.hostage.io.ByteArrayReaderWriter;
import de.tudarmstadt.informatik.hostage.io.ReaderWriter;
import de.tudarmstadt.informatik.hostage.protocol.Protocol;
import de.tudarmstadt.informatik.hostage.protocol.Protocol.TALK_FIRST;
import de.tudarmstadt.informatik.hostage.wrapper.ByteArray;

public class ByteArrayHandlerImpl extends AbstractHandler {

	public ByteArrayHandlerImpl(HoneyService service, HoneyListener listener,
			Protocol protocol, Socket client) {
		super(service, listener, protocol, client);
	}

	@Override
	protected void talkToClient(InputStream in, OutputStream out)
			throws IOException {
		ReaderWriter<ByteArray> stream = new ByteArrayReaderWriter(in, out);

		ByteArray inputLine;
		List<ByteArray> outputLine;

		if (protocol.whoTalksFirst() == TALK_FIRST.SERVER) {
			outputLine = protocol.processMessage(null);
			stream.write(outputLine);
			for (ByteArray s : outputLine) {
				log.addRecord(createRecord("SEND", s.toString()));
			}
		}

		while (!thread.isInterrupted() && (inputLine = stream.read()) != null) {
			log.addRecord(createRecord("RECEIVE", inputLine.toString()));
			outputLine = protocol.processMessage(inputLine);
			if (outputLine != null) {
				stream.write(outputLine);
				for (ByteArray s : outputLine) {
					log.addRecord(createRecord("SEND", s.toString()));
				}
			}
			if (protocol.isClosed()) {
				break;
			}
		}
	}

}