array.c: always check frozenness in Array#unshift. Fixes [Bug #15952]

Closes: https://github.com/ruby/ruby/pull/2251
This commit is contained in:
Luke Gruber authored and Nobuyoshi Nakada committed 2019-06-23 08:05:13 +09:00
1 parent 3840791b7e
commit ec8e5f5aa6
2 files changed
+13 -1

No files matched your search

+2 -1
View File
@@ -1373,6 +1373,8 @@ ary_ensure_room_for_unshift(VALUE ary, int argc)
rb_raise(rb_eIndexError, "index %ld too big", new_len);
}
rb_ary_modify(ary);
if (ARY_SHARED_P(ary)) {
VALUE shared = ARY_SHARED(ary);
capa = RARRAY_LEN(shared);
@@ -1383,7 +1385,6 @@ ary_ensure_room_for_unshift(VALUE ary, int argc)
}
}
rb_ary_modify(ary);
capa = ARY_CAPA(ary);
if (capa - (capa >> 6) <= new_len) {
ary_double_capa(ary, new_len);
+11
View File
@@ -1958,6 +1958,17 @@ class TestArray < Test::Unit::TestCase
assert_equal(@cls[@cls[1,2], nil, 'dog', 'cat'], a.unshift(@cls[1, 2]))
end
def test_unshift_frozen
bug15952 = '[Bug #15952]'
assert_raise(FrozenError, bug15952) do
a = [1] * 100
b = a[4..-1]
a.replace([1])
b.freeze
b.unshift("a")
end
end
def test_OR # '|'
assert_equal(@cls[], @cls[] | @cls[])
assert_equal(@cls[1], @cls[1] | @cls[])