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

Change transports to pack into a write fn

Browse Source
This commit is contained in:
Lonami Exo
2023-08-30 16:37:04 +02:00
parent e12845c38b
commit d5e6dbe36b
7 changed files with 49 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
client/src/telethon/_impl/mtproto/transport/full.py
View File

@@ -1,7 +1,7 @@
import struct
from zlib import crc32
from .abcs import MissingBytes, Transport
from .abcs import MissingBytes, OutFn, Transport
class Full(Transport):
@@ -24,13 +24,15 @@ class Full(Transport):
self._send_seq = 0
self._recv_seq = 0
def pack(self, input: bytes, output: bytearray) -> None:
def pack(self, input: bytes, write: OutFn) -> None:
assert len(input) % 4 == 0
length = len(input) + 12
output += struct.pack("<ii", length, self._send_seq)
output += input
output += struct.pack("<i", crc32(memoryview(output)[-(length - 4) :]))
# Unfortunately there's no hasher that can be updated multiple times,
# so a temporary buffer must be used to hash it all in one go.
tmp = struct.pack("<ii", length, self._send_seq) + input
write(tmp)
write(struct.pack("<I", crc32(tmp)))
self._send_seq += 1
def unpack(self, input: bytes, output: bytearray) -> int: