summaryrefslogtreecommitdiff
path: root/plugins/bntl_utils/cli/src/dump.rs
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/bntl_utils/cli/src/dump.rs')
-rw-r--r--plugins/bntl_utils/cli/src/dump.rs26
1 files changed, 26 insertions, 0 deletions
diff --git a/plugins/bntl_utils/cli/src/dump.rs b/plugins/bntl_utils/cli/src/dump.rs
new file mode 100644
index 00000000..fb25446b
--- /dev/null
+++ b/plugins/bntl_utils/cli/src/dump.rs
@@ -0,0 +1,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");
+ }
+}