shkatov/crypto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Добавлено декодирование словаря

Browse Source
This commit is contained in:
stirelshka8_BigARM 2025-03-02 17:52:42 +03:00
parent b26540d68a
commit e16fc55ce2
1 changed files with 9 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
main.py
View File

@ -1,45 +1,35 @@
"""
This is a simple RSA encryptor/decryptor.
It uses a key pair which is stored in a 'keys' directory.
The keys are generated when you run the script.
The script will encrypt a message for you if you provide it.
The script will decrypt a message for you if you provide it.
The script will generate a key pair and save it in the 'keys' directory.
"""
import os import os
import json
from RSA_Crypto.encrypt import RSAEncryptor from RSA_Crypto.encrypt import RSAEncryptor
from RSA_Crypto.decrypt import RSADecryptor from RSA_Crypto.decrypt import RSADecryptor
from RSA_Crypto.generate_save import RSAKeyPair from RSA_Crypto.generate_save import RSAKeyPair
# Переменные для генерации и сохранения ключей
KEYS_DIR = 'keys' KEYS_DIR = 'keys'
PRIVATE_KEY_FILE = f'{KEYS_DIR}/private-key.pem' PRIVATE_KEY_FILE = f'{KEYS_DIR}/private-key.pem'
PUBLIC_KEY_FILE = f'{KEYS_DIR}/public-key.pem' PUBLIC_KEY_FILE = f'{KEYS_DIR}/public-key.pem'
# Создание директории для ключей
if not os.path.isdir(KEYS_DIR): if not os.path.isdir(KEYS_DIR):
os.makedirs(KEYS_DIR, exist_ok=True) os.makedirs(KEYS_DIR, exist_ok=True)
# Генерация и сохранение ключей key_pair = RSAKeyPair(PRIVATE_KEY_FILE, PUBLIC_KEY_FILE)
key_pair = RSAKeyPair(PRIVATE_KEY_FILE, PUBLIC_KEY_FILE,)
key_pair.generate_and_save_keys() key_pair.generate_and_save_keys()
# Шифрование сообщения message = '{"organization": "ООО Рога, Копыта & Ко", "license_expiration_date": "2024-01-01", "license_number": "1234567890", "count_licenses": 100}'
message = "{organization: 'ООО Рога, Копыта & Ко', license_expiration_date: '2024-01-01', license_number: '1234567890', count_licenses: 100}"
rsa_encryptor = RSAEncryptor(PUBLIC_KEY_FILE) rsa_encryptor = RSAEncryptor(PUBLIC_KEY_FILE)
encrypted_message = rsa_encryptor.encrypt_message(message) encrypted_message = rsa_encryptor.encrypt_message(message)
# Запись зашифрованного сообщения в файл
with open('output.txt', 'w') as f: with open('output.txt', 'w') as f:
f.write(encrypted_message) f.write(encrypted_message)
# Дешифрование сообщения из файла
rsa_decryptor = RSADecryptor(PRIVATE_KEY_FILE) rsa_decryptor = RSADecryptor(PRIVATE_KEY_FILE)
# Чтение зашифрованного сообщения из файла
with open('output.txt', 'r') as f: with open('output.txt', 'r') as f:
encrypted_message_from_file = f.read() encrypted_message_from_file = f.read()
decrypted_message = rsa_decryptor.decrypt_message(encrypted_message_from_file) decrypted_message = rsa_decryptor.decrypt_message(encrypted_message_from_file)
print("Расшифрованное сообщение:", decrypted_message)
try:
decrypted_message_dict = json.loads(decrypted_message)
print(decrypted_message_dict['organization'])
except json.JSONDecodeError:
print("Ошибка: Не удалось декодировать сообщение как JSON.")