scripts: add cli arguments to specify signatures

This commit is contained in:
William Ballenthin
2021-03-04 15:04:33 -07:00
parent c0f4fe6867
commit c2a4629c62
6 changed files with 30 additions and 6 deletions

View File

@@ -138,6 +138,14 @@ def main(argv=None):
default="auto",
help="Select sample format, %s" % format_help,
)
parser.add_argument(
"--signature",
action="append",
dest="signatures",
type=str,
default=[],
help="use the given signatures to identify library functions, file system paths to .sig/.pat files.",
)
args = parser.parse_args(args=argv)
if args.quiet:
@@ -199,7 +207,7 @@ def main(argv=None):
else:
format = args.format
try:
extractor = capa.main.get_extractor(args.sample, args.format, capa.main.BACKEND_VIV)
extractor = capa.main.get_extractor(args.sample, args.format, capa.main.BACKEND_VIV, args.signatures)
except capa.main.UnsupportedFormatError:
logger.error("-" * 80)
logger.error(" Input file does not appear to be a PE file.")