viv: insn: fix OperandNumber reference

This commit is contained in:
Willi Ballenthin
2022-03-31 10:39:18 -06:00
parent 9da4ff10da
commit 856443319c

View File

@@ -17,7 +17,7 @@ import envi.archs.amd64.disasm
import capa.features.extractors.helpers
import capa.features.extractors.viv.helpers
from capa.features.insn import API, Number, Offset, Mnemonic, OperandOffset, OperandImmediate
from capa.features.insn import API, Number, Offset, Mnemonic, OperandOffset, OperandNumber
from capa.features.common import (
BITNESS_X32,
BITNESS_X64,
@@ -554,7 +554,7 @@ def extract_op_number_features(f, bb, insn, i, oper):
yield Number(v), insn.va
yield Number(v, bitness=get_bitness(f.vw)), insn.va
yield OperandImmediate(i, v), insn.va
yield OperandNumber(i, v), insn.va
def extract_op_offset_features(f, bb, insn, i, oper):