From a283f762f7763573d089d8e8ab250d72d525b2bc Mon Sep 17 00:00:00 2001
From: Lucino772 <lucapalmi772@gmail.com>
Date: Thu, 6 May 2021 14:30:45 +0100
Subject: [PATCH] Moved mojang.api.net.rcon to mojang.minecraft.proto.rcon

---
 mojang/minecraft/proto/rcon/__init__.py | 71 +++++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 mojang/minecraft/proto/rcon/__init__.py

diff --git a/mojang/minecraft/proto/rcon/__init__.py b/mojang/minecraft/proto/rcon/__init__.py
new file mode 100644
index 00000000..1ab33b4e
--- /dev/null
+++ b/mojang/minecraft/proto/rcon/__init__.py
@@ -0,0 +1,71 @@
+import random
+import socket
+import struct
+from contextlib import contextmanager
+from typing import IO, Tuple
+
+
+def get_request_id():
+    return random.randint(0,2**31)
+
+def _read_fully(buffer: IO, length: int):
+    data = buffer.read(length)
+    
+    while len(data) < length:
+        _data =  buffer.read(length - len(data))
+        if _data:
+            data += _data
+
+    return data
+
+def _write_packet(sock: socket.socket, packet_type: int, payload: str):
+    packet_id = get_request_id()
+    payload = payload.encode('ascii') + b'\00'
+    packet = struct.pack('<ii{}s'.format(len(payload)), packet_id, packet_type, payload)
+
+    with sock.makefile('wb') as buffer:
+        buffer.write(len(packet).to_bytes(4, 'little'))
+        buffer.write(packet)
+
+    return packet_id
+
+
+@contextmanager
+def session(addr: Tuple[str, int], password: str, timeout: float = 3):
+    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    sock.settimeout(timeout)
+    sock.connect(addr)
+
+    # Send login request
+    packet_id = _write_packet(sock, 3, password)
+
+    # Receive login response
+    with sock.makefile('rb') as buffer:
+        length = int.from_bytes(buffer.read(4), 'little')
+        r_packet_id, r_type = struct.unpack_from('<ii', buffer.read(length))
+    
+    # Check packet id and packet type
+    if r_packet_id != packet_id or r_type != 2:
+        raise Exception('Authentication failed')
+
+    def send(command: str):
+        # TODO: Parse command and command response
+
+        # Send command
+        packet_id = _write_packet(sock, 2, command)
+
+        # Receive response
+        with sock.makefile('rb') as buffer:
+            length = int.from_bytes(buffer.read(4), 'little')
+            r_packet_id, r_type, payload = struct.unpack_from('<ii{}s'.format(length-10), _read_fully(buffer, length))
+        
+        # Check packet id and packet type
+        if r_packet_id != packet_id or r_type != 0:
+            raise Exception('Command error')
+
+        return payload.decode('ascii')
+    
+    try:
+        yield send
+    finally:
+        sock.close()