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

Tune stored_msg_ids max size

Browse Source
This commit is contained in:
Dan 2021-12-15 16:53:17 +01:00
parent 2a1af2b8e9
commit c2a29c8c30
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pyrogram/crypto/mtproto.py
View File

@ -26,6 +26,8 @@ from pyrogram.raw.core import Message, Long
from . import aes from . import aes
from ..session.internals import MsgId from ..session.internals import MsgId
STORED_MSG_IDS_MAX_SIZE = 1000 * 2
def kdf(auth_key: bytes, msg_key: bytes, outgoing: bool) -> tuple: def kdf(auth_key: bytes, msg_key: bytes, outgoing: bool) -> tuple:
# https://core.telegram.org/mtproto/description#defining-aes-key-and-initialization-vector # https://core.telegram.org/mtproto/description#defining-aes-key-and-initialization-vector
@ -97,8 +99,8 @@ def unpack(
# https://core.telegram.org/mtproto/security_guidelines#checking-msg-id # https://core.telegram.org/mtproto/security_guidelines#checking-msg-id
assert message.msg_id % 2 != 0 assert message.msg_id % 2 != 0
if len(stored_msg_ids) > 200: if len(stored_msg_ids) > STORED_MSG_IDS_MAX_SIZE:
stored_msg_ids = stored_msg_ids[50:] del stored_msg_ids[:STORED_MSG_IDS_MAX_SIZE // 2]
if stored_msg_ids: if stored_msg_ids:
# Ignored message: msg_id is lower than all of the stored values # Ignored message: msg_id is lower than all of the stored values