From 7e8de24c479672261b8a5c9911cac8800bad180b Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty.wagner@gmail.com>
Date: Wed, 3 Jan 2024 15:56:01 -0700
Subject: Add GetRelocationsAt API

---
 rust/src/binaryview.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'rust/src/binaryview.rs')

diff --git a/rust/src/binaryview.rs b/rust/src/binaryview.rs
index d6916e2c..9bcbbec2 100644
--- a/rust/src/binaryview.rs
+++ b/rust/src/binaryview.rs
@@ -42,6 +42,7 @@ use crate::linearview::LinearDisassemblyLine;
 use crate::linearview::LinearViewCursor;
 use crate::metadata::Metadata;
 use crate::platform::Platform;
+use crate::relocation::Relocation;
 use crate::section::{Section, SectionBuilder};
 use crate::segment::{Segment, SegmentBuilder};
 use crate::settings::Settings;
@@ -1272,6 +1273,14 @@ pub trait BinaryViewExt: BinaryViewBase {
             Array::new(handle, count, ())
         }
     }
+
+    fn get_relocations_at(&self, addr: u64) -> Array<Relocation> {
+        unsafe {
+            let mut count = 0;
+            let handle = BNGetRelocationsAt(self.as_ref().handle, addr, &mut count);
+            Array::new(handle, count, ())
+        }
+    }
 }
 
 impl<T: BinaryViewBase> BinaryViewExt for T {}
-- 
cgit v1.3.1