From c5154605500b4a654276b103c6d2a19622febf97 Mon Sep 17 00:00:00 2001
From: rollsafe <rollsafe@users.noreply.github.com>
Date: Mon, 3 Jun 2019 18:43:32 -0400
Subject: Fix GetCodeReferencesFrom

---
 binaryview.cpp | 14 ++------------
 1 file changed, 2 insertions(+), 12 deletions(-)

(limited to 'binaryview.cpp')

diff --git a/binaryview.cpp b/binaryview.cpp
index c78e43cf..01640dcb 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -1447,12 +1447,7 @@ 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];
-
+	vector<uint64_t> result(refs, &refs[count]);
 	BNFreeCodeReferencesFrom(refs);
 	return result;
 }
@@ -1463,12 +1458,7 @@ vector<uint64_t> BinaryView::GetCodeReferencesFrom(ReferenceSource src, uint64_t
 	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];
-
+	vector<uint64_t> result(refs, &refs[count]);
 	BNFreeCodeReferencesFrom(refs);
 	return result;
 }
-- 
cgit v1.3.1