JIT support for BEGIN and END.

Author: headius

core/src/main/java/org/jruby/ir/IRBuilder.java

@@ -2917,6 +2917,9 @@ public Operand buildPostExe(PostExeNode postExeNode, IRScope s) {
         closureBuilder.addInstr(endClosure, new CopyInstr(endClosure.getCurrentModuleVariable(), new ScopeModule(0)));
         closureBuilder.build(postExeNode.getBodyNode(), endClosure);
 
+        // END does not have either explicit or implicit return, so we add one
+        closureBuilder.addInstr(endClosure, new ReturnInstr(new Nil()));
+
         // Add an instruction in 's' to record the end block in the 'topLevel' scope.
         // SSS FIXME: IR support for end-blocks that access vars in non-toplevel-scopes
         // might be broken currently. We could either fix it or consider dropping support
@@ -2936,6 +2939,9 @@ public Operand buildPreExe(PreExeNode preExeNode, IRScope s) {
         closureBuilder.addInstr(beginClosure, new CopyInstr(beginClosure.getCurrentModuleVariable(), new ScopeModule(0)));
         closureBuilder.build(preExeNode.getBodyNode(), beginClosure);
 
+        // BEGIN does not have either explicit or implicit return, so we add one
+        closureBuilder.addInstr(beginClosure, new ReturnInstr(new Nil()));
+
         // Record the begin block at IR build time
         s.getTopLevelScope().recordBeginBlock(beginClosure);
         return manager.getNil();

core/src/main/java/org/jruby/ir/instructions/RecordEndBlockInstr.java

@@ -1,9 +1,9 @@
 package org.jruby.ir.instructions;
 
 import org.jruby.ir.*;
-import org.jruby.ir.interpreter.BeginEndInterpreterContext;
 import org.jruby.ir.operands.Operand;
 import org.jruby.ir.operands.WrappedIRClosure;
+import org.jruby.ir.runtime.IRRuntimeHelpers;
 import org.jruby.ir.transformations.inlining.CloneInfo;
 import org.jruby.parser.StaticScope;
 import org.jruby.runtime.Block;
@@ -55,7 +55,7 @@ public Instr clone(CloneInfo ii) {
     @Override
     public Object interpret(ThreadContext context, StaticScope currScope, DynamicScope currDynScope, IRubyObject self, Object[] temp) {
         Block blk = (Block) endBlockClosure.retrieve(context, self, currScope, context.getCurrentScope(), temp);
-        context.runtime.pushExitBlock(context.runtime.newProc(Block.Type.LAMBDA, blk));
+        IRRuntimeHelpers.pushExitBlock(context, blk);
         return null;
     }
 

core/src/main/java/org/jruby/ir/interpreter/Interpreter.java

@@ -572,8 +572,7 @@ private static IRubyObject interpret(ThreadContext context, IRubyObject self,
         }
 
         // Control should never get here!
-        // SSS FIXME: But looks like BEGIN/END blocks get here -- needs fixing
-        return null;
+        throw context.runtime.newRuntimeError("BUG: interpreter fell through to end unexpectedly");
     }
 
     /*

core/src/main/java/org/jruby/ir/runtime/IRRuntimeHelpers.java

@@ -1287,4 +1287,9 @@ public static IRubyObject irNot(ThreadContext context, IRubyObject obj) {
     public static RaiseException newRequiredKeywordArgumentError(ThreadContext context, String name) {
         return context.runtime.newArgumentError("missing keyword: " + name);
     }
+
+    @JIT
+    public static void pushExitBlock(ThreadContext context, Block blk) {
+        context.runtime.pushExitBlock(context.runtime.newProc(Block.Type.LAMBDA, blk));
+    }
 }

core/src/main/java/org/jruby/ir/targets/JVMVisitor.java

@@ -1494,6 +1494,17 @@ public void ReceiveSelfInstr(ReceiveSelfInstr receiveselfinstr) {
         jvmStoreLocal(receiveselfinstr.getResult());
     }
 
+    @Override
+    public void RecordEndBlockInstr(RecordEndBlockInstr recordEndBlockInstr) {
+        jvmMethod().loadContext();
+
+        jvmMethod().loadContext();
+        visit(recordEndBlockInstr.getEndBlockClosure());
+        jvmMethod().invokeIRHelper("getBlockFromObject", sig(Block.class, ThreadContext.class, Object.class));
+
+        jvmMethod().invokeIRHelper("pushExitBlock", sig(void.class, ThreadContext.class, Block.class));
+    }
+
     @Override
     public void ReqdArgMultipleAsgnInstr(ReqdArgMultipleAsgnInstr reqdargmultipleasgninstr) {
         jvmMethod().loadContext();

spec/compiler/general_spec.rb

@@ -670,12 +670,12 @@ def self.remove; remove_method :gh1239; end
       run("def foo; yield; end; x = false; foo { break 5 if x; begin; ensure; x = true; redo; end; break 6}") {|result| expect(result).to eq 5 }
     end
 
-    it "compiles END BLocks", pending: "JIT support" do
+    it "compiles END Blocks" do
       # END block
       expect { run("END {}"){} }.to_not raise_error
     end
 
-    it "compiles BEGIN blocks", pending: "JIT support" do
+    it "compiles BEGIN blocks" do
       # BEGIN block
       run("BEGIN { $begin = 5 }; $begin") {|result| expect(result).to eq 5 }
     end