From f55275a627f006efabd88a41402c240737c93906 Mon Sep 17 00:00:00 2001
From: Sunpy <sunpy@osufx.com>
Date: Sun, 29 Mar 2020 23:43:08 +0200
Subject: [PATCH] Preparement for rcon support

---
 config.json            |  1 +
 sampy/client/query.py  | 34 ++++++++++++++++++++--------------
 sampy/server.py        |  3 ++-
 sampy/struct/server.py |  3 +++
 4 files changed, 26 insertions(+), 15 deletions(-)

diff --git a/config.json b/config.json
index 6b9d7fa..69ef3bc 100644
--- a/config.json
+++ b/config.json
@@ -14,6 +14,7 @@
 				"port": 7777,
 				"hostname": "Python > C",
 				"password": "",
+				"rcon_password": "changeme",
 				"max_players": 50,
 				"mode": "debug",
 				"language": "python"
diff --git a/sampy/client/query.py b/sampy/client/query.py
index a61ece0..9993b86 100644
--- a/sampy/client/query.py
+++ b/sampy/client/query.py
@@ -16,24 +16,14 @@ class QueryClient(BaseClient):
             b"i": self.query_i,
             b"r": self.query_r,
             b"c": self.query_c,
-            b"d": self.query_d
+            b"d": self.query_d,
+            b"p": self.query_p,
+            b"x": self.query_x
         }
     
     async def on_packet(self, packet: bytes):
         logger.debug("on_packet(%s)" % packet)
 
-        """
-        if len(packet) != 11: # Just a ping with some random numbers at the end (14 bytes in total)
-            await self.send(packet)
-            return
-        """
-        # ip ~~and port~~ the client connected to; (would be server ip ~~and port~~ if not though proxy)
-        #  This could be used to check if client connected remotely in some way though a proxy
-        #  SA:MP default server drops the connection if port does not match (when its client (not query protocol))
-        #server_ip = ".".join((str(x) for x in struct.unpack(b"<4B", packet[4:8])))
-        #server_port, = struct.unpack(b"<H", packet[8:10]) # This is not port, the wiki is wrong: https://wiki.sa-mp.com/wiki/Query
-        #logger.debug("[%s:%d] via %s:%d" % (self.ip, self.port, server_ip, server_port))
-
         if len(packet) <= 10: # Invalid
             return
 
@@ -109,4 +99,20 @@ class QueryClient(BaseClient):
         return packet + struct.pack(b"<H" + (b"cc%dsII" * data[0]) % tuple(len(p["nick"]) for p in players), *data)
 
     async def query_p(self, packet: bytes) -> bytes:
-        return packet
\ No newline at end of file
+        return packet
+
+    async def query_x(self, packet: bytes) -> bytes:
+        len_pswd, = struct.unpack_from(b"<H", packet, 11)
+        pswd, len_cmd = struct.unpack_from(b"<%dsH" % len_pswd, packet, 13)
+        cmd = struct.unpack_from(b"<%ds" % len_cmd, packet, 15 + len_pswd)
+
+        if len(self.server.config.rcon_password) == 0:
+            msg = b"Remote Console is not enabled on this server."
+        elif self.server.config.rcon_password.encode() != pswd:
+            msg = b"Invalid RCON password."
+            logger.warning("BAD RCON ATTEMPT BY: %s:%d" % (self.ip, self.port))
+        else:
+            # TODO: Add rcon client to command stdouts
+            return b"" # No response as all is ok
+        
+        return packet[:11] + struct.pack(b"<H%ds" % len(msg), len(msg), msg)
\ No newline at end of file
diff --git a/sampy/server.py b/sampy/server.py
index 9ad4c44..2166381 100644
--- a/sampy/server.py
+++ b/sampy/server.py
@@ -22,6 +22,7 @@ class Server:
 
     async def on_command(self, cmd: str):
         logger.debug("on_command(%s)" % cmd)
+        # TODO: When commands return a reponse we also want to forward this to potential rcon clients
     
     async def get_online_players(self): # TODO: Get data from server's client objects
         return [
@@ -48,4 +49,4 @@ class Server:
                     self.clients[addr] = Client(self, ip, port)
                 await self.clients[addr].on_packet(data)
 
-            await asyncio.sleep(0)
\ No newline at end of file
+            await asyncio.sleep(0)
diff --git a/sampy/struct/server.py b/sampy/struct/server.py
index a487e82..ae0c28e 100644
--- a/sampy/struct/server.py
+++ b/sampy/struct/server.py
@@ -4,6 +4,7 @@ class ServerConfig:
     def __init__(self,
             host: str, port: int,
             hostname: str, password: str,
+            rcon_password: str,
             max_players: int,
             mode: str, language: str):
         self.host = host
@@ -12,6 +13,8 @@ class ServerConfig:
         self.hostname = hostname
         self.password = password
 
+        self.rcon_password = rcon_password
+
         self.max_players = max_players
         
         self.mode = mode