codegen: fix use-after-free

tpapp · Jul 28, 2017 · d2f0b3c · d2f0b3c
1 parent 930a5f0
commit d2f0b3c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/codegen.cpp b/src/codegen.cpp
@@ -5695,10 +5695,15 @@ static std::unique_ptr<Module> emit_function(
  }
  }
  if (!have_real_use) {
+ Instruction *use = NULL;
  for (Use &U : root->uses()) {
+ if (use) // erase after the iterator moves on
+ use->eraseFromParent();
  User *RU = U.getUser();
- cast<Instruction>(RU)->eraseFromParent();
+ use = cast<Instruction>(RU);
  }
+ if (use)
+ use->eraseFromParent();
  root->eraseFromParent();
  if (store_value)
  store_value->eraseFromParent();