From d81f3a461e6e5ba4e13d1a49f1250cd3979bad06 Mon Sep 17 00:00:00 2001 From: Aayush Goel <81844215+Aayush-Goel-04@users.noreply.github.com> Date: Wed, 9 Aug 2023 14:20:53 +0530 Subject: [PATCH] Update show-unused-features.py --- scripts/show-unused-features.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/scripts/show-unused-features.py b/scripts/show-unused-features.py index dde77b4cd..dc5ce72a7 100644 --- a/scripts/show-unused-features.py +++ b/scripts/show-unused-features.py @@ -10,6 +10,7 @@ """ import os import sys +import typing import logging import argparse from typing import Set, Tuple @@ -52,8 +53,8 @@ def get_rules_feature_set(rules_path) -> Set[Feature]: def get_file_features( functions: Tuple[FunctionHandle, ...], extractor: capa.features.extractors.base_extractor.FeatureExtractor -) -> Counter[Feature]: - feature_map: Counter[Feature] = Counter() +) -> typing.Counter[Feature]: + feature_map: typing.Counter[Feature] = Counter() for f in functions: if extractor.is_library_function(f.address): @@ -89,7 +90,7 @@ def get_colored(s: str): return colored(s, "cyan") -def print_unused_features(feature_map: Counter[Feature], rules_feature_set: Set[Feature]): +def print_unused_features(feature_map: typing.Counter[Feature], rules_feature_set: Set[Feature]): unused_features = [] for feature, count in feature_map.items(): if feature in rules_feature_set: @@ -147,7 +148,7 @@ def main(argv=None): log_unsupported_runtime_error() return -1 - feature_map: Counter[Feature] = Counter() + feature_map: typing.Counter[Feature] = Counter() feature_map.update([feature for feature, _ in extractor.extract_global_features()]) @@ -190,7 +191,7 @@ def ida_main(): print(f"getting features for current function {hex(function)}") extractor = capa.features.extractors.ida.extractor.IdaFeatureExtractor() - feature_map: Counter[Feature] = Counter() + feature_map: typing.Counter[Feature] = Counter() feature_map.update([feature for feature, _ in extractor.extract_file_features()])