U
    g                     @   s  d dl Z d dlZd dlmZ d dlmZmZ d dlmZ d dl	m
Z
mZ d dlmZmZmZmZmZmZmZ d dlmZmZ d dlmZmZmZmZmZmZmZmZ d d	lm Z  d
d Z!G dd de"Z#G dd dej$Z%G dd dej$Z&G dd dej$Z'G dd dej$Z(G dd dej$Z)G dd dej$Z*G dd dej$Z+dd Z,G dd dej$Z-d d! Z.e.e- i fd"d#Z/e0d$krd%d& Z1ej2d'd( dS ))    N)	unhexlify)bbchr)list_test_cases)load_test_vectorsload_test_vectors_wycheproof)SHA1HMACSHA256MD5SHA224SHA384SHA512)AESDES3)PBKDF1PBKDF2_S2VHKDFscryptbcryptbcrypt_checkSP800_108_Counter)_bcrypt_decodec                 C   s,   | d krd S |  dd dd}tt|S )N  
)replacer   r   )tt2 r    I/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/SelfTest/Protocol/test_KDF.pyt2b)   s    r"   c                   @   s   e Zd ZdS )
TestVectorN)__name__
__module____qualname__r    r    r    r!   r#   0   s   r#   c                   @   s   e Zd ZdZdd ZdS )PBKDF1_Tests))password78578E5A5D63CB06   i  Z DC19847E05C64D2FAF10EBFB4A3D2A20c                 C   sF   | j d }t|d t|d |d |d t}| |t|d  d S )Nr               )	_testDatar   r"   r   assertEqual)selfvresr    r    r!   test1B   s    
$zPBKDF1_Tests.test1N)r$   r%   r&   r/   r4   r    r    r    r!   r'   4   s   	r'   c                   @   s   e Zd Zddddedfdddded	fdddd
edfddddedfddddedfddddedfddddedfddddedffZdd Zdd  Zd!d" Zd#d$ Z	d%S )&PBKDF2_Testsr(   r)      i   Z0BFDE6BE94DF7E11DD409BCE20A0255EC327CB936FFE93643Z73616c74   r+   Z(0c60c80f961f0e71f3a9b524af6012062fe037a6r,   Z(ea6c014dc72d6f8ccd1ed92ace1d41f0d8de8957i   Z(4b007901b765489abead49d926f721d065a429c1ZpasswordPASSWORDpasswordZH73616c7453414c5473616c7453414c5473616c7453414c5473616c7453414c5473616c74   Z23d2eec4fe41c849b80c8d83662c0e44a8b291a964cf2f07038z	pass wordZ
7361006c74r*   Z 56fa6aa75548099dcc37d7f03425e0c3passwd@   Z55ac046e56e3089fec1691c22544b605f94185216dde0465e68b9d57c20dacbc49ca9cccf179b645991664b39d77ef317c71b845b1e30bd509112041d3a19783ZPasswordZ4e61436ci8 Z4ddcd8f60b98be21830cee5ef22701f9641a4418d04c0414aeff08876b34ab56a1d425a1225833549adb841b51c9b3176a272bdebba1d078478f62b397f33c8dc                 C   s   dd }dd }t t| jD ]}| j| }|d }t|d }|d }|d }|d	 }	t|d
 }
|	tkrt||||}| ||
 t|||||}| ||
 qt|||||}| ||
 qd S )Nc                 S   s   t | |t S Nr	   newr   digestpsr    r    r!   prf_SHA1d   s    z$PBKDF2_Tests.test1.<locals>.prf_SHA1c                 S   s   t | |t S r;   r	   r=   r
   r>   r?   r    r    r!   
prf_SHA256g   s    z&PBKDF2_Tests.test1.<locals>.prf_SHA256r   r+   r,   r-   r.      )rangelenr/   r"   r   r   r0   )r1   rB   rD   ir2   r(   saltZout_lenZitersZhash_modexpectedr3   r    r    r!   r4   a   s"    
zPBKDF2_Tests.test1c              
   C   s.   dd }| j tttdtddd|td d S )Nc                 S   s   t | |t S r;   r<   r?   r    r    r!   rB      s    z$PBKDF2_Tests.test2.<locals>.prf_SHA1xxxyyyr*   d   )prfhmac_hash_module)assertRaises
ValueErrorr   r   r   )r1   rB   r    r    r!   test2}   s
     zPBKDF2_Tests.test2c              	      sb   t d}t d}ttttttfD ]< t||dd fddd}t||dd d}| || q d S )	NrK   rL   r*   rM   c                    s   t | |  S r;   r	   r=   r>   r?   hashmodr    r!   <lambda>       z$PBKDF2_Tests.test3.<locals>.<lambda>)rN   )rO   )	r   r   r   r   r
   r   r   r   r0   )r1   r(   rI   Zpr1Zpr2r    rT   r!   test3   s    

zPBKDF2_Tests.test3c                 C   sl   t dtddd}t tdtddd}| || t tdddd}t tdtddd}| || d S )NrK   rL   r*   
   )r   r   r0   )r1   Zk1Zk2r    r    r!   test4   s    zPBKDF2_Tests.test4N)
r$   r%   r&   r   r
   r/   r4   rR   rX   rZ   r    r    r    r!   r5   H   s$      r5   c                   @   s4   e Zd ZdddefdddefgZdd Zd	d
 ZdS )	S2V_Tests)Z0101112131415161718191a1b1c1d1e1f2021222324252627Z112233445566778899aabbccddeeZ fffefdfcfbfaf9f8f7f6f5f4f3f2f1f0Z 85632d07c6e8f37f950acd320a2ecc93)ZP00112233445566778899aabbccddeeffdeaddadadeaddadaffeeddccbbaa99887766554433221100Z102030405060708090a0Z 09f911029d74e35bd84156c5635688c0Z^7468697320697320736f6d6520706c61696e7465787420746f20656e6372797074207573696e67205349562d414553Z 7f7e7d7c7b7a79787776757473727170Z 7bdb6e3b432667eb06f4d14bff2fbd0fc                 C   s`   | j D ]T}tt|d |d }|d D ]}|t| q*| }| |t|d  qdS )z!Verify correctness of test vectorr+   r-   r   r,   N)r/   r   r=   r"   updateZderiver0   )r1   tvs2vrA   resultr    r    r!   r4      s    
zS2V_Tests.test1c                 C   st   t dd t dd  }ttfD ]N}t||}|jd d }t|D ]}|td qF| 	t
|jtd q dS )zOVerify that no more than 127(AES) and 63(TDES)
        components are accepted.r         r+   ZXXZYYN)r   r   r   r   r=   
block_sizerF   r\   r   rP   	TypeError)r1   keymoduler^   Z	max_compsrH   r    r    r!   rR      s    zS2V_Tests.test2N)r$   r%   r&   r   r/   r4   rR   r    r    r    r!   r[      s   
	r[   c                   @   s   e Zd Z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fZdd Zdd ZdS )
HKDF_TestsZ,0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0bZ000102030405060708090a0b0cZf0f1f2f3f4f5f6f7f8f9*   ZT3cb25f25faacd57a90434f64d0362f2a2d2d0a90cf1a5a4c5db02d56ecc4c5bf34007208d5b887185865Z000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f404142434445464748494a4b4c4d4e4fZ606162636465666768696a6b6c6d6e6f707172737475767778797a7b7c7d7e7f808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a3a4a5a6a7a8a9aaabacadaeafZb0b1b2b3b4b5b6b7b8b9babbbcbdbebfc0c1c2c3c4c5c6c7c8c9cacbcccdcecfd0d1d2d3d4d5d6d7d8d9dadbdcdddedfe0e1e2e3e4e5e6e7e8e9eaebecedeeeff0f1f2f3f4f5f6f7f8f9fafbfcfdfeffR   Zb11e398dc80327a1c8e7f78c596a49344f012eda2d4efad8a050cc4c19afa97c59045a99cac7827271cb41c65e590e09da3275600c2f09b8367793a9aca3db71cc30c58179ec3e87c14c01d5c1f3434f1d87NZT8da4e775a563c18f715f802a063c5a31b8a11f5c5ee1879ec3454e5f3c738d2d9d201395faa4b61a96c8Z0b0b0b0b0b0b0b0b0b0b0bZT085a01ea1b10f36933068b56efa5ad81a4f14b822f5b091568a9cdd4f155fda2c22e422478d305f3f896Z0bd770a74d1160f7c9f12cd5912a06ebff6adcae899d92191fe4305673ba2ffe8fa3f1a4e5ad79f3f334b3b202b2173c486ea37ce3d397ed034c7f9dfeb15c5e927336d0441f4c4300e2cff0d0900b52d3b4r   ZT0ac1af7002b3d761d1e55298da9d0506b9ae52057220a306e07b6b87e8df21d0ea00033de03984d34918Z,0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0cZT2c91117204d745f3500d636a62f64f0ab3bae548aa53d423b0d1f27ebba6f5e5673a081d70cce7acfc48c                    s^   | j D ]R  fdddD \}}}} fdddD \}}t||||d|}| || qd S )Nc                    s   g | ]}t  | qS r    )r"   .0xr]   r    r!   
<listcomp>I  s     z$HKDF_Tests.test1.<locals>.<listcomp>)r+   r,   r-   rE   c                    s   g | ]} | qS r    r    ri   rl   r    r!   rm   J  s     )r.   r   r+   )_test_vectorr   r0   )r1   secretrI   infoexpZkey_lenrU   outputr    rl   r!   r4   G  s
    
zHKDF_Tests.test1c                 C   s   t tddtdt}t tddtdtd\}}| |d d |dd  f||f t tddtdtd\}}}| |d d |dd |dd  f|||f d S )	NZXXXXXX   ZYYYY   r,   r.   r-   r`   )r   r   r   r0   r1   refkey1key2Zkey3r    r    r!   rR   O  s
    $zHKDF_Tests.test2)r$   r%   r&   r
   r   rn   r4   rR   r    r    r    r!   rf      sh   grf   c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	scrypt_Tests))r   r   r*   r+   r+   a%  
                    77 d6 57 62 38 65 7b 20 3b 19 ca 42 c1 8a 04 97
                    f1 6b 48 44 e3 07 4a e8 df df fa 3f ed e2 14 42
                    fc d0 06 9d ed 09 48 f8 32 6a 75 3a 0f c8 1f 17
                    e8 d3 e0 fb 2e 0d 36 28 cf 35 e2 0c 38 d1 89 06
                    )r(   ZNaCli   r`   r*   a%  
                    fd ba be 1c 9d 34 72 00 78 56 e7 19 0d 01 e9 fe
                    7c 6a d7 cb c8 23 78 30 e7 73 76 63 4b 37 31 62
                    2e af 30 d9 2e 22 a3 88 6f f1 09 27 9d 98 30 da
                    c7 27 af b9 4a 83 ee 6d 83 60 cb df a2 cc 06 40
                    )pleaseletmeinSodiumChloridei @  r`   r+   a%  
                    70 23 bd cb 3a fd 73 48 46 1c 06 cd 81 fd 38 eb
                    fd a8 fb ba 90 4f 8e 3e a9 b5 43 f6 54 5d a1 f2
                    d5 43 29 55 61 3f 0f cf 62 d4 97 05 24 2a 9a f9
                    e6 1e 85 dc 0d 65 1e 40 df cf 01 7b 45 57 58 87
                    )rz   r{      r`   r+   a%  
                    21 01 cb 9b 6a 51 1a ae ad db be 09 cf 70 f8 81
                    ec 56 8d 57 4a 2f fd 4d ab e5 ee 98 20 ad aa 47
                    8e 56 fd 8f 4b a5 d0 9f fa 1c 6d 92 7c 40 f4 c3
                    37 30 40 49 e8 a9 52 fb cb f4 5c 6f a7 7a 41 a4
                    c                 C   s~   g }| j D ]h}t }t|d |_t|d |_|d |_|d |_|d |_t|d |_	t
|j	|_|| q
|| _ d S )Nr   r+   r,   r-   r.   rE   )datar#   r   PSNrr@   r"   rr   rG   dkLenappend)r1   Znew_test_vectorsr]   Znew_tvr    r    r!   setUp  s    



zscrypt_Tests.setUpc                 C   s   | j D ]}z"t|j|j|j|j|j|j}W nZ tk
r } z<dt	|krr|jdkrrdd l
}|dt W Y qn|W 5 d }~X Y nX | ||j qd S )Nz 2 r|   r   z6Not enough memory to unit test scrypt() with N=1048576)r}   r   r~   r   r   r   r   r@   rQ   strwarningswarnRuntimeWarningr0   rr   )r1   r]   rr   er   r    r    r!   rR     s    
"
zscrypt_Tests.test2c                 C   s   t tdtddddd}t tdtdddddd\}}| |d d |dd  f||f t tdtdddddd	\}}}| |d d |dd
 |d
d  f|||f d S )Nr(   rI   rs   r*   r+   rt   r,   r.   r-   r`   )r   r   r0   ru   r    r    r!   rX     s
     $"zscrypt_Tests.test3N)r$   r%   r&   r}   r   rR   rX   r    r    r    r!   ry   [  s   	7ry   c                   @   sd   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd ZdS )bcrypt_Testsc                 C   s   |  ttdd |  ttdd |  ttdd | j ttdddd | j ttddd	d | j ttddd
d |  ttdd d S )NsI   1111111111111111111111111111111111111111111111111111111111111111111111111rY   s
   1111111111r-       r.   rW   rI      1s   11111111111111111s   1 1 1 1 1 1 1 1 1 1 )rP   rQ   r   r1   r    r    r!   test_negative_cases  s    z bcrypt_Tests.test_negative_casesc                 C   sv   t dd}td| t|}td| |d d t|d dA  }| ttd| d|dd   }| ttd| d S )Npwdr.   r+      x)r   r   	bytearrayr   rP   rQ   )r1   rv   ZbrefZwrongr    r    r!   test_bytearray_mismatch  s    


z$bcrypt_Tests.test_bytearray_mismatchc                 C   sT   dddddg}t |D ]8\}\}}}}t||t|d}| || t|| qd S )N)rW   r.      zVHmKQtGGQob.b/Nc7l9NOs<   $2a$04$zVHmKQtGGQob.b/Nc7l9NO8UlrYcW05FiuCj/SxsFO/ZtiN9.mNzy)rW   rE   r   s<   $2a$05$zVHmKQtGGQob.b/Nc7l9NOWES.1hkVBgy5IWImh9DOjKNU8atY4Iy)rW   rt   r   s<   $2a$06$zVHmKQtGGQob.b/Nc7l9NOjOl7l4oz3WSh5fJ6414Uw8IXRAUoiaO)rW      r   s<   $2a$07$zVHmKQtGGQob.b/Nc7l9NOBsj1dQpBA1HYNGpIETIByoNX9jc.hOi)rW   r`   r   s<   $2a$08$zVHmKQtGGQob.b/Nc7l9NOiLTUh/9MDpX86/DLyEzyiFjqjBFePgOr   	enumerater   r   r0   r   r1   Ztvsidxr(   ZcostZsalt64r_   rk   r    r    r!   test_empty_password  s    z bcrypt_Tests.test_empty_passwordc                 C   sZ   ddddddddg}t |D ]8\}\}}}}t||t|d	}| || t|| qd S )
N)s
   <.S.2K(Zq'r.   s   VYAclAMpaXY/oqAo9yUpkus<   $2a$04$VYAclAMpaXY/oqAo9yUpkuWmoYywaPzyhu56HxXpVltnBIfmO9tgu)s
   5.rApO%5jArE   s   kVNDrnYKvbNr5AIcxNzeIus<   $2a$05$kVNDrnYKvbNr5AIcxNzeIuRcyIF5cZk6UrwHGxENbxP5dVv.WQM/G)s
   oW++kSrQW^rt   s   QLKkRMH9Am6irtPeSKN5sOs<   $2a$06$QLKkRMH9Am6irtPeSKN5sObJGr3j47cO6Pdf5JZ0AsJXuze0IbsNm)s
   ggJ\KbTnDGr   s   4H896R09bzjhapgCPS/LYus<   $2a$07$4H896R09bzjhapgCPS/LYuMzAQluVgR5iu/ALF8L8Aln6lzzYXwbq)s
   49b0:;VkH/r`   s   hfvO2retKrSrx5f2RXikWes<   $2a$08$hfvO2retKrSrx5f2RXikWeFWdtSesPlbj08t/uXxCeZoHRWDz/xFe)s
   >9N^5jc##'	   s   XZLvl7rMB3EvM0c1.JHivus<   $2a$09$XZLvl7rMB3EvM0c1.JHivuIDPJWeNJPTVrpjZIEVRYYB/mF6cYgJK)s
   \$ch)s4WXprY   s   aIjpMOLK5qiS9zjhcHR5TOs<   $2a$10$aIjpMOLK5qiS9zjhcHR5TOU7v2NFDmcsBmSFDt5EHOgp/jeTF3O/q)s
   RYoj\_>2P7rs   s   esIAHiQAJNNBrsr5V13l7.s<   $2a$12$esIAHiQAJNNBrsr5V13l7.RFWWJI2BZFtQlkFyiWXjou05GyuREZar   r   r   r    r    r!   &test_random_password_and_salt_short_pw  s    z3bcrypt_Tests.test_random_password_and_salt_short_pwc                 C   sZ   ddddddddg}t |D ]8\}\}}}}t||t|d	}| || t|| qd S )
N)s   ^Q&"]A`%/A(BVGt>QaX0M-#<Q148&fr.   s   vrRP5vQxyD4LrqiLd/oWROs<   $2a$04$vrRP5vQxyD4LrqiLd/oWROgrrGINsw3gb4Ga5x2sn01jNmiLVECl6)s   nZa!rRf\U;OL;R?>1ghq_+":Y0CRmYrE   s   YuQvhokOGVnevctykUYpKus<   $2a$05$YuQvhokOGVnevctykUYpKutZD2pWeGGYn3auyLOasguMY3/0BbIyq)s   F%uN/j>[GuB7-jB'_Yj!Tnb7Y!u^6)rt   s   5L3vpQ0tG9O7k5gQ8nAHAes<   $2a$06$5L3vpQ0tG9O7k5gQ8nAHAe9xxQiOcOLh8LGcI0PLWhIznsDt.S.C6)s   Z>BobP32ub"Cfe*Q<<WUq3rc=[GJr-r   s   hp8IdLueqE6qFh1zYycUZ.s<   $2a$07$hp8IdLueqE6qFh1zYycUZ.twmUH8eSTPQAEpdNXKMlwms9XfKqfea)s   Ik&8N['7*[1aCc1lOm8\jWeD*H$eZMr`   s   2ANDTYCB9m7vf0Prh7rSrus<   $2a$08$2ANDTYCB9m7vf0Prh7rSrupqpO3jJOkIz2oW/QHB4lCmK7qMytGV6)s   O)=%3[E$*q+>-q-=tRSjOBh8\mLNW.r   s   nArqOfdCsD9kIbVnAixnwes<   $2a$09$nArqOfdCsD9kIbVnAixnwe6s8QvyPYWtQBpEXKir2OJF9/oNBsEFe)s   /MH51`!BP&0tj3%YCA;Xk%e3S`o\EIrY   s   ePiAc.s.yoBi3B6p1iQUCes<   $2a$10$ePiAc.s.yoBi3B6p1iQUCezn3mraLwpVJ5XGelVyYFKyp5FZn/y.u)s   ptAP"mcg6oH.";c0U2_oll.OKi<!kurs   s   aroG/pwwPj1tU5fl9a9pkOs<   $2a$12$aroG/pwwPj1tU5fl9a9pkO4rydAmkXRj/LqfHZOSnR6LGAZ.z.jwar   r   r   r    r    r!   %test_random_password_and_salt_long_pw  s    z2bcrypt_Tests.test_random_password_and_salt_long_pwc                 C   sP   dddg}t |D ]8\}\}}}}t||t|d}| || t|| qd S )N)   Q/A:k3DP;X@=<0"hg&9cr.   s   wbgDTvLMtyjQlNK7fjqwyOs<   $2a$04$wbgDTvLMtyjQlNK7fjqwyOakBoACQuYh11.VsKNarF4xUIOBWgD6S)r   rE   s   zbAaOmloOhxiKItjznRqrus<   $2a$05$zbAaOmloOhxiKItjznRqrunRqHlu3MAa7pMGv26Rr3WwyfGcwoRm6)r   rt   s   aOK0bWUvLI0qLkc3ti5jyus<   $2a$06$aOK0bWUvLI0qLkc3ti5jyuAIQoqRzuqoK09kQqQ6Ou/YKDhW50/qar   r   r   r    r    r!   "test_same_password_and_random_salt  s    z/bcrypt_Tests.test_same_password_and_random_saltc                 C   sZ   ddddddddg}t |D ]8\}\}}}}t||t|d	}| || t|| qd S )
N)   o<&+X'F4AQ8H,LU,N`&rr.      BK5u.QHk1Driey7bvnFTH.s<   $2a$04$BK5u.QHk1Driey7bvnFTH.3smGwxd91PtoK2GxH5nZ7pcBsYX4lMq)r   rE   r   s<   $2a$05$BK5u.QHk1Driey7bvnFTH.t5P.jZvFBMzDB1IY4PwkkRPOyVbEtFG)r   rt   r   s<   $2a$06$BK5u.QHk1Driey7bvnFTH.6Ea1Z5db2p25CPXZbxb/3OyKQagg3pa)r   r   r   s<   $2a$07$BK5u.QHk1Driey7bvnFTH.sruuQi8Lhv/0LWKDvNp3AGFk7ltdkm6)r   r`   r   s<   $2a$08$BK5u.QHk1Driey7bvnFTH.IE7KsaUzc4m7gzAMlyUPUeiYyACWe0q)r   r   r   s<   $2a$09$BK5u.QHk1Driey7bvnFTH.1v4Xj1dwkp44QNg0cVAoQt4FQMMrvnS)r   rY   r   s<   $2a$10$BK5u.QHk1Driey7bvnFTH.ESINe9YntUMcVgFDfkC.Vbhc9vMhNX2)r   rs   r   s<   $2a$12$BK5u.QHk1Driey7bvnFTH.QM1/nnGe/f5cTzb6XTTi/vMzcAnycqGr   r   r   r    r    r!   2test_same_password_and_salt_increasing_cost_factor  s    z?bcrypt_Tests.test_same_password_and_salt_increasing_cost_factorc                 C   sP   dddg}t |D ]8\}\}}}}t||t|d}| || t|| qd S )N)sF   g*3Q45="8NNgpT&mbMJ$Omfr.#ZeW?FP=CE$#roHd?97uL0F-]`?u73c"\[."*)qU34@VGr.   s   T2XJ5MOWvHQZRijl8LIKkOs<   $2a$04$T2XJ5MOWvHQZRijl8LIKkOQKIyX75KBfuLsuRYOJz5OjwBNF2lM8a)sF   \M+*8;&QE=Ll[>5?Ui"^ai#iQH7ZFtNMfs3AROnIncE9"BNNoEgO[[*Yk8;RQ(#S,;I+aTrE   s   wgkOlGNXIVE2fWkT3gyRoOs<   $2a$05$wgkOlGNXIVE2fWkT3gyRoOqWi4gbi1Wv2Q2Jx3xVs3apl1w.Wtj8C)sF   M.E1=dt<.L0Q&p;94NfGm_Oo23+Kpl@M5?WIAL.[@/:'S)W96G8N^AWb7_smmC]>7#fGoBrt   s   W9zTCl35nEvUukhhFzkKMes<   $2a$06$W9zTCl35nEvUukhhFzkKMekjT9/pj7M0lihRVEZrX3m8/SBNZRX7ir   r   r   r    r    r!   test_long_passwords.  s    	z bcrypt_Tests.test_long_passwordsc                 C   sj   ddddddddd	d
ddddddg}t |D ]8\}\}}}}t||t|d}| || t|| q,d S )N)   ar.      5DCebwootqWMCp59ISrMJ.s<   $2a$04$5DCebwootqWMCp59ISrMJ.l4WvgHIVg17ZawDIrDM2IjlE64GDNQS)s   aar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.AyUxBk.ThHlsLvRTH7IqcG7yVHJ3SXq)s   aaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.BxOVac5xPB6XFdRc/ZrzM9FgZkqmvbW)s   aaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.Qbr209bpCtfl5hN7UQlG/L4xiD3AKau)s   aaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.oWszihPjDZI0ypReKsaDOW1jBl7oOii)s   aaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ./k.Xxn9YiqtV/sxh3EHbnOHd0Qsq27K)s   aaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.PYJqRFQbgRbIjMd5VNKmdKS4sBVOyDe)s   aaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ..VMYfzaw1wP/SGxowpLeGf13fxCCt.q)s	   aaaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.5B0p054nO5WgAD1n04XslDY/bqY9RJi)s
   aaaaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.INBTgqm7sdlBJDg.J5mLMSRK25ri04y)s   aaaaaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.s3y7CdFD0OR5p6rsZw/eZ.Dla40KLfm)s   aaaaaaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.Jx742Djra6Q7PqJWnTAS.85c28g.Siq)s   aaaaaaaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.oKMXW3EZcPHcUV0ib5vDBnh9HojXnLu)s   aaaaaaaaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.w6nIjWpDPNSH5pZUvLjC1q25ONEQpeS)s   aaaaaaaaaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.k1b2/r9A/hxdwKEKurg6OCn4MwMdiGq)s   aaaaaaaaaaaaaaaar.   r   s<   $2a$04$5DCebwootqWMCp59ISrMJ.3prCNHVX1Ws.7Hm2bJxFUnQOX9f7DFar   r   r   r    r    r!   test_increasing_password_length>  s*    z,bcrypt_Tests.test_increasing_password_lengthc                 C   sf   ddddddddd	d
ddddg}t |D ]8\}\}}}}t||t|d}| || t|| q(d S )N)0   àèìòùÀÈÌÒÙáéíóúýÁÉÍÓÚÝðÐr.   s   D3qS2aoTVyqM7z8v8crLm.s<   $2a$04$D3qS2aoTVyqM7z8v8crLm.3nKt4CzBZJbyFB.ZebmfCvRw7BGs.Xm)r   rE   s   VA1FujiOCMPkUHQ8kF7IaOs<   $2a$05$VA1FujiOCMPkUHQ8kF7IaOg7NGaNvpxwWzSluQutxEVmbZItRTsAa)r   rt   s   TXiaNrPeBSz5ugiQlehRt.s<   $2a$06$TXiaNrPeBSz5ugiQlehRt.gwpeDQnXWteQL4z2FulouBr6G7D9KUi),   âêîôûÂÊÎÔÛãñõÃÑÕäëïöüÿr.   s   YTn1Qlvps8e1odqMn6G5x.s<   $2a$04$YTn1Qlvps8e1odqMn6G5x.85pqKql6w773EZJAExk7/BatYAI4tyO)r   rE   s   C.8k5vJKD2NtfrRI9o17DOs<   $2a$05$C.8k5vJKD2NtfrRI9o17DOfIW0XnwItA529vJnh2jzYTb1QdoY0py)r   rt   s   xqfRPj3RYAgwurrhcA6uROs<   $2a$06$xqfRPj3RYAgwurrhcA6uROtGlXDp/U6/gkoDYHwlubtcVcNft5.vW)+   ÄËÏÖÜŸåÅæÆœŒßçÇøØ¢¿¡€r.   s   y8vGgMmr9EdyxP9rmMKjH.s<   $2a$04$y8vGgMmr9EdyxP9rmMKjH.wv2y3r7yRD79gykQtmb3N3zrwjKsyay)r   rE   s   iYH4XIKAOOm/xPQs7xKP1us<   $2a$05$iYH4XIKAOOm/xPQs7xKP1upD0cWyMn3Jf0ZWiizXbEkVpS41K1dcO)r   rt   s   wCOob.D0VV8twafNDB2apes<   $2a$06$wCOob.D0VV8twafNDB2apegiGD5nqF6Y1e6K95q6Y.R8C4QGd265q)u8   ΔημοσιεύθηκεστηνΕφημερίδατηςr.   s   E5SQtS6P4568MDXW7cyUp.s<   $2a$04$E5SQtS6P4568MDXW7cyUp.18wfDisKZBxifnPZjAI1d/KTYMfHPYO)u8   АБбВвГгДдЕеЁёЖжЗзИиЙйКкЛлМмНr.   s   03e26gQFHhQwRNf81/ww9.s<   $2a$04$03e26gQFHhQwRNf81/ww9.p1UbrNwxpzWjLuT.zpTLH4t/w5WhAhC)uF   нОоПпРрСсТтУуФфХхЦцЧчШшЩщЪъЫыЬьЭэЮюr.   s   PHNoJwpXCfe32nUtLv2Upus<   $2a$04$PHNoJwpXCfe32nUtLv2UpuhJXOzd4k7IdFwnEpYwfJVCZ/f/.8Pje)u6   電电電島岛島兔兔兎龜龟亀國国国區区区r.   s   wU4/0i1TmNl2u.1jIwBX.us<   $2a$04$wU4/0i1TmNl2u.1jIwBX.uZUaOL3Rc5ID7nlQRloQh6q5wwhV/zLW)u6   诶比伊艾弗豆贝尔维吾艾尺开艾丝维贼德r.   s   P4kreGLhCd26d4WIy7DJXus<   $2a$04$P4kreGLhCd26d4WIy7DJXusPkhxLvBouzV6OXkL5EB0jux0osjsryr   r   r   r    r    r!   test_non_ascii_charactersX  s&    z&bcrypt_Tests.test_non_ascii_charactersc                 C   sb   ddddddddd	d
ddg}t |D ]8\}\}}}}t||t|d}| || t|| q$d S )N)z
-O_=*N!2JPr.      ......................s<   $2a$04$......................JjuKLOX9OOwo5PceZZXSkaLDvdmgb82)z
7B[$Q<4b>UrE   r   s<   $2a$05$......................DRiedDQZRL3xq5A5FL8y7/6NM8a2Y5W)z
>d5-I_8^.hrt   r   s<   $2a$06$......................5Mq1Ng8jgDY.uHNU4h5p/x6BedzNH2W)z
)V`/UM/]1tr.      .OC/.OC/.OC/.OC/.OC/.Os<   $2a$04$.OC/.OC/.OC/.OC/.OC/.OQIvKRDAam.Hm5/IaV/.hc7P8gwwIbmi)z
:@t2.bWuH]rE   r   s<   $2a$05$.OC/.OC/.OC/.OC/.OC/.ONDbUvdOchUiKmQORX6BlkPofa/QxW9e)z
b(#KljF5s"rt   r   s<   $2a$06$.OC/.OC/.OC/.OC/.OC/.OHfTd9e7svOu34vi1PCvOcAEq07ST7.K)z
@3YaJ^Xs]*r.      eGA.eGA.eGA.eGA.eGA.e.s<   $2a$04$eGA.eGA.eGA.eGA.eGA.e.stcmvh.R70m.0jbfSFVxlONdj1iws0C)z
'"5\!k*C(prE   r   s<   $2a$05$eGA.eGA.eGA.eGA.eGA.e.vR37mVSbfdHwu.F0sNMvgn8oruQRghy)z
edEu7C?$'Wrt   r   s<   $2a$06$eGA.eGA.eGA.eGA.eGA.e.tSq0FN8MWHQXJXNFnHTPQKtA.n2a..G)z
N7dHmg\PI^r.      999999999999999999999us<   $2a$04$999999999999999999999uCZfA/pLrlyngNDMq89r1uUk.bQ9icOu)z
"eJuHh!)7*rE   r   s<   $2a$05$999999999999999999999uj8Pfx.ufrJFAoWFLjapYBS5vVEQQ/hK)z
ZeDRJ:_tu:rt   r   s<   $2a$06$999999999999999999999u6RB0P9UmbdbQgjoQFEJsrvrKe.BoU6qr   r   r   r    r    r!   test_special_case_saltp  s"    z#bcrypt_Tests.test_special_case_saltN)r$   r%   r&   r   r   r   r   r   r   r   r   r   r   r   r    r    r    r!   r     s   	r   c                   @   sD   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )TestVectorsHKDFWycheproofc                 C   s   t j|  || _d| _d S )NNone)unittestTestCase__init___wycheproof_warnings_id)r1   wycheproof_warningsr    r    r!   r     s    z"TestVectorsHKDFWycheproof.__init__c                 C   s2   dd }dd }t d|d| d|id|id	}|S )
Nc                 S   sH   | d }|dkrt S |dkr tS |dkr,tS |dkr8tS td| d S )N	algorithmz
HKDF-SHA-1zHKDF-SHA-256zHKDF-SHA-384zHKDF-SHA-512zUnknown algorithm )r   r
   r   r   rQ   )rootZ	algo_namer    r    r!   filter_algo  s    z8TestVectorsHKDFWycheproof.add_tests.<locals>.filter_algoc                 S   s   t | d S )Nsizeint)unitr    r    r!   filter_size  s    z8TestVectorsHKDFWycheproof.add_tests.<locals>.filter_size)ProtocolZ
wycheproofzWycheproof HMAC (%s)hash_moduler   )Zroot_tagZunit_tag)r   )r1   filenamer   r   r_   r    r    r!   	add_tests  s    z#TestVectorsHKDFWycheproof.add_testsc                 C   s2   g | _ | d | d | d | d d S )Nzhkdf_sha1_test.jsonzhkdf_sha256_test.jsonzhkdf_sha384_test.jsonzhkdf_sha512_test.json)r]   r   r   r    r    r!   r     s
    


zTestVectorsHKDFWycheproof.setUpc                 C   s   | j S r;   )r   r   r    r    r!   shortDescription  s    z*TestVectorsHKDFWycheproof.shortDescriptionc                 C   s.   |j r*| jr*dd l}|d| j|jf  d S )Nr   zWycheproof warning: %s (%s))warningr   r   r   r   comment)r1   r]   r   r    r    r!   r     s    zTestVectorsHKDFWycheproof.warnc                 C   s   d|j |j|jf | _z t|j|j|j|jd|j	}W n t
k
rT   |jrPtY n,X ||jkrl|jrtn|jsvt| | d S )Nz!Wycheproof HKDF Test #%d (%s, %s)r+   )idr   r   r   r   Zikmr   rI   r   rp   rQ   ZvalidAssertionErrorZokmr   )r1   r]   rd   r    r    r!   test_verify  s     

z%TestVectorsHKDFWycheproof.test_verifyc                 C   s   | j D ]}| | qd S r;   )r]   r   )r1   r]   r    r    r!   runTest  s    
z!TestVectorsHKDFWycheproof.runTestN)
r$   r%   r&   r   r   r   r   r   r   r   r    r    r    r!   r     s   r   c                 C   s   t d|  t t dgS )NzCryptodome.Hash.r=   )
__import__globalslocals)	hash_namer    r    r!   load_hash_by_name  s    r   c                   @   s   e Zd Zdd Zdd ZdS )SP800_108_Counter_Testsc                 C   sV   dd }zt dd|dd}W n tk
r:   | d Y nX | jtt dd|dd d S )	Nc                 S   s   t | |t S r;   rC   rA   rk   r    r    r!   rN     s    z9SP800_108_Counter_Tests.test_negative_zeroes.<locals>.prf   0000000000000000r+   s   A B)labelz+SP800_108_Counter failed with zero in label)context)r   rQ   ZfailrP   )r1   rN   _r    r    r!   test_negative_zeroes  s    z,SP800_108_Counter_Tests.test_negative_zeroesc                 C   sr   dd }d}t |d|}dD ]P}t |||d| }| t|d|  | t|d | | d|| qd S )Nc                 S   s   t | |t S r;   rC   r   r    r    r!   rN     s    z7SP800_108_Counter_Tests.test_multiple_keys.<locals>.prfr      )r+   r,   r-      r   rW   )r   r0   rG   join)r1   rN   rd   rJ   r   Zdksr    r    r!   test_multiple_keys  s    z*SP800_108_Counter_Tests.test_multiple_keysN)r$   r%   r&   r   r   r    r    r    r!   r     s   r   c           
      C   s   t dddddd ipg }d }t|D ]\}}t|trtd|}|rh|dd	d
}t|}d}q&td|}|r~d}q&|st	|dkr|fdd}n|dkr|fdd}q&||j
|j|j|j|jfdd}	t| d| |	 q&d S )N)r   zKDF_SP800_108_COUNTER.txtz NIST SP 800 108 KDF Counter Modecountc                 S   s   t | S r;   r   )rk   r    r    r!   rV     rW   z-add_tests_sp800_108_counter.<locals>.<lambda>z\[HMAC-(SHA-[0-9]+)\]r+   -r   hmacz\[CMAC-AES-128\]Zcmacc                 S   s   t | || S r;   rS   rA   rk   r   r    r    r!   rN      s    z(add_tests_sp800_108_counter.<locals>.prfc                 S   s   t | |t S r;   )ZCMACr=   r   r>   r   r    r    r!   rN     s    c                 S   s:   t |t||d||}t|t|ks*t| || d S )Nr+   )r   rG   r   r0   )r1   rN   kinr   r   koutr   r_   r    r    r!   kdf_test  s    z-add_tests_sp800_108_counter.<locals>.kdf_testztest_kdf_sp800_108_counter_%d)r   r   
isinstancer   rematchgroupr   r   r   r   r   r   r   r   setattr)
clsZtest_vectors_sp800_108_counterZmac_typer   r]   r3   r   r   rN   r   r    r    r!   add_tests_sp800_108_counter  s@    


  
r   c                 C   s   |  d}|  ds4tjd d t_tjd d t_g }|tt7 }|tt7 }|tt7 }|tt7 }|t	|g7 }|tt7 }|tt
7 }|tt7 }|S )Nr   Z
slow_testsr-   )getr5   r/   ry   r}   r   r'   r[   rf   r   r   r   )configr   testsr    r    r!   	get_tests  s    

r   __main__c                   C   s   t t S r;   )r   Z	TestSuiter   r    r    r    r!   rV   (  rW   rV   suite)ZdefaultTest)3r   r   binasciir   ZCryptodome.Util.py3compatr   r   ZCryptodome.SelfTest.st_commonr   ZCryptodome.SelfTest.loaderr   r   ZCryptodome.Hashr   r	   r
   r   r   r   r   ZCryptodome.Cipherr   r   ZCryptodome.Protocol.KDFr   r   r   r   r   r   r   r   r   r"   objectr#   r   r'   r5   r[   rf   ry   r   r   r   r   r   r   r$   r   mainr    r    r    r!   <module>   s8   $(V9 h EB)
