hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-08 04:52:30 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Avoid storing memoryview in deserialized objects

Browse Source
This commit is contained in:
Lonami Exo
2023-08-31 12:25:00 +02:00
parent 352ccdd4fc
commit 9e582ca0fa
3 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
client/src/telethon/_impl/tl/core/reader.py
View File

@@ -31,17 +31,17 @@ def _bootstrap_get_ty(constructor_id: int) -> Optional[Type["Serializable"]]:
class Reader:
__slots__ = ("_view", "_pos", "_len")
def __init__(self, buffer: bytes) -> None:
def __init__(self, buffer: bytes | memoryview) -> None:
self._view = (
memoryview(buffer) if not isinstance(buffer, memoryview) else buffer
)
self._pos = 0
self._len = len(self._view)
def read_remaining(self) -> bytes:
def read_remaining(self) -> memoryview:
return self.read(self._len - self._pos)
def read(self, n: int) -> bytes:
def read(self, n: int) -> memoryview:
self._pos += n
assert self._pos <= self._len
return self._view[self._pos - n : self._pos]
@@ -52,7 +52,7 @@ class Reader:
assert self._pos <= self._len
return struct.unpack(fmt, self._view[self._pos - size : self._pos])
def read_bytes(self) -> bytes:
def read_bytes(self) -> memoryview:
if self._view[self._pos] == 254:
self._pos += 4
length = struct.unpack("<i", self._view[self._pos - 4 : self._pos])[0] >> 8

4
client/src/telethon/_impl/tl/core/serializable.py
View File

@@ -36,10 +36,6 @@ class Serializable(abc.ABC):
def __repr__(self) -> str:
fields = ((attr, getattr(self, attr)) for attr in self.__slots__)
fields = (
(name, bytes(field) if isinstance(field, memoryview) else field)
for name, field in fields
)
attrs = ", ".join(f"{name}={field!r}" for name, field in fields)
return f"{self.__class__.__name__}({attrs})"