From 17296b4f5cd7ae0cd7e18a72fe5bb7ae321bce27 Mon Sep 17 00:00:00 2001 From: frack113 <62423083+frack113@users.noreply.github.com> Date: Sat, 20 Nov 2021 11:13:18 +0100 Subject: [PATCH] Fix score error --- tools/sigma/sigma2attack.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tools/sigma/sigma2attack.py b/tools/sigma/sigma2attack.py index 2c5d6dcc1..274de9439 100755 --- a/tools/sigma/sigma2attack.py +++ b/tools/sigma/sigma2attack.py @@ -35,7 +35,7 @@ def main(): args = parser.parse_args() - + print(args.level_score) status_start = status_eq[args.status_start] status_end = status_eq[args.status_end] @@ -79,7 +79,7 @@ def main(): score_to_rules[technique_id] = [] techniques_to_rules[technique_id].append(os.path.basename(rule_file)) score_to_rules[technique_id].append(level_eq[level]) - if args.level_score: + if args.level_score == True: curr_max_technique_count = max(curr_max_technique_count, sum(score_to_rules[technique_id])) else: curr_max_technique_count = max(curr_max_technique_count, len(techniques_to_rules[technique_id])) @@ -89,9 +89,13 @@ def main(): scores = [] for technique in techniques_to_rules: + if args.level_score == True: + technique_score = sum(score_to_rules[technique]) + else: + technique_score = len(techniques_to_rules[technique]) entry = { "techniqueID": technique, - "score": sum(score_to_rules[technique]), + "score": technique_score, } if not args.no_comment: entry["comment"] = "\n".join(techniques_to_rules[technique])