관리-도구
편집 파일: test_crypto_digest.cpython-39.pyc
a f�Wc�O � @ s� d Z ddlmZmZ ddlmZ ddlZddlZddlm Z ddl m Z m Z m Z ddlmZmZmZmZ G dd � d e�ZG d d � d e�Zdd lmZmZ G d d� de�ZdS )z(tests for passlib.utils.(des|pbkdf2|md4)� )�with_statement�division)�hexlifyN)�UnknownHashError)�PY3�u�JYTHON)�TestCase� TEST_MODE� skipUnless�hbc @ sT e Zd ZdZdZddgZg d�Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dS )� HashInfoTestz test various crypto functionszpasslib.crypto.digest�hashlibZiana))�md5r zSCRAM-MD5-PLUSzMD-5)�sha1zsha-1z SCRAM-SHA-1ZSHA1)�sha256�sha-256ZSHA_256zsha2-256)Z ripemd160z ripemd-160zSCRAM-RIPEMD-160Z RIPEmd160Zripemdz SCRAM-RIPEMD)Zsha4_256zsha4-256zSHA4-256z SHA-4-256)Ztest128ztest-128ZTEST128)�test2r zTEST-2)Z test3_128z test3-128z TEST-3-128c C s� ddl m} ddlm}m} t�� }|�� | �|j � t� dd� t� dd� | � |t d��d� | � |d �d� | � t|d � ||| j�D ]J}t| j�D ]:\}}|| }|D ]$} || |�} | � | |d | |f � q�q�q�d S ) znorm_hash_name()r )�chain)�norm_hash_name�_known_hash_names�ignorez.*unknown hashz.*unsupported hashZMD4�md4s MD4Nzname=%r, format=%r:)� itertoolsr �passlib.crypto.digestr r �warnings�catch_warnings� __enter__� addCleanup�__exit__�filterwarnings� assertEqualr � assertRaises� TypeError�norm_hash_samples� enumerate�norm_hash_formats) �selfr r r �ctx�row�idx�format�correct�value�result� r/ �D/usr/lib/python3.9/site-packages/passlib/tests/test_crypto_digest.py�test_norm_hash_name. s( ��z HashInfoTest.test_norm_hash_namec C s` ddl m} | �t|d� | �t|d� | �t|d� | �|d�tjddf� zt�d � d }W n tyv d }Y n0 |r�|d �}|d }| �||d df� | �t|d �� � �d� n| �t|d � zt�d� d }W n ty� d }Y n0 |d�}|d }|�s ddl m } | � ||� | �||ddf� | �t|d �� � �d� | � |d�|d�� dS )z lookup_hash() -- constructorr �� lookup_hash�new�__name__Zsha4r � �@ ZshaTF� s abcs( 0164b8a914cd2a5e74c4f7ff082c4d97f1edf880r )r s a448017aaf21d8525fc10ae87aa6729dN) r r3 r" � ValueErrorr! r r r4 r �digestZpasslib.crypto._md4r �assertIs)r' r3 Zhas_sha�record�constZhas_md4r r/ r/ r0 �test_lookup_hash_ctorI sB � �z"HashInfoTest.test_lookup_hash_ctorc C s� ddl m} | �t|d� |ddd�}| �|j� | �td|j� | �|j d� | �|j d� | �|j d� |ddd�}| � ||� dS )z"lookup_hash() -- unknown hash namer r2 Zxxx256F��requiredzunknown hash: 'xxx256'N) r r3 r" r Z assertFalse� supported�assertRaisesRegexr= r! �name� digest_size� block_sizer; )r' r3 �infoZinfo2r/ r/ r0 � test_lookup_hash_w_unknown_namew s z,HashInfoTest.test_lookup_hash_w_unknown_namec C s� ddl m}m} |ddd�js(| �d��|� | �|d� d}| �t||d� |ddd�}| �|j |� | �t||j � | � |j d� | � |j d � d S ) z< lookup_hash() -- test set_mock_fips_mode() r )r3 �_set_mock_fips_moder Fr? zmd5 not supportedz 'md5' hash disabled for fipsr6 r7 N)r r3 rH rA ZskipTestr rB r Z assertRegexZ error_textr= r! rD rE )r' r3 rH �patrF r/ r/ r0 �test_mock_fips_mode� s z HashInfoTest.test_mock_fips_modec C s� ddl m} |d�}| �|jd� | �|jd� | �|jd� | �|jd� | �|d�|� |d�}| �|jd� | �|jd� | �|jd� | �|jd � d S ) zlookup_hash() -- metadatar r2 r r r7 � zSHA2-256r r6 N)r r3 r! rC Z iana_namerE rD r; �r' r3 rF r/ r/ r0 �test_lookup_hash_metadata� s z&HashInfoTest.test_lookup_hash_metadatac C sH ddl m} |d�}| �||�|� | �||j�|� | �t|d� dS )z lookup_hash() -- alternate typesr r2 r �{ N)r r3 r; r= r" r# rL r/ r/ r0 �test_lookup_hash_alt_types� s z'HashInfoTest.test_lookup_hash_alt_typesN) r5 � __module__� __qualname__�__doc__�descriptionPrefixr& r$ r1 r> rG rJ rM rO r/ r/ r/ r0 r s .r c @ s� e Zd ZdZdZded�ddded�fd dd dd d ed �fdd ddd ed�fdd ddd ed�fdd ddd ed�fdd ddded�fgZes�e�dd ddded�f� dd� Z dd� Z dS )� Pbkdf1_Testztest kdf helpersz passlib.crypto.digest.pbkdf1� passwordZ78578E5A5D63CB06�� r6 r Z dc19847e05c64d2faf10ebfb4a3d2a20)rU � saltrV r r � rW � r Z84� Z8475c6a8531a5d27Z 8475c6a8531a5d27e386cd496457812cNZ(4a8fd48e426ed081b535be5769892fa396293efbr Z f7f2e91100a8f96190f2dd177cb26453c C sD ddl m} | jD ],\}}}}}}||||||�}| �||� qdS )�test reference vectorsr ��pbkdf1N)r r] � pbkdf1_testsr! ) r' r] �secret�salt�rounds�keylenr: r, r. r/ r/ r0 � test_known� s zPbkdf1_Test.test_knownc s� ddl m� d� fdd� }|� | jt|dd � | jt|dd � | jt|d d � | jt|dd � | jt|dd � | jt|dd� | jt|ddd� | jt|dd� dS )�test border casesr r\ � secretrW rY r c s � || |||�S �Nr/ )r_ r` ra rb �hashr\ r/ r0 �helper� s z'Pbkdf1_Test.test_border.<locals>.helper�r_ �r` �missing)rg �ra �1����rb � )rb rg N)re rW rY rY r )r r] r" r# r9 �r' rh r/ r\ r0 � test_border� s zPbkdf1_Test.test_border) r5 rP rQ rR rS r r^ r �appendrc rr r/ r/ r/ r0 rT � s ��rT )� pbkdf2_hmac�PBKDF2_BACKENDSc $ @ s\ e Zd ZdZdd�e� Zed�ddddfed �ddd dfed �ddd d fed �dddd fed�dddd fed�dddd fed�dddd fed�ddddfed�ddd dfed �ddd dfed �d d d d!fed"�d#d$d dfed%�d&ed'�d(d)d*fed+�d,ddd)d-fed.�d/d0d1d)d-fed2�dddd d-fed3�ddd d d-fed4�ddd d d-fed5�d d d d6d-fed7�d8dd9d d-fed:�dd8d9d d-fed;�d#d$d dd-fed<�dddd d*fed=�ddd d d*fed>�ddd d d*fed?�d d dd@d*fed�dddddAfedB�dCddDddAfedE�dCddDdFdAfedG�dCddHd6dIfedJ�dCddHd6dKfg ZdLdM� Z dNdO� Z dPdQ� Z dRdS� Z dFS )T� Pbkdf2Testztest pbkdf2() supportz2passlib.crypto.digest.pbkdf2_hmac() <backends: %s>z, Z cdedb5281bb2f801565a1122b2563515rU s ATHENA.MIT.EDUraeburnrY r6 Z 01dbee7f4a9e243e988b62c73cda935d� Z@01dbee7f4a9e243e988b62c73cda935da05378b93244ec8f48a99e61ad799d86rK Z@5c08eb61fdf71e4e4ec3cf6ba1f5512ba7e52ddbc5e5142f708a31e2e62b1e13i� Z@d1daa78615f287e6a1c8b120d7062a493f98d203e6be49a6adf4fa574b6e64ees 4VxxV4� Z@139c30c0966bc32ba55fdbf212530ac9c5ec59f1a452f5cc9ad940fea0598ed1s@ XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXs pass phrase equals block sizeZ@9ccad6d468770cd51b10e6a68721be611a8b4d282601db3b36be9246915ec82asA XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXs pass phrase exceeds block sizeZ(0c60c80f961f0e71f3a9b524af6012062fe037a6rW r8 Z(ea6c014dc72d6f8ccd1ed92ace1d41f0d8de8957Z(4b007901b765489abead49d926f721d065a429c1i Z23d2eec4fe41c849b80c8d83662c0e44a8b291a964cf2f07038s passwordPASSWORDpasswords$ saltSALTsaltSALTsaltSALTsaltSALTsalt� Z 56fa6aa75548099dcc37d7f03425e0c3s pass words sa ltZ�887CFF169EA8335235D8004242AA7D6187A41E3187DF0CE14E256D85ED97A97357AAA8FF0A3871AB9EEFF458392F462F495487387F685B7472FC6C29E293F0A0s helloZ�9290F727ED06C38BA4549EF7DE25CF5642659211B7FC076F2D28FEFD71784BB8D8F6FB244A8CC5C06240631B97008565A120764C0EE9C2CB0073994D79080136i' r7 Zsha512Z�55ac046e56e3089fec1691c22544b605f94185216dde0465e68b9d57c20dacbc49ca9cccf179b645991664b39d77ef317c71b845b1e30bd509112041d3a19783s passwdr Z�4ddcd8f60b98be21830cee5ef22701f9641a4418d04c0414aeff08876b34ab56a1d425a1225833549adb841b51c9b3176a272bdebba1d078478f62b397f33c8ds Passwords NaCli�8 Z@120fb6cffcf8b32c43e7225256c4f837a86548c92ccc35480805987cb70be17bZ@ae4d0c95af6b46d32d0adff928f06dd02a303f8ef3c251dfd6e2d85a95474c43Z@c5e478d59288c841aa530db6845c4c8d962893a001ce4e11a4963873aa98134aZP348c89dbcbd32b2f32d814b8116e84cf2b17347ebc1800181c4e2a1fb8dd53e1c635518c7dac47e9�( Z@9e83f279c040f2a11aa4a02b24c418f2d3cb39560c9627fa4f47e3bcc2897c3drX i Z@ea5808411eb0c7e830deab55096cee582761e22a9bc034e3ece925225b07bf46Z 89b69d0516f829893c696226650a8687Z@867f70cf1ade02cff3752599a3a53dc4af34c7a669815ae5d513554e1c8cf252Z@e1d9c16aa681708a45f5c7c4e215ceb66e011a2e9f0040713f18aefdb866d53cZ@d197b1b33db0143e018b12f3d1d1479e6cdebdcc97c5c0f87f6902e072f457b5Z�6e23f27638084b0f7ea1734e0d9841f55dd29ea60a834466f3396bac801fac1eeb63802f03a0b4acd7603e3699c8b74437be83ff01ad7f55dac1ef60f4d56480c35ee68fd52c6936�H r Z e248fb6b13365146f8ac6307cc222812re � Z(e248fb6b13365146f8ac6307cc2228127872da6dNZPb1d5485772e6f76d5ebdc11b38d3eff0a5b2bd50dc11f937e86ecacd0cd40d1b9113e0734e3b76a3�> r ZPea014cc01f78d3883cac364bb5d054e2be238fb0b6081795a9d84512126e3129062104d2183464c4r c C sZ | j D ]N}|dd� \}}}}}t|�dkr4|d nd}t|||||�}| �||� qdS )r[ Nrx � r )�pbkdf2_test_vectors�lenrt r! ) r' r) r, r_ r` ra rb r: r. r/ r/ r0 rc � s zPbkdf2Test.test_knownc C s� ddl m} zddl}d}W n ty2 d}Y n0 | �d|v |� zddlm} |jdk}W n tyt d}Y n0 | �d |v |� dd lm } |r�| � d |� n | � d |� dS ) z$verify expected backends are presentr )ru NTF� fastpbkdf2�rt r z hashlib-ssl)r zbuiltin-from-byteszbuiltin-unpack) r ru r� � ImportErrorr! r rt rP �passlib.utils.compatr ZassertIn)r' ru r� Zhas_fastpbkdf2rt Zhas_hashlib_sslr r/ r/ r0 � test_backends� s" zPbkdf2Test.test_backendsc C s� ddd�}|� | j t|dd � | j t|d d � | j t|d d � |dd � | j t|dd � | j t|d d � | j t|d d � | j t|d d � | j t|dd� | j t|dd� | j t|dd� | j t|dd� dS )rd rU rW rY Nr c S s t || |||�S rf r� �r_ r` ra rb r: r/ r/ r0 rh � s z&Pbkdf2Test.test_border.<locals>.helperrn rl r �xro l ��O rx rj ri Zfoo�r: )rU rW rY Nr )r" r9 r# � OverflowErrorrq r/ r/ r0 rr � s zPbkdf2Test.test_borderc C s: d dd�}| � t|dd��d � | � t|d d��d � dS ) ztest keylen==NonerU rW rY Nr c S s t || |||�S rf r� r� r/ r/ r0 rh s z.Pbkdf2Test.test_default_keylen.<locals>.helperr� r8 r rK )rU rW rY Nr )r! r� rq r/ r/ r0 �test_default_keylen s z Pbkdf2Test.test_default_keylen) r5 rP rQ rR �joinru rS r r rc r� rr r� r/ r/ r/ r0 rv s� ������� ��� �� � � � � � �� � � � � � �� � � � � �� � L rv )rR Z __future__r r Zbinasciir r r Z passlib.excr r� r r r Zpasslib.tests.utilsr r r r r rT r rt ru rv r/ r/ r/ r0 �<module> s 4>