Add ECC key implementation

This commit is contained in:
Alex Stokes 2019-08-23 16:54:16 +02:00
parent 91e11f3ec0
commit 3c97a5a0ed
No known key found for this signature in database
GPG Key ID: 51CE1721B245C086
2 changed files with 56 additions and 0 deletions

55
libp2p/crypto/ecc.py Normal file
View File

@ -0,0 +1,55 @@
from Crypto.PublicKey import ECC
from Crypto.PublicKey.ECC import EccKey
from libp2p.crypto.keys import KeyPair, KeyType, PrivateKey, PublicKey
class ECCPublicKey(PublicKey):
def __init__(self, impl: EccKey) -> None:
self.impl = impl
def to_bytes(self) -> bytes:
return self.impl.export_key("DER")
@classmethod
def from_bytes(cls, data: bytes) -> "ECCPublicKey":
public_key_impl = ECC.import_key(data)
return cls(public_key_impl)
def get_type(self) -> KeyType:
return KeyType.ECC_P256
def verify(self, data: bytes, signature: bytes) -> bool:
raise NotImplementedError
class ECCPrivateKey(PrivateKey):
def __init__(self, impl: EccKey) -> None:
self.impl = impl
@classmethod
def new(cls, curve: str) -> "ECCPrivateKey":
private_key_impl = ECC.generate(curve=curve)
return cls(private_key_impl)
def to_bytes(self) -> bytes:
return self.impl.export_key("DER")
def get_type(self) -> KeyType:
return KeyType.ECC_P256
def sign(self, data: bytes) -> bytes:
raise NotImplementedError
def get_public_key(self) -> PublicKey:
return ECCPublicKey(self.impl.publickey())
def create_new_key_pair(curve: str) -> KeyPair:
"""
Returns a new RSA keypair with the requested key size (``bits``) and the given public
exponent ``e``. Sane defaults are provided for both values.
"""
private_key = ECCPrivateKey.new(curve)
public_key = private_key.get_public_key()
return KeyPair(private_key, public_key)

View File

@ -11,6 +11,7 @@ class KeyType(Enum):
Ed25519 = 1 Ed25519 = 1
Secp256k1 = 2 Secp256k1 = 2
ECDSA = 3 ECDSA = 3
ECC_P256 = 4
class Key(ABC): class Key(ABC):