[PRISM] Use frozen flag on StringNode

ruby · Dec 14, 2023 · 8e1c148 · 8e1c148
1 parent a18819e
commit 8e1c148
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/prism_compile.c b/prism_compile.c
@@ -5189,7 +5189,12 @@ pm_compile_node(rb_iseq_t *iseq, const pm_node_t *node, LINK_ANCHOR *const ret,
         if (!popped) {
             pm_string_node_t *cast = (pm_string_node_t *) node;
             VALUE value = parse_string_encoded(node, &cast->unescaped, parser);
-            ADD_INSN1(ret, &dummy_line_node, putstring, value);
+            if (node->flags & PM_STRING_FLAGS_FROZEN) {
+                ADD_INSN1(ret, &dummy_line_node, putobject, rb_str_freeze(value));
+            }
+            else {
+                ADD_INSN1(ret, &dummy_line_node, putstring, value);
+            }
         }
         return;
       }

diff --git a/test/ruby/test_compile_prism.rb b/test/ruby/test_compile_prism.rb
@@ -607,6 +607,16 @@ def test_RegularExpressionNode
 
     def test_StringNode
       assert_prism_eval('"pit"')
+      assert_prism_eval('"a".frozen?')
+
+      frozen_source = <<-CODE
+      # frozen_string_literal: true
+      "a".frozen?
+      CODE
+      ruby_eval = RubyVM::InstructionSequence.compile(frozen_source).eval
+      prism_eval = RubyVM::InstructionSequence.compile_prism(frozen_source).eval
+
+      assert_equal ruby_eval, prism_eval
     end
 
     def test_SymbolNode