plugins/bntl_utils/cli/src/dump.rs


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

use binaryninja::types::TypeLibrary;
use bntl_utils::dump::TILDump;
use clap::Args;
use std::path::PathBuf;

#[derive(Debug, Args)]
pub struct DumpArgs {
    pub input: PathBuf,
    pub output_path: Option<PathBuf>,
}

impl DumpArgs {
    pub fn execute(&self) {
        let type_lib =
            TypeLibrary::load_from_file(&self.input).expect("Failed to load type library");
        let default_output_path = self.input.with_extension("h");
        let output_path = self.output_path.as_ref().unwrap_or(&default_output_path);
        let dependencies =
            bntl_utils::helper::path_to_type_libraries(&self.input.parent().unwrap());
        let printed_types = TILDump::new()
            .with_type_libs(dependencies)
            .dump(&type_lib)
            .expect("Failed to dump type library");
        std::fs::write(output_path, printed_types).expect("Failed to write type library header");
    }
}