Cryptogtaphy Fundamentals Lab Assignment - 3 AES Code:: 17BCE0147 Vijay Surya
Cryptogtaphy Fundamentals Lab Assignment - 3 AES Code:: 17BCE0147 Vijay Surya
Cryptogtaphy Fundamentals Lab Assignment - 3 AES Code:: 17BCE0147 Vijay Surya
CRYPTOGTAPHY FUNDAMENTALS
LAB ASSIGNMENT - 3
AES
Code:
import os
from Crypto.Cipher import AES
from Crypto.Hash import SHA256
from Crypto import Random
def encrypt(key, filename):
chunksize =64 * 1024
outputFile = "(encrypted)" + filename
filesize = str(os.path.getsize(filename)).zfill(16)
IV = Random.new().read(16)
encryptor = AES.new(key, AES.MODE_CBC, IV)
while True:
chunk = infile.read(chunksize)
if len(chunk) == 0:
break
elif len(chunk) % 16 != 0:
chunk +=b ' ' * (16 – (len(chunk) % 16))
outfile.write(encryptor.encrypt(chunk))
#Decryption of files
def decrypt(key, filename):
chunksize =64 * 1024
outputFile = filename[11:]
if len(chunk) == 0:
break
outfile.write(decryptor.decrypt(chunk))
outfile.truncate(filesize)
17BCE0147 VIJAY SURYA
def getKey(password):
hasher = SHA256.new(password.encode(‘utf-8’))
return hasher.digest()
def Main():
choice = input(“Would you like to (E)ncrypt or (D)ecrypt?: “)
if choice == ‘E’ or choice == ‘e’:
filename = input(“File to encrypt: “)
password = input(“Password: “)
encrypt(getKey(password), filename)
print(“Done.”)
elif choice == ‘D’ or choice == ‘d’:
filename = input(“File to decrypt: “)
password = input(“Password: “)
decrypt(getKey(password), filename)
print(“Done.”)
else:
print(“No Option selected, closing…”)
if __name__ == ‘__main__’:
Main()