iBS d dl Z d dlmZmZ d dlmZ d dlmZmZ d dl m
Z
m Z d dl m
Z
mZ d dlmZ d Z G d d
e j Z G d d e j Z G d
de j Zi fdZedk rd Z e j d dS dS ) N)hexlify unhexlify)list_test_cases)tobytesbchr)AESDES3)SHAKE128SHA256)Counterc 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_CTR.pyget_tag_randomr ( s) z4CtrTests.test_unaligned_data_128.. % ???aq))???r r c : g | ]} | S rg rh ri s r rl z4CtrTests.test_unaligned_data_128.. rm r )r r r r( r) r, joinr* r.
plaintexts ciphertextsr/ s @r test_unaligned_data_128z CtrTests.test_unaligned_data_128 s+ !^c)
s|T\JJJ????:??? s|T\JJJ +..sxx
?S?S0T0TUUU s|T\JJJ????:??? s|T\JJJ +..sxx
?S?S0T0TUUUUUr c dgdz }t j | j t j | j fd|D }t j | j t j | j | d | d | t j | j t j | j fd|D }t j | j t j | j | d | d | d S )Nrd re r% c : g | ]} | S rg rh ri s r rl z3CtrTests.test_unaligned_data_64.. rm r r c : g | ]} | S rg rh ri s r rl z3CtrTests.test_unaligned_data_64.. rm r ) r r r r r( r5 r, ro r* rp s @r test_unaligned_data_64z CtrTests.test_unaligned_data_64 s+ !^c)
$, dkJJJ????:??? $, dkJJJ +..sxx
?S?S0T0TUUU$, dkJJJ????:??? $, dkJJJ +..sxx
?S?S0T0TUUUUUr c V | t t j | j t j d| j | t t j | j t j | j d t j | j t j | j d d S )N r% )r& unknownF)r& use_aesni)r9 r: r r r r( r) r; s r test_unknown_parametersz CtrTests.test_unknown_parameters s )SWdlCLT\ 3 3 3 )SWdlCL"&, ; ; ; clDLERRRRRRr c dD ]\}t j | j t j | j } t || d }| |d ]d S )N)r* r+ r% r )r r r r( r) getattrr, )r. funcr/ results r test_null_encryption_decryptionz(CtrTests.test_null_encryption_decryption si ( * *DWT\3< NNNF*WVT**3//F VS ) ) ) ) * *r c t j | j t j | j }| d | t |j d t j | j t j | j }| d | t |j d d S )Nr% r ) r r r r( r) r* r9 r: r+ r_ s r test_either_encrypt_or_decryptz'CtrTests.test_either_encrypt_or_decrypt s s|T\JJJs )V^S999 s|T\JJJs )V^S99999r c t j dt d dz }d}t j | j t j | }| d|z | t |j d t j | j t j | }| t |j d|dz z t j | j t j | }| d|z | t |j d t j | j t j | }| t |j d|dz z d S ) Nr r! i r% 9 )
r r r r r r( r* r9
OverflowErrorr+ )r. r& max_bytesr/ s r test_wrap_aroundzCtrTests.test_wrap_around s; +aQ" 555 s|WEEEti'((( - >>> s|WEEE - Q 9OPPP s|WEEEti'((( - >>> s|WEEE - Q 9OPPPPPr c J d}d}t j | j t j | j | }| | }t j | j t j t
| j t
| }| t
| }| || | |j |j t j | j t j | j | }| | }t j | j t j t
| j t
| } | t
| }
| ||
d S )Ns 1111111111111111rU rQ )
r r r r( r r* bytearrayr, r? r+ ) r. r ivrJ ref1rK ref2cipher3ref3cipher4ref4s r test_bytearrayzCtrTests.test_bytearray sh
& '$, $
(*, , , t$$'$, )$- 8 8(1"
7 7 7 y// t$$$
666 '$, $
(*, , , t$$'$, )$- 8 8(1"
7 7 7 y// t$$$$$r c t j dt j d }| ddz }t j | }| |d d S )Ns AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAr rD Bi@B @96204fc470476561a3a8f3b6fe6d24be85c87510b638142d1d0fb90989f8a6a6)r r r( r* r hexdigestr, )r. r/ r1 digests r test_very_long_dataz CtrTests.test_very_long_data se CL<<<
^^D7N
+
+B ))++ !cdddddr c B d}t j dt j | j }| | }t d }t j dt j | j }| || }| || | |d t j dt j | j }| || }| || | |d d S Ns 55555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555 4444444444444444rD output)r r r( r r* r r, r+ )r. r0 r/ r1 r ress r test_output_paramzCtrTests.test_output_param s
#,dmDDD
^^B
3 #,dmDDDnnRn// V$$$ d####,dmDDDnnRn// V$$$ d#####r c d}t j dt j | j }| | }t t
d }t j dt j | j }| || | || t j dt j | j }| || | || d S r ) r r r( r r*
memoryviewr r, r+ )r. r0 r/ r1 r s r test_output_param_memoryviewz%CtrTests.test_output_param_memoryview( s
#,dmDDD
^^B
IcNN++#,dmDDDr&))) V$$$#,dmDDDr&))) V$$$$$r c d}d|z }t j dt j | j }| | }t j dt j | j }| t |j |d|z t j dt j | j }| t |j |d|z t |dz
}t j dt j | j }| t |j || t j dt j | j }| t |j || d S )Nr 5r rD 0r r )
r r r( r r* r9 r: r+ r rV )r. LEN_PTr0 r/ r1 shorter_outputs r test_output_param_negz CtrTests.test_output_param_neg7 s>
F]#,dmDDD
^^B
#,dmDDD )V^Rv
NNN#,dmDDD )V^Rv
NNN"6A:..#,dmDDD *fnb PPP#,dmDDD *fnb PPPPPr N)!__name__
__module__ __qualname__r r r r r r r r5 r) r3 r6 r< rB rN rS rW r[ r` rb rs rw r| r r r r r r r r rg r r r r + s nY++G nY++G ~j!,,H ~j!,,H
W[H
-
-
-Fgk"X...G" " "" " "X X X 3 3 3 E E E6A A A*E E E,4 4 4 < < <= = = V V V
V
V
VS S S* * * : : :Q Q Q& % % %>e e e $ $ $"
%
%
% Q Q Q Q Qr r c $ e Zd ZdZd Zd Zd ZdS )SP800TestVectorszQClass exercising the CTR test vectors found in Section F.5
of NIST SP 800-38Ac d}d}d}t j dt d d }t | }t | }t | }t j |t j | }| | | | t j |t j | }| | | | d S ) Nڀ6bc1bee22e409f96e93d7e117393172aae2d8a571e03ac9c9eb76fac45af8e5130c81c46a35ce411e5fbc1191a0a52eff69f2445df4f9b17ad2b417be66c3710ڀ874d6191b620e3261bef6864990db6ce9806f66b7970fdff8617187bb9fffdff5ae4df3edbd5d35e5b4f09020db03eab1e031dda2fbe03d1792170a0f3009cee 2b7e151628aed2a6abf7158809cf4f3cr f0f1f2f3f4f5f6f7f8f9fafbfcfd nbitsr" rF r% r r r r r( r, r* r+ r. r'
ciphertextkeyr& r/ s r test_aes_128z SP800TestVectors.test_aes_128O s ; ;
; "+45S+T+T28: : : nn i(( z**
clG<<< 22J???clG<<<
33Y?????r c d}d}d}t j dt d d }t | }t | }t | }t j |t j | }| | | | t j |t j | }| | | | d S ) Nr ڀ1abc932417521ca24f2b0459fe7e6e0b090339ec0aa6faefd5ccc2c6f4ce8e941e36b26bd1ebc670d1bd1d665620abf74f78a7f6d29809585a97daec58c6b05008e73b0f7da0e6452c810f32b809079e562f8ead2522c6b7br r r r r% r r s r test_aes_192z SP800TestVectors.test_aes_192f s ; ;
K "+45S+T+T28: : : nn i(( z**
clG<<< 22J???clG<<<
33Y?????r c d}d}d}t j dt d d }t | }t | }t | }t j |t j | }| | | | t j |t j | }| | | | d S ) Nr ڀ601ec313775789a5b7a7f504bbf3d228f443e3ca4d62b59aca84e990cacaf5c52b0930daa23de94ce87017ba2d84988ddfc9c58db67aada613c2dd08457941a6@603deb1015ca71be2b73aef0857d77811f352c073b6108d72d9810a30914dff4r r r r r% r r s r test_aes_256z SP800TestVectors.test_aes_256} s ; ;
[ "+45S+T+T28: : : nn i(( z**
clG<<< 22J???clG<<<
33Y?????r N)r r r __doc__r r r rg r r r r K sV @ @ @.@ @ @.@ @ @ @ @r r c d e Zd ZdZg ZeD ]!Ze d eD "d ZdS )RFC3686TestVectors) ) 53696e676c6520626c6f636b206d7367 e4095d4fb7a7b3792d6175a3261311b8 ae6852f8121067cc4bf7a5765577f39e000000300000000000000000)@000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f@5104a106168a72d9790d41ee8edad388eb2e1efc46da57c8fce630df9141be28 7e24067817fae0d743d6ce1f32539163006cb6dbc0543b59da48d90b)H000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f20212223Hc1cf48a89f2ffdd9cf4652e9efdb72d74540a42bde6d7836d59a5ceaaef3105325b2072f 7691be035e5020a8ac6e618529f9a0dc00e0017b27777f3f4a1786f0)r 4b55384fe259c9c84e7935a003cbe928016af5b145fc9f579c175f93e3bfb0eed863d06ccfdb785150000004836733c147d6d93cb)r @453243fc609b23327edfaafa7131cd9f8490701c5ad4a79cfc1fe0ff42f4fb0007c5cb2401b3dc33c19e7340819e0f69c678c3db8e6f6a91a0096b03b020c6eadc2cb500d)r H96893fc55e5c722f540b7dd1ddf7e758d288bc95c69165884536c811662f2188abee0935002bf391ee8ecb159b959617b0965279bf59b60a786d3e0fe0007bdfd5cbd60278dcc0912)r 145ad01dbf824ec7560863dc71e3e0c0@776beff2851db06f4c8a0542c8696f6c6a81af1eec96b4d37fc1d689e6c1c10400000060db5672c97aa8f0b2)r @f05e231b3894612c49ee000b804eb2a9b8306b508f839d6a5530831d9344af1c@f6d66d6bd52d59bb0796365879eff886c66dd51a5b6a99744b50590c87a2388400faac24c1585ef15a43d875)r Heb6c52821d0bbbf7ce7594462aca4faab407df866569fd07f48cc0b583d6071f1ec0e6b8@ff7a617ce69148e4f1726e2f43581de2aa62d9f805532edff1eed687fb54153d001cc5b751a51d70a1c11148c , g | ]}t | S rg )r )rj rk s r rl z RFC3686TestVectors. s 111 ! 111r c | j D ]\ }}}}t j d| }t j |t j | }| | }| t | t | d S )Nr r! r% )bindatar r r r( r* r, r )r. r0 r1 r r" r& r/ r s r runTestzRFC3686TestVectors.runTest s #'< ; ; BC k"V444GWS#,@@@F^^B''F WR[['&// : : : : ; ;r N)r r r r r tvappendr rg r r r r se %
DN G 3 311b1112222; ; ; ; ;r r c g }|t t z
}|t t z
}|t gz
}|S r8 )r r r r )configtestss r get_testsr sE E _X
&
&&E _ -
.
..E !##
%%E Lr __main__c B t j t S r8 )unittest TestSuiter rg r r r s H&y{{33 r suite) defaultTest)r binasciir r Crypto.SelfTest.st_commonr Crypto.Util.py3compatr r
Crypto.Cipherr r Crypto.Hashr
r Crypto.Utilr r TestCaser r r r r r mainrg r r r s > ' ' ' ' ' ' ' ' 5 5 5 5 5 5 / / / / / / / / # # # # # # # # ( ( ( ( ( ( ( ( 8 8 8]Q ]Q ]Q ]Q ]Qx ]Q ]Q ]Q@ F@ F@ F@ F@ F@x( F@ F@ F@R7; 7; 7; 7; 7;* 7; 7; 7;t z 3 3EHMg&&&&&& r