call pin function on AST ctx

Author: d-netto

src/ast.c

@@ -139,6 +139,17 @@ typedef struct _jl_ast_context_t {
     arraylist_t pinned_objects;
 } jl_ast_context_t;
 
+// FIXME: Ugly hack to get a pointer to the pinned objects
+arraylist_t *extract_pinned_objects_from_ast_ctx(void *ctx)
+{
+    // This is used to extract pinned objects from the context
+    // for the purpose of pinning them in MMTk.
+    if (ctx == NULL)
+        return NULL;
+    jl_ast_context_t *jl_ctx = (jl_ast_context_t*)ctx;
+    return &jl_ctx->pinned_objects;
+}
+
 static jl_ast_context_t jl_ast_main_ctx;
 
 #ifdef __clang_gcanalyzer__

src/gc-mmtk.c

@@ -263,6 +263,17 @@ void gc_pin_objects_from_compiler_frontend(arraylist_t *objects_pinned_by_call)
             arraylist_push(objects_pinned_by_call, obj);
         }
     }
+    for (size_t i = 0; i < jl_ast_ctx_used.len; i++) {
+        void *ctx = jl_ast_ctx_used.items[i];
+        arraylist_t *pinned_objects = extract_pinned_objects_from_ast_ctx(ctx);
+        for (size_t j = 0; j < pinned_objects->len; j++) {
+            void *obj = pinned_objects->items[j];
+            unsigned char got_pinned = mmtk_pin_object(obj);
+            if (got_pinned) {
+                arraylist_push(objects_pinned_by_call, obj);
+            }
+        }
+    }
 }
 
 void gc_unpin_objects_from_compiler_frontend(arraylist_t *objects_pinned_by_call)

src/julia.h

@@ -2439,6 +2439,7 @@ JL_DLLEXPORT void jl_register_newmeth_tracer(void (*callback)(jl_method_t *trace
 
 // AST access
 JL_DLLEXPORT jl_value_t *jl_copy_ast(jl_value_t *expr JL_MAYBE_UNROOTED);
+arraylist_t *extract_pinned_objects_from_ast_ctx(void *ctx);
 extern arraylist_t jl_ast_ctx_used;
 
 // IR representation