o
    5h'                     @   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   /var/www/html/construction_image-detection-poc/venv/lib/python3.10/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     )Crypto.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   test3d   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   Crypto.Util.py3compatCrypto.Hashr   Crypto.SelfTest.st_commonr	   Crypto.Protocol.SecretSharingr
   r   r   r   r   r+   rF   rw   r(   rz   r   r   r   r    <module>   s   !*0 