Mark native arrays when the storage pointer has escaped.

Author: aardvark179

lib/truffle/truffle/cext.rb

@@ -1495,6 +1495,8 @@ def rb_data_typed_object_wrap(ruby_class, data, data_type, mark, free, size)
   end
 
   def run_marker(obj)
+    Primitive.array_mark_store(obj) if Primitive.array_store_native?(obj)
+
     mark = Primitive.object_hidden_var_get obj, DATA_MARKER
     unless Truffle::Interop.null?(mark)
       create_mark_list(obj)

lib/truffle/truffle/cext_structs.rb

@@ -295,6 +295,7 @@ def polyglot_pointer?
   end
 
   def polyglot_as_pointer
+    Primitive.cext_mark_object_on_call_exit(@array)
     Primitive.array_store_address(@array)
   end
 

src/main/java/org/truffleruby/core/array/ArrayNodes.java

@@ -2371,6 +2371,25 @@ protected boolean isStoreNative(RubyArray array,
                 @CachedLibrary("store") ArrayStoreLibrary stores) {
             return stores.isNative(store);
         }
+
+        @Specialization(guards = "!isRubyArray(array)")
+        protected boolean isStoreNativeNonArray(Object array) {
+            return false;
+        }
+    }
+
+    @Primitive(name = "array_mark_store")
+    @ImportStatic(ArrayGuards.class)
+    public abstract static class MarkNativeStoreNode extends PrimitiveArrayArgumentsNode {
+
+        @Specialization
+        protected Object markNativeStore(RubyArray array) {
+            Object store = array.getStore();
+            if (store instanceof NativeArrayStorage) {
+                ((NativeArrayStorage) store).preserveMembers();
+            }
+            return nil;
+        }
     }
 
     @Primitive(name = "array_flatten_helper", lowerFixnum = 2)