Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Use proper elliptic_curve element class

pull/73/head
tuxxy 2017-10-05 19:02:07 -06:00
parent 59d3b37440
commit 484e4908a3
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
nkms/crypto/crypto.py
View File

@ -10,26 +10,26 @@ class Crypto(object):
@staticmethod @staticmethod
def priv_bytes2ec( def priv_bytes2ec(
privkey: bytes privkey: bytes
) -> elliptic_curve.Element: ) -> elliptic_curve.ec_element:
""" """
Turns a private key, in bytes, into an elliptic_curve.Element. Turns a private key, in bytes, into an elliptic_curve.ec_element.
:param privkey: Private key to turn into an elliptic_curve.Element. :param privkey: Private key to turn into an elliptic_curve.ec_element.
:return: elliptic_curve.Element :return: elliptic_curve.ec_element
""" """
return elliptic_curve.deserialize(Crypto.PRE.ecgroup, b'\x00' + privkey) return elliptic_curve.deserialize(Crypto.PRE.ecgroup, b'\x00' + privkey)
@staticmethod @staticmethod
def pub_bytes2ec( def pub_bytes2ec(
pubkey: bytes, pubkey: bytes,
) -> elliptic_curve.Element: ) -> elliptic_curve.ec_element:
""" """
Turns a public key, in bytes, into an elliptic_curve.Element. Turns a public key, in bytes, into an elliptic_curve.ec_element.
:param pubkey: Public key to turn into an elliptic_curve.Element. :param pubkey: Public key to turn into an elliptic_curve.ec_element.
:return: elliptic_curve.Element :return: elliptic_curve.ec_element
""" """
return elliptic_curve.deserialize(Crypto.PRE.ecgroup, b'\x01' + pubkey) return elliptic_curve.deserialize(Crypto.PRE.ecgroup, b'\x01' + pubkey)
@ -68,7 +68,7 @@ class Crypto(object):
@staticmethod @staticmethod
def ecies_gen_priv( def ecies_gen_priv(
to_bytes: bool = True to_bytes: bool = True
) -> Union[bytes, elliptic_curve.Element]: ) -> Union[bytes, elliptic_curve.ec_element]:
""" """
Generates an ECIES private key. Generates an ECIES private key.
@ -83,11 +83,11 @@ class Crypto(object):
@staticmethod @staticmethod
def ecies_priv2pub( def ecies_priv2pub(
privkey: Union[bytes, elliptic_curve.Element], privkey: Union[bytes, elliptic_curve.ec_element],
to_bytes: bool = True to_bytes: bool = True
) -> Union[bytes, elliptic_curve.Element]: ) -> Union[bytes, elliptic_curve.ec_element]:
""" """
Takes a private key (secret bytes or an elliptic_curve.Element) and Takes a private key (secret bytes or an elliptic_curve.ec_element) and
derives the Public key from it. derives the Public key from it.
:param privkey: The Private key to derive the public key from :param privkey: The Private key to derive the public key from
@ -105,7 +105,7 @@ class Crypto(object):
@staticmethod @staticmethod
def ecies_encapsulate( def ecies_encapsulate(
pubkey: Union[bytes, elliptic_curve.Element], pubkey: Union[bytes, elliptic_curve.ec_element],
) -> Tuple[bytes, umbral.EncryptedKey]: ) -> Tuple[bytes, umbral.EncryptedKey]:
""" """
Encapsulates an ECIES generated symmetric key for a public key. Encapsulates an ECIES generated symmetric key for a public key.
@ -121,7 +121,7 @@ class Crypto(object):
@staticmethod @staticmethod
def ecies_decapsulate( def ecies_decapsulate(
privkey: Union[bytes, elliptic_curve.Element], privkey: Union[bytes, elliptic_curve.ec_element],
enc_key: umbral.EncryptedKey enc_key: umbral.EncryptedKey
) -> bytes: ) -> bytes:
""" """
@ -138,8 +138,8 @@ class Crypto(object):
@staticmethod @staticmethod
def ecies_rekey( def ecies_rekey(
privkey_a: Union[bytes, elliptic_curve.Element], privkey_a: Union[bytes, elliptic_curve.ec_element],
privkey_b: Union[bytes, elliptic_curve.Element], privkey_b: Union[bytes, elliptic_curve.ec_element],
to_bytes: bool = True to_bytes: bool = True
) -> Union[bytes, umbral.RekeyFrag]: ) -> Union[bytes, umbral.RekeyFrag]:
""" """
@ -163,8 +163,8 @@ class Crypto(object):
@staticmethod @staticmethod
def ecies_split_rekey( def ecies_split_rekey(
privkey_a: Union[bytes, elliptic_curve.Element], privkey_a: Union[bytes, elliptic_curve.ec_element],
privkey_b: Union[bytes, elliptic_curve.Element], privkey_b: Union[bytes, elliptic_curve.ec_element],
min_shares: int, min_shares: int,
total_shares: int total_shares: int
) -> List[umbral.RekeyFrag]: ) -> List[umbral.RekeyFrag]: