gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2025-12-22 07:10:29 -08:00
Code Issues Packages Projects Releases Wiki Activity

FeatureExtractor alias: add type casts to either StaticFeatureExtractor or DynamicFeatureExtractor

Browse Source
This commit is contained in:
Yacine Elhamer
2023-06-26 01:23:01 +01:00
parent 172e7a7649
commit 94fc7b4e9a
5 changed files with 33 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/profile-time.py
View File

@@ -46,6 +46,7 @@ import capa.helpers
import capa.features
import capa.features.common
import capa.features.freeze
from capa.features.extractors.base_extractor import FeatureExtractor
logger = logging.getLogger("capa.profile")
@@ -105,7 +106,7 @@ def main(argv=None):
with open(args.sample, "rb") as f:
extractor = capa.features.freeze.load(f.read())
else:
extractor = capa.main.get_extractor(
extractor: FeatureExtractor = capa.main.get_extractor(
args.sample, args.format, args.os, capa.main.BACKEND_VIV, sig_paths, should_save_workspace=False
)

3
scripts/show-capabilities-by-function.py
View File

@@ -70,6 +70,7 @@ import capa.render.result_document as rd
from capa.helpers import get_file_taste
from capa.features.common import FORMAT_AUTO
from capa.features.freeze import Address
from capa.features.extractors.base_extractor import FeatureExtractor
logger = logging.getLogger("capa.show-capabilities-by-function")
@@ -166,7 +167,7 @@ def main(argv=None):
should_save_workspace = os.environ.get("CAPA_SAVE_WORKSPACE") not in ("0", "no", "NO", "n", None)
try:
extractor = capa.main.get_extractor(
extractor: FeatureExtractor = capa.main.get_extractor(
args.sample, args.format, args.os, args.backend, sig_paths, should_save_workspace
)
except capa.exceptions.UnsupportedFormatError: