o 6di* @sddlmZmZddlZzddlmZWn ey!gdZYnwddeDZddlmZddlZddl Z ddl m Z ddl mZmZmZmZmZd d lmZd d lmZd d lmZmZd d lmZmZd dlmZd dlmZmZmZm Z m!Z!m"Z"d dl#m$Z$dZ% ddeDZ& gZ' e(edddD]/Z)dde(e&dddDD]Z*ej+e)eej,e*dZ-e'.d/e)e*e-j0e-j1e%edfqqej23ddde'DddZ4e j5dd Z6iZ7e j8d!krdd"l m9Z9d#e7d$<e9j:e9j;e9jd&e>d'<edFie7ee6e'd(d)Z?e j5d*d+Z@edFie>ee@d,d-ZAd.d/ZBe j5d0d1ZCd2d3ZDd4d5ZEe j5d6d7ZFd8d9ZGedFie7ee He'eGd:d;ZIedFie7ee He'e jJdee'de!ed gd?d@dAZKdBde'DZL eLdCde'D7ZLedFie7ee6eLdDdEZMdS)G)with_statementdivisionN)algorithms_available)md5sha1sha224sha256sha384sha512cCsg|]}|dvr|qS))Zmdc2Zmd2Zmd4Z whirlpoolZ ripemd160).0ir r pC:\Users\jesus\OneDrive\Desktop\erpjis_fastapi\backend\jisbackend\Lib\site-packages\ecdsa/test_malformed_sigs.py s r)partial)noteassumegivensettingsexample) SigningKey)BadSignatureError) sigencode_dersigencode_string) sigdecode_dersigdecode_string)curves)encode_integerencode_bitstringencode_octet_string encode_oidencode_sequenceencode_constructed) CurveEdTwssome data to signcCsg|] }|t|jfqSr )hashlibnew digest_size)r namer r rr2scCs|jSN)baselenxr r r?sr-)keycCs.g|]\}}d|krtjkrnn|qS)r)curver*)r r(sizer r rr@s cCs|dS)Nrr r+r r rr-B)hashfuncz{0} {1})Z sigencodezverifying_key,signaturecCs"g|] \}}}tj|||dqS))id)pytestparam)r r(Zvksigr r rrUs"cCs|j|ttds JdSNZ sigdecode)verify example_datar) verifying_key signaturer r rtest_signaturesSsr=c CsV|t|\}}}td|t|}|tjtjdt|dddd}|t |D]}||=q/td||rl|t tjdt|ddtjddd}| D] \}} ||| N<qXtd ||tjdt|d} |tj d d } |d | | || d }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}z xors: {0}max_sizeNz%Inserted at position {0} bytes: {1!r})st sampled_fromrformat bytearraylistsintegerslensortreversedZ dictionariesitemsbinarybytesr) draw keys_and_sigsr(r;Zold_sigr6 to_remover ZxorsvalZ insert_posZ insert_datar r r st_fuzzed_sig]s<    rW)) HealthCheckideadlineZsuppress_health_check Z max_examplescCH|\}}tt|j|ttdWddS1swYdSr7r4raisesrr9r:r)argsr;r6r r rtest_fuzzed_der_signatures "racCs|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 } r>rr?rXr) rGrHrTrrIintr/orderrLr"r)rSr(r;_rersr6r r rst_random_der_ecdsa_sig_values  ricCsH|\}}tt|j|ttdWddS1swYdS)zd Check if random values encoded in ECDSA-Sig-Value structure are rejected as signature. r8Nr^paramsr;r6r r rtest_random_der_ecdsa_sig_values "rlcOs(d|vrd|d<tttj|i|S)z Hypothesis strategy that returns a random positive integer as DER INTEGER. Parameters are passed to hypothesis.strategy.integer. r@r)rGbuildsrrLr`kwargsr r rst_der_integersrpcOs`|tj|i|}|r)|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. rrYr?rX)rGrQrLrJrRr)rSr`rodataZunusedr r rst_der_bit_strings  rscOstttj|i|S)z Hypothesis strategy that returns a random DER OCTET STRING object. Parameters are passed to hypothesis.strategy.binary )rGrmr rQrnr r rst_der_octet_stringsrtcCs tdS)z; Hypothesis strategy that returns DER NULL object. s)rGjustr r r r st_der_nulls rvcCsv|tjddd}|dkr|tjddd}n |tjdddd}|tjtjdddddd}t||g|RS)zI Hypothesis strategy that returns DER OBJECT IDENTIFIER objects. rrXr?'i2rE)rGrLrKr!)rSfirstsecondrestr 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. rXi)rArEcSshtddt|tddt|Btddtj|ddBtddtjdd d t|BS) NcSst|Sr))r r+r r rr-$r1z*st_der....cSs t|dS)Nr)rr+r r rr-& cSst|Sr))r"r+r r rr-(r1rEcSs t||Sr))r#)tagr,r r rr-+rr?r?)rGrmone_ofrKrL)childrenr r rr-#s zst_der..()Z max_leaves)rG recursiverurprsrtrvr|r r r rst_ders  rcCJ|\}}}tt|j|ttdWddS1swYdS)z8Check if random DER structures are rejected as signaturer8Nr^rkderr(r;rfr r rtest_random_der_as_signature3s  "rr~rEcCr)z/Check if random bytes are rejected as signaturer8Nr^rr r rtest_random_bytes_as_signature=s "rcCsBg|]\}}}t|jjts||tt||jjd|jjifqS)re) isinstancer/r$rrrer r(r;r6r r rrOs  cCs*g|]\}}}t|jjtr|||fqSr )rr/r$rr r rras cCr]r7)r4r_rr9r:rrjr r rtest_fuzzed_string_signatureslrbrr )N __future__rrr%r ImportError functoolsrr4sysZhypothesis.strategies strategiesrGZ hypothesisrrrrrkeysrrutilrrrrrrrrr r!r"r#Z ellipticcurver$r:Z hash_and_sizerTsortedr/Zhash_alggenerater&skappendrIr;signmarkZ parametrizer=Z compositerWrk version_inforZZdata_too_largeZfilter_too_muchZtoo_slowdictZ slow_paramsrarirlrprsrtrvr|rrHrrQrZkeys_and_string_sigsrr r r rs                 2       !