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

pep8

Browse Source
This commit is contained in:
William Ballenthin
2020-07-03 09:32:58 -06:00
parent 5317e1e11e
commit 80bdb4a545
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
capa/features/extractors/__init__.py
View File

@@ -1,5 +1,6 @@
import abc
class FeatureExtractor(object):
"""
FeatureExtractor defines the interface for fetching features from a sample.

3
capa/features/extractors/ida/__init__.py
View File

@@ -7,7 +7,6 @@ import capa.features.extractors.ida.file
import capa.features.extractors.ida.insn
import capa.features.extractors.ida.function
import capa.features.extractors.ida.basicblock
from capa.features.extractors import FeatureExtractor
@@ -52,6 +51,7 @@ class IdaFeatureExtractor(FeatureExtractor):
def get_functions(self):
import capa.features.extractors.ida.helpers as ida_helpers
for f in ida_helpers.get_functions(ignore_thunks=True, ignore_libs=True):
yield add_va_int_cast(f)
@@ -69,6 +69,7 @@ class IdaFeatureExtractor(FeatureExtractor):
def get_instructions(self, f, bb):
import capa.features.extractors.ida.helpers as ida_helpers
for insn in ida_helpers.get_instructions_in_range(bb.start_ea, bb.end_ea):
yield add_va_int_cast(insn)