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

use core::convert::TryFrom;
use core::fmt;

use bad64_sys::*;

/// A shift applied to a register or immediate
#[derive(Clone, Copy, Debug, Hash, Eq, PartialEq)]
#[allow(non_camel_case_types)]
pub enum Shift {
    LSL(u32),
    LSR(u32),
    ASR(u32),
    ROR(u32),
    UXTW(u32),
    SXTW(u32),
    SXTX(u32),
    UXTX(u32),
    SXTB(u32),
    SXTH(u32),
    UXTH(u32),
    UXTB(u32),
    MSL(u32),
}

#[doc(hidden)]
#[allow(non_upper_case_globals)]
impl TryFrom<&bad64_sys::InstructionOperand> for Shift {
    type Error = ();

    fn try_from(oo: &bad64_sys::InstructionOperand) -> Result<Self, Self::Error> {
        match oo.shiftType {
            ShiftType::ShiftType_LSL => Ok(Shift::LSL(oo.shiftValue)),
            ShiftType::ShiftType_LSR => Ok(Shift::LSR(oo.shiftValue)),
            ShiftType::ShiftType_ASR => Ok(Shift::ASR(oo.shiftValue)),
            ShiftType::ShiftType_ROR => Ok(Shift::ROR(oo.shiftValue)),
            ShiftType::ShiftType_UXTW => Ok(Shift::UXTW(oo.shiftValue)),
            ShiftType::ShiftType_SXTW => Ok(Shift::SXTW(oo.shiftValue)),
            ShiftType::ShiftType_UXTX => Ok(Shift::UXTX(oo.shiftValue)),
            ShiftType::ShiftType_SXTX => Ok(Shift::SXTX(oo.shiftValue)),
            ShiftType::ShiftType_SXTB => Ok(Shift::SXTB(oo.shiftValue)),
            ShiftType::ShiftType_SXTH => Ok(Shift::SXTH(oo.shiftValue)),
            ShiftType::ShiftType_UXTH => Ok(Shift::UXTH(oo.shiftValue)),
            ShiftType::ShiftType_UXTB => Ok(Shift::UXTB(oo.shiftValue)),
            ShiftType::ShiftType_MSL => Ok(Shift::MSL(oo.shiftValue)),
            ShiftType::ShiftType_NONE | ShiftType::ShiftType_END => Err(()),
        }
    }
}

impl fmt::Display for Shift {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match *self {
            Self::LSL(a) => write!(f, "lsl #{:#x}", a),
            Self::LSR(a) => write!(f, "lsr #{:#x}", a),
            Self::ASR(a) => write!(f, "asr #{:#x}", a),
            Self::ROR(a) => write!(f, "ror #{:#x}", a),
            Self::UXTW(a) => {
                write!(f, "uxtw")?;
                if a != 0 {
                    write!(f, " #{:#x}", a)?;
                }
                Ok(())
            }
            Self::SXTW(a) => {
                write!(f, "sxtw")?;
                if a != 0 {
                    write!(f, " #{:#x}", a)?;
                }
                Ok(())
            }
            Self::UXTX(a) => {
                write!(f, "uxtx")?;
                if a != 0 {
                    write!(f, " #{:#x}", a)?;
                }
                Ok(())
            }
            Self::SXTX(a) => {
                write!(f, "sxtx")?;
                if a != 0 {
                    write!(f, " #{:#x}", a)?;
                }
                Ok(())
            }
            Self::SXTB(a) => {
                write!(f, "sxtb")?;
                if a != 0 {
                    write!(f, " #{:#x}", a)?;
                }
                Ok(())
            }
            Self::SXTH(a) => {
                write!(f, "sxth")?;
                if a != 0 {
                    write!(f, " #{:#x}", a)?;
                }
                Ok(())
            }
            Self::UXTH(a) => {
                write!(f, "uxth")?;
                if a != 0 {
                    write!(f, " #{:#x}", a)?;
                }
                Ok(())
            }
            Self::UXTB(a) => {
                write!(f, "uxtb")?;
                if a != 0 {
                    write!(f, " #{:#x}", a)?;
                }
                Ok(())
            }
            Self::MSL(a) => write!(f, "msl #{:#x}", a),
        }
    }
}