[GR-12617] Small fixes for setuptools.

PullRequest: graalpython/296

Author: timfel

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/BinasciiModuleBuiltins.java

@@ -40,6 +40,9 @@
  */
 package com.oracle.graal.python.builtins.modules;
 
+import static com.oracle.graal.python.builtins.PythonBuiltinClassType.ValueError;
+
+import java.io.UnsupportedEncodingException;
 import java.util.List;
 import java.util.zip.CRC32;
 
@@ -48,7 +51,9 @@
 import com.oracle.graal.python.builtins.PythonBuiltinClassType;
 import com.oracle.graal.python.builtins.PythonBuiltins;
 import com.oracle.graal.python.builtins.objects.PNone;
+import com.oracle.graal.python.builtins.objects.bytes.BytesNodes;
 import com.oracle.graal.python.builtins.objects.bytes.PBytes;
+import com.oracle.graal.python.builtins.objects.bytes.PIBytesLike;
 import com.oracle.graal.python.builtins.objects.common.SequenceStorageNodes;
 import com.oracle.graal.python.builtins.objects.module.PythonModule;
 import com.oracle.graal.python.builtins.objects.type.PythonClass;
@@ -88,9 +93,28 @@ public void initialize(PythonCore core) {
     @GenerateNodeFactory
     static abstract class A2bBase64Node extends PythonUnaryBuiltinNode {
         @Specialization
+        PBytes doString(String data) {
+            return factory().createBytes(b64decode(data));
+        }
+
+        @Specialization
+        PBytes doBytesLike(PIBytesLike data,
+                        @Cached("create()") BytesNodes.ToBytesNode toBytesNode) {
+            return factory().createBytes(b64decode(toBytesNode.execute(data)));
+        }
+
         @TruffleBoundary
-        PBytes a2b(String data) {
-            return factory().createBytes(Base64.decode(data));
+        private static byte[] b64decode(String data) {
+            return Base64.decode(data);
+        }
+
+        @TruffleBoundary
+        private byte[] b64decode(byte[] data) {
+            try {
+                return Base64.decode(new String(data, "ascii"));
+            } catch (UnsupportedEncodingException e) {
+                throw raise(ValueError);
+            }
         }
     }
 

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/SocketModuleBuiltins.java

@@ -40,6 +40,8 @@
  */
 package com.oracle.graal.python.builtins.modules;
 
+import java.net.InetAddress;
+import java.net.UnknownHostException;
 import java.util.List;
 
 import com.oracle.graal.python.builtins.Builtin;
@@ -51,6 +53,7 @@
 import com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode;
 import com.oracle.graal.python.nodes.function.PythonBuiltinNode;
 import com.oracle.graal.python.runtime.exception.PythonErrorType;
+import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
 import com.oracle.truffle.api.dsl.GenerateNodeFactory;
 import com.oracle.truffle.api.dsl.NodeFactory;
 import com.oracle.truffle.api.dsl.Specialization;
@@ -75,4 +78,18 @@ Object socket(PythonClass cls, int family, int type, int proto, @SuppressWarning
             }
         }
     }
+
+    @Builtin(name = "gethostname", fixedNumOfPositionalArgs = 0)
+    @GenerateNodeFactory
+    public abstract static class GetHostnameNode extends PythonBuiltinNode {
+        @Specialization
+        @TruffleBoundary
+        String doGeneric() {
+            try {
+                return InetAddress.getLocalHost().getHostName();
+            } catch (UnknownHostException e) {
+                throw raise(PythonBuiltinClassType.OSError);
+            }
+        }
+    }
 }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/SysModuleBuiltins.java

@@ -174,6 +174,7 @@ public void initialize(PythonCore core) {
                         2,                      // FLT_RADIX
                         1                       // FLT_ROUNDS
         }));
+        builtinConstants.put("maxunicode", Character.MAX_CODE_POINT);
 
         String os = getPythonOSName();
         builtinConstants.put("platform", os);