*^@pl Z d dl Z d dlmZmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z d dl
mZmZmZmZmZ d dlmZ d d lmZ d d
lmZ d d lmZ d Z G d
de Z G d de j Z G d de j Z G d de j Z e dddd d d pg Z! e"e! D ].\ Z#Z$ e%e$e& r e'e$d re$j( Z)' e'e$d r4 ee$j* + Z,e,- e$j. Z/ ej0 d e)e$j1 fD Z2e$j3 ed k r2 e e$j3 Z4 ej- e2 e5e$j3 e4 Z6n ej- e2d Z6e/e6e$j7 fd!Z8e/e6e$j7 fd"Z9e$j: d k r e;e d#e#z e8 e;e d$e#z e9 0 G d% d&e j Z< G d' d(e j Z= e dd)d*d+d, i pg Z>e> e dd-d.d+d/ i pg z
Z> e"e> D ] \ Z#Z$ e%e$e& r e'e$d re$j( Z)' e'e$d0 r( ej0 d1 e)e$j1 e$j? fD Z@[ ee$j* + Z,e,- e$j. Z/e$j3 ed k r2 e e$j3 Z4 ej- e@ e5e$j3 e4 ZAn ej- e@d ZAe/eAe$j7 fd2ZB e;e=d3e#z eB
G d4 d5e j ZC G d6 d7e j ZDd8 ZE G d9 d:e j ZFi fd;ZGeHd? dS dS )@ N)bbchr
bytes_to_long)strxor)list_test_cases)load_test_vectors load_test_vectors_wycheproof)SHA1SHA224SHA256SHA384SHA512)RSA)pss) PKCS1_PSSMGF1c ^ t d| z t t dg S )Nz Crypto.Hash.new)
__import__globalslocals) hash_names /builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/Crypto/SelfTest/Signature/test_pss.pyload_hash_by_namer / s% ny0'))VXXw O OO c e Zd Zd Zd ZdS )PRNGc " || _ d| _ d S )Nr streamidx)selfr" s r __init__z
PRNG.__init__5 s r c ^ | j | j | j |z }| xj |z
c_ |S Nr! )r$ rnd_sizeresults r __call__z
PRNG.__call__9 s1 TXdh&9 9: H
r N)__name__
__module__ __qualname__r% r* r r r r 3 s2 r r c , e Zd ZdZdZdZd Zd Zd ZdS ) PSS_Testss -----BEGIN RSA PRIVATE KEY-----
MIIEowIBAAKCAQEAsvI34FgiTK8+txBvmooNGpNwk23YTU51dwNZi5yha3W4lA/Q
vcZrDalkmD7ekWQwnduxVKa6pRSI13KBgeUOIqJoGXSWhntEtY3FEwvWOHW5AE7Q
jUzTzCiYT6TVaCcpa/7YLai+p6ai2g5f5Zfh4jSawa9uYeuggFygQq4IVW796MgV
yqxYMM/arEj+/sKz3Viua9Rp9fFosertCYCX4DUTgW0mX9bwEnEOgjSI3pLOPXz1
8vx+DRZS5wMCmwCUa0sKonLn3cAUPq+sGix7+eo7T0Z12MU8ud7IYVX/75r3cXiF
PaYE2q8Le0kgOApIXbb+x74x0rNgyIh1yGygkwIDAQABAoIBABz4t1A0pLT6qHI2
EIOaNz3mwhK0dZEqkz0GB1Dhtoax5ATgvKCFB98J3lYB08IBURe1snOsnMpOVUtg
aBRSM+QqnCUG6bnzKjAkuFP5liDE+oNQv1YpKp9CsUovuzdmI8Au3ewihl+ZTIN2
UVNYMEOR1b5m+z2SSwWNOYsiJwpBrT7zkpdlDyjat7FiiPhMMIMXjhQFVxURMIcB
jUBtPzGvV/PG90cVDWi1wRGeeP1dDqti/jsnvykQ15KW1MqGrpeNKRmDdTy/Ucl1
WIoYklKw3U456lgZ/rDTDB818+Tlnk35z4yF7d5ANPM8CKfqOPcnO1BCKVFzf4eq
54wvUtkCgYEA1Zv2lp06l7rXMsvNtyYQjbFChezRDRnPwZmN4NCdRtTgGG1G0Ryd
Yz6WWoPGqZp0b4LAaaHd3W2GTcpXF8WXMKfMX1W+tMAxMozfsXRKMcHoypwuS5wT
fJRXJCG4pvd57AB0iVUEJW2we+uGKU5Zxcx//id2nXGCpoRyViIplQsCgYEA1nVC
eHupHChht0Fh4N09cGqZHZzuwXjOUMzR3Vsfz+4WzVS3NvIgN4g5YgmQFOeKwo5y
iRq5yvubcNdFvf85eHWClg0zPAyxJCVUWigCrrOanGEhJo6re4idJvNVzu4Ucg0v
6B3SJ1HsCda+ZSNz24bSyqRep8A+RoAaoVSFx5kCgYEAn3RvXPs9s+obnqWYiPF3
Re5etE6Vt2vfNKwFxx6zaR6bsmBQjuUHcABWiHb6I71S0bMPI0tbrWGG8ibrYKl1
NTLtUvVVCOS3VP7oNTWT9RTFTAnOXU7DFSo+6o/poWn3r36ff6zhDXeWWMr2OXtt
dEQ1/2lCGEGVv+v61eVmmQUCgYABFHITPTwqwiFL1O5zPWnzyPWgaovhOYSAb6eW
38CXQXGn8wdBJZL39J2lWrr4//l45VK6UgIhfYbY2JynSkO10ZGow8RARygVMILu
OUlaK9lZdDvAf/NpGdUAvzTtZ9F+iYZ2OsA2JnlzyzsGM1l//3vMPWukmJk3ral0
qoJJ8QKBgGRG3eVHnIegBbFVuMDp2NTcfuSuDVUQ1fGAwtPiFa8u81IodJnMk2pq
iXu2+0ytNA/M+SVrAnE2AgIzcaJbtr0p2srkuVM7KMWnG1vWFNjtXN8fAhf/joOv
D+NmPL/N4uE57e40tbiU/H7KdyZaDt+5QiTmdhuyAe6CBjKsF2jy
-----END RSA PRIVATE KEY-----s AAAs [c5ذ!Dݹ`P>X5ݎʼnP+ɯPL5k{=_R}b5Y~}l1-Vlk/o5Q$ = ABPG"~jqkbsbm(`F.)/_Sv/DT:s>>>>r c t j | j }t j | j }t
j |d }t | j }| t |j || d S )Ni
salt_bytesr= rA s r test_negative_2zPSS_Tests.test_negative_2R si nT\** Jtx 734000!! *hoq#>>>>>r N) r+ r, r- r3 r4 r6 r: rB rF r. r r r0 r0 ? sV vG
C v C% % % ? ? ?? ? ? ? ?r r0 c & e Zd Zd Zd Zd Zd ZdS )FIPS_PKCS1_Verify_Testsc dS )NzFIPS PKCS1 Tests (Verify)r. r$ s r shortDescriptionz(FIPS_PKCS1_Verify_Tests.shortDescription\ s **r c t | }| | }t j |t | | }| || d S NrE rand_func)r r r lenr5 r$ hashmodmessage
public_keysalt signatureprnghashedr9 s r verify_positivez'FIPS_PKCS1_Verify_Tests.verify_positive_ sT DzzW%%7:#d))tLLL *****r c t | }| | }t j |t | | }| t
|j || d S rM )r r r rP r? r@ r5 rQ s r verify_negativez'FIPS_PKCS1_Verify_Tests.verify_negativee s\ DzzW%%7:#d))tLLL *hovyIIIIIr c t j d }t j | }| | d d S )N F)r generaterT r r assertEqualcan_sign)r$ test_public_keyr9 s r
test_can_signz%FIPS_PKCS1_Verify_Tests.test_can_signk sT ,t,,77997?++ **,,e44444r N)r+ r, r- rK rY r[ rb r. r r rH rH Z sS + + ++ + + J J J 5 5 5 5 5r rH c e Zd ZdS )FIPS_PKCS1_Verify_Tests_KATNr+ r, r- r. r r rd rd q Dr rd ) Signaturez PKCS1-PSSzSigVerPSS_186-3.rspz Signature Verification 186-3c | S r' r. xs r rk x s r c | S r' r. ri s r rk rk y s r )shaalgr) npc , g | ]}t | S r. r .0rj s r
rs s J J JQ
a 0 0 J J Jr rN rD c 2 | || d S r' )r5 r$ hash_objr9 rV s r
positive_testrx s ),,,,,r c J | t |j || d S r' )r? r@ r5 rv s r
negative_testrz s$ *hoxKKKKKr ztest_positive_%dztest_negative_%dc e Zd Zd Zd ZdS )FIPS_PKCS1_Sign_Testsc dS )NzFIPS PKCS1 Tests (Sign)r. rJ s r rK z&FIPS_PKCS1_Sign_Tests.shortDescription s ((r c t j d }t j | }| | d d S )Nr] T)r r^ r r r_ r` )r$ test_private_keysigners r rb z#FIPS_PKCS1_Sign_Tests.test_can_sign sH <--)** **D11111r N)r+ r, r- rK rb r. r r r| r| s2 ) ) )2 2 2 2 2r r| c e Zd ZdS )FIPS_PKCS1_Sign_Tests_KATNre r. r r r r rf r r zSigGenPSS_186-2.txtzSignature Generation 186-2rm c | S r' r. ri s r rk rk Q r zSigGenPSS_186-3.txtzSignature Generation 186-3c | S r' r. ri s r rk rk r r ec , g | ]}t | S r. r rq s r rs rs s $U$U$U!]1%5%5$U$U$Ur c \ | | }| || d S r' )signr_ )r$ rw r r) rV s r new_testr s/ KK)) F+++++r ztest_%dc e Zd ZdZd Zd ZdS )PKCS1_Legacy_Module_TestszVerify that the legacy module Crypto.Signature.PKCS1_PSS
behaves as expected. The only difference is that the verify()
method returns True/False and does not raise exceptions.c dS )Nz&Test legacy Crypto.Signature.PKCS1_PSSr. rJ s r rK z*PKCS1_Legacy_Module_Tests.shortDescription s 77r c t j d }t j t d }t j | | }t j | }| | || d t |t d t | z }| | || d d S )Nr] TestT F)
r r^ r r r r r rT r_ r5 r r rP )r$ r7 rX good_signaturer9
bad_signatures r runTestz!PKCS1_Legacy_Module_Tests.runTest s l4 !F))$$" s++0088 = !1!122 @@$GGG ~tAww^9L9L/LMM
??GGGGGr N)r+ r, r- __doc__rK r r. r r r r sC @ @8 8 8
H
H
H
H
Hr r c e Zd Zd Zd ZdS )PKCS1_All_Hashes_Testsc dS )Nz8Test PKCS#1 PSS signature in combination with all hashesr. rJ s r rK z'PKCS1_All_Hashes_Tests.shortDescription s IIr c t j d }t j | }d}|D ]F}t | t d }| | Gddlm}m } dD ];}| |t d } | | .filter_rsa s >%/22 2r c , t | d S )Nsha)r r s r
filter_shaz6TestVectorsPSSWycheproof.add_tests..filter_sha s "5<00 0r c B | d }|dvrt d|z d S )Ntype)RsassaPssVerifyzUnknown type name )r@ )r type_names r filter_typez7TestVectorsPSSWycheproof.add_tests..filter_type s3 f
I 555 !5 !ABBB 65r c | d S )NsLenr. r s r filter_slenz7TestVectorsPSSWycheproof.add_tests..filter_slen# s = r c v | d }|dvrt d|z t | d }|fd}|S )Nmgfr z Unknown MGF mgfShac $ t | || S r' r )rj ymhs r r zCTestVectorsPSSWycheproof.add_tests..filter_mgf..mgf, s Aq"~~%r )r@ r )r r mgf1_hashs r
filter_mgfz6TestVectorsPSSWycheproof.add_tests..filter_mgf& sX ,C*$$ #!5666'h88I &
&
&
&
& Jr )rg
wycheproofz Wycheproof PSS signature (%s))r7 r r r r ) group_tag)r
)r$ filenamer r r r r r) s r add_testsz"TestVectorsPSSWycheproof.add_tests s 3 3 3 1 1 1 C C C
! ! ! ..I.6.MPX.X@JHRAL@JAL 9N 9NO O O
r c g | _ | d | d | d | d | d | d | d | d | d d S )
Nz#rsa_pss_2048_sha1_mgf1_20_test.jsonz$rsa_pss_2048_sha256_mgf1_0_test.jsonz%rsa_pss_2048_sha256_mgf1_32_test.jsonz)rsa_pss_2048_sha512_256_mgf1_28_test.jsonz)rsa_pss_2048_sha512_256_mgf1_32_test.jsonz%rsa_pss_3072_sha256_mgf1_32_test.jsonz%rsa_pss_4096_sha256_mgf1_32_test.jsonz%rsa_pss_4096_sha512_mgf1_32_test.jsonzrsa_pss_misc_test.json)tvr rJ s r setUpz TestVectorsPSSWycheproof.setUp; s <=== =>>> >??? BCCC BCCC >??? >??? >??? /00000r c | j S r' )r rJ s r rK z)TestVectorsPSSWycheproof.shortDescriptionG 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 z TestVectorsPSSWycheproof.warnJ sd
: R$3 R OOO MMM4888RZZZP Q Q Q Q Q R R R Rr c d|j |j fz | _ |j |j }t
j |j |j |j } |
||j }|j sJ |
| d S # t $ r!}|j rY d }~d S |j rJ Y d }~d S d }~ww xY w)Nz Wycheproof RSA PSS Test #%d (%s)) mask_funcrE )idr r r r r4 r r7 r r r5 sigvalidr r@ r )r$ r
hashed_msgr rV r s r test_verifyz$TestVectorsPSSWycheproof.test_verifyO s 5
8KK^''//
26bgFFF
j"&99I 8OO8 IIbMMMMM
z
x < s B
C"B>/ B>>Cc D | j D ]}| | d S r' )r r )r$ r s r r z TestVectorsPSSWycheproof.runTest^ s4 ' ! !B R ! !r N)
r+ r, r- r% r r rK r r r r. r r r r s
# # #J
1
1
1 R R R
! ! ! ! !r r c | d }g }|t t z
}|t t z
}|t t z
}|t t
z
}|t t z
}| d r.|t t z
}|t t z
}|t | gz
}|S )Nr
slow_tests)
getr r0 rH r| r r rd r r )configr testss r get_testsr c s **%:;; E _Y
'
''E _ 4
5
55E _ 2
3
33E _ 6
7
77E _ 3
4
44E
zz, <
!<===
!:;;; &':;;
<r s > ) ) ) ) ) ) ) ) , , , , , , % % % % % % 5 5 5 5 5 5 R R R R R R R R < < < < < < < < < < < < < < & & & & & & % % % % % %P P P 6 ? ? ? ? ?! ? ? ?65 5 5 5 5h / 5 5 5. ("3 ('(B,A,J7B{6Ak-C -CD D J HJ .// X XIE2z"c wr3 $wr3 ##BIOO$5$566Krv&&H J J7BD/ J J JKKJ zQQvYY tBJ 37:##bj//TRRR37:!444%-BD - - - - &.BD L L L L
yC+-?%-G WWWW+-?%-G WWWW2 2 2 2 2H - 2 2 2 1 &%&@(=(D)1;;(?A A G EG
&&'A(=(D)1;;(?A A G EGG
,-- D DIE2z"c wr3 $wr3 #cm$U$Urt?T$U$U$UVV ##BIOO$5$566Krv&&H zQQvYY tBJ RZ DQQQ333 ( , , , , G %y5'8(CCCCH H H H H 1 H H H*" " " " "X . " " "4 (Q! Q! Q! Q! Q!x 0 Q! Q! Q!h & z/ / /HMg&&&&&& r