o 6 hVF@sdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZddlmZd ZejeZd Zd Zejd krId dZnddZejdkrWddZnddZddZdDddZGdddZddZddZddZefd d!Zefd"d#Z d$d%Z!d&d'Z"d(d)Z#d*d+Z$d,d-Z%d.d/Z&d0d1Z'd2d3Z(d4d5Z)d6d7Z*d8d9Z+d:d;Z,Gdd?Z/d@dAZ0dBdCZ1dS)Ea This module includes some utility functions. The methods most typically used are the sigencode and sigdecode functions to be used with :func:`~ecdsa.keys.SigningKey.sign` and :func:`~ecdsa.keys.VerifyingKey.verify` respectively. See the :func:`sigencode_strings`, :func:`sigdecode_string`, :func:`sigencode_der`, :func:`sigencode_strings_canonize`, :func:`sigencode_string_canonize`, :func:`sigencode_der_canonize`, :func:`sigdecode_strings`, :func:`sigdecode_string`, and :func:`sigdecode_der` functions. )divisionN)sha256)PY2int2bytenext)der)normalise_bytes)riHi='r r)rr )rr r r )r cCs&tt|dddt|dS)-Convert a bytestring to string of 0's and 1'sbigr N)binint from_byteszfilllenent_256r^/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/ecdsa/util.pyentropy_to_bits4s&rcCsddd|DS)rcss*|]}tt|dddVqdS)r Nr)rordr).0xrrr <s(z"entropy_to_bits..)joinrrrrr:s)r cCstt|dS)Nr )rrrrrr bit_lengthAr$cCs |pdSNr)r$r#rrrr$F cCsdtd|dS)Nrz%xr )r)orderrrrorderlenJsr)cCsv|dksJ|dur tj}t|d}|dd} ||}t|}t|d|ddd}d|kr9|kr:|Sq)aReturn a random integer k such that 1 <= k < order, uniformly distributed across that range. Worst case should be a mean of 2 loops at (2**k)+2. Note that this function is not declared to be forwards-compatible: we may change the behavior in future releases. The entropy= argument (which should get a callable that behaves like os.urandom) can be used to achieve stability within a given release (for repeatable unit tests), but should not be used as a long-term-compatible key generation algorithm. rNr rT)baser)osurandomr$rr)r(entropyupper_2 upper_256rent_2rand_numrrr randrangeNs   r2c@s$eZdZddZddZddZdS)PRNGcCs|||_dSN)block_generator generator)selfseedrrr__init__lr%z PRNG.__init__cs,fddt|D}trd|St|S)Ncsg|]}tjqSr)rr6)rir7rr psz!PRNG.__call__..r)rangerr!bytes)r7numbytesarr;r__call__os z PRNG.__call__ccs6d} td||fD]}|Vq|d7}q)NrTz prng-%d-%sr)rencodedigest)r7r8counterbyterrrr5wszPRNG.block_generatorN)__name__ __module__ __qualname__r9rAr5rrrrr3fs r3cCsXt|dt|}tt|d|dd}d|kr"|ks*nJd||f|S)Nr r)r3r)rbinasciihexlify)r8r(r*numberrrr%randrange_from_seed__overshoot_modulos$rMcCs d|>dSr&r)numbitsrrr lsb_of_onesr'rOcCs2tt|ddd}|d}|d}|||fS)Nrr r)rmathlog)r(bitsr> extrabitsrrrbits_and_bytess rTcCsvt|\}}}|r |d7}||d|}d|t||}dtt|d}d|kr6|ks9JJ|S)NrrI)rTrCrrrJrK)r8r(hashmodrR_bytesrSr*rLrrr#randrange_from_seed__truncate_bytessrXcCstt|ddd}|dd}||d|}d|t||}d||}|r@tt|dt|@|dd}dtt |d}d|krU|ksXJJ|S)Nrr r"rrUrrI) rrPrQrCrrrrOrJrK)r8r(rVrRmaxbytesr*topbitsrLrrr"randrange_from_seed__truncate_bitss  $r[cCsr|dksJt|\}}}t|} d}|r"tt|dt|@}t|||d}d|kr7|kr8|Sq)NrT)rTr3rrrOstring_to_number)r8r(rRr>rSgenerate extrabyteguessrrr randrange_from_seed__trytryagains racCsNt|}dtd|d}t||}t||ks%Jt||f|SNz%0r r)r)strrJ unhexlifyrBrnumr(lfmt_strstringrrrnumber_to_strings rjcCs:t|}dtd|d}t||}|d|Srb)r)rcrJrdrBrerrrnumber_to_string_crops rkcCstt|dSNrI)rrJrK)rirrrr]r%r]cCs4t|}t||ksJt||ftt|dSrl)r)rrrJrK)rir(rgrrrstring_to_number_fixedlensrmcCst||}t||}||fS)aZ Encode the signature to a pair of strings in a tuple Encodes signature into raw encoding (:term:`raw encoding`) with the ``r`` and ``s`` parts of the signature encoded separately. It's expected that this function will be used as a ``sigencode=`` parameter in :func:`ecdsa.keys.SigningKey.sign` method. :param int r: first parameter of the signature :param int s: second parameter of the signature :param int order: the order of the curve over which the signature was computed :return: raw encoding of ECDSA signature :rtype: tuple(bytes, bytes) )rjrsr(r_strs_strrrrsigencode_stringss  rscCst|||\}}||S)a Encode the signature to raw format (:term:`raw encoding`) It's expected that this function will be used as a ``sigencode=`` parameter in :func:`ecdsa.keys.SigningKey.sign` method. :param int r: first parameter of the signature :param int s: second parameter of the signature :param int order: the order of the curve over which the signature was computed :return: raw encoding of ECDSA signature :rtype: bytes )rsrnrrrsigencode_stringsrtcCstt|t|S)a Encode the signature into the ECDSA-Sig-Value structure using :term:`DER`. Encodes the signature to the following :term:`ASN.1` structure:: Ecdsa-Sig-Value ::= SEQUENCE { r INTEGER, s INTEGER } It's expected that this function will be used as a ``sigencode=`` parameter in :func:`ecdsa.keys.SigningKey.sign` method. :param int r: first parameter of the signature :param int s: second parameter of the signature :param int order: the order of the curve over which the signature was computed :return: DER encoding of ECDSA signature :rtype: bytes )rencode_sequenceencode_integerrorpr(rrr sigencode_dersrxcCs||dkr ||}|S)a8 Internal function for ensuring that the ``s`` value of a signature is in the "canonical" format. :param int s: the second parameter of ECDSA signature :param int order: the order of the curve over which the signatures was computed :return: canonical value of s :rtype: int r r)rpr(rrr _canonize3s rycCt||}t|||S)a Encode the signature to a pair of strings in a tuple Encodes signature into raw encoding (:term:`raw encoding`) with the ``r`` and ``s`` parts of the signature encoded separately. Makes sure that the signature is encoded in the canonical format, where the ``s`` parameter is always smaller than ``order / 2``. Most commonly used in bitcoin. It's expected that this function will be used as a ``sigencode=`` parameter in :func:`ecdsa.keys.SigningKey.sign` method. :param int r: first parameter of the signature :param int s: second parameter of the signature :param int order: the order of the curve over which the signature was computed :return: raw encoding of ECDSA signature :rtype: tuple(bytes, bytes) )ryrsrwrrrsigencode_strings_canonizeDs  r{cCrz)aw Encode the signature to raw format (:term:`raw encoding`) Makes sure that the signature is encoded in the canonical format, where the ``s`` parameter is always smaller than ``order / 2``. Most commonly used in bitcoin. It's expected that this function will be used as a ``sigencode=`` parameter in :func:`ecdsa.keys.SigningKey.sign` method. :param int r: first parameter of the signature :param int s: second parameter of the signature :param int order: the order of the curve over which the signature was computed :return: raw encoding of ECDSA signature :rtype: bytes )ryrtrwrrrsigencode_string_canonize^s  r|cCrz)a9 Encode the signature into the ECDSA-Sig-Value structure using :term:`DER`. Makes sure that the signature is encoded in the canonical format, where the ``s`` parameter is always smaller than ``order / 2``. Most commonly used in bitcoin. Encodes the signature to the following :term:`ASN.1` structure:: Ecdsa-Sig-Value ::= SEQUENCE { r INTEGER, s INTEGER } It's expected that this function will be used as a ``sigencode=`` parameter in :func:`ecdsa.keys.SigningKey.sign` method. :param int r: first parameter of the signature :param int s: second parameter of the signature :param int order: the order of the curve over which the signature was computed :return: DER encoding of ECDSA signature :rtype: bytes )ryrxrwrrrsigencode_der_canonizeus  r}c@seZdZdZdS)MalformedSignatureaB Raised by decoding functions when the signature is malformed. Malformed in this context means that the relevant strings or integers do not match what a signature over provided curve would create. Either because the byte strings have incorrect lengths or because the encoded values are too large. N)rFrGrH__doc__rrrrr~s r~cCsdt|}t|}t|d|kstdd|t|t|d||}t||d|}||fS)a Decoder for :term:`raw encoding` of ECDSA signatures. raw encoding is a simple concatenation of the two integers that comprise the signature, with each encoded using the same amount of bytes depending on curve size/order. It's expected that this function will be used as the ``sigdecode=`` parameter to the :func:`ecdsa.keys.VerifyingKey.verify` method. :param signature: encoded signature :type signature: bytes like object :param order: order of the curve over which the signature was computed :type order: int :raises MalformedSignature: when the encoding of the signature is invalid :return: tuple with decoded ``r`` and ``s`` values of signature :rtype: tuple of ints r zWInvalid length of signature, expected {0} bytes long, provided string is {1} bytes longN)r r)rr~formatrm) signaturer(rgrorprrrsigdecode_stringsrcCst|dkstdt||\}}t|}t|}t|}t||ks/td|t|t||ks?td|t|t||}t||}||fS)a Decode the signature from two strings. First string needs to be a big endian encoding of ``r``, second needs to be a big endian encoding of the ``s`` parameter of an ECDSA signature. It's expected that this function will be used as the ``sigdecode=`` parameter to the :func:`ecdsa.keys.VerifyingKey.verify` method. :param list rs_strings: list of two bytes-like objects, each encoding one parameter of signature :param int order: order of the curve over which the signature was computed :raises MalformedSignature: when the encoding of the signature is invalid :return: tuple with decoded ``r`` and ``s`` values of signature :rtype: tuple of ints r z3Invalid number of strings provided: {0}, expected 2zjInvalid length of first string ('r' parameter), expected {0} bytes long, provided string is {1} bytes longzkInvalid length of second string ('s' parameter), expected {0} bytes long, provided string is {1} bytes long)rr~rr r)rm) rs_stringsr(rqrrrgrorprrrsigdecode_stringss.       rcCsrt|}t|\}}|dkrtdt|t|\}}t|\}}|dkr5tdt|||fS)a Decoder for DER format of ECDSA signatures. DER format of signature is one that uses the :term:`ASN.1` :term:`DER` rules to encode it as a sequence of two integers:: Ecdsa-Sig-Value ::= SEQUENCE { r INTEGER, s INTEGER } It's expected that this function will be used as as the ``sigdecode=`` parameter to the :func:`ecdsa.keys.VerifyingKey.verify` method. :param sig_der: encoded signature :type sig_der: bytes like object :param order: order of the curve over which the signature was computed :type order: int :raises UnexpectedDER: when the encoding of signature is invalid :return: tuple with decoded ``r`` and ``s`` values of signature :rtype: tuple of ints r\ztrailing junk after DER sig: %sz#trailing junk after DER numbers: %s)r rremove_sequence UnexpectedDERrJrKremove_integer)sig_derr(remptyrorestrprrr sigdecode_ders  rr4)2r __future__rr+rPrJsyshashlibrsixrrrrr_compatr oid_ecPublicKey encode_oidencoded_oid_ecPublicKeyoid_ecDH oid_ecMQV version_inforr$r)r2r3rMrOrTrXr[rarjrkr]rmrsrtrxryr{r|r} Exceptionr~rrrrrrrsX              ! .