U
    ÈZjÂ'  ã                   @   s¶   d dl mZmZmZ d dlmZmZ d dlT d dlm	Z	 d dl
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G dd„ deƒZi fdd„Zedkr²dd„ Zedd� dS )é    )ÚmainÚTestCaseÚ	TestSuite)Ú	unhexlifyÚhexlify)Ú*)ÚSHAKE128)Úlist_test_cases)ÚShamirÚ_ElementÚ	_mult_gf2Ú_div_gf2c                   @   s   e Zd Zdd„ Zdd„ ZdS )Ú	GF2_Testsc                 C   s¼   t ddƒ}|  |d¡ t ddƒ}|  |d¡ d}t ||ƒ}|  |d¡ t ||ƒ}|  |d¡ t ||ƒ}|  |d¡ ddd	d
g}d}d}t ||ƒ}d}|D ]}|t ||ƒN }q˜|  ||¡ d S )Nr   é"   é   é   é   é   é   é   é€   l        l   …    iŒ  )r   ÚassertEqual)ÚselfÚxÚzÚyÚcompsZ	sum_compsÚw© r   ún/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/Crypto/SelfTest/Protocol/test_SecretSharing.pyÚtest_mult_gf2/   s&    





zGF2_Tests.test_mult_gf2c                 C   st   ddl m} tddƒ\}}|  ||ƒ|dƒk ¡ t|dƒ|A }|  d|¡ tddƒ\}}|  |d¡ |  |d¡ d S )Nr   )Úsizei7  é   )ZCrypto.Util.numberr!   r   Z
assertTruer   r   )r   Údegr   r   r   r   r   r   Útest_div_gf2J   s    zGF2_Tests.test_div_gf2N)Ú__name__Ú
__module__Ú__qualname__r    r$   r   r   r   r   r   -   s   r   c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
ÚElement_Testsc                 C   s–   t dƒ}|  t|ƒd¡ |  | ¡ tdƒd tdƒ ¡ t tdƒd tdƒ ƒ}|  t|ƒd¡ |  | ¡ tdƒd tdƒ ¡ |  tt tdƒ¡ d S )Né   r   é   z zi  )r   r   ÚintÚencodeÚbchrÚbÚassertRaisesÚ
ValueError)r   Úer   r   r   Útest1Y   s      zElement_Tests.test1c                 C   s(   t dƒ}t dƒ}|  t|| ƒd¡ d S )Né   é
   é   ©r   r   r+   )r   r1   Úfr   r   r   Útest2f   s    zElement_Tests.test2c                 C   st   t dƒ}t dƒ}t dƒ}t dƒ| }|  t|ƒd¡ t dƒ| }|  t|ƒd¡ t dd ƒ| }|  t|ƒd¡ d S )Nr   r   é   é   é   é‡   r6   )r   ÚzeroÚoneÚtwor   r   r   r   Útest3l   s    zElement_Tests.test3c                 C   sH   t dƒ}| ¡ }|  t|ƒd¡ t dƒ}| ¡ }|  t|| ƒd¡ d S )Nr   iÓ)è)r   Zinverser   r+   )r   r>   r   r   r   r   r   Útest4{   s    zElement_Tests.test4N)r%   r&   r'   r2   r8   r@   rA   r   r   r   r   r(   W   s   r(   c                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚShamir_Testsc                 C   sh   t  ddtdƒd ¡}|  t|ƒd¡ tdƒD ]4}|  || d |d ¡ |  t|| d ƒd¡ q.d S )Nr9   r   éZ   r3   r   r   )r
   Úsplitr-   r   ÚlenÚrange)r   ÚsharesÚindexr   r   r   r2   ‰   s
    zShamir_Tests.test1c           
         s˜   ddl m} d}dd„ ‰ |D ]v‰ˆd }tˆd ƒ}d}t|tdtˆƒƒ|ƒƒD ]@\}}||krd q‡ ‡fd	d
„|D ƒ}t |d¡}	|  ||	¡ qPqd S )Nr   )Úpermutations))r9   Z d9fe73909bae28b3757854c0af7ad405z"1-594ae8964294174d95c33756d2504170z"2-d897459d29da574eb40e93ec552ffe6ez"3-5823de9bf0e068b054b5f07a28056b1bz"4-db2c1f8bff46d748f795da995bd080cb)r9   Z bf4f902d9a7efafd1f3ffd9291fd5de9z"1-557bd3b0748064b533469722d1cc7935z"2-6b2717164783c66d47cd28f2119f14d0z"3-8113548ba97d58256bb4424251ae300cz"4-179e9e5a218483ddaeda57539139cf04)r   Z ec96aa5c14c9faa699354cf1da74e904z"1-64579fbf1908d66f7239bf6e2b4e41e1z"2-6cd9428df8017b52322561e8c672ae3ez"3-e418776ef5c0579bd9299277374806ddz"4-ab3f77a0107398d23b323e581bb43f5dz"5-23fe42431db2b41bd03ecdc7ea8e97ac)r   Z 44cf249b68b80fcdc27b47be60c2c145z"1-d6515a3905cd755119b86e311c801e31z"2-16693d9ac9f10c254036ced5f8917fa3z"3-84f74338a48476b99bf5e75a84d3a0d1z"4-3fe8878dc4a5d35811cf3cbcd33dbe52z"5-ad76f92fa9d0a9c4ca0c1533af7f6132)	r   Z 5398717c982db935d968eebe53a47f5az"1-be7be2dd4c068e7ef576aaa1b1c11b01z"2-f821f5848441cb98b3eb467e2733ee21z"3-25ee52f53e203f6e29a0297b5ab486b5z"4-fc9fb58ef74dab947fbf9acd9d5d83cdz"5-b1949cce46d81552e65f248d3f74cc5cz"6-d64797f59977c4d4a7956ad916da7699z"7-ab608a6546a8b9af8820ff832b1135c7)	r   Z 4a78db90fbf35da5545d2fb728e87596z"1-08daf9a25d8aa184cfbf02b30a0ed6a0z"2-dda28261e36f0b14168c2cf153fb734ez"3-e9fdec5505d674a57f9836c417c1ecaaz"4-4dce5636ae06dee42d2c82e65f06c735z"5-3963dc118afc2ba798fa1d452b28ef00z"6-6dfe6ff5b09e94d2f84c382b12f42424z"7-6faea9d4d4a4e201bf6c90b9000630c3)r4   Z eccbf6d66d680b49b073c4f1ddf804aaz#01-7d8ac32fe4ae209ead1f3220fda34466z#02-f9144e76988aad647d2e61353a6e96d5z#03-b14c3b80179203363922d60760271c98z#04-770bb2a8c28f6cee89e00f4d5cc7f861z#05-6e3d7073ea368334ef67467871c66799z#06-248792bc74a98ce024477c13c8fb5f8dz#07-fcea4640d2db820c0604851e293d2487z#08-2776c36fb714bb1f8525a0be36fc7dbaz#09-6ee7ac8be773e473a4bf75ee5f065762z#10-33657fc073354cf91d4a68c735aacfc8z#11-7645c65094a5868bf225c516fdee2d0cz#12-840485aacb8226631ecd9c70e3018086)r4   Z 377e63bdbb5f7d4dc58a483d035212bbz#01-32c53260103be431c843b1a633afe3bdz#02-0107eb16cb8695084d452d2cc50bc7d6z#03-df1e5c66cd755287fb0446faccd72a06z#04-361bbcd5d40797f49dfa1898652da197z#05-160d3ad1512f7dec7fd9344aed318591z#06-659af6d95df4f25beca4fb9bfee3b7e8z#07-37f3b208977bad50b3724566b72bfa9dz#08-6c1de2dfc69c2986142c26a8248eb316z#09-5e19220837a396bd4bc8cd685ff314c3z#10-86e7b864fb0f3d628e46d50c1ba92f1cz#11-065d0082c80b1aea18f4abe0c49df72ez#12-84a09430c1d20ea9f388f3123c3733a3c                 S   s.   |   d¡}t| d |… ƒt| |d d … ƒfS )Nú-r   )Úfindr+   r   )ÚpÚposr   r   r   Ú	get_shareÚ   s    
z%Shamir_Tests.test2.<locals>.get_sharer   r4   r9   c                    s   g | ]}ˆ ˆ| ƒ‘qS r   r   )Ú.0r   ©rN   Útvr   r   Ú
<listcomp>å   s     z&Shamir_Tests.test2.<locals>.<listcomp>T)	Ú	itertoolsrI   r   Ú	enumeraterF   rE   r
   Úcombiner   )
r   rI   Ztest_vectorsÚkÚsecretZ	max_permsÚpermZ
shares_idxrG   Úresultr   rP   r   r8   ‘   s    C zShamir_Tests.test2c                 C   st   t  d¡}tdƒD ]\}| d¡}t dd|¡}t |d d… ¡}|  ||¡ t |d |d g¡}|  ||¡ qd S )Ns   test3éd   r3   r9   r   r   ©r   ÚnewrF   Úreadr
   rD   rU   r   )r   ÚrngÚ_rW   rG   Úsecret2Zsecret3r   r   r   r@   é   s    

zShamir_Tests.test3c                 C   s°   t  d¡}tdƒD ]B}| d¡}tjdd|dd�}tj|d d… dd�}|  ||¡ qtdƒD ]L}| d¡}tjdd|dd�}tj|d |d	 |d
 gdd�}|  ||¡ q^d S )Ns   test4r4   r3   r9   r   T)Zssssr"   r   r:   r[   )r   r^   r_   rW   rG   r`   r   r   r   rA   ú   s    


 zShamir_Tests.test4c                 C   s:   t tdƒƒ}t dd|¡}|  ttj|d |d f¡ d S )NZ 000102030405060708090a0b0c0d0e0fr9   r   r   )r   r.   r
   rD   r/   r0   rU   )r   rW   rG   r   r   r   Útest5  s    zShamir_Tests.test5N)r%   r&   r'   r2   r8   r@   rA   ra   r   r   r   r   rB   ‡   s
   XrB   c                 C   s,   g }|t tƒ7 }|t tƒ7 }|t tƒ7 }|S ©N)r	   r   r(   rB   )ÚconfigZtestsr   r   r   Ú	get_tests  s
    rd   Ú__main__c                   C   s
   t tƒ ƒS rb   )r   rd   r   r   r   r   Ú<lambda>   ó    rf   Úsuite)ZdefaultTestN)Zunittestr   r   r   Úbinasciir   r   ZCrypto.Util.py3compatZCrypto.Hashr   ZCrypto.SelfTest.st_commonr	   ZCrypto.Protocol.SecretSharingr
   r   r   r   r   r(   rB   rd   r%   rh   r   r   r   r   Ú<module>"   s   *0 