U
    g                     @   sX   d dl mZ d dlmZmZmZmZ dd Zdd Zdd Z	d	d
 Z
dd Zdd ZdS )    )long_to_bytes)EccKey	construct_import_curve25519_public_key_import_curve448_public_keyc                 C   sp   |j | j }| rtd| jdkr<t|jjddd}n0| jdkr\t|jjddd}nt|j|	 }|S )NzInvalid ECDH point
Curve25519    little)	byteorderCurve4488   )
ZpointQdZis_point_at_infinity
ValueErrorcurve	bytearrayxto_bytesr   Zsize_in_bytes)Zkey_privZkey_pubZpointPz r   :/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/Protocol/DH.py_compute_ecdh   s    

r   c                 C   s   t | }td|dS )a  Create a new X25519 public key object,
    starting from the key encoded as raw ``bytes``,
    in the format described in RFC7748.

    Args:
      encoded (bytes):
        The x25519 public key to import.
        It must be 32 bytes.

    Returns:
      :class:`Cryptodome.PublicKey.EccKey` : a new ECC key object.

    Raises:
      ValueError: when the given key cannot be parsed.
    r   r   Zpoint_x)r   r   encodedr   r   r   r   import_x25519_public_key   s    r   c                 C   s   t | ddS )a  Create a new X25519 private key object,
    starting from the key encoded as raw ``bytes``,
    in the format described in RFC7748.

    Args:
      encoded (bytes):
        The X25519 private key to import.
        It must be 32 bytes.

    Returns:
      :class:`Cryptodome.PublicKey.EccKey` : a new ECC key object.

    Raises:
      ValueError: when the given key cannot be parsed.
    r   seedr   r   r   r   r   r   import_x25519_private_key,   s    r   c                 C   s   t | }td|dS )a  Create a new X448 public key object,
    starting from the key encoded as raw ``bytes``,
    in the format described in RFC7748.

    Args:
      encoded (bytes):
        The x448 public key to import.
        It must be 56 bytes.

    Returns:
      :class:`Cryptodome.PublicKey.EccKey` : a new ECC key object.

    Raises:
      ValueError: when the given key cannot be parsed.
    r   r   )r   r   r   r   r   r   import_x448_public_key@   s    r    c                 C   s   t | ddS )a  Create a new X448 private key object,
    starting from the key encoded as raw ``bytes``,
    in the format described in RFC7748.

    Args:
      encoded (bytes):
        The X448 private key to import.
        It must be 56 bytes.

    Returns:
      :class:`Cryptodome.PublicKey.EccKey` : a new ECC key object.

    Raises:
      ValueError: when the given key cannot be parsed.
    r   r   r   r   r   r   r   import_x448_private_keyU   s    r!   c                  K   s  |  dd}|  dd}|  dd}|  dd}|  dd}|dkrLtdd}d}d}d	d
 }	|dk	r~|	||dd}|d7 }|dk	r|	||dd}|d7 }|dk	r|	||dd}|d7 }|dk	r|	||dd}|d7 }|| dk s|dks|dkrtdd}
d}|r|rt||}
|rL|rLt|t|kr@tdt||}n.|rd|rdt||}n|rz|rzt||}||
 }||S )a  Perform a Diffie-Hellman key agreement.

    Keywords:
      kdf (callable):
        A key derivation function that accepts ``bytes`` as input and returns
        ``bytes``.
      static_priv (EccKey):
        The local static private key. Optional.
      static_pub (EccKey):
        The static public key that belongs to the peer. Optional.
      eph_priv (EccKey):
        The local ephemeral private key, generated for this session. Optional.
      eph_pub (EccKey):
        The ephemeral public key, received from the peer for this session. Optional.

    At least two keys must be passed, of which one is a private key and one
    a public key.

    Returns (bytes):
      The derived secret key material.
    static_privN
static_pubeph_priveph_pubkdfz'kdf' is mandatoryr   c                 S   sX   t |tstd| |r.| s.td| | d kr>|j} n| |jkrTtd| | S )Nz'%s' must be an ECC keyz'%s' must be a private ECC keyz('%s' is defined on an incompatible curve)
isinstancer   	TypeErrorZhas_privater   )r   keynameprivater   r   r   check_curve   s    

z"key_agreement.<locals>.check_curveT   F   z'Too few keys for the ECDH key agreement    z"DH mode C(2e, 1s) is not supported)getr   r   bool)kwargsr"   r#   r$   r%   r&   Z
count_privZ	count_pubr   r,   ZsZZeZr   r   r   key_agreementi   sN    

r5   N)ZCryptodome.Util.numberr   ZCryptodome.PublicKey.ECCr   r   r   r   r   r   r   r    r!   r5   r   r   r   r   <module>   s   