From 5d0f774334c1bea3935001bf665d614c39a60d0f Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Wed, 4 Mar 2015 01:47:13 -0500
Subject: Add architecture to graph edges, and add edge routing data field for
 layout

---
 basicblock.cpp | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'basicblock.cpp')

diff --git a/basicblock.cpp b/basicblock.cpp
index 6f308307..a1becd10 100644
--- a/basicblock.cpp
+++ b/basicblock.cpp
@@ -45,14 +45,20 @@ uint64_t BasicBlock::GetLength() const
 }
 
 
-vector<BNBasicBlockEdge> BasicBlock::GetOutgoingEdges() const
+vector<BasicBlockEdge> BasicBlock::GetOutgoingEdges() const
 {
 	size_t count;
 	BNBasicBlockEdge* array = BNGetBasicBlockOutgoingEdges(m_block, &count);
 
-	vector<BNBasicBlockEdge> result;
+	vector<BasicBlockEdge> result;
 	for (size_t i = 0; i < count; i++)
-		result.push_back(array[i]);
+	{
+		BasicBlockEdge edge;
+		edge.type = array[i].type;
+		edge.target = array[i].target;
+		edge.arch = array[i].arch ? new CoreArchitecture(array[i].arch) : nullptr;
+		result.push_back(edge);
+	}
 
 	BNFreeBasicBlockOutgoingEdgeList(array);
 	return result;
-- 
cgit v1.3.1