Merge branch 'test-fix-2419' into jruby-1_7

jruby · Jan 9, 2015 · 01f3945 · 01f3945
2 parents dbf58e6 + ff54579
commit 01f3945
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 7 deletions.
diff --git a/core/src/main/java/org/jruby/util/encoding/CharsetTranscoder.java b/core/src/main/java/org/jruby/util/encoding/CharsetTranscoder.java
@@ -235,7 +235,21 @@ public RubyCoderResult transcode(ThreadContext context, ByteList value, ByteList
 
         return transcode(context, value, dest, fromEncoding, false, true);
     }
-
+
+    public RubyCoderResult econvConvert(ThreadContext context, ByteList inBuffer, ByteList outBuffer) {
+        Encoding fromEncoding = this.inEncoding != null ? this.inEncoding : inBuffer.getEncoding();
+
+        primitiveConvert(context, inBuffer.shallowDup(), outBuffer, 0, -1, fromEncoding, false, actions.ecflags);
+
+        if (lastResult != null) {
+            createLastError();
+        } else {
+            outBuffer.append(finish(inBuffer.getEncoding()));
+        }
+
+        return lastResult;
+    }
+
     public ByteList transcode(ThreadContext context, ByteList value) {
         ByteList dest = new ByteList();
 

diff --git a/core/src/main/java/org/jruby/util/encoding/Transcoder.java b/core/src/main/java/org/jruby/util/encoding/Transcoder.java
@@ -149,12 +149,17 @@ public static ByteList strConvEncOpts(ThreadContext context, ByteList value, Enc
 
         Transcoder ec = EncodingUtils.econvOpenOpts(context, fromEncoding.getName(), toEncoding.getName(), ecflags, ecopts);
         if (ec == null) return value;
-
-        ByteList ret = ec.convert(context, value, false);
-
-        ret.setEncoding(toEncoding);
-
-        return ret;
+
+        ByteList newStr = new ByteList();
+        RubyCoderResult ret = ec.econvConvert(context, value, newStr);
+
+        if (ret == null || ret.stringResult.equals("finished")) {
+            newStr.setEncoding(toEncoding);
+            return newStr;
+        } else {
+            // error result, failover to original
+            return value;
+        }
     }
 
     // rb_str_conv_enc
@@ -187,6 +192,8 @@ public static ByteList transcode(ThreadContext context, ByteList value, Encoding
 
     // rb_econv_convert
     public abstract RubyCoderResult transcode(ThreadContext context, ByteList value, ByteList dest);
+
+    public abstract RubyCoderResult econvConvert(ThreadContext context, ByteList value, ByteList dest);
 
     public abstract ByteList transcode(ThreadContext context, ByteList value);