o
    Þý°jà!  ã                   @   s’   d Z dZddlZddlmZmZmZ ddlmZ ddl	m
Z
 ddlmZ ddlT G d	d
„ d
ejƒZi fdd„ZedkrGdd„ Zejdd� dS dS )z0Self-test suite for Cryptodome.PublicKey.ElGamalz$Id$é    N)Úlist_test_casesÚa2b_hexÚb2a_hex)ÚRandom)ÚElGamal)Úbytes_to_long)Ú*c                
   @   sÎ   e Zd Zddddddddd	œd
dddddddd	œgZdddddddddœdddddddddœgZdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Z	d)d*„ Z
d+d,„ Zd;d.d/„Zd0d1„ Zd2d3„ Zd4d5„ Zd6d7„ Zd8d9„ Zd:S )<ÚElGamalTestÚ@BA4CAEAAED8CBE952AFD2126C63EB3B345D65C2A0A73D2A3AD4138B6D09BD933Ú05Ú@60D063600ECED7C7C55146020E7A31C4476E9793BEAED420FEC9E77604CAE4EFÚ.1D391BA2EE3C37FE1BA175A69B2C73A11238AD77675932Ú.F5893C5BAB4131264066F57AB3D8AD89E391A0B68A68A1Ú48656C6C6F207468657265Ú@32BFD5F487966CEA9E9356715788C491EC515E4ED48B58F0F00971E93AAA5EC7Ú@7BE8FBFF317C93E82FCEF9BD515284BA506603FEA25D01C0CB874A31F315EE68)ÚpÚgÚyÚxÚkÚptÚct1Úct2Ú€F1B18AE9F7B4E08FDA9A04832F4E919D89462FD31BF12F92791A93519F75076D6CE3942689CDFF2F344CAFF0F82D01864F69F3AECF566C774CBACF728B81A227Ú07Ú€688628C676E4F05D630E1BE39D0066178CA7AA83836B645DE5ADD359B4825A12B02EF4252E4E6FA9BEC1DB0BE90F6D7C8629CABB6E531F472B2664868156E20CÚ.14E60B1BDFD33436C0DA8A22FDC14A2CCDBBED0627CE68Ú.38DBF14E1F319BDA9BAB33EEEADCAF6B2EA5250577ACE7Ú€290F8530C2CC312EC46178724F196F308AD4C523CEABB001FACB0506BFED676083FE0F27AC688B5C749AB3CB8A80CD6F7094DBA421FB19442F5A413E06A9772BÚ€1D69AAAD1DC50493FB1B8E8721D621D683F3BF1321BE21BC4A43E11B40C9D4D9C80DE3AAC2AB60D31782B16B61112E68220889D53C4C3136EE6F6CE61F8A23A0Ú@D2F3C41EA66530838A704A48FFAC9334F4701ECE3A97CEE4C69DD01AE7129DD7Ú@C3F9417DC0DAFEA6A05C1D2333B7A95E63B3F4F28CC962254B3256984D1012E7Ú.165E4A39BE44D5A2D8B1332D416BC559616F536BC735BBÚ@C7F0C794A7EAD726E25A47FF8928013680E73C51DD3D7D99BFDA8F492585928FÚ@35CA98133779E2073EF31165AFCDEB764DD54E96ADE851715495F9C635E1E7C2Ú@0135B88B1151279FE5D8078D4FC685EE81177EE9802AB123A73925FC1CB059A7)r   r   r   r   r   ÚhÚsig1Úsig2Ú€E24CF3A4B8A6AF749DCA6D714282FE4AABEEE44A53BB6ED15FBE32B5D3C3EF9CC4124A2ECA331F3C1C1B667ACA3766825217E7B5F9856648D95F05330C6A19CFÚ0BÚ€2AD3A1049CA5D4ED207B2431C79A8719BB4073D4A94E450EA6CEE8A760EB07ADB67C0D52C275EE85D7B52789061EE45F2F37D9B2AE522A51C28329766BFE68ACÚ.16CBB4F46D9ECCF24FF9F7E63CAA3BD8936341555062ABÚ€8A3D89A4E429FD2476D7D717251FB79BF900FFE77444E6BB8299DC3F84D0DD57ABAB50732AE158EA52F5B9E7D8813E81FD9F79470AE22F8F1CF9AEC820A78C69Ú€BE001AABAFFF976EC9016198FBFEA14CBEF96B000CCC0063D3324016F9E91FE80D8F9325812ED24DDB2B4D4CF4430B169880B3CE88313B53255BD4EC0378586FÚ€5E266F3F837BA204E3BBB6DBECC0611429D96F8C7CE8F4EFDF9D4CB681C2A954468A357BF4242CEC7418B51DFC081BCD21299EF5B5A0DDEF3A139A1817503DDEc                 C   s   |   d¡ d S )Né´   )Ú_test_random_key)Úself© r4   ú�/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/SelfTest/PublicKey/test_ElGamal.pyÚtest_generate_180`   s   zElGamalTest.test_generate_180c                 C   sf   | j D ]-}|  |d¡}t |d ¡}| |d |d ¡}|  |d |d ¡ |  |d |d ¡ qd S )	NTÚkeyr   r   r   r   é   r   )ÚtveÚ
convert_tvr   Ú	constructÚ_encryptÚassertEqual)r3   ÚtvÚdr7   Úctr4   r4   r5   Útest_encryptionc   s   
ûzElGamalTest.test_encryptionc                 C   sP   | j D ]"}|  |d¡}t |d ¡}| |d |d f¡}|  ||d ¡ qd S )NTr7   r   r   r   )r9   r:   r   r;   Ú_decryptr=   )r3   r>   r?   r7   r   r4   r4   r5   Útest_decryptionk   s   
üzElGamalTest.test_decryptionc                 C   sb   | j D ]+}|  |d¡}t |d ¡}| |d |d ¡\}}|  ||d ¡ |  ||d ¡ qd S )NTr7   r'   r   r(   r)   )Útvsr:   r   r;   Ú_signr=   )r3   r>   r?   r7   r(   r)   r4   r4   r5   Útest_signingr   s   
ûzElGamalTest.test_signingc                 C   sz   | j D ]7}|  |d¡}t |d ¡}| |d |d |d f¡}|  |¡ | |d |d d |d f¡}|  |¡ qd S )NTr7   r'   r(   r)   r8   )rD   r:   r   r;   Ú_verifyÚ
assertTrueÚassertFalse)r3   r>   r?   r7   Úresr4   r4   r5   Útest_verificationz   s   

 øzElGamalTest.test_verificationc                 C   sŠ   t |  | jd d¡d ƒd d…  }}|d  d7  < |  ttj|¡ |}d|d< |  ttj|¡ |}|d d |d< |  ttj|¡ d S )Nr   r8   r7   é   é   ©Úlistr:   rD   ÚassertRaisesÚ
ValueErrorr   r;   ©r3   ÚtupÚtup0r4   r4   r5   Útest_bad_key3…   s   &zElGamalTest.test_bad_key3c                 C   sB   t |  | jd d¡d ƒ }}|d  d7  < |  ttj|¡ d S )Nr   r8   r7   rL   rN   rR   r4   r4   r5   Útest_bad_key4’   s   zElGamalTest.test_bad_key4r   c                 C   s|   d}i }|  ¡ D ]}t|| ƒ||< |s||v s|dv r$t|| ƒ||< qg |d< |D ]}|d  || g7  < ||= q+|S )zfConvert a test vector from textual form (hexadecimal ascii
        to either integers or byte strings.)r   r   r   r   )r(   r)   r7   )Úkeysr   r   )r3   r>   Úas_longsÚ	key_compsÚtv2Úcr4   r4   r5   r:   —   s   €zElGamalTest.convert_tvc                 C   sF   t  |t ¡ j¡}|  |¡ |  |¡ | ¡ }|  |¡ |  	|¡ d S ©N)
r   Úgenerater   ÚnewÚreadÚ_check_private_keyÚ_exercise_primitiveÚ	publickeyÚ_check_public_keyÚ_exercise_public_primitive)r3   ÚbitsÚelgObjÚpubr4   r4   r5   r2   ¦   s   


zElGamalTest._test_random_keyc                 C   s˜   |   | ¡ ¡ |   d|j  k o|jd k n  ¡ |  t|j|jd |jƒd¡ |   d|j  k o8|jd k n  ¡ |  t|j|j|jƒ|j¡ d S ©Nr8   )rH   Úhas_privater   r   r=   Úpowr   r   ©r3   rf   r4   r4   r5   r`   ®   s
   && zElGamalTest._check_private_keyc                 C   sV   |   | ¡ ¡ |  d|j  k o|jd k n  ¡ |  t|j|jd |jƒd¡ d S rh   )rI   ri   rH   r   r   r=   rj   rk   r4   r4   r5   rc   ¹   s   &"zElGamalTest._check_public_keyc                 C   sB   d}|  |d¡}| |¡}|  ||¡ | |d¡}| ||¡ d S )Niòð éÍ[i±hÞ:)r<   rB   r=   rE   rG   )r3   rf   Ú	plaintextÚ
ciphertextÚ
plaintextPÚ	signaturer4   r4   r5   ra   Â   s   
zElGamalTest._exercise_primitivec                 C   s   d}|  |d¡}d S )NiŒßŠrl   )r<   )r3   rf   rm   rn   r4   r4   r5   rd   Í   s   z&ElGamalTest._exercise_public_primitiveN)r   )Ú__name__Ú
__module__Ú__qualname__r9   rD   r6   rA   rC   rF   rK   rU   rV   r:   r2   r`   rc   ra   rd   r4   r4   r4   r5   r	   "   sh    ÷÷ó÷÷ô
	r	   c                 C   s   g }|t tƒ7 }|S r\   )r   r	   )ÚconfigÚtestsr4   r4   r5   Ú	get_testsÑ   s   rv   Ú__main__c                   C   s   t  tƒ ¡S r\   )ÚunittestÚ	TestSuiterv   r4   r4   r4   r5   Ú<lambda>×   s    rz   Úsuite)ÚdefaultTest)Ú__doc__Ú__revision__rx   ÚCryptodome.SelfTest.st_commonr   r   r   Ú
Cryptodomer   ÚCryptodome.PublicKeyr   ÚCryptodome.Util.numberr   ÚCryptodome.Util.py3compatÚTestCaser	   rv   rq   r{   Úmainr4   r4   r4   r5   Ú<module>   s    0þ