Restart client after receiving unknown constructor

KurimuzonAkuma · Sep 4, 2024 · 2683e04 · 2683e04
1 parent 86e26cc
commit 2683e04
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/pyrogram/session/session.py b/pyrogram/session/session.py
@@ -187,14 +187,19 @@ async def restart(self):
         await self.start()
 
     async def handle_packet(self, packet):
-        data = await self.loop.run_in_executor(
-            pyrogram.crypto_executor,
-            mtproto.unpack,
-            BytesIO(packet),
-            self.session_id,
-            self.auth_key,
-            self.auth_key_id
-        )
+        try:
+            data = await self.loop.run_in_executor(
+                pyrogram.crypto_executor,
+                mtproto.unpack,
+                BytesIO(packet),
+                self.session_id,
+                self.auth_key,
+                self.auth_key_id
+            )
+        except ValueError as e:
+            log.debug(e)
+            self.loop.create_task(self.restart())
+            return
 
         messages = (
             data.body.messages