From 14fa23601c560464a19a55e1b4e530b742df301a Mon Sep 17 00:00:00 2001 From: "Tyler St. Onge" Date: Wed, 29 Jul 2020 22:36:47 -0400 Subject: beginning research into accepting files --- .../com/tylerstonge/honeypot/Supervisor.scala | 3 +- .../tylerstonge/honeypot/ftp/FtpFileReceiver.scala | 40 +++++++++++++++++ .../honeypot/ftp/FtpFileRetriever.scala | 50 ++++++++++++++++++++++ .../com/tylerstonge/honeypot/ftp/FtpHandler.scala | 20 +++++++++ .../com/tylerstonge/honeypot/ftp/FtpListener.scala | 8 +++- 5 files changed, 118 insertions(+), 3 deletions(-) create mode 100644 src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileReceiver.scala create mode 100644 src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileRetriever.scala diff --git a/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala b/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala index ee41ea2..25f1154 100644 --- a/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala +++ b/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala @@ -8,6 +8,7 @@ import com.tylerstonge.honeypot.ftp.FtpListener import com.tylerstonge.honeypot.messages.MStartComponent import org.typelevel.jawn.ast.{JArray, JParser} + class Supervisor extends Actor { val log: LoggingAdapter = Logging(context.system, this) @@ -31,7 +32,7 @@ class Supervisor extends Actor { private def startComponent(msg: MStartComponent) { log.info("starting component :: {}", msg.name) msg.ctype match { - case "ftp" => context.actorOf(Props(new FtpListener(msg.port)), name = msg.name) + case "ftp" => context.actorOf(FtpListener.props(msg.port), name = msg.name) case _ => log.error("unknown component type: ", msg.ctype, msg.name); } } diff --git a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileReceiver.scala b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileReceiver.scala new file mode 100644 index 0000000..74ca594 --- /dev/null +++ b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileReceiver.scala @@ -0,0 +1,40 @@ +package com.tylerstonge.honeypot.ftp + +import java.net.InetSocketAddress +import java.nio.file.{Files, OpenOption, Paths, StandardOpenOption} + +import akka.actor.{Actor, Props} +import akka.event.{Logging, LoggingAdapter} +import akka.io.Tcp._ +import akka.io.{IO, Tcp} +import akka.util.{ByteString, ByteStringBuilder} + + +object FtpFileReceiver { + def props(port: Int): Props = Props(new FtpFileReceiver(port)) +} + +class FtpFileReceiver(port: Int) extends Actor { + + val log: LoggingAdapter = Logging(context.system, this) + IO(Tcp)(context.system) ! Bind(self, new InetSocketAddress("localhost", port)) + val fileData: ByteStringBuilder = ByteString.newBuilder + + override def receive: Receive = { + case Bound(localAddress) => + log.info("listening on {}", localAddress) + case CommandFailed(_: Bind) => context.stop(self) + case Connected(_, _) => + log.info("peer connected") + context.become { + case Received(data) => + fileData.addAll(data) + case PeerClosed => + log.info("peer closed connection, writing file to disk") + val out = Files.newByteChannel(Paths.get("testfile.txt"), StandardOpenOption.CREATE_NEW, StandardOpenOption.WRITE) + out.write(fileData.result().toByteBuffer) + out.close() + context.stop(self) + } + } +} diff --git a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileRetriever.scala b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileRetriever.scala new file mode 100644 index 0000000..1e441dd --- /dev/null +++ b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileRetriever.scala @@ -0,0 +1,50 @@ +package com.tylerstonge.honeypot.ftp + +import java.net.InetSocketAddress +import java.nio.file.{Files, Paths} + +import akka.actor.{Actor, Props} +import akka.event.{Logging, LoggingAdapter} +import akka.io.Tcp._ +import akka.io.{IO, Tcp} +import akka.util.{ByteString, ByteStringBuilder} + +object FtpFileRetriever { + def props(remote: InetSocketAddress): Props = Props(new FtpFileRetriever(remote)) +} + +class FtpFileRetriever(remote: InetSocketAddress) extends Actor { + + val log: LoggingAdapter = Logging(context.system, this) + IO(Tcp)(context.system) ! Connect(remote) + val fileData: ByteStringBuilder = ByteString.newBuilder + + override def preStart(): Unit = { + super.preStart() + log.info("started retrieving from {}:{}", remote.getHostString, remote.getPort) + } + + override def receive: Receive = { + case CommandFailed(_: Connect) => + context.stop(self) + case c@Connected(_, _) => + val connection = sender() + connection ! Register(self) + context.become { + case data: ByteString => + log.info("received {} bytes", data.length) + fileData.addAll(data) + case CommandFailed(w: Write) => + log.info("command failed") + case Received(data) => + log.info("received {} bytes", data.length) + fileData.addAll(data) + case _: ConnectionClosed => + log.info("peer closed connection, writing file to disk") + val out = Files.newByteChannel(Paths.get("testfile.txt")) + out.write(fileData.result().toByteBuffer) + out.close() + context.stop(self) + } + } +} diff --git a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala index 48db3f5..8bdb190 100644 --- a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala +++ b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala @@ -1,5 +1,7 @@ package com.tylerstonge.honeypot.ftp +import java.net.InetSocketAddress + import akka.actor.Actor import akka.event.{Logging, LoggingAdapter} import akka.io.Tcp.{PeerClosed, Received, Write} @@ -25,11 +27,29 @@ class FtpHandler extends Actor { "230 Login successful.\n" case "pwd" => "257 \"/\" is the current directory\n" case "quit" => "221 Goodbye.\n" + case "pasv" => + context.actorOf(FtpFileReceiver.props(1287)) + "227 entering passive mode (127,0,0,1,5,7)\n" + case "port" => + context.actorOf(FtpFileRetriever.props(new InetSocketAddress(getHostname(msg(1)), getPort(msg(1))))) + "200 OK\n" + case "stor" => + "200 OK \n" case _ => log.info("unsupported command received: {}", msg.mkString(" ")) "451 Requested action aborted. Local error in processing.\n" } + def getHostname(msg: String): String = { + val split = msg.split(",") + split.slice(0, 4).mkString(".") + } + + def getPort(msg: String): Int = { + val split = msg.split(",") + split(4).toInt * 256 + split(5).toInt + } + def sanitize(data: ByteString): Array[String] = { data.utf8String.trim.toLowerCase().split(" ") } diff --git a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala index b988ba3..07557bc 100644 --- a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala +++ b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala @@ -8,7 +8,11 @@ import akka.io.Tcp._ import akka.io.{IO, Tcp} import akka.util.ByteString -class FtpListener (port: Int) extends Actor { +object FtpListener { + def props(port: Int): Props = Props(new FtpListener(port)) +} + +class FtpListener(port: Int) extends Actor { val log: LoggingAdapter = Logging(context.system, this) IO(Tcp)(context.system) ! Bind(self, new InetSocketAddress("localhost", port)) @@ -17,7 +21,7 @@ class FtpListener (port: Int) extends Actor { case Bound(localAddress) => log.info("listening on {}", localAddress) case CommandFailed(_: Bind) => context.stop(self) - case Connected => + case Connected(_, _) => val handler = context.actorOf(Props[FtpHandler]) val connection = sender() connection ! Register(handler) -- cgit v1.1