1

我试图用python实现电报api。

基本代码分叉形式https://github.com/datamachine/twx

我想添加 checkPhone 和 sendCode 功能，但服务器总是返回无法理解的结果。

这是我的代码

import os
import struct
from io import BytesIO
from Crypto.Hash import SHA
from twx.mtproto.mtproto import MTProto as MTProtoParent, crc32
from twx.mtproto import rpc
from twx.mtproto import crypt
from hexdump import hexdump


class sendCode(rpc.TLObject):
    constructor = 0x768d5f4d

    def __init__(self, phone_number, sms_type, api_id, api_hash, lang_code):
        self.phone_number = phone_number  # bytes
        self.sms_type = sms_type  # integer
        self.api_id = int(api_id)  # integer
        self.api_hash = api_hash  # string
        self.lang_code = lang_code  # string

    def get_bytes(self):
        phone_number_io = BytesIO()
        rpc.serialize_string(phone_number_io, self.phone_number)

        params_io = BytesIO()
        rpc.serialize_string(params_io, self.api_hash.encode())
        rpc.serialize_string(params_io, self.lang_code.encode())

        return struct.pack("<I16sII40s", sendCode.constructor, phone_number_io.getvalue(), self.sms_type, self.api_id, params_io.getvalue())


class MTProto(MTProtoParent):
    def __init__(self, api_secret, api_id):
        MTProtoParent.__init__(self, api_secret, api_id)

    def get_encrypted_message(self, message_data, debug=False):
        salt = self.dc.server_salt
        session_id = os.urandom(8)
        message_id = self.dc.generate_message_id()
        seq_no = self.dc.number
        payload = message_data
        to_be_encrypted = struct.pack('<8s8sQHI', salt, session_id, message_id, seq_no, len(payload)) + payload
        to_be_encrypted_padded = to_be_encrypted + os.urandom(-len(to_be_encrypted) % 16)

        if debug:
            print("\nin get encrypted message")
            print("\nto be encrypted data:")
            hexdump(to_be_encrypted_padded)

        msg_key = SHA.new(to_be_encrypted_padded).digest()[-16:]
        auth_key_id = self.dc.auth_key_id
        auth_key = self.dc.auth_key

        # to get 256bit AES Key and AES IV
        sha1_a = SHA.new(msg_key + auth_key[0:32]).digest()
        sha1_b = SHA.new(auth_key[32:48] + msg_key + auth_key[48:64]).digest()
        sha1_c = SHA.new(auth_key[64:96] + msg_key).digest()
        sha1_d = SHA.new(msg_key + auth_key[96:128]).digest()
        aes_key = sha1_a[0:8] + sha1_b[8:20] + sha1_c[4:16]
        aes_iv = sha1_a[8:20] + sha1_b[0:8] + sha1_c[16:20] + sha1_d[0:8]

        # generate encrypted data
        encrypted_data = crypt.ige_encrypt(to_be_encrypted_padded, aes_key, aes_iv)

        message = struct.pack('<8s16s', auth_key_id, msg_key) + encrypted_data

        if debug:
            print("\nencrypted data:")
            hexdump(message)
        return message

    def send_encrypted(self, message_data, debug=False):
        message = self.get_encrypted_message(message_data, debug=False)

        # message = struct.pack('<II', len(message) + 12, self.dc.number) + message
        message = struct.pack('<II', len(message) + 12, self.dc.number) + message
        msg_chksum = crc32(message)
        message += struct.pack('<I', msg_chksum)

        if debug:
            hexdump(message)

        self.dc.sock.send(message)
        self.dc.number += 1

        # self.dc.send_message(message)

    def checkPhone(self, phone_number):
        bytes_io = BytesIO()
        rpc.serialize_string(bytes_io, phone_number)
        query = rpc.checkPhone(bytes_io.getvalue()).get_bytes()
        # hexdump(query)
        self.send_encrypted(query, debug=True)
        # self.dc.send_message(query)
        print(rpc.checkedPhone().set_params(self.dc.recv_message()))

    def sendCode(self, phone_number, sms_type, lang_code):
        query = sendCode(phone_number, sms_type, self.api_id, self.api_secret, lang_code).get_bytes()
        # hexdump(query)
        self.send_encrypted(query, debug=True)
        # self.dc.send_message(query)
        self.dc.recv_message(debug=True)


if __name__ == "__main__":
    mtp = MTProto("api_secret", "api_key")
    mtp.dc.create_auth_key()
    mtp.checkPhone(b"821011112222")

并且服务器总是响应如下，无论 auth_key_id 是什么，

00000000: 10 00 00 00 03 00 00 00  6C FE FF FF EE 28 CE 89

我的代码有什么问题？

4

0 回答 0