fix MutableBlockStmtGraph.insertBefore() : indexing of Stmt->Block wa…

…s missing
soot-oss · Sep 15, 2023 · 443db38 · 443db38
1 parent 38b9734
commit 443db38
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sootup.core/src/main/java/sootup/core/graph/MutableBlockStmtGraph.java b/sootup.core/src/main/java/sootup/core/graph/MutableBlockStmtGraph.java
@@ -847,7 +847,11 @@ public void insertBefore(
           (type, handler) ->
               successorBlock.addExceptionalSuccessorBlock(type, getOrCreateBlock(handler)));
       stmts.forEach(stmt -> addNodeToBlock(block, stmt));
-      tryMergeBlocks(block, successorBlock);
+      if (!tryMergeBlocks(block, successorBlock)) {
+        // update index: for splitted stmts
+        successorBlock.getStmts().forEach((stmt) -> stmtToBlock.put(stmt, successorBlock));
+        blocks.add(successorBlock);
+      }
     }
 
     if (beforeStmt == getStartingStmt()) {