
    g%                         d dl mZmZmZ d dlmZmZ d dl 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       yy)    )mainTestCase	TestSuite)	unhexlifyhexlify)*)list_test_cases)Shamir_Element	_mult_gf2_div_gf2c                       e Zd Zd Zd Zy)	GF2_Testsc                    t        dd      }| j                  |d       t        dd      }| j                  |d       d}t        ||      }| j                  |d       t        ||      }| j                  |d       t        ||      }| j                  |d       g d}d	}d
}t        ||      }d}|D ]  }|t        ||      z  } | j                  ||       y )Nr   "                  )r         l        l       i  )r   assertEqual)selfxzycomps	sum_compsws          n/var/www/html/brdwt/brdwt/brdwtenv/lib/python3.12/site-packages/Crypto/SelfTest/Protocol/test_SecretSharing.pytest_mult_gf2zGF2_Tests.test_mult_gf2-   s    aNA b!BaOAaOBaOB #!	i# 	!A1a A	!A    c                    ddl m} t        dd      \  }}| j                   ||       |d      k         t	        |d      |z  }| j                  d|       t        dd      \  }}| j                  |d       | j                  |d       y )Nr   )sizei7     )Crypto.Util.numberr%   r   
assertTruer   r   )r   degr   r   r    s        r!   test_div_gf2zGF2_Tests.test_div_gf2H   s}    2Q1AQ(aOaa 31AAr#   N)__name__
__module____qualname__r"   r*    r#   r!   r   r   +   s    6r#   r   c                   $    e Zd Zd Zd Zd Zd Zy)Element_Testsc                    t        d      }| j                  t        |      d       | j                  |j                         t	        d      dz  t        d      z          t        t	        d      dz  t        d      z         }| j                  t        |      d       | j                  |j                         t	        d      dz  t        d      z          | j                  t        t         t	        d             y )N   r      z zi  )r   r   intencodebchrbassertRaises
ValueError)r   es     r!   test1zElement_Tests.test1W   s    SMQ%T!WRZ!J-%?@T!WRZ!J-/0Q'T!WRZ!J-%?@ 	*hQ8r#   c                 l    t        d      }t        d      }| j                  t        ||z         d       y )N   
      r   r   r4   )r   r:   fs      r!   test2zElement_Tests.test2d   s,    TNTNQqS4(r#   c                 B   t        d      }t        d      }t        d      }t        d      |z  }| j                  t        |      d       t        d      |z  }| j                  t        |      d       t        ddz        |z  }| j                  t        |      d       y )Nr   r               r@   )r   zeroonetwor   s        r!   test3zElement_Tests.test3j   s    {qkqkQK$Q#QK#Q#QVs"Q1r#   c                     t        d      }|j                         }| j                  t        |      d       t        d      }|j                         }| j                  t        ||z        d       y )Nr   i))r   inverser   r4   )r   rI   r   r   s       r!   test4zElement_Tests.test4y   sW    qkKKMQ#XIIKQUQ'r#   N)r+   r,   r-   r;   rB   rK   rN   r.   r#   r!   r0   r0   U   s    9)2	(r#   r0   c                   *    e Zd Zd Zd Zd Zd Zd Zy)Shamir_Testsc                    t        j                  ddt        d      dz        }| j                  t	        |      d       t        d      D ]>  }| j                  ||   d   |dz          | j                  t	        ||   d         d       @ y )NrD   r   Z   r=   r   r   )r
   splitr6   r   lenrange)r   sharesindexs      r!   r;   zShamir_Tests.test1   s~    aDHRK0Va(1X 	8EVE]1-uQw7Sq!12B7	8r#   c                 D   ddl m} d}d }|D ]  }|d   }t        |d         }d}t         |t	        dt        |            |            D ]L  \  }}	||kD  r J|	D 
cg c]  }
 |||
          }}
t        j                  |d      }| j                  ||       N  y c c}
w )	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                 b    | j                  d      }t        | d |       t        | |dz   d        fS )N-r   )findr4   r   )pposs     r!   	get_sharez%Shamir_Tests.test2.<locals>.get_share   s3    &&+Cq#w<1S1WX;!777r#   r   r>   rD   T)		itertoolsrY   r   	enumeraterU   rT   r
   combiner   )r   rY   test_vectorsrg   tvksecret	max_permsperm
shares_idxr   rV   results                r!   rB   zShamir_Tests.test2   s    *A
F	8  		1B1Ar!u%FI$-l5CG;La.P$Q 1 j)#5?A9RU+AA5  01			1 Bs   Bc                    t        t        d            }t        j                  dd|      }t        j                  |d d       }| j                  ||       t        j                  |d   |d   g      }| j                  ||       y N 000102030405060708090a0b0c0d0e0frD   r   r   r   r7   r
   rS   rj   r   )r   rn   rV   secret2secret3s        r!   rK   zShamir_Tests.test3   ss    1?@AaF+..,)..6!9fQi!9:)r#   c                     t        t        d            }t        j                  dd|d      }t        j                  |d d d      }| j                  ||       y )Nru   rD   r   T)ssssrv   )r   rn   rV   rw   s       r!   rN   zShamir_Tests.test4   sL    1?@AaF6..$7)r#   c                     t        t        d            }t        j                  dd|      }| j	                  t
        t        j                  |d   |d   f       y rt   )r   r7   r
   rS   r8   r9   rj   )r   rn   rV   s      r!   test5zShamir_Tests.test5   sG    1?@AaF+*fnnvay&)6LMr#   N)r+   r,   r-   r;   rB   rK   rN   r|   r.   r#   r!   rP   rP      s    8T1l
**Nr#   rP   c                 v    g }|t        t              z  }|t        t              z  }|t        t              z  }|S N)r	   r   r0   rP   )configtestss     r!   	get_testsr     s9    E	_Y''E	_]++E	_\**ELr#   __main__c                  (    t        t                     S r~   )r   r   r.   r#   r!   <lambda>r   	  s    Iik* r#   suite)defaultTestN)unittestr   r   r   binasciir   r   Crypto.Util.py3compatCrypto.SelfTest.st_commonr	   Crypto.Protocol.SecretSharingr
   r   r   r   r   r0   rP   r   r+   r   r.   r#   r!   <module>r      sr   D / . ' # 5> >( (T-(H -(^zN8 zNz   z*EW r#   