From 522b1b7395fb4cbdba40b49c73b1f7577524fb1a Mon Sep 17 00:00:00 2001
From: rollsafe <rollsafe@users.noreply.github.com>
Date: Fri, 31 May 2019 18:36:25 -0400
Subject: Add BinaryView.get_code_refs_from

---
 binaryview.cpp | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

(limited to 'binaryview.cpp')

diff --git a/binaryview.cpp b/binaryview.cpp
index 547d1fc7..c78e43cf 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -1442,6 +1442,38 @@ vector<ReferenceSource> BinaryView::GetCodeReferences(uint64_t addr, uint64_t le
 }
 
 
+vector<uint64_t> BinaryView::GetCodeReferencesFrom(ReferenceSource src)
+{
+	size_t count;
+	BNReferenceSource _src{ src.func->m_object, src.arch->m_object, src.addr };
+	uint64_t* refs = BNGetCodeReferencesFrom(m_object, &_src, &count);
+
+	vector<uint64_t> result;
+	result.reserve(count);
+	for (size_t i = 0; i < count; i++)
+		result[i] = refs[i];
+
+	BNFreeCodeReferencesFrom(refs);
+	return result;
+}
+
+
+vector<uint64_t> BinaryView::GetCodeReferencesFrom(ReferenceSource src, uint64_t len)
+{
+	size_t count;
+	BNReferenceSource _src{ src.func->m_object, src.arch->m_object, src.addr };
+	uint64_t* refs = BNGetCodeReferencesFromInRange(m_object, &_src, len, &count);
+
+	vector<uint64_t> result;
+	result.reserve(count);
+	for (size_t i = 0; i < count; i++)
+		result[i] = refs[i];
+
+	BNFreeCodeReferencesFrom(refs);
+	return result;
+}
+
+
 vector<uint64_t> BinaryView::GetDataReferences(uint64_t addr)
 {
 	size_t count;
-- 
cgit v1.3.1