mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-13 13:06:36 +03:00
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""
|
|
This module holds the AuthKey class.
|
|
"""
|
|
import struct
|
|
from hashlib import sha1
|
|
|
|
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)
|
|
|
|
# Calculates the message key from the given data
|
|
return int.from_bytes(sha1(data).digest()[4:20], 'little', signed=True)
|
|
|
|
def __eq__(self, other):
|
|
return isinstance(other, type(self)) and other.key == self.key
|