U
    Š¾úgà!  ã                   @   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rŠdd„ Zejdd 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 )<ÚElGamalTestZ@BA4CAEAAED8CBE952AFD2126C63EB3B345D65C2A0A73D2A3AD4138B6D09BD933Z05Z@60D063600ECED7C7C55146020E7A31C4476E9793BEAED420FEC9E77604CAE4EFZ.1D391BA2EE3C37FE1BA175A69B2C73A11238AD77675932Z.F5893C5BAB4131264066F57AB3D8AD89E391A0B68A68A1Z48656C6C6F207468657265Z@32BFD5F487966CEA9E9356715788C491EC515E4ED48B58F0F00971E93AAA5EC7Z@7BE8FBFF317C93E82FCEF9BD515284BA506603FEA25D01C0CB874A31F315EE68)ÚpÚgÚyÚxÚkÚptÚct1Úct2Z€F1B18AE9F7B4E08FDA9A04832F4E919D89462FD31BF12F92791A93519F75076D6CE3942689CDFF2F344CAFF0F82D01864F69F3AECF566C774CBACF728B81A227Z07Z€688628C676E4F05D630E1BE39D0066178CA7AA83836B645DE5ADD359B4825A12B02EF4252E4E6FA9BEC1DB0BE90F6D7C8629CABB6E531F472B2664868156E20CZ.14E60B1BDFD33436C0DA8A22FDC14A2CCDBBED0627CE68Z.38DBF14E1F319BDA9BAB33EEEADCAF6B2EA5250577ACE7Z€290F8530C2CC312EC46178724F196F308AD4C523CEABB001FACB0506BFED676083FE0F27AC688B5C749AB3CB8A80CD6F7094DBA421FB19442F5A413E06A9772BZ€1D69AAAD1DC50493FB1B8E8721D621D683F3BF1321BE21BC4A43E11B40C9D4D9C80DE3AAC2AB60D31782B16B61112E68220889D53C4C3136EE6F6CE61F8A23A0Z@D2F3C41EA66530838A704A48FFAC9334F4701ECE3A97CEE4C69DD01AE7129DD7Z@C3F9417DC0DAFEA6A05C1D2333B7A95E63B3F4F28CC962254B3256984D1012E7Z.165E4A39BE44D5A2D8B1332D416BC559616F536BC735BBZ@C7F0C794A7EAD726E25A47FF8928013680E73C51DD3D7D99BFDA8F492585928FZ@35CA98133779E2073EF31165AFCDEB764DD54E96ADE851715495F9C635E1E7C2Z@0135B88B1151279FE5D8078D4FC685EE81177EE9802AB123A73925FC1CB059A7)r
   r   r   r   r   ÚhÚsig1Úsig2Z€E24CF3A4B8A6AF749DCA6D714282FE4AABEEE44A53BB6ED15FBE32B5D3C3EF9CC4124A2ECA331F3C1C1B667ACA3766825217E7B5F9856648D95F05330C6A19CFZ0BZ€2AD3A1049CA5D4ED207B2431C79A8719BB4073D4A94E450EA6CEE8A760EB07ADB67C0D52C275EE85D7B52789061EE45F2F37D9B2AE522A51C28329766BFE68ACZ.16CBB4F46D9ECCF24FF9F7E63CAA3BD8936341555062ABZ€8A3D89A4E429FD2476D7D717251FB79BF900FFE77444E6BB8299DC3F84D0DD57ABAB50732AE158EA52F5B9E7D8813E81FD9F79470AE22F8F1CF9AEC820A78C69Z€BE001AABAFFF976EC9016198FBFEA14CBEF96B000CCC0063D3324016F9E91FE80D8F9325812ED24DDB2B4D4CF4430B169880B3CE88313B53255BD4EC0378586FZ€5E266F3F837BA204E3BBB6DBECC0611429D96F8C7CE8F4EFDF9D4CB681C2A954468A357BF4242CEC7418B51DFC081BCD21299EF5B5A0DDEF3A139A1817503DDEc                 C   s   |   d¡ d S )Né´   )Ú_test_random_key)Úself© r   úN/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/SelfTest/PublicKey/test_ElGamal.pyÚtest_generate_180`   s    zElGamalTest.test_generate_180c                 C   sf   | j D ]Z}|  |d¡}t |d ¡}| |d |d ¡}|  |d |d ¡ |  |d |d ¡ qd S )	NTÚkeyr   r   r   r   é   r   )ÚtveÚ
convert_tvr   Ú	constructÚ_encryptÚassertEqual)r   ÚtvÚdr   Úctr   r   r   Útest_encryptionc   s    
zElGamalTest.test_encryptionc                 C   sP   | j D ]D}|  |d¡}t |d ¡}| |d |d f¡}|  ||d ¡ qd S )NTr   r   r   r   )r   r   r   r   Ú_decryptr!   )r   r"   r#   r   r   r   r   r   Útest_decryptionk   s
    
zElGamalTest.test_decryptionc                 C   sb   | j D ]V}|  |d¡}t |d ¡}| |d |d ¡\}}|  ||d ¡ |  ||d ¡ qd S )NTr   r   r   r   r   )Útvsr   r   r   Ú_signr!   )r   r"   r#   r   r   r   r   r   r   Útest_signingr   s    
zElGamalTest.test_signingc                 C   sz   | j D ]n}|  |d¡}t |d ¡}| |d |d |d f¡}|  |¡ | |d |d d |d f¡}|  |¡ qd S )NTr   r   r   r   r   )r(   r   r   r   Ú_verifyÚ
assertTrueÚassertFalse)r   r"   r#   r   Úresr   r   r   Ú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   r   r   é   é   ©Úlistr   r(   ÚassertRaisesÚ
ValueErrorr   r   ©r   ÚtupZtup0r   r   r   Ú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   r   r   r0   r2   r6   r   r   r   Útest_bad_key4’   s    zElGamalTest.test_bad_key4r   c                 C   s|   d}i }|  ¡ D ]8}t|| ƒ||< |s8||ks8|dkrt|| ƒ||< qg |d< |D ] }|d  || g7  < ||= qV|S )zfConvert a test vector from textual form (hexadecimal ascii
        to either integers or byte strings.)r
   r   r   r   )r   r   r   )Úkeysr   r   )r   r"   Zas_longsZ	key_compsZtv2Úcr   r   r   r   —   s    zElGamalTest.convert_tvc                 C   sF   t  |t ¡ j¡}|  |¡ |  |¡ | ¡ }|  |¡ |  	|¡ d S ©N)
r   Úgenerater   ÚnewÚreadÚ_check_private_keyÚ_exercise_primitiveZ	publickeyÚ_check_public_keyÚ_exercise_public_primitive)r   ÚbitsÚelgObjZpubr   r   r   r   ¦   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 op|jd k n  ¡ |  t|j|j|jƒ|j¡ d S ©Nr   )r,   Úhas_privater   r
   r!   Úpowr   r   ©r   rE   r   r   r   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 rF   )r-   rG   r,   r   r
   r!   rH   rI   r   r   r   rB   ¹   s    &zElGamalTest._check_public_keyc                 C   sB   d}|  |d¡}| |¡}|  ||¡ | |d¡}| ||¡ d S )Niòð éÍ[i±hÞ:)r    r&   r!   r)   r+   )r   rE   Ú	plaintextÚ
ciphertextZ
plaintextPÚ	signaturer   r   r   rA   Â   s    
zElGamalTest._exercise_primitivec                 C   s   d}|  |d¡}d S )NiŒßŠrJ   )r    )r   rE   rK   rL   r   r   r   rC   Í   s    z&ElGamalTest._exercise_public_primitiveN)r   )Ú__name__Ú
__module__Ú__qualname__r   r(   r   r%   r'   r*   r/   r8   r9   r   r   r@   rB   rA   rC   r   r   r   r   r	   "   sf   ÷÷ó÷÷ô
	r	   c                 C   s   g }|t tƒ7 }|S r<   )r   r	   )ÚconfigÚtestsr   r   r   Ú	get_testsÑ   s    rS   Ú__main__c                   C   s   t  tƒ ¡S r<   )ÚunittestZ	TestSuiterS   r   r   r   r   Ú<lambda>×   ó    rV   Úsuite)ZdefaultTest)Ú__doc__Z__revision__rU   ZCryptodome.SelfTest.st_commonr   r   r   Z
Cryptodomer   ZCryptodome.PublicKeyr   ZCryptodome.Util.numberr   ZCryptodome.Util.py3compatZTestCaser	   rS   rN   rX   Úmainr   r   r   r   Ú<module>   s    0