mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-09-21 19:28:54 +03:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""
|
|
This module holds the AuthKey class.
|
|
"""
|
|
import struct
|
|
from hashlib import sha1
|
|
|
|
from .. import helpers as utils
|
|
from ..extensions import BinaryReader
|
|
|
|
|
|
class AuthKey:
|
|
"""
|
|
Represents an authorization key, used to encrypt and decrypt
|
|
messages sent to Telegram's data centers.
|
|
"""
|
|
def __init__(self, data):
|
|
"""
|
|
Initializes a new authorization key.
|
|
|
|
:param data: the data in bytes that represent this auth key.
|
|
"""
|
|
self.key = data
|
|
|
|
with BinaryReader(sha1(self.key).digest()) 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 attributes.
|
|
|
|
:param new_nonce: the new nonce to be hashed.
|
|
:param number: number to prepend before the hash.
|
|
:return: the hash for the given new nonce.
|
|
"""
|
|
new_nonce = new_nonce.to_bytes(32, 'little', signed=True)
|
|
data = new_nonce + struct.pack('<BQ', number, self.aux_hash)
|
|
return utils.calc_msg_key(data)
|