crypto/generate_save.py

43 lines
1.3 KiB
Python

from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.primitives.asymmetric import rsa
def generate_private_key():
private_key = rsa.generate_private_key(
public_exponent=65537,
key_size=2048,
backend=default_backend()
)
return private_key
def generate_public_key(private_key):
public_key = private_key.public_key()
return public_key
def save_private_key(private_key, private_key_file):
with open(private_key_file, 'wb') as f:
f.write(private_key.private_bytes(
encoding=serialization.Encoding.PEM,
format=serialization.PrivateFormat.TraditionalOpenSSL,
encryption_algorithm=serialization.NoEncryption()
))
def save_public_key(public_key, public_key_file):
with open(public_key_file, 'wb') as f:
f.write(public_key.public_bytes(
encoding=serialization.Encoding.PEM,
format=serialization.PublicFormat.SubjectPublicKeyInfo
))
# Пример использования
if __name__ == "__main__":
private_key = generate_private_key()
public_key = generate_public_key(private_key)
save_private_key(private_key, 'private_key.pem')
save_public_key(public_key, 'public_key.pem')