o
    5hO                     @   s   d Z ddlmZmZ ddlmZ ddlZddlZddlm	Z	 ddl
mZmZmZ ddlmZmZmZmZ G dd	 d	eZG d
d deZddlmZmZ G dd deZdS )z(tests for passlib.utils.(des|pbkdf2|md4)    )with_statementdivision)hexlifyN)UnknownHashError)PY3uJYTHON)TestCase	TEST_MODE
skipUnlesshbc                   @   sT   e Zd ZdZdZddgZg dZdd Zdd	 Zd
d Z	dd Z
dd Zdd ZdS )HashInfoTestztest various crypto functionszpasslib.crypto.digesthashlibiana))md5r   zSCRAM-MD5-PLUSzMD-5)sha1zsha-1zSCRAM-SHA-1SHA1)sha256sha-256SHA_256zsha2-256)	ripemd160z
ripemd-160zSCRAM-RIPEMD-160	RIPEmd160ripemdzSCRAM-RIPEMD)sha4_256zsha4-256zSHA4-256z	SHA-4-256)test128ztest-128TEST128)test2r   zTEST-2)	test3_128z	test3-128z
TEST-3-128c              
   C   s   ddl m} ddlm}m} t }|  | |j	 t
dd t
dd | |tdd | |d	d | t|d
 ||| jD ]%}t| jD ]\}}|| }|D ]}	||	|}
| |
|d|	|f  qXqNqGd
S )znorm_hash_name()r   )chain)norm_hash_name_known_hash_namesignorez.*unknown hashz.*unsupported hashMD4md4s   MD4Nzname=%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:   t/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/passlib/tests/test_crypto_digest.pytest_norm_hash_name.   s.   
z HashInfoTest.test_norm_hash_namec                 C   s^  ddl m} | t|d | t|d | t|d | |dtjddf z	td	 d
}W n ty;   d}Y nw |r\|d	}|d }| ||ddf | t|d	 d n| t|d	 z	td d
}W n tyw   d}Y nw |d}|d }|sddl
m} | || | ||ddf | t|d	 d | |d|d dS )zlookup_hash() -- constructorr   lookup_hashnew__name__sha4r      @   shaTF   s   abcs(   0164b8a914cd2a5e74c4f7ff082c4d97f1edf880r#   )r#   s    a448017aaf21d8525fc10ae87aa6729dN)r%   r>   r-   
ValueErrorr,   r   r   r?   r   digestpasslib.crypto._md4r#   assertIs)r2   r>   has_sharecordconsthas_md4r#   r:   r:   r;   test_lookup_hash_ctorI   sF   

z"HashInfoTest.test_lookup_hash_ctorc                 C   s   ddl m} | t|d |ddd}| |j | td|j | |j	d | |j
d | |jd |ddd}| || dS )z"lookup_hash() -- unknown hash namer   r=   xxx256Frequiredzunknown hash: 'xxx256'N)r%   r>   r-   r   assertFalse	supportedassertRaisesRegexrL   r,   namedigest_size
block_sizerI   )r2   r>   infoinfo2r:   r:   r;   test_lookup_hash_w_unknown_namew   s   z,HashInfoTest.test_lookup_hash_w_unknown_namec                 C   s   ddl m}m} |dddjs| d|  | |d d}| t||d |ddd}| |j	| | t||j
 | |jd | |jd	 d
S )z<
        lookup_hash() -- test set_mock_fips_mode()
        r   )r>   _set_mock_fips_moder   FrP   zmd5 not supportedz'md5' hash disabled for fipsrB   rC   N)r%   r>   r[   rS   skipTestr)   rT   r   assertRegex
error_textrL   r,   rV   rW   )r2   r>   r[   patrX   r:   r:   r;   test_mock_fips_mode   s   
z HashInfoTest.test_mock_fips_modec                 C   s   ddl m} |d}| |jd | |jd | |jd | |jd | |d| |d}| |jd | |jd | |jd | |jd	 d
S )zlookup_hash() -- metadatar   r=   r   r   rC       zSHA2-256r   rB   N)r%   r>   r,   rU   	iana_namerW   rV   rI   r2   r>   rX   r:   r:   r;   test_lookup_hash_metadata   s   z&HashInfoTest.test_lookup_hash_metadatac                 C   sH   ddl m} |d}| ||| | ||j| | t|d dS )z lookup_hash() -- alternate typesr   r=   r   {   N)r%   r>   rI   rL   r-   r.   rc   r:   r:   r;   test_lookup_hash_alt_types   s
   z'HashInfoTest.test_lookup_hash_alt_typesN)r@   
__module____qualname____doc__descriptionPrefixr1   r/   r<   rN   rZ   r`   rd   rf   r:   r:   r:   r;   r      s    .r   c                   @   s   e Zd ZdZdZdeddddedfd	dd
dddedfdd
dddedfdd
dddedfdd
dddedfdd
dddedfgZesRedd
dddedf dd Z	dd Z
dS )Pbkdf1_Testztest kdf helperszpasslib.crypto.digest.pbkdf1   password78578E5A5D63CB06  rB   r    dc19847e05c64d2faf10ebfb4a3d2a20)rl      saltrn   r   r       rp      r   84   8475c6a8531a5d27 8475c6a8531a5d27e386cd496457812cN(4a8fd48e426ed081b535be5769892fa396293efbr#    f7f2e91100a8f96190f2dd177cb26453c           	      C   sD   ddl m} | jD ]\}}}}}}||||||}| || q	dS )test reference vectorsr   pbkdf1N)r%   r{   pbkdf1_testsr,   )	r2   r{   secretsaltroundskeylenrG   r7   r9   r:   r:   r;   
test_known   s
   zPbkdf1_Test.test_knownc                    s   ddl m  d fdd	}|  | jt|dd	 | jt|dd
 | jt|dd | jt|dd | jt|dd | jt|dd | jt|ddd | jt|dd dS )test border casesr   rz      secretrp   rr   r   c                    s    || |||S Nr:   )r}   r~   r   r   hashrz   r:   r;   helper      z'Pbkdf1_Test.test_border.<locals>.helperr}   r~   missing)r   r   1r      )r   r   N)r   rp   rr   rr   r   )r%   r{   r-   r.   rF   r2   r   r:   rz   r;   test_border   s   zPbkdf1_Test.test_border)r@   rg   rh   ri   rj   r   r|   r   appendr   r   r:   r:   r:   r;   rk      s"    rk   )pbkdf2_hmacPBKDF2_BACKENDSc                   @   s  e Zd ZdZdde Zg edddddfed	ddd
dfedddd
dfedddddfedddddfedddddfedddddfedddddfedddd
dfedddddfeddd dd!fed"d#d$ddfed%d&ed'd(d)d*fed+d,ddd)d-fed.d/d0d1d)d-fed2ddddd-fed3ddd
dd-fed4ddddd-fed5dd dd6d-fed7d8dd9dd-fed:dd8d9dd-fed;d#d$ddd-fed<ddddd*fed=ddd
dd*fed>ddddd*fed?dd dd@d*feddddddAfedBdCddDddAfedEdCddDdFdAfedGdCddHd6dIfedJdCddHd6dKfZdLdM Z	dNdO Z
dPdQ ZdRdS ZdFS )T
Pbkdf2Testztest pbkdf2() supportz2passlib.crypto.digest.pbkdf2_hmac() <backends: %s>z,  cdedb5281bb2f801565a1122b2563515rl   s   ATHENA.MIT.EDUraeburnrr   rB    01dbee7f4a9e243e988b62c73cda935d   @01dbee7f4a9e243e988b62c73cda935da05378b93244ec8f48a99e61ad799d86ra   @5c08eb61fdf71e4e4ec3cf6ba1f5512ba7e52ddbc5e5142f708a31e2e62b1e13i  @d1daa78615f287e6a1c8b120d7062a493f98d203e6be49a6adf4fa574b6e64ees   4VxxV4   @139c30c0966bc32ba55fdbf212530ac9c5ec59f1a452f5cc9ad940fea0598ed1s@   XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXs   pass phrase equals block size@9ccad6d468770cd51b10e6a68721be611a8b4d282601db3b36be9246915ec82asA   XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXs   pass phrase exceeds block size(0c60c80f961f0e71f3a9b524af6012062fe037a6rp   rE   (ea6c014dc72d6f8ccd1ed92ace1d41f0d8de8957(4b007901b765489abead49d926f721d065a429c1i   23d2eec4fe41c849b80c8d83662c0e44a8b291a964cf2f07038s   passwordPASSWORDpasswords$   saltSALTsaltSALTsaltSALTsaltSALTsalt    56fa6aa75548099dcc37d7f03425e0c3s	   pass words   sa ltڀ887CFF169EA8335235D8004242AA7D6187A41E3187DF0CE14E256D85ED97A97357AAA8FF0A3871AB9EEFF458392F462F495487387F685B7472FC6C29E293F0A0s   helloڀ9290F727ED06C38BA4549EF7DE25CF5642659211B7FC076F2D28FEFD71784BB8D8F6FB244A8CC5C06240631B97008565A120764C0EE9C2CB0073994D79080136i'  rC   sha512ڀ55ac046e56e3089fec1691c22544b605f94185216dde0465e68b9d57c20dacbc49ca9cccf179b645991664b39d77ef317c71b845b1e30bd509112041d3a19783s   passwdr   ڀ4ddcd8f60b98be21830cee5ef22701f9641a4418d04c0414aeff08876b34ab56a1d425a1225833549adb841b51c9b3176a272bdebba1d078478f62b397f33c8ds   Passwords   NaCli8 @120fb6cffcf8b32c43e7225256c4f837a86548c92ccc35480805987cb70be17b@ae4d0c95af6b46d32d0adff928f06dd02a303f8ef3c251dfd6e2d85a95474c43@c5e478d59288c841aa530db6845c4c8d962893a001ce4e11a4963873aa98134aP348c89dbcbd32b2f32d814b8116e84cf2b17347ebc1800181c4e2a1fb8dd53e1c635518c7dac47e9(   @9e83f279c040f2a11aa4a02b24c418f2d3cb39560c9627fa4f47e3bcc2897c3drq   i   @ea5808411eb0c7e830deab55096cee582761e22a9bc034e3ece925225b07bf46 89b69d0516f829893c696226650a8687@867f70cf1ade02cff3752599a3a53dc4af34c7a669815ae5d513554e1c8cf252@e1d9c16aa681708a45f5c7c4e215ceb66e011a2e9f0040713f18aefdb866d53c@d197b1b33db0143e018b12f3d1d1479e6cdebdcc97c5c0f87f6902e072f457b5ڐ6e23f27638084b0f7ea1734e0d9841f55dd29ea60a834466f3396bac801fac1eeb63802f03a0b4acd7603e3699c8b74437be83ff01ad7f55dac1ef60f4d56480c35ee68fd52c6936H   r    e248fb6b13365146f8ac6307cc222812r   
   (e248fb6b13365146f8ac6307cc2228127872da6dNPb1d5485772e6f76d5ebdc11b38d3eff0a5b2bd50dc11f937e86ecacd0cd40d1b9113e0734e3b76a3>   r   Pea014cc01f78d3883cac364bb5d054e2be238fb0b6081795a9d84512126e3129062104d2183464c4r#   c           	      C   sZ   | j D ]'}|dd \}}}}}t|dkr|d nd}t|||||}| || qdS )ry   Nr      r   )pbkdf2_test_vectorslenr   r,   )	r2   r4   r7   r}   r~   r   r   rG   r9   r:   r:   r;   r     s   
zPbkdf2Test.test_knownc                 C   s   ddl m} zddl}d}W n ty   d}Y nw | d|v | zddlm} |jdk}W n ty:   d}Y nw | d	|v | dd
lm	} |rS| 
d| dS | 
d| dS )z$verify expected backends are presentr   )r   NTF
fastpbkdf2r   r   zhashlib-ssl)r   zbuiltin-from-byteszbuiltin-unpack)r%   r   r   ImportErrorr,   r   r   rg   passlib.utils.compatr   assertIn)r2   r   r   has_fastpbkdf2r   has_hashlib_sslr   r:   r:   r;   test_backends  s&   zPbkdf2Test.test_backendsc                 C   s   ddd}|  | j t|dd	 | j t|d
d	 | j t|dd	 |dd | j t|dd | j t|d
d | j t|dd | j t|dd | j t|dd | j t|dd | j t|dd | j t|dd dS )r   rl   rp   rr   Nr   c                 S      t || |||S r   r   r}   r~   r   r   rG   r:   r:   r;   r     r   z&Pbkdf2Test.test_border.<locals>.helperr   r   r   xr   l   O r   r   r   foorG   rl   rp   rr   Nr   )r-   rF   r.   OverflowErrorr   r:   r:   r;   r     s   

zPbkdf2Test.test_borderc                 C   s:   ddd}|  t|ddd	 |  t|d
dd dS )ztest keylen==Nonerl   rp   rr   Nr   c                 S   r   r   r   r   r:   r:   r;   r     r   z.Pbkdf2Test.test_default_keylen.<locals>.helperr   rE   r   ra   r   )r,   r   r   r:   r:   r;   test_default_keylen  s   
zPbkdf2Test.test_default_keylen)r@   rg   rh   ri   joinr   rj   r   r   r   r   r   r   r:   r:   r:   r;   r     sJ   	!'-5:?KRZh
n
t
y
~
   
  
  
  
  
  $
  )  1
  9
  =
  A
  F
  Lr   )ri   
__future__r   r   binasciir   r   r&   passlib.excr   r   r   r   r   passlib.tests.utilsr	   r
   r   r   r   rk   r%   r   r   r   r:   r:   r:   r;   <module>   s     4>