py-libp2p/libp2p/crypto/key_exchange.py

30 lines
1.0 KiB
Python
Raw Normal View History

2019-08-24 05:43:36 +08:00
from typing import Callable, Tuple, cast
2019-08-23 22:54:59 +08:00
from fastecdsa.encoding import util
2019-08-23 22:54:59 +08:00
from libp2p.crypto.ecc import ECCPrivateKey, ECCPublicKey, create_new_key_pair
2019-08-23 22:54:59 +08:00
from libp2p.crypto.keys import PublicKey
SharedKeyGenerator = Callable[[bytes], bytes]
int_bytelen = util.int_bytelen
2019-08-23 22:54:59 +08:00
def create_ephemeral_key_pair(curve_type: str) -> Tuple[PublicKey, SharedKeyGenerator]:
"""Facilitates ECDH key exchange."""
2019-08-23 22:54:59 +08:00
if curve_type != "P-256":
raise NotImplementedError()
key_pair = create_new_key_pair(curve_type)
def _key_exchange(serialized_remote_public_key: bytes) -> bytes:
2019-08-24 05:43:36 +08:00
private_key = cast(ECCPrivateKey, key_pair.private_key)
remote_point = ECCPublicKey.from_bytes(serialized_remote_public_key, curve_type)
secret_point = remote_point.impl * private_key.impl
secret_x_coordinate = secret_point.x
byte_size = int_bytelen(secret_x_coordinate)
return secret_x_coordinate.to_bytes(byte_size, byteorder="big")
2019-08-23 22:54:59 +08:00
return key_pair.public_key, _key_exchange