From 30540c567569d3486ccbf59b59d903d5778f04d5 Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sat, 10 Dec 2016 08:54:40 +0000
Subject: stringio.c: chomp CR

* ext/stringio/stringio.c (strio_getline): chomp CR not only LF,
  as well as String#chomp.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@57043 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 ext/stringio/stringio.c | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'ext/stringio/stringio.c')

diff --git a/ext/stringio/stringio.c b/ext/stringio/stringio.c
index ff2a544af6..fb2eddaa7e 100644
--- a/ext/stringio/stringio.c
+++ b/ext/stringio/stringio.c
@@ -1041,6 +1041,7 @@ static inline int
 chomp_newline_width(const char *s, const char *e)
 {
     if (e > s && *--e == '\n') {
+	if (e > s && *--e == '\r') return 2;
 	return 1;
     }
     return 0;
@@ -1071,7 +1072,8 @@ strio_getline(struct getline_arg *arg, struct StringIO *ptr)
     }
     else if ((n = RSTRING_LEN(str)) == 0) {
 	p = s;
-	while (*p == '\n') {
+	while (p[(p + 1 < e) && (*p == '\r') && 0] == '\n') {
+	    p += *p == '\r';
 	    if (++p == e) {
 		return Qnil;
 	    }
@@ -1083,6 +1085,11 @@ strio_getline(struct getline_arg *arg, struct StringIO *ptr)
 		w = (arg->chomp ? 1 : 0);
 		break;
 	    }
+	    else if (*p == '\r' && p < e && p[1] == '\n') {
+		e = p + 2;
+		w = (arg->chomp ? 2 : 0);
+		break;
+	    }
 	}
 	if (!w && arg->chomp) {
 	    w = chomp_newline_width(s, e);
@@ -1092,7 +1099,7 @@ strio_getline(struct getline_arg *arg, struct StringIO *ptr)
     else if (n == 1) {
 	if ((p = memchr(s, RSTRING_PTR(str)[0], e - s)) != 0) {
 	    e = p + 1;
-	    w = (arg->chomp ? 1 : 0);
+	    w = (arg->chomp ? (p > s && *(p-1) == '\r') + 1 : 0);
 	}
 	str = strio_substr(ptr, ptr->pos, e - s - w);
     }
-- 
cgit v1.2.3