Mã hóa tin nhắn bằng Python

pip install cryptography

message_data = {
    "Huy": [
        {"message": "Chào Linh, dạo này mọi thứ thế nào?", "time": "2023-03-21 10:30:00"},
        {"message": "Không tệ lắm, mình đang làm một số dự án lập trình. Cậu có nghe về thuật toán mã hóa mới chưa?", "time": "2023-03-21 10:35:00"},
        {"message": "Nó được gọi là AES256 và được cho là rất an toàn. Cậu muốn thử áp dụng vào tin nhắn của mình không?", "time": "2023-03-21 10:40:00"},
    ],
    "Linh": [
        {"message": "Tớ ổn, cảm ơn cậu! Còn cậu thì sao?", "time": "2023-03-21 10:32:00"},
        {"message": "Chưa, đó là gì vậy?", "time": "2023-03-21 10:37:00"},
        {"message": "Được, thử làm đi!", "time": "2023-03-21 10:42:00"},
    ]
}

import os
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
from cryptography.hazmat.backends import default_backend

# Tạo khóa bí mật dùng chung
shared_secret_key = os.urandom(32)  # Khóa dài 32 byte

def encrypt_message(message, key):
    iv = os.urandom(16)  # Tạo vector khởi tạo ngẫu nhiên
    cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=default_backend())
    encryptor = cipher.encryptor()
    # Đệm tin nhắn để đảm bảo độ dài là bội số của 16
    padded_message = message + (16 - len(message) % 16) * chr(16 - len(message) % 16)
    ciphertext = encryptor.update(padded_message.encode()) + encryptor.finalize()
    return iv + ciphertext

def decrypt_message(ciphertext, key):
    iv = ciphertext[:16]  # Vector khởi tạo là 16 byte đầu tiên
    cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=default_backend())
    decryptor = cipher.decryptor()
    padded_message = decryptor.update(ciphertext[16:]) + decryptor.finalize()
    # Loại bỏ phần đệm
    return padded_message[:-padded_message[-1]].decode()

for person, messages in message_data.items():
    for message in messages:
        encrypted_message = encrypt_message(message["message"], shared_secret_key)
        message["message"] = encrypted_message.hex()  # Lưu dạng chuỗi hexa

print("Dữ liệu tin nhắn đã được mã hóa:")
print(message_data)

for person, messages in message_data.items():
    for message in messages:
        ciphertext = bytes.fromhex(message["message"])
        decrypted_message = decrypt_message(ciphertext, shared_secret_key)
        message["message"] = decrypted_message

print("Dữ liệu tin nhắn sau khi giải mã:")
print(message_data)