From c6f84e918943a0bf8db6fee556fc53180d257510 Mon Sep 17 00:00:00 2001
From: Peter Zhu <peter@peterzhu.ca>
Date: Wed, 1 Feb 2023 09:08:57 -0500
Subject: [PATCH] [Bug #19398] Memory leak in WeakMap

There's a memory leak in ObjectSpace::WeakMap due to not freeing
the `struct weakmap`. It can be seen in the following script:

```
100.times do
  10000.times do
    ObjectSpace::WeakMap.new
  end

  # Output the Resident Set Size (memory usage, in KB) of the current Ruby process
  puts `ps -o rss= -p #{$$}`
end
```
---
 gc.c                      | 1 +
 test/ruby/test_weakmap.rb | 9 +++++++++
 2 files changed, 10 insertions(+)

diff --git a/gc.c b/gc.c
index 269d46daef..3baed251f4 100644
--- a/gc.c
+++ b/gc.c
@@ -12870,6 +12870,7 @@ wmap_free(void *ptr)
     st_foreach(w->obj2wmap, wmap_free_map, 0);
     st_free_table(w->obj2wmap);
     st_free_table(w->wmap2obj);
+    xfree(w);
 }
 
 static int
diff --git a/test/ruby/test_weakmap.rb b/test/ruby/test_weakmap.rb
index 4c91661f86..86550cc221 100644
--- a/test/ruby/test_weakmap.rb
+++ b/test/ruby/test_weakmap.rb
@@ -167,4 +167,13 @@ class TestWeakMap < Test::Unit::TestCase
     assert_nothing_raised(FrozenError) {@wm[o] = 'foo'}
     assert_nothing_raised(FrozenError) {@wm['foo'] = o}
   end
+
+  def test_no_memory_leak
+    assert_no_memory_leak([], '', "#{<<~"begin;"}\n#{<<~'end;'}", "[Bug #19398]", rss: true, limit: 1.5, timeout: 60)
+    begin;
+      1_000_000.times do
+        ObjectSpace::WeakMap.new
+      end
+    end;
+  end
 end