o j @sddlmZmZmZmZddlZddlZddlZddlm Z m Z ddl m Z ddl mZmZmZejdkr8eZndd Zd gZd d ZdS) )unicode_literalsdivisionabsolute_importprint_functionN)int_from_bytes int_to_bytes)pretty_message) type_namebyte_cls int_types)cCs t|gS)N)bytes)numr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/oscrypto/_pkcs12.pychr_clss r pkcs12_kdfcCspt|tsttdt|t|tsttdt|t|ts*ttdt||dkr7ttdt|t|tsEttdt||dkrRttdt||tgdvrcttd t||tgd vrtttd t|| d  d d}t t |}ddddddd|}|dvrd} nd} t || } d} |dkr| tttt|| } t| | kr| |7} t| | ks| d| } d} |dkr| tttt|| }t| |kr| |7} t| |ks| d|} | | }ttt||}d||}td|dD]}|| |}td|dD] }||}q ||krd}t|| krC||7}t|| ks8t|d| d}tdt|| D]:}|| }|d| }|||}tt||}t|}|| kr||| d}|d||||d}qV|d|}t||}|d||d||||d}q|d|S)a KDF from RFC7292 appendix b.2 - https://tools.ietf.org/html/rfc7292#page-19 :param hash_algorithm: The string name of the hash algorithm to use: "md5", "sha1", "sha224", "sha256", "sha384", "sha512" :param password: A byte string of the password to use an input to the KDF :param salt: A cryptographic random byte string :param iterations: The numbers of iterations to use when deriving the key :param key_length: The length of the desired key in bytes :param id_: The ID of the usage - 1 for key, 2 for iv, 3 for mac :return: The derived key as a byte string z@ password must be a byte string, not %s z< salt must be a byte string, not %s z? iterations must be an integer, not %s rzC iterations must be greater than 0 - is %s z? key_length must be an integer, not %s zC key_length must be greater than 0 - is %s )md5sha1sha224sha256sha384sha512z hash_algorithm must be one of "md5", "sha1", "sha224", "sha256", "sha384", "sha512", not %s )rr z< id_ must be one of 1, 2, 3, not %s zutf-8zutf-16bes 0@)rrrrN) isinstancer TypeErrorr r r ValueErrorreprsetdecodeencodegetattrhashlibrintmathceilfloatlenrangedigestrrmin)hash_algorithmpasswordsalt iterations key_lengthid_utf16_passwordalgouvdss_lenpp_lenicara2_bnum2startendi_num2i_num2_lbeginto_copyrrrrs                   , ) __future__rrrrsysr,r._asn1rr_errorsr _typesr r r version_infochrr__all__rrrrrs