#P b:!> , d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZ d dl m
Z
d dl m Z d dl
mZmZ G d dej Z G d d
e Z G d d e Z G d
dej Z G d de Z G d de Z G d dej Z G d de Z G d de Z G d dej Z G d d e Z G d d e Z i fd Z e d k rd dlZd! Z ej d"# dS dS )$ N) unhexlifyhexlify)tobytes)strxor_c)list_test_cases)BLAKE2bBLAKE2sc \ e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
ZdS )
Blake2Testc | j | j }| j j |j fD ]}t d| j dz d D ],} || }| |j |dz -t d| j dz D ])} || }| |j | * |d| j } || j d }| || |dd| j | }| |j | j d S )
N digest_bits digest_bytes )datar A 5)r keyr ) BLAKE2newmax_bitsrange assertEqual digest_size max_bytesdigestupdate)selfhnew_funcdbitshobjdbytesdigest1digest2s /builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/Crypto/SelfTest/Hash/test_BLAKE2.pytest_new_positivez Blake2Test.test_new_positive. s KOO
O 6 6 . G GH q$-!"3Q77
?
? xE222 !15A:>>>> 4>A#566
;
; xV444 !16:::: hG$.IIIPPRRG hDN;;;BB7KKRRTTG Wg . . . H$Dt~ F F F F Fuuww )4>::::: c | j | j }| j j |j fD ]}| t || j | j | t |d | t || j dz | t |d | t |d | t || j dz | t || j d | t || j d
d S ) Nr
)r r r r r stringr r )r r )r r r assertRaises TypeErrorr
ValueError)r! r" r# s r) test_new_negativez Blake2Test.test_new_negativeD sj KOO
O 6 6 . . .H i+/>*.-
9
9
9
j( C C C j(+/>A+=
?
?
? j( B B B j( C C C j(*.-!*;
=
=
= i+/>"+
-
-
-
i+/>#,
.
.
.
. . .r+ c | j d }| t | | j d S )Ns abcr )r r r r lenr )r! r s r) test_default_digest_sizez#Blake2Test.test_default_digest_sizeY sG f --4466 Vdn55555r+ c ddg}| j | j }| |d |d | }| j | j }| |d |d z | | | d S )Ns
s, r r r )r r r r r r )r! piecesr" r s r) test_updatezBlake2Test.test_update] s / KOO O 8 8 ""6!9--- KOO O 8 8 VAY&''' V,,,,,r+ c | j | j }| t |j d d S )Nr r/ )r r r r1 r2 r )r! r" s r) test_update_negativez Blake2Test.test_update_negativef s9 KOO O 8 8 )QXy99999r+ c | j | j }| }| | | | t
|t d d S )Nr s digest)r r r r r
assertTrue
isinstancetype)r! r" r s r) test_digestzBlake2Test.test_digestj sn KOO O 8 8
V,,,
64 ??;;<<<<>>>>r+ c | j | j d }| }| | t |d }| t |j | d S )N 4r0 )r r r r verifyr r1 r3 )r! r" rM wrong_macs r) test_verifyzBlake2Test.test_verify sf KOO TO B Bhhjj
S#&& *ah :::::r+ c | j | j d }| }| | | t |j d d S )NrP r0 4556)r r r rL hexverifyr1 r3 )r! r" rM s r) test_hexverifyzBlake2Test.test_hexverify sZ KOO TO B Bkkmm C *ak6:::::r+ c d| j z dz }| j D ]}| j | | j |t
|dz z | j |d | t fd d| j fD ]*}|| j v r | t fd +d S ) Nz1.3.6.1.4.1.1722.12.2..r
r s secret)r r c j S Noidr" s r) z%Blake2Test.test_oid.. ae r+ c j S r\ r] r_ s r) r` z%Blake2Test.test_oid.. ra r+ )
oid_variantdigest_bits_oidr r r r^ strr1 AttributeErrorr )r! prefixr r" s @r) test_oidzBlake2Test.test_oid s )D,<.get_mv_ro s d## #r+ c : t t | S r\ )rv rn r6 s r) get_mv_rwz-Blake2Test.test_memoryview..get_mv_rw s ioo.. .r+ rl rm r )r r readonlyr r r )
r! r r rw ry get_mvkey_mvdata_mvrq rr s
r) test_memoryviewzBlake2Test.test_memoryview sr $ $ $ / / / !), 7 7F VC[[F fTllG d 44B g6 ::B#
% % $rr
RYY[["))++ 6 6 6 fTllG ""B ""B IIdOOO IIg #
& % RYY[["))++ 6 6 6 61 7 7r+ N)__name__
__module__ __qualname__r* r4 r8 r; r= rB rJ rN rT rX rh rs r~ r+ r) r r , s ; ; ;,. . .*6 6 6- - -: : := = =+ + +"
?
?
?; ; ;; ; ; = = = 3 3 36#7 #7 #7 #7 #7r+ r c " e Zd ZeZdZdZdZdZdS ) Blake2bTest @ ) rD i r 1N) r r r r r r r rd rc r r+ r) r r '
FHI*OKKKr+ r c " e Zd ZeZdZdZdZdZdS ) Blake2sTestrD ) r rD 2N) r r r r r r r rd rc r r+ r) r r r r+ r c e Zd Zd Zd Zd ZdS )Blake2OfficialTestVectorc 0 d}g }t |d 5 }t | D ]\ }}| dk s| d r3t j d|z | }|st
d|z | d r0t t | d }nd}|dk r|} d }|d k r|}
d
}|} d}|
| |
| f d d d n # 1 swxY w Y |S ) Ninrt #z%s: ([0-9A-Fa-f]*)&Incorrect test vector format (line %d)r r+ r hash) open enumeratestrip
startswithrematchr3 groupr r append) r! test_vector_fileexpected test_vectorstest_vector_fd line_numberlineres bin_value
input_datar results r) _load_testsz$Blake2OfficialTestVector._load_tests
s
"D
)
) C^%.~%>%>
C
C! T::<<2%% )=)=% h 4x ?FF 4$%M'2&3 4 4 4 99Q<< $ )'#))A,,*?*? @ @II #It##!*J $HH &&#C %HH &F #H ''S&(ABBBB/
C C C C C C C C C C C C C C C C2 s C)D DDc d| j f}| j dz }d| j z | _ dd l}n:# t $ r- t j d| j z t g | _ Y d S w xY wt j
|j }t j
t j
j
|g|R | }| | | _ d S )NHashz -test.txt%s testsr 'Warning: skipping extended tests for %s)namelower descriptionpycryptodome_test_vectors ImportErrorwarningswarn UserWarningr ospathdirname__file__joinr r! dir_comps file_namer init_dirfull_file_names r) setUpz Blake2OfficialTestVector.setUp( s TY' IOO%% 3 % 1 , , , , , MCdiO&
(
(
( "D FF 7??#<#EFF bgl8&Hi&H&H&H)TT , ,^ < <s ; 3A21A2c | j D ]d\ }}}| j || j }| | | | | ed S )N)r r )r r r r r r r )r! r r r rM s r) runTestz Blake2OfficialTestVector.runTest: sq )-): 3 3 %Zf+//c/GGC JJz " " " SZZ\\6 2 2 2 2 3 3r+ Nr r r r r r r r+ r) r r sA <= = =$3 3 3 3 3r+ r c e Zd ZeZdZdZdS )Blake2bOfficialTestVectorr r Nr r r r r r r r r+ r) r r A
F DIIIr+ r c e Zd ZeZdZdZdS )Blake2sOfficialTestVectorr r Nr r r r r r r r r+ r) r r J r r+ r c e Zd Zd Zd Zd ZdS )Blake2TestVector1c g }t |d 5 }t | D ]\ }}| dk s| d r3t j d| }|st
d|z | t t |
d d d d n # 1 swxY w Y |S )Nr r r zdigest: ([0-9A-Fa-f]*)r r ) r r r r r r r3 r r r r )r! r r r r r r s r) r z Blake2TestVector1._load_testsU s7
"D
)
) F^%.~%>%>
F
F! T::<<2%% )=)=% h 7>> 4$%M'2&3 4 4 4 ##Igciill.C.C$D$DEEEE
F F F F F F F F F F F F F F F F s B/CCCc d| j f}d}d| j z | _ dd l}n:# t $ r- t j d| j z t g | _ Y d S w xY wt j
|j }t j t j j |g|R | }|
| | _ d S )Nr ztv1.txtr r r r r r r r r r r r r r r r r r s r) r zBlake2TestVector1.setUpc TY' % 1 , , , , , MCdiO&
(
(
( "D FF 7??#<#EFF bgl8&Hi&H&H&H)TT , ,^ < < ! 3AAc > | j D ]}t | }d}t d D ]I}| j | }| | | |z }J| | | d S )Nr+ d r )r r7 r r r r r r )r! tvr next_data_r" s r) r zBlake2TestVector1.runTestt s # - -B r77LI3ZZ
3
3KOO O>>### HHJJ 2 QXXZZ , , , , - -r+ Nr r r+ r) r r S sA = = =" - - - - -r+ r c e Zd ZeZdZdS )Blake2bTestVector1r N)r r r r r r r r+ r) r r
F DDDr+ r c e Zd ZeZdZdS )Blake2sTestVector1r N)r r r r r r r r+ r) r r r r+ r c e Zd Zd Zd Zd ZdS )Blake2TestVector2c g }t |d 5 }t | D ]\ }}| dk s| d r3t j d| }|st
d|z t | d }t t | d }| ||f d d d n # 1 swxY w Y |S )Nr r r z"digest\(([0-9]+)\): ([0-9A-Fa-f]*)r r ) r r r r r r r3 intr r r r ) r! r r r r r r key_sizer s r) r z Blake2TestVector2._load_tests sC
"D
)
)
8^%.~%>%>
8
8! T::<<2%% )=)=% h DdKK 4$%M'2&3 4 4 4 syy||,,"7399Q<<#8#899 ##Xv$67777
8
8
8
8
8
8
8
8
8
8
8
8
8
8
8
8 s CC55C9<C9c d| j f}d}d| j z | _ dd l}n:# t $ r- t j d| j z t g | _ Y d S w xY wt j
|j }t j t j j |g|R | }|
| | _ d S )Nr ztv2.txtr r r r r s r) r zBlake2TestVector2.setUp r r c 8 | j D ]\ }}d}t d D ]R}| j | j d|z }| | | |z }S| | | d S )Nr+ r r r0 )r r r r r r r r )r! r r r r r" s r) r zBlake2TestVector2.runTest s $ 1 1 1 HfI3ZZ
3
3KOO (,x $ 9 9### HHJJ 2 QXXZZ 0 0 0 0 1 1r+ Nr r r+ r) r r sA
= = =" 1 1 1 1 1r+ r c e Zd ZeZdZdZdS )Blake2bTestVector2r r Nr r r+ r) r r r r+ r c e Zd ZeZdZdZdS )Blake2sTestVector2r r Nr r r+ r) r r r r+ r c g }|t t z
}| t | t | t |t t z
}| t | t | t |S r\ )
r r r r r r r r r r )configtestss r) get_testsr s E _[
)
))E LL*,,--- LL#%%&&& LL#%%&&& _[
)
))E LL*,,--- LL#%%&&& LL#%%&&& Lr+ __main__c B t j t S r\ )unittest TestSuiter r r+ r) suiter s !)++...r+ r ) defaultTest)!r r r r binasciir r Crypto.Util.py3compatr Crypto.Util.strxorr Crypto.SelfTest.st_commonr Crypto.Hashr r TestCaser r r r r r r r r r r r r r r mainr r+ r) r s >
' ' ' ' ' ' ' ' ) ) ) ) ) ) ' ' ' ' ' ' 5 5 5 5 5 5 ( ( ( ( ( ( ( (7 7 7 7 7" 7 7 7D
*
*
63 63 63 63 63x 0 63 63 63r 8 8 *- *- *- *- *-) *- *- *-Z * * +1 +1 +1 +1 +1) +1 +1 +1\ * *
zOOO/ / /HMg&&&&&& r+