Make WeakKeyMap safe for compaction during allocation

During allocation, the table may not have been allocated yet which would
crash in the st_foreach.
This commit is contained in:
Peter Zhu 2023-12-08 15:18:07 -05:00
Родитель 33cf8f640b
Коммит 185b7e92a3
2 изменённых файлов: 7 добавлений и 1 удалений

Просмотреть файл

@ -123,6 +123,10 @@ class TestWeakKeyMap < Test::Unit::TestCase
end;
end
def test_gc_compact_stress
EnvUtil.under_gc_compact_stress { ObjectSpace::WeakKeyMap.new }
end
private
def assert_weak_include(m, k, n = 100)

Просмотреть файл

@ -609,7 +609,9 @@ wkmap_compact(void *ptr)
{
struct weakkeymap *w = ptr;
st_foreach_with_replace(w->table, wkmap_compact_table_i, wkmap_compact_table_replace, (st_data_t)0);
if (w->table) {
st_foreach_with_replace(w->table, wkmap_compact_table_i, wkmap_compact_table_replace, (st_data_t)0);
}
}
static const rb_data_type_t weakkeymap_type = {