make communicate functions thread safe

rcon/battleye/client.py

@@ -2,6 +2,7 @@
 
 from logging import getLogger
 from socket import SOCK_DGRAM
+from threading import Lock
 from typing import Callable
 
 from rcon.battleye.proto import RESPONSE_TYPES
@@ -17,7 +18,7 @@ from rcon.exceptions import WrongPassword
 
 __all__ = ['Client']
 
-
+lock = Lock()
 MessageHandler = Callable[[ServerMessage], None]
 
 
@@ -55,10 +56,11 @@ class Client(BaseClient, socket_type=SOCK_DGRAM):
 
     def communicate(self, request: Request) -> Response:
         """Send a request and receive a response."""
-        with self._socket.makefile('wb') as file:
-            file.write(bytes(request))
+        with lock:
+            with self._socket.makefile('wb') as file:
+                file.write(bytes(request))
 
-        return self.receive()
+            return self.receive()
 
     def login(self, passwd: str) -> bool:
         """Log-in the user."""

rcon/source/client.py

@@ -1,14 +1,16 @@
 """Synchronous client."""
 
 from socket import SOCK_STREAM
+from threading import Lock
 
 from rcon.client import BaseClient
 from rcon.exceptions import SessionTimeout, WrongPassword
 from rcon.source.proto import Packet, Type
 
-
 __all__ = ['Client']
 
+lock = Lock()
+
 
 class Client(BaseClient, socket_type=SOCK_STREAM):
     """An RCON client."""
@@ -31,8 +33,9 @@ class Client(BaseClient, socket_type=SOCK_STREAM):
 
     def communicate(self, packet: Packet) -> Packet:
         """Send and receive a packet."""
-        self.send(packet)
-        return self.read()
+        with lock:
+            self.send(packet)
+            return self.read()
 
     def send(self, packet: Packet) -> None:
         """Send a packet to the server."""