LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-03-13 07:55:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Another attempt to prevent duplicates

Browse Source
This commit is contained in:
Andrey Egorov 2017-12-09 21:24:13 +03:00
parent 984f483b98
commit 7c0af2c080
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
telethon/network/mtproto_sender.py
View File

@ -15,8 +15,8 @@ from ..tl import TLMessage, MessageContainer, GzipPacked
from ..tl.all_tlobjects import tlobjects from ..tl.all_tlobjects import tlobjects
from ..tl.types import ( from ..tl.types import (
MsgsAck, Pong, BadServerSalt, BadMsgNotification, MsgsAck, Pong, BadServerSalt, BadMsgNotification,
MsgNewDetailedInfo, NewSessionCreated, MsgDetailedInfo, MsgNewDetailedInfo, MsgDetailedInfo, RpcError,
RpcError) MsgsStateReq, MsgsStateInfo, MsgsAllInfo, MsgResendReq)
from ..tl.functions.auth import LogOutRequest from ..tl.functions.auth import LogOutRequest
logging.getLogger(__name__).addHandler(logging.NullHandler()) logging.getLogger(__name__).addHandler(logging.NullHandler())
@ -121,7 +121,6 @@ class MtProtoSender:
message, remote_msg_id, remote_seq = self._decode_msg(body) message, remote_msg_id, remote_seq = self._decode_msg(body)
with BinaryReader(message) as reader: with BinaryReader(message) as reader:
await self._process_msg(remote_msg_id, remote_seq, reader) await self._process_msg(remote_msg_id, remote_seq, reader)
await self._send_acknowledge(remote_msg_id)
# endregion # endregion
@ -191,6 +190,7 @@ class MtProtoSender:
# The following codes are "parsed manually" # The following codes are "parsed manually"
if code == 0xf35c6d01: # rpc_result, (response of an RPC call) if code == 0xf35c6d01: # rpc_result, (response of an RPC call)
await self._send_acknowledge(msg_id)
return await self._handle_rpc_result(msg_id, sequence, reader) return await self._handle_rpc_result(msg_id, sequence, reader)
if code == Pong.CONSTRUCTOR_ID: if code == Pong.CONSTRUCTOR_ID:
@ -205,6 +205,14 @@ class MtProtoSender:
if code == BadServerSalt.CONSTRUCTOR_ID: if code == BadServerSalt.CONSTRUCTOR_ID:
return await self._handle_bad_server_salt(msg_id, sequence, reader) return await self._handle_bad_server_salt(msg_id, sequence, reader)
if code in (MsgsStateReq.CONSTRUCTOR_ID, MsgResendReq.CONSTRUCTOR_ID):
# just answer we don't know anything
return await self._handle_msgs_state_forgotten(msg_id, sequence, reader)
if code == MsgsAllInfo.CONSTRUCTOR_ID:
# not interesting now
return True
if code == BadMsgNotification.CONSTRUCTOR_ID: if code == BadMsgNotification.CONSTRUCTOR_ID:
return await self._handle_bad_msg_notification(msg_id, sequence, reader) return await self._handle_bad_msg_notification(msg_id, sequence, reader)
@ -231,6 +239,7 @@ class MtProtoSender:
# If the code is not parsed manually then it should be a TLObject. # If the code is not parsed manually then it should be a TLObject.
if code in tlobjects: if code in tlobjects:
await self._send_acknowledge(msg_id)
result = reader.tgread_object() result = reader.tgread_object()
self.updates_handler(result) self.updates_handler(result)
return True return True
@ -339,6 +348,11 @@ class MtProtoSender:
return True return True
async def _handle_msgs_state_forgotten(self, msg_id, sequence, reader):
req = reader.tgread_object()
await self._send_message(TLMessage(self.session, MsgsStateInfo(msg_id, chr(1) * len(req.msg_ids))))
return True
async def _handle_bad_msg_notification(self, msg_id, sequence, reader): async def _handle_bad_msg_notification(self, msg_id, sequence, reader):
self._logger.debug('Handling bad message notification') self._logger.debug('Handling bad message notification')
bad_msg = reader.tgread_object() bad_msg = reader.tgread_object()