hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-20 20:16:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support for TCP intermediate mode (#112)

Browse Source
This commit is contained in:
Lonami Exo 2017-08-29 11:39:44 +02:00
parent a3c2c462a7
commit cff7655a10
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
telethon/network/connection.py
View File

@ -9,14 +9,15 @@ from ..errors import InvalidChecksumError
class Connection: class Connection:
"""Represents an abstract connection (TCP, TCP abridged...). """Represents an abstract connection (TCP, TCP abridged...).
'mode' may be any of 'tcp_full', 'tcp_abridged'. 'mode' may be any of:
'tcp_full', 'tcp_intermediate', 'tcp_abridged', 'tcp_obfuscated'
Note that '.send()' and '.recv()' refer to messages, which Note that '.send()' and '.recv()' refer to messages, which
will be packed accordingly, whereas '.write()' and '.read()' will be packed accordingly, whereas '.write()' and '.read()'
work on plain bytes, with no further additions. work on plain bytes, with no further additions.
""" """
def __init__(self, ip, port, mode='tcp_obfuscated', def __init__(self, ip, port, mode='tcp_intermediate',
proxy=None, timeout=timedelta(seconds=5)): proxy=None, timeout=timedelta(seconds=5)):
self.ip = ip self.ip = ip
self.port = port self.port = port
@ -34,6 +35,10 @@ class Connection:
setattr(self, 'send', self._send_tcp_full) setattr(self, 'send', self._send_tcp_full)
setattr(self, 'recv', self._recv_tcp_full) setattr(self, 'recv', self._recv_tcp_full)
elif mode == 'tcp_intermediate':
setattr(self, 'send', self._send_intermediate)
setattr(self, 'recv', self._recv_intermediate)
elif mode in ('tcp_abridged', 'tcp_obfuscated'): elif mode in ('tcp_abridged', 'tcp_obfuscated'):
setattr(self, 'send', self._send_abridged) setattr(self, 'send', self._send_abridged)
setattr(self, 'recv', self._recv_abridged) setattr(self, 'recv', self._recv_abridged)
@ -52,7 +57,9 @@ class Connection:
timeout=round(self.timeout.seconds)) timeout=round(self.timeout.seconds))
if self._mode == 'tcp_abridged': if self._mode == 'tcp_abridged':
self.conn.write(int.to_bytes(239, 1, 'little')) self.conn.write(b'\xef')
elif self._mode == 'tcp_intermediate':
self.conn.write(b'\xee\xee\xee\xee')
elif self._mode == 'tcp_obfuscated': elif self._mode == 'tcp_obfuscated':
self._setup_obfuscation() self._setup_obfuscation()
@ -122,6 +129,9 @@ class Connection:
return body return body
def _recv_intermediate(self, **kwargs):
return self.read(int.from_bytes(self.read(4), 'little'))
def _recv_abridged(self, **kwargs): def _recv_abridged(self, **kwargs):
length = int.from_bytes(self.read(1), 'little') length = int.from_bytes(self.read(1), 'little')
if length >= 127: if length >= 127:
@ -149,6 +159,12 @@ class Connection:
self._send_counter += 1 self._send_counter += 1
self.write(writer.get_bytes()) self.write(writer.get_bytes())
def _send_intermediate(self, message):
with BinaryWriter() as writer:
writer.write_int(len(message))
writer.write(message)
self.write(writer.get_bytes())
def _send_abridged(self, message): def _send_abridged(self, message):
with BinaryWriter() as writer: with BinaryWriter() as writer:
length = len(message) >> 2 length = len(message) >> 2