o j @sXddlmZddlmZmZmZmZddZddZddZ d d Z d d Z d dZ dS)) long_to_bytes)EccKey construct_import_curve25519_public_key_import_curve448_public_keycCst|j|j}|rtd|jdkrt|jjddd}|S|jdkr0t|jjddd}|St|j| }|S)NzInvalid ECDH point Curve25519 little) byteorderCurve4488) pointQdis_point_at_infinity ValueErrorcurve bytearrayxto_bytesr size_in_bytes)key_privkey_pubpointPzr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/Protocol/DH.py _compute_ecdhs   rcCt|}td|dS)aCreate 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. rrpoint_x)rrencodedrrrrimport_x25519_public_key r"cC t|ddS)aCreate 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. rseedrrr!rrrimport_x25519_private_key, r)cCr)aCreate 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)rrr rrrimport_x448_public_key@r#r+cCr$)aCreate 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(rrrimport_x448_private_keyUr*r,c Ksx|dd}|dd}|dd}|dd}|dd}|dur&tdd}d}d}d d } |dur?| ||dd }|d 7}|durN| ||dd }|d 7}|dur]| ||dd }|d 7}|durl| ||dd }|d 7}||dksz|dksz|dkr~tdd} d} |r|rt||} |r|rt|t|krtdt||} n|r|rt||} n |r|rt||} | | } || S)aPerform 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_pubeph_priveph_pubkdfz'kdf' is mandatoryrcSsZt|ts td||r|std||dur |j}|S||jkr+td||S)Nz'%s' must be an ECC keyz'%s' must be a private ECC keyz('%s' is defined on an incompatible curve) isinstancer TypeError has_privater)rkeynameprivaterrr check_curves     z"key_agreement..check_curveTFz'Too few keys for the ECDH key agreementz"DH mode C(2e, 1s) is not supported)getrrbool) kwargsr-r.r/r0r1 count_priv count_pubrr8ZsZeZrrr key_agreementisN          rDN) Cryptodome.Util.numberrCryptodome.PublicKey.ECCrrrrrr"r)r+r,rDrrrrs