o
    Þý°jtO  ã                   @   s  d dl Z d dlZd dlZd dlmZmZ d dlmZmZm	Z	 d dl
mZ d dlmZ d dlmZ G dd„ dejƒZG d	d
„ d
ejƒZG dd„ dejƒZG dd„ dejƒZG dd„ dejƒZG dd„ dejƒZi fdd„Zedkr€d dlZdd„ Zejdd� dS dS )é    N)ÚhexlifyÚ	unhexlify)ÚbÚtobytesÚbchr)Ústrxor_c)Úlist_test_cases)ÚChaCha20c                   @   s\   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ ZdS )ÚChaCha20Testc                 C   sL   t jtdƒd dd�}|  |jd¡ t jtdƒd dd�}|  |jd¡ d S )NÚ0é    s   00000000©ÚkeyÚnonces   000000000000)r	   Únewr   ÚassertEqualr   )ÚselfÚcipher© r   ú›/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/SelfTest/Cipher/test_ChaCha20.pyÚtest_new_positive-   s   zChaCha20Test.test_new_positivec                 C   sf   t j}|  t|¡ | jt|tdƒd� | jt|tdƒd tdƒd� | jt|tdƒtdƒd d� d S )Nr   )r   é   )r   r   r   )r	   r   ÚassertRaisesÚ	TypeErrorr   Ú
ValueError)r   r   r   r   r   Útest_new_negative3   s
   "zChaCha20Test.test_new_negativec                 C   sN   t jtdƒd d�}t jtdƒd d�}|  t|jƒd¡ |  |j|j¡ d S )Né   r   )r   r   )r	   r   r   r   Úlenr   ÚassertNotEqual)r   Úcipher1Úcipher2r   r   r   Útest_default_nonce:   s   zChaCha20Test.test_default_noncec                 C   sH   d}d}t j||d�}|  ||j¡ d}t j||d�}|  ||j¡ d S )Nó    AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAs   PPPPPPPPr   s   QQQQQQQQQQQQ©r	   r   r   r   )r   r   Únonce1r   Únonce2r    r   r   r   Ú
test_nonce@   s   zChaCha20Test.test_noncec                 C   s„   t jtdƒd tdƒd d�}| tdƒ¡ |  t|jtdƒ¡ t jtdƒd tdƒd d�}| tdƒ¡ |  t|jtdƒ¡ dS )	zFVerify that a cipher cannot be used for both decrypting and encryptingÚ5r   Ú6r   r   Ú8Ú9N)r	   r   r   Úencryptr   r   Údecrypt)r   Úc1Úc2r   r   r   Útest_eiter_encrypt_or_decryptK   s   z*ChaCha20Test.test_eiter_encrypt_or_decryptc                 C   sœ   t dƒd }tjt dƒd t dƒd d�}tjt dƒd t dƒd d�}| |¡}|  | |¡|¡ |  | t dƒ¡t dƒ¡ |  | t dƒ¡t dƒ¡ d S )	NÚAi   r'   r   r(   r   r   Ú )r   r	   r   r+   r   r,   )r   Úptr-   r.   Úctr   r   r   Útest_round_tripV   s   
zChaCha20Test.test_round_tripc              	   C   s  ddl m} d}t|ƒ}tdƒ}t|ƒ|k r%|| |¡ ¡ 7 }t|ƒ|k stjtdƒd tdƒd d	�}| |¡}tjtdƒd tdƒd d	�}tjtdƒd tdƒd d	�}d}	|D ]0}
|  	| 
||	|	|
 … ¡||	|	|
 … ¡ |  	| ||	|	|
 … ¡||	|	|
 … ¡ |	|
7 }	q[d
S )zCVerify that an arbitrary number of bytes can be encrypted/decryptedr   )ÚSHA1)r   é   é   é   é   é   é   r1   Ú7r   Útr   r   N)ÚCryptodome.Hashr5   Úsumr   r   r   Údigestr	   r+   r   r,   )r   r5   ÚsegmentsÚtotalr2   r   r3   r    Úcipher3ÚidxÚsegmentr   r   r   Útest_streaming`   s"   ÿ
**
ýzChaCha20Test.test_streamingc                 C   sŒ   t jtdƒd tdƒd d�}d}tdƒd }| td	ƒ| ¡ | |¡}t jtdƒd tdƒd d�}| |¡ | |¡}|  ||¡ d S )
Nr*   r   Úer   r   iá  Ú1é@   r   )r	   r   r   r+   Úseekr   )r   r   Úoffsetr2   Úct1r    Úct2r   r   r   Ú	test_seekv   s   


zChaCha20Test.test_seekc                 C   sr   t dƒt dƒ t dƒd  }t dƒd }tj||d�}| d¡ ttdƒƒ}| t dƒt|ƒ ¡}|  ||¡ d S )Nr   éÿ   é   r   r   é€   Ú€72d54dfbf12ec44b362692df94137f328fea8da73990265ec1bbbea1ae9af0ca13b25aa26cb4a648cb9b9d1be65b2c0924a66c54d545ec1b7374f4872e99f096)	r   r	   r   rJ   r   r   r+   r   r   )r   r   r   r   Úexpected_key_streamr3   r   r   r   Útest_seek_tv…   s   
ÿzChaCha20Test.test_seek_tvc           
      C   sˆ   g d¢}|D ];}t |d ƒ}t |d ƒ}|d d }t |d ƒ}t |d ƒ}tj||d�}|dkr6| |¡ | |¡}	|	|ksAJ ‚qd S )	N))Ú@0000000000000000000000000000000000000000000000000000000000000000Ú000000000000000000000000r   Ú€00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000Ú€76b8e0ada0f13d90405d6ae55386bd28bdd219b8a08ded1aa836efcc8b770dc7da41597c5157488d7724e03fb8d84a376a43b8f41518a11cc387b669b2ee6586)Ú@0000000000000000000000000000000000000000000000000000000000000001Ú000000000000000000000002r   Áî  416e79207375626d697373696f6e20746f20746865204945544620696e74656e6465642062792074686520436f6e7472696275746f7220666f72207075626c69636174696f6e20617320616c6c206f722070617274206f6620616e204945544620496e7465726e65742d4472616674206f722052464320616e6420616e792073746174656d656e74206d6164652077697468696e2074686520636f6e74657874206f6620616e204945544620616374697669747920697320636f6e7369646572656420616e20224945544620436f6e747269627574696f6e222e20537563682073746174656d656e747320696e636c756465206f72616c2073746174656d656e747320696e20494554462073657373696f6e732c2061732077656c6c206173207772697474656e20616e6420656c656374726f6e696320636f6d6d756e69636174696f6e73206d61646520617420616e792074696d65206f7220706c6163652c207768696368206172652061646472657373656420746fÁî  a3fbf07df3fa2fde4f376ca23e82737041605d9f4f4f57bd8cff2c1d4b7955ec2a97948bd3722915c8f3d337f7d370050e9e96d647b7c39f56e031ca5eb6250d4042e02785ececfa4b4bb5e8ead0440e20b6e8db09d881a7c6132f420e52795042bdfa7773d8a9051447b3291ce1411c680465552aa6c405b7764d5e87bea85ad00f8449ed8f72d0d662ab052691ca66424bc86d2df80ea41f43abf937d3259dc4b2d0dfb48a6c9139ddd7f76966e928e635553ba76c5c879d7b35d49eb2e62b0871cdac638939e25e8a1e0ef9d5280fa8ca328b351c3c765989cbcf3daa8b6ccc3aaf9f3979c92b3720fc88dc95ed84a1be059c6499b9fda236e7e818b04b0bc39c1e876b193bfe5569753f88128cc08aaa9b63d1a16f80ef2554d7189c411f5869ca52c5b83fa36ff216b9c1d30062bebcfd2dc5bce0911934fda79a86f6e698ced759c3ff9b6477338f3da4f9cd8514ea9982ccafb341b2384dd902f3d1ab7ac61dd29c6f21ba5b862f3730e37cfdc4fd806c22f221)Ú@1c9240a5eb55d38af333888604f6b5f0473917c1402b80099dca5cbc207075c0rZ   é*   Úþ2754776173206272696c6c69672c20616e642074686520736c6974687920746f7665730a446964206779726520616e642067696d626c6520696e2074686520776162653a0a416c6c206d696d737920776572652074686520626f726f676f7665732c0a416e6420746865206d6f6d65207261746873206f757467726162652eÚþ62e6347f95ed87a45ffae7426f27a1df5fb69110044c0d73118effa95b01e5cf166d3df2d721caf9b21e5fb14c616871fd84c54f9d65b283196c7fe4f60553ebf39c6402c42234e32a356b3e764312a61a5532055716ead6962568f87d3f3f7704c6a8d1bcd1bf4d50d6154b6da731b187b58dfd728afa36757a797ac188d1r   r   é   rI   r6   é   r   )r   r	   r   rJ   r+   )
r   ÚtvsÚtvr   r   rK   r2   Ú	ct_expectr   r3   r   r   r   Útest_rfc7539•   s   [

õzChaCha20Test.test_rfc7539N)Ú__name__Ú
__module__Ú__qualname__r   r   r!   r&   r/   r4   rF   rN   rT   rf   r   r   r   r   r
   +   s    
r
   c                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚXChaCha20Testc                 C   sb   ddl m} d}t| dd¡ƒ}d}t| dd¡ƒ}|||ƒ}d}t| dd¡ƒ}|  ||¡ d S )	Nr   )Ú
_HChaCha20s_   00:01:02:03:04:05:06:07:08:09:0a:0b:0c:0d:0e:0f:10:11:12:13:14:15:16:17:18:19:1a:1b:1c:1d:1e:1fó   :ó    s/   00:00:00:09:00:00:00:4a:00:00:00:00:31:41:59:27sG   82413b42 27b27bfe d30e4250 8a877d73 a0f9e4d5 8a74a853 c12ec413 26d3ecdcó    )ÚCryptodome.Cipher.ChaCha20rk   r   Úreplacer   )r   rk   r   r   ÚsubkeyÚexpectedr   r   r   Útest_hchacha20  s   
zXChaCha20Test.test_hchacha20c                 C   s(   d}d}t j||d�}|  ||j¡ d S )Nr"   s   PPPPPPPPPPPPPPPPPPPPPPPPr   r#   )r   r   r   r   r   r   r   r&     s   zXChaCha20Test.test_noncec                 C   sz   d}t | dd¡ dd¡ƒ}t dƒ}t dƒ}d}t | dd¡ dd¡ƒ}tj||d�}| d	¡ | |¡}|  ||¡ d S )
Ns
  
                5468652064686f6c65202870726f6e6f756e6365642022646f6c652229206973
                20616c736f206b6e6f776e2061732074686520417369617469632077696c6420
                646f672c2072656420646f672c20616e642077686973746c696e6720646f672e
                2049742069732061626f7574207468652073697a65206f662061204765726d61
                6e20736865706865726420627574206c6f6f6b73206d6f7265206c696b652061
                206c6f6e672d6c656767656420666f782e205468697320686967686c7920656c
                757369766520616e6420736b696c6c6564206a756d70657220697320636c6173
                736966696564207769746820776f6c7665732c20636f796f7465732c206a6163
                6b616c732c20616e6420666f78657320696e20746865207461786f6e6f6d6963
                2066616d696c792043616e696461652eó   
rm   rn   s@   808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fs0   404142434445464748494a4b4c4d4e4f5051525354555658s
  
                7d0a2e6b7f7c65a236542630294e063b7ab9b555a5d5149aa21e4ae1e4fbce87
                ecc8e08a8b5e350abe622b2ffa617b202cfad72032a3037e76ffdcdc4376ee05
                3a190d7e46ca1de04144850381b9cb29f051915386b8a710b8ac4d027b8b050f
                7cba5854e028d564e453b8a968824173fc16488b8970cac828f11ae53cabd201
                12f87107df24ee6183d2274fe4c8b1485534ef2c5fbc1ec24bfc3663efaa08bc
                047d29d25043532db8391a8a3d776bf4372a6955827ccb0cdd4af403a7ce4c63
                d595c75a43e045f0cce1f29c8b93bd65afc5974922f214a40b7c402cdb91ae73
                c0b63615cdad0480680f16515a7ace9d39236464328a37743ffc28f4ddb324f4
                d0f5bbdc270c65b1749a6efff1fbaa09536175ccd29fb9e6057b307320d31683
                8a9c71f70b5b5907a66f7ea49aadc409r   rI   )r   rp   r	   r   rJ   r+   r   )r   r2   r   Úivr3   r   Úct_testr   r   r   Útest_encrypt  s   

zXChaCha20Test.test_encryptN)rg   rh   ri   rs   r&   rw   r   r   r   r   rj     s    rj   c                   @   ó   e Zd ZdZdd„ ZdS )ÚByteArrayTestú+Verify we can encrypt or decrypt bytearraysc                 C   sà   d}d}d}t |ƒ}t |ƒ}t |ƒ}tj||d�}| |¡}tj||d�}	d|d d…< d|d d…< |	 |¡}
|  ||
¡ |  |j|	j¡ t |ƒ}t |ƒ}t |ƒ}tj||d�}d|d d…< d|d d…< | |¡}|  ||¡ d S ©Ns   0123s    99999999999999999999999999999999s   ttttttttr   ó   ÿr   )Ú	bytearrayr	   r   r+   r   r   r,   )r   Údatar   r   Údata_baÚkey_baÚnonce_bar   r3   r    rv   Úct_barC   Úpt_testr   r   r   ÚrunTestF  s,   


zByteArrayTest.runTestN©rg   rh   ri   Ú__doc__r„   r   r   r   r   ry   C  ó    ry   c                   @   rx   )ÚMemoryviewTestrz   c                 C   sø   d}d}d}t t|ƒƒ}t t|ƒƒ}t t|ƒƒ}tj||d�}| |¡}tj||d�}	d|d d…< d|d d…< |	 |¡}
|  ||
¡ |  |j|	j¡ t t|ƒƒ}t t|ƒƒ}t t|ƒƒ}tj||d�}d|d d…< d|d d…< | |¡}|  ||¡ d S r{   )Ú
memoryviewr}   r	   r   r+   r   r   r,   )r   r~   r   r   Údata_mvÚkey_mvÚnonce_mvr   r3   r    rv   Úct_mvrC   rƒ   r   r   r   r„   l  s,   


zMemoryviewTest.runTestNr…   r   r   r   r   rˆ   i  r‡   rˆ   c                   @   s   e Zd Zg d¢Zdd„ ZdS )ÚChaCha20_AGL_NIR))rU   Ú0000000000000000Á   76b8e0ada0f13d90405d6ae55386bd28bdd219b8a08ded1aa836efcc8b770dc7da41597c5157488d7724e03fb8d84a376a43b8f41518a11cc387b669b2ee65869f07e7be5551387a98ba977c732d080dcb0f29a048e3656912c6533e32ee7aed29b721769ce64e43d57133b074d839d531ed1f28510afb45ace10a1f4b794d6f)rY   r�   Á   4540f05a9f1fb296d7736e7b208e3c96eb4fe1834688d2604f450952ed432d41bbe2a0b6ea7566d2a5d1e7e20d42af2c53d792b1c43fea817e9ad275ae5469633aeb5224ecf849929b9d828db1ced4dd832025e8018b8160b82284f3c949aa5a8eca00bbb4a73bdad192b5c42f73f2fd4e273644c8b36125a64addeb006c13a0)rU   Ú0000000000000001Úxde9cba7bf3d69ef5e786dc63973f653a0b49e015adbff7134fcb7df137821031e85a050278a7084527214f73efc7fa5b5277062eb7a0433e445f41e3)rU   Ú0100000000000000Ú€ef3fdfd6c61578fbf5cf35bd3dd33b8009631634d21e42ac33960bd138e50d32111e4caf237ee53ca8ad6426194a88545ddc497a0b466e7d6bbdb0041b2f586b)Ú@000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1fÚ0001020304050607Á   f798a189f195e66982105ffb640bb7757f579da31602fc93ec01ac56f85ac3c134a4547b733b46413042c9440049176905d3be59ea1c53f15916155c2be8241a38008b9a26bc35941e2444177c8ade6689de95264986d95889fb60e84629c9bd9a5acb1cc118be563eb9b3a4a472f82e09a7e778492b562ef7130e88dfe031c79db9d4f7c7a899151b9a475032b63fc385245fe054e3dd5a97a5f576fe064025d3ce042c566ab2c507b138db853e3d6959660996546cc9c4a6eafdc777c040d70eaf46f76dad3979e5c5360c3317166a1c894c94a371876a94df7628fe4eaaf2ccb27d5aaae0ad7ad0f9d4b6ad3b54098746d4524d38407a6deb3ab78fab78c9)rU   Ú0000000000000002Ú€c2c64d378cd536374ae204b9ef933fcd1a8b2288b3dfa49672ab765b54ee27c78a970e0e955c14f3a88e741b97c286f75f8fc299e8148362fa198a39531bed6dc                 C   sb   | j D ]+\}}}tjtt|ƒƒtt|ƒƒd�}tt|ƒƒ}tdƒt|ƒ }|  | |¡|¡ qd S )Nr   ú )rd   r	   r   r   r   r   r   r+   )r   r   r   ÚstreamÚcr3   r2   r   r   r   r„   Ë  s   üzChaCha20_AGL_NIR.runTestN)rg   rh   ri   rd   r„   r   r   r   r   rŽ   �  s    8rŽ   c                   @   s   e Zd Zdd„ ZdS )Ú
TestOutputc           	      C   s¤  d}d}t j||d�}d}| |¡}tt|ƒƒ}t j||d�}|j||d�}|  ||¡ |  |d ¡ t j||d�}|j||d�}|  ||¡ |  |d ¡ ttt|ƒƒƒ}t j||d�}|j||d� |  ||¡ t j||d�}|j||d� |  ||¡ t j||d�}| jt	|j|dt|ƒ d� t j||d�}| jt	|j|dt|ƒ d� tt|ƒd ƒ}t j||d�}| jt
|j||d� t j||d�}| jt
|j||d� d S )Ns    44444444444444444444444444444444s   55555555r   s,  555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555)Úoutputó   0r   )r	   r   r+   r}   r   r   r,   r‰   r   r   r   )	r   r   r   r   r2   r3   rŸ   ÚresÚshorter_outputr   r   r   r„   Õ  s<   
zTestOutput.runTestN)rg   rh   ri   r„   r   r   r   r   rž   Ó  s    rž   c                 C   sP   g }|t tƒ7 }|t tƒ7 }| tƒ ¡ | tƒ ¡ | tƒ ¡ | tƒ ¡ |S ©N)r   r
   rj   ÚappendrŽ   ry   rˆ   rž   )ÚconfigÚtestsr   r   r   Ú	get_tests  s   r§   Ú__main__c                   C   s   t  tƒ ¡S r£   )ÚunittestÚ	TestSuiter§   r   r   r   r   Ú<lambda>  s    r«   Úsuite)ÚdefaultTest)ÚosÚrer©   Úbinasciir   r   ÚCryptodome.Util.py3compatr   r   r   ÚCryptodome.Util.strxorr   ÚCryptodome.SelfTest.st_commonr   ÚCryptodome.Cipherr	   ÚTestCaser
   rj   ry   rˆ   rŽ   rž   r§   rg   r¬   Úmainr   r   r   r   Ú<module>   s*    WB&&D/ý