a fi@sdZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZmZmZmZddlmZdd lmZdd lmZmZmZd d ZdddZdS)z This module contains several functions that authenticate the client machine with Telegram's servers, effectively creating an authorization key. N)sha1) ResPQ PQInnerDataServerDHParamsFailServerDHParamsOkServerDHInnerDataClientDHInnerDataDhGenOk DhGenRetry DhGenFail)helpers)AESAuthKey Factorizationrsa) SecurityError) BinaryReader)ReqPqMultiRequestReqDHParamsRequestSetClientDHParamsRequestc$ s&tjtdddd}|t|IdH}t|ts@Jd||j|krRt dt |j }t |\}}t|t|}}tjtdd dd}ttt||||j|j|d }d \}} |jD] } t| |}|dur| } qq|dur |jD]&} tj| |dd }|dur| } q q|durJt d ddd|jD|t|j|j||| |dIdH} t| ttfsJd| | j|jkrt d| j|jkrt dt| trtjt|jdd ddddd dd} | j| krt dt| tsJd| t|j|\} }t | j!ddkrHt dt"#| j!| |}t$|:}|%d|&}t|t'sJd|Wdn1s0Y|j|jkrt d|j|jkrt dt |j(dd}|j)}t |j*dd}|j+tt,,}t tddd}t-|||}t-|||}d |krR|d ks\nt d!d |krx|d ksnt d!d |kr|d ksnt d"d#d$}||kr||ksnt d%||kr||ksnt d&tt.|j|jdt|d'}t||}t"/|| |}|t0|j|j|d(IdH}t1t2t3f}t||sxJd)||j4j5}|j|jkrt d*||j|jkrt d+|t6t|} d |7t8|}!| 9||!}"t:|d,|!}#|#|"krt d-t|t1st;d.|| |fS)/z Executes the authentication process with the Telegram servers. :param sender: a connected `MTProtoPlainSender`. :return: returns a (authorization key, time offset) tuple. bigT)signedNzStep 1 answer was %sz Step 1 invalid nonce from server little)pqpqnonce server_nonce new_nonce)NN)Zuse_oldz6Step 2 could not find a valid key for fingerprints: {}z, cSsg|] }t|qS)str).0fr"r"S/home/ec2-user/.local/lib/python3.9/site-packages/telethon/network/authenticator.py Gz%do_authentication..)rr rrZpublic_key_fingerprintencrypted_datazStep 2.1 answer was %sz Step 2 invalid nonce from serverz'Step 2 invalid server nonce from serverz(Step 2 invalid DH fail nonce from serverzStep 2.2 answer was %srzStep 3 AES block size mismatchzStep 3 answer was %sz(Step 3 Invalid nonce in encrypted answerz/Step 3 Invalid server nonce in encrypted answerFz#g_a is not within (1, dh_prime - 1)z#g_b is not within (1, dh_prime - 1)riz7g_a is not within (2^{2048-64}, dh_prime - 2^{2048-64})z7g_b is not within (2^{2048-64}, dh_prime - 2^{2048-64}))rr Zretry_idg_b)rr r)zStep 3.1 answer was %sz#Step 3 invalid {} nonce from serverz*Step 3 invalid {} server nonce from serverznew_nonce_hash{}zStep 3 invalid new nonce hashzStep 3.2 answer was %s)<int from_bytesosurandomsendr isinstancerrrget_intrrZ factorizerZget_byte_arraybytesrr Zserver_public_key_fingerprintsZencryptformatjoinrrrrto_bytesdigestnew_nonce_hashr Zgenerate_key_data_from_noncelenZencrypted_answerrZ decrypt_igerreadZ tgread_objectrdh_primegg_aZ server_timetimepowr Z encrypt_igerr r r __class____name__rindextypeZcalc_new_nonce_hashgetattrAssertionError)$ZsenderrZres_pqrrrr!Z pq_inner_dataZ cipher_textZtarget_fingerprint fingerprintZserver_dh_paramsZnnhkeyZivZplain_text_answerreaderZserver_dh_innerr>r?r@Z time_offsetbr.ZgabZ safety_rangeZclient_dh_innerZclient_dh_inner_hashedZclient_dh_encryptedZdh_genZ nonce_typesnameZauth_keyZ nonce_numberr;Zdh_hashr"r"r&do_authentications            $       rNTcCstj|d|dS)a8 Gets the specified integer from its byte array. This should be used by this module alone, as it works with big endian. :param byte_array: the byte array representing th integer. :param signed: whether the number is signed or not. :return: the integer representing the given byte array. r) byteorderr)r/r0)Z byte_arrayrr"r"r&r5s r5)T) __doc__r1rAhashlibrZtl.typesrrrrrr r r r r Zcryptorrrrerrorsr extensionsrZ tl.functionsrrrrNr5r"r"r"r&s ,   6