cryptAES/main.py
2021-09-21 16:36:13 +02:00

24 lines
542 B
Python

from Crypto.Random import get_random_bytes
from crypt import *
from compress import *
if not os.path.exists("key"):
key = get_random_bytes(32) # 32 bytes * 8 = 256 bits (1 byte = 8 bits)
open("key", 'wb').write(key)
print("Saved")
else:
key = open("key", 'rb').read()
print("Recovered")
print(key)
file = "original_file.txt"
cfile = file + '.enc'
encrypt_file(key, file, cfile)
decrypt_file(key, cfile)
rootdir = "test"
recurse(rootdir, key)
decrypt_file(key, "test/DressLens4.mp4.gz.enc", "test/DressLens4.mp4.gz")