summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTyler St. Onge <tylertstonge@gmail.com>2020-07-29 22:36:47 -0400
committerTyler St. Onge <tylertstonge@gmail.com>2020-07-29 22:36:47 -0400
commit14fa23601c560464a19a55e1b4e530b742df301a (patch)
tree08f6322343b7b614f3683a683a538595b0234e07
parent747b6af76b23650756811d896bf76b4331419784 (diff)
beginning research into accepting files
-rw-r--r--src/main/scala/com/tylerstonge/honeypot/Supervisor.scala3
-rw-r--r--src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileReceiver.scala40
-rw-r--r--src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileRetriever.scala50
-rw-r--r--src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala20
-rw-r--r--src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala8
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)