diff options
author | nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2013-04-04 08:18:54 +0000 |
---|---|---|
committer | nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2013-04-04 08:18:54 +0000 |
commit | 5e2c39ed226d084a340c6f860344062bd32d56b9 (patch) | |
tree | dee972f5bd1e55eb7302ad3efdc2ab763ee859c2 /object.c | |
parent | 75b4c8f490a882619e83a448c0a23fdd64080202 (diff) | |
download | ruby-5e2c39ed226d084a340c6f860344062bd32d56b9.tar.gz |
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
Diffstat (limited to 'object.c')
-rw-r--r-- | object.c | 47 |
1 files changed, 41 insertions, 6 deletions
@@ -2043,11 +2043,22 @@ rb_mod_const_get(int argc, VALUE *argv, VALUE mod) static 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, 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; } @@ -2164,11 +2175,23 @@ rb_obj_ivar_get(VALUE obj, VALUE iv) static VALUE 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", 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 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", 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; } |