update
This commit is contained in:
@@ -0,0 +1,63 @@
|
||||
use std::ops::Range;
|
||||
|
||||
use binaryninjacore_sys::BNGetHighLevelILIndexForInstruction;
|
||||
|
||||
use crate::basicblock::{BasicBlock, BlockContext};
|
||||
use crate::rc::Ref;
|
||||
|
||||
use super::{HighLevelILFunction, HighLevelILInstruction};
|
||||
|
||||
pub struct HighLevelILBlockIter {
|
||||
function: Ref<HighLevelILFunction>,
|
||||
range: Range<u64>,
|
||||
}
|
||||
|
||||
impl Iterator for HighLevelILBlockIter {
|
||||
type Item = HighLevelILInstruction;
|
||||
|
||||
fn next(&mut self) -> Option<Self::Item> {
|
||||
self.range
|
||||
.next()
|
||||
.map(|i| unsafe {
|
||||
BNGetHighLevelILIndexForInstruction(self.function.handle, i as usize)
|
||||
})
|
||||
.map(|i| HighLevelILInstruction::new(self.function.to_owned(), i))
|
||||
}
|
||||
}
|
||||
|
||||
pub struct HighLevelILBlock {
|
||||
pub(crate) function: Ref<HighLevelILFunction>,
|
||||
}
|
||||
|
||||
impl core::fmt::Debug for HighLevelILBlock {
|
||||
fn fmt(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result {
|
||||
write!(f, "mlil_bb {:?}", self.function)
|
||||
}
|
||||
}
|
||||
|
||||
impl BlockContext for HighLevelILBlock {
|
||||
type Iter = HighLevelILBlockIter;
|
||||
type Instruction = HighLevelILInstruction;
|
||||
|
||||
fn start(&self, block: &BasicBlock<Self>) -> HighLevelILInstruction {
|
||||
let expr_idx = unsafe {
|
||||
BNGetHighLevelILIndexForInstruction(self.function.handle, block.raw_start() as usize)
|
||||
};
|
||||
HighLevelILInstruction::new(self.function.to_owned(), expr_idx)
|
||||
}
|
||||
|
||||
fn iter(&self, block: &BasicBlock<Self>) -> HighLevelILBlockIter {
|
||||
HighLevelILBlockIter {
|
||||
function: self.function.to_owned(),
|
||||
range: block.raw_start()..block.raw_end(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Clone for HighLevelILBlock {
|
||||
fn clone(&self) -> Self {
|
||||
HighLevelILBlock {
|
||||
function: self.function.to_owned(),
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,111 @@
|
||||
use std::hash::{Hash, Hasher};
|
||||
|
||||
use binaryninjacore_sys::BNFreeHighLevelILFunction;
|
||||
use binaryninjacore_sys::BNGetHighLevelILBasicBlockList;
|
||||
use binaryninjacore_sys::BNGetHighLevelILInstructionCount;
|
||||
use binaryninjacore_sys::BNGetHighLevelILOwnerFunction;
|
||||
use binaryninjacore_sys::BNGetHighLevelILSSAForm;
|
||||
use binaryninjacore_sys::BNHighLevelILFunction;
|
||||
use binaryninjacore_sys::BNNewHighLevelILFunctionReference;
|
||||
|
||||
use crate::basicblock::BasicBlock;
|
||||
use crate::function::Function;
|
||||
use crate::rc::{Array, Ref, RefCountable};
|
||||
|
||||
use super::{HighLevelILBlock, HighLevelILInstruction, HighLevelILLiftedInstruction};
|
||||
|
||||
pub struct HighLevelILFunction {
|
||||
pub(crate) full_ast: bool,
|
||||
pub(crate) handle: *mut BNHighLevelILFunction,
|
||||
}
|
||||
|
||||
unsafe impl Send for HighLevelILFunction {}
|
||||
unsafe impl Sync for HighLevelILFunction {}
|
||||
|
||||
impl Eq for HighLevelILFunction {}
|
||||
impl PartialEq for HighLevelILFunction {
|
||||
fn eq(&self, rhs: &Self) -> bool {
|
||||
self.get_function().eq(&rhs.get_function())
|
||||
}
|
||||
}
|
||||
|
||||
impl Hash for HighLevelILFunction {
|
||||
fn hash<H: Hasher>(&self, state: &mut H) {
|
||||
self.get_function().hash(state)
|
||||
}
|
||||
}
|
||||
|
||||
impl HighLevelILFunction {
|
||||
pub(crate) unsafe fn ref_from_raw(
|
||||
handle: *mut BNHighLevelILFunction,
|
||||
full_ast: bool,
|
||||
) -> Ref<Self> {
|
||||
debug_assert!(!handle.is_null());
|
||||
Self { handle, full_ast }.to_owned()
|
||||
}
|
||||
|
||||
pub fn instruction_from_idx(&self, expr_idx: usize) -> HighLevelILInstruction {
|
||||
HighLevelILInstruction::new(self.to_owned(), expr_idx)
|
||||
}
|
||||
|
||||
pub fn lifted_instruction_from_idx(&self, expr_idx: usize) -> HighLevelILLiftedInstruction {
|
||||
self.instruction_from_idx(expr_idx).lift()
|
||||
}
|
||||
|
||||
pub fn instruction_count(&self) -> usize {
|
||||
unsafe { BNGetHighLevelILInstructionCount(self.handle) }
|
||||
}
|
||||
|
||||
pub fn ssa_form(&self) -> HighLevelILFunction {
|
||||
let ssa = unsafe { BNGetHighLevelILSSAForm(self.handle) };
|
||||
assert!(!ssa.is_null());
|
||||
HighLevelILFunction {
|
||||
handle: ssa,
|
||||
full_ast: self.full_ast,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn get_function(&self) -> Ref<Function> {
|
||||
unsafe {
|
||||
let func = BNGetHighLevelILOwnerFunction(self.handle);
|
||||
Function::from_raw(func)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn basic_blocks(&self) -> Array<BasicBlock<HighLevelILBlock>> {
|
||||
let mut count = 0;
|
||||
let blocks = unsafe { BNGetHighLevelILBasicBlockList(self.handle, &mut count) };
|
||||
let context = HighLevelILBlock {
|
||||
function: self.to_owned(),
|
||||
};
|
||||
|
||||
unsafe { Array::new(blocks, count, context) }
|
||||
}
|
||||
}
|
||||
|
||||
impl ToOwned for HighLevelILFunction {
|
||||
type Owned = Ref<Self>;
|
||||
|
||||
fn to_owned(&self) -> Self::Owned {
|
||||
unsafe { RefCountable::inc_ref(self) }
|
||||
}
|
||||
}
|
||||
|
||||
unsafe impl RefCountable for HighLevelILFunction {
|
||||
unsafe fn inc_ref(handle: &Self) -> Ref<Self> {
|
||||
Ref::new(Self {
|
||||
handle: BNNewHighLevelILFunctionReference(handle.handle),
|
||||
full_ast: handle.full_ast,
|
||||
})
|
||||
}
|
||||
|
||||
unsafe fn dec_ref(handle: &Self) {
|
||||
BNFreeHighLevelILFunction(handle.handle);
|
||||
}
|
||||
}
|
||||
|
||||
impl core::fmt::Debug for HighLevelILFunction {
|
||||
fn fmt(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result {
|
||||
write!(f, "<hlil func handle {:p}>", self.handle)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,991 @@
|
||||
use binaryninjacore_sys::BNFromVariableIdentifier;
|
||||
use binaryninjacore_sys::BNGetHighLevelILByIndex;
|
||||
use binaryninjacore_sys::BNHighLevelILOperation;
|
||||
|
||||
use crate::operand_iter::OperandIter;
|
||||
use crate::rc::Ref;
|
||||
use crate::types::{
|
||||
ConstantData, ILIntrinsic, RegisterValue, RegisterValueType, SSAVariable, Variable,
|
||||
};
|
||||
|
||||
use super::operation::*;
|
||||
use super::{HighLevelILFunction, HighLevelILLiftedInstruction, HighLevelILLiftedInstructionKind};
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct HighLevelILInstruction {
|
||||
pub function: Ref<HighLevelILFunction>,
|
||||
pub address: u64,
|
||||
pub kind: HighLevelILInstructionKind,
|
||||
}
|
||||
|
||||
#[derive(Copy, Clone)]
|
||||
pub enum HighLevelILInstructionKind {
|
||||
Nop,
|
||||
Break,
|
||||
Continue,
|
||||
Noret,
|
||||
Unreachable,
|
||||
Bp,
|
||||
Undef,
|
||||
Unimpl,
|
||||
Adc(BinaryOpCarry),
|
||||
Sbb(BinaryOpCarry),
|
||||
Rlc(BinaryOpCarry),
|
||||
Rrc(BinaryOpCarry),
|
||||
Add(BinaryOp),
|
||||
Sub(BinaryOp),
|
||||
And(BinaryOp),
|
||||
Or(BinaryOp),
|
||||
Xor(BinaryOp),
|
||||
Lsl(BinaryOp),
|
||||
Lsr(BinaryOp),
|
||||
Asr(BinaryOp),
|
||||
Rol(BinaryOp),
|
||||
Ror(BinaryOp),
|
||||
Mul(BinaryOp),
|
||||
MuluDp(BinaryOp),
|
||||
MulsDp(BinaryOp),
|
||||
Divu(BinaryOp),
|
||||
DivuDp(BinaryOp),
|
||||
Divs(BinaryOp),
|
||||
DivsDp(BinaryOp),
|
||||
Modu(BinaryOp),
|
||||
ModuDp(BinaryOp),
|
||||
Mods(BinaryOp),
|
||||
ModsDp(BinaryOp),
|
||||
CmpE(BinaryOp),
|
||||
CmpNe(BinaryOp),
|
||||
CmpSlt(BinaryOp),
|
||||
CmpUlt(BinaryOp),
|
||||
CmpSle(BinaryOp),
|
||||
CmpUle(BinaryOp),
|
||||
CmpSge(BinaryOp),
|
||||
CmpUge(BinaryOp),
|
||||
CmpSgt(BinaryOp),
|
||||
CmpUgt(BinaryOp),
|
||||
TestBit(BinaryOp),
|
||||
AddOverflow(BinaryOp),
|
||||
Fadd(BinaryOp),
|
||||
Fsub(BinaryOp),
|
||||
Fmul(BinaryOp),
|
||||
Fdiv(BinaryOp),
|
||||
FcmpE(BinaryOp),
|
||||
FcmpNe(BinaryOp),
|
||||
FcmpLt(BinaryOp),
|
||||
FcmpLe(BinaryOp),
|
||||
FcmpGe(BinaryOp),
|
||||
FcmpGt(BinaryOp),
|
||||
FcmpO(BinaryOp),
|
||||
FcmpUo(BinaryOp),
|
||||
ArrayIndex(ArrayIndex),
|
||||
ArrayIndexSsa(ArrayIndexSsa),
|
||||
Assign(Assign),
|
||||
AssignMemSsa(AssignMemSsa),
|
||||
AssignUnpack(AssignUnpack),
|
||||
AssignUnpackMemSsa(AssignUnpackMemSsa),
|
||||
Block(Block),
|
||||
Call(Call),
|
||||
Tailcall(Call),
|
||||
CallSsa(CallSsa),
|
||||
Case(Case),
|
||||
Const(Const),
|
||||
ConstPtr(Const),
|
||||
Import(Const),
|
||||
ConstData(ConstData),
|
||||
Deref(UnaryOp),
|
||||
AddressOf(UnaryOp),
|
||||
Neg(UnaryOp),
|
||||
Not(UnaryOp),
|
||||
Sx(UnaryOp),
|
||||
Zx(UnaryOp),
|
||||
LowPart(UnaryOp),
|
||||
BoolToInt(UnaryOp),
|
||||
UnimplMem(UnaryOp),
|
||||
Fsqrt(UnaryOp),
|
||||
Fneg(UnaryOp),
|
||||
Fabs(UnaryOp),
|
||||
FloatToInt(UnaryOp),
|
||||
IntToFloat(UnaryOp),
|
||||
FloatConv(UnaryOp),
|
||||
RoundToInt(UnaryOp),
|
||||
Floor(UnaryOp),
|
||||
Ceil(UnaryOp),
|
||||
Ftrunc(UnaryOp),
|
||||
DerefFieldSsa(DerefFieldSsa),
|
||||
DerefSsa(DerefSsa),
|
||||
ExternPtr(ExternPtr),
|
||||
FloatConst(FloatConst),
|
||||
For(ForLoop),
|
||||
ForSsa(ForLoopSsa),
|
||||
Goto(Label),
|
||||
Label(Label),
|
||||
If(If),
|
||||
Intrinsic(Intrinsic),
|
||||
IntrinsicSsa(IntrinsicSsa),
|
||||
Jump(Jump),
|
||||
MemPhi(MemPhi),
|
||||
Ret(Ret),
|
||||
Split(Split),
|
||||
StructField(StructField),
|
||||
DerefField(StructField),
|
||||
Switch(Switch),
|
||||
Syscall(Syscall),
|
||||
SyscallSsa(SyscallSsa),
|
||||
Trap(Trap),
|
||||
VarDeclare(Var),
|
||||
Var(Var),
|
||||
VarInit(VarInit),
|
||||
VarInitSsa(VarInitSsa),
|
||||
VarPhi(VarPhi),
|
||||
VarSsa(VarSsa),
|
||||
While(While),
|
||||
DoWhile(While),
|
||||
WhileSsa(WhileSsa),
|
||||
DoWhileSsa(WhileSsa),
|
||||
}
|
||||
impl HighLevelILInstruction {
|
||||
pub(crate) fn new(function: Ref<HighLevelILFunction>, idx: usize) -> Self {
|
||||
let op = unsafe { BNGetHighLevelILByIndex(function.handle, idx, function.full_ast) };
|
||||
use BNHighLevelILOperation::*;
|
||||
use HighLevelILInstructionKind as Op;
|
||||
let kind = match op.operation {
|
||||
HLIL_NOP => Op::Nop,
|
||||
HLIL_BREAK => Op::Break,
|
||||
HLIL_CONTINUE => Op::Continue,
|
||||
HLIL_NORET => Op::Noret,
|
||||
HLIL_UNREACHABLE => Op::Unreachable,
|
||||
HLIL_BP => Op::Bp,
|
||||
HLIL_UNDEF => Op::Undef,
|
||||
HLIL_UNIMPL => Op::Unimpl,
|
||||
HLIL_ADC => Op::Adc(BinaryOpCarry {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
carry: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_SBB => Op::Sbb(BinaryOpCarry {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
carry: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_RLC => Op::Rlc(BinaryOpCarry {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
carry: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_RRC => Op::Rrc(BinaryOpCarry {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
carry: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_ADD => Op::Add(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_SUB => Op::Sub(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_AND => Op::And(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_OR => Op::Or(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_XOR => Op::Xor(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_LSL => Op::Lsl(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_LSR => Op::Lsr(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_ASR => Op::Asr(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_ROL => Op::Rol(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_ROR => Op::Ror(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_MUL => Op::Mul(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_MULU_DP => Op::MuluDp(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_MULS_DP => Op::MulsDp(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_DIVU => Op::Divu(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_DIVU_DP => Op::DivuDp(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_DIVS => Op::Divs(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_DIVS_DP => Op::DivsDp(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_MODU => Op::Modu(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_MODU_DP => Op::ModuDp(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_MODS => Op::Mods(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_MODS_DP => Op::ModsDp(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_E => Op::CmpE(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_NE => Op::CmpNe(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_SLT => Op::CmpSlt(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_ULT => Op::CmpUlt(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_SLE => Op::CmpSle(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_ULE => Op::CmpUle(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_SGE => Op::CmpSge(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_UGE => Op::CmpUge(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_SGT => Op::CmpSgt(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CMP_UGT => Op::CmpUgt(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_TEST_BIT => Op::TestBit(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_ADD_OVERFLOW => Op::AddOverflow(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FADD => Op::Fadd(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FSUB => Op::Fsub(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FMUL => Op::Fmul(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FDIV => Op::Fdiv(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FCMP_E => Op::FcmpE(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FCMP_NE => Op::FcmpNe(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FCMP_LT => Op::FcmpLt(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FCMP_LE => Op::FcmpLe(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FCMP_GE => Op::FcmpGe(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FCMP_GT => Op::FcmpGt(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FCMP_O => Op::FcmpO(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_FCMP_UO => Op::FcmpUo(BinaryOp {
|
||||
left: op.operands[0] as usize,
|
||||
right: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_ARRAY_INDEX => Op::ArrayIndex(ArrayIndex {
|
||||
src: op.operands[0] as usize,
|
||||
index: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_ARRAY_INDEX_SSA => Op::ArrayIndexSsa(ArrayIndexSsa {
|
||||
src: op.operands[0] as usize,
|
||||
src_memory: op.operands[1],
|
||||
index: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_ASSIGN => Op::Assign(Assign {
|
||||
dest: op.operands[0] as usize,
|
||||
src: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_ASSIGN_MEM_SSA => Op::AssignMemSsa(AssignMemSsa {
|
||||
dest: op.operands[0] as usize,
|
||||
dest_memory: op.operands[1],
|
||||
src: op.operands[2] as usize,
|
||||
src_memory: op.operands[3],
|
||||
}),
|
||||
HLIL_ASSIGN_UNPACK => Op::AssignUnpack(AssignUnpack {
|
||||
num_dests: op.operands[0] as usize,
|
||||
first_dest: op.operands[1] as usize,
|
||||
src: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_ASSIGN_UNPACK_MEM_SSA => Op::AssignUnpackMemSsa(AssignUnpackMemSsa {
|
||||
num_dests: op.operands[0] as usize,
|
||||
first_dest: op.operands[1] as usize,
|
||||
dest_memory: op.operands[2],
|
||||
src: op.operands[3] as usize,
|
||||
src_memory: op.operands[4],
|
||||
}),
|
||||
HLIL_BLOCK => Op::Block(Block {
|
||||
num_params: op.operands[0] as usize,
|
||||
first_param: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_CALL => Op::Call(Call {
|
||||
dest: op.operands[0] as usize,
|
||||
num_params: op.operands[1] as usize,
|
||||
first_param: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_TAILCALL => Op::Tailcall(Call {
|
||||
dest: op.operands[0] as usize,
|
||||
num_params: op.operands[1] as usize,
|
||||
first_param: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_CALL_SSA => Op::CallSsa(CallSsa {
|
||||
dest: op.operands[0] as usize,
|
||||
num_params: op.operands[1] as usize,
|
||||
first_param: op.operands[2] as usize,
|
||||
dest_memory: op.operands[3],
|
||||
src_memory: op.operands[4],
|
||||
}),
|
||||
HLIL_CASE => Op::Case(Case {
|
||||
num_values: op.operands[0] as usize,
|
||||
first_value: op.operands[1] as usize,
|
||||
body: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_CONST => Op::Const(Const {
|
||||
constant: op.operands[0],
|
||||
}),
|
||||
HLIL_CONST_PTR => Op::ConstPtr(Const {
|
||||
constant: op.operands[0],
|
||||
}),
|
||||
HLIL_IMPORT => Op::Import(Const {
|
||||
constant: op.operands[0],
|
||||
}),
|
||||
HLIL_CONST_DATA => Op::ConstData(ConstData {
|
||||
constant_data_kind: op.operands[0] as u32,
|
||||
constant_data_value: op.operands[1] as i64,
|
||||
size: op.size,
|
||||
}),
|
||||
HLIL_DEREF => Op::Deref(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_ADDRESS_OF => Op::AddressOf(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_NEG => Op::Neg(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_NOT => Op::Not(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_SX => Op::Sx(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_ZX => Op::Zx(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_LOW_PART => Op::LowPart(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_BOOL_TO_INT => Op::BoolToInt(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_UNIMPL_MEM => Op::UnimplMem(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_FSQRT => Op::Fsqrt(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_FNEG => Op::Fneg(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_FABS => Op::Fabs(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_FLOAT_TO_INT => Op::FloatToInt(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_INT_TO_FLOAT => Op::IntToFloat(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_FLOAT_CONV => Op::FloatConv(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_ROUND_TO_INT => Op::RoundToInt(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_FLOOR => Op::Floor(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_CEIL => Op::Ceil(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_FTRUNC => Op::Ftrunc(UnaryOp {
|
||||
src: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_DEREF_FIELD_SSA => Op::DerefFieldSsa(DerefFieldSsa {
|
||||
src: op.operands[0] as usize,
|
||||
src_memory: op.operands[1],
|
||||
offset: op.operands[2],
|
||||
member_index: get_member_index(op.operands[3]),
|
||||
}),
|
||||
HLIL_DEREF_SSA => Op::DerefSsa(DerefSsa {
|
||||
src: op.operands[0] as usize,
|
||||
src_memory: op.operands[1],
|
||||
}),
|
||||
HLIL_EXTERN_PTR => Op::ExternPtr(ExternPtr {
|
||||
constant: op.operands[0],
|
||||
offset: op.operands[1],
|
||||
}),
|
||||
HLIL_FLOAT_CONST => Op::FloatConst(FloatConst {
|
||||
constant: get_float(op.operands[0], op.size),
|
||||
}),
|
||||
HLIL_FOR => Op::For(ForLoop {
|
||||
init: op.operands[0] as usize,
|
||||
condition: op.operands[1] as usize,
|
||||
update: op.operands[2] as usize,
|
||||
body: op.operands[3] as usize,
|
||||
}),
|
||||
HLIL_FOR_SSA => Op::ForSsa(ForLoopSsa {
|
||||
init: op.operands[0] as usize,
|
||||
condition_phi: op.operands[1] as usize,
|
||||
condition: op.operands[2] as usize,
|
||||
update: op.operands[3] as usize,
|
||||
body: op.operands[4] as usize,
|
||||
}),
|
||||
HLIL_GOTO => Op::Goto(Label {
|
||||
target: op.operands[0],
|
||||
}),
|
||||
HLIL_LABEL => Op::Label(Label {
|
||||
target: op.operands[0],
|
||||
}),
|
||||
HLIL_IF => Op::If(If {
|
||||
condition: op.operands[0] as usize,
|
||||
cond_true: op.operands[1] as usize,
|
||||
cond_false: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_INTRINSIC => Op::Intrinsic(Intrinsic {
|
||||
intrinsic: op.operands[0] as u32,
|
||||
num_params: op.operands[1] as usize,
|
||||
first_param: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_INTRINSIC_SSA => Op::IntrinsicSsa(IntrinsicSsa {
|
||||
intrinsic: op.operands[0] as u32,
|
||||
num_params: op.operands[1] as usize,
|
||||
first_param: op.operands[2] as usize,
|
||||
dest_memory: op.operands[3],
|
||||
src_memory: op.operands[4],
|
||||
}),
|
||||
HLIL_JUMP => Op::Jump(Jump {
|
||||
dest: op.operands[0] as usize,
|
||||
}),
|
||||
HLIL_MEM_PHI => Op::MemPhi(MemPhi {
|
||||
dest: op.operands[0],
|
||||
num_srcs: op.operands[1] as usize,
|
||||
first_src: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_RET => Op::Ret(Ret {
|
||||
num_srcs: op.operands[0] as usize,
|
||||
first_src: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_SPLIT => Op::Split(Split {
|
||||
high: op.operands[0] as usize,
|
||||
low: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_STRUCT_FIELD => Op::StructField(StructField {
|
||||
src: op.operands[0] as usize,
|
||||
offset: op.operands[1],
|
||||
member_index: get_member_index(op.operands[2]),
|
||||
}),
|
||||
HLIL_DEREF_FIELD => Op::DerefField(StructField {
|
||||
src: op.operands[0] as usize,
|
||||
offset: op.operands[1],
|
||||
member_index: get_member_index(op.operands[2]),
|
||||
}),
|
||||
HLIL_SWITCH => Op::Switch(Switch {
|
||||
condition: op.operands[0] as usize,
|
||||
default: op.operands[1] as usize,
|
||||
num_cases: op.operands[2] as usize,
|
||||
first_case: op.operands[3] as usize,
|
||||
}),
|
||||
HLIL_SYSCALL => Op::Syscall(Syscall {
|
||||
num_params: op.operands[0] as usize,
|
||||
first_param: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_SYSCALL_SSA => Op::SyscallSsa(SyscallSsa {
|
||||
num_params: op.operands[0] as usize,
|
||||
first_param: op.operands[1] as usize,
|
||||
dest_memory: op.operands[2],
|
||||
src_memory: op.operands[3],
|
||||
}),
|
||||
HLIL_TRAP => Op::Trap(Trap {
|
||||
vector: op.operands[0],
|
||||
}),
|
||||
HLIL_VAR_DECLARE => Op::VarDeclare(Var {
|
||||
var: get_var(op.operands[0]),
|
||||
}),
|
||||
HLIL_VAR => Op::Var(Var {
|
||||
var: get_var(op.operands[0]),
|
||||
}),
|
||||
HLIL_VAR_INIT => Op::VarInit(VarInit {
|
||||
dest: get_var(op.operands[0]),
|
||||
src: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_VAR_INIT_SSA => Op::VarInitSsa(VarInitSsa {
|
||||
dest: get_var_ssa((op.operands[0], op.operands[1] as usize)),
|
||||
src: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_VAR_PHI => Op::VarPhi(VarPhi {
|
||||
dest: get_var_ssa((op.operands[0], op.operands[1] as usize)),
|
||||
num_srcs: op.operands[2] as usize,
|
||||
first_src: op.operands[3] as usize,
|
||||
}),
|
||||
HLIL_VAR_SSA => Op::VarSsa(VarSsa {
|
||||
var: get_var_ssa((op.operands[0], op.operands[1] as usize)),
|
||||
}),
|
||||
HLIL_WHILE => Op::While(While {
|
||||
condition: op.operands[0] as usize,
|
||||
body: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_DO_WHILE => Op::DoWhile(While {
|
||||
condition: op.operands[0] as usize,
|
||||
body: op.operands[1] as usize,
|
||||
}),
|
||||
HLIL_WHILE_SSA => Op::WhileSsa(WhileSsa {
|
||||
condition_phi: op.operands[0] as usize,
|
||||
condition: op.operands[1] as usize,
|
||||
body: op.operands[2] as usize,
|
||||
}),
|
||||
HLIL_DO_WHILE_SSA => Op::DoWhileSsa(WhileSsa {
|
||||
condition_phi: op.operands[0] as usize,
|
||||
condition: op.operands[1] as usize,
|
||||
body: op.operands[2] as usize,
|
||||
}),
|
||||
};
|
||||
Self {
|
||||
function,
|
||||
address: op.address,
|
||||
kind,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn lift(&self) -> HighLevelILLiftedInstruction {
|
||||
use HighLevelILInstructionKind::*;
|
||||
use HighLevelILLiftedInstructionKind as Lifted;
|
||||
let kind = match self.kind {
|
||||
Nop => Lifted::Nop,
|
||||
Break => Lifted::Break,
|
||||
Continue => Lifted::Continue,
|
||||
Noret => Lifted::Noret,
|
||||
Unreachable => Lifted::Unreachable,
|
||||
Bp => Lifted::Bp,
|
||||
Undef => Lifted::Undef,
|
||||
Unimpl => Lifted::Unimpl,
|
||||
|
||||
Adc(op) => Lifted::Adc(self.lift_binary_op_carry(op)),
|
||||
Sbb(op) => Lifted::Sbb(self.lift_binary_op_carry(op)),
|
||||
Rlc(op) => Lifted::Rlc(self.lift_binary_op_carry(op)),
|
||||
Rrc(op) => Lifted::Rrc(self.lift_binary_op_carry(op)),
|
||||
|
||||
Add(op) => Lifted::Add(self.lift_binary_op(op)),
|
||||
Sub(op) => Lifted::Sub(self.lift_binary_op(op)),
|
||||
And(op) => Lifted::And(self.lift_binary_op(op)),
|
||||
Or(op) => Lifted::Or(self.lift_binary_op(op)),
|
||||
Xor(op) => Lifted::Xor(self.lift_binary_op(op)),
|
||||
Lsl(op) => Lifted::Lsl(self.lift_binary_op(op)),
|
||||
Lsr(op) => Lifted::Lsr(self.lift_binary_op(op)),
|
||||
Asr(op) => Lifted::Asr(self.lift_binary_op(op)),
|
||||
Rol(op) => Lifted::Rol(self.lift_binary_op(op)),
|
||||
Ror(op) => Lifted::Ror(self.lift_binary_op(op)),
|
||||
Mul(op) => Lifted::Mul(self.lift_binary_op(op)),
|
||||
MuluDp(op) => Lifted::MuluDp(self.lift_binary_op(op)),
|
||||
MulsDp(op) => Lifted::MulsDp(self.lift_binary_op(op)),
|
||||
Divu(op) => Lifted::Divu(self.lift_binary_op(op)),
|
||||
DivuDp(op) => Lifted::DivuDp(self.lift_binary_op(op)),
|
||||
Divs(op) => Lifted::Divs(self.lift_binary_op(op)),
|
||||
DivsDp(op) => Lifted::DivsDp(self.lift_binary_op(op)),
|
||||
Modu(op) => Lifted::Modu(self.lift_binary_op(op)),
|
||||
ModuDp(op) => Lifted::ModuDp(self.lift_binary_op(op)),
|
||||
Mods(op) => Lifted::Mods(self.lift_binary_op(op)),
|
||||
ModsDp(op) => Lifted::ModsDp(self.lift_binary_op(op)),
|
||||
CmpE(op) => Lifted::CmpE(self.lift_binary_op(op)),
|
||||
CmpNe(op) => Lifted::CmpNe(self.lift_binary_op(op)),
|
||||
CmpSlt(op) => Lifted::CmpSlt(self.lift_binary_op(op)),
|
||||
CmpUlt(op) => Lifted::CmpUlt(self.lift_binary_op(op)),
|
||||
CmpSle(op) => Lifted::CmpSle(self.lift_binary_op(op)),
|
||||
CmpUle(op) => Lifted::CmpUle(self.lift_binary_op(op)),
|
||||
CmpSge(op) => Lifted::CmpSge(self.lift_binary_op(op)),
|
||||
CmpUge(op) => Lifted::CmpUge(self.lift_binary_op(op)),
|
||||
CmpSgt(op) => Lifted::CmpSgt(self.lift_binary_op(op)),
|
||||
CmpUgt(op) => Lifted::CmpUgt(self.lift_binary_op(op)),
|
||||
TestBit(op) => Lifted::TestBit(self.lift_binary_op(op)),
|
||||
AddOverflow(op) => Lifted::AddOverflow(self.lift_binary_op(op)),
|
||||
Fadd(op) => Lifted::Fadd(self.lift_binary_op(op)),
|
||||
Fsub(op) => Lifted::Fsub(self.lift_binary_op(op)),
|
||||
Fmul(op) => Lifted::Fmul(self.lift_binary_op(op)),
|
||||
Fdiv(op) => Lifted::Fdiv(self.lift_binary_op(op)),
|
||||
FcmpE(op) => Lifted::FcmpE(self.lift_binary_op(op)),
|
||||
FcmpNe(op) => Lifted::FcmpNe(self.lift_binary_op(op)),
|
||||
FcmpLt(op) => Lifted::FcmpLt(self.lift_binary_op(op)),
|
||||
FcmpLe(op) => Lifted::FcmpLe(self.lift_binary_op(op)),
|
||||
FcmpGe(op) => Lifted::FcmpGe(self.lift_binary_op(op)),
|
||||
FcmpGt(op) => Lifted::FcmpGt(self.lift_binary_op(op)),
|
||||
FcmpO(op) => Lifted::FcmpO(self.lift_binary_op(op)),
|
||||
FcmpUo(op) => Lifted::FcmpUo(self.lift_binary_op(op)),
|
||||
|
||||
ArrayIndex(op) => Lifted::ArrayIndex(LiftedArrayIndex {
|
||||
src: self.lift_operand(op.src),
|
||||
index: self.lift_operand(op.index),
|
||||
}),
|
||||
ArrayIndexSsa(op) => Lifted::ArrayIndexSsa(LiftedArrayIndexSsa {
|
||||
src: self.lift_operand(op.src),
|
||||
src_memory: op.src_memory,
|
||||
index: self.lift_operand(op.index),
|
||||
}),
|
||||
Assign(op) => Lifted::Assign(LiftedAssign {
|
||||
dest: self.lift_operand(op.dest),
|
||||
src: self.lift_operand(op.src),
|
||||
}),
|
||||
AssignUnpack(op) => Lifted::AssignUnpack(LiftedAssignUnpack {
|
||||
dest: self.lift_instruction_list(op.first_dest, op.num_dests),
|
||||
src: self.lift_operand(op.src),
|
||||
}),
|
||||
AssignMemSsa(op) => Lifted::AssignMemSsa(LiftedAssignMemSsa {
|
||||
dest: self.lift_operand(op.dest),
|
||||
dest_memory: op.dest_memory,
|
||||
src: self.lift_operand(op.src),
|
||||
src_memory: op.src_memory,
|
||||
}),
|
||||
AssignUnpackMemSsa(op) => Lifted::AssignUnpackMemSsa(LiftedAssignUnpackMemSsa {
|
||||
dest: self.lift_instruction_list(op.first_dest, op.num_dests),
|
||||
dest_memory: op.dest_memory,
|
||||
src: self.lift_operand(op.src),
|
||||
src_memory: op.src_memory,
|
||||
}),
|
||||
Block(op) => Lifted::Block(LiftedBlock {
|
||||
body: self.lift_instruction_list(op.first_param, op.num_params),
|
||||
}),
|
||||
|
||||
Call(op) => Lifted::Call(self.lift_call(op)),
|
||||
Tailcall(op) => Lifted::Tailcall(self.lift_call(op)),
|
||||
CallSsa(op) => Lifted::CallSsa(LiftedCallSsa {
|
||||
dest: self.lift_operand(op.dest),
|
||||
params: self.lift_instruction_list(op.first_param, op.num_params),
|
||||
dest_memory: op.dest_memory,
|
||||
src_memory: op.src_memory,
|
||||
}),
|
||||
|
||||
Case(op) => Lifted::Case(LiftedCase {
|
||||
values: self.lift_instruction_list(op.first_value, op.num_values),
|
||||
body: self.lift_operand(op.body),
|
||||
}),
|
||||
Const(op) => Lifted::Const(op),
|
||||
ConstPtr(op) => Lifted::ConstPtr(op),
|
||||
Import(op) => Lifted::Import(op),
|
||||
ConstData(op) => Lifted::ConstData(LiftedConstData {
|
||||
constant_data: ConstantData::new(
|
||||
self.function.get_function(),
|
||||
RegisterValue {
|
||||
state: RegisterValueType::from_raw_value(op.constant_data_kind).unwrap(),
|
||||
value: op.constant_data_value,
|
||||
offset: 0,
|
||||
size: op.size,
|
||||
},
|
||||
),
|
||||
}),
|
||||
|
||||
Deref(op) => Lifted::Deref(self.lift_unary_op(op)),
|
||||
AddressOf(op) => Lifted::AddressOf(self.lift_unary_op(op)),
|
||||
Neg(op) => Lifted::Neg(self.lift_unary_op(op)),
|
||||
Not(op) => Lifted::Not(self.lift_unary_op(op)),
|
||||
Sx(op) => Lifted::Sx(self.lift_unary_op(op)),
|
||||
Zx(op) => Lifted::Zx(self.lift_unary_op(op)),
|
||||
LowPart(op) => Lifted::LowPart(self.lift_unary_op(op)),
|
||||
BoolToInt(op) => Lifted::BoolToInt(self.lift_unary_op(op)),
|
||||
UnimplMem(op) => Lifted::UnimplMem(self.lift_unary_op(op)),
|
||||
Fsqrt(op) => Lifted::Fsqrt(self.lift_unary_op(op)),
|
||||
Fneg(op) => Lifted::Fneg(self.lift_unary_op(op)),
|
||||
Fabs(op) => Lifted::Fabs(self.lift_unary_op(op)),
|
||||
FloatToInt(op) => Lifted::FloatToInt(self.lift_unary_op(op)),
|
||||
IntToFloat(op) => Lifted::IntToFloat(self.lift_unary_op(op)),
|
||||
FloatConv(op) => Lifted::FloatConv(self.lift_unary_op(op)),
|
||||
RoundToInt(op) => Lifted::RoundToInt(self.lift_unary_op(op)),
|
||||
Floor(op) => Lifted::Floor(self.lift_unary_op(op)),
|
||||
Ceil(op) => Lifted::Ceil(self.lift_unary_op(op)),
|
||||
Ftrunc(op) => Lifted::Ftrunc(self.lift_unary_op(op)),
|
||||
|
||||
DerefFieldSsa(op) => Lifted::DerefFieldSsa(LiftedDerefFieldSsa {
|
||||
src: self.lift_operand(op.src),
|
||||
src_memory: op.src_memory,
|
||||
offset: op.offset,
|
||||
member_index: op.member_index,
|
||||
}),
|
||||
DerefSsa(op) => Lifted::DerefSsa(LiftedDerefSsa {
|
||||
src: self.lift_operand(op.src),
|
||||
src_memory: op.src_memory,
|
||||
}),
|
||||
ExternPtr(op) => Lifted::ExternPtr(op),
|
||||
FloatConst(op) => Lifted::FloatConst(op),
|
||||
For(op) => Lifted::For(LiftedForLoop {
|
||||
init: self.lift_operand(op.init),
|
||||
condition: self.lift_operand(op.condition),
|
||||
update: self.lift_operand(op.update),
|
||||
body: self.lift_operand(op.body),
|
||||
}),
|
||||
Goto(op) => Lifted::Goto(self.lift_label(op)),
|
||||
Label(op) => Lifted::Label(self.lift_label(op)),
|
||||
ForSsa(op) => Lifted::ForSsa(LiftedForLoopSsa {
|
||||
init: self.lift_operand(op.init),
|
||||
condition_phi: self.lift_operand(op.condition_phi),
|
||||
condition: self.lift_operand(op.condition),
|
||||
update: self.lift_operand(op.update),
|
||||
body: self.lift_operand(op.body),
|
||||
}),
|
||||
If(op) => Lifted::If(LiftedIf {
|
||||
condition: self.lift_operand(op.condition),
|
||||
cond_true: self.lift_operand(op.cond_true),
|
||||
cond_false: self.lift_operand(op.cond_false),
|
||||
}),
|
||||
Intrinsic(op) => Lifted::Intrinsic(LiftedIntrinsic {
|
||||
intrinsic: ILIntrinsic::new(self.function.get_function().arch(), op.intrinsic),
|
||||
params: self.lift_instruction_list(op.first_param, op.num_params),
|
||||
}),
|
||||
IntrinsicSsa(op) => Lifted::IntrinsicSsa(LiftedIntrinsicSsa {
|
||||
intrinsic: ILIntrinsic::new(self.function.get_function().arch(), op.intrinsic),
|
||||
params: self.lift_instruction_list(op.first_param, op.num_params),
|
||||
dest_memory: op.dest_memory,
|
||||
src_memory: op.src_memory,
|
||||
}),
|
||||
Jump(op) => Lifted::Jump(LiftedJump {
|
||||
dest: self.lift_operand(op.dest),
|
||||
}),
|
||||
MemPhi(op) => Lifted::MemPhi(LiftedMemPhi {
|
||||
dest: op.dest,
|
||||
src: OperandIter::new(&*self.function, op.first_src, op.num_srcs).collect(),
|
||||
}),
|
||||
Ret(op) => Lifted::Ret(LiftedRet {
|
||||
src: self.lift_instruction_list(op.first_src, op.num_srcs),
|
||||
}),
|
||||
Split(op) => Lifted::Split(LiftedSplit {
|
||||
high: self.lift_operand(op.high),
|
||||
low: self.lift_operand(op.low),
|
||||
}),
|
||||
StructField(op) => Lifted::StructField(self.lift_struct_field(op)),
|
||||
DerefField(op) => Lifted::DerefField(self.lift_struct_field(op)),
|
||||
Switch(op) => Lifted::Switch(LiftedSwitch {
|
||||
condition: self.lift_operand(op.condition),
|
||||
default: self.lift_operand(op.default),
|
||||
cases: self.lift_instruction_list(op.first_case, op.num_cases),
|
||||
}),
|
||||
Syscall(op) => Lifted::Syscall(LiftedSyscall {
|
||||
params: self.lift_instruction_list(op.first_param, op.num_params),
|
||||
}),
|
||||
SyscallSsa(op) => Lifted::SyscallSsa(LiftedSyscallSsa {
|
||||
params: self.lift_instruction_list(op.first_param, op.num_params),
|
||||
dest_memory: op.dest_memory,
|
||||
src_memory: op.src_memory,
|
||||
}),
|
||||
Trap(op) => Lifted::Trap(op),
|
||||
VarDeclare(op) => Lifted::VarDeclare(op),
|
||||
Var(op) => Lifted::Var(op),
|
||||
VarInit(op) => Lifted::VarInit(LiftedVarInit {
|
||||
dest: op.dest,
|
||||
src: self.lift_operand(op.src),
|
||||
}),
|
||||
VarInitSsa(op) => Lifted::VarInitSsa(LiftedVarInitSsa {
|
||||
dest: op.dest,
|
||||
src: self.lift_operand(op.src),
|
||||
}),
|
||||
VarPhi(op) => Lifted::VarPhi(LiftedVarPhi {
|
||||
dest: op.dest,
|
||||
src: OperandIter::new(&*self.function, op.first_src, op.num_srcs)
|
||||
.ssa_vars()
|
||||
.collect(),
|
||||
}),
|
||||
VarSsa(op) => Lifted::VarSsa(op),
|
||||
|
||||
While(op) => Lifted::While(self.lift_while(op)),
|
||||
DoWhile(op) => Lifted::DoWhile(self.lift_while(op)),
|
||||
|
||||
WhileSsa(op) => Lifted::WhileSsa(self.lift_while_ssa(op)),
|
||||
DoWhileSsa(op) => Lifted::DoWhileSsa(self.lift_while_ssa(op)),
|
||||
};
|
||||
HighLevelILLiftedInstruction {
|
||||
function: self.function.clone(),
|
||||
address: self.address,
|
||||
kind,
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_operand(&self, expr_idx: usize) -> Box<HighLevelILLiftedInstruction> {
|
||||
Box::new(self.function.lifted_instruction_from_idx(expr_idx))
|
||||
}
|
||||
|
||||
fn lift_binary_op(&self, op: BinaryOp) -> LiftedBinaryOp {
|
||||
LiftedBinaryOp {
|
||||
left: self.lift_operand(op.left),
|
||||
right: self.lift_operand(op.right),
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_binary_op_carry(&self, op: BinaryOpCarry) -> LiftedBinaryOpCarry {
|
||||
LiftedBinaryOpCarry {
|
||||
left: self.lift_operand(op.left),
|
||||
right: self.lift_operand(op.right),
|
||||
carry: self.lift_operand(op.carry),
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_unary_op(&self, op: UnaryOp) -> LiftedUnaryOp {
|
||||
LiftedUnaryOp {
|
||||
src: self.lift_operand(op.src),
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_label(&self, op: Label) -> LiftedLabel {
|
||||
LiftedLabel {
|
||||
target: GotoLabel {
|
||||
function: self.function.get_function(),
|
||||
target: op.target,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_call(&self, op: Call) -> LiftedCall {
|
||||
LiftedCall {
|
||||
dest: self.lift_operand(op.dest),
|
||||
params: OperandIter::new(&*self.function, op.first_param, op.num_params)
|
||||
.exprs()
|
||||
.map(|expr| expr.lift())
|
||||
.collect(),
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_while(&self, op: While) -> LiftedWhile {
|
||||
LiftedWhile {
|
||||
condition: self.lift_operand(op.condition),
|
||||
body: self.lift_operand(op.body),
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_while_ssa(&self, op: WhileSsa) -> LiftedWhileSsa {
|
||||
LiftedWhileSsa {
|
||||
condition_phi: self.lift_operand(op.condition_phi),
|
||||
condition: self.lift_operand(op.condition),
|
||||
body: self.lift_operand(op.body),
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_struct_field(&self, op: StructField) -> LiftedStructField {
|
||||
LiftedStructField {
|
||||
src: self.lift_operand(op.src),
|
||||
offset: op.offset,
|
||||
member_index: op.member_index,
|
||||
}
|
||||
}
|
||||
|
||||
fn lift_instruction_list(
|
||||
&self,
|
||||
first_instruction: usize,
|
||||
num_instructions: usize,
|
||||
) -> Vec<HighLevelILLiftedInstruction> {
|
||||
OperandIter::new(&*self.function, first_instruction, num_instructions)
|
||||
.exprs()
|
||||
.map(|expr| expr.lift())
|
||||
.collect()
|
||||
}
|
||||
}
|
||||
|
||||
impl core::fmt::Debug for HighLevelILInstruction {
|
||||
fn fmt(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result {
|
||||
write!(
|
||||
f,
|
||||
"<{} at 0x{:08}>",
|
||||
core::any::type_name::<Self>(),
|
||||
self.address,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
fn get_float(value: u64, size: usize) -> f64 {
|
||||
match size {
|
||||
4 => f32::from_bits(value as u32) as f64,
|
||||
8 => f64::from_bits(value),
|
||||
// TODO how to handle this value?
|
||||
size => todo!("float size {}", size),
|
||||
}
|
||||
}
|
||||
|
||||
fn get_var(id: u64) -> Variable {
|
||||
unsafe { Variable::from_raw(BNFromVariableIdentifier(id)) }
|
||||
}
|
||||
|
||||
fn get_member_index(idx: u64) -> Option<usize> {
|
||||
(idx as i64 > 0).then_some(idx as usize)
|
||||
}
|
||||
|
||||
fn get_var_ssa(input: (u64, usize)) -> SSAVariable {
|
||||
SSAVariable::new(get_var(input.0), input.1)
|
||||
}
|
||||
@@ -0,0 +1,325 @@
|
||||
use super::{operation::*, HighLevelILFunction};
|
||||
|
||||
use crate::rc::Ref;
|
||||
use crate::types::{ConstantData, ILIntrinsic, SSAVariable, Variable};
|
||||
|
||||
#[derive(Clone)]
|
||||
pub enum HighLevelILLiftedOperand {
|
||||
ConstantData(ConstantData),
|
||||
Expr(HighLevelILLiftedInstruction),
|
||||
ExprList(Vec<HighLevelILLiftedInstruction>),
|
||||
Float(f64),
|
||||
Int(u64),
|
||||
IntList(Vec<u64>),
|
||||
Intrinsic(ILIntrinsic),
|
||||
Label(GotoLabel),
|
||||
MemberIndex(Option<usize>),
|
||||
Var(Variable),
|
||||
VarSsa(SSAVariable),
|
||||
VarSsaList(Vec<SSAVariable>),
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct HighLevelILLiftedInstruction {
|
||||
pub function: Ref<HighLevelILFunction>,
|
||||
pub address: u64,
|
||||
pub kind: HighLevelILLiftedInstructionKind,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub enum HighLevelILLiftedInstructionKind {
|
||||
Nop,
|
||||
Break,
|
||||
Continue,
|
||||
Noret,
|
||||
Unreachable,
|
||||
Bp,
|
||||
Undef,
|
||||
Unimpl,
|
||||
Adc(LiftedBinaryOpCarry),
|
||||
Sbb(LiftedBinaryOpCarry),
|
||||
Rlc(LiftedBinaryOpCarry),
|
||||
Rrc(LiftedBinaryOpCarry),
|
||||
Add(LiftedBinaryOp),
|
||||
Sub(LiftedBinaryOp),
|
||||
And(LiftedBinaryOp),
|
||||
Or(LiftedBinaryOp),
|
||||
Xor(LiftedBinaryOp),
|
||||
Lsl(LiftedBinaryOp),
|
||||
Lsr(LiftedBinaryOp),
|
||||
Asr(LiftedBinaryOp),
|
||||
Rol(LiftedBinaryOp),
|
||||
Ror(LiftedBinaryOp),
|
||||
Mul(LiftedBinaryOp),
|
||||
MuluDp(LiftedBinaryOp),
|
||||
MulsDp(LiftedBinaryOp),
|
||||
Divu(LiftedBinaryOp),
|
||||
DivuDp(LiftedBinaryOp),
|
||||
Divs(LiftedBinaryOp),
|
||||
DivsDp(LiftedBinaryOp),
|
||||
Modu(LiftedBinaryOp),
|
||||
ModuDp(LiftedBinaryOp),
|
||||
Mods(LiftedBinaryOp),
|
||||
ModsDp(LiftedBinaryOp),
|
||||
CmpE(LiftedBinaryOp),
|
||||
CmpNe(LiftedBinaryOp),
|
||||
CmpSlt(LiftedBinaryOp),
|
||||
CmpUlt(LiftedBinaryOp),
|
||||
CmpSle(LiftedBinaryOp),
|
||||
CmpUle(LiftedBinaryOp),
|
||||
CmpSge(LiftedBinaryOp),
|
||||
CmpUge(LiftedBinaryOp),
|
||||
CmpSgt(LiftedBinaryOp),
|
||||
CmpUgt(LiftedBinaryOp),
|
||||
TestBit(LiftedBinaryOp),
|
||||
AddOverflow(LiftedBinaryOp),
|
||||
Fadd(LiftedBinaryOp),
|
||||
Fsub(LiftedBinaryOp),
|
||||
Fmul(LiftedBinaryOp),
|
||||
Fdiv(LiftedBinaryOp),
|
||||
FcmpE(LiftedBinaryOp),
|
||||
FcmpNe(LiftedBinaryOp),
|
||||
FcmpLt(LiftedBinaryOp),
|
||||
FcmpLe(LiftedBinaryOp),
|
||||
FcmpGe(LiftedBinaryOp),
|
||||
FcmpGt(LiftedBinaryOp),
|
||||
FcmpO(LiftedBinaryOp),
|
||||
FcmpUo(LiftedBinaryOp),
|
||||
ArrayIndex(LiftedArrayIndex),
|
||||
ArrayIndexSsa(LiftedArrayIndexSsa),
|
||||
Assign(LiftedAssign),
|
||||
AssignMemSsa(LiftedAssignMemSsa),
|
||||
AssignUnpack(LiftedAssignUnpack),
|
||||
AssignUnpackMemSsa(LiftedAssignUnpackMemSsa),
|
||||
Block(LiftedBlock),
|
||||
Call(LiftedCall),
|
||||
Tailcall(LiftedCall),
|
||||
CallSsa(LiftedCallSsa),
|
||||
Case(LiftedCase),
|
||||
Const(Const),
|
||||
ConstPtr(Const),
|
||||
Import(Const),
|
||||
ConstData(LiftedConstData),
|
||||
Deref(LiftedUnaryOp),
|
||||
AddressOf(LiftedUnaryOp),
|
||||
Neg(LiftedUnaryOp),
|
||||
Not(LiftedUnaryOp),
|
||||
Sx(LiftedUnaryOp),
|
||||
Zx(LiftedUnaryOp),
|
||||
LowPart(LiftedUnaryOp),
|
||||
BoolToInt(LiftedUnaryOp),
|
||||
UnimplMem(LiftedUnaryOp),
|
||||
Fsqrt(LiftedUnaryOp),
|
||||
Fneg(LiftedUnaryOp),
|
||||
Fabs(LiftedUnaryOp),
|
||||
FloatToInt(LiftedUnaryOp),
|
||||
IntToFloat(LiftedUnaryOp),
|
||||
FloatConv(LiftedUnaryOp),
|
||||
RoundToInt(LiftedUnaryOp),
|
||||
Floor(LiftedUnaryOp),
|
||||
Ceil(LiftedUnaryOp),
|
||||
Ftrunc(LiftedUnaryOp),
|
||||
DerefFieldSsa(LiftedDerefFieldSsa),
|
||||
DerefSsa(LiftedDerefSsa),
|
||||
ExternPtr(ExternPtr),
|
||||
FloatConst(FloatConst),
|
||||
For(LiftedForLoop),
|
||||
ForSsa(LiftedForLoopSsa),
|
||||
Goto(LiftedLabel),
|
||||
Label(LiftedLabel),
|
||||
If(LiftedIf),
|
||||
Intrinsic(LiftedIntrinsic),
|
||||
IntrinsicSsa(LiftedIntrinsicSsa),
|
||||
Jump(LiftedJump),
|
||||
MemPhi(LiftedMemPhi),
|
||||
Ret(LiftedRet),
|
||||
Split(LiftedSplit),
|
||||
StructField(LiftedStructField),
|
||||
DerefField(LiftedStructField),
|
||||
Switch(LiftedSwitch),
|
||||
Syscall(LiftedSyscall),
|
||||
SyscallSsa(LiftedSyscallSsa),
|
||||
Trap(Trap),
|
||||
VarDeclare(Var),
|
||||
Var(Var),
|
||||
VarInit(LiftedVarInit),
|
||||
VarInitSsa(LiftedVarInitSsa),
|
||||
VarPhi(LiftedVarPhi),
|
||||
VarSsa(VarSsa),
|
||||
While(LiftedWhile),
|
||||
DoWhile(LiftedWhile),
|
||||
WhileSsa(LiftedWhileSsa),
|
||||
DoWhileSsa(LiftedWhileSsa),
|
||||
}
|
||||
|
||||
impl HighLevelILLiftedInstruction {
|
||||
pub fn operands(&self) -> Vec<(&'static str, HighLevelILLiftedOperand)> {
|
||||
use HighLevelILLiftedInstructionKind::*;
|
||||
use HighLevelILLiftedOperand as Operand;
|
||||
match &self.kind {
|
||||
Nop | Break | Continue | Noret | Unreachable | Bp | Undef | Unimpl => vec![],
|
||||
Adc(op) | Sbb(op) | Rlc(op) | Rrc(op) => vec![
|
||||
("left", Operand::Expr(*op.left.clone())),
|
||||
("right", Operand::Expr(*op.right.clone())),
|
||||
("carry", Operand::Expr(*op.carry.clone())),
|
||||
],
|
||||
Add(op) | Sub(op) | And(op) | Or(op) | Xor(op) | Lsl(op) | Lsr(op) | Asr(op)
|
||||
| Rol(op) | Ror(op) | Mul(op) | MuluDp(op) | MulsDp(op) | Divu(op) | DivuDp(op)
|
||||
| Divs(op) | DivsDp(op) | Modu(op) | ModuDp(op) | Mods(op) | ModsDp(op) | CmpE(op)
|
||||
| CmpNe(op) | CmpSlt(op) | CmpUlt(op) | CmpSle(op) | CmpUle(op) | CmpSge(op)
|
||||
| CmpUge(op) | CmpSgt(op) | CmpUgt(op) | TestBit(op) | AddOverflow(op) | Fadd(op)
|
||||
| Fsub(op) | Fmul(op) | Fdiv(op) | FcmpE(op) | FcmpNe(op) | FcmpLt(op) | FcmpLe(op)
|
||||
| FcmpGe(op) | FcmpGt(op) | FcmpO(op) | FcmpUo(op) => vec![
|
||||
("left", Operand::Expr(*op.left.clone())),
|
||||
("right", Operand::Expr(*op.right.clone())),
|
||||
],
|
||||
ArrayIndex(op) => vec![
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
("index", Operand::Expr(*op.index.clone())),
|
||||
],
|
||||
ArrayIndexSsa(op) => vec![
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
("src_memory", Operand::Int(op.src_memory)),
|
||||
("index", Operand::Expr(*op.index.clone())),
|
||||
],
|
||||
Assign(op) => vec![
|
||||
("dest", Operand::Expr(*op.dest.clone())),
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
],
|
||||
AssignMemSsa(op) => vec![
|
||||
("dest", Operand::Expr(*op.dest.clone())),
|
||||
("dest_memory", Operand::Int(op.dest_memory)),
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
("src_memory", Operand::Int(op.src_memory)),
|
||||
],
|
||||
AssignUnpack(op) => vec![
|
||||
("dest", Operand::ExprList(op.dest.clone())),
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
],
|
||||
AssignUnpackMemSsa(op) => vec![
|
||||
("dest", Operand::ExprList(op.dest.clone())),
|
||||
("dest_memory", Operand::Int(op.dest_memory)),
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
("src_memory", Operand::Int(op.src_memory)),
|
||||
],
|
||||
Block(op) => vec![("body", Operand::ExprList(op.body.clone()))],
|
||||
Call(op) | Tailcall(op) => vec![
|
||||
("dest", Operand::Expr(*op.dest.clone())),
|
||||
("params", Operand::ExprList(op.params.clone())),
|
||||
],
|
||||
CallSsa(op) => vec![
|
||||
("dest", Operand::Expr(*op.dest.clone())),
|
||||
("params", Operand::ExprList(op.params.clone())),
|
||||
("dest_memory", Operand::Int(op.dest_memory)),
|
||||
("src_memory", Operand::Int(op.src_memory)),
|
||||
],
|
||||
Case(op) => vec![
|
||||
("values", Operand::ExprList(op.values.clone())),
|
||||
("body", Operand::Expr(*op.body.clone())),
|
||||
],
|
||||
Const(op) | ConstPtr(op) | Import(op) => vec![("constant", Operand::Int(op.constant))],
|
||||
ConstData(op) => vec![(
|
||||
"constant_data",
|
||||
Operand::ConstantData(op.constant_data.clone()),
|
||||
)],
|
||||
Deref(op) | AddressOf(op) | Neg(op) | Not(op) | Sx(op) | Zx(op) | LowPart(op)
|
||||
| BoolToInt(op) | UnimplMem(op) | Fsqrt(op) | Fneg(op) | Fabs(op) | FloatToInt(op)
|
||||
| IntToFloat(op) | FloatConv(op) | RoundToInt(op) | Floor(op) | Ceil(op)
|
||||
| Ftrunc(op) => vec![("src", Operand::Expr(*op.src.clone()))],
|
||||
DerefFieldSsa(op) => vec![
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
("src_memory", Operand::Int(op.src_memory)),
|
||||
("offset", Operand::Int(op.offset)),
|
||||
("member_index", Operand::MemberIndex(op.member_index)),
|
||||
],
|
||||
DerefSsa(op) => vec![
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
("src_memory", Operand::Int(op.src_memory)),
|
||||
],
|
||||
ExternPtr(op) => vec![
|
||||
("constant", Operand::Int(op.constant)),
|
||||
("offset", Operand::Int(op.offset)),
|
||||
],
|
||||
FloatConst(op) => vec![("constant", Operand::Float(op.constant))],
|
||||
For(op) => vec![
|
||||
("init", Operand::Expr(*op.init.clone())),
|
||||
("condition", Operand::Expr(*op.condition.clone())),
|
||||
("update", Operand::Expr(*op.update.clone())),
|
||||
("body", Operand::Expr(*op.body.clone())),
|
||||
],
|
||||
ForSsa(op) => vec![
|
||||
("init", Operand::Expr(*op.init.clone())),
|
||||
("condition_phi", Operand::Expr(*op.condition_phi.clone())),
|
||||
("condition", Operand::Expr(*op.condition.clone())),
|
||||
("update", Operand::Expr(*op.update.clone())),
|
||||
("body", Operand::Expr(*op.body.clone())),
|
||||
],
|
||||
Goto(op) | Label(op) => vec![("target", Operand::Label(op.target.clone()))],
|
||||
If(op) => vec![
|
||||
("condition", Operand::Expr(*op.condition.clone())),
|
||||
("cond_true", Operand::Expr(*op.cond_true.clone())),
|
||||
("cond_false", Operand::Expr(*op.cond_false.clone())),
|
||||
],
|
||||
Intrinsic(op) => vec![
|
||||
("intrinsic", Operand::Intrinsic(op.intrinsic)),
|
||||
("params", Operand::ExprList(op.params.clone())),
|
||||
],
|
||||
IntrinsicSsa(op) => vec![
|
||||
("intrinsic", Operand::Intrinsic(op.intrinsic)),
|
||||
("params", Operand::ExprList(op.params.clone())),
|
||||
("dest_memory", Operand::Int(op.dest_memory)),
|
||||
("src_memory", Operand::Int(op.src_memory)),
|
||||
],
|
||||
Jump(op) => vec![("dest", Operand::Expr(*op.dest.clone()))],
|
||||
MemPhi(op) => vec![
|
||||
("dest", Operand::Int(op.dest)),
|
||||
("src", Operand::IntList(op.src.clone())),
|
||||
],
|
||||
Ret(op) => vec![("src", Operand::ExprList(op.src.clone()))],
|
||||
Split(op) => vec![
|
||||
("high", Operand::Expr(*op.high.clone())),
|
||||
("low", Operand::Expr(*op.low.clone())),
|
||||
],
|
||||
StructField(op) | DerefField(op) => vec![
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
("offset", Operand::Int(op.offset)),
|
||||
("member_index", Operand::MemberIndex(op.member_index)),
|
||||
],
|
||||
Switch(op) => vec![
|
||||
("condition", Operand::Expr(*op.condition.clone())),
|
||||
("default", Operand::Expr(*op.default.clone())),
|
||||
("cases", Operand::ExprList(op.cases.clone())),
|
||||
],
|
||||
Syscall(op) => vec![("params", Operand::ExprList(op.params.clone()))],
|
||||
SyscallSsa(op) => vec![
|
||||
("params", Operand::ExprList(op.params.clone())),
|
||||
("dest_memory", Operand::Int(op.dest_memory)),
|
||||
("src_memory", Operand::Int(op.src_memory)),
|
||||
],
|
||||
Trap(op) => vec![("vector", Operand::Int(op.vector))],
|
||||
VarDeclare(op) | Var(op) => vec![("var", Operand::Var(op.var))],
|
||||
VarInit(op) => vec![
|
||||
("dest", Operand::Var(op.dest)),
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
],
|
||||
VarInitSsa(op) => vec![
|
||||
("dest", Operand::VarSsa(op.dest)),
|
||||
("src", Operand::Expr(*op.src.clone())),
|
||||
],
|
||||
VarPhi(op) => vec![
|
||||
("dest", Operand::VarSsa(op.dest)),
|
||||
("src", Operand::VarSsaList(op.src.clone())),
|
||||
],
|
||||
VarSsa(op) => vec![("var", Operand::VarSsa(op.var))],
|
||||
While(op) | DoWhile(op) => vec![
|
||||
("condition", Operand::Expr(*op.condition.clone())),
|
||||
("body", Operand::Expr(*op.body.clone())),
|
||||
],
|
||||
WhileSsa(op) | DoWhileSsa(op) => vec![
|
||||
("condition_phi", Operand::Expr(*op.condition_phi.clone())),
|
||||
("condition", Operand::Expr(*op.condition.clone())),
|
||||
("body", Operand::Expr(*op.body.clone())),
|
||||
],
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,10 @@
|
||||
mod block;
|
||||
mod function;
|
||||
mod instruction;
|
||||
mod lift;
|
||||
pub mod operation;
|
||||
|
||||
pub use self::block::*;
|
||||
pub use self::function::*;
|
||||
pub use self::instruction::*;
|
||||
pub use self::lift::*;
|
||||
@@ -0,0 +1,524 @@
|
||||
use binaryninjacore_sys::BNGetGotoLabelName;
|
||||
|
||||
use crate::function::Function;
|
||||
use crate::rc::Ref;
|
||||
use crate::types::{ConstantData, ILIntrinsic, SSAVariable, Variable};
|
||||
|
||||
use super::HighLevelILLiftedInstruction;
|
||||
|
||||
#[derive(Clone, Debug, PartialEq, Eq)]
|
||||
pub struct GotoLabel {
|
||||
pub(crate) function: Ref<Function>,
|
||||
pub(crate) target: u64,
|
||||
}
|
||||
|
||||
impl GotoLabel {
|
||||
pub fn name(&self) -> &str {
|
||||
let raw_str = unsafe { BNGetGotoLabelName(self.function.handle, self.target) };
|
||||
let c_str = unsafe { core::ffi::CStr::from_ptr(raw_str) };
|
||||
c_str.to_str().unwrap()
|
||||
}
|
||||
}
|
||||
|
||||
// ADC, SBB, RLC, RRC
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct BinaryOpCarry {
|
||||
pub left: usize,
|
||||
pub right: usize,
|
||||
pub carry: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedBinaryOpCarry {
|
||||
pub left: Box<HighLevelILLiftedInstruction>,
|
||||
pub right: Box<HighLevelILLiftedInstruction>,
|
||||
pub carry: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// ADD, SUB, AND, OR, XOR, LSL, LSR, ASR, ROL, ROR, MUL, MULU_DP, MULS_DP, DIVU, DIVU_DP, DIVS, DIVS_DP, MODU, MODU_DP, MODS, MODS_DP, CMP_E, CMP_NE, CMP_SLT, CMP_ULT, CMP_SLE, CMP_ULE, CMP_SGE, CMP_UGE, CMP_SGT, CMP_UGT, TEST_BIT, ADD_OVERFLOW, FADD, FSUB, FMUL, FDIV, FCMP_E, FCMP_NE, FCMP_LT, FCMP_LE, FCMP_GE, FCMP_GT, FCMP_O, FCMP_UO
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct BinaryOp {
|
||||
pub left: usize,
|
||||
pub right: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedBinaryOp {
|
||||
pub left: Box<HighLevelILLiftedInstruction>,
|
||||
pub right: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// ARRAY_INDEX
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct ArrayIndex {
|
||||
pub src: usize,
|
||||
pub index: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedArrayIndex {
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
pub index: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// ARRAY_INDEX_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct ArrayIndexSsa {
|
||||
pub src: usize,
|
||||
pub src_memory: u64,
|
||||
pub index: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedArrayIndexSsa {
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
pub src_memory: u64,
|
||||
pub index: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// ASSIGN
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Assign {
|
||||
pub dest: usize,
|
||||
pub src: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedAssign {
|
||||
pub dest: Box<HighLevelILLiftedInstruction>,
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// ASSIGN_MEM_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct AssignMemSsa {
|
||||
pub dest: usize,
|
||||
pub dest_memory: u64,
|
||||
pub src: usize,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedAssignMemSsa {
|
||||
pub dest: Box<HighLevelILLiftedInstruction>,
|
||||
pub dest_memory: u64,
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
|
||||
// ASSIGN_UNPACK
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct AssignUnpack {
|
||||
pub first_dest: usize,
|
||||
pub num_dests: usize,
|
||||
pub src: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedAssignUnpack {
|
||||
pub dest: Vec<HighLevelILLiftedInstruction>,
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// ASSIGN_UNPACK_MEM_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct AssignUnpackMemSsa {
|
||||
pub first_dest: usize,
|
||||
pub num_dests: usize,
|
||||
pub dest_memory: u64,
|
||||
pub src: usize,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedAssignUnpackMemSsa {
|
||||
pub dest: Vec<HighLevelILLiftedInstruction>,
|
||||
pub dest_memory: u64,
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
|
||||
// BLOCK
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Block {
|
||||
pub first_param: usize,
|
||||
pub num_params: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedBlock {
|
||||
pub body: Vec<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// CALL, TAILCALL
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Call {
|
||||
pub dest: usize,
|
||||
pub first_param: usize,
|
||||
pub num_params: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedCall {
|
||||
pub dest: Box<HighLevelILLiftedInstruction>,
|
||||
pub params: Vec<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// CALL_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct CallSsa {
|
||||
pub dest: usize,
|
||||
pub first_param: usize,
|
||||
pub num_params: usize,
|
||||
pub dest_memory: u64,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedCallSsa {
|
||||
pub dest: Box<HighLevelILLiftedInstruction>,
|
||||
pub params: Vec<HighLevelILLiftedInstruction>,
|
||||
pub dest_memory: u64,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
|
||||
// CASE
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Case {
|
||||
pub first_value: usize,
|
||||
pub num_values: usize,
|
||||
pub body: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedCase {
|
||||
pub values: Vec<HighLevelILLiftedInstruction>,
|
||||
pub body: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// CONST, CONST_PTR, IMPORT
|
||||
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
|
||||
pub struct Const {
|
||||
pub constant: u64,
|
||||
}
|
||||
|
||||
// CONST_DATA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct ConstData {
|
||||
pub constant_data_kind: u32,
|
||||
pub constant_data_value: i64,
|
||||
pub size: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedConstData {
|
||||
pub constant_data: ConstantData,
|
||||
}
|
||||
|
||||
// DEREF, ADDRESS_OF, NEG, NOT, SX, ZX, LOW_PART, BOOL_TO_INT, UNIMPL_MEM, FSQRT, FNEG, FABS, FLOAT_TO_INT, INT_TO_FLOAT, FLOAT_CONV, ROUND_TO_INT, FLOOR, CEIL, FTRUNC
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct UnaryOp {
|
||||
pub src: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedUnaryOp {
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// DEREF_FIELD_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct DerefFieldSsa {
|
||||
pub src: usize,
|
||||
pub src_memory: u64,
|
||||
pub offset: u64,
|
||||
pub member_index: Option<usize>,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedDerefFieldSsa {
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
pub src_memory: u64,
|
||||
pub offset: u64,
|
||||
pub member_index: Option<usize>,
|
||||
}
|
||||
|
||||
// DEREF_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct DerefSsa {
|
||||
pub src: usize,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedDerefSsa {
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
|
||||
// EXTERN_PTR
|
||||
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
|
||||
pub struct ExternPtr {
|
||||
pub constant: u64,
|
||||
pub offset: u64,
|
||||
}
|
||||
|
||||
// FLOAT_CONST
|
||||
#[derive(Copy, Clone, Debug, PartialEq)]
|
||||
pub struct FloatConst {
|
||||
pub constant: f64,
|
||||
}
|
||||
|
||||
// FOR
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct ForLoop {
|
||||
pub init: usize,
|
||||
pub condition: usize,
|
||||
pub update: usize,
|
||||
pub body: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedForLoop {
|
||||
pub init: Box<HighLevelILLiftedInstruction>,
|
||||
pub condition: Box<HighLevelILLiftedInstruction>,
|
||||
pub update: Box<HighLevelILLiftedInstruction>,
|
||||
pub body: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// FOR_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct ForLoopSsa {
|
||||
pub init: usize,
|
||||
pub condition_phi: usize,
|
||||
pub condition: usize,
|
||||
pub update: usize,
|
||||
pub body: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedForLoopSsa {
|
||||
pub init: Box<HighLevelILLiftedInstruction>,
|
||||
pub condition_phi: Box<HighLevelILLiftedInstruction>,
|
||||
pub condition: Box<HighLevelILLiftedInstruction>,
|
||||
pub update: Box<HighLevelILLiftedInstruction>,
|
||||
pub body: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// GOTO, LABEL
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Label {
|
||||
pub target: u64,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedLabel {
|
||||
pub target: GotoLabel,
|
||||
}
|
||||
|
||||
// IF
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct If {
|
||||
pub condition: usize,
|
||||
pub cond_true: usize,
|
||||
pub cond_false: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedIf {
|
||||
pub condition: Box<HighLevelILLiftedInstruction>,
|
||||
pub cond_true: Box<HighLevelILLiftedInstruction>,
|
||||
pub cond_false: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// INTRINSIC
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Intrinsic {
|
||||
pub intrinsic: u32,
|
||||
pub first_param: usize,
|
||||
pub num_params: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedIntrinsic {
|
||||
pub intrinsic: ILIntrinsic,
|
||||
pub params: Vec<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// INTRINSIC_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct IntrinsicSsa {
|
||||
pub intrinsic: u32,
|
||||
pub first_param: usize,
|
||||
pub num_params: usize,
|
||||
pub dest_memory: u64,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedIntrinsicSsa {
|
||||
pub intrinsic: ILIntrinsic,
|
||||
pub params: Vec<HighLevelILLiftedInstruction>,
|
||||
pub dest_memory: u64,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
|
||||
// JUMP
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Jump {
|
||||
pub dest: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedJump {
|
||||
pub dest: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// MEM_PHI
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct MemPhi {
|
||||
pub dest: u64,
|
||||
pub first_src: usize,
|
||||
pub num_srcs: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedMemPhi {
|
||||
pub dest: u64,
|
||||
pub src: Vec<u64>,
|
||||
}
|
||||
|
||||
// RET
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Ret {
|
||||
pub first_src: usize,
|
||||
pub num_srcs: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedRet {
|
||||
pub src: Vec<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// SPLIT
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Split {
|
||||
pub high: usize,
|
||||
pub low: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedSplit {
|
||||
pub high: Box<HighLevelILLiftedInstruction>,
|
||||
pub low: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// STRUCT_FIELD, DEREF_FIELD
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct StructField {
|
||||
pub src: usize,
|
||||
pub offset: u64,
|
||||
pub member_index: Option<usize>,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedStructField {
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
pub offset: u64,
|
||||
pub member_index: Option<usize>,
|
||||
}
|
||||
|
||||
// SWITCH
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Switch {
|
||||
pub condition: usize,
|
||||
pub default: usize,
|
||||
pub first_case: usize,
|
||||
pub num_cases: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedSwitch {
|
||||
pub condition: Box<HighLevelILLiftedInstruction>,
|
||||
pub default: Box<HighLevelILLiftedInstruction>,
|
||||
pub cases: Vec<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// SYSCALL
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct Syscall {
|
||||
pub first_param: usize,
|
||||
pub num_params: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedSyscall {
|
||||
pub params: Vec<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// SYSCALL_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct SyscallSsa {
|
||||
pub first_param: usize,
|
||||
pub num_params: usize,
|
||||
pub dest_memory: u64,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedSyscallSsa {
|
||||
pub params: Vec<HighLevelILLiftedInstruction>,
|
||||
pub dest_memory: u64,
|
||||
pub src_memory: u64,
|
||||
}
|
||||
|
||||
// TRAP
|
||||
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
|
||||
pub struct Trap {
|
||||
pub vector: u64,
|
||||
}
|
||||
|
||||
// VAR_DECLARE, VAR
|
||||
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
|
||||
pub struct Var {
|
||||
pub var: Variable,
|
||||
}
|
||||
|
||||
// VAR_INIT
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct VarInit {
|
||||
pub dest: Variable,
|
||||
pub src: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedVarInit {
|
||||
pub dest: Variable,
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// VAR_INIT_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct VarInitSsa {
|
||||
pub dest: SSAVariable,
|
||||
pub src: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedVarInitSsa {
|
||||
pub dest: SSAVariable,
|
||||
pub src: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// VAR_PHI
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct VarPhi {
|
||||
pub dest: SSAVariable,
|
||||
pub first_src: usize,
|
||||
pub num_srcs: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedVarPhi {
|
||||
pub dest: SSAVariable,
|
||||
pub src: Vec<SSAVariable>,
|
||||
}
|
||||
|
||||
// VAR_SSA
|
||||
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
|
||||
pub struct VarSsa {
|
||||
pub var: SSAVariable,
|
||||
}
|
||||
|
||||
// WHILE, DO_WHILE
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct While {
|
||||
pub condition: usize,
|
||||
pub body: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedWhile {
|
||||
pub condition: Box<HighLevelILLiftedInstruction>,
|
||||
pub body: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
|
||||
// WHILE_SSA, DO_WHILE_SSA
|
||||
#[derive(Copy, Clone)]
|
||||
pub struct WhileSsa {
|
||||
pub condition_phi: usize,
|
||||
pub condition: usize,
|
||||
pub body: usize,
|
||||
}
|
||||
#[derive(Clone, Debug, PartialEq)]
|
||||
pub struct LiftedWhileSsa {
|
||||
pub condition_phi: Box<HighLevelILLiftedInstruction>,
|
||||
pub condition: Box<HighLevelILLiftedInstruction>,
|
||||
pub body: Box<HighLevelILLiftedInstruction>,
|
||||
}
|
||||
Reference in New Issue
Block a user