clear dst Hash on Hash#replace. [Bug #15358]

* hash.c (linear_copy): solve two issues on `Hash#replace`.
  (1) fix memory leak
      (1-1) don't allocate memory if destination already
            has a memory area.
      (1-2) free destination memory if src is NULL.
  (2) clear transient heap flag if src is NULL. [Bug #15358]


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@66091 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 21 insertions, 10 deletions

diff --git a/hash.c b/hash.c
index fc30ae9cbc..ad9f495192 100644
--- a/hash.c
+++ b/hash.c
@@ -493,6 +493,7 @@ static void
 hash_array_set(VALUE hash, struct li_table *li)
 {
     HASH_ASSERT(RHASH_ARRAY_P(hash));
+    HASH_ASSERT((RHASH_TRANSIENT_P(hash) && li == NULL) ? FALSE : TRUE);
     RHASH(hash)->as.li = li;
     hash_verify(hash);
 }
@@ -993,15 +994,17 @@ linear_copy(VALUE hash1, VALUE hash2)
 {
     li_table *old_tab = RHASH_ARRAY(hash2);
 
-    if (old_tab) {
-        li_table *new_tab;
-        new_tab = (li_table*) rb_transient_heap_alloc(hash1, sizeof(li_table));
-        if (new_tab != NULL) {
-            RHASH_SET_TRANSIENT_FLAG(hash1);
-        }
-        else {
-            RHASH_UNSET_TRANSIENT_FLAG(hash1);
-            new_tab = (li_table*)ruby_xmalloc(sizeof(li_table));
+    if (old_tab != NULL) {
+        li_table *new_tab = RHASH_ARRAY(hash1);
+        if (new_tab == NULL) {
+            new_tab = (li_table*) rb_transient_heap_alloc(hash1, sizeof(li_table));
+            if (new_tab != NULL) {
+                RHASH_SET_TRANSIENT_FLAG(hash1);
+            }
+            else {
+                RHASH_UNSET_TRANSIENT_FLAG(hash1);
+                new_tab = (li_table*)ruby_xmalloc(sizeof(li_table));
+            }
         }
         *new_tab = *old_tab;
         RHASH_ARRAY_BOUND_SET(hash1, RHASH_ARRAY_BOUND(hash2));
@@ -1014,7 +1017,15 @@ linear_copy(VALUE hash1, VALUE hash2)
     else {
         RHASH_ARRAY_BOUND_SET(hash1, RHASH_ARRAY_BOUND(hash2));
         RHASH_ARRAY_SIZE_SET(hash1, RHASH_ARRAY_SIZE(hash2));
-        RHASH_ARRAY_SET(hash1, old_tab);
+
+        if (RHASH_TRANSIENT_P(hash1)) {
+            RHASH_UNSET_TRANSIENT_FLAG(hash1);
+        }
+        else if (RHASH_ARRAY(hash1)) {
+            ruby_xfree(RHASH_ARRAY(hash1));
+        }
+
+        RHASH_ARRAY_SET(hash1, NULL);
 
         rb_gc_writebarrier_remember(hash1);
         return old_tab;