yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-11 16:44:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: try to handle startup BadMsgNotification

Browse source 
Retry to generate a new msg_id to send before raising BadMsgNotification.
This commit is contained in:
mrmissx 2023-06-30 19:20:03 +07:00
parent 3c11e99d8c
commit 2fc6402cc7
No known key found for this signature in database
GPG key ID: C913C53C78A8911D
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
pyrogram/session/session.py
View file

@ -286,7 +286,7 @@ class Session:
log.info("NetworkTask stopped")
async def send(self, data: TLObject, wait_response: bool = True, timeout: float = WAIT_TIMEOUT):
async def send(self, data: TLObject, wait_response: bool = True, timeout: float = WAIT_TIMEOUT, retry: int = 0):
message = self.msg_factory(data)
msg_id = message.msg_id
@ -330,13 +330,24 @@ class Session:
RPCError.raise_it(result, type(data))
elif isinstance(result, raw.types.BadMsgNotification):
raise BadMsgNotification(result.error_code)
if retry > 1:
raise BadMsgNotification(result.error_code)
self._handle_bad_notification()
await self.send(data, wait_response, timeout, retry + 1)
elif isinstance(result, raw.types.BadServerSalt):
self.salt = result.new_server_salt
return await self.send(data, wait_response, timeout)
else:
return result
def _handle_bad_notification(self):
new_msg_id = MsgId()
if self.stored_msg_ids[0] >= new_msg_id:
new_msg_id = self.stored_msg_ids[0] + 4
log.debug("Changing msg_id old=%s new=%s", self.stored_msg_ids, new_msg_id)
self.stored_msg_ids[0] = new_msg_id
async def invoke(
self,
query: TLObject,