object.c: avoid inadvertent symbol creation

* object.c (rb_mod_const_set): fix symbol name check.
  (rb_obj_ivar_set): ditto.
  (rb_mod_cvar_set): ditto.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@40104 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
nobu 2013-04-04 08:18:54 +00:00
Родитель 75b4c8f490
Коммит 5e2c39ed22
1 изменённых файлов: 41 добавлений и 6 удалений

Просмотреть файл

@ -2043,11 +2043,22 @@ rb_mod_const_get(int argc, VALUE *argv, VALUE mod)
static VALUE static VALUE
rb_mod_const_set(VALUE mod, VALUE name, VALUE value) rb_mod_const_set(VALUE mod, VALUE name, VALUE value)
{ {
if (!SYMBOL_P(name) && !rb_is_const_name(name)) { ID id;
if (SYMBOL_P(name)) {
id = SYM2ID(name);
if (!rb_is_const_id(id)) {
rb_name_error(id, "wrong constant name %"PRIsVALUE,
QUOTE_ID(id));
}
}
else if (!rb_is_const_name(name)) {
rb_name_error_str(name, "wrong constant name %"PRIsVALUE, rb_name_error_str(name, "wrong constant name %"PRIsVALUE,
QUOTE(name)); QUOTE(name));
} }
rb_const_set(mod, rb_to_id(name), value); else {
id = rb_to_id(name);
}
rb_const_set(mod, id, value);
return value; return value;
} }
@ -2164,11 +2175,23 @@ rb_obj_ivar_get(VALUE obj, VALUE iv)
static VALUE static VALUE
rb_obj_ivar_set(VALUE obj, VALUE iv, VALUE val) rb_obj_ivar_set(VALUE obj, VALUE iv, VALUE val)
{ {
if (!SYMBOL_P(iv) && !rb_is_instance_name(iv)) { ID id;
if (SYMBOL_P(iv)) {
id = SYM2ID(iv);
if (!rb_is_instance_id(id)) {
rb_name_error(id, "`%"PRIsVALUE"' is not allowed as an instance variable name",
QUOTE_ID(id));
}
}
else if (!rb_is_instance_name(iv)) {
rb_name_error_str(iv, "`%"PRIsVALUE"' is not allowed as an instance variable name", rb_name_error_str(iv, "`%"PRIsVALUE"' is not allowed as an instance variable name",
QUOTE(iv)); QUOTE(iv));
} }
return rb_ivar_set(obj, rb_to_id(iv), val); else {
id = rb_to_id(iv);
}
return rb_ivar_set(obj, id, val);
} }
/* /*
@ -2273,11 +2296,23 @@ rb_mod_cvar_get(VALUE obj, VALUE iv)
static VALUE static VALUE
rb_mod_cvar_set(VALUE obj, VALUE iv, VALUE val) rb_mod_cvar_set(VALUE obj, VALUE iv, VALUE val)
{ {
if (!SYMBOL_P(iv) && !rb_is_class_id(iv)) { 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_id(iv)) {
rb_name_error_str(iv, "`%"PRIsVALUE"' is not allowed as a class variable name", rb_name_error_str(iv, "`%"PRIsVALUE"' is not allowed as a class variable name",
QUOTE(iv)); QUOTE(iv));
} }
rb_cvar_set(obj, rb_to_id(iv), val); else {
id = rb_to_id(iv);
}
rb_cvar_set(obj, id, val);
return val; return val;
} }