o jJ!@sddlZddlZddlZddlZddlZddlZddlmZddlmZzddl Z dZ Wn e y7dZ YnwddZ dd Z iiifd d ZdS) N) unhexlify)FileNotFoundErrorTFcCs^d}g}Gdddt}d}d}d} |d7}|} | s)|dur&|| |S| } | ds4| s7d} q| drM|durE||d}|| q| rg|d7}d } |dur^|||d ||f|}td | } | sx|j| g7_n6| d } | d  } | | d}|durt | d dkrd | } t || t | nt || || q)zLoad and parse a test vector file Return a list of objects, one per group of adjacent KV lines or for a single line in the form "[.*]". For a group of lines, the object has one attribute per line. rc@seZdZddZdS)z_load_tests..TestVectorcSs||_||_g|_dS)N)desccountothers)self descriptionrr /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/SelfTest/loader.py__init__=s z(_load_tests..TestVector.__init__N)__name__ __module__ __qualname__r r r r r TestVector<s rNT#[Fz%s (#%d)z([A-Za-z0-9]+) = ?(.*)0)objectreadlineappendstrip startswithrematchrgrouplowergetlensetattrbinasciir) dir_compsfile_inr conversions line_numberresultsr test_vectorr new_grouplinerestokendata conversionr r r _load_tests0sT  &      r.cCsd}zItsttjttj|d||f}tjtj }tj tjj |g|R|}t |}t ||||}WdW|S1sDwYW|Sty_t jd|tddY|Sw)aLoad and parse a test vector file, formatted using the NIST style. Args: dir_comps (list of strings): The path components under the ``pycryptodome_test_vectors`` package. For instance ``("Cipher", "AES")``. file_name (string): The name of the file with the test vectors. description (string): A description applicable to the test vectors in the file. conversions (dictionary): The dictionary contains functions. Values in the file that have an entry in this dictionary will be converted usign the matching function. Otherwise, values will be considered as hexadecimal and converted to binary. Returns: A list of test vector objects. The file is formatted in the following way: - Lines starting with "#" are comments and will be ignored. - Each test vector is a sequence of 1 or more adjacent lines, where each lines is an assignement. - Test vectors are separated by an empty line, a comment, or a line starting with "[". A test vector object has the following attributes: - desc (string): description - counter (int): the order of the test vector in the file (from 1) - others (list): zero or more lines of the test vector that were not assignments - left-hand side of each assignment (lowercase): the value of the assignement, either converted or bytes. Nz %s test (%s)%Warning: skipping extended tests for r stacklevel)test_vectors_availablererrnoENOENTosstrerrorpathdirnamepycryptodome_test_vectors__file__joinopenr.warningswarn UserWarning)r" file_namerr$r&init_dirfull_file_namer#r r r load_test_vectorsus.&     rCc Cs>g}z>tsttjttj|tjtj }tj tjj |g|R|}t | } t | } Wdn1s:wYWntyUtjd|tdd|YSwGdddt} hd} | t|8} i} |D] \}}|| | |<qp| dD]}i}|D] \}}||||<q|dD]}| }| D] \}}t|||q|D] \}}t|||q|d |_|d |_| D]%}||vrz t||t||Wqtjytd ||j|fwq||_|D] \}}t||||q|d d k|_|d dk|_|d|_ ||_|!|qq|S)Nr/rr0c@s eZdZdS)z0load_test_vectors_wycheproof..TestVectorN)r r rr r r r rsr>ctivaadikmkeymsgokmsigtaginfosaltlabelpublicshared testGroupsteststcIdcommentz0Error decoding attribute '%s' (tcId=%s, file %s)resultinvalid acceptableflags)"r2rr3r4r5r6r7r8r9r:r;r<jsonloadr=r>r?rsetkeysitemsr idrUrr!Error ValueErrorfilenamevalidwarningrrYr)r"r@rroot_tag group_tagunit_tagrVrArBr#tv_treer unit_attr_hex common_rootkvr common_grouptesttvattrr r r load_test_vectors_wycheproofsp           rq)r5rrZr3r!r=rCryptodome.Util.py3compatrr9r2 ImportErrorr.rCrqr r r r s$   E>