XJ$ P d dl Z d dlmZ d dlmZmZmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d Z G d
d e j Z G d d
e j Zd Z G d de j Z G d de j Zi fdZedk rd Z e j d dS dS ) N) unhexlify)btobytesbchr)
long_to_bytes)load_test_vectors)list_test_cases)AES)SHAKE128c 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_OCB.pyget_tag_randomr + s) s cl+++ s|T]CC
^^DI
&
& s|4=III V^^DI 6 677777r c v | t t j | j t j d d S )N test12345678r ) assertRaises TypeErrorr
r r r r% s r test_nonce_must_be_bytesz!OcbTests.test_nonce_must_be_bytesH s: )SWdlCL / 1 1 1 1 1r c | t t j | j t j t
d t dd D ]5}t j | j t j | j d | 6| t t j | j t j | j d S )N r r ) r3
ValueErrorr
r r r r ranger
)r% r s r test_nonce_lengthzOcbTests.test_nonce_lengthL s *cgt|S\ !" ' ' ' Arll J JF GDL#,di6H I I I I I *cgt|S\ $ + + + + +r c t j | j t j | j }| |j t j t j | j t j j }t j | j t j j }| t | d | || d S Nr )
r
r r r r! r$
block_sizer lenassertNotEqualr% r& nonce1nonce2s r test_block_size_128z OcbTests.test_block_size_128X s s|4=III *CN;;; s|44: s|44: Vb))) FF+++++r c t j | j t j | j }| |j | j t j | j t j j }t j | j t j j }| t | d | || d S r> ) r
r r r r! r$ r rA rB rC s r test_nonce_attributez OcbTests.test_nonce_attributeb s s|4=III t}555 s|44: s|44: Vb))) FF+++++r c T | t t j | j t j | j d | t t j | j t j | j d t j | j t j | j d d S )N )r unknownF)r use_aesni)r3 r4 r
r r r r! r5 s r test_unknown_parametersz OcbTests.test_unknown_parametersl s )SWdlCL - , , , )SWdlCL $
q : : :
cl$- ! ! ! ! ! !r c dD ]v}t j | j t j | j } t || t
d }| |t
d wd S )Nr. decryptr r8 )r
r r r r! getattrr r$ )r% funcr& results r test_null_encryption_decryptionz(OcbTests.test_null_encryption_decryptionw su ( , ,DWT\3>> s|4=IIIqxx )V^QuXX>>>>>r c 6 t j | j t j | j }| t |j d t j | j t j | j }| t |j d d S )Nr ztest1234567890-*) r
r r r r! r3 r4 r. rP rW s r test_data_must_be_bytesz OcbTests.test_data_must_be_bytes sv s|4=III )V^5HIII s|4=III )V^5HIIIIIr c | t t j | j t j | j d | t t j | j t j | j d t dd D ]n}t j | j t j | j | }| | j \ }}|
t | | ot j | j t j | j }| | j \ }}|
t | d d S )NrJ r mac_len r r ) r3 r: r
r r r r! r; r" r
r$ rA )r% r] r& _r) s r test_mac_lenzOcbTests.test_mac_len s7 *cgt|S\ $
q : : : *cgt|S\ $
t = = = Q'' 0 0GWT\3zBOcbTests.test_message_chunks..break_up.. s3 # # #qD1\>)* # # #r r )r; rA )r
rt s ``r break_upz.OcbTests.test_message_chunks..break_up sJ # # # # #E!SYY 5" 5" # # #
#r )
r9 rJ
r ( P r r8 )r r
r r r r! updater" r rP r$ verifyr. ri ) r% auth_datar r&
ciphertextref_macrv rt chunkr* ct2s r test_message_chunksz OcbTests.test_message_chunks sD ##7== ";44 s|4=III
i $ 7 7 B B
G # # #
@ # #LWT\3>## #C Y , , , MM' " " " " @ 7 7LWT\3>## #C Z - - - V]]__g 6 6 6 6 7 7r c t | j }t | j }t | j }t | j }t j | j t j | j }| | j | | j | z }| }t j |t j | }d|d d< d|d d< | | d|d d< | | | z } d|d d< | }
|
|| |
||
|
|j |j t | j }t | j }t | j }~t j |t j | } d|d d< d|d d< | | d|d d< | t | t |
} |
| j | d S Nr s rx )
bytearrayr r! r
r
r r r} r. ri r$ r r# )
r% key_banonce_ba header_badata_bacipher1r( r cipher2ct_testtag_testcipher4pt_tests
r test_bytearrayzOcbTests.test_bytearray sX 4<(( T]++ di(( DI&&'$, , $
/ / / ty!!!
__TY
'
''//*;*;
;nn '& , (* * * %rr
&! y!!!' "1"
//'**W__->->>% >>## W%%% h'''
666 4<(( T]++ di(( '& , (* * * %rr
&! y!!!' "1"
,,Yw-?-?8ATATUU G,,,,,r c t t | j }t t | j }t t | j }t t | j }t j | j t
j | j }| | j | | j | z }|
}t j |t
j | }d|d d< d|d d< | | d|d d< | | | z } d|d d< |
}
| || | ||
| |j |j t t | j }t t | j }t t | j }~t j |t
j | } d|d d< d|d d< | | d|d d< |
t | t |
} | | j | d S r )
memoryviewr r r! r
r
r r r} r. ri r$ r r# )
r% key_mvnonce_mv header_mvdata_mvr r( r r r r r r s
r test_memoryviewzOcbTests.test_memoryview
s Idl 3 344 i
6 677 y 3 344 Yty 1 122'$, , $
/ / / ty!!!
__TY
'
''//*;*;
;nn '& , (* * * %rr
&! y!!!' "1"
//'**W__->->>% >>## W%%% h'''
666 Idl 3 344 i
6 677 y 3 344 '& , (* * * %rr
&! y!!!' "1"
,,Z-@-@*XBVBVWW G,,,,,r N)__name__
__module__ __qualname__r r r! r
r+ r0 r6 r< rF rH rM rT rX rZ ra rf rl r r r rq r r r r / s< nY++G ~k2..H >&# & &D" " "8 8 81 1 1
+
+
+, , ,, , , ! ! !, , , ? ? ?J J J' ' '& ' ' '" " ")7 )7 )7V-- -- --^-- -- -- -- --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 Zd Zd Zd ZdS ) OcbFSMTestsr r r r r
r c t j | j t j | j }| | j }|| z
}| }t j | j t j | j }| | | | | d S r- )
r
r r r r! r. r
ri rP r~ )r% r& r( r) s r -test_valid_init_encrypt_decrypt_digest_verifyz9OcbFSMTests.test_valid_init_encrypt_decrypt_digest_verifyC s s| #}. . .
^^DI
&
&
fnn mmoo s| #}. . .r
cr c t j | j t j | j }| | j }| t |j t j | j t j | j }|
| | t |j d S r- ) r
r r r r! r. r
r3 r4 ri rP r~ r/ s r /test_invalid_init_encrypt_decrypt_digest_verifyz;OcbFSMTests.test_invalid_init_encrypt_decrypt_digest_verifyS s s| #}. . .
^^DI
&
& )V]333 s| #}. . .r )V]33333r c l t j | j t j | j }| | j | }t j | j t j | j }| | j | | d S r- ) r
r r r r! r} r
ri r~ r% r& r) s r $test_valid_init_update_digest_verifyz0OcbFSMTests.test_valid_init_update_digest_verifya s s| #}. . .
di mmoo s| #}. . .
di
cr c t j | j t j | j }| | j | | j }|| z
}| }t j | j t j | j }| | j | | | |
| t j | j t j | j }| | j | | j d d }| | j dd \ }}t j | j t j | j }| | j | | | || d S )Nr rw )
r
r r r r! r} r
r. ri rP r~ r" r# )r% r& r( r) ct1r s r test_valid_full_pathz OcbFSMTests.test_valid_full_patho s s| #}. . .
di
^^DI
&
&
fnn mmoo s| #}. . .
di r
c s| #}. . .
di nnTYrr]++,,TYqrr];;S s| #}. . .
di s!!#s+++++r c 8 t j | j t j | j }| | j | | j | | t |j | j d S r- )
r
r r r r! r} r
r. r3 r4 rW s r test_invalid_encrypt_after_finalz,OcbFSMTests.test_invalid_encrypt_after_final s| #}. . .
di ty!!! )V^TY?????r c 8 t j | j t j | j }| | j | | j | | t |j | j d S r- )
r
r r r r! r} r
rP r3 r4 rW s r test_invalid_decrypt_after_finalz,OcbFSMTests.test_invalid_decrypt_after_final r r c t j | j t j | j }| d S r- )r
r r r r! ri rW s r test_valid_init_digestz"OcbFSMTests.test_valid_init_digest s/ s|4=III
r c t j | j t j | j }| }t j | j t j | j }| | d S r- )r
r r r r! ri r~ r s r test_valid_init_verifyz"OcbFSMTests.test_valid_init_verify s_ s|4=IIImmoo s|4=III
cr c dD ]}d t d | j | j t d z fD ]}t j | j t j | j }|| | t || } || j || j || j || j | d S )NrO 3333r ) r r
r
r r r r! r} rQ )r% method_namer r& methods r &test_valid_multiple_encrypt_or_decryptz2OcbFSMTests.test_valid_multiple_encrypt_or_decrypt s / K "AeHHdi "i!C&& 0 2
s|'+}6 6 6 (MM),,, 55ty!!!ty!!!ty!!!ty!!!
r c t j | j t j | j }| | j | }t d D ]*}| || +t j | j t j | j }| | j t d D ]}|
| d S )Nr ) r
r r r r! r} r
ri r; r$ r~ )r% r& first_macxs r $test_valid_multiple_digest_or_verifyz0OcbFSMTests.test_valid_multiple_digest_or_verify s s|4=III
di MMOO q 9 9A Y
8 8 8 8 s|4=III
di q % %A MM) $ $ $ $ % %r c t j | j t j | j }| | j | | j \ }}t j | j t j | j }| | j | || }| | j | d S r- )
r
r r r r! r} r
r" r# r$ )r% r& r( r) r' s r 0test_valid_encrypt_and_digest_decrypt_and_verifyz&# & &D 4 4 4 , , ,@@ @ @@ @ @
% % %
(
(
( - - -
1
1
1) ) ) ) )r r c t d | dz dz
z t | z }d}t d D ]}t d |z }t d|z dz |dz }t j |t j ||dz }| | || | | z | z z
}t d|z dz |dz }t j |t j ||dz }|| | | z | z z
}t d|z dz |dz }t j |t j ||dz }| | || | z z
}t d |dz }t j |t j ||dz }| | | | z S )
z.Implement the algorithm at page 18 of RFC 7253r r_ r9 r r rx r\ rw i ) r r; r r
r r r} r. ri ) keylentaglennoncelenkeyCrs SNr& s r algo_rfc7253r s q''Vq[1_
%V
4C A
3ZZ 0 0 GGaK !a%!)X] 3 3cl!Vq[III
a V^^A
!1!1
1FMMOO
CC !a%!)X] 3 3cl!Vq[III V^^A
!1!1
1FMMOO
CC !a%!)X] 3 3cl!Vq[III
a V^^
//c8q=))A
WS#,a1
E
E
EF
MM! >> fmmoo --r c 0 e Zd ZdZdZdZdZd Zd Zd Z dS ) OcbRfc7253Test 000102030405060708090A0B0C0D0E0F))BBAA99887766554433221100r8 r8 785407BFFFC8AD9EDCC5520AC9111EE6)BBAA998877665544332211010001020304050607r 06820B3657B6F615A5725BDA0D3B4EB3A257C9AF1F8F03009)BBAA99887766554433221102r r8 81017F8203F081277152FADE694A0A00)BBAA99887766554433221103r8 r 045DD69F8F5AAE72414054CD1F35D82760B2CD00D2F99BFA9)BBAA99887766554433221104r r @571D535B60B277188BE5147170A9A22C3AD7A4FF3835B8C5701C1CCEC8FC3358)BBAA99887766554433221105r r8 8CF761B6902EF764462AD86498CA6B97)BBAA99887766554433221106r8 r @5CE88EC2E0692706A915C00AEB8B2396F40E1C743F52436BDF06D8FA1ECA343D)BBAA998877665544332211070000102030405060708090A0B0C0D0E0F1011121314151617r P1CA2207308C87C010756104D8840CE1952F09673A448A122C92C62241051F57356D7F3C90BB0E07F)BBAA99887766554433221108r r8 6DC225A071FC1B9F7C69F93B0F1E10DE)BBAA99887766554433221109r8 r P221BD0DE7FA6FE993ECCD769460A0AF2D6CDED0C395B1C3CE725F32494B9F914D85C0B1EB38357FF)BBAA9988776655443322110A@000102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1Fr `BD6F6C496201C69296C11EFD138A467ABD3C707924B964DEAFFC40319AF5A48540FBBA186C5553C68AD9F592A79A4240)BBAA9988776655443322110Br r8 FE80690BEE8A485D11F32965BC9D2A32)BBAA9988776655443322110Cr8 r `2942BFC773BDA23CABC6ACFD9BFD5835BD300F0973792EF46040C53F1432BCDFB5E1DDE3BC18A5F840B52E653444D5DF)BBAA9988776655443322110DP000102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1F2021222324252627r pD5CA91748410C1751FF8A2F618255B68A0A12E093FF454606E59F9C1D0DDC54B65E8628E568BAD7AED07BA06A4A69483A7035490C5769E60)BBAA9988776655443322110Er r8 C5CD9D1850C141E358649994EE701B68)BBAA9988776655443322110Fr8 r p4412923493C57D5DE0D700F753CCE0D1D2D95060122E9F15A5DDBFC5787E50B5CC55EE507BCB084E479AD363AC366B95A98CA5F3000B1479) 0F0E0D0C0B0A09080706050403020100r r r h1792A4E31E0755FB03E31B22116E6C2DDF9EFD6E33D536F1A0124B0A55BAE884ED93481529C76B6AD0C515F4D1CDD4FDAC4F02AA) )r r 67E944D23256C5E0B6C61FA22FDF1EA2) r F673F2C3E7174AAE7BAE986CA9F29E17) r D90EB8E9C977C88B79DD793D7FFA161C)r ` 77A3D8E73589158D25D01209)r r 05D56EAD2752C86BE6932C5E)r r 5458359AC23B0CBA9E6330DD)r @ 192C9B7BD90BA06A)r r 0066BC6E0EF34E24)r r 7D4EA5D445501CBEc t t | j }| j D ]Q}d |D \ }}}}|d d |dd }}t j |t j | }| | | | | z } | || | ||
t j |t j | }| | | | | z }
| ||
| | Sd S )Nc F g | ] }t t | S rq r r rr r s r ru z(OcbRfc7253Test.test1.. s$ !>!>!>a)AaDD//!>!>!>r ir )
r r tv1_keytv1r
r r r} r. r$ ri rP r~ ) r% r tvr aadr' r( mac_tagr& r r* s r test1zOcbRfc7253Test.test1 sZ $,((( # #B!>!>2!>!>!> E3B TcT(BsttHBWS#,e<<. s$ "E"E"Eq9QqTT??"E"E"Er ir r\ )
tv2r
r r r} r. r$ ri rP r~ )
r% r r r r' r( r r& r r* s
r test2zOcbRfc7253Test.test2 s5 "E"EDH"E"E"E UCR#h344Gcl%DDD
cnnR 6>>#3#33 S!!! &--//222cl%DDD
cnnR 6>>#3#33 S!!!
g r c | j D ]G\ }}}t ||d }| t t | | Hd S )Nr )tv3r r$ r r )r% r r rS result2s r test3zOcbRfc7253Test.test3 s\ &*h < < "FFF"66266G Yqyy 1 17 ; ; ; ; < .check s6 !&&(;;F VYs^^ 4 4 4 4 4r r r C47F5F0341E15326D4D1C46F47F05062r 95B9167A38EB80495DFC561A8486E109r AFE1CDDB97028FD92F8FB3C8CFBA7D83r F471B4983BA80946DF217A545AE828BC51C24D85FA5CC7B28C8335982E2B734616CAD14Cr B553F74B85FD1E5B3B49D20E513531F9ED6DA5B1216BF8BBr CA8AFCA031BAC3F480A583BD6C50A547 D170C1DF356308079DA9A3F619147148 57F94381F2F9231EFB04AECD323757C33A618B2531ED39F260C750DC9071EB89FEDBADDA88FD286EFDF0EFB97F21A39AC4BAB5ACFAB2FF3A8DD82A13AC01D912BD0737D39D1FD0B500EA4ECFr 9E043A7140A25FB91F43BCC9DD7E0F46 680000E53908323A7F396B955B8EC641 8304B97FAACDA56E676602E1878A7E6F81F978AC9867E825D339847DEFCF2D60B24926ADA48CF5B184961DC56E917B165E58C174227AEE6C9D905A61541DE691B9E1A2F9B0E761381C7129FCrq )r% r2 s ` r test_3zOcbDkgTest.test_3 s 5 5 5 5 5
c3 ?@@@
c3 ?@@@
c3 ?@@@
c2s 6777
c2s 6777
c2s 6777
c2s .///
c2s .///
c2s ./// c3 ?@@@
c3 ?@@@
c3 ?@@@
c2s 6777
c2s 6777
c2s 6777
c2s .///
c2s .///
c2s ./// c3 ?@@@
c3 ?@@@
c3 ?@@@
c2s 6777
c2s 6777
c2s 6777
c2s .///
c2s .///
c2s ./////r c t d }t d }t d }t d }t d }t | t | z
}t d }t j |t j |d d | }| | | | \ } }
| || |
z d S )N EEDDCCBBAA9988776655443322110Dr r h07E903BFC49552411ABC865F5ECE60F6FAD1F5A9F14D3070FA2F1308A563207FFE14C1EEA44B22059C7484319D8A2C53C236A7B3hBA015C4E5AE54D76C890AE81BD40DC5703EDC30E8AC2A58BC5D8FA4D61C5BAE6C39BEAC435B2FD56A2A5085C1B135D770C8264B7r\ )r rA r
r r r} r" r$ ) r% r r APr r] buggy_resultr& C_out2tag_out2s r
test_2_bugfixzOcbDkgTest.test_2_bugfix( s :;;:;; 9
:
: 9
:
: !
"
" a&&3q66/
! ", - - cl%*gNNN
a!44Q77 v'899999r N)r r r __doc__r- rN rY rq r r r r sH CC1 1 1.%0 %0 %0N: : : : :r r c g }|t t z
}|t t z
}|t t z
}|t t z
}|S r0 )r r r r r )configtestss r get_testsr^ A sT E _X
&
&&E _[
)
))E _^
,
,,E _Z
(
((E Lr __main__c B t j t S r0 )unittest TestSuiter^ rq r r suiterc K s !)++...r rc ) defaultTest) ra binasciir Crypto.Util.py3compatr r r Crypto.Util.numberr Crypto.SelfTest.loaderr Crypto.SelfTest.st_commonr
Crypto.Cipherr
Crypto.Hashr r TestCaser r r r r r^ r rc mainrq r r rn s > 2 2 2 2 2 2 2 2 2 2 , , , , , , 4 4 4 4 4 4 5 5 5 5 5 5 8 8 8K- K- K- K- K-x K- K- K-\B) B) B) B) B)(# B) B) B)J. . .:E< E< E< E< E