refine ftp component and add configuration capabilities

7 files changed, 60 insertions, 59 deletions

diff --git a/src/main/scala/com/tylerstonge/honeypot/Main.scala b/src/main/scala/com/tylerstonge/honeypot/Main.scala
index 3c01fc8..0d47de4 100644
--- a/src/main/scala/com/tylerstonge/honeypot/Main.scala
+++ b/src/main/scala/com/tylerstonge/honeypot/Main.scala
@@ -1,9 +1,8 @@
 package com.tylerstonge.honeypot
 
 import akka.actor.{ActorSystem, Props}
-import com.tylerstonge.honeypot.ftp.FtpListener
 
 object Main extends App {
-  val system = ActorSystem("hello-system")
-  val listener = system.actorOf(Props[FtpListener], name = "ftp-listener")
-}
+  val system = ActorSystem("honeypot_system")
+  val listener = system.actorOf(Props[Supervisor], name = "honeypot-supervisor")
+}
\ No newline at end of file
diff --git a/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala b/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala
index a04a6e8..ee41ea2 100644
--- a/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala
+++ b/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala
@@ -1,14 +1,38 @@
 package com.tylerstonge.honeypot
 
-import akka.actor.Actor
+import java.nio.file.Paths
+
+import akka.actor.{Actor, Props}
+import akka.event.{Logging, LoggingAdapter}
+import com.tylerstonge.honeypot.ftp.FtpListener
+import com.tylerstonge.honeypot.messages.MStartComponent
+import org.typelevel.jawn.ast.{JArray, JParser}
 
 class Supervisor extends Actor {
 
-  override def receive: Receive = {
-    case _ => println("kk dood")
+  val log: LoggingAdapter = Logging(context.system, this)
+
+  override def preStart: Unit = {
+    val cfg = JParser.parseFromFile(Paths.get("config.json").toFile).get
+    cfg.get("components").asInstanceOf[JArray].vs.foreach(c => {
+      self ! MStartComponent(c.get("name").asString, c.get("type").asString, c.get("port").asInt)
+    })
   }
 
   override def postStop {
-    println("Supervisor::postStop")
+    log.debug("supervisor is shutting down")
+  }
+
+  override def receive: Receive = {
+    case msg: MStartComponent => startComponent(msg)
+    case _ => log.debug("supervisor received unhandled message")
+  }
+
+  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 _ => log.error("unknown component type: ", msg.ctype, msg.name);
+    }
   }
 }
\ No newline at end of file
diff --git a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala
index 8a55396..48db3f5 100644
--- a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala
+++ b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpHandler.scala
@@ -6,22 +6,31 @@ import akka.io.Tcp.{PeerClosed, Received, Write}
 import akka.util.ByteString
 
 class FtpHandler extends Actor {
+
   val log: LoggingAdapter = Logging(context.system, this)
 
   override def receive: Receive = {
-    case Received(data) =>
-      log.info(">> {}", data.utf8String)
-      sender() ! Write(ByteString.apply(parse(data.utf8String)))
+    case Received(data) => sender() ! Write(ByteString.apply(parse(sanitize(data))))
     case PeerClosed =>
-      log.info("closing connection")
+      log.info("peer closed connection")
       context.stop(self)
   }
 
-  def parse(msg: String): String = msg match {
-    case "USER anonymous\n" => "331 Please specify password.\n"
-    case "PASS password\n" => "230 Login successful.\n"
-    case "PWD\n" => "257 \"/\" is the current directory\n"
-    case "QUIT\n" => "221 Goodbye.\n"
-    case _ => "200 sure\n"
+  def parse(msg: Array[String]): String = msg(0) match {
+    case "user" =>
+      log.info("attempted login with username: {}", msg(1))
+      "331 Please specify password.\n"
+    case "pass" =>
+      log.info("attempted login with password: {}", msg(1))
+      "230 Login successful.\n"
+    case "pwd" => "257 \"/\" is the current directory\n"
+    case "quit" => "221 Goodbye.\n"
+    case _ =>
+      log.info("unsupported command received: {}", msg.mkString(" "))
+      "451 Requested action aborted. Local error in processing.\n"
+  }
+
+  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 3e6606a..b988ba3 100644
--- a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala
+++ b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala
@@ -8,16 +8,16 @@ import akka.io.Tcp._
 import akka.io.{IO, Tcp}
 import akka.util.ByteString
 
-class FtpListener extends Actor {
+class FtpListener (port: Int)  extends Actor {
 
   val log: LoggingAdapter = Logging(context.system, this)
-  IO(Tcp)(context.system) ! Bind(self, new InetSocketAddress("localhost", 2121))
+  IO(Tcp)(context.system) ! Bind(self, new InetSocketAddress("localhost", port))
 
   override def receive: Receive = {
     case Bound(localAddress) =>
       log.info("listening on {}", localAddress)
     case CommandFailed(_: Bind) => context.stop(self)
-    case Connected(remote, local) =>
+    case Connected =>
       val handler = context.actorOf(Props[FtpHandler])
       val connection = sender()
       connection ! Register(handler)
diff --git a/src/main/scala/com/tylerstonge/honeypot/http/HttpListener.scala b/src/main/scala/com/tylerstonge/honeypot/http/HttpListener.scala
deleted file mode 100644
index 8943ee0..0000000
--- a/src/main/scala/com/tylerstonge/honeypot/http/HttpListener.scala
+++ /dev/null
@@ -1,27 +0,0 @@
-package com.tylerstonge.honeypot.http
-
-import java.net.InetSocketAddress
-
-import akka.actor.{Actor, Props}
-import akka.event.Logging
-import akka.io.Tcp._
-import akka.io.{IO, Tcp}
-
-
-class HttpListener extends Actor {
-  val log = Logging(context.system, this)
-
-  import context.system
-
-  IO(Tcp) ! Bind(self, new InetSocketAddress("localhost", 7333))
-
-  override def receive: Receive = {
-    case b@Bound(localAddress) => context.parent ! b
-    case CommandFailed(_: Bind) => context.stop(self)
-    case c@Connected(remote, local) =>
-      val handler = context.actorOf(Props[SimplisticHandler])
-      val connection = sender()
-      connection ! Register(handler)
-  }
-
-}
\ No newline at end of file
diff --git a/src/main/scala/com/tylerstonge/honeypot/http/SimplisticHandler.scala b/src/main/scala/com/tylerstonge/honeypot/http/SimplisticHandler.scala
deleted file mode 100644
index 2fe1409..0000000
--- a/src/main/scala/com/tylerstonge/honeypot/http/SimplisticHandler.scala
+++ /dev/null
@@ -1,11 +0,0 @@
-package com.tylerstonge.honeypot.http
-
-import akka.actor.Actor
-import akka.io.Tcp.{PeerClosed, Received, Write}
-
-class SimplisticHandler extends Actor {
-  def receive: Receive = {
-    case Received(data) => sender() ! Write(data)
-    case PeerClosed => context.stop(self)
-  }
-}
diff --git a/src/main/scala/com/tylerstonge/honeypot/messages/MStartComponent.scala b/src/main/scala/com/tylerstonge/honeypot/messages/MStartComponent.scala
new file mode 100644
index 0000000..e4109b0
--- /dev/null
+++ b/src/main/scala/com/tylerstonge/honeypot/messages/MStartComponent.scala
@@ -0,0 +1,7 @@
+package com.tylerstonge.honeypot.messages

+

+/**

+ *

+ * @author Tyler St. Onge <tylertstonge@gmail.com>

+ */

+case class MStartComponent (name: String, ctype: String, port: Int)