[Truffle] Add io_truncate for File.truncate.

jruby · May 6, 2015 · ae25618 · nirvdrum · May 6, 2015 · nirvdrum
1 parent 9466e52
commit ae25618
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/truffle/src/main/java/org/jruby/truffle/nodes/rubinius/IOPrimitiveNodes.java b/truffle/src/main/java/org/jruby/truffle/nodes/rubinius/IOPrimitiveNodes.java
@@ -14,6 +14,7 @@
 import com.oracle.truffle.api.frame.VirtualFrame;
 import com.oracle.truffle.api.source.SourceSection;
 import jnr.constants.platform.Fcntl;
+import org.jruby.RubyEncoding;
 import org.jruby.truffle.runtime.RubyContext;
 import org.jruby.truffle.runtime.control.RaiseException;
 import org.jruby.truffle.runtime.core.RubyBasicObject;
@@ -106,6 +107,22 @@ public int open(RubyString path, int mode, int permission) {
 
     }
 
+
+    @RubiniusPrimitive(name = "io_truncate", needsSelf = false)
+    public static abstract class IOTruncatePrimitiveNode extends RubiniusPrimitiveNode {
+
+        public IOTruncatePrimitiveNode(RubyContext context, SourceSection sourceSection) {
+            super(context, sourceSection);
+        }
+
+        @Specialization
+        public int truncate(RubyString path, int offset) {
+            final String pathString = RubyEncoding.decodeUTF8(path.getByteList().getUnsafeBytes(), path.getByteList().getBegin(), path.getByteList().getRealSize());
+            return posix().truncate(pathString, offset);
+        }
+
+    }
+
     @RubiniusPrimitive(name = "io_fnmatch", needsSelf = false)
     public static abstract class IOFNMatchPrimitiveNode extends RubiniusPrimitiveNode {