Add new APIs for querying data references

1 files changed, 20 insertions, 0 deletions

diff --git a/binaryview.cpp b/binaryview.cpp
index 4447494e..1302b0cb 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -1415,6 +1415,26 @@ vector<ReferenceSource> BinaryView::GetCodeReferences(uint64_t addr, uint64_t le
 }
 
 
+vector<uint64_t> BinaryView::GetDataReferences(uint64_t addr)
+{
+	size_t count;
+	uint64_t* refs = BNGetDataReferences(m_object, addr, &count);
+	vector<uint64_t> result(refs, &refs[count]);
+	BNFreeDataReferences(refs);
+	return result;
+}
+
+
+vector<uint64_t> BinaryView::GetDataReferences(uint64_t addr, uint64_t len)
+{
+	size_t count;
+	uint64_t* refs = BNGetDataReferencesInRange(m_object, addr, len, &count);
+	vector<uint64_t> result(refs, &refs[count]);
+	BNFreeDataReferences(refs);
+	return result;
+}
+
+
 Ref<Symbol> BinaryView::GetSymbolByAddress(uint64_t addr, const NameSpace& nameSpace)
 {
 	BNNameSpace ns = nameSpace.GetAPIObject();