LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-11 03:56:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Don't send requests until there are salts

Browse Source 
This way we don't waste resources sending containers that may fail.
This commit is contained in:
Lonami Exo 2023-10-18 20:35:55 +02:00
parent 45f866c335
commit c49f0827b1
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
client/src/telethon/_impl/mtproto/mtp/encrypted.py
View File

@ -422,10 +422,6 @@ class Encrypted(Mtp):
) )
def push(self, request: bytes) -> Optional[MsgId]: def push(self, request: bytes) -> Optional[MsgId]:
if self._in_pending_ack:
self._serialize_msg(bytes(MsgsAck(msg_ids=self._in_pending_ack)), False)
self._in_pending_ack = []
if self._start_salt_time and len(self._salts) >= 2: if self._start_salt_time and len(self._salts) >= 2:
start_secs, start_instant = self._start_salt_time start_secs, start_instant = self._start_salt_time
salt = self._salts[-2] salt = self._salts[-2]
@ -434,6 +430,13 @@ class Encrypted(Mtp):
self._salts.pop() self._salts.pop()
self._try_request_salts() self._try_request_salts()
if self._salt_request_msg_id:
# Don't add anything else to the container while we still need new salts.
return None
if self._in_pending_ack:
self._serialize_msg(bytes(MsgsAck(msg_ids=self._in_pending_ack)), False)
self._in_pending_ack = []
if self._msg_count >= CONTAINER_MAX_LENGTH: if self._msg_count >= CONTAINER_MAX_LENGTH:
return None return None