Minor changes

Author: Amit Raut

classifierWindows.py

@@ -6,6 +6,7 @@
 from collections import Counter
 from datetime  import datetime as dt
 
+count = int(os.sys.argv[5])
 # Usage Message
 if len(os.sys.argv) < 5:
     print "Usage: ./classifier -i [path to malware sample] -t [path to training set directory]"
@@ -67,15 +68,15 @@ def compair(unknownFeatureList, trainingFeatureList):
 	return result
 
 def main(inSample):
-	global trainingSetPath
+	global trainingSetPath, count
 
 	# Generate nttrace file for given input sample
 
 	cmd = 'nttrace ' + str(inSample)
 	with open('C:\Users\user\AppData\Local\mlmc\\' + os.path.splitext(os.path.basename(str(inSample)))[0] + '.trace', 'w') as traceFile:
 		ps = sp.Popen(cmd, stderr=sp.PIPE, stdout=traceFile, shell=True)
 
-        time.sleep(2)
+        time.sleep(1)
 	ps.terminate()
 
 	featureLists = [] # [[3gramList],[4gramList], [5gramList], ...]
@@ -114,23 +115,24 @@ def main(inSample):
 	# print 'Similarity with 3,4,5,6,7-gram feature = {0:3.2f} %'.format(sum(finalResult) / float(len(finalResult)) * 100)
 
 	# Combine output for many samples in to a file
-	with open('Z:\Desktop\output.txt', 'a') as outFile:
-		outFile.write("{0:2.5f} \t + {1:s}  \n".format(sum(finalResult) / float(len(finalResult)) * 100, inSample))
-	print "{0:2.5f} \t {1:s} \n".format(sum(finalResult) / float(len(finalResult)) * 100, inSample)
+	with open('Z:\Desktop\output7000.txt', 'a') as outFile:
+		outFile.write("{0:2.5f} \t  {1:s}  \n".format(sum(finalResult) / float(len(finalResult)) * 100, inSample))
+	print "{0:2.5f} \t {1:s} \t".format(sum(finalResult) / float(len(finalResult)) * 100, inSample), count, '\n' 
+	count += 1
 
 	# Kill the dangling NtTrace process if any
 	os.system("taskkill /IM NtTrace.exe /F")
+	# os.system("taskkill /IM 979971590.exe /F")
+
 
 def test():
 	global inSample
 	if os.path.isfile(inSample):
-		print inSample
 		main(inSample)
 	elif os.path.isdir(inSample):
 		filelist = os.listdir(inSample)
-		for sample in filelist:
+		for sample in sorted(filelist)[count:]:
 			if sample.endswith('.malware'):
-				print inSample + sample
 				main(inSample + sample)
 
 if __name__ == '__main__':