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

Fix full transport overreading

Browse Source
This commit is contained in:
Lonami Exo
2023-09-02 01:15:10 +02:00
parent 5e43efc55d
commit 49c0645fcb
4 changed files with 31 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
client/tests/transport/abridged_test.py
View File

@@ -59,6 +59,15 @@ def test_unpack_normal() -> None:
assert input == unpacked
def test_unpack_two_at_once() -> None:
transport, input, packed = setup_pack(128)
unpacked = bytearray()
transport.pack(input, packed)
n = transport.unpack(packed[1:] + packed[1:], unpacked)
assert input == unpacked
assert n == len(packed[1:])
def unpack_large() -> None:
transport, input, packed = setup_pack(1024)
unpacked = bytearray()