*.NH d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d Z G d
d e j Z G d d
e j Z G d de j Zd Z G d de j Z G d de j Z G d de j Zi fdZedk rd Z e j d dS dS ) N) unhexlify)list_test_cases load_test_vectors_wycheproof)tobytes)ChaCha20_Poly1305)SHAKE128)strxorc l t j t | | S )N)data)r newr read)taglengths /builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/Crypto/SelfTest/Cipher/test_ChaCha20_Poly1305.pyget_tag_randomr + s) "&4<888> Vb))) FF+++++r c ` | t t j | j d d S )N )r paramr4 r7 s r test_unknown_parametersz-ChaCha20Poly1305Tests.test_unknown_parameters{ s9 )+/ "l ! # # # # #r c dD ]Q}t j | j | j } t || d }| |d Rd S )Nr! r" r r )r r
r r getattrr# )r$ funcr% results r test_null_encryption_decryptionz5ChaCha20Poly1305Tests.test_null_encryption_decryption sp ( * *D&*t|15 @ @ @F*WVT**3//F VS ) ) ) ) * *r c ^ t j | j | j }| d | t |j d t j | j | j }| d | t |j d d S )Nr r )r r
r r r! r5 r6 r" r@ s r test_either_encrypt_or_decryptz4ChaCha20Poly1305Tests.test_either_encrypt_or_decrypt s "&4<-1]< < <s )V^S999"&4<-1]< < <s )V^S99999r c
t j | j | j }| t
|j d t j | j | j }| t
|j d d S )Nr ztest1234567890-*)r r
r r r5 r6 r! r" r@ s r test_data_must_be_bytesz-ChaCha20Poly1305Tests.test_data_must_be_bytes s~ "&4<-1]< < < )V^5HIII"&4<-1]< < < )V^5HIIIIIr c t j | j | j }| | j \ }}| t | d d S )Nr r )r r
r r encrypt_and_digestr r# r- )r$ r% _macs r test_mac_lenz"ChaCha20Poly1305Tests.test_mac_len s] "&4<-1]< < <**4=993 S2&&&&&r c ( ddl m} t j | j | j }| | j \ }} ||d }t j | j | j }| t |j
|| d S )Nr )strxor_cr ) Crypto.Util.strxorrX r r
r r rS r r5 r: decrypt_and_verify)r$ rX r% r' rU invalid_macs r test_invalid_macz&ChaCha20Poly1305Tests.test_invalid_mac s //////"&4<-1]< < <++DM::C hsD)) "&4<-1]< < < *f&?% ' ' ' ' 'r c B t j | j | j }| }| | t | t j | j | j }| | d S Nr ) r r
r r hexdigestr# digestr hexverify)r$ r% mac_hexs r test_hex_macz"ChaCha20Poly1305Tests.test_hex_mac s "&4<-1]< < <""$$ )G*<*<==="&4<-1]< < <!!!!!r c t dd }t dd }t j | j | j }| | | | \ }}d }dD ]}t j | j | j } ||| D ]}| | d} ||| D ]}| | | z
} | || | | dD ]}t j | j | j } ||| D ]}| | d}
||| D ]}|
|
| z
}
| ||
| | | d S )Nzauthenticated data r r c \ fdt dt D S )Nc * g | ]}||z S ri ).0i chunk_lengthr s r
zOChaCha20Poly1305Tests.test_message_chunks..break_up.. s3 # # #qD1\>)* # # #r r )ranger- )r rl s ``r break_upz;ChaCha20Poly1305Tests.test_message_chunks..break_up sJ # # # # #E!SYY 5" 5" # # #
#r )
rY
r ( P r ) r r r
r r updaterS r" r# verifyr! ra ) r$ auth_datar r%
ciphertextref_macro rl chunkr( ct2s r test_message_chunksz)ChaCha20Poly1305Tests.test_message_chunks s& ##7== ";44 "&4<-1]< < <
i $ 7 7 B B
G # # #
@ # #L&*t|15 @ @ @F ")\::
%
%
e$$$$C!*l;;
-
-v~~e,,, Y , , , MM' " " " " @ 7 7L&*t|15 @ @ @F ")\::
%
%
e$$$$C!)\::
-
-v~~e,,, Z - - - V]]__g 6 6 6 6 7 7r c t | j }t | j }t | j }t | j }t j | j | j }| | j | | j }| }t j | j | j }d|d d< d|d d< | | d|d d< | | } d|d d< | }
| || | ||
| |j
|j
t | j }t | j }t | j }t | } t | } ~t j | j | j }
d|d d< d|d d< |
| d|d d< |
| }d| d d< |
| | || j d S Nr s rq s )
bytearrayr r r r r
rx r! ra r# r r" ry )r$ key_banonce_ba header_badata_bacipher1r' r cipher2ct_testtag_testct_batag_bacipher3pt_tests r test_bytearrayz$ChaCha20Poly1305Tests.test_bytearray sQ 4<(( T]++ dm,, DM**#'DL.2m= = =t}%%%
__T]
+
+nn #'DL.2m= = =$rr
&! y!!!' "1"
//'**% >>## W%%% h'''
666 4<(( T]++ dm,, "
3 #'DL.2m= = =$rr
&! y!!!' "1"
//%((#bqb v $-00000r c t t | j }t t | j }t t | j }t t | j }t j | j | j }| | j | | j }| }t j | j | j }d|d d< d|d d< | | d|d d< | | } d|d d< | }
|
|| |
||
|
|j |j t t | j }t t | j }t t | j }t t | } t t | } ~t j | j | j }
d|d d< d|d d< |
| d|d d< |
| }d| d d< |
| |
|| j d S r )
memoryviewr r r r r r
rx r! ra r# r r" ry )r$ key_mvnonce_mv header_mvdata_mvr r' r r r r ct_mvtag_mvr r s r test_memoryviewz%ChaCha20Poly1305Tests.test_memoryview s Idl 3 344 i
6 677 y 7 788 Yt} 5 566#'DL.2m= = =t}%%%
__T]
+
+nn #'DL.2m= = =$rr
&! y!!!' "1"
//'**% >>## W%%% h'''
666 Idl 3 344 i
6 677 y 7 788 9R==))IcNN++ #'DL.2m= = =$rr
&! y!!!' "1"
//%((#bqb v $-00000r N)__name__
__module__ __qualname__r r r r r) r1 r8 r; rA rC rG rM rO rQ rV r] rd r r r ri r r r r / s< nY++G ~j"--H ~j"--H " " "< < <21 1 1 % % %8 8 8 , , ,# # # * * * : : :J J J' ' ' ' ' '" " "*7 *7 *7X.1 .1 .1`.1 .1 .1 .1 .1r r c e Zd Zd Zd ZdS )XChaCha20Poly1305Testsc t j dd }| t |j d | |j d d S )Ns YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYs HHHHHHHHHHHHHHHHHHHHHHHHr )r r
r# r- r r@ s r r1 z!XChaCha20Poly1305Tests.test_nonceH s\ "&9-68 8 8 V\**B/// y11111r c d}t | dd dd }t d }t d }t d }d}t | dd dd }t d }t j ||
}| | | | \ }} | || | || t j ||
}| | | || d S ) Ns(
4c616469657320616e642047656e746c656d656e206f662074686520636c6173
73206f66202739393a204966204920636f756c64206f6666657220796f75206f
6e6c79206f6e652074697020666f7220746865206675747572652c2073756e73
637265656e20776f756c642062652069742e
r s 50515253c0c1c2c3c4c5c6c7s@ 808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fs0 404142434445464748494a4b4c4d4e4f5051525354555657s(
bd6d179d3e83d43b9576579493c0e939572a1700252bfaccbed2902c21396cbb
731c7f1b0b4aa6440bf3a82f4eda7e39ae64c6708c54c216cb96b72e1213b452
2f8c9ba40db5d945b11b69b982c1bb9e3f3fac2bc369488f76b2383565d3fff9
21f9664c97637da9768812f615c68b13b52es c0875924c1c7987947deafd8780acf49r )r replacer r
rx rS r# r[ )
r$ r& aadr ivr' r r% r r s
r test_encryptz#XChaCha20Poly1305Tests.test_encryptO sP 8
rzz%--55dC@@
A
A344[\\
J
K
K8
rzz%--55dC@@
A
A;<<"&3b999
c " 5 5b 9 9 W%%% h'''"&3b999
c!!"c*****r N)r r r r1 r ri r r r r F s2 2 2 2!+ !+ !+ !+ !+r r c e Zd Z edd Z edd Z edd Zd Zd Zd Z d
Z
d Z d Z d
Z
d Zd Zd Zd ZdS )ChaCha20Poly1305FSMTestsr r r r r r c 6 t j | j | j }| | j }| }t j | j | j }| | | | d S r_ ) r r
r r r! r ra r" ry r$ r% r' rU s r -test_valid_init_encrypt_decrypt_digest_verifyzFChaCha20Poly1305FSMTests.test_valid_init_encrypt_decrypt_digest_verifyy s #&4<-1]< < <
^^DM
*
*mmoo #&4<-1]< < <r
cr c @ t j | j | j }| | j | }t j | j | j }| | j | | d S r_ )r r
r r rx r ra ry r$ r% rU s r $test_valid_init_update_digest_verifyz=ChaCha20Poly1305FSMTests.test_valid_init_update_digest_verify s #&4<-1]< < <
dm$$$mmoo #&4<-1]< < <
dm$$$
cr c t j | j | j }| | j | | j }| }t j | j | j }| | j | | | | d S r_ )
r r
r r rx r r! ra r" ry r s r test_valid_full_pathz-ChaCha20Poly1305FSMTests.test_valid_full_path s #&4<-1]< < <
dm$$$
^^DM
*
*mmoo #&4<-1]< < <
dm$$$r
cr c n t j | j | j }| d S r_ )r r
r r ra r@ s r test_valid_init_digestz/ChaCha20Poly1305FSMTests.test_valid_init_digest s3 "&4<-1]< < <
r c t j | j | j }| }t j | j | j }| | d S r_ )r r
r r ra ry r s r test_valid_init_verifyz/ChaCha20Poly1305FSMTests.test_valid_init_verify sg "&4<-1]< < <mmoo"&4<-1]< < <
cr c J dD ]}d d| j | j dz fD ]}t j | j | j }|| | t
|| } || j || j || j || j d S )NrI s 333 3r )r r r
r r rx rJ )r$ method_namerz r% methods r &test_valid_multiple_encrypt_or_decryptz?ChaCha20Poly1305FSMTests.test_valid_multiple_encrypt_or_decrypt s / & &K "FDM "md 2 4
&
& *.4<59]D D D (MM),,, 55t}%%%t}%%%t}%%%t}%%%%
& & &r c t j | j | j }| | j | }t d D ]*}| || +t j | j | j }| | j t d D ]}| | d S )Nr )
r r
r r rx r ra rn r# ry )r$ r% first_macxs r $test_valid_multiple_digest_or_verifyz=ChaCha20Poly1305FSMTests.test_valid_multiple_digest_or_verify s "&4<-1]< < <
dm$$$MMOO q 9 9A Y
8 8 8 8 #&4<-1]< < <
dm$$$q % %A MM) $ $ $ $ % %r c t j | j | j }| | j | | j \ }}t j | j | j }| | j | || }| | j | d S r_ ) r r
r r rx r rS r[ r# )r$ r% r' rU r& s r 0test_valid_encrypt_and_digest_decrypt_and_verifyzIChaCha20Poly1305FSMTests.test_valid_encrypt_and_digest_decrypt_and_verify s "&4<-1]< < <
dm$$$++DM::C #&4<-1]< < <
dm$$$
&
&r3
/
/ +++++r c 2 dD ]\ }}dD ]}t j | j | j }|r| | j t
|| | j | t t
|| | j d S )N)rI )r" r! )TFr ) r r
r r rx r rJ r5 r6 )r$ method1_name method2_nameassoc_data_presentr% s r #test_invalid_mixing_encrypt_decryptz.` s> PPPYqyyb 1 1 9 9#r B BCCPPPr ri )rj tvs r rm zTestVectorsRFC.` s( lllUWPPRPPPlllr c | j D ]\ }}}}}}t j || }| | | | \ }} | || | || t j || }| | | || }
| ||
d S r_ ) test_vectorsr r
rx rS r# r[ ) r$
assoc_datar& r' rU r r r% r~ mac2r( s r runTestzTestVectorsRFC.runTestb s 373D & & /JBS%&*s%@@@F MM* % % % 11"55IC R % % % S$ ' ' ' '*s%@@@F MM* % % %++B44C R % % % % & &r N)r r r __doc__test_vectors_hexr r ri r r r r sS !!
,'
/?B ml[klllL
&
&
&
&
&r r c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestVectorsWycheproofc ` t j | || _ d| _ d S )NNone)unittestTestCase__init___wycheproof_warnings_id)r$ wycheproof_warningss r r z TestVectorsWycheproof.__init__t s, ""4((($7!r c B d }d }t d|dd|id|i }|S )Nc | d dz S )NtagSizer+ ri )groups r
filter_tagz4TestVectorsWycheproof.load_tests..filter_tag{ s #q( (r c | d S )N algorithmri )roots r filter_algoz5TestVectorsWycheproof.load_tests..filter_algo~ s
$ $r )Cipher
wycheproofz Wycheproof ChaCha20-Poly1305algotag_size)root_tag group_tagr )r$ filenamer r rL s r
load_testsz TestVectorsWycheproof.load_testsy sY ) ) ) % % % ..F/7/M9?8M:Dj9Q S S S
r c g | _ | j | d | j | d d S )Nzchacha20_poly1305_test.jsonz xchacha20_poly1305_test.json)r extendr r7 s r setUpzTestVectorsWycheproof.setUp sQ t'DEEFFF t'EFFGGGGGr c | j S N)r r7 s r shortDescriptionz&TestVectorsWycheproof.shortDescription s xr c ~ |j r3| j r.dd l}| d| j d|j d d S d S d S )Nr zWycheproof warning: z ())warningr warningswarnr comment)r$ r r s r r zTestVectorsWycheproof.warn sd
: R$3 R OOO MMM4888RZZZP Q Q Q Q Q R R R Rr c d|j d|j | _ t j |j |j }nA# t $ r4}t |j dvrdt | v sJ Y d }~d S d }~ww xY w|
|j | |j
\ }}|j rM| ||j | ||j | | d S d S )NzWycheproof Encrypt Test #r r+ r
Nonce must be)r idr r r
r r r: r- strrx r rS msgvalidr# r' r r )r$ r r% er' r s r r z"TestVectorsWycheproof.test_encrypt s 79wwwF &*rvRUCCCFF ru::W,, CFF1J1J1JJ FFFFF
bf ++BF33C
8 R ' ' ' S"& ) ) ) IIbMMMMM s :
A8)A33A8c d|j d|j | _ t j |j |j }nA# t $ r4}t |j dvrdt | v sJ Y d }~d S d }~ww xY w|
|j | |j
|j }|j sJ | ||j | | d S # t $ r
|j rJ Y d S w xY w)NzWycheproof Decrypt r r r r )r r r r r
r r r: r- r rx r r[ r' r r r# r
r )r$ r r% r r& s r test_decryptz"TestVectorsWycheproof.test_decrypt s. 79wwwF &*rvRUCCCFF ru::W,, CFF1J1J1JJ FFFFF
bf **25"&99B 8OO8 R ( ( ( IIbMMMMM x < s' :
A8)A33A8 C1 1DDc dt |j z | _ t |j dk st |j dk rd S t
j |j |j }| |j
t |j dt |j dz
z dz }| t |j ||j d S )Nz3Wycheproof Corrupt Decrypt ChaCha20-Poly1305 Test #r rY r )r r r r- r r' r r
r rx r r
r5 r: r[ r )r$ r r%
ct_corrupts r test_corrupt_decryptz*TestVectorsWycheproof.test_corrupt_decrypt s H3ru::U ru::??c"%jj1nn F"&26???
bf BE7c"%jj1n#=#GHH
*f&?RVTTTTTr c | j D ]A}| | | | | | Bd S r )r r r r )r$ r s r r z TestVectorsWycheproof.runTest s\ ' * *B b ! ! ! b ! ! ! % %b ) ) ) ) * *r N) r r r r r r r r r r r r ri r r r r r s
H H H
R R R
&U U U* * * * *r r c e Zd Zd ZdS )
TestOutputc d}d}t j || }d}| | }t d }t j || }| || }| || | |d t j || }| || }| || | |d t
t d }t j || }| || | || t j || }| || | || t j || }| t |j |d t j || }| t |j |d t d }t j || }| t |j || t j || }| t |j || d S ) Ns 44444444444444444444444444444444s 555555555555r s 5555555555555555r )outputs 0000000000000000rr )
r r
r! r r# r" r r5 r6 r: ) r$ r r r% r& r' r resshorter_outputs r r zTestOutput.runTest sf "&3e<<<
^^B
2 "&3e<<<nnRn// V$$$ d###"&3e<<<nnRn// V$$$ d###IbMM**"&3e<<<r&))) V$$$"&3e<<<r&))) V$$$"&3e<<< )V^RHHH"&3e<<< )V^RHHH"1 "&3e<<< *fnb PPP"&3e<<< *fnb PPPPPr N)r r r r ri r r r r s( *Q *Q *Q *Q *Qr r c , | d }g }|t t z
}|t t z
}|t t z
}|t gz
}|t
| gz
}|t gz
}|S )Nr )getr r r r r r r )configr testss r get_testsr s **%:;; E _ 2
3
33E _ 3
4
44E _ 5
6
66E n
E #$788
99E jll^E Lr __main__c F t j t d S r )r TestSuiter ri r r suiter# s 9;;'''''r r# ) defaultTest) r binasciir Crypto.SelfTest.st_commonr Crypto.SelfTest.loaderr Crypto.Util.py3compatr
Crypto.Cipherr Crypto.Hashr rZ r
r r r r r r r r r r r r# mainri r r r, s@ > 5 5 5 5 5 5 ? ? ? ? ? ? ) ) ) ) ) ) + + + + + + % % % % % %8 8 8T1 T1 T1 T1 T1H - T1 T1 T1n*+ *+ *+ *+ *+X . *+ *+ *+Za- a- a- a- a-x 0 a- a- a-H: : :T& T& T& T& T&X& T& T& T&nT* T* T* T* T*H - T* T* T*n,Q ,Q ,Q ,Q ,Q" ,Q ,Q ,Q^
z( ( (HMg&&&&&& r