mirror of
https://github.com/mandiant/capa.git
synced 2026-06-30 02:03:56 -07:00
fix code style
This commit is contained in:
@@ -92,7 +92,7 @@ class Shdr:
|
||||
buf: bytes
|
||||
|
||||
@classmethod
|
||||
def from_viv(cls, section, buf: bytes):
|
||||
def from_viv(cls, section, buf: bytes) -> "Shdr":
|
||||
return cls(
|
||||
section.sh_name,
|
||||
section.sh_type,
|
||||
|
||||
@@ -51,7 +51,7 @@ class VivisectFeatureExtractor(FeatureExtractor):
|
||||
def get_functions(self) -> Iterator[FunctionHandle]:
|
||||
cache = {}
|
||||
for va in sorted(self.vw.getFunctions()):
|
||||
yield FunctionHandle(address=AbsoluteVirtualAddress(va), inner=viv_utils.Function(self.vw, va), ctx={"cache": cache})
|
||||
yield FunctionHandle(address=AbsoluteVirtualAddress(va), inner=viv_utils.Function(self.vw, va), ctx={"cache":cache})
|
||||
|
||||
def extract_function_features(self, fh: FunctionHandle) -> Iterator[Tuple[Feature, Address]]:
|
||||
yield from capa.features.extractors.viv.function.extract_features(fh)
|
||||
|
||||
Reference in New Issue
Block a user