Throttle the python poc

Every 1000 files, sleep for 3 seconds

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
 import glob
 import os
 import sys
+import time
 
 try:
     import yara
@@ -14,8 +15,10 @@ if len(sys.argv) != 2:
 
 rules = yara.compile('malwares.yara')
 
-for root, dirnames, filenames in os.walk(sys.argv[1]):
+for cpt, (root, dirnames, filenames) in enumerate(os.walk(sys.argv[1])):
     for filename in fnmatch.filter(filenames, '*.ph*'):
+        if not cpt % 1000:
+            time.sleep(3)
         fname = os.path.join(root, filename)
         if os.stat(fname).st_size:
             matches = rules.match(os.path.join(root, filename), fast=True)