From a9f70dd1e588e975672034c4bf398d77dc9b4c51 Mon Sep 17 00:00:00 2001 From: Yacine Elhamer <16624109+yelhamer@users.noreply.github.com> Date: Mon, 26 Jun 2023 20:01:30 +0100 Subject: [PATCH] main.py: update extractor type casting Co-authored-by: Willi Ballenthin --- capa/main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/capa/main.py b/capa/main.py index 6000c49c..b408c55b 100644 --- a/capa/main.py +++ b/capa/main.py @@ -233,8 +233,10 @@ def find_file_capabilities(ruleset: RuleSet, extractor: FeatureExtractor, functi if isinstance(extractor, StaticFeatureExtractor): extractor_: StaticFeatureExtractor = cast(StaticFeatureExtractor, extractor) - else: + elif isinstance(extractor, DynamicFeatureExtractor): extractor_: DynamicFeatureExtractor = cast(DynamicFeatureExtractor, extractor) + else: + raise ValueError(f"unexpected extractor type: {extractor.__class__.__name__}") for feature, va in itertools.chain(extractor_.extract_file_features(), extractor_.extract_global_features()): # not all file features may have virtual addresses.