o

�h+�@s�dZd
dl
mZ
d
dlmZ
d
dlmZmZ
gd�
ZGdd�de	�Z
Gdd	�d	e	�ZGd
d�de	�ZGdd
�d
e	�Z
Gdd�de�ZdS)zG
Class for performing Elliptic-curve Diffie-Hellman (ECDH) operations.
�
)
�number_to_string)
�INFINITY)�
SigningKey�VerifyingKey)�ECDH�
NoKeyError�NoCurveError�InvalidCurveError�InvalidSharedSecretErrorc
@�eZ
dZd
ZdS)rz3ECDH. Key not found but it is needed for operation.N��__name__�
__module__�__qualname__�__doc__�rr�=/usr/local/CyberCP/lib/python3.10/site-packages/ecdsa/ecdh.pyr�
rc
@r)rz3ECDH. Curve not set but it is needed for operation.Nrrrrrrrrc
@r)r	zP
    ECDH. Raised in case the public and private keys use different curves.
    Nrrrrrr	s
r	c
@r)r
zBECDH. Raised in case the shared secret we obtained is an INFINITY.Nrrrrrr
'rr
c@s�eZ
dZd
Zd!dd�
Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
	d"dd�
Zdd�Zdd�Zdd�Zdd �ZdS)#rz�
    Elliptic-curve Diffie-Hellman (ECDH). A key agreement protocol.

    Allows two parties, each having an elliptic-curve public-private key
    pair, to establish a shared secret over an insecure channel
    NcCs6|
|_d
|_
d
|_|r|�|�

|r|�|�

d
Sd
S)a

        ECDH init.

        Call can be initialised without parameters, then the first operation
        (loading either key) will set the used curve.
        All parameters must be ultimately set before shared secret
        calculation will be allowed.

        :param curve: curve for operations
        :type curve: Curve
        :param private_key: `my` private key for ECDH
        :type private_key: SigningKey
        :param public_key:  `their` public key for ECDH
        :type public_key: VerifyingKey
        N)�curve�private_key�
public_key�load_private_key�load_received_public_key)�selfrrrrrr�__init__5s






�z
ECDH.__init__cCsr|jst
d
�
�
|jst
d�
�
|jj|jkr|
jks$td�
�

td�
�
|
jj|jjj}|t	kr5t
d�
�
|��S)Nz3Private key needs to be set to create shared secretz2Public key needs to be set to create shared secretz3Curves for public key and private key is not equal.z!Invalid shared secret (INFINITY).)rrrrr	�pubkey�point�privkey�secret_multiplierrr
�
x)r�remote_public_key�resultrrr�_get_shared_secretMs.


�

�
��
�
��
zECDH._get_shared_secretcCs
|
|_d
S)z�
        Set the working curve for ecdh operations.

        :param key_curve: curve from `curves` module
        :type key_curve: Curve
        N�
r)r�	key_curverrr�	set_curvegs
zECDH.set_curvec

Cs"|jst
d
�
�
|�tj|jd�
�
S)z�
        Generate local private key for ecdh operation with curve that was set.

        :raises NoCurveError: Curve must be set before key generation.

        :return: public (verifying) key from this private key.
        :rtype: VerifyingKey
        z*Curve must be set prior to key generation.r#)rrrr�generate�
rrrr�generate_private_keyps	

zECDH.generate_private_keycCs2|js|
j|_|j|
jkrt
d
�
�
|
|_|j��S)a�

        Load private key from SigningKey (keys.py) object.

        Needs to have the same curve as was set with set_curve method.
        If curve is not set - it sets from this SigningKey

        :param private_key: Initialised SigningKey class
        :type private_key: SigningKey

        :raises InvalidCurveError: private_key curve not the same as self.curve

        :return: public (verifying) key from this private key.
        :rtype: VerifyingKey
        �Curve mismatch.)rr	r�get_verifying_key�rrrrrr}s





zECDH.load_private_keycCs$|jst
d
�
�
|�tj|
|jd��
S)a�

        Load private key from byte string.

        Uses current curve and checks if the provided key matches
        the curve of ECDH key agreement.
        Key loads via from_string method of SigningKey class

        :param private_key: private key in bytes string format
        :type private_key: :term:`bytes-like object`

        :raises NoCurveError: Curve must be set before loading.

        :return: public (verifying) key from this private key.
        :rtype: VerifyingKey
        z$Curve must be set prior to key load.r#)rrrr�from_stringr+rrr�load_private_key_bytes�s



�zECDH.load_private_key_bytescC�|�t
�|
�
�
S)
aC
        Load private key from DER byte string.

        Compares the curve of the DER-encoded key with the ECDH set curve,
        uses the former if unset.

        Note, the only DER format supported is the RFC5915
        Look at keys.py:SigningKey.from_der()

        :param private_key_der: string with the DER encoding of private ECDSA
            key
        :type private_key_der: string

        :raises InvalidCurveError: private_key curve not the same as self.curve

        :return: public (verifying) key from this private key.
        :rtype: VerifyingKey
        )rr�from_der)r�private_key_derrrr�load_private_key_der��zECDH.load_private_key_dercCr.)
a\
        Load private key from PEM string.

        Compares the curve of the DER-encoded key with the ECDH set curve,
        uses the former if unset.

        Note, the only PEM format supported is the RFC5915
        Look at keys.py:SigningKey.from_pem()
        it needs to have `EC PRIVATE KEY` section

        :param private_key_pem: string with PEM-encoded private ECDSA key
        :type private_key_pem: string

        :raises InvalidCurveError: private_key curve not the same as self.curve

        :return: public (verifying) key from this private key.
        :rtype: VerifyingKey
        )rr�from_pem)r�private_key_pemrrr�load_private_key_pem�r2zECDH.load_private_key_pemc

Cs
|j�
�S)
z�
        Provides a public key that matches the local private key.

        Needs to be sent to the remote party.

        :return: public (verifying) key from local private key.
        :rtype: VerifyingKey
        )rr*r'rrr�get_public_key�s
	zECDH.get_public_keycCs,|js|
j|_|j|
jkrt
d
�
�
|
|_dS)a

        Load public key from VerifyingKey (keys.py) object.

        Needs to have the same curve as set as current for ecdh operation.
        If curve is not set - it sets it from VerifyingKey.

        :param public_key: Initialised VerifyingKey class
        :type public_key: VerifyingKey

        :raises InvalidCurveError: public_key curve not the same as self.curve
        r)N)rr	r)rrrrrr�s





zECDH.load_received_public_keycCs|�t
�|
|j|��
S)
a�
        Load public key from byte string.

        Uses current curve and checks if key length corresponds to
        the current curve.
        Key loads via from_string method of VerifyingKey class

        :param public_key_str: public key in bytes string format
        :type public_key_str: :term:`bytes-like object`
        :param valid_encodings: list of acceptable point encoding formats,
            supported ones are: :term:`uncompressed`, :term:`compressed`,
            :term:`hybrid`, and :term:`raw encoding` (specified with ``raw``
            name). All formats by default (specified with ``None``).
        :type valid_encodings: :term:`set-like object`
        )rrr,r)r�public_key_str�valid_encodingsrrr�load_received_public_key_bytes�s


��z#ECDH.load_received_public_key_bytescCr.)
a�

        Load public key from DER byte string.

        Compares the curve of the DER-encoded key with the ECDH set curve,
        uses the former if unset.

        Note, the only DER format supported is the RFC5912
        Look at keys.py:VerifyingKey.from_der()

        :param public_key_der: string with the DER encoding of public ECDSA key
        :type public_key_der: string

        :raises InvalidCurveError: public_key curve not the same as self.curve
        )rrr/)r�public_key_derrrr�load_received_public_key_der
�
�z!ECDH.load_received_public_key_dercCr.)
a�

        Load public key from PEM string.

        Compares the curve of the PEM-encoded key with the ECDH set curve,
        uses the former if unset.

        Note, the only PEM format supported is the RFC5912
        Look at keys.py:VerifyingKey.from_pem()

        :param public_key_pem: string with PEM-encoded public ECDSA key
        :type public_key_pem: string

        :raises InvalidCurveError: public_key curve not the same as self.curve
        )rrr3)r�public_key_pemrrr�load_received_public_key_pem
r<z!ECDH.load_received_public_key_pemc

Cst|�
�|jjj���S)
a�

        Generate shared secret from local private key and remote public key.

        The objects needs to have both private key and received public key
        before generation is allowed.

        :raises InvalidCurveError: public_key curve not the same as self.curve
        :raises NoKeyError: public_key or private_key is not set

        :return: shared secret
        :rtype: bytes
        )r�generate_sharedsecretrr�
pr'rrr�generate_sharedsecret_bytes.
s

�z ECDH.generate_sharedsecret_bytesc

Cs|�|j
�
S)
a;
        Generate shared secret from local private key and remote public key.

        The objects needs to have both private key and received public key
        before generation is allowed.

        It's the same for local and remote party,
        shared secret(local private key, remote public key) ==
        shared secret(local public key, remote private key)

        :raises InvalidCurveError: public_key curve not the same as self.curve
        :raises NoKeyError: public_key or private_key is not set

        :return: shared secret
        :rtype: int
        )r"rr'rrrr??
szECDH.generate_sharedsecret)NNN)
N)r
rrrrr"r%r(rr-r1r5r6rr9r;r>rAr?rrrrr-s$

	

�rN)r�utilr�
ellipticcurver�keysrr�__all__�	Exceptionrrr	r
�objectrrrrr�<module>
s