idalib 0.9.0+9.3.260327

Idiomatic bindings to IDA SDK
Documentation 
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

use crate::ffi::ida::msg;
use crate::ffi::plugin::PlugModBridge;
use crate::idb::IDB;
use crate::{IDA, IDAError};

pub trait IDAPlugin: Sized + Send + Sync + 'static {
    fn init(ida: &mut IDA, idb: &mut IDB) -> Result<Self, IDAError>;

    fn run(&mut self, ida: &mut IDA, idb: &mut IDB, arg: usize) -> Result<(), IDAError>;

    fn term(&mut self, _ida: &mut IDA, _idb: &mut IDB) -> Result<(), IDAError> {
        Ok(())
    }
}

trait IDAPluginErased: Send + Sync {
    fn run(&mut self, ida: &mut IDA, idb: &mut IDB, arg: usize) -> Result<(), IDAError>;
    fn term(&mut self, ida: &mut IDA, idb: &mut IDB) -> Result<(), IDAError>;
}

impl<P: IDAPlugin> IDAPluginErased for P {
    fn run(&mut self, ida: &mut IDA, idb: &mut IDB, arg: usize) -> Result<(), IDAError> {
        IDAPlugin::run(self, ida, idb, arg)
    }

    fn term(&mut self, ida: &mut IDA, idb: &mut IDB) -> Result<(), IDAError> {
        IDAPlugin::term(self, ida, idb)
    }
}

#[doc(hidden)]
pub struct PlugmodWrapper {
    name: &'static str,
    plugin: Box<dyn IDAPluginErased>,
}

impl PlugmodWrapper {
    pub fn new(name: &'static str, plugin: impl IDAPlugin) -> Self {
        Self {
            name,
            plugin: Box::new(plugin),
        }
    }
}

impl PlugModBridge for PlugmodWrapper {
    fn run(&mut self, arg: usize) -> bool {
        let result = IDB::current().and_then(|mut idb| {
            let mut ida = IDA::new(&idb);
            self.plugin.run(&mut ida, &mut idb, arg)
        });
        match result {
            Ok(()) => true,
            Err(e) => {
                let _ = unsafe { msg(&format!("[{}] `run` failed: {e}\n", self.name)) };
                false
            }
        }
    }

    fn term(&mut self) {
        let result = IDB::current().and_then(|mut idb| {
            let mut ida = IDA::new(&idb);
            self.plugin.term(&mut ida, &mut idb)
        });
        if let Err(e) = result {
            let _ = unsafe { msg(&format!("[{}] `term` failed: {e}\n", self.name)) };
        }
    }
}