1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- package de.tudarmstadt.informatik.hostage.handler;
- import java.io.IOException;
- import java.io.InputStream;
- import java.io.OutputStream;
- import java.net.Socket;
- import de.tudarmstadt.informatik.hostage.HoneyListener;
- import de.tudarmstadt.informatik.hostage.HoneyService;
- import de.tudarmstadt.informatik.hostage.logging.SQLLogger;
- import de.tudarmstadt.informatik.hostage.logging.SQLRecord;
- import de.tudarmstadt.informatik.hostage.protocol.Protocol;
- public abstract class AbstractHandler implements Runnable {
- protected static final int TIMEOUT = 30 * 1000;
- protected Protocol protocol;
- private Socket client;
- protected Thread thread;
- private int attack_id;
- private HoneyListener listener;
- protected SQLLogger log;
- 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);
- attack_id = log.getAttackCount();
- setSoTimeout(client);
- thread.start();
- }
- private void setSoTimeout(Socket client) {
- try {
- client.setSoTimeout(TIMEOUT);
- } catch (Exception e) {
- e.printStackTrace();
- }
- }
- @Override
- public void run() {
- InputStream in;
- OutputStream out;
- try {
- in = client.getInputStream();
- out = client.getOutputStream();
- talkToClient(in, out);
- } catch (Exception e) {
- e.printStackTrace();
- }
- kill();
- }
- public void kill() {
- thread.interrupt();
- try {
- client.close();
- } catch (Exception e) {
- e.printStackTrace();
- }
- listener.refreshHandlers();
- }
- public boolean isTerminated() {
- return thread.isInterrupted();
- }
- abstract protected void talkToClient(InputStream in, OutputStream out)
- throws IOException;
- protected SQLRecord createRecord(String type, String packet) {
- SQLRecord record = new SQLRecord();
- record.setAttack_id(attack_id);
- record.setProtocol(protocol.toString());
- record.setType(type);
- record.setTimestamp(String.valueOf(System.currentTimeMillis()));
- record.setLocalIP(client.getLocalAddress().getHostAddress());
- record.setLocalPort(protocol.getPort());
- record.setRemoteIP(client.getInetAddress().getHostAddress());
- record.setRemotePort(client.getPort());
- record.setPacket(packet);
- return record;
- }
- }
|