a
    kh.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s6J d S )Nhello world   hlowrdel ol   horel ollwd   hwe olordll)r=    rN   rN   L/var/www/auris/lib/python3.9/site-packages/sympy/crypto/tests/test_crypto.pytest_encipher_railfence   s    rP   c                   C   s:   t dddksJ t dddks$J t dddks6J d S )NrI   rH   rG   rK   rJ   rM   rL   )r>   rN   rN   rN   rO   test_decipher_railfence!   s    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sBJ d S )NrJ   rL   )rJ   r      rH   rR   )rR   rH   rJ   r   )r   rN   rN   rN   rO   test_cycle_list'   s    rT   c                   C   sL   t dddksJ t dddks$J t dddks6J tdddksHJ d S )NABCr   rR   BCDrS   ZZAB)r   r)   rN   rN   rN   rO   test_encipher_shift-   s    rW   c                   C   sD   t ddksJ t ddks J tddks0J tddks@J d S )NrU   NOP)r8   r9   rN   rN   rN   rO   test_encipher_rot133   s    rY   c                   C   s   t dddksJ t dddks$J t dddks6J t dddd	d
ksLJ t dddd	dksbJ t dddkstJ tdddksJ d S )NrU   )rR   r   )rR   rR   rV   )rS   r   ZAZY)rS   rR   ABCDr   BAD1231234Z214)rJ      ZQTW)r   r*   rN   rN   rN   rO   test_encipher_affine:   s    r_   c                   C   sD   t ddksJ t ddks J tddks0J tddks@J d S )NrU   ZZYX)r:   r;   rN   rN   rN   rO   test_encipher_atbashC   s    r`   c                   C   s,   t ddddksJ t ddddks(J d S )NrU   BACr\   Z1243r]   Z124)r   rN   rN   rN   rO   test_encipher_substitutionI   s    rb   c                   C   s`   t ddksJ t tddks$J t d dks8J t dddddksNJ 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>S       z%test_check_and_join.<locals>.<lambda>)r   r/   splitrD   
ValueErrorrN   rN   rN   rO   test_check_and_joinN   s
    rk   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sTJ t d	dddd	ksjJ d S )
NrU   ZACErZ   r   ZACAABZACCZACA)r	   rN   rN   rN   rO   test_encipher_vigenereV   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sTJ t dddddksjJ d S )	NrU   ZAAArZ   r   rl   ZAACZAArm   )r
   rN   rN   rN   rO   test_decipher_vigenere^   s
    ro   c                  C   s   t ddg d} td| dks"J t ddg d} td| dksDJ td| dddksZJ t ddg d} td| dddksJ td| ddd	ksJ td
| dksJ td
| dddksJ d S )NrH   rR   rH   rJ      rZ   CFIVrR   r   r   rR   r   CBABrl   CBZABAZCFGCZ)padZCFYV)r@   r   rm   rN   rN   rO   test_encipher_hillf   s    ry   c                  C   s   t ddg d} td| dks"J t ddg d} td| dksDJ td| dddksZJ t ddg d} td| dddksJ td| ddd	ksJ td
| dksJ d S )NrH   rp   rr   rZ   rs   r   rt   ru   rl   ZCFAZABAA)r@   r   rx   rN   rN   rO   test_decipher_hillu   s    rz   c                   C   sL   t dddksJ t dddks$J t dddks6J t ddd	ksHJ d S )
Nrl   CDCOrf   cZCHa bcbra   )r   rN   rN   rN   rO   test_encipher_bifid5   s    r   c                     s0   t   fdd} tdd| }td|ks,J d S )Nc                    s   t  d|  |  S )Nrq   r   ijrx   rN   rO   rg      rh   z$test_bifid5_square.<locals>.<lambda>rq    )r   r@   r   fMrN   rx   rO   test_bifid5_square   s    r   c                   C   sL   t dddksJ t dddks$J t dddks6J t dddksHJ d S )	Nrl   r|   r{   chr}   b acr   rU   )r   rN   rN   rN   rO   test_decipher_bifid5   s    r   c                   C   sL   t dddksJ t dddks$J t dddks6J t ddd	ksHJ d S )
Nrl   r{   CPrf   r}   CIr~   r   ra   )r   rN   rN   rN   rO   test_encipher_bifid6   s    r   c                   C   sL   t dddksJ t dddks$J t dddks6J t dddksHJ d S )	Nrl   r   r{   cir}   r   r   rU   )r   rN   rN   rN   rO   test_decipher_bifid6   s    r   c                     s0   t   fdd} tdd| }td|ks,J d S )Nc                    s   t  d|  |  S )N   r   r   rx   rN   rO   rg      rh   z$test_bifid6_square.<locals>.<lambda>r   r   )r   r@   r   r   rN   rx   rO   test_bifid6_square   s    r   c                   C   s|   t ddddksJ t ddddks(J tt8 t ddddksFJ t dddd	u sZJ W d    n1 sn0    Y  d S )
NrH   rJ   rR   r   rR   rq      rJ   rL   rR      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8 t dddd
ksZJ t ddddu snJ W d    n1 s0    Y  d S )NrH   rJ   rR   r   rq   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sXJ d S )NZN101565610013301240713207239558950144682174355406589305284428666903702505233009ZM89468719188754548893545560595594841381237600305314352142924213312069293984003Z65537Z8936505818327042395303988587447591295947962354408444794561435666999402846577625762582824202269399672579058991442587406384754958587400493169361356902030209)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sdJ W d    n1 sx0    Y  d S NrH   rJ   rR   rq   r   )r   r   rE   r<   )pukrN   rN   rO   test_encipher_rsa   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sdJ W d    n1 sx0    Y  d S r   )r   r   rE   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s4J |dks@J d	}td| d
 | }|dksbJ t||d
 d }||ksJ 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sJ |dksJ |dksJ |dksJ d}tt|||}|dksJ tt|||}||ks0J d S )NrH   rJ   rq            )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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sJ t|	|t|	|| ||||gksJ d S )NA5  101771576071542450680238615036930821209064871437250622834065015408225822620404699983829716714082136463818069719487950024555765445186962893346463841419427008800341257468600224049986260471922572481630144688417254769186394157267097360778136329612909110256421232977833028677441206049309220354796014376698325101693A4  28752342353095132872290181526607275886182793241660805077850801756895127977542869729522735531281818618305768362897386687452503402819969112887067641411845844290003577887448262476551386164327966696316822188398336199002306588703902894100476186823849595103239410527279605442148285816149368667083114802852804976893A5  176982292598688257768795007363501868388509619359563101343782618977186218671746306754136969481624522529192113803880017112559607315449521981157084370187887650624061033066022458512942411841187478937899723152771600850861641198795360418753353848448205660287479617671726408053319619892052000850883994343378882717849A4  68925428438585431029269182233502611027091755064643742383515623643213105828968933955293670749428083531871387944227457184196452829123186515721260426690367759918078989691645612028911275283598502265889669730331688206825220074713977607415178738015831030364290585369150502819743827343552098197095520550865360159439A4  69035483433453632820551311892368908779778144568711455301541094314870476423226953576968609257479231896350331830698238209105217117290910679774888326149322416241405010692044244589681980660015448444826108008217972129130625571421904893252804729877353352739420480574842850202181462656251626522910618936534699566291i  i  )r   r   r   r   r   )r   r   rstr   r   r   	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 ]N}t||}t||}z||ks^J W q8 ty   td|||Y q80 q8d 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 ]N}t||}t||}z||kshJ W qB ty   td|||Y qB0 qBd S )N)rJ   rq   r   r   r   r   r   r   r   r   	Z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rVqBt||}t||}z||ksxJ W qB ty   td|||Y qB0 qBd S )N)rq   rq   r   r   Z
multipowerTr   rR   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J t dddddks,J t dddddksBJ d S )NrR   rH   rq   rH   r   rJ   r   rH   )r   rN   rN   rN   rO   test_kid_rsa_public_keyw  s    r   c                   C   sF   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sBJ d S )NrR   rH   rq   rJ   r   r   rL   )r   rN   rN   rN   rO   test_kid_rsa_private_key}  s    r   c                   C   s:   t dddksJ t dddks$J t dddks6J d S )NrR   r   rH   r   rJ   r   )r   rN   rN   rN   rO   test_encipher_kid_rsa  s    r   c                   C   s:   t dddksJ t dddks$J t dddks6J d S )NrH   r   rR   rJ   r   r   )r   rN   rN   rN   rO   test_decipher_kid_rsa  s    r   c                   C   s   t ddksJ t ddks J t ddks0J t ddks@J t ddksPJ t dd	d
dksdJ t ddd
dksxJ t ddksJ t ddksJ t ddksJ d S )NrU   z.-|-...|-.-.zSMS z...|--|...||zSMS
r    z||`)sepz``z````z!@#$%^&*()_+z/-.-.--|.--.-.|...-..-|-.--.|-.--.-|..--.-|.-.-.Z12345z.----|..---|...--|....-|.....Z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
-.-|.|-.--ZKEYz.-.|..-|-.||ZRUNc                   S   s   t dS )Nz	.....----)r!   rN   rN   rN   rO   rg     rh   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sPJ t| dg| dgd| d| dgks|J td} t| dg| dgd| d| dgksJ t| dg| dgd| d| dgksJ t| dg| dgd| d| dgks
J 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   rH   rR   r   rJ   )rD   	TypeErrorrC   r   )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sNJ t|dddksbJ d S )Nc                   S   s   t dddS )NrR   rH   rJ   )r   rN   rN   rN   rO   rg     rh   z+test_lfsr_autocorrelation.<locals>.<lambda>rH   rR   r   rq   rS   )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sLJ t| d| dg| d| dgd}t||d | d ksJ d S )NrH   xrR   r   rq   )rC   r   r   r   )r   r   r   rN   rN   rO   test_lfsr_connection_polynomial  s    $$r   c                  C   sB   t dd\} }}t| sJ t|| s*J tt| dks>J d S Nd   digitf   )r"   rA   rB   lenbin)rd   r   _rN   rN   rO   test_elgamal_private_key  s    r   c                     sb   t dt} | d   d tt d | ks8J tt fdd ttfdd d S )Nrq   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$   rD   rj   )ZekrN   r   rO   test_elgamal  s     r   c                  C   sB   t dd\} }}t| sJ t|| s*J tt| dks>J d S r   )r&   rA   rB   r   r   )r   gr   rN   rN   rO   test_dh_private_key  s    r   c                  C   sT   t dd\} }}t| ||f\}}}| |ks0J ||ks<J |t||| ksPJ d S )Nr   r   )r&   r'   pow)p1Zg1rd   p2Zg2garN   rN   rO   test_dh_public_key  s
    r   c                  C   sX   t dd} t| \}}}td|}t|||f|}|t|||ksFJ ttdd  d S )Nr   r   rH   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'   rF   r(   r   rD   rj   )r   r   r   r   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   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.   rD   rj   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sDJ t tdd  td	td
ksfJ tdtdkszJ 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   Z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)rD   rj   r+   r   r   rN   rN   rN   rO   
test_bifid  s    r   c            
      C   sn   g d} g d}g d}t | |D ]F\}}t||}|D ].}t||}t||}t||}	|	|ks8J q8q"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psqsmessagesr   r   prir   ZpubencdecrN   rN   rO   test_encipher_decipher_gm  s    



r   c                   C   sD   t tdd  t tdd  t tdd  ds@J 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   rq   r!  rN   rN   rN   rO   rg     rh   r      )rD   rj   r2   rN   rN   rN   rO   test_gm_private_key  s    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   r"  rR   r   rJ   rq   c                   S   s
   t ddS )Nr   r"  r!  rN   rN   rN   rO   rg     rh   z$test_gm_public_key.<locals>.<lambda>)r2   rD   rj   rN   rN   rN   rO   test_gm_public_key  s    r$  c            
      C   sr   g d} g d }}g d}t | |D ]F\}}t||}|D ].}t||}t||}t||}	|	|ks<J q<q&d S )N)C   r   G   r  r   r   r  r   O   r"  S   r   ;   r     )r   r&  r  r   r   r  r   r'  r"  r(  r   r)  r  r*  r%  )r   iH  iW     r  i  i  r  i[  i  i  ij     )r  r6   r7   r4   r5   r  rN   rN   rO   test_encipher_decipher_bg  s    



r-  c                   C   sD   t tdd  t tdd  t tdd  ds@J 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*  )rD   rj   r6   rN   rN   rN   rO   test_bg_private_key)  s    r0  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%  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   rD   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.external.gmpyr?   Zsympy.matricesr@   Zsympy.ntheoryrA   rB   Zsympy.polys.domainsrC   Zsympy.testing.pytestrD   rE   Zsympy.core.randomrF   rP   rQ   rT   rW   rY   r_   r`   rb   rk   rn   ro   ry   rz   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>   st   		
!2			