llvmir2hll/statment.cpp: remove observers when removing predecessors.

Related to #512.
avast · Mar 18, 2019 · 6f60474 · 6f60474
1 parent 301f4e6
commit 6f60474
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/llvmir2hll/ir/statement.cpp b/src/llvmir2hll/ir/statement.cpp
@@ -300,7 +300,8 @@ void Statement::removeStatement(ShPtr<Statement> stmt) {
 	// target's predecessors - target is not goto's successor.
 	if (auto gotoStmt = cast<GotoStmt>(stmt)) {
 		if (gotoStmt->getTarget()) {
-			gotoStmt->getTarget()->preds.erase(stmt);
+			gotoStmt->getTarget()->removeObserver(stmt);
+			gotoStmt->getTarget()->removePredecessor(stmt);
 			preserveLabel(stmt, gotoStmt->getTarget());
 		}
 	}
@@ -503,7 +504,8 @@ void Statement::replaceStatement(ShPtr<Statement> oldStmt,
 	// target's predecessors - target is not goto's successor.
 	if (auto gotoStmt = cast<GotoStmt>(oldStmt)) {
 		if (gotoStmt->getTarget()) {
-			gotoStmt->getTarget()->preds.erase(oldStmt);
+			gotoStmt->getTarget()->removeObserver(oldStmt);
+			gotoStmt->getTarget()->removePredecessor(oldStmt);
 		}
 	}