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

library-detection: add json output format

Browse Source
This commit is contained in:
Willi Ballenthin
2024-10-21 10:42:30 +00:00
parent c005de0a0d
commit 70c96a29b4
1 changed files with 52 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
capa/analysis/libraries.py
View File

@@ -54,22 +54,18 @@ class Method(str, Enum):
class FunctionClassification(BaseModel):
va: int
classification: Classification
method: Method
# if is library, this must be provided
method: Optional[Method]
# if is library, these can optionally be provided.
library_name: Optional[str] = None
library_version: Optional[str] = None
class Layout(BaseModel):
functions: List[int]
class FunctionIdResults(BaseModel):
function_classifications: List[FunctionClassification]
# layout: Layout
@contextlib.contextmanager
def ida_session(input_path: Path, use_temp_dir=True):
@@ -115,6 +111,7 @@ def main(argv=None):
capa.main.install_common_args(parser, wanted={"input_file"})
parser.add_argument("--store-idb", action="store_true", default=False, help="store IDA database file")
parser.add_argument("--min-string-length", type=int, default=8, help="minimum string length")
parser.add_argument("-j", "--json", action="store_true", help="emit JSON instead of text")
args = parser.parse_args(args=argv)
try:
@@ -152,6 +149,24 @@ def main(argv=None):
)
)
if args.json:
doc = FunctionIdResults(function_classifications=[])
classifications_by_va = capa.analysis.strings.create_index(function_classifications, "va")
for va in idautils.Functions(start=0, end=None):
if classifications := classifications_by_va.get(va):
doc.function_classifications.extend(classifications)
else:
doc.function_classifications.append(
FunctionClassification(
va=va,
classification=Classification.UNKNOWN,
method=None,
)
)
print(doc.model_dump_json()) # noqa: T201 print found
else:
table = rich.table.Table()
table.add_column("FVA")
table.add_column("CLASSIFICATION")
@@ -167,7 +182,7 @@ def main(argv=None):
if classifications := classifications_by_va.get(va):
classification = {c.classification for c in classifications}
method = {c.method for c in classifications}
method = {c.method for c in classifications if c.method}
extra = {f"{c.library_name}@{c.library_version}" for c in classifications if c.library_name}
table.add_row(