lief 1.0.0

Official Rust bindings for LIEF
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use lief_ffi as ffi;

use crate::common::FromFFI;

pub mod immediate;
pub mod memory;
pub mod pc_relative;
pub mod register;

#[doc(inline)]
pub use register::Register;

#[doc(inline)]
pub use pc_relative::PCRelative;

#[doc(inline)]
pub use immediate::Immediate;

#[doc(inline)]
pub use memory::Memory;

/// Trait shared by **all** [`Operands`]
pub trait Operand {
    #[doc(hidden)]
    fn as_generic(&self) -> &ffi::asm_riscv_Operand;

    /// Pretty representation of the operand
    fn to_string(&self) -> String {
        self.as_generic().to_string().to_string()
    }
}

/// This enum represents the different kind of operands associated with [`super::Instruction`]
pub enum Operands {
    /// A register operand
    Reg(Register),

    /// A PC-relative operand
    PCRelative(PCRelative),

    /// An immediate value
    Imm(Immediate),

    /// A memory operand
    Mem(Memory),

    /// Operand that is not correctly supported
    Unknown(Unknown),
}

impl FromFFI<ffi::asm_riscv_Operand> for Operands {
    fn from_ffi(ptr: cxx::UniquePtr<ffi::asm_riscv_Operand>) -> Self {
        unsafe {
            let op_ref = ptr.as_ref().unwrap();
            if ffi::asm_riscv_operands_Memory::classof(op_ref) {
                let raw = {
                    type From = cxx::UniquePtr<ffi::asm_riscv_Operand>;
                    type To = cxx::UniquePtr<ffi::asm_riscv_operands_Memory>;
                    std::mem::transmute::<From, To>(ptr)
                };
                return Operands::Mem(Memory::from_ffi(raw));
            } else if ffi::asm_riscv_operands_Register::classof(op_ref) {
                let raw = {
                    type From = cxx::UniquePtr<ffi::asm_riscv_Operand>;
                    type To = cxx::UniquePtr<ffi::asm_riscv_operands_Register>;
                    std::mem::transmute::<From, To>(ptr)
                };
                return Operands::Reg(Register::from_ffi(raw));
            } else if ffi::asm_riscv_operands_Immediate::classof(op_ref) {
                let raw = {
                    type From = cxx::UniquePtr<ffi::asm_riscv_Operand>;
                    type To = cxx::UniquePtr<ffi::asm_riscv_operands_Immediate>;
                    std::mem::transmute::<From, To>(ptr)
                };
                return Operands::Imm(Immediate::from_ffi(raw));
            } else if ffi::asm_riscv_operands_PCRelative::classof(op_ref) {
                let raw = {
                    type From = cxx::UniquePtr<ffi::asm_riscv_Operand>;
                    type To = cxx::UniquePtr<ffi::asm_riscv_operands_PCRelative>;
                    std::mem::transmute::<From, To>(ptr)
                };
                return Operands::PCRelative(PCRelative::from_ffi(raw));
            }
            Operands::Unknown(Unknown::from_ffi(ptr))
        }
    }
}

impl Operand for Operands {
    #[doc(hidden)]
    fn as_generic(&self) -> &ffi::asm_riscv_Operand {
        match &self {
            Operands::Reg(op) => op.as_generic(),

            Operands::Imm(op) => op.as_generic(),

            Operands::Mem(op) => op.as_generic(),

            Operands::PCRelative(op) => op.as_generic(),

            Operands::Unknown(op) => op.as_generic(),
        }
    }
}

pub struct Unknown {
    ptr: cxx::UniquePtr<ffi::asm_riscv_Operand>,
}

impl FromFFI<ffi::asm_riscv_Operand> for Unknown {
    fn from_ffi(ptr: cxx::UniquePtr<ffi::asm_riscv_Operand>) -> Self {
        Self { ptr }
    }
}

impl Operand for Unknown {
    #[doc(hidden)]
    fn as_generic(&self) -> &ffi::asm_riscv_Operand {
        self.ptr.as_ref().unwrap()
    }
}