a m?Z?ddl@mAZAmBZBddlCmDZDddlEmFZFmGZGddlHmIZIdd ZJd d ZKd d ZLddZMddZNddZOddZPddZQddZRddZSddZTddZUd d!ZVd"d#ZWd$d%ZXd&d'ZYd(d)ZZd*d+Z[d,d-Z\d.d/Z]d0d1Z^d2d3Z_d4d5Z`d6d7Zad8d9Zbd:d;Zcdd?Zed@dAZfdBdCZgdDdEZhdFdGZidHdIZjdJdKZkdLdMZldNdOZmdPdQZndRdSZodTdUZpdVdWZqdXdYZrdZd[Zsd\d]Ztd^d_Zud`daZvdbdcZwdddeZxdfdgZydhdiZzdjdkZ{dldmZ|dnS)osymbols); cycle_listencipher_shiftencipher_affineencipher_substitutioncheck_and_joinencipher_vigeneredecipher_vigenere encipher_hill decipher_hillencipher_bifid5encipher_bifid6 bifid5_square bifid6_squarebifid5bifid6decipher_bifid5decipher_bifid6encipher_kid_rsadecipher_kid_rsakid_rsa_private_keykid_rsa_public_key decipher_rsarsa_private_keyrsa_public_key encipher_rsalfsr_connection_polynomiallfsr_autocorrelation lfsr_sequence encode_morse decode_morseelgamal_private_keyelgamal_public_keyencipher_elgamaldecipher_elgamaldh_private_key dh_public_key dh_shared_keydecipher_shiftdecipher_affineencipher_bifiddecipher_bifid bifid_square padded_keyuniq decipher_gm encipher_gm gm_public_keygm_private_key encipher_bg decipher_bgbg_private_key bg_public_keyencipher_rot13decipher_rot13encipher_atbashdecipher_atbashNonInvertibleCipherWarningencipher_railfencedecipher_railfence)Matrix)isprimeis_primitive_root)FF)raiseswarns) randrangecCs:tdddksJtdddks$Jtdddks6JdS)N hello world hlowrdel ol horel ollwd hwe olordll)r=rMrMn/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/crypto/tests/test_crypto.pytest_encipher_railfencesrOcCs:tdddksJtdddks$Jtdddks6JdS)NrHrGrFrJrIrLrK)r>rMrMrMrNtest_decipher_railfence srPcCsFtddgdksJtddgdks,JtddgdksBJdS)NrIrK)rIrrGrQ)rQrGrIr)rrMrMrMrNtest_cycle_list&srScCsLtdddksJtdddks$Jtdddks6JtdddksHJdS)NABCrrQBCDrRZZAB)rr)rMrMrMrNtest_encipher_shift,srVcCsDtddksJtddks Jtddks0Jtddks@JdS)NrTNOP)r8r9rMrMrMrNtest_encipher_rot132srXcCstdddksJtdddks$Jtdddks6Jtdddd d ksLJtd dd d d ksbJtdddkstJtdddksJdS)NrT)rQr)rQrQrU)rRrZAZY)rRrQABCDrBAD1231234Z214)rIZQTW)rr*rMrMrMrNtest_encipher_affine9sr^cCsDtddksJtddks Jtddks0Jtddks@JdS)NrTZZYX)r:r;rMrMrMrNtest_encipher_atbashBsr_cCs,tddddksJtddddks(JdS)NrTBACr[Z1243r\Z124)rrMrMrMrNtest_encipher_substitutionHsracCs`tddksJttddks$Jtddks8JtdddddksNJttdddS) NabcZaaabczab caT)filtercSs tddS)Nabrc)rrMrMrMrNRz%test_check_and_join..)rr/splitrC ValueErrorrMrMrMrNtest_check_and_joinMs rjcCsntdddksJtdddddks(Jtdddddks>JtdddddksTJtd dddd ksjJdS) NrTZACErYrZACAABZACCZACA)r rMrMrMrNtest_encipher_vigenereUs rmcCsntdddksJtdddddks(Jtdddddks>JtdddddksTJtdddddksjJdS) NrTZAAArYrrkZAACZAArl)r rMrMrMrNtest_decipher_vigenere]s rncCstddgd}td|dks"Jtddgd}td|dksDJtd|dddksZJtddgd}td|dddksJtd|ddd ksJtd |d ksJtd |d d dksJdS)NrGrQrGrIrYCFIVrQrrrQrCBABrkCBZABAZCFGCZ)padZCFYV)r?r rlrMrMrNtest_encipher_hillesrxcCstddgd}td|dks"Jtddgd}td|dksDJtd|dddksZJtddgd}td|dddksJtd|ddd ksJtd |d ksJdS) NrGrorqrYrrrrsrtrkZCFAZABAA)r?r rwrMrMrNtest_decipher_hilltsrycCsLtdddksJtdddks$Jtdddks6Jtddd ksHJdS) NrkCDCOrecZCHa bcbr`)r rMrMrMrNtest_encipher_bifid5srcs0tfdd}tdd|}td|ks,JdS)Ncstd||S)NrprijrwrMrNrfrgz$test_bifid5_square..rp)rr?rfMrMrwrNtest_bifid5_squares  rcCsLtdddksJtdddks$Jtdddks6JtdddksHJdS) Nrkr{rzchr|b acr~rT)rrMrMrMrNtest_decipher_bifid5srcCsLtdddksJtdddks$Jtdddks6Jtddd ksHJdS) NrkrzCPrer|CIr}r~r`)rrMrMrMrNtest_encipher_bifid6srcCsLtdddksJtdddks$Jtdddks6JtdddksHJdS) Nrkrrzcir|rr~rT)rrMrMrMrNtest_decipher_bifid6srcs0tfdd}tdd|}td|ks,JdS)Ncstd||S)NrrrwrMrNrfrgz$test_bifid6_square..rr)rr?rrrMrwrNtest_bifid6_squares  rcCs|tddddksJtddddks(Jtt8tddddksFJtdddd usZJWdn1sn0YdS) NrGrIrQrrQrprIrKrQF)rrDr<rMrMrMrNtest_rsa_public_keys  rcCstddddksJtddddks(Jtdddd ks.)r!rCKeyErrorrMrMrMrNtest_decode_morsesrcCsttddttddtd}t|dg|dgd|d|dgksPJt|dg|dgd|d|dgks|Jtd}t|dg|dgd|d|dgksJt|dg|dgd|d|dgksJt|dg|dgd|d|dgks JdS)NcSstddgdSNrQrrMrMrMrNrfrgz$test_lfsr_sequence..cSstdgddSrrrMrMrMrNrfrgrGrQrrI)rC TypeErrorrBr)FrMrMrNtest_lfsr_sequences,,,,rcCsfttddtd}t|d|dg|d|dgd}t|dddksNJt|dddksbJdS)NcSs tdddS)NrQrGrI)rrMrMrMrNrfrgz+test_lfsr_autocorrelation..rGrQrrprR)rCrrBrr)rrrMrMrNtest_lfsr_autocorrelations $rcCstd}td}t|d|dg|d|dgd}t||ddksLJt|d|dg|d|dgd}t||d|dksJdS)NrGxrQrrp)rBrrr)rrrrMrMrNtest_lfsr_connection_polynomials $$rcCsBtdd\}}}t|sJt||s*Jtt|dks>JdSNddigitf)r"r@rAlenbin)rcr~_rMrMrNtest_elgamal_private_keys rcsbtdt}|ddttd|ks8JttfddttfdddS)NrprrQcs tS)Nr$rMPdkrMrNrfrgztest_elgamal..cs tdS)NrRrrM)rrMrNrfrg)r"r#r%r$rCri)ZekrMrrN test_elgamals  rcCsBtdd\}}}t|sJt||s*Jtt|dks>JdSr)r&r@rArr)rgrrMrMrNtest_dh_private_keys rcCsTtdd\}}}t|||f\}}}||ks0J||ks.)r&r'rEr(rrCri)rrrrr~skrMrMrNtest_dh_shared_keys   rcCs2tdddksJttddttdddS)Nr~rebacSs tddS)NreZacer.rMrMrMrNrfrgz!test_padded_key..cSs tddS)NreZabbarrMrMrMrNrfrg)r.rCrirMrMrMrNtest_padded_keysrcCs~ttddtddddks"JttddtddddksDJttddtd td ksfJtd td kszJdS) NcSs tdddS)Nrbr~abcde)r+rMrMrMrNrfrgztest_bifid..rbr~ZabcdbdbcSs tdddS)Nrr~r)r,rMrMrMrNrfrgcSstdS)Nr)r-rMrMrMrNrfrgBZBACDEFGHIKLMNOPQRSTUVWXYZB0Z$B0ACDEFGHIJKLMNOPQRSTUVWXYZ123456789)rCrir+rrrMrMrMrN test_bifidsrc Csngd}gd}gd}t||D]F\}}t||}|D].}t||}t||}t||} | |ks8Jq8q"dS)N))Yregkmqrrrrrrr) riWii9iS(iiii+iiH)zipr3r2r1r0 ZpsqsmessagesrrprirZpubencdecrMrMrNtest_encipher_decipher_gms    r cCsDttddttddttddds@JdtddkdS)NcSs tddS)Nrrr2rMrMrMrNrf rgz%test_gm_private_key..cSs tddS)Nrr!rMrMrMrNrf rgcSs tddS)Nrrpr!rMrMrMrNrfrgr)rCrir2rMrMrMrNtest_gm_private_key sr#cCs>dtdddksJdtdddks,Jttdd dS) NiCrr"rQrrIrpcSs tddS)Nrr"r!rMrMrMrNrfrgz$test_gm_public_key..)r2rCrirMrMrMrNtest_gm_public_keysr$c Csrgd}gd}}gd}t||D]F\}}t||}|D].}t||}t||}t||} | |ks.cSs tddS)Nrr.rMrMrMrNrf+rgcSs tddSNrrr.rMrMrMrNrf,rgrr*)rCrir6rMrMrMrNtest_bg_private_key)sr0cCs6dtddksJdtddks$JttdddS) Nir%r'irr*cSs tddSr/r.rMrMrMrNrf2rgz$test_bg_public_key..)r7rCrirMrMrMrNtest_bg_public_key/sr1N)}Z sympy.corerZsympy.crypto.cryptorrrrrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>Zsympy.matricesr?Z sympy.ntheoryr@rAZsympy.polys.domainsrBZsympy.testing.pytestrCrDZsympy.core.randomrErOrPrSrVrXr^r_rarjrmrnrxryrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r#r$r-r0r1rMrMrMrNsr          !2