[GR-20296] Backport to 19.3.1: Implement rb_gc_register_mark_object and rb_enc_str_asciionly_p.

PullRequest: truffleruby/1227

Author: ezzarghili

CHANGELOG.md

@@ -1,3 +1,9 @@
+# 19.3.1
+
+Compatibility:
+
+* Implemented `rb_gc_register_mark_object` and `rb_enc_str_asciionly_p` (#1856, @chrisseaton).
+
 # 19.3.0
 
 New features:

lib/truffle/truffle/cext.rb

@@ -1418,6 +1418,12 @@ def rb_gc
     GC.start
   end
 
+  GC_ROOTS = []
+
+  def rb_gc_register_mark_object(obj)
+    GC_ROOTS.push obj
+  end
+
   def rb_nativethread_self
     Thread.current
   end

spec/ruby/optional/capi/encoding_spec.rb

@@ -479,4 +479,14 @@
       length.should == 4
     end
   end
+
+  describe "rb_enc_str_asciionly_p" do
+    it "returns true for an ASCII string" do
+      @s.rb_enc_str_asciionly_p("hello").should be_true
+    end
+
+    it "returns false for a non-ASCII string" do
+      @s.rb_enc_str_asciionly_p("hüllo").should be_false
+    end
+  end
 end

spec/ruby/optional/capi/ext/encoding_spec.c

@@ -196,6 +196,14 @@ static VALUE encoding_spec_rb_enc_codepoint_len(VALUE self, VALUE str) {
   return rb_ary_new3(2, LONG2NUM(codepoint), LONG2NUM(len));
 }
 
+static VALUE encoding_spec_rb_enc_str_asciionly_p(VALUE self, VALUE str) {
+  if (rb_enc_str_asciionly_p(str)) {
+    return Qtrue;
+  } else {
+    return Qfalse;
+  }
+}
+
 void Init_encoding_spec(void) {
   VALUE cls = rb_define_class("CApiEncodingSpecs", rb_cObject);
   rb_define_method(cls, "ENC_CODERANGE_ASCIIONLY",
@@ -242,6 +250,7 @@ void Init_encoding_spec(void) {
   rb_define_method(cls, "rb_to_encoding_index", encoding_spec_rb_to_encoding_index, 1);
   rb_define_method(cls, "rb_enc_nth", encoding_spec_rb_enc_nth, 2);
   rb_define_method(cls, "rb_enc_codepoint_len", encoding_spec_rb_enc_codepoint_len, 1);
+  rb_define_method(cls, "rb_enc_str_asciionly_p", encoding_spec_rb_enc_str_asciionly_p, 1);
 }
 
 #ifdef __cplusplus

spec/ruby/optional/capi/ext/gc_spec.c

@@ -30,6 +30,11 @@ static VALUE gc_spec_rb_gc() {
 }
 
 
+static VALUE gc_spec_rb_gc_register_mark_object(VALUE self, VALUE obj) {
+  rb_gc_register_mark_object(obj);
+  return Qnil;
+}
+
 void Init_gc_spec(void) {
   VALUE cls = rb_define_class("CApiGCSpecs", rb_cObject);
   registered_tagged_value    = INT2NUM(10);
@@ -43,6 +48,7 @@ void Init_gc_spec(void) {
   rb_define_method(cls, "rb_gc_enable", gc_spec_rb_gc_enable, 0);
   rb_define_method(cls, "rb_gc_disable", gc_spec_rb_gc_disable, 0);
   rb_define_method(cls, "rb_gc", gc_spec_rb_gc, 0);
+  rb_define_method(cls, "rb_gc_register_mark_object", gc_spec_rb_gc_register_mark_object, 1);
 }
 
 #ifdef __cplusplus

spec/ruby/optional/capi/gc_spec.rb

@@ -51,4 +51,9 @@
 
   end
 
+  describe "rb_gc_register_mark_object" do
+    it "can be called with an object" do
+      @f.rb_gc_register_mark_object(Object.new).should be_nil
+    end
+  end
 end

src/main/c/cext/ruby.c

@@ -3245,7 +3245,7 @@ long rb_str_coderange_scan_restartable(const char *s, const char *e, rb_encoding
 }
 
 int rb_enc_str_asciionly_p(VALUE str) {
-  rb_tr_error("rb_enc_str_asciionly_p not implemented");
+  return polyglot_as_boolean(RUBY_INVOKE_NO_WRAP(str, "ascii_only?"));
 }
 
 int rb_enc_unicode_p(rb_encoding *enc) {
@@ -4711,7 +4711,7 @@ void rb_define_virtual_variable(const char *name, VALUE (*getter)(ANYARGS), void
 }
 
 void rb_gc_register_mark_object(VALUE obj) {
-  rb_tr_error("rb_gc_register_mark_object not implemented");
+  RUBY_CEXT_INVOKE_NO_WRAP("rb_gc_register_mark_object", obj);
 }
 
 ID rb_check_id(volatile VALUE *namep) {