gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2026-07-05 12:17:16 -07:00
Code Issues Packages Projects Releases Wiki Activity

fixes 191

Browse Source
This commit is contained in:
Michael Hunhoff
2020-07-23 17:38:52 -06:00
committed by Ana María Martínez Gómez
parent 41a8199770
commit 4c5a104055
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
capa/features/extractors/viv/function.py
+3 -1
View File
@@ -78,7 +78,9 @@ def extract_function_loop(f):
for bb in f.basic_blocks:
if len(bb.instructions) > 0:
for bva, bflags in bb.instructions[-1].getBranches():
if bflags & vivisect.envi.BR_COND or bflags & vivisect.envi.BR_FALL or bflags & vivisect.envi.BR_TABLE:
# vivisect does not set branch flags for non-conditional jmp so add explicit check
if bflags & vivisect.envi.BR_COND or bflags & vivisect.envi.BR_FALL or bflags & vivisect.envi.BR_TABLE \
or bb.instructions[-1].mnem == "jmp":
edges.append((bb.va, bva))
if edges and loops.has_loop(edges):