[GR-17457] Implement rb_imemo_tmpbuf allocation for ripper.

PullRequest: truffleruby/3319

Author: aardvark179

CHANGELOG.md

@@ -19,6 +19,7 @@ Compatibility:
 * Fix `Process.wait2` to return `nil` when the `WNOHANG` flag is given and the child process is still running (@bjfish).
 * Disable most `nokogiri` C extension patches when system libraries are not being used (#2693, @aardvark179).
 * Implement `rb_gc_mark_maybe` and `rb_global_variable` to ensure `VALUE` stay live in C extensions (@aardvark179).
+* Implement `rb_imemo_tmpbuf` allocation for `ripper` (@aardvark179).
 
 Performance:
 

lib/cext/ABI_version.txt

@@ -1 +1 @@
-2
+3

src/main/c/cext/alloc.c

@@ -95,3 +95,15 @@ void* rb_imemo_tmpbuf_set_ptr(VALUE imemo, void *ptr) {
   polyglot_invoke(RUBY_CEXT, "rb_imemo_tmpbuf_set_ptr", rb_tr_unwrap(imemo), ptr);
   return ptr;
 }
+
+rb_imemo_tmpbuf_t *rb_imemo_tmpbuf_parser_heap(void *buf, rb_imemo_tmpbuf_t *old_heap, size_t cnt) {
+  /* This differs from CRuby as this does not produce an object known to the
+     GC. This is not a problem for Ripper because we also mod that to free the
+     heap when freeing the parser structure, but it might be a problem if other
+     extensions use this function. */
+  rb_imemo_tmpbuf_t *imemo = malloc(sizeof(rb_imemo_tmpbuf_t));
+  imemo->ptr = buf;
+  imemo->next = old_heap;
+  imemo->cnt = cnt;
+  return imemo;
+}

src/main/c/ripper/ripper.c

@@ -20248,8 +20248,10 @@ parser_mark(void *ptr)
     rb_gc_mark(p->debug_buffer);
     rb_gc_mark(p->debug_output);
 #ifdef YYMALLOC
+#ifndef TRUFFLERUBY
     rb_gc_mark((VALUE)p->heap);
 #endif
+#endif
 }
 
 static void
@@ -20258,6 +20260,17 @@ parser_free(void *ptr)
     struct parser_params *p = (struct parser_params*)ptr;
     struct local_vars *local, *prev;
 
+#ifdef TRUFFLERUBY
+    rb_imemo_tmpbuf_t *heap = p->heap;
+
+    while (heap != NULL) {
+        if (heap->ptr != NULL) {
+            xfree(ptr);
+        }
+        heap = heap->next;
+    }
+#endif
+
     if (p->tokenbuf) {
         ruby_sized_xfree(p->tokenbuf, p->toksiz);
     }