From ee51d0f9fd224695cbdb2b0b36ddb00bbdf69eaf Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Wed, 27 Oct 2004 00:18:08 +0000
Subject: * string.c (str_gsub): use a string object for exception safeness.  
 [ruby-dev:24601]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@7117 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 ChangeLog |  7 ++++++-
 string.c  | 58 ++++++++++++++++++++++++++++++----------------------------
 2 files changed, 36 insertions(+), 29 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index a25945b0d4..0655e0ca75 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,6 +1,11 @@
+Wed Oct 27 09:17:30 2004  Nobuyoshi Nakada  <nobu@ruby-lang.org>
+
+	* string.c (str_gsub): use a string object for exception safeness.
+	  [ruby-dev:24601]
+
 Tue Oct 26 23:30:39 2004  Dave Thomas  <dave@pragprog.com>
 
-	* lib/rdoc/code_objects.rb (RDoc::Context::add_class_or_module): 
+	* lib/rdoc/code_objects.rb (RDoc::Context::add_class_or_module):
 	  Restore correct :nopdoc: behavior with nested classes and modules.
 
 Mon Oct 25 18:35:39 2004  WATANABE Hirofumi  <eban@ruby-lang.org>
diff --git a/string.c b/string.c
index bcdd6c3c2c..9b82bb5174 100644
--- a/string.c
+++ b/string.c
@@ -36,6 +36,17 @@ VALUE rb_cString;
 
 VALUE rb_fs;
 
+static inline void
+str_mod_check(s, p, len)
+    VALUE s;
+    char *p;
+    long len;
+{
+    if (RSTRING(s)->ptr != p || RSTRING(s)->len != len) {
+	rb_raise(rb_eRuntimeError, "string modified");
+    }
+}
+
 static VALUE str_alloc _((VALUE));
 static VALUE
 str_alloc(klass)
@@ -2008,7 +2019,7 @@ str_gsub(argc, argv, str, bang)
     VALUE str;
     int bang;
 {
-    VALUE pat, val, repl, match;
+    VALUE pat, val, repl, match, dest;
     struct re_registers *regs;
     long beg, n;
     long offset, blen, len;
@@ -2037,7 +2048,8 @@ str_gsub(argc, argv, str, bang)
     }
 
     blen = RSTRING(str)->len + 30; /* len + margin */
-    buf = ALLOC_N(char, blen);
+    dest = rb_str_new5(str, 0, blen);
+    buf = RSTRING(dest)->ptr;
     bp = buf;
     cp = RSTRING(str)->ptr;
 
@@ -2048,9 +2060,7 @@ str_gsub(argc, argv, str, bang)
 	if (iter) {
 	    rb_match_busy(match);
 	    val = rb_obj_as_string(rb_yield(rb_reg_nth_match(0, match)));
-	    if (RSTRING(str)->ptr == buf) {
-		rb_raise(rb_eRuntimeError, "gsub reentered");
-	    }
+	    str_mod_check(dest, buf, blen);
 	    rb_backref_set(match);
 	}
 	else {
@@ -2061,7 +2071,9 @@ str_gsub(argc, argv, str, bang)
 	if (blen < len) {
 	    while (blen < len) blen *= 2;
 	    len = bp - buf;
-	    REALLOC_N(buf, char, blen);
+	    RESIZE_CAPA(dest, blen);
+	    RSTRING(dest)->len = blen;
+	    buf = RSTRING(dest)->ptr;
 	    bp = buf + len;
 	}
 	len = beg - offset;	/* copy pre-match substr */
@@ -2087,30 +2099,32 @@ str_gsub(argc, argv, str, bang)
     }
     if (RSTRING(str)->len > offset) {
 	len = bp - buf;
-	if (blen - len < RSTRING(str)->len - offset + 1) {
-	    REALLOC_N(buf, char, len + RSTRING(str)->len - offset + 1);
+	if (blen - len < RSTRING(str)->len - offset) {
+	    blen = len + RSTRING(str)->len - offset;
+	    RESIZE_CAPA(dest, blen);
+	    buf = RSTRING(dest)->ptr;
 	    bp = buf + len;
 	}
 	memcpy(bp, cp, RSTRING(str)->len - offset);
 	bp += RSTRING(str)->len - offset;
     }
     rb_backref_set(match);
+    *bp = '\0';
     if (bang) {
 	if (str_independent(str)) {
 	    free(RSTRING(str)->ptr);
 	}
 	FL_UNSET(str, ELTS_SHARED|STR_ASSOC);
+	RSTRING(str)->ptr = buf;
+	RSTRING(str)->aux.capa = blen;
+	RSTRING(dest)->ptr = 0;
+	RSTRING(dest)->len = 0;
     }
     else {
-	VALUE dup = str_alloc(rb_obj_class(str));
-
-	OBJ_INFECT(dup, str);
-	str = dup;
+	OBJ_INFECT(dest, str);
+	str = dest;
     }
-    RSTRING(str)->ptr = buf;
-    RSTRING(str)->len = len = bp - buf;
-    RSTRING(str)->ptr[len] = '\0';
-    RSTRING(str)->aux.capa = len;
+    RSTRING(str)->len = bp - buf;
 
     if (tainted) OBJ_TAINT(str);
     return str;
@@ -3592,18 +3606,6 @@ rb_f_split(argc, argv)
     return rb_str_split_m(argc, argv, uscore_get());
 }
 
-
-static inline void
-str_mod_check(s, p, len)
-    VALUE s;
-    char *p;
-    long len;
-{
-    if (RSTRING(s)->ptr != p || RSTRING(s)->len != len) {
-	rb_raise(rb_eRuntimeError, "string modified");
-    }
-}
-
 /*
  *  call-seq:
  *     str.each(separator=$/) {|substr| block }        => str
-- 
cgit v1.2.3