67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
import os
|
|
from cryptography.hazmat.backends import default_backend
|
|
from cryptography.hazmat.primitives import serialization
|
|
from cryptography.hazmat.primitives.asymmetric import rsa
|
|
from encrypt import encrypt_message
|
|
from decrypt import decrypt_message
|
|
|
|
KEYS_DIR = 'keys'
|
|
PRIVATE_KEY_FILE = f'{KEYS_DIR}/private-key.pem'
|
|
PUBLIC_KEY_FILE = f'{KEYS_DIR}/public-key.pem'
|
|
|
|
if not os.path.exists(KEYS_DIR):
|
|
os.makedirs(KEYS_DIR)
|
|
|
|
|
|
def generate_keys():
|
|
private_key = rsa.generate_private_key(
|
|
public_exponent=65537,
|
|
key_size=2048,
|
|
backend=default_backend()
|
|
)
|
|
public_key = private_key.public_key()
|
|
return private_key, public_key
|
|
|
|
|
|
def save_keys(private_key, public_key):
|
|
# Сохранение закрытого ключа
|
|
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()
|
|
))
|
|
|
|
# Сохранение открытого ключа
|
|
with open(PUBLIC_KEY_FILE, 'wb') as f:
|
|
f.write(public_key.public_bytes(
|
|
encoding=serialization.Encoding.PEM,
|
|
format=serialization.PublicFormat.SubjectPublicKeyInfo
|
|
))
|
|
|
|
|
|
# Запрос выбора действия
|
|
action = input("Хотите сгенерировать новые ключи (1) или использовать существующие (2)? ")
|
|
|
|
if action == '1':
|
|
# Генерация новых ключей
|
|
private_key, public_key = generate_keys()
|
|
save_keys(private_key, public_key)
|
|
print(f"Новые ключи сгенерированы и сохранены в '{PRIVATE_KEY_FILE}' и '{PUBLIC_KEY_FILE}'.")
|
|
elif action == '2':
|
|
# Загрузка существующих ключей
|
|
if os.path.exists(PRIVATE_KEY_FILE) and os.path.exists(PUBLIC_KEY_FILE):
|
|
pass
|
|
else:
|
|
print(f"Файлы ключей не найдены. Пожалуйста, сгенерируйте новые ключи.")
|
|
exit()
|
|
|
|
# Шифрование сообщения
|
|
message = "Это лицензионный ключ."
|
|
encrypted_message = encrypt_message(message, PUBLIC_KEY_FILE)
|
|
print("Зашифрованное сообщение (Base64):", encrypted_message)
|
|
|
|
# Дешифрование сообщения
|
|
decrypted_message = decrypt_message(encrypted_message, PRIVATE_KEY_FILE)
|
|
print("Расшифрованное сообщение:", decrypted_message)
|