diff options
author | akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2011-01-12 10:27:48 +0000 |
---|---|---|
committer | akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2011-01-12 10:27:48 +0000 |
commit | e1680f8cc1c6ba464710bc164720e47e749b0ac8 (patch) | |
tree | 45bb1b6453be703b634f63a735b4722638683604 /enum.c | |
parent | ef0736604a327e30568ea97ca07637c7d144544e (diff) |
* enum.c (sort_by_i): reenter check more strictly.
(sort_by_cmp): ditto.
[ruby-dev:43003] reported by Usaku NAKAMURA.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@30521 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'enum.c')
-rw-r--r-- | enum.c | 18 |
1 files changed, 14 insertions, 4 deletions
@@ -782,14 +782,20 @@ sort_by_i(VALUE i, VALUE _data, int argc, VALUE *argv) { struct sort_by_data *data = (struct sort_by_data *)_data; VALUE ary = data->ary; + VALUE v; ENUM_WANT_SVALUE(); + v = rb_yield(i); + if (RBASIC(ary)->klass) { rb_raise(rb_eRuntimeError, "sort_by reentered"); } - - RARRAY_PTR(data->buf)[data->n*2] = rb_yield(i); + if (RARRAY_LEN(data->buf) != SORT_BY_BUFSIZE*2) { + rb_raise(rb_eRuntimeError, "sort_by reentered"); + } + + RARRAY_PTR(data->buf)[data->n*2] = v; RARRAY_PTR(data->buf)[data->n*2+1] = i; data->n++; if (data->n == SORT_BY_BUFSIZE) { @@ -802,13 +808,17 @@ sort_by_i(VALUE i, VALUE _data, int argc, VALUE *argv) static int sort_by_cmp(const void *ap, const void *bp, void *data) { - VALUE a = *(VALUE *)ap; - VALUE b = *(VALUE *)bp; + VALUE a; + VALUE b; VALUE ary = (VALUE)data; if (RBASIC(ary)->klass) { rb_raise(rb_eRuntimeError, "sort_by reentered"); } + + a = *(VALUE *)ap; + b = *(VALUE *)bp; + return rb_cmpint(rb_funcall(a, id_cmp, 1, b), a, b); } |