From 09b11850a5df44d8e0ba5d031c57146e0c41547b Mon Sep 17 00:00:00 2001
From: "Yukihiro \"Matz\" Matsumoto" <matz@ruby-lang.org>
Date: Fri, 10 Jun 2016 10:12:48 +0900
Subject: mrb_gc_unregister() to remove one registration; close #3160

when multiple mrb_gc_register() were called for the same object
---
 src/gc.c | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

(limited to 'src')

diff --git a/src/gc.c b/src/gc.c
index e7948e4f5..86fa054ff 100644
--- a/src/gc.c
+++ b/src/gc.c
@@ -453,7 +453,7 @@ mrb_gc_unregister(mrb_state *mrb, mrb_value obj)
   mrb_sym root = mrb_intern_lit(mrb, GC_ROOT_NAME);
   mrb_value table = mrb_gv_get(mrb, root);
   struct RArray *a;
-  mrb_int i, j;
+  mrb_int i, len;
 
   if (mrb_nil_p(table)) return;
   if (mrb_type(table) != MRB_TT_ARRAY) {
@@ -462,12 +462,14 @@ mrb_gc_unregister(mrb_state *mrb, mrb_value obj)
   }
   a = mrb_ary_ptr(table);
   mrb_ary_modify(mrb, a);
-  for (i=j=0; i<a->len; i++) {
-    if (!mrb_obj_eq(mrb, a->ptr[i], obj)) {
-      a->ptr[j++] = a->ptr[i];
+  len = a->len-1;
+  for (i=0; i<len; i++) {
+    if (mrb_obj_eq(mrb, a->ptr[i], obj)) {
+      memmove(&a->ptr[i], &a->ptr[i+1], len-i);
+      break;
     }
   }
-  a->len = j;
+  a->len--;
 }
 
 MRB_API struct RBasic*
-- 
cgit v1.2.3