From 14fa23601c560464a19a55e1b4e530b742df301a Mon Sep 17 00:00:00 2001
From: "Tyler St. Onge" <tylertstonge@gmail.com>
Date: Wed, 29 Jul 2020 22:36:47 -0400
Subject: beginning research into accepting files

---
 .../tylerstonge/honeypot/ftp/FtpFileReceiver.scala | 40 ++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileReceiver.scala

(limited to 'src/main/scala/com/tylerstonge/honeypot/ftp/FtpFileReceiver.scala')

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)
+      }
+  }
+}
-- 
cgit v1.1