mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-27 03:43:45 +03:00
f5f0c84553
Reduce abstraction leaks. Now the transport can hold any state, rather than just the tag. It's also responsible to initialize on the first connection, and they can be cleanly reset. asyncio connections are no longer used, in favour of raw sockets, which should avoid some annoyances. For the time being, more obscure transport modes have been removed, as well as proxy support, until further cleaning is done.
44 lines
1006 B
Python
44 lines
1006 B
Python
from .transport import Transport
|
|
import struct
|
|
|
|
|
|
class Abridged(Transport):
|
|
def __init__(self):
|
|
self._init = False
|
|
|
|
def recreate_fresh(self):
|
|
return type(self)()
|
|
|
|
def pack(self, input: bytes) -> bytes:
|
|
if self._init:
|
|
header = b''
|
|
else:
|
|
header = b'\xef'
|
|
self._init = True
|
|
|
|
length = len(data) >> 2
|
|
if length < 127:
|
|
length = struct.pack('B', length)
|
|
else:
|
|
length = b'\x7f' + int.to_bytes(length, 3, 'little')
|
|
|
|
return header + length + data
|
|
|
|
def unpack(self, input: bytes) -> (int, bytes):
|
|
if len(input) < 4:
|
|
raise EOFError()
|
|
|
|
length = input[0]
|
|
if length < 127:
|
|
offset = 1
|
|
else:
|
|
offset = 4
|
|
length = struct.unpack('<i', input[1:4] + b'\0')[0]
|
|
|
|
length = (length << 2) + offset
|
|
|
|
if len(input) < length:
|
|
raise EOFError()
|
|
|
|
return length, input[offset:length]
|