U #i, @sddlmZmZddlZzddlmZWn$ek rLdddddd gZYnXd d eDZdd lmZddlZddl Z ddl m Z dd l mZmZmZmZmZddlmZddlmZddlmZmZddlmZmZddlmZmZmZddlmZm Z m!Z!m"Z"m#Z#m$Z$ddl%m&Z&dZ'dd eDZ(de j)kr>eegZgZ*e+edddD]bZ,dd e+e(dddDD]@Z-ej.e,eej/e-dZ0e*1d2e,e-e0j3e0j4e'ed fqpqRej56d!d"d e*Dd#d$Z7e j8d%d&Z9iZ:e j;d'krdd(l meZId?d@ZJefeAee Ke*eJdAdBZLefeAee Ke*e jMdCdDee*de#ededee*de#ededdEee*de#edgdFdGdHZNdId e*DZOeOdJd e*D7ZOefeAee9eOdKdLZPdS)M)with_statementdivisionN)algorithms_availablemd5sha1sha224sha256sha384sha512cCsg|]}|dkr|qS))Zmdc2Zmd2Zmd4Z whirlpoolZ ripemd160).0ir r a/var/www/html/me.goteku.com/backend/venv/lib/python3.8/site-packages/ecdsa/test_malformed_sigs.py sr)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 signcCsg|]}|t|jfqSr )hashlibnew digest_size)r namer r rr2sz--fastcCs|jSN)baselenxr r rCr/)keycCs.g|]&\}}d|kr"tjkrnq|qS)r)curver,)r r*sizer r rrDs  cCs|dS)Nrr r-r r rr/Fr0)hashfuncz{0} {1})Z sigencodezverifying_key,signaturecCs"g|]\}}}tj|||dqS))id)pytestparam)r r*Zvksigr r rrYscCs|j|ttdstdSNZ sigdecode)verify example_datarAssertionError) verifying_key signaturer r rtest_signaturesWs r@c Cs^|t|\}}}td|t|}|tjtjdt|dddd}|t |D] }||=q^td|d}|r|t tjdt|ddtjdd d}| D]\}} ||| N<qtd ||tjdt|d} |tj d d } |d| | || d}td | | t |}t|pF|pF| 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 bytearraylistsintegerslensortreversedZ dictionariesitemsbinarybytesr) draw keys_and_sigsr*r>Zold_sigr8Z to_remover ZxorsvalZ insert_posZ insert_datar r r st_fuzzed_sigas>    rY)) HealthCheckideadlineZsuppress_health_checkZ max_examples c Cs2|\}}tt|j|ttdW5QRXdSr9r6raisesrr;r<r)argsr>r8r r rtest_fuzzed_der_signaturess rccCs|tt\}}}td|t|jj}|tjd|d>dtj|d?|ddB}|tjd|d>dtj|d?|ddB}t t |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 } rArrBrZr) rJrKrWrrLintr2orderrOr$r )rVr*r>_rfrsr8r r rst_random_der_ecdsa_sig_values  rjc Cs2|\}}tt|j|ttdW5QRXdS)zd Check if random values encoded in ECDSA-Sig-Value structure are rejected as signature. r:Nr`paramsr>r8r r rtest_random_der_ecdsa_sig_values rmcOs$d|krd|d<tttj||S)z Hypothesis strategy that returns a random positive integer as DER INTEGER. Parameters are passed to hypothesis.strategy.integer. rCr)rJbuildsr rOrbkwargsr r rst_der_integersrqcOs\|tj||}|rN|tjddd}t|}|dd| M<t|}nd}t||S)z| Hypothesis strategy that returns a random DER BIT STRING. Parameters are passed to hypothesis.strategy.binary. rr[rBrZ)rJrTrOrMrUr!)rVrbrpdataZunusedr r rst_der_bit_strings rtcOstttj||S)z Hypothesis strategy that returns a random DER OCTET STRING object. Parameters are passed to hypothesis.strategy.binary )rJrnr"rTror r rst_der_octet_stringsrucCs tdS)z; Hypothesis strategy that returns DER NULL object. s)rJjustr r r r st_der_nullsrwcCst|tjddd}|dkr.|tjddd}n|tjdddd}|tjtjdddddd}t||f|S)zI Hypothesis strategy that returns DER OBJECT IDENTIFIER objects. rrZrB'i2rH)rJrOrNr#)rVfirstsecondrestr r r st_der_oidsr}cCsHtjtdtdddBtddBtddBtBtBddd d 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. r0rZi)rDrHcSs`ttt|tddt|Btddtj|ddBtttjdddt|BS) NcSs t|dS)Nr)r!r-r r rr/.r0z*st_der....cSst|Sr+)r$r-r r rr/0r0rHr?rB)rJrnr"Zone_ofrNr%rO)childrenr r rr/-s  zst_der..()Z max_leaves)rJ recursivervrqrtrurwr}r r r rst_ders  rc Cs4|\}}}tt|j|ttdW5QRXdS)z8Check if random DER structures are rejected as signaturer:Nr`rlderr*r>rgr r rtest_random_der_as_signature;s  rr~rHc Cs4|\}}}tt|j|ttdW5QRXdS)z/Check if random bytes are rejected as signaturer:Nr`rr r rtest_random_bytes_as_signatureEs  rcCsBg|]:\}}}t|jjts||tt||jjd|jjifqS)rf) isinstancer2r&rrrfr r*r>r8r r rrWs  cCs*g|]"\}}}t|jjtr|||fqSr )rr2r&rr r rris c Cs2|\}}tt|j|ttdW5QRXdSr9)r6rarr;r<rrkr r rtest_fuzzed_string_signaturests r)Q __future__rrr'r ImportError functoolsrr6sysZhypothesis.strategiesZ strategiesrJZ hypothesisrrrrrkeysrrutilrrrrrrrrr r!r"r#r$r%Z ellipticcurver&r<Z hash_and_sizeargvrWsortedr2Zhash_alggenerater(skappendrLr>signmarkZ parametrizer@Z compositerYrl version_infor\Zdata_too_largeZfilter_too_muchZtoo_slowdictZ slow_paramsrcrjrmrqrtrurwr}rrKrrTrZkeys_and_string_sigsrr r r rs                3        !