[ruby-dev:11138] copy-on-write for substr

前田です。

rb_str_substr()をcopy-on-writeにするパッチです。
dupされたものと区別するためにフラグを増やしたんですが、問題がある
でしょうか?

-- 
前田 修吾

Index: string.c
===================================================================
RCS file: /home/cvs/ruby/string.c,v
retrieving revision 1.46
diff -u -r1.46 string.c
--- string.c	2000/10/02 07:48:33	1.46
+++ string.c	2000/10/06 08:14:23
@@ -28,6 +28,7 @@
 VALUE rb_cString;
 
 #define STR_NO_ORIG FL_USER2
+#define STR_SUBSTR FL_USER3
 
 VALUE rb_fs;
 
@@ -99,7 +100,7 @@
 {
     if (OBJ_FROZEN(orig)) return orig;
     if (RSTRING(orig)->orig) {
-	if (FL_TEST(orig, STR_NO_ORIG)) {
+	if (FL_TEST(orig, STR_NO_ORIG|STR_SUBSTR)) {
 	    orig = rb_str_new(RSTRING(orig)->ptr, RSTRING(orig)->len);
 	    OBJ_FREEZE(orig);
 	    return orig;
@@ -195,7 +196,12 @@
 	klass = (VALUE)RCLASS(klass)->super;
     }
 
-    if (OBJ_FROZEN(str)) str2 = rb_str_new3(str);
+    if (FL_TEST(str, STR_SUBSTR)) {
+	str2 = rb_str_new(RSTRING(str)->ptr, RSTRING(str)->len);
+    }
+    else if (OBJ_FROZEN(str)) {
+	str2 = rb_str_new3(str);
+    }
     else if (FL_TEST(str, STR_NO_ORIG)) {
 	str2 = rb_str_new(RSTRING(str)->ptr, RSTRING(str)->len);
     }
@@ -348,9 +354,32 @@
 	len = 0;
     }
     if (len == 0) return rb_str_new(0,0);
-
-    str2 = rb_str_new(RSTRING(str)->ptr+beg, len);
+    
+    if (OBJ_FROZEN(str)) {
+	str2 = rb_str_new3(str);
+	RSTRING(str2)->ptr = RSTRING(str)->ptr + beg;
+	RSTRING(str2)->len = len;
+    }
+    else if (FL_TEST(str, STR_NO_ORIG)) {
+	str2 = rb_str_new(RSTRING(str)->ptr+beg, len);
+    }
+    else if (RSTRING(str)->orig) {
+	str2 = rb_str_new3(RSTRING(str)->orig);
+	RSTRING(str2)->ptr = RSTRING(str)->ptr + beg;
+	RSTRING(str2)->len = len;
+    }
+    else {
+	VALUE orig;
+	NEWOBJ(str2, struct RString);
+	OBJSETUP(str2, rb_cString, T_STRING);
+
+	orig = rb_str_new4(str);
+	RSTRING(str2)->ptr = RSTRING(orig)->ptr + beg;
+	RSTRING(str2)->len = len;
+	RSTRING(str2)->orig = orig;
+    }
     if (OBJ_TAINTED(str)) OBJ_TAINT(str2);
+    FL_SET(str2, STR_SUBSTR);
 
     return str2;
 }
@@ -394,7 +423,7 @@
 rb_str_dup_frozen(str)
     VALUE str;
 {
-    if (RSTRING(str)->orig && !FL_TEST(str, STR_NO_ORIG)) {
+    if (RSTRING(str)->orig && !FL_TEST(str, STR_NO_ORIG|STR_SUBSTR)) {
 	OBJ_FREEZE(RSTRING(str)->orig);
 	return RSTRING(str)->orig;
     }