1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

Fix Method#super_method for aliased methods

Previously, Method#super_method looked at the called_id to
determine the method id to use, but that isn't correct for
aliased methods, because the super target depends on the
original method id, not the called_id.

Additionally, aliases can reference methods defined in other
classes and modules, and super lookup needs to start in the
super of the defined class in such cases.

This adds tests for Method#super_method for both types of
aliases, one that uses VM_METHOD_TYPE_ALIAS and another that
does not.  Both check that the results for calling super
methods return the expected values.

To find the defined class for alias methods, add an rb_ prefix
to find_defined_class_by_owner in vm_insnhelper.c and make it
non-static, so that it can be called from method_super_method
in proc.c.

This bug was original discovered while researching [Bug #11189].

Fixes [Bug #17130]
This commit is contained in:
Jeremy Evans 2020-08-27 08:37:03 -07:00 committed by GitHub
parent f41bd0d708
commit c60aaed185
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
Notes: git 2020-08-28 00:37:38 +09:00
Merged: https://github.com/ruby/ruby/pull/3458

Merged-By: jeremyevans <code@jeremyevans.net>
3 changed files with 94 additions and 5 deletions

13
proc.c
View file

@ -3116,6 +3116,8 @@ method_to_proc(VALUE method)
return procval;
}
extern VALUE rb_find_defined_class_by_owner(VALUE current_class, VALUE target_owner);
/*
* call-seq:
* meth.super_method -> method
@ -3135,8 +3137,15 @@ method_super_method(VALUE method)
TypedData_Get_Struct(method, struct METHOD, &method_data_type, data);
iclass = data->iclass;
if (!iclass) return Qnil;
super_class = RCLASS_SUPER(RCLASS_ORIGIN(iclass));
mid = data->me->called_id;
if (data->me->def->type == VM_METHOD_TYPE_ALIAS) {
super_class = RCLASS_SUPER(rb_find_defined_class_by_owner(data->me->defined_class,
data->me->def->body.alias.original_me->owner));
mid = data->me->def->body.alias.original_me->def->original_id;
}
else {
super_class = RCLASS_SUPER(RCLASS_ORIGIN(iclass));
mid = data->me->def->original_id;
}
if (!super_class) return Qnil;
me = (rb_method_entry_t *)rb_callable_method_entry_with_refinements(super_class, mid, &iclass);
if (!me) return Qnil;

View file

@ -1080,6 +1080,86 @@ class TestMethod < Test::Unit::TestCase
'[ruby-core:85231] [Bug #14421]'
end
def test_super_method_alias
c0 = Class.new do
def m1
[:C0_m1]
end
def m2
[:C0_m2]
end
end
c1 = Class.new(c0) do
def m1
[:C1_m1] + super
end
alias m2 m1
end
c2 = Class.new(c1) do
def m2
[:C2_m2] + super
end
end
o1 = c2.new
assert_equal([:C2_m2, :C1_m1, :C0_m1], o1.m2)
m = o1.method(:m2)
assert_equal([:C2_m2, :C1_m1, :C0_m1], m.call)
m = m.super_method
assert_equal([:C1_m1, :C0_m1], m.call)
m = m.super_method
assert_equal([:C0_m1], m.call)
assert_nil(m.super_method)
end
def test_super_method_alias_to_prepended_module
m = Module.new do
def m1
[:P_m1] + super
end
def m2
[:P_m2] + super
end
end
c0 = Class.new do
def m1
[:C0_m1]
end
end
c1 = Class.new(c0) do
def m1
[:C1_m1] + super
end
prepend m
alias m2 m1
end
o1 = c1.new
assert_equal([:P_m2, :P_m1, :C1_m1, :C0_m1], o1.m2)
m = o1.method(:m2)
assert_equal([:P_m2, :P_m1, :C1_m1, :C0_m1], m.call)
m = m.super_method
assert_equal([:P_m1, :C1_m1, :C0_m1], m.call)
m = m.super_method
assert_equal([:C1_m1, :C0_m1], m.call)
m = m.super_method
assert_equal([:C0_m1], m.call)
assert_nil(m.super_method)
end
def rest_parameter(*rest)
rest
end

View file

@ -2673,8 +2673,8 @@ vm_call_bmethod(rb_execution_context_t *ec, rb_control_frame_t *cfp, struct rb_c
return vm_call_bmethod_body(ec, calling, cd, argv);
}
static VALUE
find_defined_class_by_owner(VALUE current_class, VALUE target_owner)
MJIT_FUNC_EXPORTED VALUE
rb_find_defined_class_by_owner(VALUE current_class, VALUE target_owner)
{
VALUE klass = current_class;
@ -2702,7 +2702,7 @@ aliased_callable_method_entry(const rb_callable_method_entry_t *me)
const rb_callable_method_entry_t *cme;
if (orig_me->defined_class == 0) {
VALUE defined_class = find_defined_class_by_owner(me->defined_class, orig_me->owner);
VALUE defined_class = rb_find_defined_class_by_owner(me->defined_class, orig_me->owner);
VM_ASSERT(RB_TYPE_P(orig_me->owner, T_MODULE));
cme = rb_method_entry_complement_defined_class(orig_me, me->called_id, defined_class);