diff options
Diffstat (limited to 'src/main/scala/com/tylerstonge/honeypot')
5 files changed, 118 insertions, 3 deletions
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) |