summaryrefslogtreecommitdiff
path: root/object.c
diff options
context:
space:
mode:
Diffstat (limited to 'object.c')
-rw-r--r--object.c18
1 files changed, 14 insertions, 4 deletions
diff --git a/object.c b/object.c
index 6861525e8e..355a7569e5 100644
--- a/object.c
+++ b/object.c
@@ -2296,11 +2296,21 @@ rb_mod_cvar_get(VALUE obj, VALUE iv)
static VALUE
rb_mod_cvar_set(VALUE obj, VALUE iv, VALUE val)
{
- ID id = rb_to_id(iv);
+ ID id;
- if (!rb_is_class_id(id)) {
- rb_name_error(id, "`%"PRIsVALUE"' is not allowed as a class variable name",
- QUOTE_ID(id));
+ if (SYMBOL_P(iv)) {
+ id = SYM2ID(iv);
+ if (!rb_is_class_id(id)) {
+ rb_name_error(id, "`%"PRIsVALUE"' is not allowed as an class variable name",
+ QUOTE_ID(id));
+ }
+ }
+ else if (!rb_is_class_name(iv)) {
+ rb_name_error_str(iv, "`%"PRIsVALUE"' is not allowed as a class variable name",
+ QUOTE(iv));
+ }
+ else {
+ id = rb_to_id(iv);
}
rb_cvar_set(obj, id, val);
return val;