
    1xh'                         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e          Z G d	 d
e          Z G d de          Zi fdZedk    rd Z ed           dS dS )    )mainTestCase	TestSuite)	unhexlifyhexlify)*)SHAKE128)list_test_cases)Shamir_Element	_mult_gf2_div_gf2c                       e Zd Zd Zd ZdS )	GF2_Testsc                    t          dd          }|                     |d           t          dd          }|                     |d           d}t          ||          }|                     |d           t          ||          }|                     |d           t          ||          }|                     |d           g d}d	}d
}t          ||          }d}|D ]}|t          ||          z  }|                     ||           d S )Nr   "                  )r         l        l       i  )r   assertEqual)selfxzycomps	sum_compsws          k/home/visionen/capture/venv/lib/python3.11/site-packages/Cryptodome/SelfTest/Protocol/test_SecretSharing.pytest_mult_gf2zGF2_Tests.test_mult_gf2/   s$   aNNA b!BaOOAaOOBaOOB #""!	i## 	! 	!A1a AAA    c                 ^   ddl m} t          dd          \  }}|                      ||           |d          k                t	          |d          |z  }|                     d|           t          dd          \  }}|                     |d           |                     |d           d S )Nr   )sizei7     )Cryptodome.Util.numberr&   r   
assertTruer   r   )r   degr   r   r!   s        r"   test_div_gf2zGF2_Tests.test_div_gf2J   s    666666Q1AQ(((aOOaa   31AAr$   N)__name__
__module____qualname__r#   r+    r$   r"   r   r   -   s2          6    r$   r   c                   &    e Zd Zd Zd Zd Zd ZdS )Element_Testsc                    t          d          }|                     t          |          d           |                     |                                t	          d          dz  t          d          z              t          t	          d          dz  t          d          z             }|                     t          |          d           |                     |                                t	          d          dz  t          d          z              |                     t          t           t	          d                     d S )N   r      z zi  )r   r   intencodebchrbassertRaises
ValueError)r   es     r"   test1zElement_Tests.test1Y   s    SMMQ%%%T!WWRZ!J--%?@@@T!WWRZ!J--/00Q'''T!WWRZ!J--%?@@@ 	*hQ88888r$   c                     t          d          }t          d          }|                     t          ||z             d           d S )N   
      r   r   r5   )r   r;   fs      r"   test2zElement_Tests.test2f   s>    TNNTNNQqS4(((((r$   c                    t          d          }t          d          }t          d          }t          d          |z  }|                     t          |          d           t          d          |z  }|                     t          |          d           t          ddz            |z  }|                     t          |          d           d S )Nr   r               rA   )r   zeroonetwor   s        r"   test3zElement_Tests.test3l   s    {{qkkqkkQKK$Q###QKK#Q###QVs"Q11111r$   c                 $   t          d          }|                                }|                     t          |          d           t          d          }|                                }|                     t          ||z            d           d S )Nr   i))r   inverser   r5   )r   rJ   r   r   s       r"   test4zElement_Tests.test4{   sv    qkkKKMMQ###XIIKKQUQ'''''r$   N)r,   r-   r.   r<   rC   rL   rO   r/   r$   r"   r1   r1   W   sP        9 9 9) ) )2 2 2	( 	( 	( 	( 	(r$   r1   c                   ,    e Zd Zd Zd Zd Zd Zd ZdS )Shamir_Testsc                 d   t          j        ddt          d          dz            }|                     t	          |          d           t          d          D ]V}|                     ||         d         |dz              |                     t	          ||         d                   d           Wd S )NrE   r   Z   r>   r   r   )r   splitr7   r   lenrange)r   sharesindexs      r"   r<   zShamir_Tests.test1   s    aDHHRK00Va(((1XX 	8 	8EVE]1-uQw777Sq!122B7777	8 	8r$   c                 `  
 ddl m} d}d 
|D ]d         }t          d                   }d}t           |t	          dt                              |                    D ]G\  }}||k    r n;
fd|D             }t          j        |d	          }	|                     ||	           Hd S )
Nr   )permutations))rE    d9fe73909bae28b3757854c0af7ad405z"1-594ae8964294174d95c33756d2504170z"2-d897459d29da574eb40e93ec552ffe6ez"3-5823de9bf0e068b054b5f07a28056b1bz"4-db2c1f8bff46d748f795da995bd080cb)rE    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)r?    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)r?    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                     |                      d          }t          | d |                   t          | |dz   d                    fS )N-r   )findr5   r   )pposs     r"   	get_sharez%Shamir_Tests.test2.<locals>.get_share   s>    &&++Cq#w<<1S1WXX;!7!777r$   r   r?   rE   c                 2    g | ]} |                   S r/   r/   ).0r   rh   tvs     r"   
<listcomp>z&Shamir_Tests.test2.<locals>.<listcomp>   s'    AAA99RU++AAAr$   T)		itertoolsrZ   r   	enumeraterV   rU   r   combiner   )r   rZ   test_vectorsksecret	max_permsperm
shares_idxrW   resultrh   rk   s             @@r"   rC   zShamir_Tests.test2   s    ******A
F	8 	8 	8  		1 		1B1Ar!u%%FI$-ll5CGG;L;La.P.P$Q$Q 1 1 j)##EAAAAAZAAA55  0000		1 		1r$   c                 |   t          j        d          }t          d          D ]}|                    d          }t	          j        dd|          }t	          j        |d d                   }|                     ||           t	          j        |d         |d         g          }|                     ||           d S )Ns   test3d   r>   rE   r   r   r	   newrV   readr   rT   ro   r   )r   rng_rr   rW   secret2secret3s          r"   rL   zShamir_Tests.test3   s     l8$$s 
	. 
	.AXXb\\F\!Q//FnVBQBZ00GVW---nvay&)%=>>GVW----
	. 
	.r$   c                    t          j        d          }t          d          D ]c}|                    d          }t	          j        dd|d          }t	          j        |d d         d          }|                     ||           dt          d          D ]p}|                    d          }t	          j        dd|d          }t	          j        |d         |d	         |d
         gd          }|                     ||           qd S )Ns   test4r?   r>   rE   r   T)ssssr'   r   rF   ry   )r   r|   r}   rr   rW   r~   s         r"   rO   zShamir_Tests.test4   s
    l8$$r 	. 	.AXXb\\F\!QT:::FnVBQBZd;;;GVW----r 	. 	.AXXb\\F\!QT:::FnfQiF1I%FTRRRGVW----	. 	.r$   c                     t          t          d                    }t          j        dd|          }|                     t
          t          j        |d         |d         f           d S )N 000102030405060708090a0b0c0d0e0frE   r   r   )r   r8   r   rT   r9   r:   ro   )r   rr   rW   s      r"   test5zShamir_Tests.test5  sX    1?@@AAaF++*fnvay&)6LMMMMMr$   N)r,   r-   r.   r<   rC   rL   rO   r   r/   r$   r"   rQ   rQ      sh        8 8 8V1 V1 V1p. . .". . .,N N N N Nr$   rQ   c                     g }|t          t                    z  }|t          t                    z  }|t          t                    z  }|S N)r
   r   r1   rQ   )configtestss     r"   	get_testsr     sB    E	_Y'''E	_]+++E	_\***ELr$   __main__c                  8    t          t                                S r   )r   r   r/   r$   r"   <lambda>r      s    Iikk** r$   suite)defaultTestN)unittestr   r   r   binasciir   r   Cryptodome.Util.py3compatCryptodome.Hashr	   Cryptodome.SelfTest.st_commonr
   !Cryptodome.Protocol.SecretSharingr   r   r   r   r   r1   rQ   r   r,   r   r/   r$   r"   <module>r      s  D / . . . . . . . . . ' ' ' ' ' ' ' ' ' ' ' ' $ $ $ $ $ $ 9 9 9 9 9 9> > > > > > > > > > > >( ( ( ( ( ( ( (T-( -( -( -( -(H -( -( -(`NN NN NN NN NN8 NN NN NNb      z**EDW r$   