diff --git a/loki.py b/loki.py index 337dbe9a..5f45ed0e 100644 --- a/loki.py +++ b/loki.py @@ -345,15 +345,20 @@ def scan_path(self, path): # Set fileData to an empty value fileData = "" + print_filesize_info = False + # Evaluations ------------------------------------------------------- # Evaluate size - if fileSize > (int(args.s) * 1024): + fileSizeLimit = int(args.s) * 1024 + if fileSize > fileSizeLimit: # Print files do_intense_check = False + print_filesize_info = True # Some file types will force intense check if fileType == "MDMP": do_intense_check = True + print_filesize_info = False # Intense Check switch if do_intense_check: @@ -363,6 +368,9 @@ def scan_path(self, path): if args.printall: logger.log("INFO", "FileScan", "Checking %s TYPE: %s SIZE: %s" % (fileNameCleaned, fileType, fileSize)) + if print_filesize_info and args.printall: + logger.log("INFO", "FileScan", "Skipping file due to file size: %s TYPE: %s SIZE: %s CURRENT SIZE LIMIT(kilobytes): %d" % (fileNameCleaned, fileType, fileSize, fileSizeLimit)) + # Hash Check ------------------------------------------------------- # Do the check if do_intense_check: