TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 12:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Read integers first when size matches

Browse Source
This commit is contained in:
Dan 2021-09-25 10:12:12 +02:00 committed by GitHub
parent 362441a74a
commit 428cbf56a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pyrogram/raw/core/primitives/vector.py
View File

@ -31,21 +31,19 @@ class Vector(bytes, TLObject):
# i.e., RpcResult body starts with 0x1cb5c415 (Vector Id) - e.g., messages.GetMessagesViews. # i.e., RpcResult body starts with 0x1cb5c415 (Vector Id) - e.g., messages.GetMessagesViews.
@staticmethod @staticmethod
def read_bare(b: BytesIO, size: int) -> Union[int, Any]: def read_bare(b: BytesIO, size: int) -> Union[int, Any]:
try: if size == 4:
return TLObject.read(b) return Int.read(b)
except KeyError:
b.seek(-4, 1)
if size == 4: if size == 8:
return Int.read(b) return Long.read(b)
else:
return Long.read(b) return TLObject.read(b)
@classmethod @classmethod
def read(cls, data: BytesIO, t: Any = None, *args: Any) -> List: def read(cls, data: BytesIO, t: Any = None, *args: Any) -> List:
count = Int.read(data) count = Int.read(data)
left = len(data.read()) left = len(data.read())
size = (left // count) if count else 0 size = (left / count) if count else 0
data.seek(-left, 1) data.seek(-left, 1)
return List( return List(