Support loading IDBs from projects in idb_import

Vector35 · Oct 14, 2024 · a9d3753 · a9d3753
1 parent f95056e
commit a9d3753
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/rust/examples/idb_import/src/lib.rs b/rust/examples/idb_import/src/lib.rs
@@ -20,8 +20,14 @@ use anyhow::Result;
 struct IDBDebugInfoParser;
 impl CustomDebugInfoParser for IDBDebugInfoParser {
     fn is_valid(&self, view: &BinaryView) -> bool {
-        view.file().filename().as_str().ends_with(".i64")
-            || view.file().filename().as_str().ends_with(".idb")
+        if let Some(project_file) = view.file().get_project_file() {
+            project_file.name().as_str().ends_with(".i64")
+                || project_file.name().as_str().ends_with(".idb")
+        }
+        else {
+            view.file().filename().as_str().ends_with(".i64")
+                || view.file().filename().as_str().ends_with(".idb")
+        }
     }
 
     fn parse_info(
@@ -44,7 +50,12 @@ impl CustomDebugInfoParser for IDBDebugInfoParser {
 struct TILDebugInfoParser;
 impl CustomDebugInfoParser for TILDebugInfoParser {
     fn is_valid(&self, view: &BinaryView) -> bool {
-        view.file().filename().as_str().ends_with(".til")
+        if let Some(project_file) = view.file().get_project_file() {
+            project_file.name().as_str().ends_with(".til")
+        }
+        else {
+            view.file().filename().as_str().ends_with(".til")
+        }
     }
 
     fn parse_info(