mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 06:04:21 +01:00
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import os.path
|
|
from sys import exit
|
|
|
|
from cryptography.hazmat.backends import default_backend
|
|
from cryptography.hazmat.primitives import serialization
|
|
from cryptography.hazmat.primitives.asymmetric import ec
|
|
|
|
|
|
# Simple tool to generate an ECDSA private key using the secp256k1 curve and save private and public keys
|
|
# as 'pisa_sk.der' 'and pisa_pk.der', respectively.
|
|
|
|
SK_FILE_NAME = "../pisa_sk.der"
|
|
PK_FILE_NAME = "../pisa_pk.der"
|
|
|
|
|
|
def save_sk(sk, filename):
|
|
der = sk.private_bytes(
|
|
encoding=serialization.Encoding.DER,
|
|
format=serialization.PrivateFormat.TraditionalOpenSSL,
|
|
encryption_algorithm=serialization.NoEncryption(),
|
|
)
|
|
|
|
with open(filename, "wb") as der_out:
|
|
der_out.write(der)
|
|
|
|
|
|
def save_pk(pk, filename):
|
|
der = pk.public_bytes(encoding=serialization.Encoding.DER, format=serialization.PublicFormat.SubjectPublicKeyInfo)
|
|
with open(filename, "wb") as der_out:
|
|
der_out.write(der)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if os.path.exists(SK_FILE_NAME):
|
|
print('A key with name "{}" already exists. Aborting.'.format(SK_FILE_NAME))
|
|
exit(1)
|
|
|
|
sk = ec.generate_private_key(ec.SECP256K1, default_backend())
|
|
pk = sk.public_key()
|
|
|
|
save_sk(sk, SK_FILE_NAME)
|
|
save_pk(pk, PK_FILE_NAME)
|
|
print('Saved private key "{}" and public key "{}".'.format(SK_FILE_NAME, PK_FILE_NAME))
|