Better error message in case of empty data in buffer

delivrance · delivrance · commit e4f99df07c2f · 2021-04-26T16:09:39.000+02:00
diff --git a/pyrogram/crypto/mtproto.py b/pyrogram/crypto/mtproto.py
@@ -63,6 +63,9 @@ def unpack(b: BytesIO, session_id: bytes, auth_key: bytes, auth_key_id: bytes) -
     try:
         message = Message.read(data)
     except KeyError as e:
+        if e.args[0] == 0:
+            raise ConnectionError(f"Received empty data. Check your internet connection.")
+
         left = data.read().hex()
 
         left = [left[i:i + 64] for i in range(0, len(left), 64)]
