mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
110 lines
2.7 KiB
Python
110 lines
2.7 KiB
Python
# Separate all crypto code so that we can easily test several implementations
|
|
|
|
import base64
|
|
|
|
import base58
|
|
import ed25519
|
|
|
|
from bigchaindb.crypto.asymmetric import SigningKey, VerifyingKey
|
|
|
|
|
|
class Ed25519SigningKey(ed25519.SigningKey, SigningKey):
|
|
"""
|
|
PrivateKey instance
|
|
"""
|
|
|
|
def __init__(self, key):
|
|
"""
|
|
Instantiate the private key with the private_value encoded in base58
|
|
|
|
Args:
|
|
key (base58): base58 encoded private key
|
|
"""
|
|
private_base64 = self.decode(key)
|
|
super().__init__(private_base64, encoding='base64')
|
|
|
|
def sign(self, data, encoding="base64"):
|
|
"""
|
|
Sign data with private key
|
|
|
|
Args:
|
|
data (str, bytes): data to sign
|
|
encoding (str): base64, hex
|
|
"""
|
|
if not isinstance(data, bytes):
|
|
data = data.encode('utf-8')
|
|
return super().sign(data, encoding=encoding)
|
|
|
|
@staticmethod
|
|
def encode(private_base64):
|
|
"""
|
|
Encode the base64 number private_base64 to base58
|
|
Args:
|
|
private_base64:
|
|
"""
|
|
return base58.b58encode(base64.b64decode(private_base64))
|
|
|
|
@staticmethod
|
|
def decode(key):
|
|
"""
|
|
Decode the base58 private_value to base64
|
|
|
|
Args:
|
|
key:
|
|
"""
|
|
return base64.b64encode(base58.b58decode(key))
|
|
|
|
|
|
class Ed25519VerifyingKey(ed25519.VerifyingKey, VerifyingKey):
|
|
|
|
def __init__(self, key):
|
|
"""
|
|
Instantiate the public key with the compressed public value encoded in base58
|
|
"""
|
|
public_base64 = self.decode(key)
|
|
super().__init__(public_base64, encoding='base64')
|
|
|
|
def verify(self, data, signature, encoding='base64'):
|
|
try:
|
|
if encoding:
|
|
data = data.encode('utf-8')
|
|
super().verify(signature, data, encoding=encoding)
|
|
except ed25519.BadSignatureError:
|
|
return False
|
|
|
|
return True
|
|
|
|
@staticmethod
|
|
def encode(public_base64):
|
|
"""
|
|
Encode the public key represented by base64 to base58
|
|
|
|
Args:
|
|
public_base64
|
|
"""
|
|
return Ed25519SigningKey.encode(public_base64)
|
|
|
|
@staticmethod
|
|
def decode(public_base58):
|
|
"""
|
|
Decode the base58 public_value to base64
|
|
|
|
Args:
|
|
public_base58
|
|
"""
|
|
return Ed25519SigningKey.decode(public_base58)
|
|
|
|
|
|
def ed25519_generate_key_pair():
|
|
"""
|
|
Generate a new key pair and return the pair encoded in base58
|
|
"""
|
|
sk, vk = ed25519.create_keypair()
|
|
# Private key
|
|
private_value_base58 = base58.b58encode(sk.to_bytes())
|
|
|
|
# Public key
|
|
public_value_compressed_base58 = base58.b58encode(vk.to_bytes())
|
|
|
|
return (private_value_base58, public_value_compressed_base58)
|