LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-29 12:53:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update mtprotoplainsender.py asserts

Browse Source
This commit is contained in:
Lonami Exo 2018-06-28 09:48:03 +02:00
parent 9bb5cfd871
commit fb40e7b508
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
telethon/network/mtprotoplainsender.py
View File

@ -39,10 +39,15 @@ class MTProtoPlainSender:
raise BrokenAuthKeyError() raise BrokenAuthKeyError()
with BinaryReader(body) as reader: with BinaryReader(body) as reader:
assert reader.read_long() == 0 # auth_key_id assert reader.read_long() == 0, 'Bad auth_key_id' # auth_key_id
assert reader.read_long() > msg_id # msg_id
assert reader.read_int() # length
assert reader.read_long() != 0, 'Bad msg_id' # msg_id
# ^ We should make sure that the read ``msg_id`` is greater
# than our own ``msg_id``. However, under some circumstances
# (bad system clock/working behind proxies) this seems to not
# be the case, which would cause endless assertion errors.
assert reader.read_int() > 0, 'Bad length' # length
# We could read length bytes and use those in a new reader to read # We could read length bytes and use those in a new reader to read
# the next TLObject without including the padding, but since the # the next TLObject without including the padding, but since the
# reader isn't used for anything else after this, it's unnecessary. # reader isn't used for anything else after this, it's unnecessary.