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

Update capa/features/extractors/dnfile/extractor.py

Browse Source 
Co-authored-by: Mike Hunhoff <mike.hunhoff@gmail.com>
This commit is contained in:
Willi Ballenthin
2022-06-14 17:01:20 -06:00
committed by GitHub
parent aff6191b11
commit df101e5a60
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
capa/features/extractors/dnfile/extractor.py
View File

@@ -62,7 +62,10 @@ class DnfileFeatureExtractor(FeatureExtractor):
def get_instructions(self, fh, bbh):
for insn in bbh.inner.instructions:
yield InsnHandle(address=DNTokenOffsetAddress(bbh.address.token, insn.offset - fh.inner.offset), inner=insn)
yield InsnHandle(
address=DNTokenOffsetAddress(bbh.address.token, insn.offset - (fh.inner.offset + fh.inner.header_size)),
inner=insn,
)
def extract_insn_features(self, fh, bbh, ih) -> Iterator[Tuple[Feature, Address]]:
yield from capa.features.extractors.dnfile.insn.extract_features(fh, bbh, ih)