diff options
-rw-r--r-- | ChangeLog | 5 | ||||
-rw-r--r-- | io.c | 32 | ||||
-rw-r--r-- | test/ruby/test_io.rb | 3 |
3 files changed, 24 insertions, 16 deletions
@@ -1,3 +1,8 @@ +Wed Feb 25 12:06:09 2009 <nobu@ruby-lang.org> + + * io.c (rb_io_getline_1): enables limit even if rs is given. + [ruby-core:22434] + Wed Feb 25 02:28:51 2009 Nobuyoshi Nakada <nobu@ruby-lang.org> * string.c (rb_str_chomp_bang): coderange may change. @@ -2246,7 +2246,7 @@ rb_io_getline_1(VALUE rs, long limit, VALUE io) GetOpenFile(io, fptr); rb_io_check_readable(fptr); - if (NIL_P(rs)) { + if (NIL_P(rs) && limit < 0) { str = read_all(fptr, 0, Qnil); if (RSTRING_LEN(str) == 0) return Qnil; } @@ -2258,24 +2258,26 @@ rb_io_getline_1(VALUE rs, long limit, VALUE io) return rb_io_getline_fast(fptr, enc); } else { - int c, newline; - const char *rsptr; - long rslen; + int c, newline = -1; + const char *rsptr = 0; + long rslen = 0; int rspara = 0; int extra_limit = 16; - rslen = RSTRING_LEN(rs); - if (rslen == 0) { - rsptr = "\n\n"; - rslen = 2; - rspara = 1; - swallow(fptr, '\n'); - rs = 0; - } - else { - rsptr = RSTRING_PTR(rs); + if (!NIL_P(rs)) { + rslen = RSTRING_LEN(rs); + if (rslen == 0) { + rsptr = "\n\n"; + rslen = 2; + rspara = 1; + swallow(fptr, '\n'); + rs = 0; + } + else { + rsptr = RSTRING_PTR(rs); + } + newline = (unsigned char)rsptr[rslen - 1]; } - newline = (unsigned char)rsptr[rslen - 1]; /* MS - Optimisation */ enc = io_read_encoding(fptr); diff --git a/test/ruby/test_io.rb b/test/ruby/test_io.rb index 0f448dbf39..678a74a1d0 100644 --- a/test/ruby/test_io.rb +++ b/test/ruby/test_io.rb @@ -102,8 +102,9 @@ class TestIO < Test::Unit::TestCase def test_gets_limit_extra_arg with_pipe {|r, w| r, w = IO.pipe - w << "0123456789" + w << "0123456789\n0123456789" w.close + assert_equal("0123456789\n0", r.gets(nil, 12)) assert_raise(TypeError) { r.gets(3,nil) } } end |