diff options
-rw-r--r-- | proc.c | 6 | ||||
-rw-r--r-- | vm.c | 9 |
2 files changed, 9 insertions, 6 deletions
@@ -14,7 +14,7 @@ #include "gc.h" #include "iseq.h" -NODE *rb_vm_cref_in_context(VALUE self); +const NODE *rb_vm_cref_in_context(VALUE self, VALUE cbase); struct METHOD { VALUE recv; @@ -1622,9 +1622,9 @@ rb_mod_define_method(int argc, VALUE *argv, VALUE mod) ID id; VALUE body; int noex = NOEX_PUBLIC; - const NODE *cref = rb_vm_cref_in_context(mod); + const NODE *cref = rb_vm_cref_in_context(mod, mod); - if (cref && cref->nd_clss == mod) { + if (cref) { noex = (int)cref->nd_visi; } @@ -930,13 +930,16 @@ rb_vm_cref(void) return rb_vm_get_cref(cfp->iseq, cfp->ep); } -NODE * -rb_vm_cref_in_context(VALUE self) +const NODE * +rb_vm_cref_in_context(VALUE self, VALUE cbase) { rb_thread_t *th = GET_THREAD(); const rb_control_frame_t *cfp = rb_vm_get_ruby_level_next_cfp(th, th->cfp); + const NODE *cref; if (cfp->self != self) return NULL; - return rb_vm_get_cref(cfp->iseq, cfp->ep); + cref = rb_vm_get_cref(cfp->iseq, cfp->ep); + if (cref->nd_clss != cbase) return NULL; + return cref; } #if 0 |