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

extractors: smda: fix missing yield from

Browse Source
This commit is contained in:
William Ballenthin
2021-09-04 16:55:37 -06:00
parent 32244b2641
commit 7a3287fa25
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
capa/features/extractors/smda/extractor.py
View File

@@ -29,25 +29,25 @@ class SmdaFeatureExtractor(FeatureExtractor):
yield from self.global_features
def extract_file_features(self):
capa.features.extractors.smda.file.extract_features(self.smda_report, self.buf)
yield from capa.features.extractors.smda.file.extract_features(self.smda_report, self.buf)
def get_functions(self):
for function in self.smda_report.getFunctions():
yield function
def extract_function_features(self, f):
capa.features.extractors.smda.function.extract_features(f)
yield from capa.features.extractors.smda.function.extract_features(f)
def get_basic_blocks(self, f):
for bb in f.getBlocks():
yield bb
def extract_basic_block_features(self, f, bb):
capa.features.extractors.smda.basicblock.extract_features(f, bb)
yield from capa.features.extractors.smda.basicblock.extract_features(f, bb)
def get_instructions(self, f, bb):
for smda_ins in bb.getInstructions():
yield smda_ins
def extract_insn_features(self, f, bb, insn):
capa.features.extractors.smda.insn.extract_features(f, bb, insn)
yield from capa.features.extractors.smda.insn.extract_features(f, bb, insn)