varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-08 06:00:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add public key implementation

Browse Source
This commit is contained in:
Alex Stokes
2019-09-09 20:23:44 -04:00
parent fa7d1d66a8
commit 5fdca2ffb2
1 changed files with 40 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
libp2p/crypto/ed25519.py
View File

@ -1,13 +1,15 @@
from Crypto.Hash import SHA256 from Crypto.Hash import SHA256
from libp2p.crypto.keys import KeyType, PublicKey from libp2p.crypto.keys import KeyPair, KeyType, PrivateKey, PublicKey
from nacl.exceptions import BadSignatureError from nacl.exceptions import BadSignatureError
from nacl.public import PublicKey as Ed255129PublicKeyImpl from nacl.public import PrivateKey as PrivateKeyImpl
from nacl.signing import VerifyKey from nacl.public import PublicKey as PublicKeyImpl
from nacl.signing import SigningKey, VerifyKey
import nacl.utils as utils
class Ed25519PublicKey(PublicKey): class Ed25519PublicKey(PublicKey):
def __init__(self, impl: Ed255129PublicKeyImpl) -> None: def __init__(self, impl: PublicKeyImpl) -> None:
self.impl = impl self.impl = impl
def to_bytes(self) -> bytes: def to_bytes(self) -> bytes:
@ -15,7 +17,7 @@ class Ed25519PublicKey(PublicKey):
@classmethod @classmethod
def from_bytes(cls, key_bytes: bytes) -> "Ed25519PublicKey": def from_bytes(cls, key_bytes: bytes) -> "Ed25519PublicKey":
return cls(Ed255129PublicKeyImpl(key_bytes)) return cls(PublicKeyImpl(key_bytes))
def get_type(self) -> KeyType: def get_type(self) -> KeyType:
return KeyType.Ed25519 return KeyType.Ed25519
@ -28,3 +30,36 @@ class Ed25519PublicKey(PublicKey):
except BadSignatureError: except BadSignatureError:
return False return False
return True return True
class Ed25519PrivateKey(PrivateKey):
def __init__(self, impl: PrivateKeyImpl) -> None:
self.impl = impl
@classmethod
def new(cls, seed: bytes = None) -> "Ed25519PrivateKey":
if not seed:
seed = utils.random()
private_key_impl = PrivateKeyImpl.from_seed(seed)
return cls(private_key_impl)
def to_bytes(self) -> bytes:
return bytes(self.impl)
def get_type(self) -> KeyType:
return KeyType.Ed25519
def sign(self, data: bytes) -> bytes:
h = SHA256.new(data)
signing_key = SigningKey(self.to_bytes())
return signing_key.sign(h)
def get_public_key(self) -> PublicKey:
return Ed25519PublicKey(self.impl.public_key)
def create_new_key_pair(seed: bytes = None) -> KeyPair:
private_key = Ed25519PrivateKey.new(seed)
public_key = private_key.get_public_key()
return KeyPair(private_key, public_key)