hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-08 12:59:46 +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

11
client/tests/transport/full_test.py
View File

@@ -73,6 +73,17 @@ def test_unpack_normal() -> None:
assert output == expected_output
def test_unpack_two_at_once() -> None:
expected_output, transport, input, output = setup_unpack(128)
n = transport.unpack(input + input, output)
assert output == expected_output
assert n == len(input)
with raises(ValueError) as e:
transport.unpack(input, output)
e.match("bad seq")
assert output == expected_output
def test_unpack_twice() -> None:
transport, input, packed = setup_pack(128)
unpacked = bytearray()