mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-23 09:53:45 +03:00
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
import hashlib
|
|
|
|
from .tcpobfuscated import ConnectionTcpObfuscated
|
|
from ...crypto import AESModeCTR
|
|
|
|
|
|
class ConnectionTcpMTProxy(ConnectionTcpObfuscated):
|
|
"""
|
|
Wrapper around the "obfuscated2" mode that modifies it a little and allows
|
|
user to connect to the Telegram proxy servers commonly known as MTProxy.
|
|
Implemented very ugly due to the leaky abstractions in Telethon networking
|
|
classes that should be refactored later (TODO).
|
|
"""
|
|
@staticmethod
|
|
def address_info(proxy_info):
|
|
if proxy_info is None:
|
|
raise ValueError("No proxy info specified for MTProxy connection")
|
|
return proxy_info[:2]
|
|
|
|
def __init__(self, ip, port, dc_id, *, loop, loggers, proxy=None):
|
|
proxy_host, proxy_port = self.address_info(proxy)
|
|
super().__init__(
|
|
proxy_host, proxy_port, dc_id, loop=loop, loggers=loggers)
|
|
|
|
# TODO: Implement the dd-secret secure mode (adds noise to fool DPI)
|
|
self._secret = bytes.fromhex(proxy[2])
|
|
if len(self._secret) != 16:
|
|
raise ValueError(
|
|
"MTProxy secure mode is not implemented for now"
|
|
if len(self._secret) == 17 and self._secret[0] == 0xDD else
|
|
"MTProxy secret must be a hex-string representing 16 bytes"
|
|
)
|
|
|
|
def _compose_key(self, data):
|
|
return hashlib.sha256(data + self._secret).digest()
|
|
|
|
def _compose_tail(self, data):
|
|
dc_id_bytes = self._dc_id.to_bytes(2, "little", signed=True)
|
|
return super()._compose_tail(data[:60] + dc_id_bytes + data[62:])
|