From 5069c5f5214ce68df8b3954321ad9114c5368dc3 Mon Sep 17 00:00:00 2001 From: Jeremy Evans Date: Sat, 12 Oct 2019 00:01:37 -0700 Subject: Honor refinements for modules that prepend other modules This previously did not work, and the reason it did not work is that: 1) Refining a module or class that prepends other modules places the refinements in the class itself and not the origin iclass. 2) Inclusion of a module that prepends other modules skips the module itself, including only iclasses for the prepended modules and the origin iclass. Those two behaviors combined meant that the method table for the refined methods for the included module never ends up in the method lookup chain for the class including the module. Fix this by not skipping the module itself when the module is included. This requires some code rearranging in rb_include_class_new to make sure the correct method tables and origin settings are used for the created iclass. As origin iclasses shouldn't be exposed to Ruby, this also requires skipping modules that have origin iclasses in Module#ancestors (classes that have origin iclasses were already skipped). Fixes [Bug #16242] --- class.c | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) (limited to 'class.c') diff --git a/class.c b/class.c index c253b4b642..e03ba22576 100644 --- a/class.c +++ b/class.c @@ -825,10 +825,20 @@ VALUE rb_include_class_new(VALUE module, VALUE super) { VALUE klass = class_alloc(T_ICLASS, rb_cClass); + RCLASS_SET_ORIGIN(klass, klass); + + RCLASS_M_TBL(OBJ_WB_UNPROTECT(klass)) = + RCLASS_M_TBL(OBJ_WB_UNPROTECT(module)); /* TODO: unprotected? */ if (BUILTIN_TYPE(module) == T_ICLASS) { + if (module != RCLASS_ORIGIN(module)) { + RCLASS_SET_ORIGIN(klass, RCLASS_ORIGIN(module)); + } module = RBASIC(module)->klass; } + else if (module != RCLASS_ORIGIN(module)) { + RCLASS_SET_ORIGIN(klass, RCLASS_ORIGIN(module)); + } if (!RCLASS_IV_TBL(module)) { RCLASS_IV_TBL(module) = st_init_numtable(); } @@ -838,9 +848,6 @@ rb_include_class_new(VALUE module, VALUE super) RCLASS_IV_TBL(klass) = RCLASS_IV_TBL(module); RCLASS_CONST_TBL(klass) = RCLASS_CONST_TBL(module); - RCLASS_M_TBL(OBJ_WB_UNPROTECT(klass)) = - RCLASS_M_TBL(OBJ_WB_UNPROTECT(RCLASS_ORIGIN(module))); /* TODO: unprotected? */ - RCLASS_SET_SUPER(klass, super); if (RB_TYPE_P(module, T_ICLASS)) { RBASIC_SET_CLASS(klass, RBASIC(module)->klass); @@ -894,8 +901,6 @@ include_modules_at(const VALUE klass, VALUE c, VALUE module, int search_super) int superclass_seen = FALSE; struct rb_id_table *tbl; - if (RCLASS_ORIGIN(module) != module) - goto skip; if (klass_m_tbl && klass_m_tbl == RCLASS_M_TBL(module)) return -1; /* ignore if the module included already in superclasses */ @@ -1091,10 +1096,11 @@ rb_mod_ancestors(VALUE mod) VALUE p, ary = rb_ary_new(); for (p = mod; p; p = RCLASS_SUPER(p)) { + if (p != RCLASS_ORIGIN(p)) continue; if (BUILTIN_TYPE(p) == T_ICLASS) { rb_ary_push(ary, RBASIC(p)->klass); } - else if (p == RCLASS_ORIGIN(p)) { + else { rb_ary_push(ary, p); } } -- cgit v1.2.3