120 lines
3.7 KiB
Python
120 lines
3.7 KiB
Python
import os
|
|
import base64
|
|
from cryptography.hazmat.backends import default_backend
|
|
from cryptography.hazmat.primitives import serialization, hashes
|
|
from cryptography.hazmat.primitives.asymmetric import rsa, padding
|
|
|
|
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
|
|
))
|
|
|
|
|
|
def load_keys_private():
|
|
# Загрузка закрытого ключа
|
|
with open(PRIVATE_KEY_FILE, 'rb') as f:
|
|
private_key_data = f.read()
|
|
|
|
private_key = serialization.load_pem_private_key(
|
|
private_key_data,
|
|
password=None,
|
|
backend=default_backend()
|
|
)
|
|
|
|
return private_key
|
|
|
|
|
|
def load_keys_public():
|
|
# Загрузка открытого ключа
|
|
with open(PUBLIC_KEY_FILE, 'rb') as f:
|
|
public_key_data = f.read()
|
|
|
|
public_key = serialization.load_pem_public_key(
|
|
public_key_data,
|
|
backend=default_backend()
|
|
)
|
|
|
|
return public_key
|
|
|
|
|
|
def encrypt_message(message):
|
|
public_key = load_keys_public()
|
|
encrypted = public_key.encrypt(
|
|
message.encode(),
|
|
padding.OAEP(
|
|
mgf=padding.MGF1(algorithm=hashes.SHA256()),
|
|
algorithm=hashes.SHA256(),
|
|
label=None
|
|
)
|
|
)
|
|
return base64.b64encode(encrypted).decode('utf-8')
|
|
|
|
|
|
def decrypt_message(encrypted_message):
|
|
private_key = load_keys_private()
|
|
encrypted_bytes = base64.b64decode(encrypted_message)
|
|
decrypted = private_key.decrypt(
|
|
encrypted_bytes,
|
|
padding.OAEP(
|
|
mgf=padding.MGF1(algorithm=hashes.SHA256()),
|
|
algorithm=hashes.SHA256(),
|
|
label=None
|
|
)
|
|
)
|
|
return decrypted.decode()
|
|
|
|
|
|
# Запрос выбора действия
|
|
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)
|
|
print("Зашифрованное сообщение (Base64):", encrypted_message)
|
|
|
|
# Дешифрование сообщения
|
|
decrypted_message = decrypt_message(encrypted_message)
|
|
print("Расшифрованное сообщение:", decrypted_message)
|