diff --git a/main.py b/main.py index dc45e01..1c17923 100644 --- a/main.py +++ b/main.py @@ -2,9 +2,13 @@ from cryptography.hazmat.primitives import hashes from cryptography.hazmat.backends import default_backend from cryptography.exceptions import InvalidKey, InvalidTag -from ed25519 import BadSignatureError + + +from nacl.signing import VerifyKey +from nacl.exceptions import BadSignatureError + import argparse -import ed25519 +#import ed25519 import base64 import json import sys @@ -47,14 +51,13 @@ # Verify using Ed25519 try: - verify_key = ed25519.VerifyingKey( - os.environ['KEYGEN_PUBLIC_KEY'].encode(), - encoding='hex', + verify_key = VerifyKey( + bytes.fromhex(os.environ['KEYGEN_PUBLIC_KEY']) ) verify_key.verify( - base64.b64decode(sig), ('license/%s' % enc).encode(), + base64.b64decode(sig), ) except (AssertionError, BadSignatureError): print('[error] verification failed!') diff --git a/requirements.txt b/requirements.txt index 45dcc28..5705f08 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ argparse cryptography -ed25519 +#ed25519 +pynacl \ No newline at end of file