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

result document: mypy

Browse Source
This commit is contained in:
Willi Ballenthin
2023-04-03 13:44:09 +02:00
parent 3f5d9c79f9
commit d2fc740278
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
capa/render/result_document.py
View File

@@ -581,7 +581,7 @@ class ResultDocument(FrozenModel):
def to_capa(self) -> Tuple[Dict, Dict]:
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():
# Parse the YAML source into a Rule instance
@@ -604,12 +604,12 @@ class ResultDocument(FrozenModel):
result = capa.features.common.Result(
statement=statement,
success=match.success,
locations=[frz.Address.to_capa(loc) for loc in match.locations],
locations={loc.to_capa() for loc in match.locations},
children=[],
)
if rule_name not in capabilities:
capabilities[rule_name] = []
capabilities[rule_name].append((frz.Address.from_capa(addr), result))
capabilities[rule_name].append((addr.to_capa(), result))
return meta, capabilities