LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-06 05:00:23 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add a lock around connection.recv()

Browse Source
This commit is contained in:
Lonami Exo 2018-02-10 12:44:09 +01:00
parent 50515aa528
commit d8376ee50d
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
telethon/network/mtproto_sender.py
View File

@ -49,6 +49,7 @@ class MtProtoSender:
self.session = session self.session = session
self.connection = connection self.connection = connection
self._loop = loop if loop else asyncio.get_event_loop() self._loop = loop if loop else asyncio.get_event_loop()
self._recv_lock = asyncio.Lock()
# Requests (as msg_id: Message) sent waiting to be received # Requests (as msg_id: Message) sent waiting to be received
self._pending_receive = {} self._pending_receive = {}
@ -123,7 +124,10 @@ class MtProtoSender:
Update and Updates objects. Update and Updates objects.
""" """
try: try:
body = await self.connection.recv() with await self._recv_lock:
# Receiving items is not an "atomic" operation since we
# need to read the length and then upcoming parts separated.
body = await self.connection.recv()
except (BufferError, InvalidChecksumError): except (BufferError, InvalidChecksumError):
# TODO BufferError, we should spot the cause... # TODO BufferError, we should spot the cause...
# "No more bytes left"; something wrong happened, clear # "No more bytes left"; something wrong happened, clear