# find files containing a specific strings like encrypted malware
#Import os module
import os
import fnmatch
# Ask the user to enter string to search
search_path = input("Enter directory path to search : ")
file_type = input("File Type : ")
search_str = input("Enter the search string : ")
result_txt_file = input("Output file : ")
def sessionSsave(nSess):
sessFile = open(result_txt_file,"a+")
#sessnum = sessFile.readlines()
#numer = len(sessnum)
sessFile.write(str(nSess))
sessFile.close()
try:
# Append a directory separator if not already present
if not (search_path.endswith("/") or search_path.endswith("\\") ):
search_path = search_path + "/"
# If path does not exist, set search path to current directory
if not os.path.exists(search_path):
search_path ="."
i = 1
# Repeat for each file in the directory
#for fname in os.walk(search_path):
for root, dirs, files in os.walk(search_path):
for filename in fnmatch.filter(files, file_type):
# Open file for reading
fo = open(os.path.join(root, filename), encoding="ISO-8859-1")
#print (i)
#i +=1
# Read the first line from the file
line = fo.readline()
# Initialize counter for line number
line_no = 1
try:
# Loop until EOF
while line != '' :
# Search for string in line
index = line.find(search_str)
if ( index != -1) :
print(filename, "[", line_no, ",", index, "] ", line, sep="")
txtline =str(root)+"/"+str( filename)+ "["+ str(line_no) + ","+ str(index) + "] "+ str(line)
sessionSsave( txtline)
# Read next line
line = fo.readline()
# Increment line counter
line_no += 1
except Exception as e:
print ("eroor message" + str(e))
# Close the files
fo.close()
except Exception as e:
print ("eroor message" + str(e))
