1 files changed, 37 insertions, 0 deletions
diff --git a/plugins/bntl_utils/cli/src/diff.rs b/plugins/bntl_utils/cli/src/diff.rs
new file mode 100644
index 00000000..1eedb970
--- /dev/null
+++ b/plugins/bntl_utils/cli/src/diff.rs
@@ -0,0 +1,37 @@
+use binaryninja::types::TypeLibrary;
+use bntl_utils::diff::TILDiff;
+use clap::Args;
+use std::path::PathBuf;
+
+#[derive(Debug, Args)]
+pub struct DiffArgs {
+    pub file_a: PathBuf,
+    pub file_b: PathBuf,
+    /// Path to write the `.diff` file to.
+    pub output_path: PathBuf,
+    /// Timeout in seconds for the diff operation to complete, if provided the diffing will begin
+    /// to approximate after the deadline has passed.
+    #[clap(long)]
+    pub timeout: Option<u64>,
+}
+
+impl DiffArgs {
+    pub fn execute(&self) {
+        let type_lib_a =
+            TypeLibrary::load_from_file(&self.file_a).expect("Failed to load type library");
+        let type_lib_b =
+            TypeLibrary::load_from_file(&self.file_b).expect("Failed to load type library");
+
+        let diff_result =
+            match TILDiff::new().diff((&self.file_a, &type_lib_a), (&self.file_b, &type_lib_b)) {
+                Ok(diff_result) => diff_result,
+                Err(err) => {
+                    tracing::error!("Failed to diff type libraries: {}", err);
+                    return;
+                }
+            };
+        tracing::info!("Similarity Ratio: {}", diff_result.ratio);
+        std::fs::write(&self.output_path, diff_result.diff).unwrap();
+        tracing::info!("Diff written to: {}", self.output_path.display());
+    }
+}