binutils 0.1.1

A Rust library that ease interacting with the binutils disassembly engine
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

// Guillaume Valadon <guillaume@valadon.net>
// binutils - utils.rs

use Error;
use bfd::{Bfd, arch_list};
use helpers;
use opcodes::DisassembleInfo;

pub fn disassemble_buffer(
    arch_name: &str,
    buffer: &[u8],
    offset: u64,
) -> Result<DisassembleInfo, Error> {
    // Create a bfd structure
    let mut bfd = Bfd::empty();

    // Check if the architecture is supported
    if !arch_list().iter().any(|arch| arch == arch_name) {
        let error = Error::CommonError(format!("Unsupported architecture ({})!", arch_name));
        return Err(error);
    }

    // Set bfd_arch and bfd_mach from the architecture name
    let _ = bfd.set_arch_mach(arch_name);

    // Create a disassemble_info structure
    let mut info = DisassembleInfo::new()?;

    // Configure the disassemble_info structure
    info.init_buffer(buffer, bfd, offset)?;

    Ok(info)
}

pub(crate) fn check_null_pointer<T>(pointer: *const T, message: &str) -> Result<(), Error> {
    if pointer.is_null() {
        Err(Error::NullPointerError(message.to_string()))
    } else {
        Ok(())
    }
}

pub fn opcode_buffer_append(string: &str) {
    unsafe {
        let tmp = match helpers::CURRENT_OPCODE {
            Some(ref o) => o,
            None => "",
        };
        helpers::CURRENT_OPCODE = Some(format!("{}{}", tmp, string));
    }
}

#[cfg(test)]
mod tests {
    #[test]
    fn test_opcode_buffer_append() {
        use helpers;
        use utils;

        unsafe {
            helpers::CURRENT_OPCODE = None;
            utils::opcode_buffer_append("te");
            assert_eq!(helpers::CURRENT_OPCODE, Some("te".to_string()));
            utils::opcode_buffer_append("st!");
            assert_eq!(helpers::CURRENT_OPCODE, Some("test!".to_string()));
            helpers::CURRENT_OPCODE = None;
        }
    }
}