Recheck elements type after `to_str` conversion

https://hackerone.com/reports/244786
This commit is contained in:
Nobuyoshi Nakada 2020-01-20 00:29:40 +09:00
Родитель 92a30acbfb
Коммит 2b2821acd3
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4BC7D6DF58D8DF60
2 изменённых файлов: 14 добавлений и 2 удалений

Просмотреть файл

@ -2286,7 +2286,7 @@ recursive_join(VALUE obj, VALUE argp, int recur)
return Qnil; return Qnil;
} }
static void static long
ary_join_0(VALUE ary, VALUE sep, long max, VALUE result) ary_join_0(VALUE ary, VALUE sep, long max, VALUE result)
{ {
long i; long i;
@ -2295,10 +2295,12 @@ ary_join_0(VALUE ary, VALUE sep, long max, VALUE result)
if (max > 0) rb_enc_copy(result, RARRAY_AREF(ary, 0)); if (max > 0) rb_enc_copy(result, RARRAY_AREF(ary, 0));
for (i=0; i<max; i++) { for (i=0; i<max; i++) {
val = RARRAY_AREF(ary, i); val = RARRAY_AREF(ary, i);
if (!RB_TYPE_P(val, T_STRING)) break;
if (i > 0 && !NIL_P(sep)) if (i > 0 && !NIL_P(sep))
rb_str_buf_append(result, sep); rb_str_buf_append(result, sep);
rb_str_buf_append(result, val); rb_str_buf_append(result, val);
} }
return i;
} }
static void static void
@ -2374,7 +2376,7 @@ rb_ary_join(VALUE ary, VALUE sep)
int first; int first;
result = rb_str_buf_new(len + (RARRAY_LEN(ary)-i)*10); result = rb_str_buf_new(len + (RARRAY_LEN(ary)-i)*10);
rb_enc_associate(result, rb_usascii_encoding()); rb_enc_associate(result, rb_usascii_encoding());
ary_join_0(ary, sep, i, result); i = ary_join_0(ary, sep, i, result);
first = i == 0; first = i == 0;
ary_join_1(ary, ary, sep, i, result, &first); ary_join_1(ary, ary, sep, i, result, &first);
return result; return result;

Просмотреть файл

@ -2447,6 +2447,16 @@ class TestArray < Test::Unit::TestCase
assert_equal("12345", [1,[2,[3,4],5]].join) assert_equal("12345", [1,[2,[3,4],5]].join)
end end
def test_join_recheck_elements_type
x = Struct.new(:ary).new
def x.to_str
ary[2] = [0, 1, 2]
"z"
end
(x.ary = ["a", "b", "c", x])
assert_equal("ab012z", x.ary.join(""))
end
def test_to_a2 def test_to_a2
klass = Class.new(Array) klass = Class.new(Array)
a = klass.new.to_a a = klass.new.to_a