o Xzj@sZdZddlmZddlmZmZddlmZmZddl m Z ddl m Z Gdd d Z d S) z3 This module holds the CdnDecrypter utility class. )sha256)GetCdnFileRequestReuploadCdnFileRequest)CdnFileReuploadNeededCdnFile) AESModeCTR)CdnFileTamperedErrorc@s8eZdZdZddZeddZddZedd Zd S) CdnDecrypterz Used when downloading a file results in a 'FileCdnRedirect' to both prepare the redirect, decrypt the file as it downloads, and ensure the file hasn't been tampered. https://core.telegram.org/cdn cCs||_||_||_||_dS)a. Initializes the CDN decrypter. :param cdn_client: a client connected to a CDN. :param file_token: the token of the file to be used. :param cdn_aes: the AES CTR used to decrypt the file. :param cdn_file_hashes: the hashes the decrypted file must match. N)client file_tokencdn_aescdn_file_hashes)self cdn_clientr r rrW/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telethon/crypto/cdndecrypter.py__init__s  zCdnDecrypter.__init__cst|j|jddtdd}t||j||j}|t|j|jdj|jdj dIdH}t |t rI|t |j|j dIdH|}||fS|j|j|_|jd}||j|||fS)a9 Prepares a new CDN decrypter. :param client: a TelegramClient connected to the main servers. :param cdn_client: a new client connected to the CDN. :param cdn_redirect: the redirect file object that caused this call. :return: (CdnDecrypter, first chunk file data) N )keyivr)r offsetlimit)r request_token)rencryption_key encryption_ivbytesr r rrrr isinstancerrrget_filer encryptpopcheck)r r cdn_redirectr decryptercdn_filecdn_hashrrrprepare_decrypter s2     zCdnDecrypter.prepare_decryptercCs\|jr&|jd}|t|j|j|j}|j|j |_ | |j ||St t d}|S)z Calls GetCdnFileRequest and decrypts its bytes. Also ensures that the file hasn't been tampered. :return: the CdnFile result. r) rr!r rr rrr r rr"r)rr&r%rrrrLs   zCdnDecrypter.get_filecCst||jkr tdS)z Checks the integrity of the given data. Raises CdnFileTamperedError if the integrity check fails. :param data: the data to be hashed. :param cdn_hash: the expected hash. N)rdigesthashr )datar&rrrr"_s zCdnDecrypter.checkN) __name__ __module__ __qualname____doc__r staticmethodr'rr"rrrrr s +r N)r.hashlibrtl.functions.uploadrrtl.types.uploadrrcryptorerrorsr r rrrrs