金盾视频加密软件

from Crypto.Cipher import AES
from Crypto.Random import get_random_bytes
from Crypto.Util.Padding import pad, unpad
import base64

# 加密函数
def encrypt_video(file_name, key):
    cipher = AES.new(key, AES.MODE_CBC)
    with open(file_name, 'rb') as f:
        file_data = f.read()
    encrypted_data = cipher.encrypt(pad(file_data, AES.block_size))
    iv = base64.b64encode(cipher.iv).decode('utf-8')
    encrypted_data = base64.b64encode(encrypted_data).decode('utf-8')
    return iv + ':' + encrypted_data

# 解密函数
def decrypt_video(encrypted_data, key):
    iv, encrypted_data = encrypted_data.split(':')
    iv = base64.b64decode(iv)
    encrypted_data = base64.b64decode(encrypted_data)
    cipher = AES.new(key, AES.MODE_CBC, iv)
    original_data = unpad(cipher.decrypt(encrypted_data), AES.block_size)
    return original_data

# 示例使用
key = get_random_bytes(16)  # 16字节密钥
encrypted = encrypt_video('example.mp4', key)
print(f'Encrypted: {encrypted}')

decrypted = decrypt_video(encrypted, key)
with open('decrypted_example.mp4', 'wb') as f:
    f.write(decrypted)