diff options
| -rw-r--r-- | doc/mrbgems/README.md | 4 | ||||
| -rw-r--r-- | mrbgems/mruby-array-ext/mrblib/array.rb | 54 | ||||
| -rw-r--r-- | mrbgems/mruby-array-ext/test/array.rb | 34 | ||||
| -rw-r--r-- | mrblib/range.rb | 14 | ||||
| -rw-r--r-- | src/codegen.c | 2 | ||||
| -rw-r--r-- | src/debug.c | 2 | ||||
| -rw-r--r-- | src/dump.c | 18 | ||||
| -rw-r--r-- | src/load.c | 12 | ||||
| -rw-r--r-- | src/string.c | 6 | ||||
| -rw-r--r-- | src/symbol.c | 3 |
10 files changed, 108 insertions, 41 deletions
diff --git a/doc/mrbgems/README.md b/doc/mrbgems/README.md index 866198f79..863f7bd30 100644 --- a/doc/mrbgems/README.md +++ b/doc/mrbgems/README.md @@ -159,10 +159,6 @@ Its format is same as argument of method `MRuby::Build#gem`, expect that it can' When a special version of depedency is required, use `MRuby::Build#gem` in *build_config.rb* to override default gem. -__ATTENTION:__ -The dependency system is currently (May 2013) under development and doesn't check -or resolve dependencies! - In case your GEM has more complex build requirements you can use the following options additionally inside of your GEM specification: diff --git a/mrbgems/mruby-array-ext/mrblib/array.rb b/mrbgems/mruby-array-ext/mrblib/array.rb index 49d0db0d5..bf9c09223 100644 --- a/mrbgems/mruby-array-ext/mrblib/array.rb +++ b/mrbgems/mruby-array-ext/mrblib/array.rb @@ -313,7 +313,7 @@ class Array def fill(arg0=nil, arg1=nil, arg2=nil, &block) if arg0 == nil && arg1 == nil && arg2 == nil && !block - raise ArgumentError, "wrong number of arguments (0 for 1..3)" + raise ArgumentError, "wrong number of arguments (0 for 1..3)" end beg = len = 0 @@ -323,11 +323,13 @@ class Array # ary.fill { |index| block } -> ary beg = 0 len = self.size - elsif arg0 != nil && arg0.respond_to?(:begin) && arg0.respond_to?(:end) + elsif arg0 != nil && arg0.kind_of?(Range) # ary.fill(range) { |index| block } -> ary beg = arg0.begin beg += self.size if beg < 0 - len = arg0.end - beg + 1 + len = arg0.end + len += self.size if len < 0 + len += 1 unless arg0.exclude_end? elsif arg0 != nil # ary.fill(start [, length] ) { |index| block } -> ary beg = arg0 @@ -342,20 +344,22 @@ class Array if arg0 != nil && arg1 == nil && arg2 == nil # ary.fill(obj) -> ary beg = 0 - len = self.size - elsif arg0 != nil && arg1 != nil && arg1.respond_to?(:begin) && arg1.respond_to?(:end) - # ary.fill(obj, range ) -> ary len = self.size + elsif arg0 != nil && arg1 != nil && arg1.kind_of?(Range) + # ary.fill(obj, range ) -> ary beg = arg1.begin - len = arg1.end - beg + 1 + beg += self.size if beg < 0 + len = arg1.end + len += self.size if len < 0 + len += 1 unless arg1.exclude_end? elsif arg0 != nil && arg1 != nil # ary.fill(obj, start [, length]) -> ary beg = arg1 beg += self.size if beg < 0 - if arg2 == nil + if arg2 == nil len = self.size else - len = arg1 + arg2 + len = beg + arg2 end end end @@ -509,4 +513,36 @@ class Array self[idx, 0] = args self end + + ## + # call-seq: + # ary.delete_if { |item| block } -> ary + # ary.delete_if -> Enumerator + # + # Deletes every element of +self+ for which block evaluates to +true+. + # + # The array is changed instantly every time the block is called, not after + # the iteration is over. + # + # See also Array#reject! + # + # If no block is given, an Enumerator is returned instead. + # + # scores = [ 97, 42, 75 ] + # scores.delete_if {|score| score < 80 } #=> [97] + + def delete_if(&block) + return to_enum :delete_if unless block_given? + + idx = 0 + len = self.size + while idx < self.size do + if block.call(self[idx]) + self.delete_at(idx) + else + idx += 1 + end + end + self + end end diff --git a/mrbgems/mruby-array-ext/test/array.rb b/mrbgems/mruby-array-ext/test/array.rb index d15ea2a64..ba1d4fd75 100644 --- a/mrbgems/mruby-array-ext/test/array.rb +++ b/mrbgems/mruby-array-ext/test/array.rb @@ -125,14 +125,31 @@ end assert("Array#fill") do a = [ "a", "b", "c", "d" ] assert_equal ["x", "x", "x", "x"], a.fill("x") - assert_equal ["x", "x", "x", "w"], a.fill("w", -1) + assert_equal ["x", "x", "x", "w"], a.fill("w", -1) assert_equal ["x", "x", "z", "z"], a.fill("z", 2, 2) assert_equal ["y", "y", "z", "z"], a.fill("y", 0..1) - assert_equal [0, 1, 4, 9], a.fill { |i| i*i } + assert_equal [0, 1, 4, 9], a.fill { |i| i*i } assert_equal [0, 1, 8, 27], a.fill(-2) { |i| i*i*i } assert_equal [0, 2, 3, 27], a.fill(1, 2) { |i| i+1 } assert_equal [1, 2, 3, 27], a.fill(0..1) { |i| i+1 } assert_raise(ArgumentError) { a.fill } + + assert_equal([0, 1, 2, 3, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, -2, 1)) + assert_equal([0, 1, 2, 3, -1, -1, -1], [0, 1, 2, 3, 4, 5].fill(-1, -2, 3)) + assert_equal([0, 1, 2, -1, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, 3..4)) + assert_equal([0, 1, 2, -1, 4, 5], [0, 1, 2, 3, 4, 5].fill(-1, 3...4)) + assert_equal([0, 1, -1, -1, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, 2..-2)) + assert_equal([0, 1, -1, -1, 4, 5], [0, 1, 2, 3, 4, 5].fill(-1, 2...-2)) + assert_equal([0, 1, 2, 13, 14, 5], [0, 1, 2, 3, 4, 5].fill(3..4){|i| i+10}) + assert_equal([0, 1, 2, 13, 4, 5], [0, 1, 2, 3, 4, 5].fill(3...4){|i| i+10}) + assert_equal([0, 1, 12, 13, 14, 5], [0, 1, 2, 3, 4, 5].fill(2..-2){|i| i+10}) + assert_equal([0, 1, 12, 13, 4, 5], [0, 1, 2, 3, 4, 5].fill(2...-2){|i| i+10}) + + assert_equal [1, 2, 3, 4, 'x', 'x'], [1, 2, 3, 4, 5, 6].fill('x', -2..-1) + assert_equal [1, 2, 3, 4, 'x', 6], [1, 2, 3, 4, 5, 6].fill('x', -2...-1) + assert_equal [1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6].fill('x', -2...-2) + assert_equal [1, 2, 3, 4, 'x', 6], [1, 2, 3, 4, 5, 6].fill('x', -2..-2) + assert_equal [1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6].fill('x', -2..0) end assert("Array#reverse_each") do @@ -206,3 +223,16 @@ assert("Array#insert") do b = ["a", "b", "c", "d"] assert_equal ["a", "b", "c", "d", nil, nil, 99], b.insert(6, 99) end + +assert("Array#delete_if") do + a = [1, 2, 3, 4, 5] + assert_equal [1, 2, 3, 4, 5], a.delete_if { false } + assert_equal [1, 2, 3, 4, 5], a + + a = [1, 2, 3, 4, 5] + assert_equal [], a.delete_if { true } + assert_equal [], a + + a = [ 1, 2, 3, 4, 5 ] + assert_equal [1, 2, 3], a.delete_if { |val| val > 3 } +end diff --git a/mrblib/range.rb b/mrblib/range.rb index d587cab45..1ec9ac508 100644 --- a/mrblib/range.rb +++ b/mrblib/range.rb @@ -13,11 +13,23 @@ class Range return to_enum :each unless block_given? val = self.first + last = self.last + + if val.kind_of?(Fixnum) && last.kind_of?(Fixnum) # fixnums are special + lim = last + lim += 1 unless exclude_end? + i = val + while i < lim + block.call(i) + i += 1 + end + return self + end + unless val.respond_to? :succ raise TypeError, "can't iterate" end - last = self.last return self if (val <=> last) > 0 while((val <=> last) < 0) diff --git a/src/codegen.c b/src/codegen.c index a36d609c6..d8fd52665 100644 --- a/src/codegen.c +++ b/src/codegen.c @@ -719,7 +719,7 @@ attrsym(codegen_scope *s, mrb_sym a) + 1 /* '=' */ + 1 /* '\0' */ ); - mrb_assert(len <= SIZE_MAX); + mrb_assert(len > 0); memcpy(name2, name, (size_t)len); name2[len] = '='; name2[len+1] = '\0'; diff --git a/src/debug.c b/src/debug.c index 0af0f48f7..ea1aa1ddc 100644 --- a/src/debug.c +++ b/src/debug.c @@ -95,7 +95,7 @@ mrb_debug_get_line(mrb_irep *irep, uint32_t pc) mrb_assert(f->lines.flat_map <= ret && ret < (f->lines.flat_map + f->line_entry_count)); /* check pc range */ mrb_assert(ret->start_pos <= pc && - pc < (((ret + 1 - f->lines.flat_map) < f->line_entry_count) + pc < (((uint32_t)(ret + 1 - f->lines.flat_map) < f->line_entry_count) ? (ret+1)->start_pos : irep->debug_info->pc_count)); return ret->line; diff --git a/src/dump.c b/src/dump.c index e3d1b779f..97e0fc3c9 100644 --- a/src/dump.c +++ b/src/dump.c @@ -89,8 +89,7 @@ get_pool_block_size(mrb_state *mrb, mrb_irep *irep) str = mrb_fixnum_to_str(mrb, irep->pool[pool_no], 10); { mrb_int len = RSTRING_LEN(str); - mrb_assert(len >= 0); - mrb_assert(len <= SIZE_MAX); + mrb_assert(len >= 0 && (size_t)len <= SIZE_MAX); size += (size_t)len; } break; @@ -99,8 +98,7 @@ get_pool_block_size(mrb_state *mrb, mrb_irep *irep) { int len; len = mrb_float_to_str(buf, mrb_float(irep->pool[pool_no])); - mrb_assert(len >= 0); - mrb_assert(len <= SIZE_MAX); + mrb_assert(len >= 0 && (size_t)len <= SIZE_MAX); size += (size_t)len; } break; @@ -108,8 +106,7 @@ get_pool_block_size(mrb_state *mrb, mrb_irep *irep) case MRB_TT_STRING: { mrb_int len = RSTRING_LEN(irep->pool[pool_no]); - mrb_assert(len >= 0); - mrb_assert(len <= SIZE_MAX); + mrb_assert(len >= 0 && (size_t)len <= SIZE_MAX); size += (size_t)len; } break; @@ -416,11 +413,11 @@ write_lineno_record_1(mrb_state *mrb, mrb_irep *irep, uint8_t* bin) diff = cur - bin; mrb_assert(diff >= 0); - mrb_assert(diff <= UINT32_MAX); + mrb_assert((uint32_t)diff <= UINT32_MAX); uint32_to_bin((uint32_t)diff, bin); /* record size */ - mrb_assert(diff <= SIZE_MAX); + mrb_assert((size_t)diff <= SIZE_MAX); return (size_t)diff; } @@ -596,11 +593,10 @@ write_debug_record_1(mrb_state *mrb, mrb_irep *irep, uint8_t *bin, mrb_sym const } ret = cur - bin; - mrb_assert(ret >= 0); - mrb_assert(ret <= UINT32_MAX); + mrb_assert(ret >= 0 && (uint32_t)ret <= UINT32_MAX); uint32_to_bin(ret, bin); - mrb_assert(ret <= SIZE_MAX); + mrb_assert(ret >= 0 && (size_t)ret <= SIZE_MAX); return (size_t)ret; } diff --git a/src/load.c b/src/load.c index d97776a16..776fec561 100644 --- a/src/load.c +++ b/src/load.c @@ -156,8 +156,7 @@ read_irep_record_1(mrb_state *mrb, const uint8_t *bin, size_t *len, mrb_bool all irep->reps = (mrb_irep**)mrb_malloc(mrb, sizeof(mrb_irep*)*irep->rlen); diff = src - bin; - mrb_assert(diff >= 0); - mrb_assert(diff <= SIZE_MAX); + mrb_assert(diff >= 0 && (size_t)diff <= SIZE_MAX); *len = (size_t)diff; return irep; @@ -334,8 +333,7 @@ read_debug_record(mrb_state *mrb, const uint8_t *start, mrb_irep* irep, size_t * } diff = bin - start; - mrb_assert(diff >= 0); - mrb_assert(diff <= SIZE_MAX); + mrb_assert(diff >= 0 && (size_t)diff <= SIZE_MAX); if (record_size != (size_t)diff) { return MRB_DUMP_GENERAL_FAILURE; @@ -351,8 +349,7 @@ read_debug_record(mrb_state *mrb, const uint8_t *start, mrb_irep* irep, size_t * } diff = bin - start; - mrb_assert(diff >= 0); - mrb_assert(diff <= SIZE_MAX); + mrb_assert(diff >=0 && (size_t)diff <= SIZE_MAX); *record_len = (size_t)diff; return MRB_DUMP_OK; @@ -394,8 +391,7 @@ read_section_debug(mrb_state *mrb, const uint8_t *start, mrb_irep *irep, mrb_boo bin += len; diff = bin - start; - mrb_assert(diff >= 0); - mrb_assert(diff <= UINT32_MAX); + mrb_assert(diff >= 0 && (size_t)diff <= UINT32_MAX); if ((uint32_t)diff != bin_to_uint32(header->section_size)) { result = MRB_DUMP_GENERAL_FAILURE; } diff --git a/src/string.c b/src/string.c index 1e3c7db3a..dd0889d92 100644 --- a/src/string.c +++ b/src/string.c @@ -32,11 +32,11 @@ }\ } while (0) #define RSTRING_EMBED_LEN(s) \ - (size_t)((RSTRING(s)->flags & MRB_STR_EMBED_LEN_MASK) >> MRB_STR_EMBED_LEN_SHIFT) + (mrb_int)((RSTRING(s)->flags & MRB_STR_EMBED_LEN_MASK) >> MRB_STR_EMBED_LEN_SHIFT) #define STR_EMBED_LEN(s)\ - (size_t)(((s)->flags & MRB_STR_EMBED_LEN_MASK) >> MRB_STR_EMBED_LEN_SHIFT) + (mrb_int)(((s)->flags & MRB_STR_EMBED_LEN_MASK) >> MRB_STR_EMBED_LEN_SHIFT) #define STR_PTR(s) ((STR_EMBED_P(s)) ? (s)->as.ary : (s)->as.heap.ptr) -#define STR_LEN(s) ((STR_EMBED_P(s)) ? STR_EMBED_LEN(s) : (size_t)(s)->as.heap.len) +#define STR_LEN(s) ((STR_EMBED_P(s)) ? STR_EMBED_LEN(s) : (s)->as.heap.len) const char mrb_digitmap[] = "0123456789abcdefghijklmnopqrstuvwxyz"; diff --git a/src/symbol.c b/src/symbol.c index fd0b116bd..de2cc2881 100644 --- a/src/symbol.c +++ b/src/symbol.c @@ -401,7 +401,8 @@ sym_inspect(mrb_state *mrb, mrb_value sym) sp = RSTRING_PTR(str); RSTRING_PTR(str)[0] = ':'; memcpy(sp+1, name, len); - if (!symname_p(name) || strlen(name) != len) { + mrb_assert(len > 0 && (size_t)len <= SIZE_MAX); + if (!symname_p(name) || strlen(name) != (size_t)len) { str = mrb_str_dump(mrb, str); sp = RSTRING_PTR(str); sp[0] = ':'; |
