From 54d534f04313a0437a613516169cb243b695ccaf Mon Sep 17 00:00:00 2001
From: matz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Tue, 30 Oct 2001 08:43:28 +0000
Subject: * string.c (rb_str_chomp_bang): do smart chomp if $/ == '\n'.

* io.c (rb_io_puts): don't treat Array specially.

* bignum.c (rb_big_cmp): should convert bignum to float.

* eval.c (rb_f_eval): can't modify untainted binding.

* regex.c (re_compile_pattern): should preverve p0 value.


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@1803 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 string.c | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'string.c')

diff --git a/string.c b/string.c
index a4612221c5..c57ebb5a5b 100644
--- a/string.c
+++ b/string.c
@@ -2661,15 +2661,37 @@ rb_str_chomp_bang(argc, argv, str)
     long len = RSTRING(str)->len;
 
     if (rb_scan_args(argc, argv, "01", &rs) == 0) {
+	if (len == 0) return Qnil;
 	rs = rb_rs;
+	if (rs == rb_default_rs) {
+	  smart_chomp:
+	    rb_str_modify(str);
+	    if (RSTRING(str)->ptr[len-1] == '\n') {
+		RSTRING(str)->len--;
+		if (RSTRING(str)->len > 0 &&
+		    RSTRING(str)->ptr[RSTRING(str)->len-1] == '\r') {
+		    RSTRING(str)->len--;
+		}
+		return str;
+	    }
+	    else if (RSTRING(str)->ptr[len-1] == '\r') {
+		RSTRING(str)->len--;
+		return str;
+	    }
+	    return Qnil;
+	}
     }
     if (NIL_P(rs)) return Qnil;
+    if (len == 0) return Qnil;
 
     StringValue(rs);
+    rb_str_modify(str);
     rslen = RSTRING(rs)->len;
     if (rslen == 0) {
 	while (len>0 && p[len-1] == '\n') {
 	    len--;
+	    if (len>0 && p[len-1] == '\r')
+		len--;
 	}
 	if (len < RSTRING(str)->len) {
 	    rb_str_modify(str);
@@ -2681,6 +2703,8 @@ rb_str_chomp_bang(argc, argv, str)
     }
     if (rslen > len) return Qnil;
     newline = RSTRING(rs)->ptr[rslen-1];
+    if (rslen == 1 && newline == '\n')
+	goto smart_chomp;
 
     if (p[len-1] == newline &&
 	(rslen <= 1 ||
-- 
cgit v1.2.3