@vzm> d dl Z d dlZd dlZd dlmZmZ d dlmZmZm Z d dl
m Z d dl m
Z
d dlmZ G d dej Z G d d
ej Z G d d ej Z G d
dej Z G d dej Z G d dej Zi fdZedk rd dlZd Z ej d dS dS ) N)hexlify unhexlify)btobytesbchr)strxor_c)list_test_cases)ChaCha20c J e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d S ) ChaCha20Testc
t j t d dz d }| |j d t j t d dz d }| |j d d S )N0 s 00000000keynonces 000000000000)r
newr assertEqualr )selfciphers /builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/Crypto/SelfTest/Cipher/test_ChaCha20.pytest_new_positivez ChaCha20Test.test_new_positive- st !C&&)6::: x000 !C&&)7;;; y11111 c t j }| t | | t |t d | t
|t d dz t d | t
|t d t d dz d S )Nr )r )r r r )r
r assertRaises TypeErrorr
ValueError)r r s r test_new_negativez ChaCha20Test.test_new_negative3 s l )S))) )S#777 *c3qvvFFF *c3QsVVBYGGGGGr c * t j t d dz }t j t d dz }| t |j d | |j |j d S )N r )r r )r
r r r lenr assertNotEqual)r cipher1cipher2s r test_default_noncez ChaCha20Test.test_default_nonce: sz ,477R<000,477R<000 W]++Q/// GM7=99999r c d}d}t j || }| ||j d}t j || }| ||j d S )N AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAs PPPPPPPPr s QQQQQQQQQQQQr
r r r )r r nonce1r$ nonce2r% s r
test_noncezChaCha20Test.test_nonce@ so ,3f555 ///,3f555 /////r c t j t d dz t d dz }| t d | t
|j t d t j t d dz t d dz }| t d | t
|j t d dS ) zFVerify that a cipher cannot be used for both decrypting and encrypting5r 6r r 89N)r
r r encryptr r decrypt)r c1c2s r test_eiter_encrypt_or_decryptz*ChaCha20Test.test_eiter_encrypt_or_decryptK s \affrk3!
<
<
<
1S66 )RZ3888
\affrk3!
<
<
<
1S66 )RZ388888r c t d dz }t j t d dz t d dz }t j t d dz t d dz }| | }| | | | | | t d t d | | t d t d d S ) NAi r. r r/ r r )r r
r r2 r r3 )r ptr4 r5 cts r test_round_tripz ChaCha20Test.test_round_tripV s
sVVd]
\affrk3!
<
<
<
\affrk3!
<
<
<
ZZ^^ B ,,, AbEE**AbEE222 AbEE**AbEE22222r c t ddl m} d}t | }t d }t | |k r=|| | z
}t | |k =t j t d dz t d dz }| | }t j t d dz t d dz }t j t d dz t d dz }d} |D ]}
| |
|| | |
z || | |
z | | || | |
z || | |
z | |
z
} d
S ) zCVerify that an arbitrary number of bytes can be encrypted/decryptedr )SHA1)r! r9 7r tr r N) Crypto.Hashr> sumr r" r digestr
r2 r r3 ) r r> segmentstotalr: r$ r; r% cipher3idxsegments r test_streamingzChaCha20Test.test_streaming` s $$$$$$+H
rUU"ggoo $((2,,%%'' 'B "ggoo ,1S66B;affqjAAA
__R
,1S66B;affqjAAA,1S66B;affqjAAA G W__RCK-@ A A2c#g+oCV W W W W__RCK-@ A A2c#g+oCV W W W 7NCC r c t j t d dz t d dz }d}t d dz }| t d |z | | }t j t d dz t d dz }| | | | }| || d S )
Nr1 r er r i 1@ r )r
r r r2 seekr )r r$ offsetr: ct1r% ct2s r test_seekzChaCha20Test.test_seekv s ,1S66B;affqjAAA
sVVb[#(((oob!!,1S66B;affqjAAA V oob!! c"""""r c t d t d z t d dz z }t d dz }t j || }| d t t d }| t d t | z }| || d S )Nr r r ڀ72d54dfbf12ec44b362692df94137f328fea8da73990265ec1bbbea1ae9af0ca13b25aa26cb4a648cb9b9d1be65b2c0924a66c54d545ec1b7374f4872e99f096) r r
r rT r r r2 r" r )r r r r expected_key_streamr; s r test_seek_tvzChaCha20Test.test_seek_tv s 1ggS !DGGbL0Q! #U333 F '
/) ) ^^DGGc*=&>&> >
?
? ,b11111r c r g d}|D ]}t |d }t |d }|d dz }t |d }t |d }t j || }|dk r| | | | } | |k sJ d S ) N))@0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000r ڀ00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000ڀ76b8e0ada0f13d90405d6ae55386bd28bdd219b8a08ded1aa836efcc8b770dc7da41597c5157488d7724e03fb8d84a376a43b8f41518a11cc387b669b2ee6586)@0000000000000000000000000000000000000000000000000000000000000001000000000000000000000002r! 416e79207375626d697373696f6e20746f20746865204945544620696e74656e6465642062792074686520436f6e7472696275746f7220666f72207075626c69636174696f6e20617320616c6c206f722070617274206f6620616e204945544620496e7465726e65742d4472616674206f722052464320616e6420616e792073746174656d656e74206d6164652077697468696e2074686520636f6e74657874206f6620616e204945544620616374697669747920697320636f6e7369646572656420616e20224945544620436f6e747269627574696f6e222e20537563682073746174656d656e747320696e636c756465206f72616c2073746174656d656e747320696e20494554462073657373696f6e732c2061732077656c6c206173207772697474656e20616e6420656c656374726f6e696320636f6d6d756e69636174696f6e73206d61646520617420616e792074696d65206f7220706c6163652c207768696368206172652061646472657373656420746f a3fbf07df3fa2fde4f376ca23e82737041605d9f4f4f57bd8cff2c1d4b7955ec2a97948bd3722915c8f3d337f7d370050e9e96d647b7c39f56e031ca5eb6250d4042e02785ececfa4b4bb5e8ead0440e20b6e8db09d881a7c6132f420e52795042bdfa7773d8a9051447b3291ce1411c680465552aa6c405b7764d5e87bea85ad00f8449ed8f72d0d662ab052691ca66424bc86d2df80ea41f43abf937d3259dc4b2d0dfb48a6c9139ddd7f76966e928e635553ba76c5c879d7b35d49eb2e62b0871cdac638939e25e8a1e0ef9d5280fa8ca328b351c3c765989cbcf3daa8b6ccc3aaf9f3979c92b3720fc88dc95ed84a1be059c6499b9fda236e7e818b04b0bc39c1e876b193bfe5569753f88128cc08aaa9b63d1a16f80ef2554d7189c411f5869ca52c5b83fa36ff216b9c1d30062bebcfd2dc5bce0911934fda79a86f6e698ced759c3ff9b6477338f3da4f9cd8514ea9982ccafb341b2384dd902f3d1ab7ac61dd29c6f21ba5b862f3730e37cfdc4fd806c22f221)@1c9240a5eb55d38af333888604f6b5f0473917c1402b80099dca5cbc207075c0rf * 2754776173206272696c6c69672c20616e642074686520736c6974687920746f7665730a446964206779726520616e642067696d626c6520696e2074686520776162653a0a416c6c206d696d737920776572652074686520626f726f676f7665732c0a416e6420746865206d6f6d65207261746873206f757467726162652e62e6347f95ed87a45ffae7426f27a1df5fb69110044c0d73118effa95b01e5cf166d3df2d721caf9b21e5fb14c616871fd84c54f9d65b283196c7fe4f60553ebf39c6402c42234e32a356b3e764312a61a5532055716ead6962568f87d3f3f7704c6a8d1bcd1bf4d50d6154b6da731b187b58dfd728afa36757a797ac188d1r r! rS r? r )r r
r rT r2 )
r tvstvr r rU r: ct_expectr r; s
r test_rfc7539zChaCha20Test.test_rfc7539 s Y
Y
Y
v $ $BBqE""C be$$EURZF2a5!!B!"Q%((I \c777F{{ F### ##B????? $ $r N)
__name__
__module__ __qualname__r r r& r, r6 r< rO rX r_ rr r r r r + s 2 2 2 H H H: : : 0 0 0 9 9 93 3 3 ,
#
#
# 2 2 2 i$ i$ i$ i$ i$r r c e Zd Zd Zd Zd ZdS )
XChaCha20Testc 4 ddl m} d}t | dd }d}t | dd } ||| }d}t | dd }| || 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 :r 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 )Crypto.Cipher.ChaCha20rz r replacer )r rz r r subkeyexpecteds r test_hchacha20z XChaCha20Test.test_hchacha20 s 655555p D#..//B%--c2233C''] X - -dC 8 899 *****r c p d}d}t j || }| ||j d S )Nr( s PPPPPPPPPPPPPPPPPPPPPPPPr r) )r r r r s r r, zXChaCha20Test.test_nonce s> #U333 -----r c d}t | dd dd }t d }t d }d}t | dd dd }t j || }| d | | }| || d S )
Ns
5468652064686f6c65202870726f6e6f756e6365642022646f6c652229206973
20616c736f206b6e6f776e2061732074686520417369617469632077696c6420
646f672c2072656420646f672c20616e642077686973746c696e6720646f672e
2049742069732061626f7574207468652073697a65206f662061204765726d61
6e20736865706865726420627574206c6f6f6b73206d6f7265206c696b652061
206c6f6e672d6c656767656420666f782e205468697320686967686c7920656c
757369766520616e6420736b696c6c6564206a756d70657220697320636c6173
736966696564207769746820776f6c7665732c20636f796f7465732c206a6163
6b616c732c20616e6420666f78657320696e20746865207461786f6e6f6d6963
2066616d696c792043616e696461652e
r r| s@ 808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fs0 404142434445464748494a4b4c4d4e4f5051525354555658s
7d0a2e6b7f7c65a236542630294e063b7ab9b555a5d5149aa21e4ae1e4fbce87
ecc8e08a8b5e350abe622b2ffa617b202cfad72032a3037e76ffdcdc4376ee05
3a190d7e46ca1de04144850381b9cb29f051915386b8a710b8ac4d027b8b050f
7cba5854e028d564e453b8a968824173fc16488b8970cac828f11ae53cabd201
12f87107df24ee6183d2274fe4c8b1485534ef2c5fbc1ec24bfc3663efaa08bc
047d29d25043532db8391a8a3d776bf4372a6955827ccb0cdd4af403a7ce4c63
d595c75a43e045f0cce1f29c8b93bd65afc5974922f214a40b7c402cdb91ae73
c0b63615cdad0480680f16515a7ace9d39236464328a37743ffc28f4ddb324f4
d0f5bbdc270c65b1749a6efff1fbaa09536175ccd29fb9e6057b307320d31683
8a9c71f70b5b5907a66f7ea49aadc409r rS )r r~ r
r rT r2 r )r r: r ivr; r ct_tests r test_encryptzXChaCha20Test.test_encrypt s
4 rzz%--55dC@@
A
A[\\
J
K
K
4 rzz%--55dC@@
A
A #R000 B..$$ W%%%%%r N)rs rt ru r r, r rv r r rx rx sA + + +$. . . #& #& #& #& #&r rx c e Zd ZdZd ZdS )
ByteArrayTest+Verify we can encrypt or decrypt bytearraysc d}d}d}t | }t | }t | }t j || }| | }t j || } d|d d< d|d d< | | }
| ||
| |j | j t | }t | }t | } t j || } d|d d< d|d d< | | }
| ||
d S Ns 0123s 99999999999999999999999999999999s ttttttttr r! ) bytearrayr
r r2 r r r3 )r datar r data_bakey_banonce_bar$ r; r% r ct_barL pt_tests r runTestzByteArrayTest.runTestF sJ D//3 U##,3e444
__T
"
",6::: rr
! //'** W%%%
666 3 U##"
,6::: rr
! //%(( w'''''r Nrs rt ru __doc__r rv r r r r C ) 55 ( ( ( ( (r r c e Zd ZdZd ZdS )MemoryviewTestr c 4 d}d}d}t t | }t t | }t t | }t j || }| | }t j || } d|d d< d|d d< | | }
| ||
| |j | j t t | }t t | }t t | } t j || } d|d d< d|d d< | | }
| ||
d S r )
memoryviewr r
r r2 r r r3 )r r r r data_mvkey_mvnonce_mvr$ r; r% r ct_mvrL r s r r zMemoryviewTest.runTestl sz Yt__--IcNN++ i . .//,3e444
__T
"
",6::: rr
! //'** W%%%
666 IcNN++ i . .//9R==)),6::: rr
! //%(( w'''''r Nr rv r r r r i r r r c e Zd Zg dZd ZdS )ChaCha20_AGL_NIR))ra 0000000000000000 76b8e0ada0f13d90405d6ae55386bd28bdd219b8a08ded1aa836efcc8b770dc7da41597c5157488d7724e03fb8d84a376a43b8f41518a11cc387b669b2ee65869f07e7be5551387a98ba977c732d080dcb0f29a048e3656912c6533e32ee7aed29b721769ce64e43d57133b074d839d531ed1f28510afb45ace10a1f4b794d6f)re r 4540f05a9f1fb296d7736e7b208e3c96eb4fe1834688d2604f450952ed432d41bbe2a0b6ea7566d2a5d1e7e20d42af2c53d792b1c43fea817e9ad275ae5469633aeb5224ecf849929b9d828db1ced4dd832025e8018b8160b82284f3c949aa5a8eca00bbb4a73bdad192b5c42f73f2fd4e273644c8b36125a64addeb006c13a0)ra 0000000000000001xde9cba7bf3d69ef5e786dc63973f653a0b49e015adbff7134fcb7df137821031e85a050278a7084527214f73efc7fa5b5277062eb7a0433e445f41e3)ra 0100000000000000ڀef3fdfd6c61578fbf5cf35bd3dd33b8009631634d21e42ac33960bd138e50d32111e4caf237ee53ca8ad6426194a88545ddc497a0b466e7d6bbdb0041b2f586b)@000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f0001020304050607 f798a189f195e66982105ffb640bb7757f579da31602fc93ec01ac56f85ac3c134a4547b733b46413042c9440049176905d3be59ea1c53f15916155c2be8241a38008b9a26bc35941e2444177c8ade6689de95264986d95889fb60e84629c9bd9a5acb1cc118be563eb9b3a4a472f82e09a7e778492b562ef7130e88dfe031c79db9d4f7c7a899151b9a475032b63fc385245fe054e3dd5a97a5f576fe064025d3ce042c566ab2c507b138db853e3d6959660996546cc9c4a6eafdc777c040d70eaf46f76dad3979e5c5360c3317166a1c894c94a371876a94df7628fe4eaaf2ccb27d5aaae0ad7ad0f9d4b6ad3b54098746d4524d38407a6deb3ab78fab78c9)ra 0000000000000002ڀc2c64d378cd536374ae204b9ef933fcd1a8b2288b3dfa49672ab765b54ee27c78a970e0e955c14f3a88e741b97c286f75f8fc299e8148362fa198a39531bed6dc ~ | j D ]\ }}}t j t t | t t | }t t | }t d t | z }| | | | d S )Nr )rp r
r r r r" r r2 )r r r streamcr; r: s r r zChaCha20_AGL_NIR.runTest s $(G 0 0 S% 1S66!2!2)AeHH:M:MNNNA1V99%%B6SWW$B QYYr]]B / / / / 0 0r N)rs rt ru rp r rv r r r r s5 6
6
6
Bp0 0 0 0 0r r c e Zd Zd ZdS )
TestOutputc d}d}t j || }d}| | }t t | }t j || }| || }| || | |d t j || }| || }| || | |d t t t | }t j || }| || | || t j || }| || | || t j || }| t |j |dt | z t j || }| t |j |dt | z t t | dz
}t j || }| t |j || t j || }| t |j || d S )Ns 44444444444444444444444444444444s 55555555r s, 555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555)output 0r! ) r
r r2 r r" r r3 r r r r ) r r r r r: r; r resshorter_outputs r r zTestOutput.runTest s #U333
^^B
3r77## #U333nnRn// V$$$ d### #U333nnRn// V$$$ d###Ic"gg . .// #U333r&))) V$$$ #U333r&))) V$$$ #U333 )V^RSWW MMM #U333 )V^RSWW MMM"3r77Q;// #U333 *fnb PPP #U333 *fnb PPPPPr N)rs rt ru r rv r r r r s( *Q *Q *Q *Q *Qr r c n g }|t t z
}|t t z
}| t | t | t
| t |S N)r r rx appendr r r r )configtestss r get_testsr s E _\
*
**E _]
+
++E LL!##$$$ LL !!! LL !!""" LL Lr __main__c B t j t S r )unittest TestSuiter rv r r r s H&y{{33 r suite) defaultTest)osrer binasciir r Crypto.Util.py3compatr r r Crypto.Util.strxorr Crypto.SelfTest.st_commonr
Crypto.Cipherr
TestCaser rx r r r r r rs r mainrv r r r s >
' ' ' ' ' ' ' ' 2 2 2 2 2 2 2 2 2 2 ' ' ' ' ' ' 5 5 5 5 5 5 " " " " " "S$ S$ S$ S$ S$8$ S$ S$ S$l?& ?& ?& ?& ?&H% ?& ?& ?&D#( #( #( #( #(H% #( #( #(L#( #( #( #( #(X& #( #( #(LA0 A0 A0 A0 A0x( A0 A0 A0H,Q ,Q ,Q ,Q ,Q" ,Q ,Q ,Q^ zOOO 3 3EHMg&&&&&& r