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

pep8

Browse Source
This commit is contained in:
William Ballenthin
2021-08-11 09:23:41 -06:00
parent 97092c91db
commit 753b003107
4 changed files with 29 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
capa/features/extractors/pefile.py
View File

@@ -14,9 +14,8 @@ import capa.features.extractors
import capa.features.extractors.helpers
import capa.features.extractors.strings
from capa.features.file import Export, Import, Section
from capa.features.common import String, Characteristic
from capa.features.common import CHARACTERISTIC_PE, CHARACTERISTIC_WINDOWS, String, Characteristic
from capa.features.extractors.base_extractor import FeatureExtractor
from capa.features.common import CHARACTERISTIC_WINDOWS, CHARACTERISTIC_PE
logger = logging.getLogger(__name__)

13
capa/rules.py
View File

@@ -34,10 +34,15 @@ import capa.features.insn
import capa.features.common
import capa.features.basicblock
from capa.engine import Statement, FeatureSet
from capa.features.common import MAX_BYTES_FEATURE_SIZE, Feature
from capa.features.common import CHARACTERISTIC_WINDOWS, CHARACTERISTIC_LINUX, CHARACTERISTIC_MACOS
from capa.features.common import CHARACTERISTIC_PE, CHARACTERISTIC_ELF
from capa.features.common import (
CHARACTERISTIC_PE,
CHARACTERISTIC_ELF,
CHARACTERISTIC_LINUX,
CHARACTERISTIC_MACOS,
CHARACTERISTIC_WINDOWS,
MAX_BYTES_FEATURE_SIZE,
Feature,
)
logger = logging.getLogger(__name__)

10
tests/fixtures.py
View File

@@ -21,8 +21,14 @@ import capa.features.file
import capa.features.insn
import capa.features.common
import capa.features.basicblock
from capa.features.common import ARCH_X32, ARCH_X64
from capa.features.common import CHARACTERISTIC_WINDOWS, CHARACTERISTIC_LINUX, CHARACTERISTIC_PE, CHARACTERISTIC_ELF
from capa.features.common import (
ARCH_X32,
ARCH_X64,
CHARACTERISTIC_PE,
CHARACTERISTIC_ELF,
CHARACTERISTIC_LINUX,
CHARACTERISTIC_WINDOWS,
)
CD = os.path.dirname(__file__)

13
tests/test_rules.py
View File

@@ -15,7 +15,16 @@ import capa.engine
import capa.features.common
from capa.features.file import FunctionName
from capa.features.insn import Number, Offset
from capa.features.common import ARCH_X32, ARCH_X64, CHARACTERISTIC_PE, CHARACTERISTIC_WINDOWS, OS_WINDOWS, FORMAT_PE, String, Characteristic
from capa.features.common import (
ARCH_X32,
ARCH_X64,
FORMAT_PE,
OS_WINDOWS,
CHARACTERISTIC_PE,
CHARACTERISTIC_WINDOWS,
String,
Characteristic,
)
def test_rule_ctor():
@@ -979,4 +988,4 @@ def test_format_features():
r = capa.rules.Rule.from_yaml(rule)
children = list(r.statement.get_children())
assert (CHARACTERISTIC_PE in children) == True
assert (CHARACTERISTIC_ELF not in children) == True
assert (CHARACTERISTIC_ELF not in children) == True