mirror of
https://github.com/morgan9e/noiseprotocol
synced 2026-04-14 00:14:05 +09:00
e84db3c232
* Created NoiseBackend class serving as a base for backends
* Refactored NoiseProtocol name parsing
* Refactored existing spec-defined functions into abstract classes.
Implementing classes are connecting crypto primitives to expected
interfaces.
* Refactored existing usage of Cryptography as source of crypto into
"default" backend (along with in-house implementation of X448).
* Provisioned "experimental" backend, it will contain e.g. non-default
crypto algorithms
* Backend can be chosen while creating NoiseConnection, though by
default, the Cryptography backend ("default") is used
Closes #7
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from cryptography.hazmat.primitives.asymmetric import x25519
|
|
|
|
from noise.backends.default.crypto import X448
|
|
from noise.backends.default.keypairs import KeyPair25519, KeyPair448
|
|
from noise.exceptions import NoiseValueError
|
|
from noise.functions.dh import DH
|
|
|
|
|
|
class ED25519(DH):
|
|
@property
|
|
def klass(self):
|
|
return KeyPair25519
|
|
|
|
@property
|
|
def dhlen(self):
|
|
return 32
|
|
|
|
def generate_keypair(self) -> 'KeyPair':
|
|
private_key = x25519.X25519PrivateKey.generate()
|
|
public_key = private_key.public_key()
|
|
return KeyPair25519(private_key, public_key, public_key.public_bytes())
|
|
|
|
def dh(self, private_key, public_key) -> bytes:
|
|
if not isinstance(private_key, x25519.X25519PrivateKey) or not isinstance(public_key, x25519.X25519PublicKey):
|
|
raise NoiseValueError('Invalid keys! Must be x25519.X25519PrivateKey and x25519.X25519PublicKey instances')
|
|
return private_key.exchange(public_key)
|
|
|
|
|
|
class ED448(DH):
|
|
@property
|
|
def klass(self):
|
|
return KeyPair448
|
|
|
|
@property
|
|
def dhlen(self):
|
|
return 56
|
|
|
|
def generate_keypair(self) -> 'KeyPair':
|
|
return KeyPair448.new()
|
|
|
|
def dh(self, private_key, public_key) -> bytes:
|
|
if len(private_key) != self.dhlen or len(public_key) != self.dhlen:
|
|
raise NoiseValueError('Invalid length of keys! Should be {}'.format(self.dhlen))
|
|
return X448.mul(private_key, public_key)
|