summaryrefslogtreecommitdiff
path: root/poc.py
diff options
context:
space:
mode:
Diffstat (limited to 'poc.py')
-rw-r--r--poc.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/poc.py b/poc.py
index 9d66e30..8fddc5a 100644
--- a/poc.py
+++ b/poc.py
@@ -2,6 +2,7 @@ import fnmatch
2import glob 2import glob
3import os 3import os
4import sys 4import sys
5import time
5 6
6try: 7try:
7 import yara 8 import yara
@@ -14,8 +15,10 @@ if len(sys.argv) != 2:
14 15
15rules = yara.compile('malwares.yara') 16rules = yara.compile('malwares.yara')
16 17
17for root, dirnames, filenames in os.walk(sys.argv[1]): 18for cpt, (root, dirnames, filenames) in enumerate(os.walk(sys.argv[1])):
18 for filename in fnmatch.filter(filenames, '*.ph*'): 19 for filename in fnmatch.filter(filenames, '*.ph*'):
20 if not cpt % 1000:
21 time.sleep(3)
19 fname = os.path.join(root, filename) 22 fname = os.path.join(root, filename)
20 if os.stat(fname).st_size: 23 if os.stat(fname).st_size:
21 matches = rules.match(os.path.join(root, filename), fast=True) 24 matches = rules.match(os.path.join(root, filename), fast=True)