1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

* array.c (rb_ary_unshift_m): a bug in lfree shift length

calculation.


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@11179 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
matz 2006-10-16 14:35:51 +00:00
parent eb2661510d
commit 58c0100a77
3 changed files with 50 additions and 38 deletions

View file

@ -1,3 +1,8 @@
Mon Oct 16 23:33:18 2006 Yukihiro Matsumoto <matz@ruby-lang.org>
* array.c (rb_ary_unshift_m): a bug in lfree shift length
calculation.
Mon Oct 16 08:30:43 2006 Nobuyoshi Nakada <nobu@ruby-lang.org> Mon Oct 16 08:30:43 2006 Nobuyoshi Nakada <nobu@ruby-lang.org>
* mkconfig.rb: *OBJS are not needed for extension libraries. * mkconfig.rb: *OBJS are not needed for extension libraries.

81
array.c
View file

@ -248,7 +248,7 @@ rb_ary_free(VALUE ary)
{ {
if (!ARY_SHARED_P(ary) && !ARY_EMBED_P(ary)) { if (!ARY_SHARED_P(ary) && !ARY_EMBED_P(ary)) {
if (ARY_LFREE_P(ary)) { if (ARY_LFREE_P(ary)) {
xfree(RARRAY(ary)->as.heap.ptr - RARRAY(ary)->as.heap.ptr[-1]); xfree(RARRAY(ary)->as.heap.ptr - LFREE_SIZE(ary));
} }
else { else {
xfree(RARRAY(ary)->as.heap.ptr); xfree(RARRAY(ary)->as.heap.ptr);
@ -569,9 +569,10 @@ rb_ary_pop(VALUE ary)
long n; long n;
rb_ary_modify_check(ary); rb_ary_modify_check(ary);
if (RARRAY_LEN(ary) == 0) return Qnil; if (RARRAY_LEN(ary) == 0) return Qnil;
if (!FL_TEST(ary, ELTS_SHARED) && if (!ARY_SHARED_P(ary) &&
RARRAY_LEN(ary) * 3 < ARY_CAPA(ary) && RARRAY_LEN(ary) * 3 < ARY_CAPA(ary) &&
ARY_CAPA(ary) > ARY_DEFAULT_SIZE) { ARY_CAPA(ary) > ARY_DEFAULT_SIZE)
{
RESIZE_CAPA(ary, RARRAY_LEN(ary) * 2); RESIZE_CAPA(ary, RARRAY_LEN(ary) * 2);
} }
n = RARRAY_LEN(ary)-1; n = RARRAY_LEN(ary)-1;
@ -622,7 +623,7 @@ rb_ary_shift(VALUE ary)
} }
if (!ARY_SHARED_P(ary)) { if (!ARY_SHARED_P(ary)) {
if (ARY_LFREE_P(ary)) { if (ARY_LFREE_P(ary)) {
RARRAY(ary)->as.heap.ptr[0] = RARRAY(ary)->as.heap.ptr[-1]+1; RARRAY(ary)->as.heap.ptr[0] = LFREE_SIZE(ary)+1;
} }
else { else {
FL_SET(ary, ARY_LFREE); FL_SET(ary, ARY_LFREE);
@ -664,10 +665,6 @@ rb_ary_shift_m(int argc, VALUE *argv, VALUE ary)
} }
rb_ary_modify_check(ary); rb_ary_modify_check(ary);
if (ARY_EMBED_P(ary)) {
}
result = ary_shared_first(argc, argv, ary, Qfalse); result = ary_shared_first(argc, argv, ary, Qfalse);
n = RARRAY_LEN(result); n = RARRAY_LEN(result);
if (ARY_SHARED_P(ary)) { if (ARY_SHARED_P(ary)) {
@ -697,43 +694,54 @@ rb_ary_shift_m(int argc, VALUE *argv, VALUE ary)
static VALUE static VALUE
rb_ary_unshift_m(int argc, VALUE *argv, VALUE ary) rb_ary_unshift_m(int argc, VALUE *argv, VALUE ary)
{ {
long lfree = ARY_LFREE_P(ary) ? RARRAY_PTR(ary)[-1] : 0; long lfree = ARY_LFREE_P(ary) ? LFREE_SIZE(ary) : 0;
long free2 = lfree;
rb_ary_modify_check(ary); rb_ary_modify_check(ary);
if (argc == 0) return ary; if (argc == 0) return ary;
if (lfree < argc) { if (lfree < argc) {
int shared = ARY_SHARED_P(ary); int shared = ARY_SHARED_P(ary);
long len = RARRAY_LEN(ary); long len = RARRAY_LEN(ary);
long rfree = shared ? 0 : (ARY_CAPA(ary)-len)/2; long free = shared ? RARRAY_LEN(ary) : ARY_CAPA(ary);
long free2 = argc+(RARRAY_LEN(ary)+argc)/2;
VALUE *ptr; VALUE *ptr;
if (free2 < ARY_DEFAULT_SIZE) {
free2 = ARY_DEFAULT_SIZE; if (!ARY_EMBED_P(ary) && free > len + argc) {
} free += lfree;
ptr = ALLOC_N(VALUE,len+free2)+(free2-rfree); free2 = (free - len - argc) / 2 + argc;
MEMCPY(ptr, RARRAY_PTR(ary), VALUE, len); ptr = RARRAY(ary)->as.heap.ptr-lfree+free2;
if (shared) { MEMMOVE(ptr, RARRAY(ary)->as.heap.ptr, VALUE,
FL_UNSET(ary, ELTS_SHARED); RARRAY(ary)->as.heap.len);
}
else if (ARY_EMBED_P(ary)) {
ARY_SET_NOEMBED(ary);
} }
else { else {
free(RARRAY_PTR(ary)-lfree); free = (len+argc) * 1.5;
} if (free - len < ARY_DEFAULT_SIZE) {
RARRAY(ary)->as.heap.ptr = ptr; free += ARY_DEFAULT_SIZE;
RARRAY(ary)->as.heap.len = len; }
RARRAY(ary)->as.heap.aux.capa = len+rfree; free2 = (free - len - argc)/2 + argc;
lfree = free2-rfree; ptr = ALLOC_N(VALUE,free)+free2;
FL_SET(ary, ARY_LFREE); MEMCPY(ptr, RARRAY_PTR(ary), VALUE, len);
if (shared) {
FL_UNSET(ary, ELTS_SHARED);
}
else if (ARY_EMBED_P(ary)) {
ARY_SET_NOEMBED(ary);
}
else {
xfree(RARRAY(ary)->as.heap.ptr-lfree);
}
}
RARRAY(ary)->as.heap.ptr = ptr;
RARRAY(ary)->as.heap.len = len;
RARRAY(ary)->as.heap.aux.capa = free-free2;
} }
RARRAY(ary)->as.heap.ptr -= argc; RARRAY(ary)->as.heap.ptr -= argc;
RARRAY(ary)->as.heap.len += argc; RARRAY(ary)->as.heap.len += argc;
RARRAY(ary)->as.heap.aux.capa += argc; RARRAY(ary)->as.heap.aux.capa += argc;
lfree -= argc; free2 -= argc;
if (lfree > 0) { if (free2 > 0) {
RARRAY(ary)->as.heap.ptr[-1] = lfree; RARRAY(ary)->as.heap.ptr[-1] = free2;
FL_SET(ary, ARY_LFREE);
} else { } else {
FL_UNSET(ary, ARY_LFREE); FL_UNSET(ary, ARY_LFREE);
} }
@ -1878,10 +1886,9 @@ rb_ary_delete_at(VALUE ary, long pos)
rb_ary_modify(ary); rb_ary_modify(ary);
del = RARRAY_PTR(ary)[pos]; del = RARRAY_PTR(ary)[pos];
for (i = pos + 1; i < len; i++, pos++) { MEMMOVE(RARRAY_PTR(ary)+pos, RARRAY_PTR(ary)+pos+1, VALUE,
RARRAY_PTR(ary)[pos] = RARRAY_PTR(ary)[i]; RARRAY_LEN(ary)-pos-1);
} ARY_SET_LEN(ary, RARRAY_LEN(ary)-1);
ARY_SET_LEN(ary, pos);
return del; return del;
} }

View file

@ -91,7 +91,7 @@ module Test
end end
find_test_cases(already_gathered).each{|t| add_suite(suites, t.suite)} find_test_cases(already_gathered).each{|t| add_suite(suites, t.suite)}
ensure ensure
$:.delete_at($:.rindex(dir)) if(dir) $:.delete_at($:.index(dir)) if dir
end end
def realdir(path) def realdir(path)