mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-25 02:43:45 +03:00
1dac866118
The initial release contains the most basic implementation of TLSharp core. This is also fully untested, since no test can be done until more work is done.
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
|
|
import time
|
|
from utils.binary_writer import BinaryWriter
|
|
from utils.binary_reader import BinaryReader
|
|
|
|
|
|
class MtProtoPlainSender:
|
|
|
|
def __init__(self, transport):
|
|
self._sequence = 0
|
|
self._time_offset = 0
|
|
self._last_msg_id = 0
|
|
self._transport = transport
|
|
|
|
def send(self, data):
|
|
with BinaryWriter() as writer:
|
|
writer.write_long(0)
|
|
writer.write_int(self.get_new_msg_id())
|
|
writer.write_int(len(data))
|
|
writer.write(data)
|
|
|
|
packet = writer.get_bytes()
|
|
self._transport.send(packet)
|
|
|
|
def receive(self):
|
|
result = self._transport.receive()
|
|
with BinaryReader(result.body) as reader:
|
|
auth_key_id = reader.read_long()
|
|
message_id = reader.read_long()
|
|
message_length = reader.read_int()
|
|
|
|
response = reader.read(message_length)
|
|
return response
|
|
|
|
def get_new_msg_id(self):
|
|
new_msg_id = int(self._time_offset + time.time() * 1000) # multiply by 1000 to get milliseconds
|
|
|
|
if self._last_msg_id >= new_msg_id:
|
|
new_msg_id = self._last_msg_id + 4
|
|
|
|
self._last_msg_id = new_msg_id
|
|
return new_msg_id
|