Adding RCON Connector

src/main/scala/org/codeoverflow/chatoverflow/requirement/service/rcon/RconConnector.scala

@@ -0,0 +1,129 @@
+package org.codeoverflow.chatoverflow.requirement.service.rcon
+
+import java.io.{DataInputStream, InputStream, OutputStream}
+import java.net.{Socket, SocketException}
+import java.nio.{ByteBuffer, ByteOrder}
+import java.util.Random
+
+import org.codeoverflow.chatoverflow.WithLogger
+import org.codeoverflow.chatoverflow.connector.Connector
+
+class RconConnector(override val sourceIdentifier: String) extends Connector(sourceIdentifier) with WithLogger {
+  override protected var requiredCredentialKeys: List[String] = List("password", "address")
+  override protected var optionalCredentialKeys: List[String] = List("port")
+
+  private var socket: Socket = _
+  private var outputStream: OutputStream = _
+  private var inputStream: InputStream = _
+  private var requestId: Int = 0
+  private var loggedIn = false
+
+  def sendCommand(command: String): String = {
+    if (!loggedIn) {
+      logger error "Could not execute RCON Command due to wrong password or no connection"
+      return null
+    }
+    logger debug s"Sending $command to RCON"
+    requestId += 1
+    if (write(2, command.getBytes("ASCII"))) {
+      return read()
+    }
+    null
+  }
+
+
+  /**
+    * Starts the connector, e.g. creates a connection with its platform.
+    */
+  override def start(): Boolean = {
+    logger info s"Starting rcon connection to ${credentials.get.getValue("address").get}"
+    var port: Int = 25575
+    if (credentials.get.exists("port")) {
+      port = credentials.get.getValue("port").get.toInt
+      if (port < 1 || port > 65535) {
+        return false
+      }
+    }
+    socket = new Socket(credentials.get.getValue("address").get, port)
+    socket.setKeepAlive(true)
+    outputStream = socket.getOutputStream
+    inputStream = socket.getInputStream
+    login()
+    Thread.sleep(5000)
+    true
+  }
+
+  private def login(): Unit = {
+    requestId = new Random().nextInt(Integer.MAX_VALUE)
+    logger info "Logging RCON in..."
+    val password = credentials.get.getValue("password").get
+    if (write(3, password.getBytes("ASCII"))) {
+      if (read() == null) {
+        logger error "Could not log in to RCON Server. Password is Wrong!"
+      } else {
+        logger debug "Login to RCON was successful"
+        loggedIn = true
+      }
+    }
+  }
+
+  private def write(packageType: Int, payload: Array[Byte]): Boolean = {
+    try  {
+      val length = 4 + 4 + payload.length + 1 + 1
+      var byteBuffer: ByteBuffer = ByteBuffer.allocate(length + 4)
+      byteBuffer.order(ByteOrder.LITTLE_ENDIAN)
+
+      byteBuffer.putInt(length)
+      byteBuffer.putInt(requestId)
+      byteBuffer.putInt(packageType)
+      byteBuffer.put(payload)
+      byteBuffer.put(0x00.toByte)
+      byteBuffer.put(0x00.toByte)
+
+      outputStream.write(byteBuffer.array())
+      outputStream.flush()
+    } catch {
+      case e: NullPointerException => {
+        logger error "There was and is no Connection to the RCON Server, please try restarting."
+        return false
+      }
+      case e: SocketException => {
+        logger error "Connection Error to RCON Server. This request will not be sended!"
+        return false
+      }
+    }
+    true
+  }
+
+  private def read(): String = {
+    try {
+      val header: Array[Byte] = Array.ofDim[Byte](4*3)
+      inputStream.read(header)
+      val headerBuffer: ByteBuffer = ByteBuffer.wrap(header)
+      headerBuffer.order(ByteOrder.LITTLE_ENDIAN)
+      val length = headerBuffer.getInt()
+      val packageType = headerBuffer.getInt
+      val payload: Array[Byte] = Array.ofDim[Byte](length - 4 - 4 - 2)
+      val dataInputStream: DataInputStream = new DataInputStream(inputStream)
+      dataInputStream.readFully(payload)
+      dataInputStream.read(Array.ofDim[Byte](2))
+      if (packageType == -1) {
+        return null
+      }
+      new String(payload, "ASCII")
+    } catch {
+      case e: NegativeArraySizeException => null;
+    }
+  }
+
+  private[rcon] def isLoggedIn: Boolean = loggedIn
+
+  /**
+    * This stops the activity of the connector, e.g. by closing the platform connection.
+    */
+  override def stop(): Boolean = {
+    logger info s"Stopped RCON connector to ${credentials.get.getValue("address")}!"
+    socket.close()
+    true
+  }
+}

src/main/scala/org/codeoverflow/chatoverflow/requirement/service/rcon/impl/RconInputImpl.scala

@@ -0,0 +1,19 @@
+package org.codeoverflow.chatoverflow.requirement.service.rcon.impl
+
+import org.codeoverflow.chatoverflow.WithLogger
+import org.codeoverflow.chatoverflow.api.io.input.RconInput
+import org.codeoverflow.chatoverflow.registry.Impl
+import org.codeoverflow.chatoverflow.requirement.InputImpl
+import org.codeoverflow.chatoverflow.requirement.service.rcon.RconConnector
+
+@Impl(impl = classOf[RconInput], connector = classOf[RconConnector])
+class RconInputImpl extends InputImpl[RconConnector] with RconInput with WithLogger {
+  override def getCommandOutput(command: String): String = sourceConnector.get.sendCommand(command)
+
+  /**
+    * Start the input, called after source connector did init
+    *
+    * @return true if starting the input was successful, false if some problems occurred
+    */
+  override def start(): Boolean = sourceConnector.get.isLoggedIn
+}

src/main/scala/org/codeoverflow/chatoverflow/requirement/service/rcon/impl/RconOutputImpl.scala

@@ -0,0 +1,21 @@
+package org.codeoverflow.chatoverflow.requirement.service.rcon.impl
+
+import org.codeoverflow.chatoverflow.WithLogger
+import org.codeoverflow.chatoverflow.api.io.output.RconOutput
+import org.codeoverflow.chatoverflow.registry.Impl
+import org.codeoverflow.chatoverflow.requirement.OutputImpl
+import org.codeoverflow.chatoverflow.requirement.service.rcon.RconConnector
+
+@Impl(impl = classOf[RconOutput], connector = classOf[RconConnector])
+class RconOutputImpl extends OutputImpl[RconConnector] with RconOutput with WithLogger {
+  override def sendCommand(command: String): Boolean = {
+    sourceConnector.get.sendCommand(command) != null
+  }
+
+  /**
+    * Start the input, called after source connector did init
+    *
+    * @return true if starting the input was successful, false if some problems occurred
+    */
+  override def start(): Boolean = sourceConnector.get.isLoggedIn
+}