Notes

Для реализации протокола Диффи-Хеллмана на языке Python можно использовать библиотеку cryptography. Для интеграции с OpenSSL можно воспользоваться модулем subprocess.

Ниже приведен пример кода, который выполняет обмен ключами между двумя сторонами и использует полученный секретный ключ для шифрования файла с помощью OpenSSL:

python
from cryptography.hazmat.primitives.asymmetric import dh
from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.primitives.serialization import load_pem_private_key
import subprocess

# Генерация параметров Диффи-Хеллмана
parameters = dh.generate_parameters(generator=2, key_size=2048)

# Генерация закрытого и открытого ключей для стороны A
private_key_a = parameters.generate_private_key()
public_key_a = private_key_a.public_key()

# Сериализация открытого ключа стороны A в формат PEM
public_key_a_pem = public_key_a.public_bytes(
encoding=serialization.Encoding.PEM,
format=serialization.PublicFormat.SubjectPublicKeyInfo
)

# Генерация закрытого и открытого ключей для стороны B
private_key_b = parameters.generate_private_key()
public_key_b = private_key_b.public_key()

# Сериализация открытого ключа стороны B в формат PEM
public_key_b_pem = public_key_b.public_bytes(
encoding=serialization.Encoding.PEM,
format=serialization.PublicFormat.SubjectPublicKeyInfo
)

# Обмен открытыми ключами между сторонами A и B

# Сторона A получает открытый ключ стороны B
received_public_key_b = serialization.load_pem_public_key(
public_key_b_pem,
)

# Сторона B получает открытый ключ стороны A
received_public_key_a = serialization.load_pem_public_key(
public_key_a_pem,
)

# Вычисление общего секретного ключа на стороне A
shared_key_a = private_key_a.exchange(received_public_key_b)

# Вычисление общего секретного ключа на стороне B
shared_key_b = private_key_b.exchange(received_public_key_a)

# Проверка, что общие секретные ключи совпадают
assert shared_key_a == shared_key_b

# Шифрование файла с помощью OpenSSL и полученного секретного ключа

# Запись секретного ключа в файл для использования в OpenSSL
with open('shared_key.bin', 'wb') as f:
f.write(shared_key_a)

# Шифрование файла file_to_encrypt.txt с помощью OpenSSL и полученного секретного ключа
subprocess.run(['openssl', 'enc', '-aes-256-cbc', '-salt', '-in', 'file_to_encrypt.txt', '-out', 'encrypted_file.enc', '-pass', 'file:shared_key.bin'])

# Расшифровка файла encrypted_file.enc с помощью OpenSSL и полученного секретного ключа
subprocess.run(['openssl', 'enc', '-d', '-aes-256-cbc', '-in', 'encrypted_file.enc', '-out', 'decrypted_file.txt', '-pass', 'file:shared_key.bin'])


В данном примере сначала генерируются параметры Диффи-Хеллмана, затем генерируются закрытые и открытые ключи для сторон A и B. Открытые ключи сериализуются в формат PEM и обмениваются между сторонами. Затем на каждой стороне вычисляется общий секретный ключ, который используется для шифрования и расшифровки файла с помощью OpenSSL.

Обратите внимание, что для работы данного кода требуется установленная библиотека cryptography и установленный OpenSSL на вашей системе.