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

Fix alternative connection modes

Browse Source
This commit is contained in:
Lonami Exo 2018-10-04 17:34:58 +02:00
parent ebde3be820
commit a5d4e97922
3 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
telethon/network/connection/tcpabridged.py
View File

@ -11,7 +11,8 @@ class ConnectionTcpAbridged(Connection):
""" """
async def connect(self, timeout=None, ssl=None): async def connect(self, timeout=None, ssl=None):
await super().connect(timeout=timeout, ssl=ssl) await super().connect(timeout=timeout, ssl=ssl)
await self.send(b'\xef') self._writer.write(b'\xef')
await self._writer.drain()
def _write(self, data): def _write(self, data):
""" """

3
telethon/network/connection/tcpintermediate.py
View File

@ -10,7 +10,8 @@ class ConnectionTcpIntermediate(Connection):
""" """
async def connect(self, timeout=None, ssl=None): async def connect(self, timeout=None, ssl=None):
await super().connect(timeout=timeout, ssl=ssl) await super().connect(timeout=timeout, ssl=ssl)
await self.send(b'\xee\xee\xee\xee') self._writer.write(b'\xee\xee\xee\xee')
await self._writer.drain()
def _send(self, data): def _send(self, data):
self._writer.write(struct.pack('<i', len(data)) + data) self._writer.write(struct.pack('<i', len(data)) + data)

4
telethon/network/connection/tcpobfuscated.py
View File

@ -1,6 +1,5 @@
import os import os
from .connection import Connection
from .tcpabridged import ConnectionTcpAbridged from .tcpabridged import ConnectionTcpAbridged
from ...crypto import AESModeCTR from ...crypto import AESModeCTR
@ -48,4 +47,5 @@ class ConnectionTcpObfuscated(ConnectionTcpAbridged):
self._aes_decrypt = AESModeCTR(decrypt_key, decrypt_iv) self._aes_decrypt = AESModeCTR(decrypt_key, decrypt_iv)
random[56:64] = self._aes_encrypt.encrypt(bytes(random))[56:64] random[56:64] = self._aes_encrypt.encrypt(bytes(random))[56:64]
await self._send(random) self._writer.write(random)
await self._writer.drain()