yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-03 14:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Lock TCP send()

Browse source
This commit is contained in:
Dan 2018-06-30 11:03:55 +02:00
parent 335a2e06c8
commit 984e989a4b
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pyrogram/connection/transport/tcp/tcp.py
View file

@ -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,6 +78,7 @@ class TCP:
self.socket.close()
async def send(self, data: bytes):
with await self.lock:
self.writer.write(data)
await self.writer.drain()