fix: potential endless recursion in hybrid instrumentation

discopop-project · Aug 27, 2024 · 4e6f868 · 4e6f868
1 parent 7c063d6
commit 4e6f868
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 7 deletions.
diff --git a/DiscoPoP/hybrid_analysis/InstructionCFG.cpp b/DiscoPoP/hybrid_analysis/InstructionCFG.cpp
@@ -30,10 +30,11 @@ InstructionCFG::InstructionCFG(dputil::VariableNameFinder *_VNF, Function &F) :
     }
     // Add edges from last instruction in current block to first instruction all
     // the successor blocks
-    if (previousInstruction != nullptr)
-      findAndAddFirstRelevantInstructionInSuccessorBlocks(&BB, previousInstruction);
+    if (previousInstruction != nullptr){
+      std::set<std::pair<BasicBlock*, Instruction*>> visited;
+      findAndAddFirstRelevantInstructionInSuccessorBlocks(&BB, previousInstruction, &visited);
+    }
   }
-
   // Conect entry/exit nodes
   for (auto instNode : Graph::getInstructionNodes()) {
     if (instNode != entry && instNode != exit) {
@@ -47,7 +48,17 @@ InstructionCFG::InstructionCFG(dputil::VariableNameFinder *_VNF, Function &F) :
 }
 
 void InstructionCFG::findAndAddFirstRelevantInstructionInSuccessorBlocks(BasicBlock *BB,
-                                                                         Instruction *previousInstruction) {
+                                                                         Instruction *previousInstruction,
+                                                                         std::set<std::pair<BasicBlock*, Instruction*>> *visited) {
+  // Check for and break out of cycles
+  std::pair<BasicBlock*, Instruction*> tmp_pair = std::make_pair(BB, previousInstruction);
+  const bool already_visited = visited->find(tmp_pair) != visited->end();
+  if(already_visited){
+    return;
+  }
+  // register visited node
+  visited->insert(tmp_pair);
+
   bool hasSuccessors = false;
   for (BasicBlock *S : successors(BB)) {
     hasSuccessors = true;
@@ -62,8 +73,9 @@ void InstructionCFG::findAndAddFirstRelevantInstructionInSuccessorBlocks(BasicBl
         Graph::addEdge(Graph::getInstructionNode(previousInstruction), exit);
       }
     }
-    if (S != BB)
-      findAndAddFirstRelevantInstructionInSuccessorBlocks(S, previousInstruction);
+    if (S != BB){
+      findAndAddFirstRelevantInstructionInSuccessorBlocks(S, previousInstruction, visited);
+    }
   next:;
   }
 }

diff --git a/DiscoPoP/hybrid_analysis/InstructionCFG.hpp b/DiscoPoP/hybrid_analysis/InstructionCFG.hpp
@@ -33,7 +33,7 @@ class InstructionCFG : public Graph<Instruction *> {
   dputil::VariableNameFinder *VNF;
   set<Instruction *> highlightedInstructionNodes;
 
-  void findAndAddFirstRelevantInstructionInSuccessorBlocks(BasicBlock *BB, Instruction *previousInstruction);
+  void findAndAddFirstRelevantInstructionInSuccessorBlocks(BasicBlock *BB, Instruction *previousInstruction, std::set<std::pair<BasicBlock*, Instruction*>> *visited);
 
 public:
   InstructionCFG(dputil::VariableNameFinder *_VNF, Function &F);