gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2025-12-12 15:49:46 -08:00
Code Issues Packages Projects Releases Wiki Activity

main.py: update extractor type casting

Browse Source 
Co-authored-by: Willi Ballenthin <willi.ballenthin@gmail.com>
This commit is contained in:
Yacine Elhamer
2023-06-26 20:01:30 +01:00
committed by GitHub
parent 94fc7b4e9a
commit a9f70dd1e5
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
capa/main.py
View File

@@ -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.