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

pep8

Browse Source
This commit is contained in:
William Ballenthin
2021-09-04 15:59:38 -06:00
parent 2ea4dc9d7e
commit 39e4e47763
5 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
capa/main.py
View File

@@ -88,12 +88,16 @@ def find_function_capabilities(ruleset: RuleSet, extractor: FeatureExtractor, f:
# - basic blocks # - basic blocks
bb_features = collections.defaultdict(set) bb_features = collections.defaultdict(set)
for feature, va in itertools.chain(extractor.extract_basic_block_features(f, bb), extractor.extract_global_features()): for feature, va in itertools.chain(
extractor.extract_basic_block_features(f, bb), extractor.extract_global_features()
):
bb_features[feature].add(va) bb_features[feature].add(va)
function_features[feature].add(va) function_features[feature].add(va)
for insn in extractor.get_instructions(f, bb): for insn in extractor.get_instructions(f, bb):
for feature, va in itertools.chain(extractor.extract_insn_features(f, bb, insn), extractor.extract_global_features()): for feature, va in itertools.chain(
extractor.extract_insn_features(f, bb, insn), extractor.extract_global_features()
):
bb_features[feature].add(va) bb_features[feature].add(va)
function_features[feature].add(va) function_features[feature].add(va)