TL schema parsing. PQ factorization
TODO: dynamic methods and constructors generator
This commit is contained in:
Anton Grigoryev
parent
57afc246ee
commit
7d22608fbd
111
mtproto.py
111
mtproto.py
|
|
@ -4,12 +4,13 @@ Created on Tue Sep 2 19:26:15 2014
|
|||
|
||||
@author: agrigoryev
|
||||
"""
|
||||
import binascii
|
||||
from binascii import crc32
|
||||
import struct
|
||||
import socket
|
||||
import re
|
||||
from datetime import datetime
|
||||
import sys
|
||||
import io
|
||||
|
||||
current_module = sys.modules[__name__]
|
||||
|
||||
|
|
@ -18,33 +19,36 @@ def vis(bs):
|
|||
n = len(bs) // 8
|
||||
for i in range(n):
|
||||
print(" ".join(["%02X" % b for b in bs[i*8:i*8+8]]))
|
||||
print(" ".join(["%02X" % b for b in bs[i*8+8:]]))
|
||||
print(" ".join(["%02X" % b for b in bs[i*8+8:]])+"\n")
|
||||
|
||||
|
||||
class TlElement:
|
||||
def __init__(self, type_string):
|
||||
tl_re = re.compile("""([a-z]\w*) #name
|
||||
\#([0-9a-f]{8}) #id
|
||||
\s+(.*)\s+ #arguments list
|
||||
=\s+([A-Z]\w*); #result""", re.X)
|
||||
tl_re = re.compile("""([a-z]\w*) #name
|
||||
\#?([0-9a-f]{8})?\s+ #id
|
||||
(\{.*\}\s+)? #subtype
|
||||
(.*)\s+ #arguments list
|
||||
=\s+([A-Z]\w*) #result
|
||||
(\s+(\w+))?; #subresult""", re.X)
|
||||
assert isinstance(type_string, str)
|
||||
x = tl_re.match(type_string)
|
||||
if x is not None:
|
||||
self.name = x.groups()[0]
|
||||
self.id = x.groups()[1]
|
||||
self.args = self.get_arg_list(x.groups()[2])
|
||||
self.result = x.groups()[3]
|
||||
if x.groups()[1] is not None:
|
||||
self.id = int(x.groups()[1], 16)
|
||||
else:
|
||||
self.id = crc32(re.sub("(#[0-9a-f]{8})|([;\n{}])", "", type_string).encode())
|
||||
self.args = self.get_arg_list(x.groups()[3])
|
||||
self.result = x.groups()[4]
|
||||
else:
|
||||
raise SyntaxError
|
||||
|
||||
@staticmethod
|
||||
def get_arg_list(arg_string):
|
||||
arg_re = re.compile("(\w+):(\w+)(<(\w+)>)?")
|
||||
arg_re = re.compile("([\w0-9]+)?:?([\w0-9]+)(<([\w0-0]+)>)?")
|
||||
res = []
|
||||
for s in arg_re.findall(arg_string):
|
||||
d = {'name': s[0], 'type': s[1]}
|
||||
if s[2]:
|
||||
d['subtype'] = s[3]
|
||||
d = {'name': s[0], 'type': s[1], 'subtype': s[3] if s[2] is not None else None}
|
||||
res.append(d)
|
||||
return res
|
||||
|
||||
|
|
@ -65,7 +69,7 @@ class TL:
|
|||
try:
|
||||
z = TlElement(line)
|
||||
self.obj_dict_id[z.id] = z
|
||||
self.obj_dict_name[z.name] = z
|
||||
self.obj_dict_name[z.result] = z
|
||||
except SyntaxError:
|
||||
pass
|
||||
|
||||
|
|
@ -81,6 +85,7 @@ class TL:
|
|||
except SyntaxError:
|
||||
pass
|
||||
|
||||
|
||||
def tl_serialize(self, elem, kwargs):
|
||||
assert isinstance(elem, TlElement)
|
||||
for arg in elem.args:
|
||||
|
|
@ -94,6 +99,69 @@ class TL:
|
|||
answer = session.recv_message()
|
||||
return deserialize(answer, TL)
|
||||
|
||||
def class_generator(self, tl_element):
|
||||
class Dummy:
|
||||
def __init__(self, bstring):
|
||||
assert isinstance(bstring, bytes)
|
||||
f = io.BytesIO(bstring)
|
||||
dict = self.deserialize(f, type=tl_element.type)
|
||||
|
||||
|
||||
def deserialize(self, bytes_io, type=None, subtype=None):
|
||||
assert isinstance(bytes_io, io.BytesIO)
|
||||
if type == 'int':
|
||||
x = struct.unpack('<i', bytes_io.read(4))[0]
|
||||
elif type == '#':
|
||||
x = struct.unpack('<I', bytes_io.read(4))[0]
|
||||
elif type == 'long':
|
||||
x = struct.unpack('<q', bytes_io.read(8))[0]
|
||||
elif type == 'double':
|
||||
x = struct.unpack('<d', bytes_io.read(8))[0]
|
||||
elif type == 'int128':
|
||||
t = struct.unpack('<16s', bytes_io.read(16))[0]
|
||||
x = int.from_bytes(t, 'little')
|
||||
elif type == 'int256':
|
||||
t = struct.unpack('<32s', bytes_io.read(32))[0]
|
||||
x = int.from_bytes(t, 'little')
|
||||
elif type == 'bytes':
|
||||
l = int.from_bytes(bytes_io.read(1), 'little')
|
||||
x = bytes_io.read(l)
|
||||
bytes_io.read(-(l+1) % 4) # skip padding bytes
|
||||
elif type == 'string':
|
||||
l = int.from_bytes(bytes_io.read(1), 'little')
|
||||
assert l <=254
|
||||
if l == 254:
|
||||
# We have a long string
|
||||
long_len = int.from_bytes(bytes_io.read(3), 'little')
|
||||
x = bytes_io.read(long_len)
|
||||
bytes_io.read(-long_len % 4) # skip padding bytes
|
||||
else:
|
||||
# We have a short string
|
||||
x = bytes_io.read(l)
|
||||
bytes_io.read(-(l+1) % 4) # skip padding bytes
|
||||
assert isinstance(x, bytes)
|
||||
elif type == 'vector':
|
||||
assert subtype is not None
|
||||
count = int.from_bytes(bytes_io.read(4), 'little')
|
||||
x = [self.deserialize(bytes_io, type=subtype) for i in range(count)]
|
||||
else:
|
||||
# Detect what type
|
||||
i = struct.unpack('<i', bytes_io.read(4))[0]
|
||||
try:
|
||||
tl_elem = self.obj_dict_id[i]
|
||||
except:
|
||||
raise Exception("Could not extract type: %s" % type)
|
||||
base_boxed_types = ["Vector", "Int", "Long", "Double", "String", "Int128", "Int256"]
|
||||
if tl_elem.result in base_boxed_types:
|
||||
x = self.deserialize(bytes_io, type=tl_elem.name, subtype=subtype)
|
||||
|
||||
else: # other types
|
||||
x = {}
|
||||
for arg in tl_elem.args:
|
||||
x[arg['name']] = self.deserialize(bytes_io, type=arg['type'], subtype=arg['subtype'])
|
||||
return x
|
||||
|
||||
|
||||
|
||||
class Session:
|
||||
def __init__(self, IP_adress, port):
|
||||
|
|
@ -109,23 +177,26 @@ class Session:
|
|||
struct.pack('<L', len(message)))
|
||||
|
||||
def send_message(self, message):
|
||||
print('>>')
|
||||
vis(message)
|
||||
print()
|
||||
#self.number += 1
|
||||
data = self.header(message) + message
|
||||
step1 = struct.pack('<LL', len(data)+12, self.number) + data
|
||||
step2 = step1 + struct.pack('<L', binascii.crc32(step1))
|
||||
vis(step2)
|
||||
step2 = step1 + struct.pack('<L', crc32(step1))
|
||||
self.sock.send(step2)
|
||||
|
||||
|
||||
def recv_message(self):
|
||||
packet_length_data = self.sock.recv(4)
|
||||
if len(packet_length_data) > 0: # if we have smth. in the socket
|
||||
packet_length = struct.unpack("<L", packet_length_data)[0]
|
||||
packet = self.sock.recv(packet_length - 4)
|
||||
self.number = struct.unpack("<L", packet[0:4])[0]
|
||||
data = packet[4:-4]
|
||||
auth_key_id = struct.unpack("<8s", packet[4:12])[0]
|
||||
message_id = struct.unpack("<8s", packet[12:20])[0]
|
||||
message_length = struct.unpack("<I", packet[20:24])[0]
|
||||
data = packet[24:24+message_length]
|
||||
crc = packet[-4:]
|
||||
if binascii.crc32(packet_length_data + packet[0:-4]).to_bytes(4,'little') == crc:
|
||||
if crc32(packet_length_data + packet[0:-4]).to_bytes(4,'little') == crc:
|
||||
print('<<')
|
||||
vis(data)
|
||||
return data
|
||||
|
|
|
|||
Loading…
Reference in New Issue