better coverage for variable_clear()

bpftrace · Feb 26, 2019 · 34fdded · 34fdded
1 parent 83e7136
commit 34fdded
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/ast/codegen_llvm.cpp b/src/ast/codegen_llvm.cpp
@@ -1241,14 +1241,12 @@ void CodegenLLVM::visit(Probe &probe)
     BasicBlock *entry = BasicBlock::Create(module_->getContext(), "entry", func);
     b_.SetInsertPoint(entry);
 
-    variables_.clear(); // make sure variables are local to the probe
-
     ctx_ = func->arg_begin();
 
     if (probe.pred) {
       probe.pred->accept(*this);
     }
-
+    variables_.clear();
     for (Statement *stmt : *probe.stmts) {
       stmt->accept(*this);
     }
@@ -1308,6 +1306,7 @@ void CodegenLLVM::visit(Probe &probe)
         if (probe.pred) {
           probe.pred->accept(*this);
         }
+        variables_.clear();
         for (Statement *stmt : *probe.stmts) {
           stmt->accept(*this);
         }