a db( @sZddlmZmZddlZzddlmZWneyBgdZYn0ddlmZddlZddl Z ddl m Z ddl mZmZmZmZmZddlmZdd lmZdd lmZmZdd lmZmZdd lmZdd lmZmZmZm Z m!Z!m"Z"dZ#ddeDZ$gZ%e&edddD]bZ'dde&e$dddDD]@Z(ej)e'eej*e(dZ+e%,d-e'e(e+j.e+j/e#edfq0qej01ddde%DddZ2e j3ddZ4iZ5e j6dkrdd l m7Z7d!e5d"<e7j8e7j9e7j:ge5d#<e;e5Zefie<ee>d*d+Z?d,d-Z@e j3d.d/ZAd0d1ZBd2d3ZCe j3d4d5ZDd6d7ZEefie5ee Fe%eEd8d9ZGefie5ee Fe%e jHd:d;ee%de!ededee%de!ededd<ee%de!edgd=d>d?ZId@de%DZJefie5ee4eJdAdBZKdS)C)with_statementdivisionN)algorithms_available)md5sha1sha224sha256sha384sha512)partial)noteassumegivensettingsexample) SigningKey)BadSignatureError) sigencode_dersigencode_string) sigdecode_dersigdecode_string)curves)encode_integerencode_bitstringencode_octet_string encode_oidencode_sequenceencode_constructedssome data to signcCsg|]}|t|jfqS)hashlibnew digest_size).0namerrj/home/tom/ab/renpy-build/tmp/install.linux-x86_64/lib/python3.9/site-packages/ecdsa/test_malformed_sigs.py )sr&cCs|jSN)baselenxrrr%6r+)keycCs.g|]&\}}d|kr"tjkrnq|qS)r)curver()r#r$sizerrr%r&7scCs|dS)Nrrr)rrr%r+9r,)hashfuncz{0} {1})Z sigencodezverifying_key,signaturecCs"g|]\}}}tj|||dqS))id)pytestparam)r#r$Zvksigrrr%r&Lr,cCs|j|ttdsJdSNZ sigdecode)verify example_datar) verifying_key signaturerrr%test_signaturesJsr;c CsZ|t|\}}}td|t|}|tjtjdt|dddd}|t |D] }||=q^td||r|t tjdt|ddtjddd}| D]\}} ||| N<qtd ||tjdt|d} |tj d d } |d | | || d }td | | t |}t|pB|pB| 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_fromr format bytearraylistsintegerslensortreversedZ dictionariesitemsbinarybytesr ) draw keys_and_sigsr$r9Zold_sigr4 to_removeiZxorsvalZ insert_posZ insert_datarrr% st_fuzzed_sigTs<    rV)) HealthCheckideadlineZsuppress_health_check Z max_examplescCsF|\}}tt |j|ttdWdn1s80YdSr5r2raisesrr7r8r)argsr9r4rrr%test_fuzzed_der_signaturess r_cCs|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=rWr) rErFrRr rGintr.orderrJrr)rQr$r9_rbrsr4rrr%st_random_der_ecdsa_sig_values  rfcCsF|\}}tt |j|ttdWdn1s80YdS)zd Check if random values encoded in ECDSA-Sig-Value structure are rejected as signature. r6Nr\paramsr9r4rrr%test_random_der_ecdsa_sig_values ricOs(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)rEbuildsrrJr^kwargsrrr%st_der_integersrmcOs`|tj|i|}|rR|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. rrXr=rW)rErOrJrHrPr)rQr^rldataZunusedrrr%st_der_bit_strings rpcOstttj|i|S)z Hypothesis strategy that returns a random DER OCTET STRING object. Parameters are passed to hypothesis.strategy.binary )rErjrrOrkrrr%st_der_octet_stringsrqcCs tdS)z; Hypothesis strategy that returns DER NULL object. s)rEjustrrrr% st_der_nullsrscCsv|tjddd}|dkr.|tjddd}n|tjdddd}|tjtjdddddd}t||g|RS)zI Hypothesis strategy that returns DER OBJECT IDENTIFIER objects. rrWr='i2rC)rErJrIr)rQfirstsecondrestrrr% st_der_oidsrycCsHtjtdtdddBtddBtddBtBtBddd 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,rWi)r?rCcSshtddt|tddt|Btddtj|ddBtddtjdd d t|BS) NcSst|Sr')rr)rrr%r+r,z*st_der....cSs t|dS)Nr)rr)rrr%r+r,cSst|Sr')rr)rrr%r+r,rCcSs t||Sr')r)tagr*rrr%r+"r,r?r=)rErjZone_ofrIrJ)childrenrrr%r+s zst_der..()Z max_leaves)rE recursiverrrmrprqrsryrrrr%st_der s  rcCsH|\}}}tt |j|ttdWdn1s:0YdS)z8Check if random DER structures are rejected as signaturer6Nr\rhderr$r9rcrrr%test_random_der_as_signature*s  rrzrCcCsH|\}}}tt |j|ttdWdn1s:0YdS)z/Check if random bytes are rejected as signaturer6Nr\rrrr%test_random_bytes_as_signature4s  rcCs4g|],\}}}||tt||jjd|jjifqS)rb)rrr.rb)r#r$r9r4rrr%r&Fs  cCsF|\}}tt |j|ttdWdn1s80YdSr5)r2r]rr7r8rrgrrr%test_fuzzed_string_signaturesWs r)L __future__rrr r ImportError functoolsr r2sysZhypothesis.strategies strategiesrEZ hypothesisr r rrrkeysrrutilrrrrrrrrrrrrr8Z hash_and_sizerRsortedr.Zhash_alggenerater!skappendrGr9signmarkZ parametrizer;Z compositerVrh version_inforYZdata_too_largeZfilter_too_muchZtoo_slowdictZ slow_paramsr_rfrirmrprqrsryrrFrrOrZkeys_and_string_sigsrrrrr%s               2      !