mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-27 03:43:45 +03:00
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
|
import errno
|
||
|
|
import struct
|
||
|
|
from datetime import timedelta
|
||
|
|
from zlib import crc32
|
||
|
|
|
||
|
|
from .common import Connection
|
||
|
|
from ...errors import InvalidChecksumError
|
||
|
|
from ...extensions import TcpClient
|
||
|
|
|
||
|
|
|
||
|
|
class ConnectionTcpFull(Connection):
|
||
|
|
"""
|
||
|
|
Default Telegram mode. Sends 12 additional bytes and
|
||
|
|
needs to calculate the CRC value of the packet itself.
|
||
|
|
"""
|
||
|
|
def __init__(self, proxy=None, timeout=timedelta(seconds=5)):
|
||
|
|
super().__init__(proxy, timeout)
|
||
|
|
self._send_counter = 0
|
||
|
|
self.conn = TcpClient(proxy=self._proxy, timeout=self._timeout)
|
||
|
|
self.read = self.conn.read
|
||
|
|
self.write = self.conn.write
|
||
|
|
|
||
|
|
def connect(self, ip, port):
|
||
|
|
try:
|
||
|
|
self.conn.connect(ip, port)
|
||
|
|
except OSError as e:
|
||
|
|
if e.errno == errno.EISCONN:
|
||
|
|
return # Already connected, no need to re-set everything up
|
||
|
|
else:
|
||
|
|
raise
|
||
|
|
|
||
|
|
self._send_counter = 0
|
||
|
|
|
||
|
|
def get_timeout(self):
|
||
|
|
return self.conn.timeout
|
||
|
|
|
||
|
|
def is_connected(self):
|
||
|
|
return self.conn.connected
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
self.conn.close()
|
||
|
|
|
||
|
|
def clone(self):
|
||
|
|
return ConnectionTcpFull(self._proxy, self._timeout)
|
||
|
|
|
||
|
|
def recv(self):
|
||
|
|
packet_len_seq = self.read(8) # 4 and 4
|
||
|
|
packet_len, seq = struct.unpack('<ii', packet_len_seq)
|
||
|
|
body = self.read(packet_len - 12)
|
||
|
|
checksum = struct.unpack('<I', self.read(4))[0]
|
||
|
|
|
||
|
|
valid_checksum = crc32(packet_len_seq + body)
|
||
|
|
if checksum != valid_checksum:
|
||
|
|
raise InvalidChecksumError(checksum, valid_checksum)
|
||
|
|
|
||
|
|
return body
|
||
|
|
|
||
|
|
def send(self, message):
|
||
|
|
# https://core.telegram.org/mtproto#tcp-transport
|
||
|
|
# total length, sequence number, packet and checksum (CRC32)
|
||
|
|
length = len(message) + 12
|
||
|
|
data = struct.pack('<ii', length, self._send_counter) + message
|
||
|
|
crc = struct.pack('<I', crc32(data))
|
||
|
|
self._send_counter += 1
|
||
|
|
self.write(data + crc)
|