From 91a079cc86b01699d11481bdd8f9636df334a404 Mon Sep 17 00:00:00 2001
From: matz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Wed, 3 Oct 2007 06:48:06 +0000
Subject: * variable.c (rb_cvar_set): check whether class variable is   defined
 in superclasses.  root classes have higher priority.   removes lower class
 variable entry from IV_TBL (if it's defined   in classes, not modules).

* variable.c (rb_cvar_get): ditto.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@13604 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 variable.c | 70 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 61 insertions(+), 9 deletions(-)

(limited to 'variable.c')

diff --git a/variable.c b/variable.c
index 5f15fbc74d..a0f68f2073 100644
--- a/variable.c
+++ b/variable.c
@@ -1701,15 +1701,53 @@ rb_define_global_const(const char *name, VALUE val)
     rb_define_const(rb_cObject, name, val);
 }
 
+static VALUE
+original_module(c)
+    VALUE c;
+{
+    if (TYPE(c) == T_ICLASS)
+	return RBASIC(c)->klass;
+    return c;
+}
+
 void
 rb_cvar_set(VALUE klass, ID id, VALUE val)
 {
-    mod_av_set(klass, id, val, Qfalse);
+    VALUE tmp;
+    VALUE front = 0, target = 0;
+
+    tmp = klass;
+    while (tmp) {
+	if (RCLASS_IV_TBL(tmp) && st_lookup(RCLASS_IV_TBL(tmp),id,0)) {
+	    if (!front) front = tmp;
+	    target = tmp;
+	}
+	tmp = RCLASS_SUPER(tmp);
+    }
+    if (target) {
+	if (front && target != front) {
+	    ID did = id;
+
+	    if (RTEST(ruby_verbose)) {
+		rb_warning("class variable %s of %s is overtaken by %s",
+			   rb_id2name(id), rb_class2name(original_module(front)),
+			   rb_class2name(original_module(target)));
+	    }
+	    if (BUILTIN_TYPE(front) == T_CLASS) {
+		st_delete(RCLASS_IV_TBL(front),&did,0);
+	    }
+	}
+    }
+    else {
+	target = klass;
+    }
+
+    mod_av_set(target, id, val, Qfalse);
 }
 
 #define CVAR_LOOKUP(v,r) do {\
     if (RCLASS_IV_TBL(klass) && st_lookup(RCLASS_IV_TBL(klass),id,(v))) {\
-	return (r);\
+	r;\
     }\
     if (FL_TEST(klass, FL_SINGLETON) ) {\
 	VALUE obj = rb_iv_get(klass, "__attached__");\
@@ -1728,7 +1766,7 @@ rb_cvar_set(VALUE klass, ID id, VALUE val)
     }\
     while (klass) {\
 	if (RCLASS_IV_TBL(klass) && st_lookup(RCLASS_IV_TBL(klass),id,(v))) {\
-	    return (r);\
+	    r;\
 	}\
 	klass = RCLASS_SUPER(klass);\
     }\
@@ -1737,20 +1775,34 @@ rb_cvar_set(VALUE klass, ID id, VALUE val)
 VALUE
 rb_cvar_get(VALUE klass, ID id)
 {
-    VALUE value, tmp;
+    VALUE value, tmp, front = 0, target = 0;
 
     tmp = klass;
-    CVAR_LOOKUP(&value, value);
-    rb_name_error(id,"uninitialized class variable %s in %s",
-		  rb_id2name(id), rb_class2name(tmp));
-    return Qnil;		/* not reached */
+    CVAR_LOOKUP(&value, {if (!front) front = klass; target = klass;});
+    if (!target) {
+	rb_name_error(id,"uninitialized class variable %s in %s",
+		      rb_id2name(id), rb_class2name(tmp));
+    }
+    if (front && target != front) {
+	ID did = id;
+
+	if (RTEST(ruby_verbose)) {
+	    rb_warning("class variable %s of %s is overtaken by %s",
+		       rb_id2name(id), rb_class2name(original_module(front)),
+		       rb_class2name(original_module(target)));
+	}
+	if (BUILTIN_TYPE(front) == T_CLASS) {
+	    st_delete(RCLASS_IV_TBL(front),&did,0);
+	}
+    }
+    return value;
 }
 
 VALUE
 rb_cvar_defined(VALUE klass, ID id)
 {
     if (!klass) return Qfalse;
-    CVAR_LOOKUP(0,Qtrue);
+    CVAR_LOOKUP(0,return Qtrue);
     return Qfalse;
 }
 
-- 
cgit v1.2.3