o
    e"                     @   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 )z,Self-test suite for Crypto.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ct2ZF1B18AE9F7B4E08FDA9A04832F4E919D89462FD31BF12F92791A93519F75076D6CE3942689CDFF2F344CAFF0F82D01864F69F3AECF566C774CBACF728B81A227Z07Z688628C676E4F05D630E1BE39D0066178CA7AA83836B645DE5ADD359B4825A12B02EF4252E4E6FA9BEC1DB0BE90F6D7C8629CABB6E531F472B2664868156E20CZ.14E60B1BDFD33436C0DA8A22FDC14A2CCDBBED0627CE68Z.38DBF14E1F319BDA9BAB33EEEADCAF6B2EA5250577ACE7Z290F8530C2CC312EC46178724F196F308AD4C523CEABB001FACB0506BFED676083FE0F27AC688B5C749AB3CB8A80CD6F7094DBA421FB19442F5A413E06A9772BZ1D69AAAD1DC50493FB1B8E8721D621D683F3BF1321BE21BC4A43E11B40C9D4D9C80DE3AAC2AB60D31782B16B61112E68220889D53C4C3136EE6F6CE61F8A23A0Z@D2F3C41EA66530838A704A48FFAC9334F4701ECE3A97CEE4C69DD01AE7129DD7Z@C3F9417DC0DAFEA6A05C1D2333B7A95E63B3F4F28CC962254B3256984D1012E7Z.165E4A39BE44D5A2D8B1332D416BC559616F536BC735BBZ@C7F0C794A7EAD726E25A47FF8928013680E73C51DD3D7D99BFDA8F492585928FZ@35CA98133779E2073EF31165AFCDEB764DD54E96ADE851715495F9C635E1E7C2Z@0135B88B1151279FE5D8078D4FC685EE81177EE9802AB123A73925FC1CB059A7)r
   r   r   r   r   hsig1sig2ZE24CF3A4B8A6AF749DCA6D714282FE4AABEEE44A53BB6ED15FBE32B5D3C3EF9CC4124A2ECA331F3C1C1B667ACA3766825217E7B5F9856648D95F05330C6A19CFZ0BZ2AD3A1049CA5D4ED207B2431C79A8719BB4073D4A94E450EA6CEE8A760EB07ADB67C0D52C275EE85D7B52789061EE45F2F37D9B2AE522A51C28329766BFE68ACZ.16CBB4F46D9ECCF24FF9F7E63CAA3BD8936341555062ABZ8A3D89A4E429FD2476D7D717251FB79BF900FFE77444E6BB8299DC3F84D0DD57ABAB50732AE158EA52F5B9E7D8813E81FD9F79470AE22F8F1CF9AEC820A78C69ZBE001AABAFFF976EC9016198FBFEA14CBEF96B000CCC0063D3324016F9E91FE80D8F9325812ED24DDB2B4D4CF4430B169880B3CE88313B53255BD4EC0378586FZ5E266F3F837BA204E3BBB6DBECC0611429D96F8C7CE8F4EFDF9D4CB681C2A954468A357BF4242CEC7418B51DFC081BCD21299EF5B5A0DDEF3A139A1817503DDEc                 C   s   |  d d S )N   )_test_random_key)self r   VD:\Projects\ConvertPro\env\Lib\site-packages\Crypto/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)r   tvdr   ctr   r   r   test_encryptionc   s   
zElGamalTest.test_encryptionc                 C   sP   | j 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 ]+}| |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 ]7}| |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 ]}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   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 o8|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 [ih:)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	   "   sh    
	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>   s    rV   suite)ZdefaultTest)__doc__Z__revision__rU   ZCrypto.SelfTest.st_commonr   r   r   ZCryptor   ZCrypto.PublicKeyr   ZCrypto.Util.numberr   ZCrypto.Util.py3compatZTestCaser	   rS   rN   rW   mainr   r   r   r   <module>   s    0