From d1b4d61ce5357bda486e7b380a0c52bfc2b58ad0 Mon Sep 17 00:00:00 2001 From: Terrtia Date: Tue, 24 Nov 2020 15:44:11 +0100 Subject: [PATCH] fix: [Yara Tracker] catch yara timeout --- bin/trackers/Tracker_Yara.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/bin/trackers/Tracker_Yara.py b/bin/trackers/Tracker_Yara.py index 55e5a00b..b0356b55 100755 --- a/bin/trackers/Tracker_Yara.py +++ b/bin/trackers/Tracker_Yara.py @@ -71,10 +71,12 @@ if __name__ == "__main__": item_id = p.get_from_set() if item_id is not None: item_content = item_basic.get_item_content(item_id) - yara_match = rules.match(data=item_content, callback=yara_rules_match, which_callbacks=yara.CALLBACK_MATCHES, timeout=60) - if yara_match: - print(f'{item_id}: {yara_match}') - + try: + yara_match = rules.match(data=item_content, callback=yara_rules_match, which_callbacks=yara.CALLBACK_MATCHES, timeout=60) + if yara_match: + print(f'{item_id}: {yara_match}') + except yara.TimeoutError as e: + print(f'{item_id}: yara scanning timed out') else: time.sleep(5)