30 lines
862 B
Python
30 lines
862 B
Python
|
from Crypto.Hash import SHA256
|
||
|
|
|
||
|
|
from libp2p.crypto.keys import KeyType, PublicKey
|
||
|
|
from nacl.public import PublicKey as Ed255129PublicKeyImpl
|
||
|
|
from nacl.signing import BadSignatureError, VerifyKey
|
||
|
|
|
||
|
|
|
||
|
|
class Ed25519PublicKey(PublicKey):
|
||
|
|
def __init__(self, impl: Ed255129PublicKeyImpl) -> None:
|
||
|
|
self.impl = impl
|
||
|
|
|
||
|
|
def to_bytes(self) -> bytes:
|
||
|
|
return bytes(self.impl)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_bytes(cls, key_bytes: bytes) -> "Ed25519PublicKey":
|
||
|
|
return cls(Ed255129PublicKeyImpl(key_bytes))
|
||
|
|
|
||
|
|
def get_type(self) -> KeyType:
|
||
|
|
return KeyType.Ed25519
|
||
|
|
|
||
|
|
def verify(self, data: bytes, signature: bytes) -> bool:
|
||
|
|
verify_key = VerifyKey(self.to_bytes())
|
||
|
|
h = SHA256.new(data)
|
||
|
|
try:
|
||
|
|
verify_key.verify(h, signature)
|
||
|
|
except BadSignatureError:
|
||
|
|
return False
|
||
|
|
return True
|