N$$[ d Z ddlZddlmZmZmZ ddlmZ ddlm Z G d d Z
e
fdZ G d d
ej Z
G d d e
Z G d
dej Z G d dej Z G d dej Z G d dej Z G d dej Z G d dej Z e fdZd ZdS )z&Self-testing for PyCrypto hash modules N)a2b_hexb2a_hexhexlify)b)strxor_cc e Zd ZdS )
_NoDefaultN)__name__
__module__ __qualname__ /builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/Crypto/SelfTest/Cipher/common.pyr r ! s r r c X | | }n # t $ r |t u r |cY S w xY w| |= |S )zAGet an item from a dictionary, and remove it from the dictionary.)KeyErrorr )dkdefaultretvals r _extractr " sR 1 j
! Ms $$c , e Zd Zd Zd Zd Zd Zd ZdS )CipherSelfTestc t j | || _ | }t |d | _ t t |d | _ t t |d | _ t t |d | _
t |dd | _ t |dd | _ t |dd | _
| j rt | j
| _
t |dd }t | | _ |qt! | j d |z | _ t |d
d | _ | j t |d d | _ | j t | j | _ n=d | _ t |d
d | _ | j t | j | _ || _ d S N descriptionkey plaintext
ciphertext module_name
assoc_datamacmodeMODE_ivnonceunittestTestCase__init__modulecopyr r r r r r r r r! str mode_namegetattrr" r$ extra_paramsselfr* paramsr" s r r) zCipherSelfTest.__init__0 s ""4((( #FM::Xfe,,--8FK8899HV\ : :;;#FM4@@"6<>>FE400 ? #{{DH-- T Wt^< >>>r c t | j }t | j }g }| j rd | j D }d }d }t d D ]}| }| }|D ],} | | | | -t | | }
t | | } |r,|
||
|
|| |
| }}|
| j | |
| j | | j ret | } |
| j | |
t | j d S d S )Nc F g | ] }t t | S r
)r r .0xs r
z*CipherSelfTest.runTest..o s$ BBBQ71Q44==BBBr )r r r r ranger< updater encryptdecrypt assertEqualr! digestverify)
r1 r r r ctpticipherdeciphercompctXptXr! s
r runTestzCipherSelfTest.runTestj s DN++ T_--
? CBB$/BBBJ
q AYY[[Fyy{{H #
&
&
d###%%%%&..3344C(**:6677C
* S))) S)))#BB "--- ,,, 8 /&--//**C TXs + + + OOGDH - - . . . . . / /r N)r
r r r) r6 r< rA rX r
r r r r . s` $# $# $#L
:
:
:? ? ?
%/ %/ %/ %/ %/r r c e Zd Zd Zd ZdS )CipherStreamingSelfTestc D | j }| j |d| j dz
}|dS )Nz in z modez# should behave like a stream cipher)r r" r- )r1 descs r r6 z(CipherStreamingSelfTest.shortDescription s4 9 DT^^^5 5D:>$$@@r c & t | j }t | j }g }| }t dt | d D ]5}| | |||dz 6t t d
| }| | j | g }| }t dt | d D ]5}| | |||dz 6t t d
| }| | j | d S )Nr ) r r r r< rI lenappendrK r r joinrM )r1 r r ct3rS rR pt3s r rX z CipherStreamingSelfTest.runTest sV DN++ T_--
q#i..!,, 9 9A JJv~~i!A#&677 8 8 8 8aeejjoo&& #... q#j//1-- : :A JJv~~j1Q3&788 9 9 9 9aeejjoo&& -----r N)r
r r r6 rX r
r r rZ rZ s5 A A A . . . . .r rZ c e Zd Zd Zd Zd ZdS )
RoundtripTestc @ ddl m} t j | || _ | |j | _ t |d | _
dt |d z | _ | dd | _
d S )Nr )Randomr d r r )Cryptorh r' r( r) r* get_random_bytes
block_sizer$ r r r getr )r1 r* r2 rh s r r) zRoundtripTest.__init__ s !!!!!!""4((( ))&*;<<VE]## q !4 5 55!::mT::r c | j dS )Nz6 .decrypt() output of .encrypt() should not be garbled)r r5 s r r6 z RoundtripTest.shortDescription s QUQaQaQaccr c f | j j }| j t | j | }| | j }| j t | j | }| | }| | j | d S r4 ) r* MODE_ECBr9 r r rK r rL rM )r1 r" encryption_cipherr decryption_cipherdecrypted_plaintexts r rX zRoundtripTest.runTest s {# KOOGDH,=,=t D D&..t~>>
KOOGDH,=,=t D D / 7 7
C C )<=====r N)r
r r r) r6 rX r
r r rf rf sD ; ; ;d d d> > > > >r rf c & e Zd Zd Zd Zd Zd ZdS ) IVLengthTestc t j | || _ t |d | _ d S Nr r' r( r) r* r r r1 r* r2 s r r) zIVLengthTest.__init__ 7 ""4((( VE]##r c dS )NzTCheck that all modes except MODE_ECB and MODE_CTR require an IV of the proper lengthr
r5 s r r6 z IVLengthTest.shortDescription s eer c | t | j j t | j | j j t d d S )Nr_ ) assertRaises TypeErrorr* r9 r r rp r r5 s r rX zIVLengthTest.runTest sH )T[_gdh6G6G $aee - - - - -r c d| j j z S )N )r* rl r5 s r _dummy_counterzIVLengthTest._dummy_counter s dk,,,r N)r
r r r) r6 rX r r
r r ru ru sS $ $ $
f f f- - -- - - - -r ru c e Zd Zd Zd ZdS )NoDefaultECBTestc t j | || _ t |d | _ d S rw rx ry s r r) zNoDefaultECBTest.__init__ rz r c v | t | j j t | j d S r4 )r} r~ r* r9 r r r5 s r rX zNoDefaultECBTest.runTest s. )T[_gdh6G6GHHHHHr Nr
r r r) rX r
r r r r s7 $ $ $
I I I I Ir r c e Zd Zd Zd ZdS )
BlockSizeTestc t j | || _ t t |d | _ d S rw )r' r( r) r* r r r ry s r r) zBlockSizeTest.__init__ s? ""4((( 1VE]++,,r c | j | j | j j }| |j | j j d S r4 )r* r9 r rp rM rl )r1 rS s r rX zBlockSizeTest.runTest sB 4;+?@@ *DK,BCCCCCr Nr r
r r r r s7 - - -
D D D D Dr r c $ e Zd ZdZd Zd Zd ZdS )
ByteArrayTestz;Verify we can use bytearray's for encrypting and decryptingc t j | || _ | }t |d | _ t t |d | _ t t |d | _ t t |d | _
t |dd | _ t |dd | _ t |dd | _
| j rt | j
| _
t |dd }t | | _ |qt! | j d |z | _ t |d
d | _ | j t |d d | _ | j t | j | _ n=d | _ t |d
d | _ | j t | j | _ || _ d S r r& r0 s r r) zByteArrayTest.__init__ ""4((( #FM::Xfe,,--8FK8899HV\ : :;;#FM4@@"6<>>FE400 ? #{{DH-- T Wt^<.) s, MMM9WQqTT]]33MMMr )r r r r r< rJ r rK r rL rM r! rN rO
r1 r r r rS rT rU rP rQ r! s
r rX zByteArrayTest.runTest# sv DN++ T_--
? NMMT_MMMJ99;; " "D MM$ OOD ! ! ! !
V^^Ii$8$899
:
:
X%%i
&;&;<<
=
= "--- ,,, 8 :&--//**C TXs + + + OOIgdh&7&7 8 8 9 9 9 9 9 : :r Nr
r r __doc__r) r< rX r
r r r r sH EE## ## ##J
:
:
:: : : : :r r c $ e Zd ZdZd Zd Zd ZdS )MemoryviewTestz;Verify we can use memoryviews for encrypting and decryptingc t j | || _ | }t |d | _ t t |d | _ t t |d | _ t t |d | _
t |dd | _ t |dd | _ t |dd | _
| j rt | j
| _
t |dd }t | | _ |qt! | j d |z | _ t |d
d | _ | j t |d d | _ | j t | j | _ n=d | _ t |d
d | _ | j t | j | _ || _ d S r r& r0 s r r) zMemoryviewTest.__init__B r r c | j }t | j }g }| j | j g}| j |t | j gz
} | j j |g|R i |S r4 r8 r: s r r< zMemoryviewTest._newg r= r c R t | j }t | j }g }| j rd | j D }| }| }|D ],}| | | | -t
| t | }t
| t | }|
| j | |
| j | | j rrt
| } |
| j | |
t t | j d S d S )Nc ` g | ]+}t t t | ,S r
)
memoryviewr r rD s r rG z*MemoryviewTest.runTest..y s, NNN:gaddmm44NNNr )r r r r r< rJ r rK r rL rM r! rN rO r s
r rX zMemoryviewTest.runTests sv DN++ T_--
? ONNdoNNNJ99;; " "D MM$ OOD ! ! ! !
V^^Jy$9$9::
;
;
X%%j&<&<==
>
> "--- ,,, 8 ;&--//**C TXs + + + OOJwtx'8'8 9 9 : : : : : ; ;r Nr r
r r r r ? sH EE## ## ##J
:
:
:; ; ; ; ;r r c
g }d}t t | D ]}|| }i }t | dk r|\ |d< |d< |d< nt | dk r|\ |d< |d< |d< |d< n]t | dk r*|\ |d< |d< |d< |d< } | | n t d t | fz d
|vrd |d
< | }
t |
d } t |
d } t |
d }
t |
d
}t |
dd }||}n |d k r |
s d | d
| }n d | d
| d|
}d||dz |fz }||d< ||d< | | |sR|t
| | t | | t | | t | | t | | gz
}d}| t | | |S )NFr^ r r r r Unsupported tuple size %dr" ECBp=, k=,
%s #%d: %s r T)
rI r` rJ AssertionErrorr+ r rf ru r r r ra r )r* r test_dataadditional_paramstestsextra_tests_addedrR rowr2 r/ p2p_key p_plaintext p_ciphertextp_mode
p_descriptionr r@ s r make_block_testsr su E
3y>>
"
" 75 75l s88q==IL FVK
& "6u
XX]]`c ]VK
& "6u
vmG\G\
XX]]nq kVK
& "6u
vmG\^j MM, ' ' ' ' !<C{!JKK K "F6N [[]]U## r;// L11 "f%% ]D99
$'KK
u__R__*5++uu=KK.9kk555""EK{AaC == $} +}
'((( ! % ff-- VV,, 00 ff-- ff--
E !% ^FF334444 Lr c g }d}t t | D ]}|| }i }t | dk r|\ |d< |d< |d< nt | dk r|\ |d< |d< |d< |d< n]t | dk r*|\ |d< |d< |d< |d< }| | n t d t | fz | } t | d }
t | d } t | d } t | dd }
|
|
}n| s d
| d |
}n d
| d |
d | }d
||dz |fz }||d< ||d< |s9|t
| | gz
}| t | | d}| t | | | t | | |S )NFr^ r r r r r r r r r r r r r T) rI r` rJ r r+ r r ra r r rZ )r* r r r r rR r r2 r/ r r r r r r r@ s r make_stream_testsr sK E
3y>>
"
" +> +>l s88q==IL FVK
& "6u
XX]]`c ]VK
& "6u
vmG\G\
XX]]nq kVK
& "6u
vmG\^j MM, ' ' ' ' !<C{!JKK K [[]]U## r;// L11 ]D99
$'KK F F*5++uu=KK F /:kk555""EK{AaC == $} +} ! % ff--
E
LL 77 8 8 8 $ ^FF33444
,VV<<==== Lr )r r' binasciir r r Crypto.Util.py3compatr Crypto.Util.strxorr r r r( r rZ rf ru r r r r dictr r r
r r r sS 2 - , . . . . . . . . . . # # # # # # ' ' ' ' ' ' % a/ a/ a/ a/ a/X& a/ a/ a/F . . . . .n . . .@> > > > >H% > > >0- - - - -8$ - - -"I I I I Ix( I I ID D D D DH% D D DM: M: M: M: M:H% M: M: M:`M; M; M; M; M;X& M; M; M;` HLtvv < < < <|/ / / / /r