qtj, ddlmZmZddlZ ddlmZn#e$rgdZYnwxYwdeDZddlmZddlZddl Z ddl m Z ddl mZmZmZmZmZdd lmZdd lmZdd lmZmZdd lmZmZdd lmZmZmZddlmZm Z m!Z!m"Z"m#Z#m$Z$ddl%m&Z&dZ' deDZ( de j)vreegZgZ* e+edD]Z,de+e(dDD]iZ-ej.e,eej/e-Z0e*1d2e,e-e0j3e04e'efjej56dde*DdZ7e j8dZ9iZ:e j;dkr"ddl mers. @@@@@@)partial)noteassumegivensettingsexample) SigningKey)BadSignatureError) sigencode_dersigencode_string) sigdecode_dersigdecode_string)curves SECP112r2 SECP128r1)encode_integerencode_bitstringencode_octet_string encode_oidencode_sequenceencode_constructed) CurveEdTwssome data to signcDg|]}|tj|jfSr)hashlibnew digest_size)rnames rrr2s7.2T7;t   ()rz--fastc|jSN)baselenxs rr9Cs!)r)keycLg|]!\}}d|cxkrtjknn|"S)r)curver6)rr3sizes rrrDsJ D$ t $ $ $ $u} $ $ $ $ $ $ $ $rc|dS)Nrrr7s rr9r9Fs adr)hashfuncz{0} {1}) sigencodezverifying_key,signaturecDg|]\}}}tj|||S))id)pytestparam)rr3vksigs rrrYs/GGG b#V\"cd###GGGrcN||ttsJdSN sigdecode)verify example_datar#) verifying_key signatures rtest_signaturesrOWs<   <=    rc |tj|\}}}td|t |}|tjtjdt|dz d}|t|D]}||=td|d}|r|tj tjdt|dz tjdd }| D]\}} ||xx| zcc<td ||tjdt|} |tj d } |d| | z|| dz}td | | t|}t|p|p| t||k||fS)z Hypothesis strategy that generates pairs of VerifyingKey and malformed signatures created by fuzzing of a valid signature. Configuration: {0}rr min_value max_valueT)uniquezRemove bytes: {0}Nz xors: {0}max_sizez%Inserted at position {0} bytes: {1!r})st sampled_fromrformat bytearraylistsintegerslensortreversed dictionariesitemsbinarybytesr) draw keys_and_sigsr3rMold_sigrF to_removerxorsval insert_pos insert_datas r st_fuzzed_sigroasR $(4 (F(F#G#G D-  $ $T * *+++ G  C qCHHqLAAA$OOOINN i  FF  # #I . ./// D 't O a3s88a<@@@ a3777     jjll  FAs FFFcMFFFF [   % %&&&bkASBBBCCJ$ry#...//K kzk [ (3z{{+; ;C/66z;OO **C 9 + + ,,, 3'> # r)) HealthCheckideadlinesuppress_health_check max_examples c|\}}tjt5||tt ddddS#1swxYwYdSrHrCraisesr rKrLr#)argsrMrFs rtest_fuzzed_der_signaturesr|sM3 ( ) )IIS,-HHHIIIIIIIIIIIIIIIIII#AAAc|tjt\}}}td|t |jj}|tjd|dztj|dz |dzz}|tjd|dztj|dz |dzz}tt|t|}||fS)z Hypothesis strategy for selecting random values and encoding them to ECDSA-Sig-Value object:: ECDSA-Sig-Value ::= SEQUENCE { r INTEGER, s INTEGER } rQrrRrpr) rZr[rhrr\intr<orderr_r,r()rgr3rM_rrsrFs rst_random_der_ecdsa_sig_valuers"T"/-"@"@AAD-  $ $T * *+++  #) * *E  a5A:666 + eai @ @ @ A  A  a5A:666 + eai @ @ @ A  A .++^A->-> ? ?C # rc|\}}tjt5||tt ddddS#1swxYwYdS)zd Check if random values encoded in ECDSA-Sig-Value structure are rejected as signature. rINryparamsrMrFs rtest_random_der_ecdsa_sig_valuers M3 ( ) )IIS,-HHHIIIIIIIIIIIIIIIIIIr}cfd|vrd|d<tjttj|i|S)z Hypothesis strategy that returns a random positive integer as DER INTEGER. Parameters are passed to hypothesis.strategy.integer. rSr)rZbuildsr(r_r{kwargss rst_der_integerrs: &  { 9^R[$%A&%A%A B BBrc|tj|i|}|rR|tjdd}t|}|dxxd|z zcc<t |}nd}t ||S)z| Hypothesis strategy that returns a random DER BIT STRING. Parameters are passed to hypothesis.strategy.binary. rrqrRrp)rZrer_r]rfr))rgr{rdataunuseds rst_der_bit_stringrs 4 4*6** + +D bkA;;;<< RaiL T{{ D& ) ))rcTtjttj|i|S)z Hypothesis strategy that returns a random DER OCTET STRING object. Parameters are passed to hypothesis.strategy.binary )rZrr*rers rst_der_octet_stringrs& 9(")T*DV*D*D E EErc*tjdS)z; Hypothesis strategy that returns DER NULL object. s)rZjustrrr st_der_nullrs 7;  rc X|tjdd}|dkr |tjdd}n"|tjdddz}|tjtjdddzd}t||g|RS)zI Hypothesis strategy that returns DER OBJECT IDENTIFIER objects. rrprR'i2rX)rZr_r^r+)rgfirstsecondrests r st_der_oidrs DqA666 7 7E qyybkA<<<==bkAC@@@AA 4 qAsF;;;bIII  D eV +d + + ++rctjtjdtddzzt dzt dzt ztzdd S) z Hypothesis strategy that returns random DER structures. A valid DER structure is any primitive object, an octet encoding of a valid DER structure, sequence of valid DER objects or a constructed encoding of any of the above. rrpi)rTrXctjttj|tjdtj|ztjdtj|dztjt tjddtj|zS)Nc"t|dS)Nr)r)r7s rr9z*st_der.....s.q!44rct|Sr5)r,r7s rr9z*st_der....0s oq)rrXr?rR)rZrr*one_ofr^r-r_)childrens rr9zst_der..-s#6 (8K8KLL )44bi6I6I J JK ) ) )28Hs+K+K+K   )  K!t 4 4 4 Ih     r() max_leaves)rZ recursiverrrrrrrrrst_derrs <  1d7 + + + , W - - - . w / / / 0 --   ,,   #   rc|\}}}tjt5||tt ddddS#1swxYwYdS)z8Check if random DER structures are rejected as signaturerINryrderr3rMrs rtest_random_der_as_signaturer;s$D- ( ) )IIS,-HHHIIIIIIIIIIIIIIIIII#AAArrXc|\}}}tjt5||tt ddddS#1swxYwYdS)z/Check if random bytes are rejected as signaturerINryrs rtest_random_bytes_as_signaturerEs$D- ( ) )IIS,-HHHIIIIIIIIIIIIIIIIIIrc g|]V\}}}t|jjt%||tt ||jjd|jjifWS)r) isinstancer<r.r"r#rrr3rMrFs rrrWs|    !mS m)/ ; ;   3 3 9 : : %+     rc^g|]*\}}}t|jjt%|||f+Sr)rr<r.rs rrrisQ !mS-%+Y77  rc|\}}tjt5||tt ddddS#1swxYwYdSrH)rCrzr rKrLr$rs rtest_fuzzed_string_signaturesrts M3 ( ) )LLS,:JKKKLLLLLLLLLLLLLLLLLLr}r)Q __future__rrr0r ImportError functoolsrrCsyshypothesis.strategies strategiesrZ hypothesisrrrrrkeysrr utilr!r"r#r$r%r&r'rr(r)r*r+r,r- ellipticcurver.rL hash_and_sizeargvrhsortedr<hash_alggenerater1skappendr\rMsignmark parametrizerO compositeror version_inforrdata_too_largefilter_too_muchtoo_slowdict slow_paramsr|rrrrrrrrr[rrerkeys_and_string_sigsrrrrrsZ//////// ,,,,,,, !  """"""==============######11111111111111110000000000%$$$$$$ I6J &  sx #F IVF 3 3 4 4 4  E &NNCCC   !Z GGGK::      11  >>      &GGGGG  000f v&&&&&&F:"#'F "#  sxF>d6ll  sx"#K"$K K}}]##$$II%$IB K$$&&''II('ICCC * * * FFF    , , ,: Kr}%%vvxx00II10I Kr}%%yry''B'B'BCC!oonnQ&7&79J9JKK !ONN1%%~~a'8'899GC q ??^^A->->,?!,CDEEIIFE DCI  %2     %2  K}})**++LL,+LLLs  !!