Raise EncodingError at parse time for a Hash literal with invalid Symbol key

Author: andrykonchin

CHANGELOG.md

@@ -70,6 +70,7 @@ Compatibility:
 * Fix `IO#write` and support multiple arguments with different encodings (#2829, @andrykonchin).
 * Fix `Array` methods `reject`, `reject!`, `inject`, `map`, `select`, `each_index` and handle a case when array is modified by a passed block like CRuby does (#2822, andrykonchin, @eregon).
 * Fix `EncodingError` exception message when Symbol has invalid encoding (#2850, @andrykonchin).
+* Raise `EncodingError` at parse time when Hash literal contains a Symbol key with invalid encoding (#2848, @andrykonchin).
 
 Performance:
 

spec/ruby/language/hash_spec.rb

@@ -177,6 +177,22 @@ def h.to_hash; {:b => 2, :c => 3}; end
     utf8_hash.keys.first.should == usascii_hash.keys.first
     usascii_hash.keys.first.encoding.should == Encoding::US_ASCII
   end
+
+  it "raises an EncodingError at parse time when Symbol key with invalid bytes" do
+    ScratchPad.record []
+    -> {
+      eval 'ScratchPad << 1; {:"\xC3" => 1}'
+    }.should raise_error(EncodingError, 'invalid symbol in encoding UTF-8 :"\xC3"')
+    ScratchPad.recorded.should == []
+  end
+
+  it "raises an EncodingError at parse time when Symbol key with invalid bytes and 'key: value' syntax used" do
+    ScratchPad.record []
+    -> {
+      eval 'ScratchPad << 1; {"\xC3": 1}'
+    }.should raise_error(EncodingError, 'invalid symbol in encoding UTF-8 :"\xC3"')
+    ScratchPad.recorded.should == []
+  end
 end
 
 describe "The ** operator" do

src/main/java/org/truffleruby/parser/BodyTranslator.java

@@ -1307,13 +1307,38 @@ public RubyNode visitDStrNode(DStrParseNode node) {
     public RubyNode visitDSymbolNode(DSymbolParseNode node) {
         SourceIndexLength sourceSection = node.getPosition();
 
+        // A special case for a symbol in Hash literal {"key": value}
+        if (node.size() == 1 && node.getLast() instanceof StrParseNode) {
+            final StrParseNode strParseNode = (StrParseNode) node.getLast();
+            final SymbolParseNode symbolParseNode = asSymbol(sourceSection, strParseNode);
+            return visitSymbolNode(symbolParseNode);
+        }
+
         final RubyNode stringNode = translateInterpolatedString(sourceSection, node.getEncoding(), node.children());
 
         final RubyNode ret = StringToSymbolNodeGen.create(stringNode);
         ret.unsafeSetSourceSection(sourceSection);
         return addNewlineIfNeeded(node, ret);
     }
 
+    /** Same as {@link ParserSupport#asSymbol(SourceIndexLength position, ParseNode value)} but without needing a
+     * ParserSupport instance and only for StrParseNode argument. */
+    private SymbolParseNode asSymbol(SourceIndexLength position, StrParseNode value) {
+        final SymbolParseNode symbolParseNode = new SymbolParseNode(position, value.getValue(),
+                value.encoding);
+
+        if (!symbolParseNode.getTString().isValidUncached(symbolParseNode.getRubyEncoding().tencoding)) {
+            final RubyContext context = RubyLanguage.getCurrentContext();
+            throw new RaiseException(
+                    RubyLanguage.getCurrentContext(),
+                    context.getCoreExceptions().encodingError(
+                            symbolParseNode.getTString(),
+                            symbolParseNode.getRubyEncoding(),
+                            null));
+        }
+        return symbolParseNode;
+    }
+
     private RubyNode translateInterpolatedString(SourceIndexLength sourceSection,
             Encoding encoding, ParseNode[] childNodes) {
         final ToSNode[] children = new ToSNode[childNodes.length];