U Mfm @s,ddlmZddlmZddZddZdS)) long_to_bytes)EccKeycCs0|j|j}|rtdt|j|}|S)NzInvalid ECDH point)ZpointQdZis_point_at_infinity ValueErrorrxZ size_in_bytes)Zkey_privZkey_pubZpointPzr:/tmp/pip-unpacked-wheel-l_0d1exj/Cryptodome/Protocol/DH.py _compute_ecdhs  r c Ks|dd}|dd}|dd}|dd}|dd}|dkrLtdd}d}d}d d } |dk r~| ||dd }|d 7}|dk r| ||dd }|d 7}|dk r| ||dd }|d 7}|dk r| ||dd }|d 7}||dks|dks|dkrtdd} d} |r|rt||} |rL|rLt|t|kr@tdt||} n.|rd|rdt||} n|rz|rzt||} | | } || 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 mandatoryrcSsXt|tstd||r.|s.td||dkr>|j}n||jkrTtd||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_privatecurve)rkeynameprivaterrr check_curve2s      z"key_agreement..check_curveTFz'Too few keys for the ECDH key agreementz"DH mode C(2e, 1s) is not supported)getrr bool) kwargsr r r rrZ count_privZ count_pubrrZsZZeZrrr key_agreementsN              rN)ZCryptodome.Util.numberrZCryptodome.PublicKey.ECCrr rrrrr s