U ʹh, @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_availableZmd5Zsha1Zsha224Zsha256Zsha384Zsha512cCsg|]}|dkr|qS))Zmdc2Zmd2Zmd4Z whirlpoolZ ripemd160).0irr?./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)hashlibnewZ digest_size)rnamerrrr 2sz--fastcCs|jSN)baselenxrrrCr()keycCs.g|]&\}}d|kr"tjkrnq|qS)r)curver%)rr#sizerrrr Ds  cCs|dS)Nrrr&rrrr(Fr))Zhashfuncz{0} {1})Z sigencodezverifying_key,signaturecCs"g|]\}}}tj|||dqS))id)pytestZparam)rr#Zvksigrrrr YscCs|j|ttdstdSNZ sigdecode)verify example_datarAssertionError) verifying_keyZ signaturerrrtest_signaturesWs r6c 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}Zmax_sizez%Inserted at position {0} bytes: {1!r})st sampled_fromr format bytearraylistsintegerslensortreversedZ dictionariesitemsbinarybytesr ) draw keys_and_sigsr#r5Zold_sigr/Z to_removerZxorsvalZ insert_posZ insert_datarrr st_fuzzed_sigas>    rN)) HealthCheckiZdeadlineZsuppress_health_checkZ max_examples c Cs2|\}}tt|j|ttdW5QRXdSr0r.raisesrr2r3r)argsr5r/rrrtest_fuzzed_der_signaturess rWcCs|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 } r7rr8rOr) r?r@rLr rAintr+orderrDrr)rKr#r5_rZrsr/rrrst_random_der_ecdsa_sig_values  r^c Cs2|\}}tt|j|ttdW5QRXdS)zd Check if random values encoded in ECDSA-Sig-Value structure are rejected as signature. r1NrTparamsr5r/rrrtest_random_der_ecdsa_sig_values racOs$d|krd|d<tttj||S)z Hypothesis strategy that returns a random positive integer as DER INTEGER. Parameters are passed to hypothesis.strategy.integer. r9r)r?buildsrrDrVkwargsrrrst_der_integersrecOs\|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. rrPr8rO)r?rIrDrBrJr)rKrVrddataZunusedrrrst_der_bit_strings rhcOstttj||S)z Hypothesis strategy that returns a random DER OCTET STRING object. Parameters are passed to hypothesis.strategy.binary )r?rbrrIrcrrrst_der_octet_stringsricCs tdS)z; Hypothesis strategy that returns DER NULL object. s)r?justrrrr st_der_nullsrkcCst|tjddd}|dkr.|tjddd}n|tjdddd}|tjtjdddddd}t||f|S)zI Hypothesis strategy that returns DER OBJECT IDENTIFIER objects. rrOr8'i2r>)r?rDrCr)rKfirstsecondrestrrr st_der_oidsrqcCsHtjtdtdddBtddBtddBtBtBddd 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. r)rOi)r:r>cSs`ttt|tddt|Btddtj|ddBtttjdddt|BS) NcSs t|dS)Nr)rr&rrrr(.r)z*st_der....cSst|Sr$)rr&rrrr(0r)r>r?r8)r?rbrZone_ofrCrrD)Zchildrenrrrr(-s  zst_der..()Z max_leaves)r? recursiverjrerhrirkrqrrrrst_ders  rwc Cs4|\}}}tt|j|ttdW5QRXdS)z8Check if random DER structures are rejected as signaturer1NrTr`derr#r5r[rrrtest_random_der_as_signature;s  rzrrr>c Cs4|\}}}tt|j|ttdW5QRXdS)z/Check if random bytes are rejected as signaturer1NrTrxrrrtest_random_bytes_as_signatureEs  r}cCsBg|]:\}}}t|jjts||tt||jjd|jjifqS)rZ) isinstancer+r rrrZrr#r5r/rrrr Ws  cCs*g|]"\}}}t|jjtr|||fqSr)r~r+r rrrrr is c Cs2|\}}tt|j|ttdW5QRXdSr0)r.rUrr2r3rr_rrrtest_fuzzed_string_signaturests r)QZ __future__rrr!r ImportError functoolsr r.sysZhypothesis.strategiesZ strategiesr?Z hypothesisr r r rrkeysrrutilrrrrrrrryrrrrrrZ ellipticcurver r3Z hash_and_sizeargvrLsortedr+Zhash_algZgenerater"skappendrAr5signZmarkZ parametrizer6Z compositerNr` version_inforQZdata_too_largeZfilter_too_muchZtoo_slowdictZ slow_paramsrWr^rarerhrirkrqrwr@rzrIr}Zkeys_and_string_sigsrrrrrs                3        !