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

elffile: fix mypy

Browse Source
This commit is contained in:
William Ballenthin
2021-09-13 13:32:09 -06:00
parent 297d9aaa32
commit c121e9219c
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
capa/features/extractors/elffile.py
View File

@@ -8,7 +8,7 @@
import io
import logging
import contextlib
from typing import Tuple
from typing import Tuple, Iterator
from elftools.elf.elffile import ELFFile, SymbolTableSection
@@ -79,9 +79,9 @@ def extract_file_arch(elf, **kwargs):
logger.warning("unsupported architecture: %s", arch)
def extract_file_features(elf: ELFFile, buf: bytes) -> Tuple[Feature, int]:
def extract_file_features(elf: ELFFile, buf: bytes) -> Iterator[Tuple[Feature, int]]:
for file_handler in FILE_HANDLERS:
for feature, va in file_handler(elf=elf, buf=buf):
for feature, va in file_handler(elf=elf, buf=buf): # type: ignore
yield feature, va