From 64f2b9f99055aaaab9abbf53da1f72b2825df503 Mon Sep 17 00:00:00 2001 From: nobu Date: Sat, 26 Oct 2013 10:08:02 +0000 Subject: [PATCH] range.c: modify check * range.c (range_initialize_copy): disallow to modify after initialized. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@43433 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ChangeLog | 5 +++++ range.c | 24 +++++++++++++++++++----- test/ruby/test_range.rb | 1 + 3 files changed, 25 insertions(+), 5 deletions(-) diff --git a/ChangeLog b/ChangeLog index 0bfae76310..745f07b983 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,8 @@ +Sat Oct 26 19:08:00 2013 Nobuyoshi Nakada + + * range.c (range_initialize_copy): disallow to modify after + initialized. + Sat Oct 26 17:48:54 2013 Tanaka Akira * lib/open-uri.rb (meta_add_field): : Re-implemented. diff --git a/range.c b/range.c index 99fdf9cede..ebd0811a98 100644 --- a/range.c +++ b/range.c @@ -82,6 +82,15 @@ rb_range_new(VALUE beg, VALUE end, int exclude_end) return range; } +static void +range_modify(VALUE range) +{ + /* Ranges are immutable, so that they should be initialized only once. */ + if (RANGE_EXCL(range) != Qnil) { + rb_name_error(idInitialize, "`initialize' called twice"); + } +} + /* * call-seq: * Range.new(begin, end, exclude_end=false) -> rng @@ -97,15 +106,19 @@ range_initialize(int argc, VALUE *argv, VALUE range) VALUE beg, end, flags; rb_scan_args(argc, argv, "21", &beg, &end, &flags); - /* Ranges are immutable, so that they should be initialized only once. */ - if (RANGE_EXCL(range) != Qnil) { - rb_name_error(idInitialize, "`initialize' called twice"); - } + range_modify(range); range_init(range, beg, end, RBOOL(RTEST(flags))); return Qnil; } -#define range_initialize_copy rb_struct_init_copy /* :nodoc: */ +/* :nodoc: */ +static VALUE +range_initialize_copy(VALUE range, VALUE orig) +{ + range_modify(range); + rb_struct_init_copy(range, orig); + return range; +} /* * call-seq: @@ -1244,6 +1257,7 @@ range_loader(VALUE range, VALUE obj) rb_raise(rb_eTypeError, "not a dumped range object"); } + range_modify(range); RANGE_SET_BEG(range, rb_ivar_get(obj, id_beg)); RANGE_SET_END(range, rb_ivar_get(obj, id_end)); RANGE_SET_EXCL(range, rb_ivar_get(obj, id_excl)); diff --git a/test/ruby/test_range.rb b/test/ruby/test_range.rb index 186ea678c6..914baf9ecf 100644 --- a/test/ruby/test_range.rb +++ b/test/ruby/test_range.rb @@ -82,6 +82,7 @@ class TestRange < Test::Unit::TestCase def test_initialize_twice r = eval("1..2") assert_raise(NameError) { r.instance_eval { initialize 3, 4 } } + assert_raise(NameError) { r.instance_eval { initialize_copy 3..4 } } end def test_uninitialized_range