gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2026-01-24 18:24:44 -08:00
Code Issues Packages Projects Releases Wiki Activity

test scripts and fix show-features

Browse Source
This commit is contained in:
Moritz Raabe
2021-06-28 13:18:30 +02:00
parent 02658d6962
commit 5c8a4aafd7
5 changed files with 67 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/show-features.py
View File

@@ -64,6 +64,7 @@ Example::
insn: 0x10001027: mnemonic(shl)
...
"""
import os
import sys
import logging
import os.path
@@ -106,8 +107,11 @@ def main(argv=None):
with open(args.sample, "rb") as f:
extractor = capa.features.freeze.load(f.read())
else:
should_save_workspace = os.environ.get("CAPA_SAVE_WORKSPACE") not in ("0", "no", "NO", "n", None)
try:
extractor = capa.main.get_extractor(args.sample, args.format, capa.main.BACKEND_VIV, sigpaths=sig_paths)
extractor = capa.main.get_extractor(
args.sample, args.format, capa.main.BACKEND_VIV, sig_paths, should_save_workspace
)
except capa.main.UnsupportedFormatError:
logger.error("-" * 80)
logger.error(" Input file does not appear to be a PE file.")