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

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/intermediate_test.py
View File

@@ -50,3 +50,12 @@ def test_unpack_normal() -> None:
transport.pack(input, packed)
transport.unpack(packed[4:], unpacked)
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[4:] + packed[4:], unpacked)
assert input == unpacked
assert n == len(packed[4:])