diff options
author | Peter Zhu <[email protected]> | 2024-01-08 11:32:48 -0500 |
---|---|---|
committer | Peter Zhu <[email protected]> | 2024-01-10 11:20:26 -0500 |
commit | 82b57d7bfeefd717c10f7a5a3484aca6b3e708a3 (patch) | |
tree | 29dd965c9e8347f2de3656e4543bc916127dd282 | |
parent | f165fa09e7bc36f90d3862e537f071f0e1339428 (diff) |
Fix memory leak when duplicating too complex object
[Bug #20162]
Creating a ST table then calling st_replace leaks memory because the
st_replace overwrites the ST table without freeing any of the existing
memory. This commit changes it to use st_copy instead.
For example:
RubyVM::Shape.exhaust_shapes
o = Object.new
o.instance_variable_set(:@a, 0)
10.times do
100_000.times { o.dup }
puts `ps -o rss= -p #{$$}`
end
Before:
23264
33600
42672
52160
61600
71728
81056
90528
100560
109840
After:
14752
14816
15584
15584
15664
15664
15664
15664
15664
15664
-rw-r--r-- | object.c | 3 | ||||
-rw-r--r-- | test/ruby/test_shapes.rb | 13 |
2 files changed, 14 insertions, 2 deletions
@@ -302,8 +302,7 @@ rb_obj_copy_ivar(VALUE dest, VALUE obj) if (rb_shape_obj_too_complex(obj)) { // obj is TOO_COMPLEX so we can copy its iv_hash - st_table * table = rb_st_init_numtable_with_size(rb_st_table_size(ROBJECT_IV_HASH(obj))); - st_replace(table, ROBJECT_IV_HASH(obj)); + st_table *table = st_copy(ROBJECT_IV_HASH(obj)); rb_obj_convert_to_too_complex(dest, table); return; diff --git a/test/ruby/test_shapes.rb b/test/ruby/test_shapes.rb index 885b762eb9..ee99fbad6d 100644 --- a/test/ruby/test_shapes.rb +++ b/test/ruby/test_shapes.rb @@ -942,6 +942,19 @@ class TestShapes < Test::Unit::TestCase assert_shape_equal(RubyVM::Shape.of(obj), RubyVM::Shape.of(obj2)) end + def test_duplicating_too_complex_objects_memory_leak + assert_no_memory_leak([], "#{<<~'begin;'}", "#{<<~'end;'}", "[Bug #20162]", rss: true) + RubyVM::Shape.exhaust_shapes + + o = Object.new + o.instance_variable_set(:@a, 0) + begin; + 1_000_000.times do + o.dup + end + end; + end + def test_freezing_and_duplicating_object obj = Object.new.freeze obj2 = obj.dup |