o
    Þý°jÖ'  ã                   @   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 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¡ g d¢}d	}d
}t ||ƒ}d}|D ]	}|t ||ƒN }qJ|  ||¡ d S )Nr   é"   é   é   é   é   é   )r   é   é€   l        l   …    iŒ  )r   ÚassertEqual)ÚselfÚxÚzÚyÚcompsÚ	sum_compsÚw© r   ú¢/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/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  é   )ÚCryptodome.Util.numberr"   r   Ú
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   r4   Ú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   é   é   é   é‡   r9   )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   Úinverser   r.   )r   rA   r   r   r   r   r    Útest4{   s   zElement_Tests.test4N)r(   r)   r*   r5   r;   rC   rE   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 ]}|  || d |d ¡ |  t|| d ƒd¡ qd S )Nr<   r   éZ   r6   r   r   )r
   Úsplitr0   r   ÚlenÚrange)r   ÚsharesÚindexr   r   r    r5   ‰   s   þzShamir_Tests.test1c           
         s˜   ddl m} d}dd„ ‰ |D ];‰ˆd }tˆd ƒ}d}t|tdtˆƒƒ|ƒƒD ] \}}||kr2 n‡ ‡fd	d
„|D ƒ}t |d¡}	|  ||	¡ q(qd S )Nr   )Úpermutations))r<   Ú d9fe73909bae28b3757854c0af7ad405z"1-594ae8964294174d95c33756d2504170z"2-d897459d29da574eb40e93ec552ffe6ez"3-5823de9bf0e068b054b5f07a28056b1bz"4-db2c1f8bff46d748f795da995bd080cb)r<   Ú bf4f902d9a7efafd1f3ffd9291fd5de9z"1-557bd3b0748064b533469722d1cc7935z"2-6b2717164783c66d47cd28f2119f14d0z"3-8113548ba97d58256bb4424251ae300cz"4-179e9e5a218483ddaeda57539139cf04)r   Ú ec96aa5c14c9faa699354cf1da74e904z"1-64579fbf1908d66f7239bf6e2b4e41e1z"2-6cd9428df8017b52322561e8c672ae3ez"3-e418776ef5c0579bd9299277374806ddz"4-ab3f77a0107398d23b323e581bb43f5dz"5-23fe42431db2b41bd03ecdc7ea8e97ac)r   Ú 44cf249b68b80fcdc27b47be60c2c145z"1-d6515a3905cd755119b86e311c801e31z"2-16693d9ac9f10c254036ced5f8917fa3z"3-84f74338a48476b99bf5e75a84d3a0d1z"4-3fe8878dc4a5d35811cf3cbcd33dbe52z"5-ad76f92fa9d0a9c4ca0c1533af7f6132)	r   Ú 5398717c982db935d968eebe53a47f5az"1-be7be2dd4c068e7ef576aaa1b1c11b01z"2-f821f5848441cb98b3eb467e2733ee21z"3-25ee52f53e203f6e29a0297b5ab486b5z"4-fc9fb58ef74dab947fbf9acd9d5d83cdz"5-b1949cce46d81552e65f248d3f74cc5cz"6-d64797f59977c4d4a7956ad916da7699z"7-ab608a6546a8b9af8820ff832b1135c7)	r   Ú 4a78db90fbf35da5545d2fb728e87596z"1-08daf9a25d8aa184cfbf02b30a0ed6a0z"2-dda28261e36f0b14168c2cf153fb734ez"3-e9fdec5505d674a57f9836c417c1ecaaz"4-4dce5636ae06dee42d2c82e65f06c735z"5-3963dc118afc2ba798fa1d452b28ef00z"6-6dfe6ff5b09e94d2f84c382b12f42424z"7-6faea9d4d4a4e201bf6c90b9000630c3)r7   Ú 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)r7   Ú 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   r7   r<   c                    s   g | ]}ˆ ˆ| ƒ‘qS r   r   )Ú.0r   ©rZ   Útvr   r    Ú
<listcomp>å   s    z&Shamir_Tests.test2.<locals>.<listcomp>T)	Ú	itertoolsrM   r   Ú	enumeraterJ   rI   r
   Úcombiner   )
r   rM   Útest_vectorsÚkÚsecretÚ	max_permsÚpermÚ
shares_idxrK   Úresultr   r\   r    r;   ‘   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   r6   r<   r   r   ©r   ÚnewrJ   Úreadr
   rH   ra   r   )r   ÚrngÚ_rd   rK   Úsecret2Úsecret3r   r   r    rC   é   s   

özShamir_Tests.test3c                 C   s°   t  d¡}tdƒD ]!}| d¡}tjdd|dd�}tj|d d… dd�}|  ||¡ q	tdƒD ]&}| d¡}tjdd|dd�}tj|d |d	 |d
 gdd�}|  ||¡ q/d S )Ns   test4r7   r6   r<   r   T)Ússssr#   r   r=   rj   )r   rm   rn   rd   rK   ro   r   r   r    rE   ú   s   


 úzShamir_Tests.test4c                 C   s:   t tdƒƒ}t dd|¡}|  ttj|d |d f¡ d S )NÚ 000102030405060708090a0b0c0d0e0fr<   r   r   )r   r1   r
   rH   r2   r3   ra   )r   rd   rK   r   r   r    Útest5  s    zShamir_Tests.test5N)r(   r)   r*   r5   r;   rC   rE   rs   r   r   r   r    rF   ‡   s    XrF   c                 C   s,   g }|t tƒ7 }|t tƒ7 }|t tƒ7 }|S ©N)r	   r   r+   rF   )ÚconfigÚtestsr   r   r    Ú	get_tests  s
   rw   Ú__main__c                   C   s
   t tƒ ƒS rt   )r   rw   r   r   r   r    Ú<lambda>   s   
 ry   Úsuite)ÚdefaultTestN)Úunittestr   r   r   Úbinasciir   r   ÚCryptodome.Util.py3compatÚCryptodome.Hashr   ÚCryptodome.SelfTest.st_commonr	   Ú!Cryptodome.Protocol.SecretSharingr
   r   r   r   r   r+   rF   rw   r(   rz   r   r   r   r    Ú<module>   s   !*0 þ