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

fix mypy and codestyle issues

Browse Source
This commit is contained in:
Yacine Elhamer
2023-06-27 11:32:21 +01:00
parent a99ff813cb
commit 06aea6b97c
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
capa/features/extractors/cape/extractor.py
View File

@@ -26,6 +26,7 @@ class CapeExtractor(DynamicFeatureExtractor):
self.behavior = behavior self.behavior = behavior
self.global_features = capa.features.extractors.cape.global_.extract_features(self.static) self.global_features = capa.features.extractors.cape.global_.extract_features(self.static)
def extract_global_features(self) -> Iterator[Tuple[Feature, Address]]: def extract_global_features(self) -> Iterator[Tuple[Feature, Address]]:
yield from self.global_features yield from self.global_features

3
capa/main.py
View File

@@ -21,7 +21,7 @@ import textwrap
import itertools import itertools
import contextlib import contextlib
import collections import collections
from typing import Any, Dict, List, Tuple, Callable, cast, Union from typing import Any, Dict, List, Tuple, Union, Callable, cast
import halo import halo
import tqdm import tqdm
@@ -786,6 +786,7 @@ def collect_metadata(
sha1 = hashlib.sha1() sha1 = hashlib.sha1()
sha256 = hashlib.sha256() sha256 = hashlib.sha256()
assert isinstance(extractor, StaticFeatureExtractor)
with open(sample_path, "rb") as f: with open(sample_path, "rb") as f:
buf = f.read() buf = f.read()

1
scripts/show-features.py
View File

@@ -84,7 +84,6 @@ from capa.helpers import get_auto_format, log_unsupported_runtime_error
from capa.features.common import FORMAT_AUTO, FORMAT_FREEZE, DYNAMIC_FORMATS, is_global_feature from capa.features.common import FORMAT_AUTO, FORMAT_FREEZE, DYNAMIC_FORMATS, is_global_feature
from capa.features.extractors.base_extractor import FeatureExtractor, StaticFeatureExtractor, DynamicFeatureExtractor from capa.features.extractors.base_extractor import FeatureExtractor, StaticFeatureExtractor, DynamicFeatureExtractor
logger = logging.getLogger("capa.show-features") logger = logging.getLogger("capa.show-features")