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
81 lines
1.6 KiB
Python
81 lines
1.6 KiB
Python
import abc
|
|
from functools import partial
|
|
|
|
from cryptography.hazmat.backends import default_backend
|
|
from cryptography.hazmat.primitives import hashes
|
|
from cryptography.hazmat.primitives.hmac import HMAC
|
|
|
|
from noise.functions.hash import Hash
|
|
|
|
cryptography_backend = default_backend()
|
|
|
|
|
|
class CryptographyHash(Hash, metaclass=abc.ABCMeta):
|
|
def hash(self, data):
|
|
digest = hashes.Hash(self.fn(), cryptography_backend)
|
|
digest.update(data)
|
|
return digest.finalize()
|
|
|
|
|
|
class SHA256Hash(CryptographyHash):
|
|
@property
|
|
def fn(self):
|
|
return hashes.SHA256
|
|
|
|
@property
|
|
def hashlen(self):
|
|
return 32
|
|
|
|
@property
|
|
def blocklen(self):
|
|
return 64
|
|
|
|
|
|
class SHA512Hash(CryptographyHash):
|
|
@property
|
|
def fn(self):
|
|
return hashes.SHA512
|
|
|
|
@property
|
|
def hashlen(self):
|
|
return 64
|
|
|
|
@property
|
|
def blocklen(self):
|
|
return 128
|
|
|
|
|
|
class BLAKE2sHash(CryptographyHash):
|
|
@property
|
|
def fn(self):
|
|
return partial(hashes.BLAKE2s, digest_size=self.hashlen)
|
|
|
|
@property
|
|
def hashlen(self):
|
|
return 32
|
|
|
|
@property
|
|
def blocklen(self):
|
|
return 64
|
|
|
|
|
|
class BLAKE2bHash(CryptographyHash):
|
|
@property
|
|
def fn(self):
|
|
return partial(hashes.BLAKE2b, digest_size=self.hashlen)
|
|
|
|
@property
|
|
def hashlen(self):
|
|
return 64
|
|
|
|
@property
|
|
def blocklen(self):
|
|
return 128
|
|
|
|
|
|
def hmac_hash(key, data, algorithm):
|
|
# Applies HMAC using the HASH() function.
|
|
hmac = HMAC(key=key, algorithm=algorithm(), backend=cryptography_backend)
|
|
hmac.update(data=data)
|
|
return hmac.finalize()
|