V.j, * ddlmZmZddlZ ddlmZeDcgc]}|dvr| c}ZddlmZddlZddl Z ddl m Z ddl mZmZmZmZmZdd lmZdd lmZdd lmZmZdd lmZmZdd lmZmZmZddlmZm Z m!Z!m"Z"m#Z#m$Z$ddl%m&Z&dZ' eDcgc]}|ejQ|jRf!c}Z* de jVvreegZgZ, e-edD]Z.e-e*dDcgc]\}}d|cxkre.j^krnn|!c}}D]fZ0ejbe.eejPe0Z2e,jgdjie.e0e2jje2jme'efhejnjqde,Dcgc]\}}}ejs|||c}}}dZ:e jvdZr.Cs!)))keyc |dS)Nrr+s r-r.r.Fsadr/)hashfuncz{0} {1}) sigencodezverifying_key,signature)idc@|j|ttsJyN sigdecode)verify example_datar) verifying_key signatures r-test_signaturesr>Ws(   <=   r/c |tj|\}}}tdj|t |}|tj tj dt|dz d}|jt|D]}||=tdj|d}|r|tjtj dt|dz tj dd }|jD]\}} ||xx| zcc<td j||tj dt|} |tjd } |d| | z|| dz}td j| | t|}t|xs|xs| t||k7||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_sigsnamer<old_sigsig to_removeixorsval insert_pos insert_datas r- st_fuzzed_sigraas $( (F#G D-  $ $T *+ G C qCHqLA$OINN i  F!  # #I ./ D  OO a3s8a<@ a37   jjlFAs FcMF# [   %&bkkASBCJryy#./K kz [ (3z{+; ;C/66z;O *C 9 + + , 3'> # r/)) HealthCheckideadlinesuppress_health_check max_examples c|\}}tjt5|j|tt dddy#1swYyxYwr7pytestraisesrr:r;r)argsr<rZs r-test_fuzzed_der_signaturesros<M3 ( )S,-H * ) ) AAc|tjt\}}}tdj |t |j j}|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 } r@rrArbr) rIrJrWrrKintcurveorderrNr%r!)rVrXr<_rursrZs r-st_random_der_ecdsa_sig_valuerys""//-"@AD-  $ $T *+  ##)) *E  a5A:6 ++ eai @ A A  a5A:6 ++ eai @ A A .+^A-> ?C # r/c|\}}tjt5|j|tt dddy#1swYyxYw)zd Check if random values encoded in ECDSA-Sig-Value structure are rejected as signature. r8Nrkparamsr<rZs r-test_random_der_ecdsa_sig_valuer}s< M3 ( )S,-H * ) )rpcnd|vrd|d<tjttj|i|S)z Hypothesis strategy that returns a random positive integer as DER INTEGER. Parameters are passed to hypothesis.strategy.integer. rBr)rIbuildsr!rNrnkwargss r-st_der_integerrs5 & { 99^R[[$%A&%A BBr/c|tj|i|}|rE|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. rrcrArb)rIrTrNrLrUr")rVrnrdataunuseds r-st_der_bit_stringrsl  4*6* +D bkkA;< RaiL T{ D& ))r/c\tjttj|i|S)z Hypothesis strategy that returns a random DER OCTET STRING object. Parameters are passed to hypothesis.strategy.binary )rIrr#rTrs r-st_der_octet_stringrs$ 99("))T*DV*D EEr/c,tjdS)z; Hypothesis strategy that returns DER NULL object. s)rIjustr2r/r- st_der_nullrs 77; r/c F|tjdd}|dkr|tjdd}n |tjdddz}|tjtjdddzd}t||g|S)zI Hypothesis strategy that returns DER OBJECT IDENTIFIER objects. rrbrA'i2rG)rIrNrMr$)rVfirstsecondrests r- st_der_oidrs qA6 7E qybkkA<=bkkAC@A  qAsF;bI D eV +d ++r/ctjtjdtddzzt 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. r/rbi)rCrGctjttj|tjdtj|ztjdtj|dztjt tj ddtj|zS)Nct|dS)Nr)r"r+s r-r.z*st_der.....s .q!4r/ct|Sr))r%r+s r-r.z*st_der....0s oq)r/rGr?rA)rIrr#one_ofrMr&rN)childrens r-r.zst_der..-s#6 (8KL ))4bii6I JK )) )288Hs+K  ))  KK!t 4 IIh   r/() max_leaves)rI recursiverrrrrrr2r/r-st_derrso <<  1d7 + , W - . w / 0 -   ,   # r/c|\}}}tjt5|j|tt dddy#1swYyxYw)z8Check if random DER structures are rejected as signaturer8Nrkr|derrXr<rvs r-test_random_der_as_signaturer;s>$D- ( )S,-H * ) ) AArrGc|\}}}tjt5|j|tt dddy#1swYyxYw)z/Check if random bytes are rejected as signaturer8Nrkrs r-test_random_bytes_as_signaturerEs>$D- ( )S,-H * ) )rruc|\}}tjt5|j|tt dddy#1swYyxYwr7)rlrmrr:r;rr{s r-test_fuzzed_string_signaturesrts= M3 ( )S,:JK * ) )rpr2)V __future__rrhashlibr ImportError functoolsrrlsyshypothesis.strategies strategiesrI hypothesisrrrrrkeysrrutilrrrrrrr rr!r"r#r$r%r& ellipticcurver'r;new digest_size hash_and_sizeargvrWsortedrtr*hash_alggenerateskappendrKr<signmark parametrizeparamr> compositerar| version_inforddata_too_largefilter_too_muchtoo_slowdict slow_paramsroryr}rrrrrrrJrrTr isinstancerukeys_and_string_sigsr)r\rXsizevkrZr<s000000r-rsy/ ," !@@ !  "==#1100%$ I7K6JdT7;;t  ( ()6J &  sxx #F IF 3 4E!NCCJD$ t $u}} $ C !Z GGKK:    1   >  5(9FG b#V\\"cd\#G  00f v&F:""##'F "#  sxxF>6l  sxx"#K"$K K}]#$I%IB K$&'I(IC * * F  , ,: Kr}%vx0I1I Kr}%yryy''BC!onQ&79JK !N1%~a'89GC q ?^A->,?!,CDEIF DI"%2 %2 mS m))// ;  3 3 3 9 9 : %%++ %2   %2  %2 mS-%%++Y7   %2  K})*+L,L_ >$*H| $s5Q Q0 $Q5&$Q:RA"R -R Q-,Q-