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

fix mypy issues

Browse Source
This commit is contained in:
mr-tz
2024-06-12 09:27:13 +00:00
parent 97a3fba2c9
commit 9ec9a6f439
1 changed files with 24 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
scripts/capa2sarif.py
View File

@@ -98,7 +98,7 @@ def main() -> int:
# Create baseline sarif structure to be populated from json data # Create baseline sarif structure to be populated from json data
sarif_structure: Optional[dict] = _sarif_boilerplate(json_data["meta"], json_data["rules"]) sarif_structure: Optional[dict] = _sarif_boilerplate(json_data["meta"], json_data["rules"])
if sarif_structure is None: if sarif_structure is None:
logger.errort("An Error has occured creating default sarif structure.") logger.error("An Error has occured creating default sarif structure.")
return -3 return -3
_populate_artifact(sarif_structure, json_data["meta"]) _populate_artifact(sarif_structure, json_data["meta"])
@@ -244,37 +244,36 @@ def _populate_invocations(sarif_log: dict, meta_data: dict) -> None:
def _enumerate_evidence(node: dict, related_count: int) -> List[dict]: def _enumerate_evidence(node: dict, related_count: int) -> List[dict]:
related_locations = [] related_locations = []
if node.get("success") and node.get("node").get("type") != "statement": if node.get("success") and node.get("node", {}).get("type") != "statement":
label = "" label = ""
if node.get("node").get("type") == "feature": if node.get("node", {}).get("type") == "feature":
if node.get("node").get("feature").get("type") == "api": if node.get("node", {}).get("feature", {}).get("type") == "api":
label = "api: " + node.get("node").get("feature").get("api") label = "api: " + node.get("node", {}).get("feature", {}).get("api")
elif node.get("node").get("feature").get("type") == "match": elif node.get("node", {}).get("feature", {}).get("type") == "match":
label = "match: " + node.get("node").get("feature").get("match") label = "match: " + node.get("node", {}).get("feature", {}).get("match")
elif node.get("node").get("feature").get("type") == "number": elif node.get("node", {}).get("feature", {}).get("type") == "number":
label = f"number: {node.get('node').get('feature').get('description')} ({node.get('node').get('feature').get('number')})" label = f"number: {node.get('node', {}).get('feature', {}).get('description')} ({node.get('node', {}).get('feature', {}).get('number')})"
elif node.get("node").get("feature").get("type") == "offset": elif node.get("node", {}).get("feature", {}).get("type") == "offset":
label = f"offset: {node.get('node').get('feature').get('description')} ({node.get('node').get('feature').get('offset')})" label = f"offset: {node.get('node', {}).get('feature', {}).get('description')} ({node.get('node', {}).get('feature', {}).get('offset')})"
elif node.get("node").get("feature").get("type") == "mnemonic": elif node.get("node", {}).get("feature", {}).get("type") == "mnemonic":
label = f"mnemonic: {node.get('node').get('feature').get('mnemonic')}" label = f"mnemonic: {node.get('node', {}).get('feature', {}).get('mnemonic')}"
elif node.get("node").get("feature").get("type") == "characteristic": elif node.get("node", {}).get("feature", {}).get("type") == "characteristic":
label = f"characteristic: {node.get('node').get('feature').get('characteristic')}" label = f"characteristic: {node.get('node', {}).get('feature', {}).get('characteristic')}"
elif node.get("node").get("feature").get("type") == "os": elif node.get("node", {}).get("feature", {}).get("type") == "os":
label = f"os: {node.get('node').get('feature').get('os')}" label = f"os: {node.get('node', {}).get('feature', {}).get('os')}"
elif node.get("node").get("feature").get("type") == "operand number": elif node.get("node", {}).get("feature", {}).get("type") == "operand number":
label = f"operand: ({node.get('node').get('feature').get('index')} ) {node.get('node').get('feature').get('description')} ({node.get('node').get('feature').get('operand_number')})" label = f"operand: ({node.get('node', {}).get('feature', {}).get('index')} ) {node.get('node', {}).get('feature', {}).get('description')} ({node.get('node', {}).get('feature', {}).get('operand_number')})"
else: else:
logger.error( logger.error(
"Not implemented %s", "Not implemented %s",
node.get("node").get("feature").get("type"), node.get("node", {}).get("feature", {}).get("type"),
file=sys.stderr,
) )
return [] return []
else: else:
logger.error("Not implemented %s", node.get("node").get("type"), file=sys.stderr) logger.error("Not implemented %s", node.get("node", {}).get("type"))
return [] return []
for loc in node.get("locations"): for loc in node.get("locations", []):
if loc["type"] != "absolute": if loc["type"] != "absolute":
continue continue
@@ -287,8 +286,8 @@ def _enumerate_evidence(node: dict, related_count: int) -> List[dict]:
) )
related_count += 1 related_count += 1
if node.get("success") and node.get("node").get("type") == "statement": if node.get("success") and node.get("node", {}).get("type") == "statement":
for child in node.get("children"): for child in node.get("children", []):
related_locations += _enumerate_evidence(child, related_count) related_locations += _enumerate_evidence(child, related_count)
return related_locations return related_locations