aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2025-12-17 06:04:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Captures TypeErrors in Cryptographer.get_compressed_pk and adds unit tests

Browse Source
This commit is contained in:
Sergi Delgado Segura
2020-03-27 16:41:19 +01:00
parent 7e6a4849e6
commit ae68cd33da
2 changed files with 32 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
common/cryptographer.py
View File

@@ -315,7 +315,7 @@ class Cryptographer:
if "failed to recover ECDSA public key" in str(e):
logger.error("Cannot recover public key from signature".format(type(rsig_recid)))
else:
logger.error("Unknown exception", error=e)
logger.error("Unknown exception", error=str(e))
return None
@@ -334,7 +334,6 @@ class Cryptographer:
return pk.point() == rpk.point()
# TODO: UNITTEST
@staticmethod
def get_compressed_pk(pk):
"""
@@ -351,4 +350,10 @@ class Cryptographer:
logger.error("The received data is not a PublicKey object")
return None
return hexlify(pk.format(compressed=True)).decode("utf-8")
try:
compressed_pk = pk.format(compressed=True)
return hexlify(compressed_pk).decode("utf-8")
except TypeError as e:
logger.error("PublicKey has invalid initializer", error=str(e))
return None