U
    -e3M                     @   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tt dddks$tt dddks6td S )Nhello world   hlowrdel ol   horel ollwd   hwe olordll)r=   AssertionError rN   rN   _/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/sympy/crypto/tests/test_crypto.pytest_encipher_railfence   s    rP   c                   C   s:   t dddkstt dddks$tt dddks6td S )NrH   rG   rF   rJ   rI   rL   rK   )r>   rM   rN   rN   rN   rO   test_decipher_railfence    s    rQ   c                   C   sR   t ddddddgkstt ddddddgks4tt ddddddgksNtd S )NrI   rK   r      rG   )r   rM   rN   rN   rN   rO   test_cycle_list&   s    rT   c                   C   sL   t dddkstt dddks$tt dddks6ttdddksHtd S )NABCr   rR   BCDrS   ZZAB)r   rM   r)   rN   rN   rN   rO   test_encipher_shift,   s    rW   c                   C   sD   t ddkstt ddks ttddks0ttddks@td S )NrU   NOP)r8   rM   r9   rN   rN   rN   rO   test_encipher_rot132   s    rY   c                   C   s   t dddkstt dddks$tt dddks6tt dddd	d
ksLtt dddd	dksbtt dddkstttdddkstd S )NrU   )rR   r   )rR   rR   rV   )rS   r   ZAZY)rS   rR   ABCDr   BAD1231234Z214)rI      ZQTW)r   rM   r*   rN   rN   rN   rO   test_encipher_affine9   s    r_   c                   C   sD   t ddkstt ddks ttddks0ttddks@td S )NrU   ZZYX)r:   rM   r;   rN   rN   rN   rO   test_encipher_atbashB   s    r`   c                   C   s,   t ddddkstt ddddks(td S )NrU   BACr\   Z1243r]   Z124)r   rM   rN   rN   rN   rO   test_encipher_substitutionH   s    rb   c                   C   s`   t ddkstt tddks$tt d dks8tt dddddksNtttdd  d S )	NabcZaaabczab caT)filterc                   S   s
   t ddS )Nabrd   )r   rN   rN   rN   rO   <lambda>R       z%test_check_and_join.<locals>.<lambda>)r   rM   r/   splitrC   
ValueErrorrN   rN   rN   rO   test_check_and_joinM   s
    rk   c                   C   sn   t dddkstt dddddks(tt dddddks>tt dddddksTtt d	dddd	ksjtd S )
NrU   ZACErZ   r   ZACAABZACCZACA)r	   rM   rN   rN   rN   rO   test_encipher_vigenereU   s
    rn   c                   C   sn   t dddkstt dddddks(tt dddddks>tt dddddksTtt dddddksjtd S )	NrU   ZAAArZ   r   rl   ZAACZAArm   )r
   rM   rN   rN   rN   rO   test_decipher_vigenere]   s
    ro   c                  C   s   t ddddddg} td| dks&tt ddddddg} td| dksLttd| dddksbtt ddddddg} td| ddd	ksttd
| dddksttd| dksttd| dddkstd S )NrG   rR   rI      rZ   CFIVr   r   CBABrl   CBZABAZCFGCZ)padZCFYV)r?   r   rM   rm   rN   rN   rO   test_encipher_hille   s    rw   c                  C   s   t ddddddg} td| dks&tt ddddddg} td| dksLttd| dddksbtt ddddddg} td	| dddksttd
| dddksttd| dkstd S )NrG   rR   rI   rp   rq   rZ   r   r   rr   rs   rl   ZCFAZABAA)r?   r   rM   rv   rN   rN   rO   test_decipher_hillt   s    rx   c                   C   sL   t dddkstt dddks$tt dddks6tt ddd	ksHtd S )
Nrl   CDCOrf   cZCHa bcbra   )r   rM   rN   rN   rN   rO   test_encipher_bifid5   s    r~   c                     s0   t   fdd} tdd| }td|ks,td S )Nc                    s   t  d|  |  S )Nrp   r   ijrv   rN   rO   rg      rh   z$test_bifid5_square.<locals>.<lambda>rp    )r   r?   r   rM   fMrN   rv   rO   test_bifid5_square   s    r   c                   C   sL   t dddkstt dddks$tt dddks6tt dddksHtd S )	Nrl   rz   ry   chr{   b acr}   rU   )r   rM   rN   rN   rN   rO   test_decipher_bifid5   s    r   c                   C   sL   t dddkstt dddks$tt dddks6tt ddd	ksHtd S )
Nrl   ry   CPrf   r{   CIr|   r}   ra   )r   rM   rN   rN   rN   rO   test_encipher_bifid6   s    r   c                   C   sL   t dddkstt dddks$tt dddks6tt dddksHtd S )	Nrl   r   ry   cir{   r   r}   rU   )r   rM   rN   rN   rN   rO   test_decipher_bifid6   s    r   c                     s0   t   fdd} tdd| }td|ks,td S )Nc                    s   t  d|  |  S )N   r   r   rv   rN   rO   rg      rh   z$test_bifid6_square.<locals>.<lambda>r   r   )r   r?   r   rM   r   rN   rv   rO   test_bifid6_square   s    r   c                	   C   sh   t ddddkstt ddddks(ttt. t ddddksFtt dddd	ksZtW 5 Q R X d S )
NrG   rI   rR   r   rR   rp      rI   rK   rR      F)r   rM   rD   r<   rN   rN   rN   rO   test_rsa_public_key   s
    
r   c                	   C   s|   t ddddkstt ddddks(tt dddd	ks<ttt. t dddd
ksZtt ddddksntW 5 Q R X d S )NrG   rI   rR   r   rp   r         )i  i  r   r   F)r   rM   rD   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<tt| ||| | |fksXtd S )NZN101565610013301240713207239558950144682174355406589305284428666903702505233009ZM89468719188754548893545560595594841381237600305314352142924213312069293984003Z65537Z8936505818327042395303988587447591295947962354408444794561435666999402846577625762582824202269399672579058991442587406384754958587400493169361356902030209)intr   rM   r   )pqedrN   rN   rO   test_rsa_large_key   s    r   c               	   C   sr   t ddd} td| dkstt ddd} td| dks<ttt$ t ddd} td| dksdtW 5 Q R X d S NrG   rI   rR   rp   r   )r   r   rM   rD   r<   )pukrN   rN   rO   test_encipher_rsa   s    
r   c               	   C   sr   t ddd} td| dkstt ddd} td| dks<ttt$ t ddd} td| dksdtW 5 Q R X d S r   )r   r   rM   rD   r<   )prkrN   rN   rO   test_decipher_rsa   s    
r   c            	      C   s4  t ddddddd} tddddddd}| dks4t|dks@td	}td| d
 | }|dksbtt||d
 d }||kst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st|dkst|dkst|dkstd}tt|||}|dksttt|||}||ks0td S )NrG   rI   rp            )Nu  r   )r   i7  
   r   iF  5   )   +   /   a   )C r   )r   i| )r   r   )r   i! i!0  i )r   r   rM   r   r   )	r   r   msg	encrypted	decryptedZpuk1Zprk1Zpuk2Z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stt|	|t|	|| ||||gkstd S )NA5  101771576071542450680238615036930821209064871437250622834065015408225822620404699983829716714082136463818069719487950024555765445186962893346463841419427008800341257468600224049986260471922572481630144688417254769186394157267097360778136329612909110256421232977833028677441206049309220354796014376698325101693A4  28752342353095132872290181526607275886182793241660805077850801756895127977542869729522735531281818618305768362897386687452503402819969112887067641411845844290003577887448262476551386164327966696316822188398336199002306588703902894100476186823849595103239410527279605442148285816149368667083114802852804976893A5  176982292598688257768795007363501868388509619359563101343782618977186218671746306754136969481624522529192113803880017112559607315449521981157084370187887650624061033066022458512942411841187478937899723152771600850861641198795360418753353848448205660287479617671726408053319619892052000850883994343378882717849A4  68925428438585431029269182233502611027091755064643742383515623643213105828968933955293670749428083531871387944227457184196452829123186515721260426690367759918078989691645612028911275283598502265889669730331688206825220074713977607415178738015831030364290585369150502819743827343552098197095520550865360159439A4  69035483433453632820551311892368908779778144568711455301541094314870476423226953576968609257479231896350331830698238209105217117290910679774888326149322416241405010692044244589681980660015448444826108008217972129130625571421904893252804729877353352739420480574842850202181462656251626522910618936534699566291i  i  )r   r   r   r   rM   r   )r   r   rstr   r   r   Z	plaintextZciphertext_1Zciphertext_2rN   rN   rO   test_rsa_crt_extreme  s2    
r   c               
   C   s   d\} }d}t | ||dd}t| ||dd}t|d D ]P}t||}t||}z||ks^tW q8 tk
r   td|||Y q8X q8d S )N)=   r      
Carmichael)totientr   RThe RSA is not correctly decrypted (Original : {}, Encrypted : {}, Decrypted : {})r   r   ranger   r   rM   format)r   r   r   r   r   r   r   r   rN   rN   rO   test_rsa_exhaustive6  s"    

  r   c            	   
   C   s   ddddg} d}| |g }t |ddi}t|ddi}|d }t|D ]P}t||}t||}z||ksltW qF tk
r   td|||Y qFX qFd S )	NrI   rp   r   r   r   r   r   r   r   )	primesr   argsr   r   nr   r   r   rN   rN   rO   test_rsa_multiprime_exhanstiveI  s&    


  r   c            
   
   C   s   ddl m}  dddg}d}||g }t|ddi}t|ddi}|d }t|D ]`}| ||dkrdqPt||}t||}	z|	|kstW qP tk
r   td|||	Y qPX qPd S )	Nr   )igcdrp   r   Z
multipowerTrR   r   )	Zsympy.core.numbersr   r   r   r   r   r   rM   r   )
r   r   r   r   r   r   r   r   r   r   rN   rN   rO   test_rsa_multipower_exhanstive^  s,    



  r   c                   C   sF   t dddddkstt dddddks,tt dddddksBtd S )NrR   rG   rp   rG   r   rI   r   rG   )r   rM   rN   rN   rN   rO   test_kid_rsa_public_keyw  s    r   c                   C   sF   t dddddkstt dddddks,tt dddddksBtd S )NrR   rG   rp   rI   r   r   rK   )r   rM   rN   rN   rN   rO   test_kid_rsa_private_key}  s    r   c                   C   s:   t dddkstt dddks$tt dddks6td S )NrR   r   rG   r   rI   r   )r   rM   rN   rN   rN   rO   test_encipher_kid_rsa  s    r   c                   C   s:   t dddkstt dddks$tt dddks6td S )NrG   r   rR   rI   r   r   )r   rM   rN   rN   rN   rO   test_decipher_kid_rsa  s    r   c                   C   s   t ddkstt ddks tt ddks0tt ddks@tt ddksPtt dd	d
dksdtt ddd
dksxtt ddkstt ddkstt ddkstd S )NrU   z.-|-...|-.-.zSMS z...|--|...||zSMS
r    z||`)sepz``z````z!@#$%^&*()_+z/-.-.--|.--.-.|...-..-|-.--.|-.--.-|..--.-|.-.-.Z12345z.----|..---|...--|....-|.....Z67890z-....|--...|---..|----.|-----)r    rM   rN   rN   rN   rO   test_encode_morse  s    r   c                   C   s2   t ddkstt ddks tttdd  d S )Nz
-.-|.|-.--ZKEYz.-.|..-|-.||ZRUNc                   S   s   t dS )Nz	.....----)r!   rN   rN   rN   rO   rg     rh   z#test_decode_morse.<locals>.<lambda>)r!   rM   rC   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sPtt| dg| dgd| d| dgks|ttd} t| dg| dgd| d| dgkstt| dg| dgd| d| dgkstt| dg| dgd| d| dgks
td S )Nc                   S   s   t ddgdS NrR   r   rN   rN   rN   rO   rg     rh   z$test_lfsr_sequence.<locals>.<lambda>c                   S   s   t dgddS r   r   rN   rN   rN   rO   rg     rh   rG   rR   r   rI   )rC   	TypeErrorrB   r   rM   )FrN   rN   rO   test_lfsr_sequence  s    ,,,,r   c                  C   sf   t tdd  td} t| d| dg| d| dgd}t|dddksNtt|dddksbtd S )Nc                   S   s   t dddS )NrR   rG   rI   )r   rN   rN   rN   rO   rg     rh   z+test_lfsr_autocorrelation.<locals>.<lambda>rG   rR   r   rp   rS   )rC   r   rB   r   r   rM   )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sLtt| d| dg| d| dgd}t||d | d kstd S )NrG   xrR   r   rp   )rB   r   r   r   rM   )r   r   r   rN   rN   rO   test_lfsr_connection_polynomial  s    $$r   c                  C   sB   t dd\} }}t| stt|| s*ttt| dks>td S Nd   digitf   )r"   r@   rM   rA   lenbin)rd   r}   _rN   rN   rO   test_elgamal_private_key  s    r   c                     sb   t dt} | d   d tt d | ks8ttt fdd ttfdd d S )Nrp   r   rR   c                      s
   t  S )Nr$   rN   PdkrN   rO   rg     rh   ztest_elgamal.<locals>.<lambda>c                      s
   t d S )NrS   r   rN   )r   rN   rO   rg     rh   )r"   r#   r%   r$   rM   rC   rj   )ZekrN   r   rO   test_elgamal  s     r   c                  C   sB   t dd\} }}t| stt|| s*ttt| dks>td S r   )r&   r@   rM   rA   r   r   )r   gr   rN   rN   rO   test_dh_private_key  s    r   c                  C   sT   t dd\} }}t| ||f\}}}| |ks0t||ks<t|t||| ksPtd S )Nr   r   )r&   r'   rM   pow)p1g1rd   p2g2garN   rN   rO   test_dh_public_key  s
    r   c                  C   sX   t dd} t| \}}}td|}t|||f|}|t|||ksFtttdd  d S )Nr   r   rG   c                   S   s
   t ddS )N)i     i5  i  )r(   rN   rN   rN   rO   rg     rh   z$test_dh_shared_key.<locals>.<lambda>)r&   r'   rE   r(   r   rM   rC   rj   )r   r   r   r   r}   skrN   rN   rO   test_dh_shared_key  s    

r   c                   C   s2   t dddkstttdd  ttdd  d S )Nr}   rf   bac                   S   s
   t ddS )Nrf   Zacer.   rN   rN   rN   rO   rg     rh   z!test_padded_key.<locals>.<lambda>c                   S   s
   t ddS )Nrf   Zabbar   rN   rN   rN   rO   rg     rh   )r.   rM   rC   rj   rN   rN   rN   rO   test_padded_key  s    r   c                   C   s~   t tdd  tddddks"tt tdd  tddddksDtt tdd  td	td
ksfttdtdksztd S )Nc                   S   s   t dddS )Nrc   r}   abcde)r+   rN   rN   rN   rO   rg     rh   ztest_bifid.<locals>.<lambda>rc   r}   abcdbdbc                   S   s   t dddS )Nr   r}   r   )r,   rN   rN   rN   rO   rg     rh   c                   S   s   t dS )Nr   )r-   rN   rN   rN   rO   rg     rh   BZBACDEFGHIKLMNOPQRSTUVWXYZB0Z$B0ACDEFGHIJKLMNOPQRSTUVWXYZ123456789)rC   rj   r+   rM   r   r   rN   rN   rN   rO   
test_bifid  s    r   c            
      C   s   ddddddddd	d
dddddg} dddddddddddddddg}dddddddd d!d"d#d$d%g}t | |D ]F\}}t||}|D ].}t||}t||}t||}	|	|kstqqld S )&N                                             Y   r   e   g   k   m   q      r   r   iW  i  i9     iS( i    i  i  i+ i  iH  )zipr3   r2   r1   r0   rM   
Zpsqsmessagesr   r   prir   ZpubencdecrN   rN   rO   test_encipher_decipher_gm  sN                           



r   c                   C   sD   t tdd  t tdd  t tdd  ds@tdtddkd S )Nc                   S   s
   t ddS )Nr   r   r2   rN   rN   rN   rO   rg     rh   z%test_gm_private_key.<locals>.<lambda>c                   S   s
   t ddS )Nr   r!  rN   rN   rN   rO   rg     rh   c                   S   s
   t ddS )Nr   rp   r!  rN   rN   rN   rO   rg     rh   r      )rC   rj   rM   r2   rN   rN   rN   rO   test_gm_private_key  s    r#  c                   C   s>   dt ddd kstdt ddd ks,tttdd	  d S )
NiC  r   r"  rR   r   rI   rp   c                   S   s
   t ddS )Nr   r"  r!  rN   rN   rN   rO   rg     rh   z$test_gm_public_key.<locals>.<lambda>)r2   rM   rC   rj   rN   rN   rN   rO   test_gm_public_key  s    r$  c            
      C   s   ddddddddd	d
dddddg} dddddddd	d
ddddddg }}dddddddddddddg}t | |D ]F\}}t||}|D ].}t||}t||}t||}	|	|kstqqpd S )NC   r   G   r  r   r   r  r   O   r"  S   r   ;   r     r   iH  iW     r  i  i  r  i[  i  i  ij     )r  r6   r7   r4   r5   rM   r  rN   rN   rO   test_encipher_decipher_bg  sP                            



r-  c                   C   sD   t tdd  t tdd  t tdd  ds@tdtddkd S )Nc                   S   s
   t ddS )Nr   r^   r6   rN   rN   rN   rO   rg   *  rh   z%test_bg_private_key.<locals>.<lambda>c                   S   s
   t ddS )Nr   r.  rN   rN   rN   rO   rg   +  rh   c                   S   s
   t ddS Nr   r   r.  rN   rN   rN   rO   rg   ,  rh   r   r*  )rC   rj   rM   r6   rN   rN   rN   rO   test_bg_private_key)  s    r0  c                   C   s6   dt ddkstdt ddks$tttdd  d S )	Ni  r%  r'  i  r   r*  c                   S   s
   t ddS r/  r.  rN   rN   rN   rO   rg   2  rh   z$test_bg_public_key.<locals>.<lambda>)r7   rM   rC   rj   rN   rN   rN   rO   test_bg_public_key/  s    r1  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sympy.core.randomrE   rP   rQ   rT   rW   rY   r_   r`   rb   rk   rn   ro   rw   rx   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  rN   rN   rN   rO   <module>   sr   		
!2			