From 5352b89b827c4f33fadbb21be19222eb2b714cc4 Mon Sep 17 00:00:00 2001 From: Rusty Wagner Date: Tue, 27 Jun 2017 22:44:07 -0400 Subject: Compute back edges in the core --- basicblock.cpp | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'basicblock.cpp') diff --git a/basicblock.cpp b/basicblock.cpp index 7eb37c57..c2a2bddf 100644 --- a/basicblock.cpp +++ b/basicblock.cpp @@ -125,6 +125,7 @@ vector BasicBlock::GetOutgoingEdges() const BasicBlockEdge edge; edge.type = array[i].type; edge.target = array[i].target ? new BasicBlock(BNNewBasicBlockReference(array[i].target)) : nullptr; + edge.backEdge = array[i].backEdge; result.push_back(edge); } @@ -144,6 +145,7 @@ vector BasicBlock::GetIncomingEdges() const BasicBlockEdge edge; edge.type = array[i].type; edge.target = array[i].target ? new BasicBlock(BNNewBasicBlockReference(array[i].target)) : nullptr; + edge.backEdge = array[i].backEdge; result.push_back(edge); } @@ -397,5 +399,10 @@ void BasicBlock::SetUserBasicBlockHighlight(uint8_t r, uint8_t g, uint8_t b, uin bool BasicBlock::IsBackEdge(BasicBlock* source, BasicBlock* target) { - return source->GetDominators().count(target) != 0; + for (auto& i : source->GetOutgoingEdges()) + { + if (i.target->GetObject() == target->GetObject()) + return i.backEdge; + } + return false; } -- cgit v1.3.1