forked from STI/Astroport.ONE
119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
|
# This file is dual licensed under the terms of the Apache License, Version
|
||
|
# 2.0, and the BSD License. See the LICENSE file in the root of this repository
|
||
|
# for complete details.
|
||
|
|
||
|
from __future__ import annotations
|
||
|
|
||
|
import abc
|
||
|
|
||
|
from cryptography.exceptions import UnsupportedAlgorithm, _Reasons
|
||
|
from cryptography.hazmat.bindings._rust import openssl as rust_openssl
|
||
|
from cryptography.hazmat.primitives import _serialization
|
||
|
|
||
|
|
||
|
class Ed448PublicKey(metaclass=abc.ABCMeta):
|
||
|
@classmethod
|
||
|
def from_public_bytes(cls, data: bytes) -> Ed448PublicKey:
|
||
|
from cryptography.hazmat.backends.openssl.backend import backend
|
||
|
|
||
|
if not backend.ed448_supported():
|
||
|
raise UnsupportedAlgorithm(
|
||
|
"ed448 is not supported by this version of OpenSSL.",
|
||
|
_Reasons.UNSUPPORTED_PUBLIC_KEY_ALGORITHM,
|
||
|
)
|
||
|
|
||
|
return rust_openssl.ed448.from_public_bytes(data)
|
||
|
|
||
|
@abc.abstractmethod
|
||
|
def public_bytes(
|
||
|
self,
|
||
|
encoding: _serialization.Encoding,
|
||
|
format: _serialization.PublicFormat,
|
||
|
) -> bytes:
|
||
|
"""
|
||
|
The serialized bytes of the public key.
|
||
|
"""
|
||
|
|
||
|
@abc.abstractmethod
|
||
|
def public_bytes_raw(self) -> bytes:
|
||
|
"""
|
||
|
The raw bytes of the public key.
|
||
|
Equivalent to public_bytes(Raw, Raw).
|
||
|
"""
|
||
|
|
||
|
@abc.abstractmethod
|
||
|
def verify(self, signature: bytes, data: bytes) -> None:
|
||
|
"""
|
||
|
Verify the signature.
|
||
|
"""
|
||
|
|
||
|
@abc.abstractmethod
|
||
|
def __eq__(self, other: object) -> bool:
|
||
|
"""
|
||
|
Checks equality.
|
||
|
"""
|
||
|
|
||
|
|
||
|
if hasattr(rust_openssl, "ed448"):
|
||
|
Ed448PublicKey.register(rust_openssl.ed448.Ed448PublicKey)
|
||
|
|
||
|
|
||
|
class Ed448PrivateKey(metaclass=abc.ABCMeta):
|
||
|
@classmethod
|
||
|
def generate(cls) -> Ed448PrivateKey:
|
||
|
from cryptography.hazmat.backends.openssl.backend import backend
|
||
|
|
||
|
if not backend.ed448_supported():
|
||
|
raise UnsupportedAlgorithm(
|
||
|
"ed448 is not supported by this version of OpenSSL.",
|
||
|
_Reasons.UNSUPPORTED_PUBLIC_KEY_ALGORITHM,
|
||
|
)
|
||
|
|
||
|
return rust_openssl.ed448.generate_key()
|
||
|
|
||
|
@classmethod
|
||
|
def from_private_bytes(cls, data: bytes) -> Ed448PrivateKey:
|
||
|
from cryptography.hazmat.backends.openssl.backend import backend
|
||
|
|
||
|
if not backend.ed448_supported():
|
||
|
raise UnsupportedAlgorithm(
|
||
|
"ed448 is not supported by this version of OpenSSL.",
|
||
|
_Reasons.UNSUPPORTED_PUBLIC_KEY_ALGORITHM,
|
||
|
)
|
||
|
|
||
|
return rust_openssl.ed448.from_private_bytes(data)
|
||
|
|
||
|
@abc.abstractmethod
|
||
|
def public_key(self) -> Ed448PublicKey:
|
||
|
"""
|
||
|
The Ed448PublicKey derived from the private key.
|
||
|
"""
|
||
|
|
||
|
@abc.abstractmethod
|
||
|
def sign(self, data: bytes) -> bytes:
|
||
|
"""
|
||
|
Signs the data.
|
||
|
"""
|
||
|
|
||
|
@abc.abstractmethod
|
||
|
def private_bytes(
|
||
|
self,
|
||
|
encoding: _serialization.Encoding,
|
||
|
format: _serialization.PrivateFormat,
|
||
|
encryption_algorithm: _serialization.KeySerializationEncryption,
|
||
|
) -> bytes:
|
||
|
"""
|
||
|
The serialized bytes of the private key.
|
||
|
"""
|
||
|
|
||
|
@abc.abstractmethod
|
||
|
def private_bytes_raw(self) -> bytes:
|
||
|
"""
|
||
|
The raw bytes of the private key.
|
||
|
Equivalent to private_bytes(Raw, Raw, NoEncryption()).
|
||
|
"""
|
||
|
|
||
|
|
||
|
if hasattr(rust_openssl, "x448"):
|
||
|
Ed448PrivateKey.register(rust_openssl.ed448.Ed448PrivateKey)
|