aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorPikachuEXE <pikachuexe@gmail.com>2021-08-05 15:51:15 +0800
committerusa <usa@garbagecollect.jp>2021-11-24 16:47:15 +0900
commiteeabac4c55e0968e3664728bcce3c0eb23cd9321 (patch)
treec0ba580e13a87fef23c69f51185864f095549446
parente0b323632f5ea07e2646a2ec0b72f56093348265 (diff)
downloadruby-eeabac4c55e0968e3664728bcce3c0eb23cd9321.tar.gz
Do not allow Module#include to insert modules before the origin in the lookup chain
Module#include should only be able to insert modules after the origin, otherwise it ends up working like Module#prepend. This fixes the case where one of the modules in the included module chain is included in a module that is already prepended to the receiver. Fixes [Bug #7844] Backport of https://github.com/ruby/ruby/pull/3796 to 2.7
-rw-r--r--class.c7
-rw-r--r--test/ruby/test_module.rb10
2 files changed, 16 insertions, 1 deletions
diff --git a/class.c b/class.c
index 2e4b31d7c6..f17e7f053a 100644
--- a/class.c
+++ b/class.c
@@ -911,17 +911,22 @@ include_modules_at(const VALUE klass, VALUE c, VALUE module, int search_super)
}
while (module) {
+ int origin_seen = FALSE;
int superclass_seen = FALSE;
struct rb_id_table *tbl;
+ if (klass == c)
+ origin_seen = TRUE;
if (klass_m_tbl && klass_m_tbl == RCLASS_M_TBL(module))
return -1;
/* ignore if the module included already in superclasses */
for (p = RCLASS_SUPER(klass); p; p = RCLASS_SUPER(p)) {
int type = BUILTIN_TYPE(p);
+ if (c == p)
+ origin_seen = TRUE;
if (type == T_ICLASS) {
if (RCLASS_M_TBL(p) == RCLASS_M_TBL(module)) {
- if (!superclass_seen) {
+ if (!superclass_seen && origin_seen) {
c = p; /* move insertion point */
}
goto skip;
diff --git a/test/ruby/test_module.rb b/test/ruby/test_module.rb
index 5cfda018f3..231ccd2072 100644
--- a/test/ruby/test_module.rb
+++ b/test/ruby/test_module.rb
@@ -472,6 +472,16 @@ class TestModule < Test::Unit::TestCase
assert_equal([Comparable, Kernel], String.included_modules - mixins)
end
+ def test_include_with_prepend
+ c = Class.new{def m; [:c] end}
+ p = Module.new{def m; [:p] + super end}
+ q = Module.new{def m; [:q] + super end; include p}
+ r = Module.new{def m; [:r] + super end; prepend q}
+ s = Module.new{def m; [:s] + super end; include r}
+ a = Class.new(c){def m; [:a] + super end; prepend p; include s}
+ assert_equal([:p, :a, :s, :q, :r, :c], a.new.m)
+ end
+
def test_instance_methods
assert_equal([:user, :user2], User.instance_methods(false).sort)
assert_equal([:user, :user2, :mixin].sort, User.instance_methods(true).sort)