Adjust stack position of NODE_RESCUE.

mattetti · Feb 24, 2013 · c6d2cde · c6d2cde
1 parent bbd50cc
commit c6d2cde
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/codegen.c b/src/codegen.c
@@ -1083,6 +1083,7 @@ codegen(codegen_scope *s, node *tree, int val)
           }
           if (n3->cdr->cdr->car) {
             codegen(s, n3->cdr->cdr->car, val);
+            if (val) pop();
           }
           tmp = new_label(s);
           genop(s, MKOP_sBx(OP_JMP, exend));
@@ -1102,6 +1103,9 @@ codegen(codegen_scope *s, node *tree, int val)
       if (tree->car) {
         codegen(s, tree->car, val);
       }
+      else if (val) {
+        push();
+      }
       dispatch_linked(s, exend);
       loop_pop(s, NOVAL);
     }