archive/ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2025-08-15 13:39:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

[Bug #20936] Fix #size for Range#reverse_each

Browse source
This commit is contained in:
Kouhei Yanagita 2024-12-10 11:51:49 +09:00 • committed by Nobuyoshi Nakada
parent d5abcae435
commit 3422bfcab6
Notes: git 2024-12-10 14:12:45 +00:00
2 changed files with 67 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

51
range.c
View file

@ -908,6 +908,10 @@ sym_each_i(VALUE v, VALUE arg)
return each_i(rb_str_intern(v), arg);
}
#define CANT_ITERATE_FROM(x) \
rb_raise(rb_eTypeError, "can't iterate from %s", \
rb_obj_classname(x))
/*
* call-seq:
* size -> non_negative_integer or Infinity or nil
@ -944,13 +948,48 @@ range_size(VALUE range)
}
if (!discrete_object_p(b)) {
rb_raise(rb_eTypeError, "can't iterate from %s",
rb_obj_classname(b));
CANT_ITERATE_FROM(b);
}
return Qnil;
}
static VALUE
range_reverse_size(VALUE range)
{
VALUE b = RANGE_BEG(range), e = RANGE_END(range);
if (NIL_P(e)) {
CANT_ITERATE_FROM(e);
}
if (RB_INTEGER_TYPE_P(b)) {
if (rb_obj_is_kind_of(e, rb_cNumeric)) {
return ruby_num_interval_step_size(b, e, INT2FIX(1), EXCL(range));
}
else {
CANT_ITERATE_FROM(e);
}
}
if (NIL_P(b)) {
if (RB_INTEGER_TYPE_P(e)) {
return DBL2NUM(HUGE_VAL);
}
else {
CANT_ITERATE_FROM(e);
}
}
if (!discrete_object_p(b)) {
CANT_ITERATE_FROM(e);
}
return Qnil;
}
#undef CANT_ITERATE_FROM
/*
* call-seq:
* to_a -> array
@ -979,6 +1018,12 @@ range_enum_size(VALUE range, VALUE args, VALUE eobj)
return range_size(range);
}
static VALUE
range_enum_reverse_size(VALUE range, VALUE args, VALUE eobj)
{
return range_reverse_size(range);
}
RBIMPL_ATTR_NORETURN()
static void
range_each_bignum_endless(VALUE beg)
@ -1225,7 +1270,7 @@ range_reverse_each_negative_bignum_section(VALUE beg, VALUE end)
static VALUE
range_reverse_each(VALUE range)
{
RETURN_SIZED_ENUMERATOR(range, 0, 0, range_enum_size);
RETURN_SIZED_ENUMERATOR(range, 0, 0, range_enum_reverse_size);
VALUE beg = RANGE_BEG(range);
VALUE end = RANGE_END(range);