ZJIT: Support Regexp type

Author: st0012

zjit/bindgen/src/main.rs

@@ -188,6 +188,7 @@ fn main() {
         .allowlist_var("rb_cHash")
         .allowlist_var("rb_cSet")
         .allowlist_var("rb_cClass")
+        .allowlist_var("rb_cRegexp")
         .allowlist_var("rb_cISeq")
 
         // From include/ruby/internal/fl_type.h

zjit/src/cruby_bindings.inc.rs

@@ -6310,4 +6310,18 @@ mod opt_tests {
               Return v7
         "#]]);
     }
+
+    #[test]
+    fn test_regexp_type() {
+        eval("
+            def test = /a/
+            test
+        ");
+        assert_optimized_method_hir("test", expect![[r#"
+            fn test:
+            bb0(v0:BasicObject):
+              v2:RegexpExact[VALUE(0x1000)] = Const Value(VALUE(0x1000))
+              Return v2
+        "#]]);
+    }
 }

zjit/src/hir.rs

@@ -73,6 +73,7 @@ def base_type name
 base_type "Hash"
 base_type "Range"
 base_type "Set"
+base_type "Regexp"
 
 (integer, integer_exact) = base_type "Integer"
 # CRuby partitions Integer into immediate and non-immediate variants.

zjit/src/hir_type/gen_hir_type.rb

@@ -1,6 +1,6 @@
 #![allow(non_upper_case_globals)]
 use crate::cruby::{Qfalse, Qnil, Qtrue, VALUE, RUBY_T_ARRAY, RUBY_T_STRING, RUBY_T_HASH};
-use crate::cruby::{rb_cInteger, rb_cFloat, rb_cArray, rb_cHash, rb_cString, rb_cSymbol, rb_cObject, rb_cTrueClass, rb_cFalseClass, rb_cNilClass, rb_cRange, rb_cSet};
+use crate::cruby::{rb_cInteger, rb_cFloat, rb_cArray, rb_cHash, rb_cString, rb_cSymbol, rb_cObject, rb_cTrueClass, rb_cFalseClass, rb_cNilClass, rb_cRange, rb_cSet, rb_cRegexp};
 use crate::cruby::ClassRelationship;
 use crate::cruby::get_class_name;
 use crate::cruby::ruby_sym_to_rust_string;
@@ -197,6 +197,9 @@ impl Type {
         else if is_string_exact(val) {
             Type { bits: bits::StringExact, spec: Specialization::Object(val) }
         }
+        else if val.class_of() == unsafe { rb_cRegexp } {
+            Type { bits: bits::RegexpExact, spec: Specialization::Object(val) }
+        }
         else if val.class_of() == unsafe { rb_cSet } {
             Type { bits: bits::SetExact, spec: Specialization::Object(val) }
         }
@@ -292,6 +295,7 @@ impl Type {
         if class == unsafe { rb_cNilClass } { return true; }
         if class == unsafe { rb_cObject } { return true; }
         if class == unsafe { rb_cRange } { return true; }
+        if class == unsafe { rb_cRegexp } { return true; }
         if class == unsafe { rb_cString } { return true; }
         if class == unsafe { rb_cSymbol } { return true; }
         if class == unsafe { rb_cTrueClass } { return true; }