Only compile iseqs with simple parameters. Enable get_local_wc1

1 files changed, 11 insertions, 3 deletions

diff --git a/ujit_codegen.c b/ujit_codegen.c
index ad5d1b18b0..50bc4f77b1 100644
--- a/ujit_codegen.c
+++ b/ujit_codegen.c
@@ -1112,6 +1112,8 @@ gen_opt_swb_cfunc(jitstate_t* jit, ctx_t* ctx, struct rb_call_data * cd, const r
     return true;
 }
 
+bool rb_simple_iseq_p(const rb_iseq_t *iseq);
+
 static bool
 gen_opt_swb_iseq(jitstate_t* jit, ctx_t* ctx, struct rb_call_data * cd, const rb_callable_method_entry_t *cme, int32_t argc)
 {
@@ -1120,13 +1122,19 @@ gen_opt_swb_iseq(jitstate_t* jit, ctx_t* ctx, struct rb_call_data * cd, const rb
     int num_params = iseq->body->param.size;
     int num_locals = iseq->body->local_table_size - num_params;
 
-    rb_gc_register_mark_object((VALUE)iseq); // FIXME: intentional LEAK!
-
     if (num_params != argc) {
         //fprintf(stderr, "param argc mismatch\n");
         return false;
     }
 
+    if (!rb_simple_iseq_p(iseq)) {
+        // Only handle iseqs that have simple parameters.
+        // See vm_callee_setup_arg().
+        return false;
+    }
+
+    rb_gc_register_mark_object((VALUE)iseq); // FIXME: intentional LEAK!
+
     // Create a size-exit to fall back to the interpreter
     uint8_t* side_exit = ujit_side_exit(jit, ctx);
 
@@ -1375,7 +1383,7 @@ ujit_init_codegen(void)
     ujit_reg_op(BIN(putobject_INT2FIX_1_), gen_putobject_int2fix, false);
     ujit_reg_op(BIN(putself), gen_putself, false);
     ujit_reg_op(BIN(getlocal_WC_0), gen_getlocal_wc0, false);
-    //ujit_reg_op(BIN(getlocal_WC_1), gen_getlocal_wc1, false);
+    ujit_reg_op(BIN(getlocal_WC_1), gen_getlocal_wc1, false);
     ujit_reg_op(BIN(setlocal_WC_0), gen_setlocal_wc0, false);
     ujit_reg_op(BIN(getinstancevariable), gen_getinstancevariable, false);
     ujit_reg_op(BIN(setinstancevariable), gen_setinstancevariable, false);