aboutsummaryrefslogtreecommitdiffstats
path: root/range.c
diff options
context:
space:
mode:
Diffstat (limited to 'range.c')
-rw-r--r--range.c26
1 files changed, 15 insertions, 11 deletions
diff --git a/range.c b/range.c
index 6431fc9cee..37151ad28a 100644
--- a/range.c
+++ b/range.c
@@ -436,19 +436,23 @@ range_each(VALUE range)
rb_yield(LONG2FIX(i));
}
}
- else if (TYPE(beg) == T_STRING) {
- VALUE args[2];
-
- args[0] = end;
- args[1] = EXCL(range) ? Qtrue : Qfalse;
- rb_block_call(beg, rb_intern("upto"), 2, args, rb_yield, 0);
- }
else {
- if (!rb_respond_to(beg, id_succ)) {
- rb_raise(rb_eTypeError, "can't iterate from %s",
- rb_obj_classname(beg));
+ VALUE tmp = rb_check_string_type(beg);
+
+ if (!NIL_P(tmp)) {
+ VALUE args[2];
+
+ args[0] = end;
+ args[1] = EXCL(range) ? Qtrue : Qfalse;
+ rb_block_call(tmp, rb_intern("upto"), 2, args, rb_yield, 0);
+ }
+ else {
+ if (!rb_respond_to(beg, id_succ)) {
+ rb_raise(rb_eTypeError, "can't iterate from %s",
+ rb_obj_classname(beg));
+ }
+ range_each_func(range, each_i, NULL);
}
- range_each_func(range, each_i, NULL);
}
return range;
}