Telethon/crypto/auth_key.py

# This file is based on TLSharp
# https://github.com/sochix/TLSharp/blob/master/TLSharp.Core/MTProto/Crypto/AuthKey.cs
import utils
from errors import *
from utils import BinaryWriter, BinaryReader


class AuthKey:
    def __init__(self, gab=None, data=None):
        if gab:
            self.key = utils.get_byte_array(gab, signed=False)
        elif data:
            self.key = data
        else:
            raise InvalidParameterError('Either a gab integer or data bytes array must be provided')

        with BinaryReader(utils.sha1(self.key)) as reader:
            self.aux_hash = reader.read_long(signed=False)
            reader.read(4)
            self.key_id = reader.read_long(signed=False)

    def calc_new_nonce_hash(self, new_nonce, number):
        """Calculates the new nonce hash based on the current class fields' values"""
        with BinaryWriter() as writer:
            writer.write(new_nonce)
            writer.write_byte(number)
            writer.write_long(self.aux_hash, signed=False)

            new_nonce_hash = utils.sha1(writer.get_bytes())[4:20]
            return new_nonce_hash