diff options
Diffstat (limited to 'ext/stringio')
-rw-r--r-- | ext/stringio/stringio.c | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/ext/stringio/stringio.c b/ext/stringio/stringio.c index da4eb964cb..609355a51b 100644 --- a/ext/stringio/stringio.c +++ b/ext/stringio/stringio.c @@ -1067,6 +1067,11 @@ strio_each(int argc, VALUE *argv, VALUE self) struct StringIO *ptr = StringIO(self); VALUE line; + if(argc > 0 && TYPE(argv[argc-1]) == T_FIXNUM && + FIX2INT(argv[argc-1]) == 0) { + rb_raise(rb_eArgError, "invalid limit: 0 for readlines"); + } + RETURN_ENUMERATOR(self, argc, argv); while (!NIL_P(line = strio_getline(argc, argv, readable(ptr)))) { @@ -1088,6 +1093,12 @@ strio_readlines(int argc, VALUE *argv, VALUE self) { struct StringIO *ptr = StringIO(self); VALUE ary = rb_ary_new(), line; + + if(argc > 0 && TYPE(argv[argc-1]) == T_FIXNUM && + FIX2INT(argv[argc-1]) == 0) { + rb_raise(rb_eArgError, "invalid limit: 0 for readlines"); + } + while (!NIL_P(line = strio_getline(argc, argv, readable(ptr)))) { rb_ary_push(ary, line); } |