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

fix: try to handle startup BadMsgNotification #11

Browse source 
Retry to generate a new msg_id to send before raising BadMsgNotification.

Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
mrmissx 2023-06-30 19:20:03 +07:00 committed by wulan17
parent e80ebe1a9e
commit ed4cbc34ca
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5
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[len(self.stored_msg_ids)-1] >= new_msg_id:
new_msg_id = self.stored_msg_ids[len(self.stored_msg_ids)-1] + 4
log.debug("Changing msg_id old=%s new=%s", self.stored_msg_ids[len(self.stored_msg_ids)-1], new_msg_id)
self.stored_msg_ids[len(self.stored_msg_ids)-1] = new_msg_id
async def invoke(
self,
query: TLObject,