Fix alternative connection modes

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

View File

@ -11,7 +11,8 @@ class ConnectionTcpAbridged(Connection):
"""
async def connect(self, timeout=None, ssl=None):
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):
"""

View File

@ -10,7 +10,8 @@ class ConnectionTcpIntermediate(Connection):
"""
async def connect(self, timeout=None, ssl=None):
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):
self._writer.write(struct.pack('<i', len(data)) + data)

View File

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