main.py: add asserts for checking filters are not empty

mandiant · Aug 20, 2024 · ac50103 · ac50103
1 parent b0d8071
commit ac50103
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/capa/main.py b/capa/main.py
@@ -827,8 +827,10 @@ def get_extractor_filters_from_cli(args, input_format) -> FilterConfig:
 
 def apply_extractor_filters(extractor: FeatureExtractor, extractor_filters: FilterConfig):
     if isinstance(extractor, StaticFeatureExtractor):
+        assert extractor_filters["functions"]
         return FunctionFilter(extractor, extractor_filters["functions"])
     elif isinstance(extractor, DynamicFeatureExtractor):
+        assert extractor_filters["processes"]
         return ProcessFilter(extractor, extractor_filters["processes"])
     else:
         raise ShouldExitError(E_INVALID_FEATURE_EXTRACTOR)