From 84d70f00f436803ae3941d94ba91a02e69813878 Mon Sep 17 00:00:00 2001
From: Richard Neumann <mail@richard-neumann.de>
Date: Tue, 22 Jun 2021 09:59:43 +0200
Subject: [PATCH] Wait for SERVERDATA_AUTH_RESPONSE in async client

---
 rcon/asyncio.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/rcon/asyncio.py b/rcon/asyncio.py
index 46fc496..af3de82 100644
--- a/rcon/asyncio.py
+++ b/rcon/asyncio.py
@@ -4,7 +4,7 @@ from asyncio import open_connection
 from typing import IO
 
 from rcon.exceptions import RequestIdMismatch, WrongPassword
-from rcon.proto import Packet
+from rcon.proto import Packet, Type
 
 
 __all__ = ['rcon']
@@ -26,6 +26,11 @@ async def rcon(command: str, *arguments: str, host: str, port: int,
     login = Packet.make_login(passwd)
     response = await communicate(reader, writer, login)
 
+    # Wait for SERVERDATA_AUTH_RESPONSE according to:
+    # https://developer.valvesoftware.com/wiki/Source_RCON_Protocol
+    while response.type != Type.SERVERDATA_AUTH_RESPONSE:
+        response = await Packet.aread(reader)
+
     if response.id == -1:
         raise WrongPassword()