mirror of
https://github.com/mandiant/capa.git
synced 2025-12-19 18:45:00 -08:00
result document: mypy
This commit is contained in:
@@ -581,7 +581,7 @@ class ResultDocument(FrozenModel):
|
|||||||
|
|
||||||
def to_capa(self) -> Tuple[Dict, Dict]:
|
def to_capa(self) -> Tuple[Dict, Dict]:
|
||||||
meta = self.meta.to_capa()
|
meta = self.meta.to_capa()
|
||||||
capabilities: Dict[str, List[Tuple[frz.Address, capa.features.common.Result]]] = {}
|
capabilities: Dict[str, List[Tuple[capa.features.address.Address, capa.features.common.Result]]] = {}
|
||||||
|
|
||||||
for rule_name, rule_match in self.rules.items():
|
for rule_name, rule_match in self.rules.items():
|
||||||
# Parse the YAML source into a Rule instance
|
# Parse the YAML source into a Rule instance
|
||||||
@@ -604,12 +604,12 @@ class ResultDocument(FrozenModel):
|
|||||||
result = capa.features.common.Result(
|
result = capa.features.common.Result(
|
||||||
statement=statement,
|
statement=statement,
|
||||||
success=match.success,
|
success=match.success,
|
||||||
locations=[frz.Address.to_capa(loc) for loc in match.locations],
|
locations={loc.to_capa() for loc in match.locations},
|
||||||
children=[],
|
children=[],
|
||||||
)
|
)
|
||||||
|
|
||||||
if rule_name not in capabilities:
|
if rule_name not in capabilities:
|
||||||
capabilities[rule_name] = []
|
capabilities[rule_name] = []
|
||||||
capabilities[rule_name].append((frz.Address.from_capa(addr), result))
|
capabilities[rule_name].append((addr.to_capa(), result))
|
||||||
|
|
||||||
return meta, capabilities
|
return meta, capabilities
|
||||||
|
|||||||
Reference in New Issue
Block a user