o
    jg.M                     @   s  d dl mZ d dlmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z= d dl>m?Z? d dl@mAZA d dlBmCZCmDZD d dlEmFZF d dlGmHZHmIZI d dlJmKZK d	d
 ZLdd ZMdd ZNdd ZOdd ZPdd ZQdd ZRdd ZSdd ZTdd ZUdd ZVdd  ZWd!d" ZXd#d$ ZYd%d& ZZd'd( Z[d)d* Z\d+d, Z]d-d. Z^d/d0 Z_d1d2 Z`d3d4 Zad5d6 Zbd7d8 Zcd9d: Zdd;d< Zed=d> Zfd?d@ ZgdAdB ZhdCdD ZidEdF ZjdGdH ZkdIdJ ZldKdL ZmdMdN ZndOdP ZodQdR ZpdSdT ZqdUdV ZrdWdX ZsdYdZ Ztd[d\ Zud]d^ Zvd_d` Zwdadb Zxdcdd Zydedf Zzdgdh Z{didj Z|dkdl Z}dmdn Z~doS )p    symbols);
cycle_listencipher_shiftencipher_affineencipher_substitutioncheck_and_joinencipher_vigeneredecipher_vigenereencipher_hilldecipher_hillencipher_bifid5encipher_bifid6bifid5_squarebifid6_squarebifid5bifid6decipher_bifid5decipher_bifid6encipher_kid_rsadecipher_kid_rsakid_rsa_private_keykid_rsa_public_keydecipher_rsarsa_private_keyrsa_public_keyencipher_rsalfsr_connection_polynomiallfsr_autocorrelationlfsr_sequenceencode_morsedecode_morseelgamal_private_keyelgamal_public_keyencipher_elgamaldecipher_elgamaldh_private_keydh_public_keydh_shared_keydecipher_shiftdecipher_affineencipher_bifiddecipher_bifidbifid_square
padded_keyuniqdecipher_gmencipher_gmgm_public_keygm_private_keyencipher_bgdecipher_bgbg_private_keybg_public_keyencipher_rot13decipher_rot13encipher_atbashdecipher_atbashNonInvertibleCipherWarningencipher_railfencedecipher_railfence)gcd)Matrix)isprimeis_primitive_root)FF)raiseswarns)	randrangec                   C   s:   t dddks	J t dddksJ t dddksJ d S )Nhello world   hlowrdel ol   horel ollwd   hwe olordll)r=    rN   rN   V/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/crypto/tests/test_crypto.pytest_encipher_railfence      rP   c                   C   s:   t dddks	J t dddksJ t dddksJ d S )NrI   rH   rG   rK   rJ   rM   rL   )r>   rN   rN   rN   rO   test_decipher_railfence!   rQ   rR   c                   C   sF   t ddg dksJ t ddg dksJ t ddg dks!J d S )NrJ   rL   )rJ   r      rH   rS   )rS   rH   rJ   r   )r   rN   rN   rN   rO   test_cycle_list'      rU   c                   C   sL   t dddks	J t dddksJ t dddksJ tdddks$J d S )NABCr   rS   BCDrT   ZAB)r   r)   rN   rN   rN   rO   test_encipher_shift-      rZ   c                   C   D   t ddksJ t ddksJ tddksJ tddks J d S )NrW   NOP)r8   r9   rN   rN   rN   rO   test_encipher_rot133      r^   c                   C   s   t dddks	J t dddksJ t dddksJ t dddd	d
ks&J t dddd	dks1J t dddks:J tdddksCJ d S )NrW   )rS   r   )rS   rS   rX   )rT   r   AZY)rT   rS   ABCDr   BAD1231234214)rJ      QTW)r   r*   rN   rN   rN   rO   test_encipher_affine:   s   rh   c                   C   r\   )NrW   ZYX)r:   r;   rN   rN   rN   rO   test_encipher_atbashC   r_   rj   c                   C   s,   t ddddks
J t ddddksJ d S )NrW   BACrc   1243rd   124)r   rN   rN   rN   rO   test_encipher_substitutionI   s   rn   c                   C   s`   t ddksJ t tddksJ t d dksJ t dddddks'J ttdd  d S )	Nabcaaabczab caT)filterc                   S   
   t ddS )Nabrq   )r   rN   rN   rN   rO   <lambda>S      
 z%test_check_and_join.<locals>.<lambda>)r   r/   splitrD   
ValueErrorrN   rN   rN   rO   test_check_and_joinN   s
   ry   c                   C   sn   t dddks	J t dddddksJ t dddddksJ t dddddks*J t d	dddd	ks5J d S )
NrW   ACEra   r   ACAABACCACA)r	   rN   rN   rN   rO   test_encipher_vigenereV   
   r   c                   C   sn   t dddks	J t dddddksJ t dddddksJ t dddddks*J t dddddks5J d S )	NrW   AAAra   r   r|   AACAAr   )r
   rN   rN   rN   rO   test_decipher_vigenere^   r   r   c                  C   s   t ddg d} td| dksJ t ddg d} td| dks"J td| dddks-J t ddg d} td| dddks@J td| ddd	ksKJ td
| dksTJ td
| dddks_J d S )NrH   rS   rH   rJ      ra   CFIVrS   r   r   rS   r   CBABr|   CBABACFGCZ)padCFYV)r@   r   r   rN   rN   rO   test_encipher_hillf   s   r   c                  C   s   t ddg d} td| dksJ t ddg d} td| dks"J td| dddks-J t ddg d} td| dddks@J td| ddd	ksKJ td
| dksTJ d S )NrH   r   r   ra   r   r   r   r   r|   CFAABAA)r@   r   r   rN   rN   rO   test_decipher_hillu   s   r   c                   C   L   t dddks	J t dddksJ t dddksJ t ddd	ks$J d S )
Nr|   CDCOrt   cCHa bcbrk   )r   rN   rN   rN   rO   test_encipher_bifid5   r[   r   c                     0   t   fdd} tdd| }td|ksJ d S )Nc                       t  d|  |  S )Nr   r   ijr   rN   rO   ru          z$test_bifid5_square.<locals>.<lambda>r    )r   r@   r   fMrN   r   rO   test_bifid5_square      r   c                   C   L   t dddks	J t dddksJ t dddksJ t dddks$J d S )	Nr|   r   r   chr   b acr   rW   )r   rN   rN   rN   rO   test_decipher_bifid5   r[   r   c                   C   r   )
Nr|   r   CPrt   r   CIr   r   rk   )r   rN   rN   rN   rO   test_encipher_bifid6   r[   r   c                   C   r   )	Nr|   r   r   cir   r   r   rW   )r   rN   rN   rN   rO   test_decipher_bifid6   r[   r   c                     r   )Nc                    r   )N   r   r   r   rN   rO   ru      r   z$test_bifid6_square.<locals>.<lambda>r   r   )r   r@   r   r   rN   r   rO   test_bifid6_square   r   r   c                   C   s~   t ddddks
J t ddddksJ tt t ddddks#J t dddd	u s-J W d    d S 1 s8w   Y  d S )
NrH   rJ   rS   r   rS   r      rJ   rL   rS      F)r   rE   r<   rN   rN   rN   rO   test_rsa_public_key   s   
"r   c                   C   s   t ddddks
J t ddddksJ t dddd	ksJ tt t dddd
ks-J t ddddu s7J W d    d S 1 sBw   Y  d S )NrH   rJ   rS   r   r   r         )i  i  r   r   F)r   rE   r<   rN   rN   rN   rO   test_rsa_private_key   s   
"r   c                  C   s\   t d} t d}t d}t d}t| ||| | |fksJ t| ||| | |fks,J d S )NN101565610013301240713207239558950144682174355406589305284428666903702505233009M8946871918875454889354556059559484138123760030531435214292421331206929398400365537ښ8936505818327042395303988587447591295947962354408444794561435666999402846577625762582824202269399672579058991442587406384754958587400493169361356902030209)intr   r   )pqedrN   rN   rO   test_rsa_large_key   s    r   c                  C   s   t ddd} td| dksJ t ddd} td| dksJ tt t ddd} td| dks2J W d    d S 1 s=w   Y  d S NrH   rJ   rS   r   r   )r   r   rE   r<   )pukrN   rN   rO   test_encipher_rsa      
"r   c                  C   s   t ddd} td| dksJ t ddd} td| dksJ tt t ddd} td| dks2J W d    d S 1 s=w   Y  d S r   )r   r   rE   r<   )prkrN   rN   rO   test_decipher_rsa   r   r   c            	      C   s0  t ddddddd} tddddddd}| dksJ |dks J d	}td| d
 | }|dks1J t||d
 d }||ks@J t ddddd}tddddd}t ddddd}tddddd}|dksfJ |dkslJ |dksrJ |dksxJ d}tt|||}|dksJ tt|||}||ksJ d S )NrH   rJ   r            )Nu  r   )r   i7  
   r   iF  5   )   +   /   a   )C r   )r   i| )r   r   )r   i! i!0  i )r   r   r   r   )	r   r   msg	encrypted	decryptedpuk1prk1puk2prk2rN   rN   rO   !test_mutltiprime_rsa_full_example   s,   r   c               	   C   s   t d} t d}t d}t d}t d}d}t| |||||}t| |||||}d}t||}	t||| ||||g}
|	|
ks@J t|	|t|	|| ||||gksRJ d S )N5  1017715760715424506802386150369308212090648714372506228340650154082258226204046999838297167140821364638180697194879500245557654451869628933464638414194270088003412574686002240499862604719225724816301446884172547691863941572670973607781363296129091102564212329778330286774412060493092203547960143766983251016934  287523423530951328722901815266072758861827932416608050778508017568951279775428697295227355312818186183057683628973866874525034028199691128870676414118458442900035778874482624765513861643279666963168221883983361990023065887039028941004761868238495951032394105272796054421482858161493686670831148028528049768935  1769822925986882577687950073635018683885096193595631013437826189771862186717463067541369694816245225291921138038800171125596073154495219811570843701878876506240610330660224585129424118411874789378997231527716008508616411987953604187533538484482056602874796176717264080533196198920520008508839943433788827178494  689254284385854310292691822335026110270917550646437423835156236432131058289689339552936707494280835318713879442274571841964528291231865157212604266903677599180789896916456120289112752835985022658896697303316882068252200747139776074151787380158310303642905853691505028197438273435520981970955205508653601594394  69035483433453632820551311892368908779778144568711455301541094314870476423226953576968609257479231896350331830698238209105217117290910679774888326149322416241405010692044244589681980660015448444826108008217972129130625571421904893252804729877353352739420480574842850202181462656251626522910618936534699566291i  i  )r   r   r   r   r   )r   r   rstr   r   r   	plaintextciphertext_1ciphertext_2rN   rN   rO   test_rsa_crt_extreme  s2   
r   c               
   C   s   d\} }d}t | ||dd}t| ||dd}t|d D ]$}t||}t||}z||ks/J W q ty@   td|||w d S )N)=   r      
Carmichael)totientr   RThe RSA is not correctly decrypted (Original : {}, Encrypted : {}, Decrypted : {})r   r   ranger   r   AssertionErrorformat)r   r   r   r   r   r   r   r   rN   rN   rO   test_rsa_exhaustive7  s    


r   c            	   
   C   s   g d} d}| |g }t |ddi}t|ddi}|d }t|D ]$}t||}t||}z||ks4J W q! tyE   td|||w d S )N)rJ   r   r   r   r   r   r   r   r   r   	primesr   argsr   r   nr   r   r   rN   rN   rO   test_rsa_multiprime_exhanstiveJ  s$   



r   c            	   
   C   s   g d} d}| |g }t |ddi}t|ddi}|d }t|D ],}t||dkr+q!t||}t||}z||ks<J W q! tyM   td|||w d S )N)r   r   r   r   
multipowerTr   rS   r   )r   r   r   r?   r   r   r   r   r   rN   rN   rO   test_rsa_multipower_exhanstive_  s(   



r   c                   C   F   t dddddksJ t dddddksJ t dddddks!J d S )NrS   rH   r   rH   r   rJ   r   rH   )r   rN   rN   rN   rO   test_kid_rsa_public_keyw  rV   r  c                   C   r   )NrS   rH   r   rJ   r   r   rL   )r   rN   rN   rN   rO   test_kid_rsa_private_key}  rV   r  c                   C   s:   t dddks	J t dddksJ t dddksJ d S )NrS   r   rH   r   rJ   r  )r   rN   rN   rN   rO   test_encipher_kid_rsa  rQ   r  c                   C   s:   t dddks	J t dddksJ t dddksJ d S )NrH   r  rS   rJ   r   r  )r   rN   rN   rN   rO   test_decipher_kid_rsa  rQ   r  c                   C   s   t ddksJ t ddksJ t ddksJ t ddks J t ddks(J t dd	d
dks2J t ddd
dks<J t ddksDJ t ddksLJ t ddksTJ d S )NrW   z.-|-...|-.-.zSMS z...|--|...||zSMS
r    z||`)sepz``z````z!@#$%^&*()_+z/-.-.--|.--.-.|...-..-|-.--.|-.--.-|..--.-|.-.-.12345z.----|..---|...--|....-|.....67890z-....|--...|---..|----.|-----)r    rN   rN   rN   rO   test_encode_morse  s   r  c                   C   s2   t ddksJ t ddksJ ttdd  d S )Nz
-.-|.|-.--KEYz.-.|..-|-.||RUNc                   S      t dS )Nz	.....----)r!   rN   rN   rN   rO   ru         z#test_decode_morse.<locals>.<lambda>)r!   rD   KeyErrorrN   rN   rN   rO   test_decode_morse  s   r  c                  C   s  t tdd  t tdd  td} t| dg| dgd| d| dgks(J t| dg| dgd| d| dgks>J td} t| dg| dgd| d| dgksXJ t| dg| dgd| d| dgksnJ t| dg| dgd| d| dgksJ d S )Nc                   S   s   t ddgdS NrS   r   rN   rN   rN   rO   ru         z$test_lfsr_sequence.<locals>.<lambda>c                   S   s   t dgddS r  r  rN   rN   rN   rO   ru     r  rH   rS   r   rJ   )rD   	TypeErrorrC   r   )FrN   rN   rO   test_lfsr_sequence  s   ,,,,0r  c                  C   sf   t tdd  td} t| d| dg| d| dgd}t|dddks'J t|dddks1J d S )Nc                   S      t dddS )NrS   rH   rJ   )r   rN   rN   rN   rO   ru         z+test_lfsr_autocorrelation.<locals>.<lambda>rH   rS   r   r   rT   )rD   r  rC   r   r   )r  r   rN   rN   rO   test_lfsr_autocorrelation  s
   $r  c                  C   s   t d} td}t| d| dg| d| dgd}t||d d ks&J t| d| dg| d| dgd}t||d | d ksFJ d S )NrH   xrS   r   r   )rC   r   r   r   )r  r  r   rN   rN   rO   test_lfsr_connection_polynomial  s   $$ r  c                  C   B   t dd\} }}t| sJ t|| sJ tt| dksJ d S Nd   digitf   )r"   rA   rB   lenbin)rq   r   _rN   rN   rO   test_elgamal_private_key     r(  c                     sb   t dt} | d   d tt d | ksJ tt fdd ttfdd d S )Nr   r   rS   c                      s
   t  S )Nr$   rN   PdkrN   rO   ru     rv   ztest_elgamal.<locals>.<lambda>c                      s
   t d S )NrT   r*  rN   )r-  rN   rO   ru     rv   )r"   r#   r%   r$   rD   rx   )ekrN   r+  rO   test_elgamal  s    r/  c                  C   r  r   )r&   rA   rB   r%  r&  )r   gr'  rN   rN   rO   test_dh_private_key  r)  r1  c                  C   sT   t dd\} }}t| ||f\}}}| |ksJ ||ksJ |t||| ks(J d S )Nr!  r"  )r&   r'   pow)p1g1rq   p2g2garN   rN   rO   test_dh_public_key  s
   r8  c                  C   sX   t dd} t| \}}}td|}t|||f|}|t|||ks#J ttdd  d S )Nr!  r"  rH   c                   S   rs   )N)i     i5  i  )r(   rN   rN   rN   rO   ru     rv   z$test_dh_shared_key.<locals>.<lambda>)r&   r'   rF   r(   r2  rD   rx   )r   r   r'  r7  r   skrN   rN   rO   test_dh_shared_key  s   

r;  c                   C   s2   t dddks	J ttdd  ttdd  d S )Nr   rt   bac                   S   rs   )Nrt   acer.   rN   rN   rN   rO   ru     rv   z!test_padded_key.<locals>.<lambda>c                   S   rs   )Nrt   abbar>  rN   rN   rN   rO   ru     rv   )r.   rD   rx   rN   rN   rN   rO   test_padded_key  s   r@  c                   C   s~   t tdd  tddddksJ t tdd  tddddks"J t tdd  td	td
ks3J tdtdks=J d S )Nc                   S   r  )Nro   r   abcde)r+   rN   rN   rN   rO   ru     r  ztest_bifid.<locals>.<lambda>ro   r   abcdbdbc                   S   r  )NrC  r   rA  )r,   rN   rN   rN   rO   ru     r  c                   S   r  )NrA  )r-   rN   rN   rN   rO   ru     r  BBACDEFGHIKLMNOPQRSTUVWXYZB0$B0ACDEFGHIJKLMNOPQRSTUVWXYZ123456789)rD   rx   r+   r   r   rN   rN   rN   rO   
test_bifid  s   rH  c            
      C   sn   g d} g d}g d}t | |D ]#\}}t||}|D ]}t||}t||}t||}	|	|ks3J qqd S )N)                                             )Y   r   e   g   k   m   q      rI  rJ  rK  rL  rM  rN  r   )r   iW  i  i9     iS( i    i  i  i+ i  iH  )zipr3   r2   r1   r0   
psqsmessagesr   r   prir   pubencdecrN   rN   rO   test_encipher_decipher_gm  s   



rj  c                   C   .   t tdd  t tdd  t tdd  d S )Nc                   S   rs   )Nr   r   r2   rN   rN   rN   rO   ru     rv   z%test_gm_private_key.<locals>.<lambda>c                   S   
   t ddS )Nr   rl  rN   rN   rN   rO   ru     rv   c                   S   rs   )Nr   r   rl  rN   rN   rN   rO   ru     rv   )rD   rx   r2   rN   rN   rN   rO   test_gm_private_key     rn  c                   C   s>   dt ddd ksJ dt ddd ksJ ttdd	  d S )
NiC  r      rS   r   rJ   r   c                   S   rs   )Nr   rp  rl  rN   rN   rN   rO   ru     rv   z$test_gm_public_key.<locals>.<lambda>)r2   rD   rx   rN   rN   rN   rO   test_gm_public_key  s   rq  c            
      C   sr   g d} g d }}g d}t | |D ]#\}}t||}|D ]}t||}t||}t||}	|	|ks5J qqd S )N)C   r   G   rZ  r   r   r[  r   O   rp  S   r   ;   r^     )r   rs  rZ  r   r   r[  r   rt  rp  ru  r   rv  r^  rw  rr  )r   iH  iW     r_  i  i  r`  i[  i  i  ij     )ra  r6   r7   r4   r5   rb  rN   rN   rO   test_encipher_decipher_bg  s   



rz  c                   C   rk  )Nc                   S   rs   )Nr   rf   r6   rN   rN   rN   rO   ru   *  rv   z%test_bg_private_key.<locals>.<lambda>c                   S   rm  )Nr   r{  rN   rN   rN   rO   ru   +  rv   c                   S   rs   Nr   r   r{  rN   rN   rN   rO   ru   ,  rv   )rD   rx   r6   rN   rN   rN   rO   test_bg_private_key)  ro  r}  c                   C   s6   dt ddks	J dt ddksJ ttdd  d S )	Ni  rr  rt  i  r   rw  c                   S   rs   r|  r{  rN   rN   rN   rO   ru   2  rv   z$test_bg_public_key.<locals>.<lambda>)r7   rD   rx   rN   rN   rN   rO   test_bg_public_key/  s   r~  N)
sympy.corer   sympy.crypto.cryptor   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   sympy.external.gmpyr?   sympy.matricesr@   sympy.ntheoryrA   rB   sympy.polys.domainsrC   sympy.testing.pytestrD   rE   sympy.core.randomrF   rP   rR   rU   rZ   r^   rh   rj   rn   ry   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r  r  r(  r/  r1  r8  r;  r@  rH  rj  rn  rq  rz  r}  r~  rN   rN   rN   rO   <module>   sv    		
!2			