refine ftp component and add configuration capabilities

1 files changed, 19 insertions, 10 deletions

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(" ")
   }
 }