dfi* ddlmZmZddlZ ddlmZeDcgc]}|dvr| c}ZddlmZddlZddl Z ddl m Z ddl mZmZmZmZmZdd lmZdd lmZdd lmZmZdd lmZmZdd lmZddlmZmZmZm Z m!Z!m"Z"ddl#m$Z$dZ% eDcgc] }|ejL|jNf"c}Z( gZ) e*edD]Z+e*e(dDcgc]\}}d|cxkre+jXkrnn|!c}}D]fZ-ej\e+eejLe-Z/e)jadjce+e-e/jde/jge%efhejhjkde)Dcgc]\}}}ejl|||c}}}dZ7e jpdZ9iZ:e jvdk\r1ddl mr,?s !)))keyc |dS)Nrr)s r+r,r,Bs adr-)hashfuncz{0} {1}) sigencodezverifying_key,signature)idc@|j|ttsJyN sigdecode)verify example_datar) verifying_key signatures r+test_signaturesr<Ss(   <=   r-c |tj|\}}}tdj|t |}|tj tj dt|dz d}|jt|D]}||=tdj||r|tjtj dt|dz tj dd}|jD]\}} ||xx| zcc<td j||tj dt|} |tjd } |d | | z|| d z}td j| | t|}t|xsxs| 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}z xors: {0}max_sizeNz%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_sigr_]s $( (F#G D-  $ $T *+ G C qCHqLA$OINN i  F  # #I ./  OO a3s8a<@ a37   jjl FAs FcMF  [   %&bkkASBCJryy#./K kz [ (3z{+; ;C/66z;O *C 9 + + , 3'> # r-)) HealthCheckideadlinesuppress_health_check max_examplesc|\}}tjt5|j|tt dddy#1swYyxYwr5pytestraisesrr8r9r)argsr:rXs r+test_fuzzed_der_signaturesrlsIM3 ( )IS,-HIII 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>rr?r`r) rGrHrUrrIintcurveorderrLr#r)rTrVr:_rrrsrXs r+st_random_der_ecdsa_sig_valuervs""//-"@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. r6Nrhparamsr:rXs r+test_random_der_ecdsa_sig_valuerzsI M3 ( )IS,-HIIIrmcnd|vrd|d<tjttj|i|S)z Hypothesis strategy that returns a random positive integer as DER INTEGER. Parameters are passed to hypothesis.strategy.integer. r@r)rGbuildsrrLrkkwargss 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. rrar?r`)rGrRrLrJrSr )rTrkr~dataunuseds 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 )rGr|r!rRr}s r+st_der_octet_stringrs$ 99("))T*DV*D EEr-c,tjdS)z; Hypothesis strategy that returns DER NULL object. s)rGjustr0r-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. rr`r?'i2rE)rGrLrKr")rTfirstsecondrests 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-r`i)rArEctjdtj|tjdtj|ztjdtj|dztjdtjdd tj|zS) Nct|Sr')r!r)s r+r,z*st_der....$s )!,r-ct|dS)Nr)r r)s r+r,z*st_der....&s.q!4r-ct|Sr')r#r)s r+r,z*st_der....(s oq)r-rEct||Sr')r$)tagr*s r+r,z*st_der....+s-c15r-r?r?)rGr|one_ofrKrL)childrens r+r,zst_der..#s ,bii.A  ))4bii6I JK )) )288Hs+K   )) 5 KK!t 4 IIh    r-() max_leaves)rG recursiverrrrrrr0r-r+st_derrso <<  1d7 + , W - . w / 0 -   ,   ' r-c|\}}}tjt5|j|tt dddy#1swYyxYw)z8Check if random DER structures are rejected as signaturer6NrhryderrVr:rss r+test_random_der_as_signaturer3sK$D- ( )IS,-HIII AArrEc|\}}}tjt5|j|tt dddy#1swYyxYw)z/Check if random bytes are rejected as signaturer6Nrhrs r+test_random_bytes_as_signaturer=sK$D- ( )IS,-HIIIrrrc|\}}tjt5|j|tt dddy#1swYyxYwr5)rirjrr8r9rrxs r+test_fuzzed_string_signaturesrlsJ M3 ( )LS,:JKLLLrmr0)S __future__rrhashlibr ImportError functoolsrrisyshypothesis.strategies strategiesrG hypothesisrrrrrkeysrrutilrrrrrrrr r!r"r#r$ ellipticcurver%r9new digest_size hash_and_sizerUsortedrqr(hash_alggenerateskappendrIr:signmark parametrizeparamr< compositer_ry version_inforbdata_too_largefilter_too_muchtoo_slowdict slow_paramsrlrvrzrrrrrrrHrrRr isinstancerrkeys_and_string_sigsr)rZrVsizevkrXr:s000000r+rs(/ ," @@  "==#11%$ I7K.2T;7;;t  ( () &  IF 3 4 E!NC D$ t $u}} $   !Z GGKK:    1   >    (9FGG b#\V\\"cd#G  //d v&F:""##'F "# 6l  N F}]#$I%IB K$&'I(IC * * F  , ,> Fr}%vx0I1I Fr}%yryy''BC!onQ&79JK !N1%~a'89GC q ?^A->,?!,CDEIF DI"%2   !mS m))// ;  3 3 3 9 9 : %%++    %2  !mS-%%++Y7   F})*+L,LO >*Ht $s5P% P5 %P:$P? Q"A"Q -Q % P21P2