wip

Author: XrXr

yjit/src/codegen.rs

@@ -2448,6 +2448,8 @@ fn gen_get_ep(asm: &mut Assembler, level: u32) -> Opnd {
 /// EP returned may not be `VM_ENV_FLAG_LOCAL`. Practically, ISeqs that don't
 /// meet this condition, such as `ISEQ_TYPE_TOP`, also don't need to use this operation.
 fn gen_get_lep(jit: &JITState, asm: &mut Assembler) -> Opnd {
+    assert_eq!(ISEQ_TYPE_METHOD, unsafe { rb_get_iseq_body_type(rb_get_iseq_body_local_iseq(jit.iseq)) });
+
     // GET_LEP() chases the parent environment pointer to reach the local environment. Inside
     // a descendant of ISEQ_TYPE_METHOD, it chases the same number of times as chasing
     // the parent iseq pointer to reach the local iseq. See get_lvar_level() in compile.c
@@ -9904,6 +9906,12 @@ fn gen_invokesuper_specialized(
         return None;
     }
 
+    let local_iseq = unsafe { rb_get_iseq_body_local_iseq(jit.iseq) };
+    if matches!(block, BlockHandler::LEPSpecVal) && unsafe { rb_get_iseq_body_type(local_iseq) } != ISEQ_TYPE_METHOD {
+        gen_counter_incr(jit, asm, Counter::invokesuper_find_block_handler_from_block);
+        return None;
+    }
+
     // FIXME: We should track and invalidate this block when this cme is invalidated
     let current_defined_class = unsafe { (*me).defined_class };
     let mid = unsafe { get_def_original_id((*me).def) };

yjit/src/stats.rs

@@ -398,6 +398,7 @@ make_counters! {
 
     invokesuper_defined_class_mismatch,
     invokesuper_forwarding,
+    invokesuper_find_block_handler_from_block,
     invokesuper_kw_splat,
     invokesuper_kwarg,
     invokesuper_megamorphic,