LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-03 21:24:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix mtp message and container handling

Browse Source
This commit is contained in:
Lonami Exo 2023-08-31 12:26:43 +02:00
parent 9e582ca0fa
commit dfc540c472
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
client/src/telethon/_impl/mtproto/mtp/encrypted.py
View File

@ -4,6 +4,7 @@ import time
from typing import List, Optional, Tuple, Union from typing import List, Optional, Tuple, Union
from ...crypto import AuthKey, decrypt_data_v2, encrypt_data_v2 from ...crypto import AuthKey, decrypt_data_v2, encrypt_data_v2
from ...tl.core.reader import Reader
from ...tl.mtproto.abcs import BadMsgNotification as AbcBadMsgNotification from ...tl.mtproto.abcs import BadMsgNotification as AbcBadMsgNotification
from ...tl.mtproto.abcs import DestroySessionRes from ...tl.mtproto.abcs import DestroySessionRes
from ...tl.mtproto.abcs import MsgDetailedInfo as AbcMsgDetailedInfo from ...tl.mtproto.abcs import MsgDetailedInfo as AbcMsgDetailedInfo
@ -116,6 +117,7 @@ class Encrypted(Mtp):
DestroySessionOk.constructor_id(): self._handle_destroy_session, DestroySessionOk.constructor_id(): self._handle_destroy_session,
DestroySessionNone.constructor_id(): self._handle_destroy_session, DestroySessionNone.constructor_id(): self._handle_destroy_session,
NewSessionCreated.constructor_id(): self._handle_new_session_created, NewSessionCreated.constructor_id(): self._handle_new_session_created,
MsgContainer.constructor_id(): self._handle_container,
GzipPacked.constructor_id(): self._handle_gzip_packed, GzipPacked.constructor_id(): self._handle_gzip_packed,
HttpWait.constructor_id(): self._handle_http_wait, HttpWait.constructor_id(): self._handle_http_wait,
} }
@ -191,9 +193,9 @@ class Encrypted(Mtp):
self._handlers.get(constructor_id, self._handle_update)(message) self._handlers.get(constructor_id, self._handle_update)(message)
def _handle_rpc_result(self, message: Message) -> None: def _handle_rpc_result(self, message: Message) -> None:
assert isinstance(message.body, RpcResult) rpc_result = RpcResult.from_bytes(message.body)
req_msg_id = message.body.req_msg_id req_msg_id = rpc_result.req_msg_id
result = message.body.result result = rpc_result.result
msg_id = MsgId(req_msg_id) msg_id = MsgId(req_msg_id)
inner_constructor = struct.unpack_from("<I", result)[0] inner_constructor = struct.unpack_from("<I", result)[0]
@ -401,7 +403,7 @@ class Encrypted(Mtp):
if client_id != self._client_id: if client_id != self._client_id:
raise RuntimeError("wrong session id") raise RuntimeError("wrong session id")
self._process_message(Message.from_bytes(memoryview(plaintext)[16:])) self._process_message(Message._read_from(Reader(memoryview(plaintext)[16:])))
result = Deserialization(rpc_results=self._rpc_results, updates=self._updates) result = Deserialization(rpc_results=self._rpc_results, updates=self._updates)
self._rpc_results = [] self._rpc_results = []