From d3b57e17262bad4cc23b6064627d94352c36d4ea Mon Sep 17 00:00:00 2001 From: sanchitjain123 Date: Mon, 28 Sep 2020 18:12:09 +0530 Subject: [PATCH] Fix bug - Change encryption key generation Encryption key generation should use secrets package rather than generating key from a variable defined. This fixes bug #8 --- File_Encryptor.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/File_Encryptor.py b/File_Encryptor.py index 9b28974..11c3b70 100644 --- a/File_Encryptor.py +++ b/File_Encryptor.py @@ -1,8 +1,8 @@ import hashlib import pyAesCrypt -import random import os,time import shutil +import secrets import email, smtplib, ssl from email import encoders @@ -25,8 +25,7 @@ fname=input(str('Enter file path (without quotes): ')) #fname='test.zip' try: - sub_key="abcdefg!~hijklm#$&*nopqrstuvw-+@xyz02345+/67890A-)(%BCDEFGHIJKLMNOPQRSTUVWXYZ" - enc_key = "".join(random.sample(sub_key,30)) + enc_key = secrets.token_hex(30) pyAesCrypt.encryptFile(fname,fname+'.aes',enc_key,buffer) os.remove(fname) print('''----------Encryption Sucessfull----------\n \nUser Warning: Make sure to notedown the encryption key, @@ -59,8 +58,7 @@ #fol='Dataset' if not os.path.exists('encrypted_folder'): os.mkdir('encrypted_folder') - sub_key="abcdefg!~hijklm#$&*nopqrstuvw-+@xyz02345+/67890A-)(%BCDEFGHIJKLMNOPQRSTUVWXYZ" - enc_key = "".join(random.sample(sub_key,30)) + enc_key = secrets.token_hex(30) for i in os.listdir(fol): if i[-4:]!='.aes': try: