[Truffle] Add POSIX and Pointer nodes for File.readlink

jruby · May 7, 2015 · 5b296eb · nirvdrum · May 7, 2015 · bjfish
1 parent 440a90f
commit 5b296eb
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 6 deletions.
diff --git a/spec/truffle/tags/core/file/readlink_tags.txt b/spec/truffle/tags/core/file/readlink_tags.txt
diff --git a/truffle/src/main/java/org/jruby/truffle/nodes/rubinius/PointerPrimitiveNodes.java b/truffle/src/main/java/org/jruby/truffle/nodes/rubinius/PointerPrimitiveNodes.java
@@ -18,7 +18,9 @@
 import org.jruby.truffle.runtime.RubyContext;
 import org.jruby.truffle.runtime.core.RubyBasicObject;
 import org.jruby.truffle.runtime.core.RubyClass;
+import org.jruby.truffle.runtime.core.RubyString;
 import org.jruby.util.unsafe.UnsafeHolder;
+import sun.misc.Unsafe;
 
 public abstract class PointerPrimitiveNodes {
 
@@ -172,6 +174,25 @@ protected boolean isSigned(boolean signed) {
 
     }
 
+    @RubiniusPrimitive(name = "pointer_read_string")
+    public static abstract class PointerReadStringPrimitiveNode extends ReadAddressPrimitiveNode {
+
+        public PointerReadStringPrimitiveNode(RubyContext context, SourceSection sourceSection) {
+            super(context, sourceSection);
+        }
+
+        @Specialization
+        public RubyString readString(RubyBasicObject pointer, int length) {
+            final long address = getAddress(pointer);
+            final byte[] bytes = new byte[length];
+            for (int n = 0; n < bytes.length; n++) {
+                bytes[n] = UnsafeHolder.U.getByte(address + n * Unsafe.ARRAY_BYTE_INDEX_SCALE);
+            }
+            return getContext().makeString(bytes);
+        }
+
+    }
+
     @RubiniusPrimitive(name = "pointer_set_autorelease")
     public static abstract class PointerSetAutoreleasePrimitiveNode extends ReadAddressPrimitiveNode {
 

diff --git a/truffle/src/main/java/org/jruby/truffle/nodes/rubinius/PosixNodes.java b/truffle/src/main/java/org/jruby/truffle/nodes/rubinius/PosixNodes.java
@@ -9,6 +9,7 @@
  */
 package org.jruby.truffle.nodes.rubinius;
 
+import com.oracle.truffle.api.CompilerDirectives;
 import com.oracle.truffle.api.dsl.Specialization;
 import com.oracle.truffle.api.source.SourceSection;
 import jnr.constants.platform.Fcntl;
@@ -18,6 +19,7 @@
 import org.jruby.truffle.nodes.core.CoreMethod;
 import org.jruby.truffle.nodes.core.CoreMethodArrayArgumentsNode;
 import org.jruby.truffle.runtime.RubyContext;
+import org.jruby.truffle.runtime.control.RaiseException;
 import org.jruby.truffle.runtime.core.RubyBasicObject;
 import org.jruby.truffle.runtime.core.RubyNilClass;
 import org.jruby.truffle.runtime.core.RubyString;
@@ -201,6 +203,31 @@ public RubyBasicObject memset(RubyBasicObject pointer, int c, long length) {
 
     }
 
+    @CoreMethod(names = "readlink", isModuleFunction = true, required = 3)
+    public abstract static class ReadlinkNode extends PointerPrimitiveNodes.ReadAddressPrimitiveNode {
+
+        public ReadlinkNode(RubyContext context, SourceSection sourceSection) {
+            super(context, sourceSection);
+        }
+
+        @Specialization
+        public int readlink(RubyString path, RubyBasicObject pointer, int bufsize) {
+            final String pathString = RubyEncoding.decodeUTF8(path.getByteList().getUnsafeBytes(), path.getByteList().getBegin(), path.getByteList().getRealSize());
+            final long address = getAddress(pointer);
+            final byte[] buffer = new byte[bufsize];
+            final int result = posix().readlink(pathString, buffer, bufsize);
+            if (result == -1) {
+                CompilerDirectives.transferToInterpreter();
+                throw new RaiseException(getContext().getCoreLibrary().errnoError(posix().errno(), this));
+            }
+            for (int n = 0; n < buffer.length; n++) {
+                UnsafeHolder.U.putInt(address + n * Unsafe.ARRAY_BYTE_INDEX_SCALE, buffer[n]);
+            }
+            return result;
+        }
+
+    }
+
     @CoreMethod(names = "link", isModuleFunction = true, required = 2)
     public abstract static class LinkNode extends PointerPrimitiveNodes.ReadAddressPrimitiveNode {