o
    8Va(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@mAZAmBZB d dlCmDZD d dlEmFZFmGZG d dlHmIZI dd	 ZJd
d ZKd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]d0d1 Z^d2d3 Z_d4d5 Z`d6d7 Zad8d9 Zbd:d; Zcd<d= Zdd>d? Zed@dA ZfdBdC ZgdDdE ZhdFdG ZidHdI ZjdJdK ZkdLdM ZldNdO ZmdPdQ ZndRdS ZodTdU ZpdVdW ZqdXdY ZrdZd[ Zsd\d] Ztd^d_ Zud`da Zvdbdc Zwddde Zxdfdg Zydhdi Zzdjdk Z{dldm Z|dnS )o    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)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=    rM   rM   @/usr/lib/python3/dist-packages/sympy/crypto/tests/test_crypto.pytest_encipher_railfence      rO   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   rG   rF   rJ   rI   rL   rK   )r>   rM   rM   rM   rN   test_decipher_railfence    rP   rQ   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 )NrI   rK   )rI   r      rG   rR   )rR   rG   rI   r   )r   rM   rM   rM   rN   test_cycle_list&      rT   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   rR   BCDrS   ZZAB)r   r)   rM   rM   rM   rN   test_encipher_shift,      rX   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 )NrV   ZNOP)r8   r9   rM   rM   rM   rN   test_encipher_rot132      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 )NrV   )rR   r   )rR   rR   rW   )rS   r   ZAZY)rS   rR   ABCDr   ZBAD1231234Z214)rI      ZQTW)r   r*   rM   rM   rM   rN   test_encipher_affine9   s   ra   c                   C   rZ   )NrV   ZZYX)r:   r;   rM   rM   rM   rN   test_encipher_atbashB   r\   rb   c                   C   s,   t ddddks
J t ddddksJ d S )NrV   BACr^   Z1243r_   Z124)r   rM   rM   rM   rN   test_encipher_substitutionH   s   rd   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abcZaaabczab caT)filterc                   S   
   t ddS )Nabrf   )r   rM   rM   rM   rN   <lambda>R      
 z%test_check_and_join.<locals>.<lambda>)r   r/   splitrC   
ValueErrorrM   rM   rM   rN   test_check_and_joinM   s
   rn   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 )
NrV   ZACEr]   r   ZACAABZACCZACA)r	   rM   rM   rM   rN   test_encipher_vigenereU   
   rq   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 )	NrV   ZAAAr]   r   ro   ZAACZAArp   )r
   rM   rM   rM   rN   test_decipher_vigenere]   rr   rs   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 )NrG   rR   rG   rI      r]   CFIVrR   r   r   rR   r   CBABro   CBZABAZCFGCZ)ZpadZCFYV)r?   r   rp   rM   rM   rN   test_encipher_hille   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 )NrG   rt   rv   r]   rw   r   rx   ry   ro   ZCFAZABAA)r?   r   r{   rM   rM   rN   test_decipher_hillt   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 )
Nro   CDCOri   cZCHa bcbrc   )r   rM   rM   rM   rN   test_encipher_bifid5   rY   r   c                     0   t   fdd} tdd| }td|ksJ d S )Nc                       t  d|  |  S )Nru   r   ijr{   rM   rN   rj          z$test_bifid5_square.<locals>.<lambda>ru    )r   r?   r   fMrM   r{   rN   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 )	Nro   r   r   Zchr   b acr   rV   )r   rM   rM   rM   rN   test_decipher_bifid5   rY   r   c                   C   r~   )
Nro   r   CPri   r   ZCIr   r   rc   )r   rM   rM   rM   rN   test_encipher_bifid6   rY   r   c                   C   r   )	Nro   r   r   Zcir   r   r   rV   )r   rM   rM   rM   rN   test_decipher_bifid6   rY   r   c                     r   )Nc                    r   )N   r   r   r{   rM   rN   rj      r   z$test_bifid6_square.<locals>.<lambda>r   r   )r   r?   r   r   rM   r{   rN   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 )
NrG   rI   rR   r   rR   ru      rI   rK   rR      F)r   rD   r<   rM   rM   rM   rN   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 )NrG   rI   rR   r   ru   r         )i  i  r   r   F)r   rD   r<   rM   rM   rM   rN   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 )NZN101565610013301240713207239558950144682174355406589305284428666903702505233009ZM89468719188754548893545560595594841381237600305314352142924213312069293984003Z65537Z8936505818327042395303988587447591295947962354408444794561435666999402846577625762582824202269399672579058991442587406384754958587400493169361356902030209)intr   r   )pqedrM   rM   rN   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 NrG   rI   rR   ru   r   )r   r   rD   r<   )pukrM   rM   rN   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   rD   r<   )prkrM   rM   rN   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 )NrG   rI   ru            )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	decryptedZpuk1Zprk1Zpuk2Zprk2rM   rM   rN   !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 )NA5  101771576071542450680238615036930821209064871437250622834065015408225822620404699983829716714082136463818069719487950024555765445186962893346463841419427008800341257468600224049986260471922572481630144688417254769186394157267097360778136329612909110256421232977833028677441206049309220354796014376698325101693A4  28752342353095132872290181526607275886182793241660805077850801756895127977542869729522735531281818618305768362897386687452503402819969112887067641411845844290003577887448262476551386164327966696316822188398336199002306588703902894100476186823849595103239410527279605442148285816149368667083114802852804976893A5  176982292598688257768795007363501868388509619359563101343782618977186218671746306754136969481624522529192113803880017112559607315449521981157084370187887650624061033066022458512942411841187478937899723152771600850861641198795360418753353848448205660287479617671726408053319619892052000850883994343378882717849A4  68925428438585431029269182233502611027091755064643742383515623643213105828968933955293670749428083531871387944227457184196452829123186515721260426690367759918078989691645612028911275283598502265889669730331688206825220074713977607415178738015831030364290585369150502819743827343552098197095520550865360159439A4  69035483433453632820551311892368908779778144568711455301541094314870476423226953576968609257479231896350331830698238209105217117290910679774888326149322416241405010692044244589681980660015448444826108008217972129130625571421904893252804729877353352739420480574842850202181462656251626522910618936534699566291i  i  )r   r   r   r   r   )r   r   rstr   r   r   Z	plaintextZciphertext_1Zciphertext_2rM   rM   rN   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   rM   rM   rN   test_rsa_exhaustive6  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)rI   ru   r   r   r   r   r   r   r   r   )	primesr   argsr   r   nr   r   r   rM   rM   rN   test_rsa_multiprime_exhanstiveI  s$   



r   c            
   
   C   s   ddl m}  g d}d}||g }t|ddi}t|ddi}|d }t|D ],}| ||dkr1q't||}t||}	z|	|ksBJ W q' tyS   td|||	w d S )	Nr   )igcd)ru   ru   r   r   Z
multipowerTrR   r   )	Zsympy.core.numbersr   r   r   r   r   r   r   r   )
r   r   r   r   r   r   r   r   r   r   rM   rM   rN   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 )NrR   rG   ru   rG   r   rI   r   rG   )r   rM   rM   rM   rN   test_kid_rsa_public_keyw  rU   r   c                   C   r   )NrR   rG   ru   rI   r   r   rK   )r   rM   rM   rM   rN   test_kid_rsa_private_key}  rU   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 )NrR   r   rG   r   rI   r   )r   rM   rM   rM   rN   test_encipher_kid_rsa  rP   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 )NrG   r   rR   rI   r   r   )r   rM   rM   rM   rN   test_decipher_kid_rsa  rP   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 )NrV   z.-|-...|-.-.zSMS z...|--|...||zSMS
r    z||`)sepz``z````z!@#$%^&*()_+z/-.-.--|.--.-.|...-..-|-.--.|-.--.-|..--.-|.-.-.Z12345z.----|..---|...--|....-|.....Z67890z-....|--...|---..|----.|-----)r    rM   rM   rM   rN   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
-.-|.|-.--ZKEYz.-.|..-|-.||ZRUNc                   S      t dS )Nz	.....----)r!   rM   rM   rM   rN   rj         z#test_decode_morse.<locals>.<lambda>)r!   rC   KeyErrorrM   rM   rM   rN   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 NrR   r   rM   rM   rM   rN   rj         z$test_lfsr_sequence.<locals>.<lambda>c                   S   s   t dgddS r   r   rM   rM   rM   rN   rj     r   rG   rR   r   rI   )rC   	TypeErrorrB   r   )FrM   rM   rN   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 )NrR   rG   rI   )r   rM   rM   rM   rN   rj         z+test_lfsr_autocorrelation.<locals>.<lambda>rG   rR   r   ru   rS   )rC   r   rB   r   r   )r   r   rM   rM   rN   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 )NrG   xrR   r   ru   )rB   r   r   r   )r   r   r   rM   rM   rN   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   Zdigitf   )r"   r@   rA   lenbin)rf   r   _rM   rM   rN   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 )Nru   r   rR   c                      s
   t  S )Nr$   rM   PdkrM   rN   rj     rk   ztest_elgamal.<locals>.<lambda>c                      s
   t d S )NrS   r   rM   )r   rM   rN   rj     rk   )r"   r#   r%   r$   rC   rm   )ZekrM   r   rN   test_elgamal  s    r   c                  C   r   r   )r&   r@   rA   r   r   )r   gr   rM   rM   rN   test_dh_private_key  r   r   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)Zp1Zg1rf   Zp2Zg2garM   rM   rN   test_dh_public_key  s
   r  c                  C   sX   t dd} t| \}}}td|}t|||f|}|t|||ks#J ttdd  d S )Nr   r   rG   c                   S   rh   )N)i     i5  i  )r(   rM   rM   rM   rN   rj     rk   z$test_dh_shared_key.<locals>.<lambda>)r&   r'   rE   r(   r   rC   rm   )r   r   r   r   r   skrM   rM   rN   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   ri   Zbac                   S   rh   )Nri   Zacer.   rM   rM   rM   rN   rj     rk   z!test_padded_key.<locals>.<lambda>c                   S   rh   )Nri   Zabbar  rM   rM   rM   rN   rj     rk   )r.   rC   rm   rM   rM   rM   rN   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   )Nre   r   abcde)r+   rM   rM   rM   rN   rj     r   ztest_bifid.<locals>.<lambda>re   r   Zabcdbdbc                   S   r   )Nr  r   r  )r,   rM   rM   rM   rN   rj     r   c                   S   r   )Nr  )r-   rM   rM   rM   rN   rj     r   BZBACDEFGHIKLMNOPQRSTUVWXYZZB0Z$B0ACDEFGHIJKLMNOPQRSTUVWXYZ123456789)rC   rm   r+   r   r   rM   rM   rM   rN   
test_bifid  s   r
  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      r  r  r  r  r  r  r   )r   iW  i  i9     iS( i    i  i  i+ i  iH  )zipr3   r2   r1   r0   
ZpsZqsmessagesr   r   Zprir   ZpubencZdecrM   rM   rN   test_encipher_decipher_gm  s   



r'  c                   C   .   t tdd  t tdd  t tdd  d S )Nc                   S   rh   )Nr   r   r2   rM   rM   rM   rN   rj     rk   z%test_gm_private_key.<locals>.<lambda>c                   S   
   t ddS )Nr   r)  rM   rM   rM   rN   rj     rk   c                   S   rh   )Nr   ru   r)  rM   rM   rM   rN   rj     rk   )rC   rm   r2   rM   rM   rM   rN   test_gm_private_key     r+  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      rR   r   rI   ru   c                   S   rh   )Nr   r-  r)  rM   rM   rM   rN   rj     rk   z$test_gm_public_key.<locals>.<lambda>)r2   rC   rm   rM   rM   rM   rN   test_gm_public_key  s   r.  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   r  r   r   r  r   O   r-  S   r   ;   r      )r   r0  r  r   r   r  r   r1  r-  r2  r   r3  r   r4  r/  )r   iH  iW     r!  i  i  r"  i[  i  i  ij     )r#  r6   r7   r4   r5   r$  rM   rM   rN   test_encipher_decipher_bg  s   



r7  c                   C   r(  )Nc                   S   rh   )Nr   r`   r6   rM   rM   rM   rN   rj   *  rk   z%test_bg_private_key.<locals>.<lambda>c                   S   r*  )Nr   r8  rM   rM   rM   rN   rj   +  rk   c                   S   rh   Nr   r   r8  rM   rM   rM   rN   rj   ,  rk   )rC   rm   r6   rM   rM   rM   rN   test_bg_private_key)  r,  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  r/  r1  i  r   r4  c                   S   rh   r9  r8  rM   rM   rM   rN   rj   2  rk   z$test_bg_public_key.<locals>.<lambda>)r7   rC   rm   rM   rM   rM   rN   test_bg_public_key/  s   r;  N)}Z
sympy.corer   Z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>   Zsympy.matricesr?   Zsympy.ntheoryr@   rA   Zsympy.polys.domainsrB   Zsympy.testing.pytestrC   rD   ZrandomrE   rO   rQ   rT   rX   r[   ra   rb   rd   rn   rq   rs   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.  r7  r:  r;  rM   rM   rM   rN   <module>   st    		
!2			