
    ¿ig%                         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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)*)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/var/www/html/afkarena/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_gf2H   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.test1W   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   r4   )r   r:   fs      r!   test2zElement_Tests.test2d   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               r@   )r   zeroonetwor   s        r!   test3zElement_Tests.test3j   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   r4   )r   rI   r   r   s       r!   test4zElement_Tests.test4y   sv    qkkKKMMQ###XIIKKQUQ'''''r#   N)r+   r,   r-   r;   rB   rK   rN   r.   r#   r!   r0   r0   U   sP        9 9 9) ) )2 2 2	( 	( 	( 	( 	(r#   r0   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 )NrD   r   Z   r=   r   r   )r
   splitr6   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))rD    d9fe73909bae28b3757854c0af7ad405z"1-594ae8964294174d95c33756d2504170z"2-d897459d29da574eb40e93ec552ffe6ez"3-5823de9bf0e068b054b5f07a28056b1bz"4-db2c1f8bff46d748f795da995bd080cb)rD    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   )findr4   r   )pposs     r!   	get_sharez%Shamir_Tests.test2.<locals>.get_share   s>    &&++Cq#w<<1S1WXX;!7!777r#   r   r>   rD   c                 2    g | ]} |                   S r.   r.   ).0r   rg   tvs     r!   
<listcomp>z&Shamir_Tests.test2.<locals>.<listcomp>   s'    AAA99RU++AAAr#   T)		itertoolsrY   r   	enumeraterU   rT   r
   combiner   )r   rY   test_vectorsksecret	max_permsperm
shares_idxrV   resultrg   rj   s             @@r!   rB   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          t          d                    }t          j        dd|          }t          j        |d d                   }|                     ||           t          j        |d         |d         g          }|                     ||           d S N 000102030405060708090a0b0c0d0e0frD   r   r   r   r7   r
   rS   rn   r   )r   rq   rV   secret2secret3s        r!   rK   zShamir_Tests.test3   s    1?@@AAaF++.,,))).6!9fQi!9::)))))r#   c                     t          t          d                    }t          j        dd|d          }t          j        |d d         d          }|                     ||           d S )Nrx   rD   r   T)ssssry   )r   rq   rV   rz   s       r!   rN   zShamir_Tests.test4   sh    1?@@AAaF666.$777)))))r#   c                     t          t          d                    }t          j        dd|          }|                     t
          t          j        |d         |d         f           d S rw   )r   r7   r
   rS   r8   r9   rn   )r   rq   rV   s      r!   test5zShamir_Tests.test5   sX    1?@@AAaF++*fnvay&)6LMMMMMr#   N)r+   r,   r-   r;   rB   rK   rN   r   r.   r#   r!   rP   rP      sh        8 8 8T1 T1 T1l
* 
* 
** * *N N N N Nr#   rP   c                     g }|t          t                    z  }|t          t                    z  }|t          t                    z  }|S N)r	   r   r0   rP   )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.SelfTest.st_commonr	   !Cryptodome.Protocol.SecretSharingr
   r   r   r   r   r0   rP   r   r+   r   r.   r#   r!   <module>r      sx  D / . . . . . . . . . ' ' ' ' ' ' ' ' ' ' ' ' 9 9 9 9 9 9> > > > > > > > > > > >( ( ( ( ( ( ( (T-( -( -( -( -(H -( -( -(^zN zN zN zN zN8 zN zN zNz      z**EDW r#   