Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Raise directly when not checking a boolean expression

Browse Source
This commit is contained in:
Dan 2021-12-16 21:39:52 +01:00
parent 8aa358129c
commit ea3281b5f6
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pyrogram/crypto/mtproto.py
View File

@ -106,21 +106,21 @@ def unpack(
if stored_msg_ids:
# Ignored message: msg_id is lower than all of the stored values
if message.msg_id < stored_msg_ids[0]:
SecurityCheckMismatch.check(False)
raise SecurityCheckMismatch
# Ignored message: msg_id is equal to any of the stored values
if message.msg_id in stored_msg_ids:
SecurityCheckMismatch.check(False)
raise SecurityCheckMismatch
time_diff = (message.msg_id - MsgId()) / 2 ** 32
# Ignored message: msg_id belongs over 30 seconds in the future
if time_diff > 30:
SecurityCheckMismatch.check(False)
raise SecurityCheckMismatch
# Ignored message: msg_id belongs over 300 seconds in the past
if time_diff < -300:
SecurityCheckMismatch.check(False)
raise SecurityCheckMismatch
bisect.insort(stored_msg_ids, message.msg_id)