Lock TCP send()

delivrance · delivrance · commit 984e989a4b03 · 2018-06-30T11:03:55.000+02:00
diff --git a/pyrogram/connection/transport/tcp/tcp.py b/pyrogram/connection/transport/tcp/tcp.py
@@ -39,6 +39,8 @@ class TCP:
     def __init__(self, proxy: dict):
         self.proxy = proxy
 
+        self.lock = asyncio.Lock()
+
         self.socket = socks.socksocket()
         self.reader = None  # type: asyncio.StreamReader
         self.writer = None  # type: asyncio.StreamWriter
@@ -76,8 +78,9 @@ def close(self):
                 self.socket.close()
 
     async def send(self, data: bytes):
-        self.writer.write(data)
-        await self.writer.drain()
+        with await self.lock:
+            self.writer.write(data)
+            await self.writer.drain()
 
     async def recv(self, length: int = 0):
         data = b""
